Suwayomi-Server/server/src/main/kotlin/suwayomi/tachidesk/graphql/mutations/ExtensionMutation.kt

package suwayomi.tachidesk.graphql.mutations

import eu.kanade.tachiyomi.source.local.LocalSource
import graphql.execution.DataFetcherResult
import io.javalin.http.UploadedFile
import org.jetbrains.exposed.sql.selectAll
import org.jetbrains.exposed.sql.transactions.transaction
import suwayomi.tachidesk.graphql.asDataFetcherResult
import suwayomi.tachidesk.graphql.directives.RequireAuth
import suwayomi.tachidesk.graphql.types.ExtensionType
import suwayomi.tachidesk.manga.impl.extension.Extension
import suwayomi.tachidesk.manga.impl.extension.ExtensionsList
import suwayomi.tachidesk.manga.model.table.ExtensionTable
import suwayomi.tachidesk.server.JavalinSetup.future
import java.util.concurrent.CompletableFuture

class ExtensionMutation {
    data class UpdateExtensionPatch(
        val install: Boolean? = null,
        val update: Boolean? = null,
        val uninstall: Boolean? = null,
    )

    data class UpdateExtensionPayload(
        val clientMutationId: String?,
        val extension: ExtensionType?,
    )

    data class UpdateExtensionInput(
        val clientMutationId: String? = null,
        val id: String,
        val patch: UpdateExtensionPatch,
    )

    data class UpdateExtensionsPayload(
        val clientMutationId: String?,
        val extensions: List<ExtensionType>,
    )

    data class UpdateExtensionsInput(
        val clientMutationId: String? = null,
        val ids: List<String>,
        val patch: UpdateExtensionPatch,
    )

    private suspend fun updateExtensions(
        ids: List<String>,
        patch: UpdateExtensionPatch,
    ) {
        val extensions =
            transaction {
                ExtensionTable
                    .selectAll()
                    .where { ExtensionTable.pkgName inList ids }
                    .map { ExtensionType(it) }
            }

        if (patch.update == true) {
            extensions.filter { it.hasUpdate }.forEach {
                Extension.updateExtension(it.pkgName)
            }
        }

        if (patch.install == true) {
            extensions.filterNot { it.isInstalled }.forEach {
                Extension.installExtension(it.pkgName)
            }
        }

        if (patch.uninstall == true) {
            extensions.filter { it.isInstalled }.forEach {
                Extension.uninstallExtension(it.pkgName)
            }
        }
    }

    @RequireAuth
    fun updateExtension(input: UpdateExtensionInput): CompletableFuture<DataFetcherResult<UpdateExtensionPayload>> {
        val (clientMutationId, id, patch) = input

        return future {
            asDataFetcherResult {
                updateExtensions(listOf(id), patch)

                val extension =
                    transaction {
                        ExtensionTable
                            .selectAll()
                            .where { ExtensionTable.pkgName eq id }
                            .firstOrNull()
                            ?.let { ExtensionType(it) }
                    }

                UpdateExtensionPayload(
                    clientMutationId = clientMutationId,
                    extension = extension,
                )
            }
        }
    }

    @RequireAuth
    fun updateExtensions(input: UpdateExtensionsInput): CompletableFuture<DataFetcherResult<UpdateExtensionsPayload>> {
        val (clientMutationId, ids, patch) = input

        return future {
            asDataFetcherResult {
                updateExtensions(ids, patch)

                val extensions =
                    transaction {
                        ExtensionTable
                            .selectAll()
                            .where { ExtensionTable.pkgName inList ids }
                            .map { ExtensionType(it) }
                    }

                UpdateExtensionsPayload(
                    clientMutationId = clientMutationId,
                    extensions = extensions,
                )
            }
        }
    }

    data class FetchExtensionsInput(
        val clientMutationId: String? = null,
    )

    data class FetchExtensionsPayload(
        val clientMutationId: String?,
        val extensions: List<ExtensionType>,
    )

    @RequireAuth
    fun fetchExtensions(input: FetchExtensionsInput): CompletableFuture<DataFetcherResult<FetchExtensionsPayload>> {
        val (clientMutationId) = input

        return future {
            asDataFetcherResult {
                ExtensionsList.fetchExtensions()

                val extensions =
                    transaction {
                        ExtensionTable
                            .selectAll()
                            .where { ExtensionTable.name neq LocalSource.EXTENSION_NAME }
                            .map { ExtensionType(it) }
                    }

                FetchExtensionsPayload(
                    clientMutationId = clientMutationId,
                    extensions = extensions,
                )
            }
        }
    }

    data class InstallExternalExtensionInput(
        val clientMutationId: String? = null,
        val extensionFile: UploadedFile,
    )

    data class InstallExternalExtensionPayload(
        val clientMutationId: String?,
        val extension: ExtensionType,
    )

    @RequireAuth
    fun installExternalExtension(
        input: InstallExternalExtensionInput,
    ): CompletableFuture<DataFetcherResult<InstallExternalExtensionPayload>> {
        val (clientMutationId, extensionFile) = input

        return future {
            asDataFetcherResult {
                Extension.installExternalExtension(extensionFile.content(), extensionFile.filename())

                val dbExtension =
                    transaction { ExtensionTable.selectAll().where { ExtensionTable.apkName eq extensionFile.filename() }.first() }

                InstallExternalExtensionPayload(
                    clientMutationId,
                    extension = ExtensionType(dbExtension),
                )
            }
        }
    }
}