mihon/app/src/main/java/exh/ui/lock/LockActivityDelegate.kt

package exh.ui.lock

import android.view.WindowManager
import androidx.fragment.app.FragmentActivity
import com.bluelinelabs.conductor.Router
import com.bluelinelabs.conductor.RouterTransaction
import eu.kanade.tachiyomi.data.preference.PreferencesHelper
import eu.kanade.tachiyomi.data.preference.getOrDefault
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.flow.launchIn
import kotlinx.coroutines.flow.onEach
import uy.kohesive.injekt.injectLazy

object LockActivityDelegate {
    private val preferences by injectLazy<PreferencesHelper>()

    var willLock: Boolean = true

    private val uiScope = CoroutineScope(Dispatchers.Main)

    fun doLock(router: Router, animate: Boolean = false) {
        router.pushController(
            RouterTransaction.with(LockController())
                .popChangeHandler(LockChangeHandler(animate))
        )
    }

    fun onCreate(activity: FragmentActivity) {
        preferences.secureScreen().asFlow()
            .onEach {
                if (it) {
                    activity.window.setFlags(WindowManager.LayoutParams.FLAG_SECURE, WindowManager.LayoutParams.FLAG_SECURE)
                } else {
                    activity.window.clearFlags(WindowManager.LayoutParams.FLAG_SECURE)
                }
            }
            .launchIn(uiScope)
    }

    fun onResume(activity: FragmentActivity, router: Router) {
        if (lockEnabled() && !isAppLocked(router) && willLock && !preferences.eh_lockManually().getOrDefault()) {
            doLock(router)
            willLock = false
        }
    }

    private fun isAppLocked(router: Router): Boolean {
        return router.backstack.lastOrNull()?.controller() is LockController
    }
}