package de.mm20.launcher2.ui.settings.searchactions

import android.content.ComponentName
import android.content.Context
import android.content.Intent
import android.net.Uri
import android.os.Bundle
import androidx.compose.runtime.derivedStateOf
import androidx.compose.runtime.mutableStateOf
import androidx.compose.ui.unit.Density
import androidx.compose.ui.unit.dp
import androidx.core.net.toUri
import androidx.lifecycle.ViewModel
import androidx.lifecycle.viewModelScope
import de.mm20.launcher2.searchactions.SearchActionService
import de.mm20.launcher2.searchactions.actions.SearchActionIcon
import de.mm20.launcher2.searchactions.builders.AppSearchActionBuilder
import de.mm20.launcher2.searchactions.builders.CustomIntentActionBuilder
import de.mm20.launcher2.searchactions.builders.CustomWebsearchActionBuilder
import de.mm20.launcher2.searchactions.builders.CustomizableSearchActionBuilder
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.flow.flow
import kotlinx.coroutines.launch
import org.koin.core.component.KoinComponent
import org.koin.core.component.inject
import java.io.File
import java.text.Collator
import kotlin.math.roundToInt

class EditSearchActionSheetVM : ViewModel(), KoinComponent {

    private val searchActionService: SearchActionService by inject()

    private var initialCustomIcon: String? = null

    val currentPage = mutableStateOf<EditSearchActionPage?>(null)
    val createNew = mutableStateOf(false)

    val searchAction = mutableStateOf<CustomizableSearchActionBuilder?>(null)

    fun init(searchAction: CustomizableSearchActionBuilder?) {
        initialCustomIcon = searchAction?.customIcon
        currentPage.value = when (searchAction) {
            is AppSearchActionBuilder -> EditSearchActionPage.CustomizeAppSearch
            is CustomWebsearchActionBuilder -> EditSearchActionPage.CustomizeWebSearch
            is CustomIntentActionBuilder -> EditSearchActionPage.CustomizeCustomIntent
            else -> EditSearchActionPage.SelectType
        }
        createNew.value = searchAction == null
        this.searchAction.value = searchAction
    }

    fun getSearchableApps(context: Context) = flow {
        val items = searchActionService.getSearchActivities().map {
            SearchableApp(
                label = context.packageManager.getActivityInfo(it, 0)
                    .loadLabel(context.packageManager).toString(),
                componentName = it
            )
        }.sorted()
        emit(items)
    }

    fun initAppSearch() {
        currentPage.value = EditSearchActionPage.InitAppSearch
    }

    fun initWebSearch() {
        currentPage.value = EditSearchActionPage.InitWebSearch
    }

    fun selectSearchableApp(app: SearchableApp) {
        searchAction.value = AppSearchActionBuilder(
            label = app.label,
            baseIntent = Intent().apply {
                setComponent(app.componentName)
            },
            icon = SearchActionIcon.Custom,
            customIcon = null,
            iconColor = 1,
        )
        currentPage.value = EditSearchActionPage.CustomizeAppSearch
    }


    fun initCustomIntent() {
        searchAction.value = CustomIntentActionBuilder(
            label = "",
            queryKey = "",
            baseIntent = Intent(),
            icon = SearchActionIcon.Search,
            customIcon = null,
            iconColor = 0,
        )
        currentPage.value = EditSearchActionPage.CustomizeCustomIntent
    }

    fun setLabel(label: String) {
        val action = searchAction.value ?: return

        val newAction = when (action) {
            is CustomIntentActionBuilder -> action.copy(label = label)
            is AppSearchActionBuilder -> action.copy(label = label)
            is CustomWebsearchActionBuilder -> action.copy(label = label)
        }

        searchAction.value = newAction
    }

    fun setComponentName(componentName: ComponentName) {
        val action = searchAction.value ?: return

        val newAction = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                baseIntent = action.baseIntent.setComponent(
                    componentName
                )
            )

            is AppSearchActionBuilder -> action.also {
                it.baseIntent.setComponent(componentName)
            }

