koishi-plugin-wikit-querier-remodified 0.0.1

package/lib/graphql/index.d.ts

@@ -0,0 +1,6 @@
+export declare const queries: {
+    titleQuery: string;
+    userQuery: string;
+    userGlobalQuery: string;
+    userRankQuery: string;
+};

package/lib/index.d.ts

@@ -0,0 +1,22 @@
+import { Context, Schema } from "koishi";
+declare module "koishi" {
+    interface Tables {
+        wikitQuerier: WikitQuerierTable;
+    }
+}
+interface WikitQuerierTable {
+    id?: number;
+    platform: string;
+    channelId: string;
+    defaultwiki: string;
+}
+export declare const name: string;
+export declare const inject: string[];
+export interface Config {
+    bannedUsers: string[];
+    bannedTitles: string[];
+    bannedTags: string[];
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;
+export {};

package/lib/index.js

@@ -0,0 +1,421 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name2 in all)
+    __defProp(target, name2, { get: all[name2], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.tsx
+var src_exports = {};
+__export(src_exports, {
+  Config: () => Config,
+  apply: () => apply,
+  inject: () => inject,
+  name: () => name
+});
+module.exports = __toCommonJS(src_exports);
+var import_koishi = require("koishi");
+
+// src/graphql/index.ts
+var gql = /* @__PURE__ */ __name((query, ...substitutions) => String.raw(query, ...substitutions), "gql");
+var queries = {
+  titleQuery: gql`
+    query titleQuery($anyBaseUrl: [String], $query: String) {
+      articles(wiki: $anyBaseUrl, titleKeyword: $query, page: 1, pageSize: 20) {
+        nodes {
+          title
+          url
+          author
+          rating
+        }
+        pageInfo {
+          total
+          page
+          pageSize
+          hasNextPage
+        }
+      }
+    }
+  `,
+  userQuery: gql`
+  query userQuery($query: String!, $baseUrl: String!) {
+    authorWikiRank(
+      wiki: $baseUrl
+      name: $query
+      by: RATING
+    ) {
+      rank
+      name
+      value
+    }
+
+    articles(
+      author: $query
+      wiki: [$baseUrl]
+    ) {
+      pageInfo {
+        total
+      }
+    }
+  }
+  `,
+  userGlobalQuery: gql`
+    query userGlobalQuery($query: String!) {
+      authorGlobalRank(
+        name: $query
+        by: RATING
+      ) {
+        rank
+        name
+        value
+      }
+      
+      articles(author: $query) {
+        pageInfo {
+          total
+        }
+      }
+    }
+  `,
+  userRankQuery: gql`
+    query userRankQuery($baseUrl: String) {
+      authorRanking(wiki: $baseUrl, by: RATING) {
+        rank
+        name
+        value
+      }
+    }
+  `
+};
+
+// src/lib.ts
+var apiList = [
+  "https://wikit.unitreaty.org/apiv1/graphql",
+  "https://wikittest.unitreaty.org/apiv1/graphql"
+];
+var wikiInfo = {
+  /* 维基名称格式："站点简写": { wiki: "Wikit里的Wiki全名" }, 例如："ubmh": { wiki: "ubmh" } */
+  "ubmh": { wiki: "ubmh" },
+  "scp-cloud": { wiki: "scp-wiki-cloud" },
+  "cloud": { wiki: "backroom-wiki-cn" },
+  "scr": { wiki: "scr-wiki" },
+  "dfc": { wiki: "deep-forest-club" },
+  "rule": { wiki: "rule-wiki" },
+  "as": { wiki: "asbackroom" },
+  "lm": { wiki: "lostmedia" },
+  "if": { wiki: "if-backrooms" },
+  "rpc": { wiki: "rpc-wiki-cn" },
+  "warma": { wiki: "warma-world" },
+  "fr": { wiki: "backrooms-split-library" },
+  "f": { wiki: "backrooms-f" }
+};
+function levenshtein(a, b) {
+  const matrix = Array.from({ length: a.length + 1 }, () => Array(b.length + 1).fill(0));
+  for (let i = 0; i <= a.length; i++) matrix[i][0] = i;
+  for (let j = 0; j <= b.length; j++) matrix[0][j] = j;
+  for (let i = 1; i <= a.length; i++) {
+    for (let j = 1; j <= b.length; j++) {
+      const cost = a[i - 1] === b[j - 1] ? 0 : 1;
+      matrix[i][j] = Math.min(
+        matrix[i - 1][j] + 1,
+        matrix[i][j - 1] + 1,
+        matrix[i - 1][j - 1] + cost
+      );
+    }
+  }
+  return matrix[a.length][b.length];
+}
+__name(levenshtein, "levenshtein");
+async function wikitApiRequest(param, name2, endpointIndex = 0, queryString) {
+  if (endpointIndex >= apiList.length) {
+    throw new Error("所有API端点均已尝试但均失败");
+  }
+  let variables = {};
+  const wikiLongName = wikiInfo[name2]?.wiki;
+  if (queryString.includes("query titleQuery")) {
+    variables = { query: param, anyBaseUrl: wikiLongName ? [wikiLongName] : null };
+  } else if (queryString.includes("query userQuery")) {
+    variables = { query: param, baseUrl: wikiLongName };
+  } else if (queryString.includes("query userRankQuery")) {
+    variables = { baseUrl: wikiLongName };
+  } else if (queryString.includes("query userGlobalQuery")) {
+    variables = { query: param };
+  }
+  try {
+    const response = await fetch(apiList[endpointIndex], {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ query: queryString, variables })
+    });
+    if (!response.ok) {
+      throw new Error(`请求失败，状态码: ${response.status}`);
+    }
+    const { data, errors } = await response.json();
+    if (errors && errors.length > 0) {
+      return await wikitApiRequest(param, name2, endpointIndex + 1, queryString);
+    }
+    if (queryString.includes("query titleQuery") && data.articles?.nodes?.length) {
+      const articles = data.articles.nodes;
+      const lowerParam = param.toLowerCase();
+      let bestArticle = null;
+      let minDistance = Infinity;
+      for (const article of articles) {
+        const distance = levenshtein(
+          lowerParam,
+          article.title.toLowerCase()
+        );
+        if (distance < minDistance) {
+          minDistance = distance;
+          bestArticle = article;
+        }
+      }
+      if (bestArticle) {
+        return {
+          articles: { nodes: [bestArticle] }
+        };
+      }
+      return { articles: { nodes: [] } };
+    }
+    return data;
+  } catch (error) {
+    if (endpointIndex < apiList.length - 1) {
+      return await wikitApiRequest(param, name2, endpointIndex + 1, queryString);
+    }
+    throw error;
+  }
+}
+__name(wikitApiRequest, "wikitApiRequest");
+
+// src/index.tsx
+var import_jsx_runtime = require("@satorijs/element/jsx-runtime");
+var name = "wikit-querier";
+var inject = ["database"];
+var Config = import_koishi.Schema.object({
+  bannedUsers: import_koishi.Schema.array(import_koishi.Schema.string()).description("禁止查询的用户列表"),
+  bannedTitles: import_koishi.Schema.array(import_koishi.Schema.string()).description("禁止查询的文章列表"),
+  bannedTags: import_koishi.Schema.array(import_koishi.Schema.string()).description("禁止查询的标签列表")
+}).description("禁止查询配置");
+function apply(ctx, config) {
+  ctx.model.extend("wikitQuerier", {
+    id: "unsigned",
+    platform: "string(64)",
+    channelId: "string(64)",
+    defaultwiki: "string(64)"
+  });
+  const normalizeUrl = /* @__PURE__ */ __name((url) => url.replace(/^https?:\/\/backrooms-wiki-cn.wikidot.com/, "https://brcn.backroomswiki.cn").replace(/^https?:\/\/scp-wiki-cn.wikidot.com/, "https://scpcn.backroomswiki.cn").replace(/^https?:\/\/([a-z]+-wiki-cn|nationarea)/, "https://$1"), "normalizeUrl");
+  const getDefaultwiki = /* @__PURE__ */ __name(async (session) => {
+    const platform = session.event.platform;
+    const channelId = session.event.channel.id;
+    const data = await ctx.database.get("wikitQuerier", {
+      platform,
+      channelId
+    });
+    if (data.length > 0) {
+      return data[0].defaultwiki;
+    }
+    return void 0;
+  }, "getDefaultwiki");
+  let cmd = ctx;
+  cmd.command("wikitre", "（重制版）作者及页面信息查询");
+  cmd.command("wikitre.about", "此插件的相关信息。").action(async (argv) => {
+    return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+      /* @__PURE__ */ (0, import_jsx_runtime.jsx)("quote", { id: argv.session.event.message.id }),
+      "此组件由 lestday233 基于 Wikit API 编写",
+      /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+      "修改自 https://github.com/Laimuslime/koishi-plugin-crom-querier-modified"
+    ] });
+  });
+  cmd.command("wikitre.suppost-list", "列出所有支持的网站及对应的地址。").alias("wikitre-list").alias("wikitre.list").action(async (argv) => {
+    const entries = Object.entries(wikiInfo);
+    if (entries.length === 0) return "当前没有配置任何维基信息。";
+    const lines = entries.map(([key, value]) => `${key} → https://${value.wiki}.wikidot.com/`);
+    return `支持的维基列表：
+${lines.join("\n")}`;
+  });
+  cmd.command("wikitre.default-wiki <维基名称:string>", "设置默认维基。").alias("wikitre-db").action(async (argv, wiki) => {
+    const platform = argv.session.event.platform;
+    const channelId = argv.session.event.channel.id;
+    if (!wiki || !Object.keys(wikiInfo).includes(wiki) || wiki === "all") {
+      return "维基名称不正确。";
+    }
+    ctx.database.upsert("wikitQuerier", [{ channelId, platform, defaultwiki: wiki }], ["platform", "channelId"]);
+    return `已将本群默认查询维基设置为: ${wiki}`;
+  });
+  cmd.command("wikitre.author <作者:string> [维基名称:string]", "查询作者信息。默认搜索所有支持的维基。").alias("wikitre-au").action(async (argv, author, wiki) => {
+    const isRankQuery = /^#[0-9]{1,15}$/.test(author);
+    const rankNumber = isRankQuery ? Number(author.slice(1)) : null;
+    let queryString = isRankQuery ? queries.userRankQuery : queries.userQuery;
+    const validwikies = ["all", ...Object.keys(wikiInfo)];
+    const authorName = wiki && !validwikies.includes(wiki) || !author ? validwikies.includes(argv.args.at(-1)) ? argv.args.slice(0, -1).join(" ") : argv.args.join(" ") : author;
+    const User = /* @__PURE__ */ __name(({ object }) => {
+      const dataArray = object.authorRanking ? object.authorRanking : object.authorGlobalRank ? [object.authorGlobalRank] : object.authorWikiRank ? [object.authorWikiRank] : [];
+      if (!dataArray || dataArray.length === 0) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("template", { children: "未找到用户。" });
+      }
+      let user;
+      if (rankNumber !== null) {
+        user = dataArray.find(
+          (u) => u.rank === rankNumber && !config.bannedUsers.includes(u.name)
+        );
+      } else {
+        user = dataArray.find(
+          (u) => !config.bannedUsers.includes(u.name)
+        );
+      }
+      if (!user) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("template", { children: "未找到用户。" });
+      }
+      const total = object.articles?.pageInfo?.total ?? "未知";
+      let average = "未知";
+      if (typeof total === "number" && total > 0) {
+        average = (user.value / total).toFixed(2);
+      } else if (total === 0) {
+        average = 0;
+      }
+      return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("quote", { id: argv.session.event.message.id }),
+        "🙎‍♂️ ",
+        user.name,
+        " (#",
+        user.rank,
+        ")",
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+        "⭐ 总分：",
+        user.value,
+        " | 📑 页面数：",
+        total,
+        " | 📈 平均分：",
+        average
+      ] });
+    }, "User");
+    try {
+      let finalwiki = wiki;
+      if (!finalwiki) {
+        finalwiki = await getDefaultwiki(argv.session);
+      }
+      if (!finalwiki || finalwiki === "all") {
+        queryString = isRankQuery ? queries.userRankQuery : queries.userGlobalQuery;
+        finalwiki = "all";
+      }
+      let result = await wikitApiRequest(authorName, finalwiki, 0, queryString);
+      if (isRankQuery && result.authorRanking) {
+        const rankData = result;
+        const matchedUser = rankData.authorRanking.find(
+          (u) => u.rank === rankNumber && !config.bannedUsers.includes(u.name)
+        );
+        if (matchedUser) {
+          let secondQuery = !finalwiki || finalwiki === "all" ? queries.userGlobalQuery : queries.userQuery;
+          result = await wikitApiRequest(matchedUser.name, finalwiki, 0, secondQuery);
+        }
+      }
+      const response = /* @__PURE__ */ (0, import_jsx_runtime.jsx)(User, { object: result });
+      const sentMessages = await argv.session.send(response);
+      scheduleChecks(0, argv.session, sentMessages[0]);
+      return;
+    } catch (err) {
+      return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+        "查询失败: ",
+        err.message || "未知错误"
+      ] });
+    }
+  });
+  cmd.command("wikitre.search <标题:string> [维基名称:string]", "查询文章信息。默认搜索所有支持的维基。").alias("wikitre-sr").action(async (argv, title, wiki) => {
+    const titleName = wiki && !Object.keys(wikiInfo).includes(wiki) || !title ? Object.keys(wikiInfo).includes(argv.args.at(-1)) ? argv.args.slice(0, -1).join(" ") : argv.args.join(" ") : title;
+    const Author = /* @__PURE__ */ __name(({ authorName }) => {
+      return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+        "🙎‍♂️作者：",
+        authorName || "已注销用户"
+      ] });
+    }, "Author");
+    const TitleProceed = /* @__PURE__ */ __name(({ titleData }) => {
+      const articles = titleData?.articles?.nodes;
+      if (!articles || articles.length === 0) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("template", { children: "未找到文章。" });
+      }
+      const selectedIndex = articles.findIndex((article2) => {
+        const isBannedTitle = config.bannedTitles.includes(article2.title);
+        const isBannedUser = config.bannedUsers.includes(article2.author);
+        return !(isBannedTitle || isBannedUser);
+      });
+      if (selectedIndex === -1) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("template", { children: "未找到符合条件的文章。" });
+      }
+      const article = articles[selectedIndex];
+      return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("quote", { id: argv.session.event.message.id }),
+        "📝页面：",
+        article.title,
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+        "⭐评分：",
+        article.rating,
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)(Author, { authorName: article.author }),
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+        "🔗",
+        normalizeUrl(article.url)
+      ] });
+    }, "TitleProceed");
+    try {
+      let finalwiki = wiki;
+      if (!finalwiki) {
+        finalwiki = await getDefaultwiki(argv.session);
+      }
+      const result = await wikitApiRequest(titleName, wiki, 0, queries.titleQuery);
+      const response = /* @__PURE__ */ (0, import_jsx_runtime.jsx)(TitleProceed, { titleData: result });
+      const sentMessages = await argv.session.send(response);
+      scheduleChecks(0, argv.session, sentMessages[0]);
+      return;
+    } catch (err) {
+      return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+        "查询失败：",
+        err.message || "未知错误"
+      ] });
+    }
+  });
+  const checkTimes = [1e4, 3e4, 6e4, 9e4, 11e3, 12e3];
+  const checkAndDelete = /* @__PURE__ */ __name(async (session, sentMessage) => {
+    try {
+      const message = await session.onebot.getMsg(session.messageId);
+      if (message?.raw_message === "") {
+        await session.onebot.deleteMsg(sentMessage);
+        return true;
+      }
+      return false;
+    } catch (error) {
+      ctx.logger("wikit-querier").warn("检测或撤回消息失败:", error);
+      return false;
+    }
+  }, "checkAndDelete");
+  const scheduleChecks = /* @__PURE__ */ __name((index, session, sentMessage) => {
+    if (index >= checkTimes.length) return;
+    ctx.setTimeout(
+      async () => {
+        const deleted = await checkAndDelete(session, sentMessage);
+        if (!deleted) {
+          scheduleChecks(index + 1, session, sentMessage);
+        }
+      },
+      index === 0 ? checkTimes[0] : checkTimes[index] - checkTimes[index - 1]
+    );
+  }, "scheduleChecks");
+}
+__name(apply, "apply");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  inject,
+  name
+});

