koishi-plugin-oni-sync-bot 0.8.8 → 0.9.0

package/lib/index.js

@@ -503,6 +503,20 @@ var RouteRedirect = class {
       )}`;
       router.redirect(targetUrl);
     });
+    ctx.server.use("/ggwiki", async (router, next) => {
+      const fullPath = router.path.replace("/ggwiki", "") || "";
+      const decodedPath = decodeURIComponent(fullPath);
+      if (decodedPath.includes("..")) {
+        return router.body = "❌ 非法路径访问！";
+      }
+      if (!/^(\/[\p{L}\p{N}_\-:.%]+)*\/?$/u.test(fullPath)) {
+        return router.body = "❌ 路径包含非法字符！";
+      }
+      const queryString = router.querystring ? `?${router.querystring}` : "";
+      const targetUrl = `https://${this.config.main_site}${fullPath}${queryString}`;
+      router.redirect(targetUrl);
+      return;
+    });
   }
 };
 ((RouteRedirect2) => {
@@ -1839,45 +1853,65 @@ bwiki: https://${this.config.domain}/bw/${match.id}`;
 // src/plugins/updateCommands.ts
 var import_koishi10 = require("koishi");
 var UpdateCommands = class {
+  constructor(ctx, config) {
+    this.ctx = ctx;
+    this.config = config;
+    this.log = ctx.logger("oni-sync");
+    this.registerCommands(ctx);
+    this.registerCronJobs(ctx);
+  }
   static {
     __name(this, "UpdateCommands");
   }
-  static inject = ["database", "wikiBot"];
+  static inject = ["database", "wikiBot", "cron"];
   config;
   log;
-  constructor(ctx, config) {
-    this.config = config;
-    this.log = ctx.logger("oni-sync");
-    this.registerCommands(ctx);
+  registerCronJobs(ctx) {
+    const cronExpression = this.config.updateCron || "0 2 * * *";
+    ctx.cron(cronExpression, async () => {
+      if (!this.ctx.wikiBot.isReady) {
+        this.log.warn("定时任务：WikiBot 服务未就绪，跳过页面缓存更新");
+        return;
+      }
+      this.log.info("定时任务：开始每日自动更新本地页面缓存（主站）");
+      try {
+        const result = await this.updatePageCache();
+        this.log.info(`定时任务完成：更新 ${result.count} 条页面到本地缓存`);
+      } catch (err) {
+        this.log.error("定时任务失败：自动更新页面缓存", err);
+      }
+    });
+  }
+  async updatePageCache() {
+    const res = await this.ctx.wikiBot.getGGBot().request({
+      action: "query",
+      list: "allpages",
+      format: "json",
+      aplimit: "max"
+    });
+    logger.info("主站页面查询成功");
+    const pages = res.query.allpages || [];
+    const pageData = pages.map((page) => {
+      const { pinyin_full, pinyin_first } = generatePinyinInfo(page.title);
+      return {
+        id: page.pageid,
+        title: page.title,
+        pinyin_full,
+        pinyin_first
+      };
+    });
+    if (pageData.length > 0) {
+      await this.ctx.database.upsert("wikipages", pageData);
+    }
+    logger.info(`检索到 ${pages.length} 个页面，已更新至数据库`);
+    return { count: pages.length };
   }
   registerCommands(ctx) {
     ctx.command("update", "更新本地页面缓存（主站）", { authority: 2 }).action(async ({ session }) => {
       await session.execute("update.status");
       try {
-        const res = await ctx.wikiBot.getGGBot().request({
-          action: "query",
-          list: "allpages",
-          format: "json",
-          aplimit: "max"
-        });
-        logger.info("主站页面查询成功");
-        const pages = res.query.allpages || [];
-        const pageData = pages.map((page) => {
-          const { pinyin_full, pinyin_first } = generatePinyinInfo(
-            page.title
-          );
-          return {
-            id: page.pageid,
-            title: page.title,
-            pinyin_full,
-            pinyin_first
-          };
-        });
-        if (pageData.length > 0) {
-          await ctx.database.upsert("wikipages", pageData);
-        }
-        session.send(`✅ 检索到 ${pages.length} 个页面，已更新至数据库`);
-        logger.info(`检索到 ${pages.length} 个页面，已更新至数据库`);
+        const result = await this.updatePageCache();
+        session.send(`✅ 检索到 ${result.count} 个页面，已更新至数据库`);
       } catch (err) {
         this.log.error("主站缓存更新失败", err);
         session.send("❌ 主站缓存更新失败，请联系管理员查看日志");
@@ -1950,7 +1984,8 @@ var UpdateCommands = class {
 };
 ((UpdateCommands2) => {
   UpdateCommands2.Config = import_koishi10.Schema.object({
-    logsUrl: import_koishi10.Schema.string().description("日志查看地址").default("https://klei.vip/onilogs")
+    logsUrl: import_koishi10.Schema.string().description("日志查看地址").default("https://klei.vip/onilogs"),
+    updateCron: import_koishi10.Schema.string().description("页面缓存自动更新 cron 表达式（默认每天凌晨 2 点）").default("0 2 * * *")
   });
 })(UpdateCommands || (UpdateCommands = {}));
 

package/lib/plugins/updateCommands.d.ts

@@ -1,12 +1,16 @@
 import { Context, Logger, Schema } from "koishi";
 export interface UpdateCommandsConfig {
     logsUrl: string;
+    updateCron?: string;
 }
 export declare class UpdateCommands {
+    ctx: Context;
     static readonly inject: string[];
     config: UpdateCommandsConfig;
     log: Logger;
     constructor(ctx: Context, config: UpdateCommandsConfig);
+    private registerCronJobs;
+    private updatePageCache;
     private registerCommands;
 }
 export declare namespace UpdateCommands {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-oni-sync-bot",
   "description": "缺氧Wiki站镜像点同步-测试",
-  "version": "0.8.8",
+  "version": "0.9.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -30,10 +30,10 @@
   },
   "dependencies": {
     "@types/node-fetch": "^2.6.13",
-    "form-data": "^4.0.5",
+    "form-data": "^4.0.6",
     "koishi-plugin-cron": "^3.1.0",
-    "mwn": "^3.0.1",
+    "mwn": "^3.0.3",
     "node-fetch": "^3.3.2",
-    "pinyin-pro": "^3.28.0"
+    "pinyin-pro": "^3.28.1"
   }
 }