From 53675ca59805a494923c889ceab44d95d4f91211 Mon Sep 17 00:00:00 2001
From: InsanusMokrassar <ovsyannikov.alexey95@gmail.com>
Date: Wed, 7 Sep 2022 00:23:14 +0600
Subject: [PATCH] fixes and fills

---
 gradle/libs.versions.toml                     |  2 +
 .../src/commonMain/kotlin/DefaultSelector.kt  |  5 ++-
 .../src/commonMain/kotlin/Selector.kt         |  3 +-
 .../kotlin/models/SelectorConfig.kt           |  5 ++-
 .../commonMain/kotlin/models/TimeConfig.kt    |  3 +-
 runner/config.json                            |  7 ++-
 settings.gradle                               |  1 +
 triggers/command/build.gradle                 |  1 +
 triggers/command/src/jvmMain/kotlin/Plugin.kt |  8 ++--
 triggers/selector_with_timer/build.gradle     | 24 +++++++++++
 .../src/commonMain/kotlin/PackageInfo.kt      |  1 +
 .../src/jvmMain/kotlin/Plugin.kt              | 43 +++++++++++++++++++
 .../src/main/AndroidManifest.xml              |  1 +
 13 files changed, 91 insertions(+), 13 deletions(-)
 create mode 100644 triggers/selector_with_timer/build.gradle
 create mode 100644 triggers/selector_with_timer/src/commonMain/kotlin/PackageInfo.kt
 create mode 100644 triggers/selector_with_timer/src/jvmMain/kotlin/Plugin.kt
 create mode 100644 triggers/selector_with_timer/src/main/AndroidManifest.xml

diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
index 62b9baf..15df308 100644
--- a/gradle/libs.versions.toml
+++ b/gradle/libs.versions.toml
@@ -7,6 +7,7 @@ plagubot = "2.3.0"
 tgbotapi = "3.2.0"
 microutils = "0.12.6"
 kslog = "0.5.1"
+krontab = "0.8.0"
 
 psql = "42.3.6"
 
