feat: GtiHubReleasePluginUpdater

Author: cssxsh

src/main/kotlin/io/github/gnuf0rce/mirai/github/GitHubHelperPlugin.kt

@@ -87,6 +87,10 @@ public object GitHubHelperPlugin : KotlinPlugin(
                 regex findingReply replier
             }
         }
+        val target = resolveConfigFile("update.dict.json")
+        logger.info { "1.3.0 起提供从 github 更新<其他插件>的功能, 如有需要, 请编辑:\n $target" }
+        GitHubReleasePluginUpdater.reload(target)
+        GitHubReleasePluginUpdater.update()
     }
 
     override fun onDisable() {

src/main/kotlin/io/github/gnuf0rce/mirai/github/GitHubReleasePluginUpdater.kt

@@ -0,0 +1,163 @@
+/*
+ * Copyright 2021-2023 dsstudio Technologies and contributors.
+ *
+ *  此源代码的使用受 GNU AFFERO GENERAL PUBLIC LICENSE version 3 许可证的约束, 可以在以下链接找到该许可证.
+ *  Use of this source code is governed by the GNU AGPLv3 license that can be found through the following link.
+ *
+ *  https://github.com/gnuf0rce/github-helper/blob/master/LICENSE
+ */
+
+
+package io.github.gnuf0rce.mirai.github
+
+import io.github.gnuf0rce.github.*
+import io.ktor.client.request.*
+import io.ktor.client.statement.*
+import io.ktor.util.cio.*
+import io.ktor.utils.io.*
+import kotlinx.coroutines.*
+import kotlinx.serialization.*
+import net.mamoe.mirai.console.plugin.*
+import net.mamoe.mirai.console.plugin.jvm.*
+import net.mamoe.mirai.console.util.*
+import java.io.File
+
+public object GitHubReleasePluginUpdater {
+
+    @JvmStatic
+    public val dict: MutableMap<String, String> = hashMapOf(
+        "cn.whitrayhb.grasspics" to "NLR-DevTeam/GrassPictures",
+
+        "com.evolvedghost.mirai.steamhelper.steamhelper" to "EvolvedGhost/Steamhelper",
+        "com.evolvedghost.mutegames" to "EvolvedGhost/MuteGames",
+
+        "com.happysnaker.HRobot" to "happysnaker/mirai-plugin-HRobot",
+
+        "com.hcyacg.bilibili" to "Nekoer/mirai-plugins-bilibili",
+        "com.hcyacg.github-notice" to "Nekoer/mirai-github-notice",
+        "com.hcyacg.novelai" to "Nekoer/mirai-plugins-novelai",
+        "com.hcyacg.pixiv" to "Nekoer/mirai-plugins-pixiv",
+
+        "com.hrs.kloping.AutoReply" to "Kloping/Mirai_Plugins_Auto_Reply",
+
+        "com.kasukusakura.mlss" to "KasukuSakura/mirai-login-solver-sakura",
+
+        "com.khjxiaogu.mirai.MiraiSongPlugin" to "khjxiaogu/MiraiSongPlugin",
+
+        "com.xtex.repeater" to "xtexChooser/mirai-repeater",
+
+        "io.github.samarium150.mirai.plugin.mirai-console-drift-bottle" to "Samarium150/mirai-console-drift-bottle",
+        "io.github.samarium150.mirai.plugin.mirai-console-loafers-calendar" to "Samarium150/mirai-console-loafers-calendar",
+        "io.github.samarium150.mirai.plugin.mirai-console-lolicon" to "Samarium150/mirai-console-lolicon",
+
+        "me.jie65535.mirai-console-jnr-plugin" to "jie65535/mirai-console-jnr-plugin",
+
+        "me.stageguard.obms.OsuMapSuggester" to "StageGuard/OsuMapSuggester",
+        "me.stageguard.sctimetable" to "StageGuard/SuperCourseTimetableBot",
+
+        "org.laolittle.plugin.SkikoMirai" to "LaoLittle/SkikoMirai",
+        "org.laolittle.plugin.draw.DrawMeme" to "LaoLittle/DrawMeme",
+
+        "top.colter.bilibili-dynamic-mirai-plugin" to "Colter23/bilibili-dynamic-mirai-plugin",
+        "top.colter.genshin-sign" to "Colter23/genshin-sign-mirai-plugin",
+
+        "top.cutestar.antiRecall" to "Pmaru-top/AntiRecall",
+
+        "top.limbang.mcmod" to "limbang/mirai-console-mcmod-plugin",
+        "top.limbang.mcsm" to "limbang/mirai-console-mcsm-plugin",
+        "top.limbang.minecraft" to "limbang/mirai-console-minecraft-plugin",
+
+        "top.jie65535.mirai.grasscutter-command" to "jie65535/JGrasscutterCommand",
+        "top.jie65535.mail-notify" to "jie65535/JMailNotify",
+        "top.jie65535.mirai-console-jms-plugin" to "jie65535/mirai-console-jms-plugin",
+        "top.jie65535.jcf" to "jie65535/mirai-console-jcf-plugin",
+        "top.jie65535.mirai-console-jcr-plugin" to "jie65535/mirai-console-jcr-plugin",
+        "top.jie65535.mirai-console-jhr-plugin" to "jie65535/mirai-console-jhr-plugin",
+        "top.jie65535.mirai-console-jcab-arg-plugin" to "jie65535/mirai-console-jcab-arg-plugin",
+
+        "xmmt.dituon.petpet" to "Dituon/petpet",
+
+        "xyz.cssxsh.mirai.plugin.novelai-helper" to "cssxsh/novelai-helper",
+        "xyz.cssxsh.mirai.plugin.pixiv-helper" to "cssxsh/pixiv-helper"
+    )
+
+    public fun reload(file: File) {
+        if (file.exists()) {
+            dict.clear()
+            dict.putAll(GitHubJson.decodeFromString(file.readText()))
+        } else {
+            file.writeText(GitHubJson.encodeToString(dict))
+            val last = file.lastModified()
+            runBlocking {
+                ConsoleInput.requestInput(hint = "update.dict.json 已生成，你可以打开浏览一下 (输入回车结束等待)")
+            }
+            if (last < file.lastModified()) {
+                dict.clear()
+                dict.putAll(GitHubJson.decodeFromString(file.readText()))
+            }
+        }
+    }
+
+    public fun update() {
+        for (plugin in PluginManager.plugins) {
+            if (plugin !is JvmPlugin) continue
+            val id = dict[plugin.description.id]
+                ?: dict[plugin.description.name]
+                ?: continue
+            val classLoader = plugin::class.java.classLoader as? java.net.URLClassLoader ?: continue
+            val source = classLoader
+                .urLs.singleOrNull()
+                ?.let { File(it.path) }
+                ?: continue
+            var needUpdate = false
+            var target = PluginManager.pluginsFolder.resolve(plugin.description.id)
+
+            plugin.launch {
+                val latest = github.repo(id).releases.latest()
+                val jar = latest.assets.find { it.name.endsWith(".mirai2.jar") }
+                    ?: latest.assets.find { it.name.endsWith(".mirai.jar") }
+                    ?: latest.assets.find { it.name.endsWith(".jar") }
+                    ?: return@launch
+                val updated = jar.updatedAt.toInstant().toEpochMilli()
+                needUpdate = try {
+                    SemVersion(latest.tagName.removeSuffix("v")) > plugin.description.version
+                } catch (_: IllegalArgumentException) {
+                    updated > source.lastModified()
+                }
+
+                if (needUpdate.not()) return@launch
+
+                plugin.logger.info("从 ${latest.htmlUrl} 尝试升级")
+                target = PluginManager.pluginsFolder.resolve(jar.name)
+                github.useHttpClient { http ->
+                    http.get(jar.browserDownloadUrl)
+                        .bodyAsChannel()
+                        .copyAndClose(target.writeChannel())
+                }
+                check(target.length() == jar.size) {
+                    target.delete()
+                    "从 ${jar.browserDownloadUrl} 下载失败(文件大小校验失败 ${target.length()}!=${jar.size})"
+                }
+                target.setLastModified(updated)
+                source.delete()
+                plugin.logger.info("从 ${latest.htmlUrl} 升级成功")
+
+            }.invokeOnCompletion { cause ->
+                if (cause != null) {
+                    plugin.logger.warning("从 $id 升级失败")
+                    target.deleteOnExit()
+                    Runtime.getRuntime().addShutdownHook(Thread {
+                        target.delete()
+                    })
+                } else if (needUpdate && source.exists()) {
+                    plugin.logger.warning("旧版插件 ${source.name} 删除失败，将尝试添加退出时删除，请在下次启动时手动检查")
+                    source.deleteOnExit()
+                    Runtime.getRuntime().addShutdownHook(Thread {
+                        classLoader.close()
+                        source.delete()
+                    })
+                }
+            }
+        }
+    }
+}