            is CustomWebsearchActionBuilder -> action
        }

        searchAction.value = newAction
    }


    val initWebsearchUrl = mutableStateOf("")

    /**
     * Last imported URL that failed (if the current URL is equal to this, show an error banner)
     */
    private val websearchImportErrorUrl = mutableStateOf<String?>(null)
    val websearchImportError =
        derivedStateOf { websearchImportErrorUrl.value == initWebsearchUrl.value }
    val loadingWebsearch = mutableStateOf(false)

    val skipWebsearchImport =
        derivedStateOf { websearchImportError.value || initWebsearchUrl.value.isEmpty() }

    fun importWebsearch(density: Density) {
        if (loadingWebsearch.value) return
        viewModelScope.launch {
            val url = initWebsearchUrl.value
            loadingWebsearch.value = true
            val action = searchActionService.importWebsearch(
                url,
                with(density) { 20.dp.toPx().roundToInt() })
            if (action == null) {
                websearchImportErrorUrl.value = url
            } else {
                websearchImportErrorUrl.value = null
                searchAction.value = action
                currentPage.value = EditSearchActionPage.CustomizeWebSearch
            }
            loadingWebsearch.value = false
        }
    }

    fun skipWebsearchImport() {
        searchAction.value = CustomWebsearchActionBuilder(
            urlTemplate = "",
            iconColor = 0,
            icon = SearchActionIcon.Search,
            label = "",
        )
        currentPage.value = EditSearchActionPage.CustomizeWebSearch
    }

    fun setUrlTemplate(template: String) {
        val action = searchAction.value ?: return
        if (action is CustomWebsearchActionBuilder) {
            searchAction.value = action.copy(
                urlTemplate = template
            )
        }
    }


    private val invalidWebsearchUrl = mutableStateOf<String?>(null)
    val websearchInvalidUrlError =
        derivedStateOf { invalidWebsearchUrl.value == (searchAction.value as? CustomWebsearchActionBuilder)?.urlTemplate }
    val customIntentTemplateError = mutableStateOf(false)
    val customIntentKeyError = mutableStateOf(false)
    fun validate(): Boolean {
        val action = searchAction.value ?: return false

        if (action is CustomWebsearchActionBuilder) {
            val valid = action.urlTemplate.contains("\${1}")
            invalidWebsearchUrl.value = if (valid) null else action.urlTemplate
            return valid
        }
        if (action is CustomIntentActionBuilder) {
            customIntentTemplateError.value = !(action.queryTemplate.isNullOrEmpty() || action.queryTemplate!!.contains("\${1}"))
            customIntentKeyError.value = action.queryKey != null && action.queryKey!!.isEmpty()
            return !customIntentTemplateError.value && !customIntentKeyError.value
        }
        return true
    }

    fun onSave() {
        val action = searchAction.value ?: return
        if (initialCustomIcon != action.customIcon) deleteCustomIcon(initialCustomIcon)
    }

    fun onDismiss() {
        currentPage.value = null
        val action = searchAction.value ?: return
        val newIcon = action.customIcon
        if (newIcon != initialCustomIcon) {
            deleteCustomIcon(newIcon)
        }
    }

    fun setIcon(icon: SearchActionIcon) {
        val action = searchAction.value ?: return
        if (action.customIcon != initialCustomIcon) {
            deleteCustomIcon(action.customIcon)
        }
        searchAction.value = when (action) {
            is CustomWebsearchActionBuilder -> action.copy(
                icon = icon,
                customIcon = null,
                iconColor = 0
            )

            is CustomIntentActionBuilder -> action.copy(
                icon = icon,
                customIcon = null,
                iconColor = 0
            )

            is AppSearchActionBuilder -> action.copy(icon = icon, customIcon = null, iconColor = 0)
        }
    }

    fun setCustomIcon(iconPath: String?) {
        val action = searchAction.value ?: return
        if (action.customIcon != initialCustomIcon) {
            deleteCustomIcon(action.customIcon)
        }
        searchAction.value = when (action) {
            is CustomWebsearchActionBuilder -> action.copy(
                customIcon = iconPath,
                iconColor = 1,
                icon = SearchActionIcon.Custom
            )

            is CustomIntentActionBuilder -> action.copy(
                customIcon = iconPath,
                iconColor = 1,
                icon = SearchActionIcon.Custom
            )

            is AppSearchActionBuilder -> action.copy(
                customIcon = iconPath,
                iconColor = 1,
                icon = SearchActionIcon.Custom
            )
        }
    }

    private fun deleteCustomIcon(path: String?) {
        path ?: return
        viewModelScope.launch(Dispatchers.IO) {
            File(path).delete()
        }
    }

    fun openIconPicker() {
        currentPage.value = EditSearchActionPage.PickIcon
    }

    fun applyIcon() {
        currentPage.value = when (searchAction.value) {
            is AppSearchActionBuilder -> EditSearchActionPage.CustomizeAppSearch
            is CustomWebsearchActionBuilder -> EditSearchActionPage.CustomizeWebSearch
            is CustomIntentActionBuilder -> EditSearchActionPage.CustomizeCustomIntent
            null -> EditSearchActionPage.SelectType
        }
    }

    fun setIconColor(color: Int) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomWebsearchActionBuilder -> action.copy(iconColor = color)
            is CustomIntentActionBuilder -> action.copy(iconColor = color)
            is AppSearchActionBuilder -> action.copy(iconColor = color)
        }
    }

    fun importIcon(uri: Uri, size: Int) {
        viewModelScope.launch {
            val path = searchActionService.createIcon(uri, size)
            setCustomIcon(path)
        }
    }

    fun removeExtra(key: String) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.remove(key)
                    it.replaceExtras(extras)
                },
            )

            is AppSearchActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.remove(key)
                    it.replaceExtras(extras)
                }
            )

            else -> action
        }
    }

    fun putStringExtra(key: String, value: String = "") {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putString(key, value)
                    it.replaceExtras(extras)
                },
            )

            is AppSearchActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putString(key, value)
                    it.replaceExtras(extras)
                },
            )

            else -> action
        }
    }

    fun putIntExtra(key: String, value: Int = 0) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putInt(key, value)
                    it.replaceExtras(extras)
                },
            )

            is AppSearchActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putInt(key, value)
                    it.replaceExtras(extras)
                },
            )

            else -> action
        }
    }

    fun putLongExtra(key: String, value: Long = 0L) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putLong(key, value)
                    it.replaceExtras(extras)
                },
            )

            is AppSearchActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putLong(key, value)
                    it.replaceExtras(extras)
                },
            )

            else -> action
        }
    }

    fun putFloatExtra(key: String, value: Float = 0f) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putFloat(key, value)
                    it.replaceExtras(extras)
                },
            )

            is AppSearchActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putFloat(key, value)
                    it.replaceExtras(extras)
                },
            )

            else -> action
        }
    }

    fun putDoubleExtra(key: String, value: Double = 0.0) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putDouble(key, value)
                    it.replaceExtras(extras)
                },
            )

            is AppSearchActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putDouble(key, value)
                    it.replaceExtras(extras)
                },
            )

            else -> action
        }
    }

    fun putBooleanExtra(key: String, value: Boolean = false) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putBoolean(key, value)
                    it.replaceExtras(extras)
                },
            )

            is AppSearchActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    extras.putBoolean(key, value)
                    it.replaceExtras(extras)
                },
            )

            else -> action
        }
    }

    fun setIntentAction(action: String) {
        val searchAction = searchAction.value ?: return
        this.searchAction.value = when (searchAction) {
            is CustomIntentActionBuilder -> searchAction.copy(
                baseIntent = searchAction.baseIntent.cloneFilter().also {
                    val extras = searchAction.baseIntent.extras?.deepCopy() ?: Bundle()
                    it.replaceExtras(extras)
                    it.action = action
                },
            )

            else -> searchAction
        }
    }

    fun setIntentCategory(category: String) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                baseIntent = action.baseIntent.cloneFilter().also {
                    val extras = action.baseIntent.extras?.deepCopy() ?: Bundle()
                    it.replaceExtras(extras)
                    val oldCategory = it.categories?.firstOrNull()
                    if (oldCategory != null) it.removeCategory(oldCategory)
                    if (category.isNotBlank()) it.addCategory(category)
                },
            )

            else -> action
        }
    }

    fun setIntentType(type: String) {
        val searchAction = searchAction.value ?: return
        this.searchAction.value = when (searchAction) {
            is CustomIntentActionBuilder -> searchAction.copy(
                baseIntent = searchAction.baseIntent.cloneFilter().also {
                    val extras = searchAction.baseIntent.extras?.deepCopy() ?: Bundle()
                    it.replaceExtras(extras)
                    it.setDataAndType(it.data, type.takeIf { it.isNotBlank() })
                },
            )

            else -> searchAction
        }
    }

    fun setIntentData(data: String) {
        val searchAction = searchAction.value ?: return
        this.searchAction.value = when (searchAction) {
            is CustomIntentActionBuilder -> searchAction.copy(
                baseIntent = searchAction.baseIntent.cloneFilter().also {
                    val extras = searchAction.baseIntent.extras?.deepCopy() ?: Bundle()
                    it.replaceExtras(extras)
                    it.setDataAndType(data.takeIf { it.isNotEmpty() }?.toUri(), it.type)
                },
            )

            else -> searchAction
        }
    }

    fun setIntentPackage(pkg: String) {
        val searchAction = searchAction.value ?: return
        this.searchAction.value = when (searchAction) {
            is CustomIntentActionBuilder -> searchAction.copy(
                baseIntent = searchAction.baseIntent.cloneFilter().also {
                    val extras = searchAction.baseIntent.extras?.deepCopy() ?: Bundle()
                    it.replaceExtras(extras)
                    if (pkg.isBlank()) {
                        it.`package` = null
                        it.component = null
                    } else if (it.component != null) {
                        it.component = ComponentName(pkg, it.component!!.className)
                    } else {
                        it.`package` = pkg
                    }
                },
            )
            else -> searchAction
        }
    }
    fun setIntentClassName(className: String) {
        val searchAction = searchAction.value ?: return
        this.searchAction.value = when (searchAction) {
            is CustomIntentActionBuilder -> searchAction.copy(
                baseIntent = searchAction.baseIntent.cloneFilter().also {
                    val extras = searchAction.baseIntent.extras?.deepCopy() ?: Bundle()
                    it.replaceExtras(extras)

                    val pkg = it.component?.packageName ?: it.`package`

                    if (className.isBlank()) {
                        it.`package` = pkg
                        it.component = null
                    } else if (pkg != null) {
                        it.`package` = null
                        it.component = ComponentName(pkg, className)
                    }
                },
            )
            else -> searchAction
        }
    }

    fun setQueryKey(key: String?) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                queryKey = key,
            )

            else -> action
        }
    }

    fun setIntentQueryTemplate(template: String) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomIntentActionBuilder -> action.copy(
                queryTemplate = template.takeIf { it.isNotEmpty() },
            )

            else -> action
        }
    }

    fun setQueryEncoding(encoding: CustomWebsearchActionBuilder.QueryEncoding) {
        val action = searchAction.value ?: return
        searchAction.value = when (action) {
            is CustomWebsearchActionBuilder -> action.copy(
                encoding = encoding
            )

            else -> action
        }
    }
}

enum class EditSearchActionPage {
    SelectType,
    InitAppSearch,
    InitWebSearch,
    CustomizeAppSearch,
    CustomizeWebSearch,
    CustomizeCustomIntent,
    PickIcon,
}

data class SearchableApp(
    val label: String,
    val componentName: ComponentName,
): Comparable<SearchableApp> {
    override fun compareTo(other: SearchableApp): Int {
        val label1 = label
        val label2 = other.label
        return Collator.getInstance().apply { strength = Collator.SECONDARY }
            .compare(label1, label2)
    }

}