package com.margelo.nitro.co.zyke.dfu

import com.facebook.react.TurboReactPackage
import com.facebook.react.bridge.NativeModule
import com.facebook.react.bridge.ReactApplicationContext
import com.facebook.react.module.model.ReactModuleInfoProvider

class NitroDfuPackage : TurboReactPackage() {
    private var contextSet = false

    override fun getModule(name: String, reactContext: ReactApplicationContext): NativeModule? {
        if (!contextSet) {
            NativeDfuManager.setContext(reactContext)
            contextSet = true
        }
        return null
    }

    override fun getReactModuleInfoProvider(): ReactModuleInfoProvider =
        ReactModuleInfoProvider { HashMap() }

    override fun createNativeModules(reactContext: ReactApplicationContext): MutableList<NativeModule> {
        NativeDfuManager.setContext(reactContext)
        return super.createNativeModules(reactContext).toMutableList()
    }

    companion object {
        init {
            NitroDfuOnLoad.initializeNative()
        }
    }
}