Add Source Meta (#875)

server/src/main/kotlin/suwayomi/tachidesk/graphql/dataLoaders/MetaDataLoader.kt

@@ -12,9 +12,11 @@ import suwayomi.tachidesk.graphql.types.CategoryMetaType
 import suwayomi.tachidesk.graphql.types.ChapterMetaType
 import suwayomi.tachidesk.graphql.types.GlobalMetaType
 import suwayomi.tachidesk.graphql.types.MangaMetaType
+import suwayomi.tachidesk.graphql.types.SourceMetaType
 import suwayomi.tachidesk.manga.model.table.CategoryMetaTable
 import suwayomi.tachidesk.manga.model.table.ChapterMetaTable
 import suwayomi.tachidesk.manga.model.table.MangaMetaTable
+import suwayomi.tachidesk.manga.model.table.SourceMetaTable
 import suwayomi.tachidesk.server.JavalinSetup.future
 
 class GlobalMetaDataLoader : KotlinDataLoader<String, GlobalMetaType?> {
@@ -88,3 +90,21 @@ class CategoryMetaDataLoader : KotlinDataLoader<Int, List<CategoryMetaType>> {
             }
         }
 }
+
+class SourceMetaDataLoader : KotlinDataLoader<Long, List<SourceMetaType>> {
+    override val dataLoaderName = "SourceMetaDataLoader"
+
+    override fun getDataLoader(): DataLoader<Long, List<SourceMetaType>> =
+        DataLoaderFactory.newDataLoader<Long, List<SourceMetaType>> { ids ->
+            future {
+                transaction {
+                    addLogger(Slf4jSqlDebugLogger)
+                    val metasByRefId =
+                        SourceMetaTable.select { SourceMetaTable.ref inList ids }
+                            .map { SourceMetaType(it) }
+                            .groupBy { it.sourceId }
+                    ids.map { metasByRefId[it].orEmpty() }
+                }
+            }
+        }
+}

server/src/main/kotlin/suwayomi/tachidesk/graphql/mutations/SourceMutation.kt

@@ -5,11 +5,15 @@ import androidx.preference.EditTextPreference
 import androidx.preference.ListPreference
 import androidx.preference.MultiSelectListPreference
 import androidx.preference.SwitchPreferenceCompat
+import org.jetbrains.exposed.sql.SqlExpressionBuilder.eq
+import org.jetbrains.exposed.sql.and
+import org.jetbrains.exposed.sql.deleteWhere
 import org.jetbrains.exposed.sql.select
 import org.jetbrains.exposed.sql.transactions.transaction
 import suwayomi.tachidesk.graphql.types.FilterChange
 import suwayomi.tachidesk.graphql.types.MangaType
 import suwayomi.tachidesk.graphql.types.Preference
+import suwayomi.tachidesk.graphql.types.SourceMetaType
 import suwayomi.tachidesk.graphql.types.SourceType
 import suwayomi.tachidesk.graphql.types.preferenceOf
 import suwayomi.tachidesk.graphql.types.updateFilterList
@@ -17,11 +21,69 @@ import suwayomi.tachidesk.manga.impl.MangaList.insertOrGet
 import suwayomi.tachidesk.manga.impl.Source
 import suwayomi.tachidesk.manga.impl.util.source.GetCatalogueSource
 import suwayomi.tachidesk.manga.model.table.MangaTable
