mirrors/Tachidesk
1
0
Fork 0
mirror of https://github.com/Suwayomi/Tachidesk.git synced 2026-01-06 03:42:34 +01:00
Code Issues 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 eu.kanade.tachiyomi.source.ConfigurableSource
import eu.kanade.tachiyomi.source.getPreferenceKey
import eu.kanade.tachiyomi.source.online.HttpSource
import mu.KotlinLogging
import org.jetbrains.exposed.sql.select
import org.jetbrains.exposed.sql.selectAll
@@ -32,16 +33,25 @@ import xyz.nulldev.androidcompat.androidimpl.CustomContext
object Source {
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> {
return transaction {
SourceTable.selectAll().map {
val httpSource = getHttpSource(it[SourceTable.id].value)
SourceDataClass(
it[SourceTable.id].value.toString(),
it[SourceTable.name],
it[SourceTable.lang],
getExtensionIconUrl(ExtensionTable.select { ExtensionTable.id eq it[SourceTable.extension] }.first()[ExtensionTable.apkName]),
getHttpSource(it[SourceTable.id].value).supportsLatest,
getHttpSource(it[SourceTable.id].value) is ConfigurableSource
getExtensionIconUrl(
ExtensionTable.select { ExtensionTable.id eq it[SourceTable.extension] }
.first()[ExtensionTable.apkName]
),
httpSource.supportsLatest,
httpSource is ConfigurableSource,
httpSource.isNsfw
)
}
}
@@ -50,14 +60,21 @@ object Source {
fun getSource(sourceId: Long): SourceDataClass {
return transaction {
val source = SourceTable.select { SourceTable.id eq sourceId }.firstOrNull()
val httpSource = source?.let { getHttpSource(sourceId) }
SourceDataClass(
sourceId.toString(),
source?.get(SourceTable.name),
source?.get(SourceTable.lang),
source?.let { getExtensionIconUrl(ExtensionTable.select { ExtensionTable.id eq source[SourceTable.extension] }.first()[ExtensionTable.apkName]) },
source?.let { getHttpSource(sourceId).supportsLatest },
source?.let { getHttpSource(sourceId) is ConfigurableSource },
source?.let {
getExtensionIconUrl(
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)
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)
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 iconUrl: String?,
val supportsLatest: Boolean?,
val isConfigurable: Boolean?
val isConfigurable: Boolean?,
val isNSFW: Boolean?,
)