XuqmGroup-AndroidSDK/sdk-update/src/main/java/com/xuqm/sdk/update/UpdateSDK.kt

package com.xuqm.sdk.update

import android.content.Context
import android.content.Intent
import android.net.Uri
import android.os.Build
import androidx.core.content.FileProvider
import com.xuqm.sdk.XuqmSDK
import com.xuqm.sdk.file.FileSDK
import com.xuqm.sdk.core.ServiceEndpointRegistry
import com.xuqm.sdk.network.ApiClient
import com.xuqm.sdk.update.api.UpdateApi
import com.xuqm.sdk.update.model.UpdateInfo
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.withContext
import java.io.File

object UpdateSDK {

    private val api: UpdateApi get() = ApiClient.create(UpdateApi::class.java, ServiceEndpointRegistry.updateBaseUrl)

    private fun normalizeDownloadUrl(rawUrl: String?): String? {
        if (rawUrl.isNullOrBlank()) return rawUrl

        return runCatching {
            val uri = Uri.parse(rawUrl)
            if (uri.path?.startsWith("/files/apk/") == true) {
                "${uri.scheme}://${uri.authority}/api/v1/updates${uri.path}"
            } else {
                rawUrl
            }
        }.getOrDefault(rawUrl)
    }

    suspend fun checkAppUpdate(context: Context): UpdateInfo? = withContext(Dispatchers.IO) {
        awaitInitialization()
        val packageInfo = context.packageManager.getPackageInfo(context.packageName, 0)
        val versionCode = if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.P) {
            packageInfo.longVersionCode.toInt()
        } else {
            @Suppress("DEPRECATION")
            packageInfo.versionCode
        }
        val userId = XuqmSDK.currentLoginSession?.userId
        runCatching {
            api.checkUpdate(XuqmSDK.appKey, "ANDROID", versionCode, userId).data?.let {
                it.copy(downloadUrl = normalizeDownloadUrl(it.downloadUrl) ?: it.downloadUrl)
            }
        }.getOrNull()
    }

    private suspend fun awaitInitialization() {
        val xuqmClazz = runCatching { Class.forName("com.xuqm.sdk.XuqmSDK") }.getOrNull()
        if (xuqmClazz != null) {
            val isInitMethod = xuqmClazz.getMethod("isInitialized")
            val maxWaitMs = 30000L
            val start = System.currentTimeMillis()
            while (!(isInitMethod.invoke(null) as Boolean)) {
                if (System.currentTimeMillis() - start > maxWaitMs) {
                    throw IllegalStateException("UpdateSDK initialization timed out waiting for XuqmSDK")
                }
                kotlinx.coroutines.delay(100)
            }
        }
        XuqmSDK.requireInit()
    }

    suspend fun downloadAndInstall(
        context: Context,
        downloadUrl: String,
        onProgress: (Int) -> Unit = {},
    ) = withContext(Dispatchers.IO) {
        val apkFile = File(context.getExternalFilesDir(null), "update.apk")
        FileSDK.downloadToFile(downloadUrl, apkFile, onProgress)
        withContext(Dispatchers.Main) { installApk(context, apkFile) }
    }

    private fun installApk(context: Context, apkFile: File) {
        val intent = Intent(Intent.ACTION_VIEW).apply {
            flags = Intent.FLAG_ACTIVITY_NEW_TASK or Intent.FLAG_GRANT_READ_URI_PERMISSION
            val uri = FileProvider.getUriForFile(context, "${context.packageName}.fileprovider", apkFile)
            setDataAndType(uri, "application/vnd.android.package-archive")
        }
        context.startActivity(intent)
    }
}