@@ -38,6 +39,7 @@ microutils-repos-common = { module = "dev.inmo:micro_utils.repos.common", versio
 microutils-repos-exposed = { module = "dev.inmo:micro_utils.repos.exposed", version.ref = "microutils" }
 microutils-repos-cache = { module = "dev.inmo:micro_utils.repos.cache", version.ref = "microutils" }
 kslog = { module = "dev.inmo:kslog", version.ref = "kslog" }
+krontab = { module = "dev.inmo:krontab", version.ref = "krontab" }
 
 psql = { module = "org.postgresql:postgresql", version.ref = "psql" }
 
diff --git a/ratings/selector/src/commonMain/kotlin/DefaultSelector.kt b/ratings/selector/src/commonMain/kotlin/DefaultSelector.kt
index 482f74d..02b52ca 100644
--- a/ratings/selector/src/commonMain/kotlin/DefaultSelector.kt
+++ b/ratings/selector/src/commonMain/kotlin/DefaultSelector.kt
@@ -1,5 +1,6 @@
 package dev.inmo.plaguposter.ratings.selector
 
+import com.soywiz.klock.DateTime
 import dev.inmo.plaguposter.posts.models.PostId
 import dev.inmo.plaguposter.ratings.repo.RatingsRepo
 import dev.inmo.plaguposter.ratings.selector.models.SelectorConfig
@@ -8,11 +9,11 @@ class DefaultSelector (
     private val config: SelectorConfig,
     private val repo: RatingsRepo
 ) : Selector {
-    override suspend fun take(n: Int): List<PostId> {
+    override suspend fun take(n: Int, now: DateTime): List<PostId> {
         val result = mutableListOf<PostId>()
 
         do {
-            val selected = config.active ?.ratings ?.select(repo, result) ?: break
+            val selected = config.active(now.time) ?.ratings ?.select(repo, result) ?: break
             result.add(selected)
         } while (result.size < n)
 
diff --git a/ratings/selector/src/commonMain/kotlin/Selector.kt b/ratings/selector/src/commonMain/kotlin/Selector.kt
index c35d18e..813aa20 100644
--- a/ratings/selector/src/commonMain/kotlin/Selector.kt
+++ b/ratings/selector/src/commonMain/kotlin/Selector.kt
@@ -1,7 +1,8 @@
 package dev.inmo.plaguposter.ratings.selector
 
+import com.soywiz.klock.DateTime
 import dev.inmo.plaguposter.posts.models.PostId
 
 interface Selector {
-    suspend fun take(n: Int = 1): List<PostId>
+    suspend fun take(n: Int = 1, now: DateTime = DateTime.now()): List<PostId>
 }
diff --git a/ratings/selector/src/commonMain/kotlin/models/SelectorConfig.kt b/ratings/selector/src/commonMain/kotlin/models/SelectorConfig.kt
index c7909c0..da35cf5 100644
--- a/ratings/selector/src/commonMain/kotlin/models/SelectorConfig.kt
+++ b/ratings/selector/src/commonMain/kotlin/models/SelectorConfig.kt
@@ -1,11 +1,12 @@
 package dev.inmo.plaguposter.ratings.selector.models
 
+import com.soywiz.klock.DateTime
+import com.soywiz.klock.Time
 import kotlinx.serialization.Serializable
 
 @Serializable
 data class SelectorConfig(
     val items: List<SelectorConfigItem>
 ) {
-    val active: SelectorConfigItem?
-        get() = items.firstOrNull { it.time.isActive }
+    fun active(now: Time = DateTime.now().time) = items.firstOrNull { it.time.isActive(now) }
 }
diff --git a/ratings/selector/src/commonMain/kotlin/models/TimeConfig.kt b/ratings/selector/src/commonMain/kotlin/models/TimeConfig.kt
index 6937834..826b579 100644
--- a/ratings/selector/src/commonMain/kotlin/models/TimeConfig.kt
+++ b/ratings/selector/src/commonMain/kotlin/models/TimeConfig.kt
@@ -17,8 +17,7 @@ class TimeConfig(
     @Transient
     val range = from .. to
 
-    val isActive: Boolean
-        get() = DateTime.now().time in range
+    fun isActive(now: Time): Boolean = now in range
 
 
     object TimeSerializer : KSerializer<Time> {
diff --git a/runner/config.json b/runner/config.json
index e75da5b..7af286a 100644
--- a/runner/config.json
+++ b/runner/config.json
@@ -10,7 +10,9 @@
     "dev.inmo.plaguposter.posts.Plugin",
     "dev.inmo.plaguposter.posts.registrar.Plugin",
     "dev.inmo.plaguposter.ratings.Plugin",
-    "dev.inmo.plaguposter.ratings.source.Plugin"
+    "dev.inmo.plaguposter.ratings.source.Plugin",
+    "dev.inmo.plaguposter.ratings.selector.Plugin",
+    "dev.inmo.plaguposter.triggers.selector_with_timer.Plugin"
   ],
   "posts": {
     "targetChat": 12345678,
@@ -61,5 +63,8 @@
         }
       }
     ]
+  },
+  "timer_trigger": {
+    "krontab": "0 30 2/4 * *"
   }
 }
diff --git a/settings.gradle b/settings.gradle
index b09f9ae..c7199d7 100644
--- a/settings.gradle
+++ b/settings.gradle
@@ -8,6 +8,7 @@ String[] includes = [
     ":ratings:source",
     ":ratings:selector",
     ":triggers:command",
+    ":triggers:selector_with_timer",
 //    ":settings",
     ":runner"
 ]
diff --git a/triggers/command/build.gradle b/triggers/command/build.gradle
index 779ca41..4f8449e 100644
--- a/triggers/command/build.gradle
+++ b/triggers/command/build.gradle
@@ -12,6 +12,7 @@ kotlin {
             dependencies {
                 api project(":plaguposter.common")
                 api project(":plaguposter.posts")
+                api project(":plaguposter.ratings.selector")
             }
         }
         jvmMain {
diff --git a/triggers/command/src/jvmMain/kotlin/Plugin.kt b/triggers/command/src/jvmMain/kotlin/Plugin.kt
index f37ffe1..c12be69 100644
--- a/triggers/command/src/jvmMain/kotlin/Plugin.kt
+++ b/triggers/command/src/jvmMain/kotlin/Plugin.kt
@@ -6,6 +6,7 @@ import dev.inmo.plagubot.Plugin
 import dev.inmo.plaguposter.common.SuccessfulSymbol
 import dev.inmo.plaguposter.posts.repo.PostsRepo
 import dev.inmo.plaguposter.posts.sending.PostPublisher
+import dev.inmo.plaguposter.ratings.selector.Selector
 import dev.inmo.tgbotapi.extensions.api.edit.edit
 import dev.inmo.tgbotapi.extensions.api.send.reply
 import dev.inmo.tgbotapi.extensions.api.send.send
@@ -38,10 +39,7 @@ object Plugin : Plugin {
     override suspend fun BehaviourContextWithFSM<State>.setupBotPlugin(koin: Koin) {
         val postsRepo = koin.get<PostsRepo>()
         val publisher = koin.get<PostPublisher>()
-        strictlyOn { state: PublishState ->
-
-            null
-        }
+        val selector = koin.getOrNull<Selector>()
 
         onCommand("publish_post") {
             val messageInReply = it.replyTo ?.contentMessageOrNull() ?: let { _ ->
@@ -49,7 +47,7 @@ object Plugin : Plugin {
 
                 return@onCommand
             }
-            val postId = postsRepo.getIdByChatAndMessage(messageInReply.chat.id, messageInReply.messageId)
+            val postId = postsRepo.getIdByChatAndMessage(messageInReply.chat.id, messageInReply.messageId) ?: selector ?.take(1) ?.firstOrNull()
             if (postId == null) {
                 reply(
                     it,
diff --git a/triggers/selector_with_timer/build.gradle b/triggers/selector_with_timer/build.gradle
new file mode 100644
index 0000000..eadb025
--- /dev/null
+++ b/triggers/selector_with_timer/build.gradle
@@ -0,0 +1,24 @@
+plugins {
+    id "org.jetbrains.kotlin.multiplatform"
+    id "org.jetbrains.kotlin.plugin.serialization"
+    id "com.android.library"
+}
+
+apply from: "$mppProjectWithSerializationPresetPath"
+
+kotlin {
+    sourceSets {
+        commonMain {
+            dependencies {
+                api project(":plaguposter.common")
+                api project(":plaguposter.posts")
+                api project(":plaguposter.ratings.selector")
+                api libs.krontab
+            }
+        }
+        jvmMain {
+            dependencies {
+            }
+        }
+    }
+}
diff --git a/triggers/selector_with_timer/src/commonMain/kotlin/PackageInfo.kt b/triggers/selector_with_timer/src/commonMain/kotlin/PackageInfo.kt
new file mode 100644
index 0000000..349c445
--- /dev/null
+++ b/triggers/selector_with_timer/src/commonMain/kotlin/PackageInfo.kt
@@ -0,0 +1 @@
+package dev.inmo.plaguposter.triggers.selector_with_timer
diff --git a/triggers/selector_with_timer/src/jvmMain/kotlin/Plugin.kt b/triggers/selector_with_timer/src/jvmMain/kotlin/Plugin.kt
new file mode 100644
index 0000000..22e5dc3
--- /dev/null
+++ b/triggers/selector_with_timer/src/jvmMain/kotlin/Plugin.kt
@@ -0,0 +1,43 @@
+package dev.inmo.plaguposter.triggers.selector_with_timer
+
+import dev.inmo.krontab.KrontabTemplate
+import dev.inmo.krontab.toSchedule
+import dev.inmo.krontab.utils.asFlow
+import dev.inmo.micro_utils.coroutines.subscribeSafelyWithoutExceptions
+import dev.inmo.plagubot.Plugin
+import dev.inmo.plaguposter.posts.sending.PostPublisher
+import dev.inmo.plaguposter.ratings.selector.Selector
+import dev.inmo.tgbotapi.extensions.behaviour_builder.BehaviourContext
+import kotlinx.coroutines.FlowPreview
+import kotlinx.serialization.*
+import kotlinx.serialization.json.*
+import org.jetbrains.exposed.sql.Database
+import org.koin.core.Koin
+import org.koin.core.module.Module
+
+object Plugin : Plugin {
+    @Serializable
+    internal data class Config(
+        @SerialName("krontab")
+        val krontabTemplate: KrontabTemplate
+    ) {
+        @Transient
+        val krontab by lazy {
+            krontabTemplate.toSchedule()
+        }
+    }
+    override fun Module.setupDI(database: Database, params: JsonObject) {
+        single { get<Json>().decodeFromJsonElement(Config.serializer(), params["timer_trigger"] ?: return@single null) }
+    }
+
+    @OptIn(FlowPreview::class)
+    override suspend fun BehaviourContext.setupBotPlugin(koin: Koin) {
+        val publisher = koin.get<PostPublisher>()
+        val selector = koin.get<Selector>()
+        koin.get<Config>().krontab.asFlow().subscribeSafelyWithoutExceptions(this) {
+            selector.take(now = it).forEach { postId ->
+                publisher.publish(postId)
+            }
+        }
+    }
+}
diff --git a/triggers/selector_with_timer/src/main/AndroidManifest.xml b/triggers/selector_with_timer/src/main/AndroidManifest.xml
new file mode 100644
index 0000000..eba3bbd
--- /dev/null
+++ b/triggers/selector_with_timer/src/main/AndroidManifest.xml
@@ -0,0 +1 @@
+<manifest package="dev.inmo.plaguposter.triggers.selector_with_timer"/>