koishi-plugin-crom-querier-modified 2.0.0

package/lib/graphql/index.d.ts

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

package/lib/index.d.ts

@@ -0,0 +1,22 @@
+import { Context, Schema } from "koishi";
+declare module "koishi" {
+    interface Tables {
+        cromQuerier: CromQuerierTable;
+    }
+}
+interface CromQuerierTable {
+    id?: number;
+    platform: string;
+    channelId: string;
+    defaultBranch: 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,389 @@
+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: 10) {
+        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
+      }
+    }
+  `,
+  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"
+];
+var branchInfo = {
+  // br
+  "cn": { url: "http://backrooms-wiki-cn.wikidot.com" },
+  "en": { url: "http://backrooms-wiki.wikidot.com" },
+  "es": { url: "http://es-backrooms-wiki.wikidot.com" },
+  "fr": { url: "http://fr-backrooms-wiki.wikidot.com" },
+  "id": { url: "http://id-backrooms-wiki.wikidot.com" },
+  "jp": { url: "http://japan-backrooms-wiki.wikidot.com" },
+  "pl": { url: "http://pl-backrooms-wiki.wikidot.com" },
+  "ptbr": { url: "http://pt-br-backrooms-wiki.wikidot.com" },
+  "ru": { url: "http://ru-backrooms-wiki.wikidot.com" },
+  "vn": { url: "http://backrooms-vn.wikidot.com" },
+  "it": { url: "http://it-backrooms-wiki.wikidot.com" },
+  "de": { url: "http://de-backrooms-wiki.wikidot.com" },
+  // scp
+  "scp-cn": { url: "http://scp-wiki-cn.wikidot.com" },
+  "scp-cs": { url: "http://scp-cs.wikidot.com" },
+  "scp-de": { url: "http://scp-wiki-de.wikidot.com" },
+  "scp-en": { url: "http://scp-wiki.wikidot.com" },
+  "scp-es": { url: "http://lafundacionscp.wikidot.com" },
+  "scp-fr": { url: "http://fondationscp.wikidot.com" },
+  "scp-int": { url: "http://scp-int.wikidot.com" },
+  "scp-it": { url: "http://fondazionescp.wikidot.com" },
+  "scp-jp": { url: "http://scp-jp.wikidot.com" },
+  "scp-ko": { url: "http://scpko.wikidot.com" },
+  "scp-pl": { url: "http://scp-pl.wikidot.com" },
+  "scp-ptbr": { url: "http://scp-pt-br.wikidot.com" },
+  "scp-ru": { url: "http://scp-ru.wikidot.com" },
+  "scp-th": { url: "http://scp-th.wikidot.com" },
+  "scp-uk": { url: "http://scp-ukrainian.wikidot.com" },
+  "scp-vn": { url: "http://scp-vn.wikidot.com" },
+  "scp-el": { url: "http://scp-el.wikidot.com" },
+  "scp-id": { url: "http://scp-id.wikidot.com" },
+  // others
+  "na": { url: "http://nationarea.wikidot.com" },
+  "ci-cn": { url: "http://ci-cn-wiki.wikidot.com" },
+  // wl
+  "wl-en": { url: "http://wanderers-library.wikidot.com" },
+  "wl-pl": { url: "http://wanderers-library-pl.wikidot.com" },
+  "wl-jp": { url: "http://wanderers-library-jp.wikidot.com" },
+  "wl-cs": { url: "http://wanderers-library-cs.wikidot.com" },
+  "wl-ko": { url: "http://wanderers-library-ko.wikidot.com" },
+  "wl-vn": { url: "http://wanderers-library-vn.wikidot.com" }
+};
+async function cromApiRequest(titleQuery, baseUrl, endpointIndex = 0, queryString) {
+  if (endpointIndex >= apiList.length) {
+    throw new Error("所有API端点均已尝试但均失败");
+  }
+  try {
+    const response = await fetch(apiList[endpointIndex], {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({
+        query: queryString,
+        variables: {
+          query: titleQuery,
+          anyBaseUrl: baseUrl !== "" ? baseUrl : null,
+          baseUrl,
+          rank: parseInt(titleQuery.replace(/#([0-9]{1,15})/, "$1")) || 0
+        }
+      })
+    });
+    if (!response.ok) {
+      throw new Error(`请求失败，状态码: ${response.status}`);
+    }
+    const { data, errors } = await response.json();
+    if (errors && errors.length > 0) {
+      return await cromApiRequest(titleQuery, baseUrl, endpointIndex + 1, queryString);
+    }
+    return data;
+  } catch (error) {
+    if (endpointIndex < apiList.length - 1) {
+      return await cromApiRequest(titleQuery, baseUrl, endpointIndex + 1, queryString);
+    }
+    throw error;
+  }
+}
+__name(cromApiRequest, "cromApiRequest");
+
+// src/index.tsx
+var import_jsx_runtime = require("@satorijs/element/jsx-runtime");
+var name = "crom-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("cromQuerier", {
+    id: "unsigned",
+    platform: "string(64)",
+    channelId: "string(64)",
+    defaultBranch: "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 getBranchUrl = /* @__PURE__ */ __name(async (branch, lastStr, { platform, channel: { id: channelId } }) => {
+    const branchUrls = await ctx.database.get("cromQuerier", { platform, channelId });
+    if (Object.keys(branchInfo).includes(lastStr)) {
+      return branchInfo[lastStr].url;
+    } else if (branch && Object.keys(branchInfo).includes(branch)) {
+      return branchInfo[branch].url;
+    } else if (branchUrls.length > 0) {
+      return branchInfo[branchUrls[0].defaultBranch].url;
+    } else {
+      return branchInfo.cn.url;
+    }
+  }, "getBranchUrl");
+  ctx.command("default-branch <分部名称:string>", "设置默认分部。").alias("默认分部").alias("默认").alias("db").action(async (argv, branch) => {
+    const platform = argv.session.event.platform;
+    const channelId = argv.session.event.channel.id;
+    if (!branch || !Object.keys(branchInfo).includes(branch) || branch === "all") {
+      return "分部名称不正确。";
+    }
+    ctx.database.upsert("cromQuerier", [{ channelId, platform, defaultBranch: branch }], ["platform", "channelId"]);
+    return `已将本群默认查询分部设置为: ${branch}`;
+  });
+  ctx.command("author <作者:string> [分部名称:string]", "查询作者信息。\n默认搜索后室中文站。").alias("作者").alias("作").alias("au").action(async (argv, author, branch) => {
+    const branchUrl = await getBranchUrl(branch, argv.args.at(-1), argv.session.event);
+    const isRankQuery = /^#[0-9]{1,15}$/.test(author);
+    const queryString = isRankQuery ? queries.userRankQuery : queries.userQuery;
+    const authorName = branch && !Object.keys(branchInfo).includes(branch) || !author ? Object.keys(branchInfo).includes(argv.args.at(-1)) ? argv.args.slice(0, -1).join(" ") : argv.args.join(" ") : author;
+    const authorpageOutput = /* @__PURE__ */ __name((author2, authorInfos, branch2) => {
+      if (!authorInfos || authorInfos.length === 0) {
+        return "";
+      }
+      const filteredInfos = authorInfos.filter(
+        (info) => info.authorPage.translationOf == null && !info.authorPage.url.includes("old:") && !info.authorPage.url.includes("deleted:")
+      );
+      if (filteredInfos.length === 0) return "";
+      const matchingAuthorInfos = filteredInfos.filter(
+        (info) => info.authorPage.url.includes(author2)
+      );
+      return matchingAuthorInfos.length > 0 ? matchingAuthorInfos.find((info) => info.site === branch2)?.authorPage.url ?? matchingAuthorInfos[0]?.authorPage.url : filteredInfos.find((info) => info.site === branch2)?.authorPage.url ?? filteredInfos[0]?.authorPage.url ?? "";
+    }, "authorpageOutput");
+    const User = /* @__PURE__ */ __name(({
+      object,
+      branch: branch2,
+      isRank
+    }) => {
+      const dataArray = isRank ? object.usersByRank : object.searchUsers;
+      if (!dataArray || dataArray.length === 0) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("template", { children: "未找到用户。" });
+      }
+      const selectedIndex = dataArray.findIndex(
+        (user2) => !config.bannedUsers.some((banned) => user2.name === banned)
+      );
+      if (selectedIndex === -1) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("template", { children: "未找到符合条件的用户。" });
+      }
+      const user = dataArray[selectedIndex];
+      const userTotalRating = user.statistics.totalRating;
+      const userPageCount = user.statistics.pageCount;
+      const userAuthorPageUrl = normalizeUrl(authorpageOutput(user.name, user.authorInfos, branch2));
+      const averageRating = userPageCount > 0 ? (userTotalRating / userPageCount).toFixed(2) : "0.00";
+      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.statistics.rank,
+        ")",
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+        "总分：",
+        userTotalRating,
+        " 总页面数：",
+        userPageCount,
+        " 平均分：",
+        averageRating,
+        userAuthorPageUrl && /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+          /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+          "作者页：",
+          userAuthorPageUrl
+        ] })
+      ] });
+    }, "User");
+    try {
+      const result = await cromApiRequest(authorName, branchUrl, 0, queryString);
+      const response = /* @__PURE__ */ (0, import_jsx_runtime.jsx)(User, { object: result, branch: branchUrl, isRank: isRankQuery });
+      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 || "未知错误"
+      ] });
+    }
+  });
+  ctx.command("search <标题:string> [分部名称:string]", "查询文章信息。\n默认搜索后室中文站。").alias("搜索").alias("搜").alias("sr").action(async (argv, title, branch) => {
+    const branchUrl = await getBranchUrl(branch, argv.args.at(-1), argv.session.event);
+    const titleName = branch && !Object.keys(branchInfo).includes(branch) || !title ? Object.keys(branchInfo).includes(argv.args.at(-1)) ? argv.args.slice(0, -1).join(" ") : argv.args.join(" ") : title;
+    const Author = /* @__PURE__ */ __name(({ article, isTranslation }) => {
+      const prefix = isTranslation ? "译者：" : "作者：";
+      if (!article.attributions) return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+        prefix,
+        " 未知"
+      ] });
+      const authors = article.attributions.map((attr) => attr.user.name).join("、");
+      if (!isTranslation) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+          prefix,
+          authors
+        ] });
+      }
+      if (!article.translationOf || !article.translationOf.attributions) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+          prefix,
+          authors,
+          " 作者：未知"
+        ] });
+      }
+      const originalAuthors = article.translationOf.attributions.map((attr) => attr.user.name).join("、");
+      return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+        prefix,
+        authors,
+        " 作者：",
+        originalAuthors
+      ] });
+    }, "Author");
+    const TitleProceed = /* @__PURE__ */ __name(({ titleData }) => {
+      if (!titleData.searchPages || titleData.searchPages.length === 0) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("template", { children: "未找到文章。" });
+      }
+      const selectedIndex = titleData.searchPages.findIndex((article2) => {
+        const isBannedTitle = config.bannedTitles.includes(article2.wikidotInfo.title);
+        const isBannedUser = config.bannedUsers.some(
+          (user) => article2.attributions.some((attr) => attr.user.name === user)
+        );
+        const isBannedTag = config.bannedTags.some(
+          (tag) => article2.wikidotInfo.tags.includes(tag)
+        );
+        return !(isBannedTitle || isBannedUser || isBannedTag);
+      });
+      if (selectedIndex === -1) {
+        return /* @__PURE__ */ (0, import_jsx_runtime.jsx)("template", { children: "未找到符合条件的文章。" });
+      }
+      const article = titleData.searchPages[selectedIndex];
+      const { rating, voteCount } = article.wikidotInfo;
+      const positiveVotes = Math.round((voteCount + rating) / 2);
+      const negativeVotes = Math.round((voteCount - rating) / 2);
+      const alternateTitle = article.alternateTitles && article.alternateTitles.length > 0 ? /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(import_jsx_runtime.Fragment, { children: [
+        " - ",
+        article.alternateTitles[selectedIndex].title
+      ] }) : null;
+      return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)("template", { children: [
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("quote", { id: argv.session.event.message.id }),
+        article.wikidotInfo.title,
+        alternateTitle,
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+        "评分：",
+        rating,
+        " (+",
+        positiveVotes,
+        ", -",
+        negativeVotes,
+        ")",
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)(Author, { article, isTranslation: Boolean(article.translationOf) }),
+        /* @__PURE__ */ (0, import_jsx_runtime.jsx)("br", {}),
+        normalizeUrl(article.url)
+      ] });
+    }, "TitleProceed");
+    try {
+      const result = await cromApiRequest(titleName, branchUrl, 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("crom-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,6 @@
+import type { TitleQuery, UserQuery, UserRankQuery } from "./types";
+export declare const branchInfo: Record<string, {
+    url: string;
+}>;
+export declare function cromApiRequest(titleQuery: string, baseUrl: string, endpointIndex: number, queryString: string): Promise<TitleQuery & UserQuery & UserRankQuery>;
+export declare function getBranchUrl(branch: string): string;

package/lib/types.d.ts

@@ -0,0 +1,59 @@
+export interface UserWikidotInfo {
+    displayName: string;
+    wikidotId: number;
+    unixName: string;
+}
+export interface AuthorInfo {
+    site: string;
+    authorPage: {
+        translationOf: {
+            url: string;
+        } | null;
+        url: string;
+    };
+}
+export interface Statistics {
+    rank: number;
+    totalRating: number;
+    pageCount: number;
+}
+export interface User {
+    name: string;
+    wikidotInfo: UserWikidotInfo;
+    authorInfos: AuthorInfo[];
+    statistics: Statistics;
+}
+export interface UserQuery {
+    searchUsers: User[];
+}
+export interface UserRankQuery {
+    usersByRank: User[];
+}
+export interface Attribution {
+    user: {
+        name: string;
+    };
+}
+interface TitleWikidotInfo {
+    title: string;
+    rating: number;
+    voteCount: number;
+    createdAt: Date;
+    tags: string[];
+}
+export interface Title {
+    url: string;
+    wikidotInfo: TitleWikidotInfo;
+    alternateTitles: {
+        title: string;
+    }[];
+    translationOf: {
+        url: string;
+        attributions: Attribution[];
+    } | null;
+    attributions: Attribution[];
+}
+export interface TitleQuery {
+    searchPages: Title[];
+}
+export {};

package/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "koishi-plugin-crom-querier-modified",
+  "description": "向crom查询并返回文本的插件，自用",
+  "version": "2.0.0",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "homepage": "https://github.com/WindyPear-Team/koishi-plugin-crom-querier-modified",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/WindyPeat-Team/koishi-plugin-crom-querier-modified.git"
+  },
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "peerDependencies": {
+    "koishi": "^4.18.7"
+  },
+  "dependencies": {
+    "@satorijs/element": "^3.1.8"
+  }
+}

package/readme.md

@@ -0,0 +1,5 @@
+# koishi-plugin-crom-querier
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-crom-querier?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-crom-querier)
+
+向crom查询并返回文本的插件，自用