add timers

Author: lightless233

build.gradle.kts

@@ -24,6 +24,8 @@ dependencies {
     implementation("org.apache.logging.log4j:log4j-api:2.13.1")
     implementation("org.apache.logging.log4j:log4j-core:2.13.1")
     implementation("org.apache.logging.log4j:log4j-slf4j-impl:2.13.1")
+
+    implementation("org.jetbrains.exposed", "exposed-jodatime", "0.23.1")
 }
 
 

izumi_config.template.yml

@@ -20,4 +20,9 @@ enabledCommands: [
 # 启用的 pipeline 列表
 enabledPipelines: [
   repeat
+]
+
+# 启用的 timer 列表
+enabledTimers: [
+  life
 ]

src/main/kotlin/me/lightless/izumi/Constant.kt

@@ -4,5 +4,6 @@ object Constant {
 
     const val COMMAND_PACKAGE_NAME = "me.lightless.izumi.plugin.command.impl"
     const val PIPELINE_PACKAGE_NAME = "me.lightless.izumi.plugin.pipeline.impl"
+    const val TIMER_PACKAGE_NAME = "me.lightless.izumi.plugin.timer.impl"
 
 }

src/main/kotlin/me/lightless/izumi/MainApp.kt

@@ -3,6 +3,7 @@ package me.lightless.izumi
 import kotlinx.coroutines.runBlocking
 import me.lightless.izumi.config.ConfigParser
 import me.lightless.izumi.core.Dispatcher
+import me.lightless.izumi.plugin.timer.TimerLoader
 import net.mamoe.mirai.BotFactory
 import net.mamoe.mirai.alsoLogin
 import net.mamoe.mirai.event.events.GroupMessageEvent
@@ -53,6 +54,7 @@ class MainApp {
         this.logger.info("MessageDispatcher done.")
 
         // 开启定时器
+        TimerLoader.loadAndStart()
 
         logger.info("izumi start finished.")
     }

src/main/kotlin/me/lightless/izumi/plugin/timer/ITimer.kt

@@ -0,0 +1,13 @@
+package me.lightless.izumi.plugin.timer
+
+interface ITimer {
+
+    val name: String
+
+    @Suppress("unused")
+    val daemon: Boolean get() = true
+
+    val period: Long
+
+    suspend fun process()
+}

src/main/kotlin/me/lightless/izumi/plugin/timer/TimerLoader.kt

@@ -0,0 +1,52 @@
+package me.lightless.izumi.plugin.timer
+
+import kotlinx.coroutines.GlobalScope
+import kotlinx.coroutines.async
+import me.lightless.izumi.ApplicationContext
+import me.lightless.izumi.Constant
+import me.lightless.izumi.util.BotLoader
+import org.slf4j.LoggerFactory
+import kotlin.reflect.full.createInstance
+
+object TimerLoader {
+
+    private val logger = LoggerFactory.getLogger(javaClass)
+    private val timers = mutableListOf<ITimer>()
+
+    suspend fun loadAndStart() {
+        val timerClass = BotLoader.load(Constant.TIMER_PACKAGE_NAME)
+        logger.debug("timer class name: $timerClass")
+
+        var allowedTimers = ApplicationContext.botConfig?.enabledTimers
+        if (allowedTimers == null) {
+            allowedTimers = emptyList()
+        }
+
+        timerClass.forEach {
+            try {
+                val p = Class.forName(it).kotlin.createInstance() as ITimer
+                if (p.name in allowedTimers) {
+                    timers.add(p)
+                }
+            } catch (e: ClassNotFoundException) {
+                logger.error("Can't register timer class: $it")
+            }
+        }
+        logger.debug("load finished. count: ${timers.size}, " +
+                "list: ${timers.joinToString { it.name }}")
+
+        // 启动所有的 timer，每个timer单独一个协程
+        // TODO 在 global scope 上开启一堆协程不是一个好事，先跑起来，下次一定改
+        timers.forEach {
+            GlobalScope.async {
+                try {
+                    it.process()
+                } catch (e: Exception) {
+                    logger.error("timer error, name: ${it.name}")
+                    e.printStackTrace()
+                }
+            }
+        }
+    }
+
+}

src/main/kotlin/me/lightless/izumi/plugin/timer/impl/Life.kt

@@ -0,0 +1,66 @@
+package me.lightless.izumi.plugin.timer.impl
+
+import kotlinx.coroutines.delay
+import kotlinx.coroutines.launch
+import me.lightless.izumi.ApplicationContext
+import me.lightless.izumi.plugin.timer.ITimer
+import net.mamoe.mirai.message.data.buildMessageChain
+import org.joda.time.DateTime
+import org.slf4j.LoggerFactory
+import java.math.BigDecimal
+import java.math.RoundingMode
+
+@Suppress("unused")
+class Life: ITimer {
+
+    private val logger = LoggerFactory.getLogger(javaClass)
+
+    override val name: String
+        get() = "life"
+
+    // 每分钟运行一次
+    override val period: Long
+        get() = 60 * 1000
+
+    override suspend fun process() {
+        logger.debug("$name start!")
+
+        val bot = ApplicationContext.bot
+        if (bot == null) {
+            logger.debug("[$name] bot instance is null!")
+            return
+        }
+        val groupNumber = ApplicationContext.botConfig?.allowedGroups ?: listOf()
+        logger.debug("groupNumber: $groupNumber")
+
+        while (true) {
+            val datetime = DateTime()
+            val h: Int = datetime.hourOfDay
+            val m: Int = datetime.minuteOfHour
+
+            // 每天早上 9 点 30 分，发送今年的进度
+            if (h == 9 && m == 30) {
+                val days = datetime.dayOfYear
+                val percent = if (datetime.year().isLeap) {
+                    BigDecimal(days / 366.00 * 100.00).setScale(2, RoundingMode.HALF_UP)
+                } else {
+                    BigDecimal(days / 365.00 * 100.00).setScale(2, RoundingMode.HALF_UP)
+                }
+
+                val message = "【虚度光阴小助手】 \n今天是 ${datetime.year} 年的第 $days 天，今年已经过了 $percent% 啦！"
+                bot.launch {
+                    groupNumber.forEach {
+                        bot.getGroup(it)?.sendMessage(buildMessageChain {
+                            add(message)
+                        })
+                    }
+                }
+                // 多 sleep 5 秒，防止同一分钟内发两次消息
+                delay(1000 * 65)
+            } else {
+                // 如果不是 9 点 30 分就直接 sleep
+                delay(1000 * 60)
+            }
+        }
+    }
+}