16 Commits

Author SHA1 Message Date
renovate[bot]
d6383d12f2 Update dependency dev.inmo:tgbotapi to v28 2025-09-10 12:30:26 +00:00
ecffb1dd07 updates 2024-11-16 17:39:18 +06:00
34a8267568 0.0.5: update dependencies 2024-11-16 17:17:03 +06:00
a8c317af44 fix of build 2024-10-22 17:15:49 +06:00
3e2b6e67e9 Update docker-publish.yml 2024-10-22 17:08:14 +06:00
b81be8404c Update build.yml 2024-10-22 17:06:00 +06:00
d60898f4f1 Update settings.gradle 2024-10-22 17:02:22 +06:00
115eb8bf3a Update gradle-wrapper.properties 2024-10-22 16:58:50 +06:00
ebc4b80ef3 Update libs.versions.toml 2024-10-22 16:58:30 +06:00
16e85b0878 Merge pull request #5 from InsanusMokrassar/renovate/ktor-monorepo
Update dependency io.ktor:ktor-client-okhttp to v2.3.12
2024-10-22 16:55:35 +06:00
renovate[bot]
b7c33ded79 Update dependency io.ktor:ktor-client-okhttp to v2.3.12 2024-10-22 10:50:46 +00:00
8863afa5a7 Merge pull request #3 from InsanusMokrassar/renovate/configure
Configure Renovate
2024-10-22 16:49:50 +06:00
renovate[bot]
9ff5ffa9c8 Add renovate.json 2024-10-22 10:46:08 +00:00
41549fbef1 improvements 2023-10-12 14:08:13 +06:00
4e2c5ea31e update dependencies 2023-10-11 19:03:10 +06:00
046a8bede4 fix in build 2023-10-05 22:44:09 +06:00
16 changed files with 185 additions and 35 deletions

View File

@@ -8,9 +8,9 @@ jobs:
steps:
- uses: actions/checkout@v2
- name: Set up JDK 11
- name: Set up JDK 17
uses: actions/setup-java@v1
with:
java-version: 11
java-version: 17
- name: Build with Gradle
run: ./gradlew build

View File

@@ -14,6 +14,10 @@ jobs:
steps:
- name: Checkout repository
uses: actions/checkout@v2
- name: Set up JDK 17
uses: actions/setup-java@v1
with:
java-version: 17
- name: Log into registry
uses: docker/login-action@28218f9b04b4f3f62068d7b6ce6ca5b26e35336c
with:

View File

@@ -1,4 +1,4 @@
FROM adoptopenjdk/openjdk11
FROM bellsoft/liberica-openjdk-alpine:19
ADD ./build/distributions/booru_grabber_bot.tar /
RUN chown -R 1000:1000 "/booru_grabber_bot"

View File