+import suwayomi.tachidesk.manga.model.table.SourceMetaTable
 import suwayomi.tachidesk.manga.model.table.SourceTable
 import suwayomi.tachidesk.server.JavalinSetup.future
 import java.util.concurrent.CompletableFuture
 
 class SourceMutation {
+    data class SetSourceMetaInput(
+        val clientMutationId: String? = null,
+        val meta: SourceMetaType,
+    )
+
+    data class SetSourceMetaPayload(
+        val clientMutationId: String?,
+        val meta: SourceMetaType,
+    )
+
+    fun setSourceMeta(input: SetSourceMetaInput): SetSourceMetaPayload {
+        val (clientMutationId, meta) = input
+
+        Source.modifyMeta(meta.sourceId, meta.key, meta.value)
+
+        return SetSourceMetaPayload(clientMutationId, meta)
+    }
+
+    data class DeleteSourceMetaInput(
+        val clientMutationId: String? = null,
+        val sourceId: Long,
+        val key: String,
+    )
+
+    data class DeleteSourceMetaPayload(
+        val clientMutationId: String?,
+        val meta: SourceMetaType?,
+        val source: SourceType?,
+    )
+
+    fun deleteSourceMeta(input: DeleteSourceMetaInput): DeleteSourceMetaPayload {
+        val (clientMutationId, sourceId, key) = input
+
+        val (meta, source) =
+            transaction {
+                val meta =
+                    SourceMetaTable.select { (SourceMetaTable.ref eq sourceId) and (SourceMetaTable.key eq key) }
+                        .firstOrNull()
+
+                SourceMetaTable.deleteWhere { (SourceMetaTable.ref eq sourceId) and (SourceMetaTable.key eq key) }
+
+                val source =
+                    transaction {
+                        SourceTable.select { SourceTable.id eq sourceId }.firstOrNull()
+                            ?.let { SourceType(it) }
+                    }
+
+                if (meta != null) {
+                    SourceMetaType(meta)
+                } else {
+                    null
+                } to source
+            }
+
+        return DeleteSourceMetaPayload(clientMutationId, meta, source)
+    }
+
     enum class FetchSourceMangaType {
         SEARCH,
         POPULAR,

server/src/main/kotlin/suwayomi/tachidesk/graphql/server/TachideskDataLoaderRegistryFactory.kt

@@ -30,6 +30,7 @@ import suwayomi.tachidesk.graphql.dataLoaders.MangaForIdsDataLoader
 import suwayomi.tachidesk.graphql.dataLoaders.MangaForSourceDataLoader
 import suwayomi.tachidesk.graphql.dataLoaders.MangaMetaDataLoader
 import suwayomi.tachidesk.graphql.dataLoaders.SourceDataLoader
+import suwayomi.tachidesk.graphql.dataLoaders.SourceMetaDataLoader
 import suwayomi.tachidesk.graphql.dataLoaders.SourcesForExtensionDataLoader
 import suwayomi.tachidesk.graphql.dataLoaders.TrackRecordDataLoader
 import suwayomi.tachidesk.graphql.dataLoaders.TrackRecordsForMangaIdDataLoader
@@ -64,6 +65,7 @@ class TachideskDataLoaderRegistryFactory {
                 CategoriesForMangaDataLoader(),
                 SourceDataLoader(),
                 SourcesForExtensionDataLoader(),
+                SourceMetaDataLoader(),
                 ExtensionDataLoader(),
                 ExtensionForSourceDataLoader(),
                 TrackerDataLoader(),

server/src/main/kotlin/suwayomi/tachidesk/graphql/types/MetaType.kt

@@ -12,6 +12,7 @@ import suwayomi.tachidesk.graphql.server.primitives.PageInfo
 import suwayomi.tachidesk.manga.model.table.CategoryMetaTable
 import suwayomi.tachidesk.manga.model.table.ChapterMetaTable
 import suwayomi.tachidesk.manga.model.table.MangaMetaTable
+import suwayomi.tachidesk.manga.model.table.SourceMetaTable
 import java.util.concurrent.CompletableFuture
 
 interface MetaType : Node {
@@ -67,6 +68,22 @@ class CategoryMetaType(
     }
 }
 
+class SourceMetaType(
+    override val key: String,
+    override val value: String,
+    val sourceId: Long,
+) : MetaType {
+    constructor(row: ResultRow) : this(
+        key = row[SourceMetaTable.key],
+        value = row[SourceMetaTable.value],
+        sourceId = row[SourceMetaTable.ref],
+    )
+
+    fun source(dataFetchingEnvironment: DataFetchingEnvironment): CompletableFuture<SourceType> {
+        return dataFetchingEnvironment.getValueFromDataLoader<Long, SourceType>("SourceDataLoader", sourceId)
+    }
+}
+
 class GlobalMetaType(
     override val key: String,
     override val value: String,

server/src/main/kotlin/suwayomi/tachidesk/graphql/types/SourceType.kt

@@ -82,6 +82,10 @@ class SourceType(
     fun filters(): List<Filter> {
         return getCatalogueSourceOrStub(id).getFilterList().map { filterOf(it) }
     }
+
+    fun meta(dataFetchingEnvironment: DataFetchingEnvironment): CompletableFuture<List<SourceMetaType>> {
+        return dataFetchingEnvironment.getValueFromDataLoader<Long, List<SourceMetaType>>("SourceMetaDataLoader", id)
+    }
 }
 
 @Suppress("ktlint:standard:function-naming")