package expo.modules.nativewechat

import android.os.Bundle
import androidx.core.os.bundleOf
import com.tencent.mm.opensdk.constants.ConstantsAPI
import com.tencent.mm.opensdk.modelbase.BaseResp
import com.tencent.mm.opensdk.modelbiz.WXLaunchMiniProgram
import com.tencent.mm.opensdk.modelbiz.WXOpenBusinessView
import com.tencent.mm.opensdk.modelmsg.SendAuth

object NativeWechatRespDataHelper {
    private var type: String = ""

    fun downcastResp(baseResp: BaseResp?): Bundle {
        if (baseResp == null)
            return bundleOf()

        val argument = bundleOf()

        if (baseResp.type == ConstantsAPI.COMMAND_SENDAUTH) {
            type = "SendAuthResp"
            val resp = baseResp as SendAuth.Resp

            argument.putString("code", resp.code)
            argument.putString("state", resp.state)
            argument.putString("lang", resp.lang)
            argument.putString("country", resp.country)
        }

        if (baseResp.type == ConstantsAPI.COMMAND_LAUNCH_WX_MINIPROGRAM) {
            type = "WXLaunchMiniProgramResp"
            val resp = baseResp as WXLaunchMiniProgram.Resp

            argument.putString("extMsg", resp.extMsg)
        }

        if (baseResp.type == ConstantsAPI.COMMAND_PAY_BY_WX) {
            type = "PayResp"
        }

        if (baseResp.type == ConstantsAPI.COMMAND_OPEN_BUSINESS_VIEW) {
            type = "WXOpenBusinessViewResp"
            val resp = baseResp as WXOpenBusinessView.Resp

            argument.putString("businessType", resp.businessType)
            argument.putString("extMsg", resp.extMsg)
        }

        return wrapResponse(baseResp, argument)
    }

    private fun wrapResponse(
        baseResp: BaseResp,
        data: Bundle
    ): Bundle {
        return bundleOf(
            "type" to type,
            "errorCode" to baseResp.errCode,
            "errorStr" to baseResp.errStr,
            "transaction" to baseResp.transaction,
            "data" to data
        )
    }
}