@@ -33,6 +33,7 @@ dependencies {
implementation libs.krontab
implementation libs.psql
implementation libs.imageboard
implementation libs.ktor.client.okhttp
}
application {
@@ -40,6 +41,6 @@ application {
}
java {
sourceCompatibility = JavaVersion.VERSION_11
targetCompatibility = JavaVersion.VERSION_11
sourceCompatibility = JavaVersion.VERSION_17
targetCompatibility = JavaVersion.VERSION_17
}

View File

@@ -10,3 +10,8 @@ services:
POSTGRES_DB: "test"
ports:
- "8092:5432"
# booru_grabber_bot:
# image: insanusmokrassar/booru_grabber_bot
# container_name: "booru_grabber_bot"
# volumes:
# - "path_to_file:/booru_grabber_bot/config.json:ro"

View File

@@ -4,4 +4,4 @@ org.gradle.parallel=true
kotlin.js.generate.externals=true
kotlin.incremental=true
version=0.0.3
docker_version=0.0.5

View File

@@ -1,14 +1,15 @@
[versions]
kotlin = "1.9.10"
tgbotapi = "9.2.1"
microutils = "0.20.6"
imageboard = "2.6.0"
krontab = "2.2.1"
kslog = "1.2.1"
exposed = "0.43.0"
psql = "42.6.0"
clikt = "4.2.1"
kotlin = "2.0.21"
tgbotapi = "28.0.2"
microutils = "0.23.0"
imageboard = "2.7.0"
krontab = "2.6.1"
kslog = "1.3.6"
ktor = "3.0.1"
exposed = "0.55.0"
psql = "42.7.4"
clikt = "5.0.1"
[libraries]
@@ -23,6 +24,8 @@ kslog = { module = "dev.inmo:kslog", version.ref = "kslog" }
exposed = { module = "org.jetbrains.exposed:exposed-jdbc", version.ref = "exposed" }
psql = { module = "org.postgresql:postgresql", version.ref = "psql" }
ktor-client-okhttp = { module = "io.ktor:ktor-client-okhttp", version.ref = "ktor" }
imageboard = { module = "com.github.Kodehawa:imageboard-api", version.ref = "imageboard" }
clikt = { module = "com.github.ajalt.clikt:clikt", version.ref = "clikt" }

View File

@@ -1,5 +1,5 @@
distributionBase=GRADLE_USER_HOME
distributionPath=wrapper/dists
distributionUrl=https\://services.gradle.org/distributions/gradle-7.6.1-bin.zip
distributionUrl=https\://services.gradle.org/distributions/gradle-8.10.2-bin.zip
zipStoreBase=GRADLE_USER_HOME
zipStorePath=wrapper/dists

View File

@@ -14,7 +14,7 @@ function assert_success() {
}
app=booru_grabber_bot
version="`grep ./gradle.properties -e "^version=" | sed -e "s/version=\(.*\)/\1/"`"
version="`grep ./gradle.properties -e "^docker_version=" | sed -e "s/docker_version=\(.*\)/\1/"`"
server=insanusmokrassar
assert_success docker build -t $app:"$version" .

6
renovate.json Normal file
View File

@@ -0,0 +1,6 @@
{
"$schema": "https://docs.renovatebot.com/renovate-schema.json",
"extends": [
"config:recommended"
]
}

View File

@@ -1,3 +1 @@
rootProject.name = 'booru_grabber_bot'
enableFeaturePreview("VERSION_CATALOGS")

View File

@@ -1,3 +1,4 @@
import com.github.ajalt.clikt.core.parse
import dev.inmo.krontab.utils.asFlowWithDelays
import dev.inmo.micro_utils.coroutines.*
import dev.inmo.micro_utils.pagination.utils.doForAllWithNextPaging
@@ -6,12 +7,12 @@ import dev.inmo.micro_utils.repos.cache.full.fullyCached
import dev.inmo.micro_utils.repos.exposed.keyvalue.ExposedKeyValueRepo
import dev.inmo.micro_utils.repos.exposed.onetomany.ExposedKeyValuesRepo
import dev.inmo.micro_utils.repos.mappers.withMapper
import dev.inmo.tgbotapi.bot.ktor.telegramBot
import dev.inmo.tgbotapi.extensions.api.bot.getMe
import dev.inmo.tgbotapi.extensions.api.bot.setMyCommands
import dev.inmo.tgbotapi.extensions.api.delete
import dev.inmo.tgbotapi.extensions.api.send.media.*
import dev.inmo.tgbotapi.extensions.api.send.reply
import dev.inmo.tgbotapi.extensions.api.telegramBot
import dev.inmo.tgbotapi.extensions.behaviour_builder.buildBehaviourWithLongPolling
import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onCommand
import dev.inmo.tgbotapi.requests.abstracts.FileUrl
@@ -20,6 +21,7 @@ import dev.inmo.tgbotapi.types.chat.ChannelChat
import dev.inmo.tgbotapi.types.chat.PrivateChat
import dev.inmo.tgbotapi.types.media.TelegramMediaPhoto
import dev.inmo.tgbotapi.utils.code
import io.ktor.client.engine.okhttp.*
import java.io.File
import kotlinx.coroutines.*
import kotlinx.coroutines.sync.Mutex
@@ -29,6 +31,9 @@ import models.Config
import net.kodehawa.lib.imageboards.ImageBoard
import net.kodehawa.lib.imageboards.boards.DefaultBoards
import net.kodehawa.lib.imageboards.entities.BoardImage
import okhttp3.OkHttpClient
internal lateinit var InternalBoards: Boards
/**
* This method by default expects one argument in [args] field: telegram bot configuration
@@ -39,7 +44,14 @@ suspend fun main(args: Array<String>) {
// decode config
val config: Config = json.decodeFromString(Config.serializer(), File(args.first()).readText())
// that is your bot
val bot = telegramBot(config.token)
val bot = telegramBot(config.token, OkHttp) {
config.client ?.apply { setupConfig() }
}
InternalBoards = Boards(
OkHttpClient.Builder().apply {
config.client ?.apply { setupConfig() }
}.build()
)
ImageBoard.setUserAgent("WhoAmI?")
@@ -52,9 +64,9 @@ suspend fun main(args: Array<String>) {
{ text("config") },
"configs"
).withMapper(
{ chatId },
{ chatId.long },
{ json.encodeToString(ChatSettings.serializer(), this) },
{ ChatId(this) },
{ ChatId(RawChatId(this)) },
{ json.decodeFromString(ChatSettings.serializer(), this) },
).fullyCached(scope = scope)
@@ -64,9 +76,9 @@ suspend fun main(args: Array<String>) {
{ text("url") },
"chatsUrlsSeen"
).withMapper(
{ chatId },
{ chatId.long },
{ this },
{ ChatId(this) },
{ ChatId(RawChatId(this)) },
{ this },
)

25
src/main/kotlin/Boards.kt Normal file
View File

@@ -0,0 +1,25 @@
import net.kodehawa.lib.imageboards.ImageBoard
import net.kodehawa.lib.imageboards.boards.DefaultBoards
import net.kodehawa.lib.imageboards.entities.impl.*
import okhttp3.OkHttpClient
import java.util.concurrent.TimeUnit
class Boards(
private val client: OkHttpClient = OkHttpClient.Builder()
.connectTimeout(3, TimeUnit.SECONDS)
.readTimeout(3, TimeUnit.SECONDS)
.build()
) {
val E621 = ImageBoard(client, DefaultBoards.E621, FurryImage::class.java)
val KONACHAN = ImageBoard(client, DefaultBoards.KONACHAN, KonachanImage::class.java)
val RULE34 = ImageBoard(client, DefaultBoards.R34, Rule34Image::class.java)
val YANDERE = ImageBoard(client, DefaultBoards.YANDERE, YandereImage::class.java)
val DANBOORU = ImageBoard(client, DefaultBoards.DANBOORU, DanbooruImage::class.java)
val SAFEBOORU = ImageBoard(
client, DefaultBoards.SAFEBOORU,
SafebooruImage::class.java
)
val E926 = ImageBoard(client, DefaultBoards.E926, SafeFurryImage::class.java)
val GELBOORU = ImageBoard(client, DefaultBoards.GELBOORU, GelbooruImage::class.java)
}

View File

@@ -6,7 +6,6 @@ import kotlinx.serialization.builtins.serializer
import kotlinx.serialization.descriptors.SerialDescriptor
import kotlinx.serialization.encoding.Decoder
import kotlinx.serialization.encoding.Encoder
import net.kodehawa.lib.imageboards.DefaultImageBoards
import net.kodehawa.lib.imageboards.ImageBoard
import net.kodehawa.lib.imageboards.boards.DefaultBoards
import net.kodehawa.lib.imageboards.entities.BoardImage
@@ -29,14 +28,14 @@ data class ChatSettings(
val board: ImageBoard<*>
get() = when (boardBase) {
DefaultBoards.R34 -> DefaultImageBoards.RULE34
DefaultBoards.E621 -> DefaultImageBoards.E621
DefaultBoards.KONACHAN -> DefaultImageBoards.KONACHAN
DefaultBoards.YANDERE -> DefaultImageBoards.YANDERE
DefaultBoards.DANBOORU -> DefaultImageBoards.DANBOORU
DefaultBoards.SAFEBOORU -> DefaultImageBoards.SAFEBOORU
DefaultBoards.GELBOORU -> DefaultImageBoards.GELBOORU
DefaultBoards.E926 -> DefaultImageBoards.E926
DefaultBoards.R34 -> InternalBoards.RULE34
DefaultBoards.E621 -> InternalBoards.E621
DefaultBoards.KONACHAN -> InternalBoards.KONACHAN
DefaultBoards.YANDERE -> InternalBoards.YANDERE
DefaultBoards.DANBOORU -> InternalBoards.DANBOORU
DefaultBoards.SAFEBOORU -> InternalBoards.SAFEBOORU
DefaultBoards.GELBOORU -> InternalBoards.GELBOORU
DefaultBoards.E926 -> InternalBoards.E926
}
suspend fun makeRequest(page: Int): List<BoardImage> {

View File

@@ -5,5 +5,6 @@ import kotlinx.serialization.Serializable
@Serializable
data class Config(
val token: String,
val database: DatabaseConfig
val database: DatabaseConfig,
val client: HttpClientConfig? = null
)

View File

@@ -0,0 +1,96 @@
package models
import dev.inmo.tgbotapi.types.MilliSeconds
import io.ktor.client.HttpClientConfig
import io.ktor.client.engine.okhttp.*
import io.ktor.client.plugins.*
import io.ktor.client.request.*
import io.ktor.http.*
import kotlinx.serialization.Serializable
import okhttp3.OkHttpClient
import java.net.*
import java.util.*
import java.util.concurrent.TimeUnit
@Serializable
data class HttpClientConfig(
val connectionTimeoutMillis: MilliSeconds? = null,
val requestTimeoutMillis: MilliSeconds? = null,
val responseTimeoutMillis: MilliSeconds? = null,
val proxy: ProxyConfig? = null
) {
@Serializable
data class ProxyConfig(
val hostname: String,
val type: ProxyType = ProxyType.socks,
val port: Int = type.defaultPort,
val username: String? = null,
val password: String? = null
) {
@Serializable
enum class ProxyType(val defaultPort: Int, val proxyType: Proxy.Type) {
socks(1080, Proxy.Type.SOCKS),
http(3128, Proxy.Type.HTTP),
}
val socketAddress
get() = InetSocketAddress(hostname, port)
}
fun OkHttpClient.Builder.setupConfig() {
// setting up connection timeout millis
connectionTimeoutMillis ?.let { connectTimeout(it, TimeUnit.MILLISECONDS) }
// setting up write timeout millis
requestTimeoutMillis ?.let { writeTimeout(it, TimeUnit.MILLISECONDS) }
// setting up read timeout millis
responseTimeoutMillis ?.let { readTimeout(it, TimeUnit.MILLISECONDS) }
// Start setup bot client engine proxy
this@HttpClientConfig.proxy ?.let { proxyConfig ->
proxy(
Proxy(
proxyConfig.type.proxyType,
proxyConfig.socketAddress
)
)
proxyConfig.username ?.let { username ->
if (proxyConfig.type == ProxyConfig.ProxyType.socks) {
val passwordAuthentication = PasswordAuthentication(
username,
proxyConfig.password ?.toCharArray() ?: error("For Socks proxy you should use both username and password or do not use authentication at all")
)
Authenticator.setDefault(object : Authenticator() {
override fun getPasswordAuthentication(): PasswordAuthentication? {
return if (requestingHost.lowercase() == proxyConfig.hostname.lowercase()) {
passwordAuthentication
} else {
null
}
}
})
}
}
}
}
fun HttpClientConfig<OkHttpConfig>.setupConfig() {
// setting up telegram bot client
engine {
// Start setup bot client engine configuration
config {
setupConfig()
}
}
proxy ?.username ?.let { username ->
if (proxy.type == ProxyConfig.ProxyType.http) {
val passwordSuffix = proxy.password ?.let { ":$it" }
val credentials = Base64.getEncoder().encodeToString("${username}${passwordSuffix}".toByteArray())
this@setupConfig.defaultRequest {
header(HttpHeaders.ProxyAuthorization, "Basic $credentials")
}
}
}
}
}