package/lib/lib.d.ts

@@ -0,0 +1,5 @@
+import type { TitleQueryResponse, UserQueryResponse, UserRankQueryResponse } from "./types";
+export declare const wikiInfo: Record<string, {
+    wiki: string;
+}>;
+export declare function wikitApiRequest(param: string, name: string, endpointIndex: number, queryString: string): Promise<TitleQueryResponse | UserQueryResponse | UserRankQueryResponse>;

package/lib/types.d.ts

@@ -0,0 +1,36 @@
+export interface Article {
+    title: string;
+    url: string;
+    author: string;
+    rating: number;
+}
+export interface PageInfo {
+    total: number;
+    page: number;
+    pageSize: number;
+    hasNextPage: boolean;
+}
+export interface TitleQueryResponse {
+    articles: {
+        nodes: Article[];
+        pageInfo: PageInfo;
+    };
+}
+export interface AuthorRank {
+    total: number;
+    rank: number;
+    name: string;
+    value: number;
+}
+export interface UserQueryResponse {
+    authorWikiRank?: AuthorRank;
+    authorGlobalRank?: AuthorRank;
+    articles?: {
+        pageInfo: {
+            total: number;
+        };
+    };
+}
+export interface UserRankQueryResponse {
+    authorRanking: AuthorRank[];
+}

package/package.json

@@ -0,0 +1,29 @@
+{
+  "name": "koishi-plugin-wikit-querier-remodified",
+  "description": "Wikit，修改版，自用",
+  "version": "0.0.1",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "homepage": "https://github.com/lest-day/koishi-plugin-wikit-querier-remodified.git",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/lest-day/koishi-plugin-wikit-querier-remodified.git"
+  },
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "peerDependencies": {
+    "koishi": "^4.18.7"
+  },
+  "dependencies": {
+    "@satorijs/element": "^3.1.8",
+    "koishi-plugin-adapter-onebot": "^6.8.0"
+  }
+}

package/readme.md

@@ -0,0 +1,6 @@
+# koishi-plugin-wikit-querier-remodified
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-wikit-querier-remodified?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-wikit-querier-remodified)
+
+"# koishi-plugin-wikit-querier" 
+"# koishi-plugin-crom-querier-modified"