temple/Suwayomi-Server
1
0
Fork 0
mirror of https://github.com/Suwayomi/Suwayomi-Server.git synced 2026-07-01 01:44:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

add isNsfw to SourceDataClass

Browse Source
This commit is contained in:
Aria Moradi
2021-08-23 21:46:28 +04:30
parent 65d9021c37
commit 45a50ca0c1
2 changed files with 27 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
server/src/main/kotlin/suwayomi/tachidesk/manga/impl/Source.kt
View File

@@ -12,6 +12,7 @@ import android.content.Context
import androidx.preference.PreferenceScreen import androidx.preference.PreferenceScreen
import eu.kanade.tachiyomi.source.ConfigurableSource import eu.kanade.tachiyomi.source.ConfigurableSource
import eu.kanade.tachiyomi.source.getPreferenceKey import eu.kanade.tachiyomi.source.getPreferenceKey
import eu.kanade.tachiyomi.source.online.HttpSource
import mu.KotlinLogging import mu.KotlinLogging
import org.jetbrains.exposed.sql.select import org.jetbrains.exposed.sql.select
import org.jetbrains.exposed.sql.selectAll import org.jetbrains.exposed.sql.selectAll
@@ -32,16 +33,25 @@ import xyz.nulldev.androidcompat.androidimpl.CustomContext
object Source { object Source {
private val logger = KotlinLogging.logger {} private val logger = KotlinLogging.logger {}
val HttpSource.isNsfw: Boolean
get() = this::class.annotations.any { it.toString() == "@eu.kanade.tachiyomi.annotations.Nsfw()" }
fun getSourceList(): List<SourceDataClass> { fun getSourceList(): List<SourceDataClass> {
return transaction { return transaction {
SourceTable.selectAll().map { SourceTable.selectAll().map {
val httpSource = getHttpSource(it[SourceTable.id].value)
SourceDataClass( SourceDataClass(
it[SourceTable.id].value.toString(), it[SourceTable.id].value.toString(),
it[SourceTable.name], it[SourceTable.name],
it[SourceTable.lang], it[SourceTable.lang],
getExtensionIconUrl(ExtensionTable.select { ExtensionTable.id eq it[SourceTable.extension] }.first()[ExtensionTable.apkName]), getExtensionIconUrl(
getHttpSource(it[SourceTable.id].value).supportsLatest, ExtensionTable.select { ExtensionTable.id eq it[SourceTable.extension] }
getHttpSource(it[SourceTable.id].value) is ConfigurableSource .first()[ExtensionTable.apkName]
),
httpSource.supportsLatest,
httpSource is ConfigurableSource,
httpSource.isNsfw
) )
} }
} }
@@ -50,14 +60,21 @@ object Source {
fun getSource(sourceId: Long): SourceDataClass { fun getSource(sourceId: Long): SourceDataClass {
return transaction { return transaction {
val source = SourceTable.select { SourceTable.id eq sourceId }.firstOrNull() val source = SourceTable.select { SourceTable.id eq sourceId }.firstOrNull()
val httpSource = source?.let { getHttpSource(sourceId) }
SourceDataClass( SourceDataClass(
sourceId.toString(), sourceId.toString(),
source?.get(SourceTable.name), source?.get(SourceTable.name),
source?.get(SourceTable.lang), source?.get(SourceTable.lang),
source?.let { getExtensionIconUrl(ExtensionTable.select { ExtensionTable.id eq source[SourceTable.extension] }.first()[ExtensionTable.apkName]) }, source?.let {
source?.let { getHttpSource(sourceId).supportsLatest }, getExtensionIconUrl(
source?.let { getHttpSource(sourceId) is ConfigurableSource }, ExtensionTable.select { ExtensionTable.id eq source[SourceTable.extension] }
.first()[ExtensionTable.apkName]
)
},
httpSource?.supportsLatest,
httpSource?.let { it is ConfigurableSource },
httpSource?.isNsfw
) )
} }
} }
@@ -85,7 +102,8 @@ object Source {
val source = getHttpSource(sourceId) val source = getHttpSource(sourceId)
if (source is ConfigurableSource) { if (source is ConfigurableSource) {
val sourceShardPreferences = Injekt.get<Application>().getSharedPreferences(source.getPreferenceKey(), Context.MODE_PRIVATE) val sourceShardPreferences =
Injekt.get<Application>().getSharedPreferences(source.getPreferenceKey(), Context.MODE_PRIVATE)
val screen = PreferenceScreen(context) val screen = PreferenceScreen(context)
screen.sharedPreferences = sourceShardPreferences screen.sharedPreferences = sourceShardPreferences

3
server/src/main/kotlin/suwayomi/tachidesk/manga/model/dataclass/SourceDataClass.kt
View File

@@ -13,5 +13,6 @@ data class SourceDataClass(
val lang: String?, val lang: String?,
val iconUrl: String?, val iconUrl: String?,
val supportsLatest: Boolean?, val supportsLatest: Boolean?,
val isConfigurable: Boolean? val isConfigurable: Boolean?,
val isNSFW: Boolean?,
) )