koishi-plugin-oni-wiki-qq 0.4.7 → 0.4.9

package/lib/index.d.ts

@@ -1,6 +1,6 @@
 import { Context, Schema } from "koishi";
 export declare const name = "oni-wiki-qq";
-export declare const usage = "\n  - 0.4.6 \u79FB\u9664\u6CA1\u5FC5\u8981\u7684\u529F\u80FD\n  - 0.4.5 \u68C0\u6D4B\u6559\u7A0B\u9875\u9762\n";
+export declare const usage = "\n  - 0.4.9 \u6DFB\u52A0\u91CD\u5B9A\u5411\u529F\u80FD\n  - 0.4.8 \u91CD\u542Fbwiki\u66F4\u65B0\n  - 0.4.6 \u79FB\u9664\u6CA1\u5FC5\u8981\u7684\u529F\u80FD\n  - 0.4.5 \u68C0\u6D4B\u6559\u7A0B\u9875\u9762\n";
 export declare const inject: string[];
 declare module "koishi" {
     interface Tables {
@@ -10,10 +10,12 @@ declare module "koishi" {
 export interface WikiPages {
     id: number;
     title: string;
+    redirect: string;
 }
 export interface Config {
     bot_username: string;
     bot_password: string;
+    bwiki_session: string;
 }
 export declare const Config: Schema<Config>;
 export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -31,20 +31,40 @@ var import_koishi = require("koishi");
 var import_mwn = require("mwn");
 var name = "oni-wiki-qq";
 var usage = `
+  - 0.4.9 添加重定向功能
+  - 0.4.8 重启bwiki更新
   - 0.4.6 移除没必要的功能
   - 0.4.5 检测教程页面
 `;
 var inject = ["database"];
 var Config = import_koishi.Schema.object({
   bot_username: import_koishi.Schema.string().description("机器人用户名"),
-  bot_password: import_koishi.Schema.string().description("机器人密码")
+  bot_password: import_koishi.Schema.string().description("机器人密码"),
+  bwiki_session: import_koishi.Schema.string().description("bwiki的session,无法连接到gg时使用")
 });
 function apply(ctx, config) {
   const logger = ctx.logger;
   let wikibot;
   ctx.model.extend("wikipages", {
     id: "integer",
-    title: "string"
+    title: "string",
+    redirect: "string"
+  });
+  ctx.on("ready", async () => {
+    wikibot = new import_mwn.Mwn({
+      apiUrl: "https://oxygennotincluded.wiki.gg/zh/api.php",
+      username: config.bot_username,
+      password: config.bot_password,
+      userAgent: "Charles`Bot/2.1",
+      defaultParams: {
+        assert: "user"
+      }
+    });
+    wikibot.login().then(() => {
+      logger.info("Wiki机器人登录成功");
+    }).catch((err) => {
+      logger.error("Wiki机器人登录失败", err);
+    });
   });
   ctx.command("x <itemName>", "查询缺氧中文wiki").alias("/查wiki").action(async ({ session }, itemName = "电解器") => {
     if (/教程/.test(itemName)) {
@@ -55,31 +75,58 @@ function apply(ctx, config) {
 镜像站:  http://klei.vip/oni/usiz6d/${encodeURI(`教程`)}`;
     }
     const res = await ctx.database.get("wikipages", {
-      title: [`${itemName}`]
+      $or: [
+        {
+          title: itemName
+        },
+        {
+          redirect: itemName
+        }
+      ]
     });
     if (res.length == 0) {
       return `在Wiki里没找到,如有需要,请按照游戏内名称重新发起查询....`;
     }
+    const pageName = res[0].title;
     return `请点击链接前往站点查看:
 原站点:  http://oni.wiki/${encodeURI(
-      itemName
+      pageName
     )}
-镜像站:  http://klei.vip/oni/usiz6d/${encodeURI(itemName)}`;
+镜像站:  http://klei.vip/oni/usiz6d/${encodeURI(pageName)}`;
   });
-  ctx.on("ready", async () => {
-    wikibot = new import_mwn.Mwn({
-      apiUrl: "https://oxygennotincluded.wiki.gg/zh/api.php",
-      username: config.bot_username,
-      password: config.bot_password,
-      userAgent: "Charles`Bot/2.1",
-      defaultParams: {
-        assert: "user"
-      }
+  ctx.command("x.redirect <itemName:string> <redirect:string>", "物品重定向", { authority: 2 }).alias("重定向").action(async ({ session }, itemName, redirect) => {
+    if (!itemName || !redirect) {
+      return `请提供物品名称或重定向链接，格式如下：
+重定向 页面名称 重定向到页面物品名称`;
+    }
+    const database_res = await ctx.database.get("wikipages", {
+      $or: [
+        {
+          title: itemName
+        },
+        {
+          redirect: itemName
+        }
+      ]
     });
-    wikibot.login().then(() => {
-      logger.info("Wiki机器人登录成功");
-    }).catch((err) => {
-      logger.error("Wiki机器人登录失败", err);
+    if (database_res.length == 0) {
+      return `在数据库里没找到 ${itemName},如有需要,请按照游戏内名称重新发起重定向或联系管理员`;
+    }
+    if (database_res[0].redirect) {
+      return `${itemName} 已存在重定向,如需修改请联系管理员`;
+    }
+    await ctx.database.upsert("wikipages", [{
+      id: database_res[0].id,
+      title: itemName,
+      redirect
+    }]).then(() => {
+      wikibot.save(`${itemName}`, `#REDIRECT [[${redirect}]]`).then(() => {
+        session.send(`已成功将 ${itemName} 重定向至 ${redirect}并写入GG原站点`);
+      }).catch((err) => {
+        logger.error("写入重定向至原站点失败：", err);
+        session.send("写入重定向至原站点失败,请联系管理员检查日志或自行前往原站点编辑");
+      });
+      return `已成功将 ${itemName} 重定向至 ${redirect} 并写入数据库`;
     });
   });
   ctx.command("update", "更新本地页面缓存", { authority: 2 }).action(async ({ session }) => {
@@ -107,6 +154,31 @@ function apply(ctx, config) {
     session.send(`已删除 ${count.removed} 条本地缓存`);
     logger.info(`已删除 ${count.removed} 条本地缓存`);
   });
+  ctx.command("update.bwiki", "使用bwiki的session更新本地页面缓存", { authority: 2 }).action(async ({ session }) => {
+    const headers = {
+      "Content-Type": "application/json",
+      "user-agent": "Charles'queryBot",
+      Cookie: `SESSDATA=${config.bwiki_session}`
+    };
+    const url = `https://wiki.biligame.com/oni/api.php?action=query&list=allpages&apnamespace=0&aplimit=5000&format=json`;
+    ctx.http.get(url, { headers }).then((res) => {
+      res["query"]["allpages"].forEach((page) => {
+        ctx.database.upsert("wikipages", () => [
+          { id: page.pageid, title: page.title }
+        ]);
+      });
+      session.send(`检索到 ${res["query"]["allpages"].length} 个页面，已尝试更新至数据库`);
+      logger.info(`检索到 ${res["query"]["allpages"].length} 个页面，已尝试更新至数据库`);
+    }).catch((err) => {
+      session.send("更新失败,请联系管理员检查日志");
+      logger.error("更新失败", err);
+    });
+  });
+  ctx.command("update.status", "查询本地页面缓存数量", { authority: 1 }).action(async ({ session }) => {
+    const count = await ctx.database.get("wikipages", {});
+    session.send(`数据库中缓存了 ${count.length} 条页面`);
+    logger.info(`数据库中缓存了 ${count.length} 条页面`);
+  });
 }
 __name(apply, "apply");
 // Annotate the CommonJS export names for ESM import in node:

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-oni-wiki-qq",
   "description": "缺氧wiki查询，自用",
-  "version": "0.4.7",
+  "version": "0.4.9",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [