package moe.yuuta.workmode.suspend

import android.content.Context
import android.os.Bundle
import moe.yuuta.workmode.access.AccessorStarter
import moe.yuuta.workmode.access.DumpResult
import moe.yuuta.workmode.suspend.data.TransferableSuspendedApp
import java.util.concurrent.CompletableFuture
import java.util.function.Function
import java.util.function.Supplier

/**
 * An async suspender which wraps suspend tasks and run them in the background
 */
class AsyncSuspender(private val mContext: Context) {
    fun isSuspended(packages: List<TransferableSuspendedApp>): CompletableFuture<Boolean> = generalCall(Function {
            return@Function it.isPackageSuspended(packages)
        })

    fun suspend(packages: List<TransferableSuspendedApp>, suspended: Boolean): CompletableFuture<Array<String>> = generalCall(
        Function {
            return@Function it.suspend(packages, suspended)
        })

    fun getInstalledApplicationsAcrossUser(flags: Int): CompletableFuture<List<TransferableSuspendedApp>> = generalCall(
        Function {
            return@Function it.getInstalledApplicationsAcrossUser(flags)
        })

    private fun <T> generalCall(thenApply: java.util.function.Function<AccessorStarter, T>): CompletableFuture<T> {
        return CompletableFuture.supplyAsync(Supplier {
            return@Supplier AccessorStarter.start(mContext, true, thenApply)
        })
    }

    fun applyFromSettings(): CompletableFuture<Unit> = generalCall(Function { t -> t.apply() })

    fun getSuspendedPackageAppExtras(packageInfo: TransferableSuspendedApp): CompletableFuture<Bundle?> = generalCall(Function {
        return@Function it.getSuspendedPackageAppExtras(packageInfo)
    })

    fun getSuspendedPackageLauncherExtras(packageInfo: TransferableSuspendedApp): CompletableFuture<Bundle?> = generalCall(Function {
        return@Function it.getSuspendedPackageLauncherExtras(packageInfo)
    })

    fun dump(packageInfo: TransferableSuspendedApp): CompletableFuture<DumpResult> = generalCall(Function {
        return@Function it.dump(packageInfo)
    })

    private data class DataWrapper<T>(val accessorStarter: AccessorStarter?, val data: T?)
}