koishi-plugin-phimg2 1.0.3 → 1.0.5

package/lib/index.d.ts

@@ -1,4 +1,5 @@
 import { Context, Schema } from 'koishi';
+export declare const name = "phimg2";
 export declare const inject: string[];
 declare module 'koishi' {
     interface Tables {
@@ -11,12 +12,11 @@ export interface Config {
     defaultTags: string[];
     enabledByDefault: boolean;
     useGlobalTagsByDefault: boolean;
-    proxy: string;
-    timeout: number;
+    filterId: number;
 }
 export declare const Config: Schema<Config>;
 interface GroupConfig {
-    id?: number;
+    id: number;
     groupId: string;
     enabled: boolean;
     useGlobalTags: boolean;

package/lib/index.js

@@ -1,13 +1,11 @@
-var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
-var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
+  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") {
@@ -17,14 +15,6 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
-  // If the importer is in node compatibility mode or this is not an ESM
-  // file that has been converted to a CommonJS file using a Babel-
-  // compatible transform (i.e. "__esModule" has not been set), then set
-  // "default" to the CommonJS "module.exports" for node compatibility.
-  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
-  mod
-));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/index.ts
@@ -32,12 +22,13 @@ var src_exports = {};
 __export(src_exports, {
   Config: () => Config,
   apply: () => apply,
-  inject: () => inject
+  inject: () => inject,
+  name: () => name
 });
 module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
-var import_axios = __toESM(require("axios"));
-var inject = ["database"];
+var name = "phimg2";
+var inject = ["database", "http"];
 var translationTable = {
   "；": ";",
   "：": ":",
@@ -59,13 +50,12 @@ function translateText(text) {
 }
 __name(translateText, "translateText");
 var Config = import_koishi.Schema.object({
-  apiKey: import_koishi.Schema.string().description("Philomena API 密钥").default(""),
-  apiUrl: import_koishi.Schema.string().description("Philomena API 域名 (例如 derpibooru.org)").default("derpibooru.org"),
+  apiKey: import_koishi.Schema.string().description("Philomena API 密钥").role("secret").default(""),
+  apiUrl: import_koishi.Schema.string().description("Philomena API 域名 (无需 https://)").default("derpibooru.org"),
   defaultTags: import_koishi.Schema.array(String).description("全局默认标签").default(["safe"]),
   enabledByDefault: import_koishi.Schema.boolean().description("新群聊默认启用搜图功能").default(true),
   useGlobalTagsByDefault: import_koishi.Schema.boolean().description("新群聊默认启用全局标签").default(true),
-  proxy: import_koishi.Schema.string().description("代理服务器 (例如 http://127.0.0.1:7890)").default(""),
-  timeout: import_koishi.Schema.number().description("请求超时时间 (秒)").default(30)
+  filterId: import_koishi.Schema.number().description("搜索使用的 Filter ID (例如 100073)").default(100073)
 });
 function apply(ctx, config) {
   ctx.model.extend("phimg_config", {
@@ -82,82 +72,72 @@ function apply(ctx, config) {
   const getGroupConfig = /* @__PURE__ */ __name(async (groupId) => {
     let [groupConfig] = await ctx.database.get("phimg_config", { groupId });
     if (!groupConfig) {
-      groupConfig = await ctx.database.create("phimg_config", {
-        groupId,
-        enabled: config.enabledByDefault,
-        useGlobalTags: config.useGlobalTagsByDefault,
-        customTags: []
-      });
+      try {
+        groupConfig = await ctx.database.create("phimg_config", {
+          groupId,
+          enabled: config.enabledByDefault,
+          useGlobalTags: config.useGlobalTagsByDefault,
+          customTags: []
+        });
+      } catch (e) {
+        [groupConfig] = await ctx.database.get("phimg_config", { groupId });
+      }
     }
     return groupConfig;
   }, "getGroupConfig");
   const updateGroupConfig = /* @__PURE__ */ __name(async (groupId, data) => {
+    await getGroupConfig(groupId);
     await ctx.database.set("phimg_config", { groupId }, data);
   }, "updateGroupConfig");
   const makeRequest = /* @__PURE__ */ __name(async (method, params) => {
-    let domain = config.apiUrl;
-    if (domain.includes("://")) {
-      domain = domain.split("://")[1];
-    }
-    domain = domain.split("/")[0];
-    const url = `https://${domain}/api/v1/json/search/${method}`;
-    const axiosConfig = {
-      headers: {
-        "Accept": "application/json",
-        "User-Agent": "Phimg for Koishi"
-      },
-      timeout: config.timeout * 1e3
+    const host = config.apiUrl.replace(/^https?:\/\//, "").replace(/\/$/, "");
+    const endpoint = `https://${host}/api/v1/json/search/${method}`;
+    const queryParams = {
+      filter_id: config.filterId
     };
-    if (config.proxy) {
-      try {
-        const urlParsed = new URL(config.proxy);
-        axiosConfig.proxy = {
-          host: urlParsed.hostname,
-          port: parseInt(urlParsed.port),
-          protocol: urlParsed.protocol.replace(":", "")
-        };
-      } catch (e) {
-        ctx.logger.warn(`Invalid proxy URL: ${config.proxy}`);
-      }
+    if (params.key) {
+      queryParams.key = params.key;
+      delete params.key;
     }
     try {
-      let response;
+      let responseData;
       if (method === "images") {
-        response = await import_axios.default.get(url, { ...axiosConfig, params });
+        responseData = await ctx.http.get(endpoint, {
+          params: { ...queryParams, ...params },
+          headers: { "User-Agent": "Phimg for Koishi" }
+        });
       } else {
-        const formData = new URLSearchParams();
-        for (const key in params) {
-          formData.append(key, params[key]);
-        }
-        response = await import_axios.default.post(url, formData, axiosConfig);
+        responseData = await ctx.http.post(endpoint, params, {
+          params: queryParams,
+          headers: {
+            "User-Agent": "Phimg for Koishi",
+            "Content-Type": "application/x-www-form-urlencoded"
+          }
+        });
       }
-      if (!response.data.images || response.data.images.length === 0) {
+      if (!responseData.images || responseData.images.length === 0) {
         throw new Error("未找到匹配的图片");
       }
-      return response.data;
+      return responseData;
     } catch (error) {
-      if (import_axios.default.isAxiosError(error)) {
-        if (error.response?.status === 404) throw new Error("未找到匹配的图片");
-        throw new Error(`API 请求失败: ${error.message}`);
-      }
-      throw error;
+      if (error.response?.status === 404) throw new Error("未找到匹配的图片");
+      ctx.logger("phimg").warn(error);
+      throw new Error(`${error.message}`);
     }
   }, "makeRequest");
   const VIDEO_TYPES = ["webm", "mp4"];
   const getMediaElement = /* @__PURE__ */ __name((selected) => {
+    if (!selected?.representations) return import_koishi.h.text("图片数据解析错误");
     const file = selected.representations.full;
     const url = file.endsWith(".webm") ? selected.representations.medium : selected.representations.large;
     const fileType = file.split(".").pop()?.toLowerCase() || "";
     if (VIDEO_TYPES.includes(fileType)) {
       return import_koishi.h.video(url);
-    } else {
-      return import_koishi.h.image(url);
     }
+    return import_koishi.h.image(url);
   }, "getMediaElement");
   const searchHelp = `用法: /搜图 [tags|distance]
 
-输入标签为tags搜图；输入匹配距离为图搜图
-
 可选项:
   --tags             获取当前群聊内置标签列表
   --status           获取当前群聊的搜图功能状态
@@ -167,12 +147,8 @@ function apply(ctx, config) {
   --sd [sd]          排序方向，默认为desc
   --i [index]        选择结果索引，默认为-1（即随机）
 
-示例：
-  /搜图 [tags] # 直接通过标签搜索图片
-
 提示: 
-  图搜图使用方式为引用图片，默认匹配距离为0.25
-  所有参数及变量全部遵循呆站（Philomena系图站）搜索API规范`;
+  图搜图使用方式为引用图片，默认匹配距离为0.25`;
   const configHelp = `用法: 搜图-c [选项]
 
 可选项:
@@ -181,11 +157,10 @@ function apply(ctx, config) {
   --on               开启当前群聊的搜图功能
   --off              关闭当前群聊的搜图功能
   --onglobal         启用全局标签
-  --offglobal        关闭全局标签
-`;
-  ctx.command("搜图 [...params]", "从图站搜索图片").option("tags", "--tags 获取当前群聊内置标签列表").option("status", "--status 获取当前群聊的搜图功能状态").option("pp", "--pp <per_page:number> 每页结果数量，默认为50", { fallback: 50 }).option("p", "--p <page:number> 页码，默认为1", { fallback: 1 }).option("sf", "--sf <sf:string> 排序字段，默认为score", { fallback: "score" }).option("sd", "--sd <sd:string> 排序方向，默认为desc", { fallback: "desc" }).option("i", "--i <index:number> 选择结果索引，默认为-1（即随机）", { fallback: -1 }).action(async ({ session, options }, ...paramsArray) => {
+  --offglobal        关闭全局标签`;
+  ctx.command("搜图 [...params]", "从图站搜索图片").option("tags", "--tags").option("status", "--status").option("pp", "--pp <per_page:number>", { fallback: 50 }).option("p", "--p <page:number>", { fallback: 1 }).option("sf", "--sf <sf:string>", { fallback: "score" }).option("sd", "--sd <sd:string>", { fallback: "desc" }).option("i", "--i <index:number>", { fallback: -1 }).action(async ({ session, options }, ...paramsArray) => {
     if (!session?.guildId) return "搜图仅限群聊使用。";
-    let params = paramsArray.join(" ");
+    let params = paramsArray.join(" ").trim();
     if (!params && !session.quote && !options.tags && !options.status) {
       return searchHelp;
     }
@@ -193,9 +168,9 @@ function apply(ctx, config) {
     const groupConfig = await getGroupConfig(groupId);
     if (options.status) {
       return `当前群聊搜图功能状态：
- 启用：${groupConfig.enabled}
- 标签：${groupConfig.customTags.join(", ") || "无"}
- 全局标签：${groupConfig.useGlobalTags ? "启用" : "禁用"}`;
+启用：${groupConfig.enabled}
+标签：${groupConfig.customTags.join(", ") || "无"}
+全局标签：${groupConfig.useGlobalTags ? "启用" : "禁用"}`;
     }
     if (options.tags) {
       return `当前群聊内置标签：${groupConfig.customTags.join(", ") || "无"}`;
@@ -207,17 +182,18 @@ function apply(ctx, config) {
     let imageUrl;
     const quote = session.quote;
     if (quote) {
-      const imgElement = import_koishi.h.select(quote.content, "image")[0];
+      const imgElement = import_koishi.h.select(quote.content, "image")[0] || import_koishi.h.select(quote.content, "img")[0];
       if (imgElement) {
-        imageUrl = imgElement.attrs.url;
+        imageUrl = imgElement.attrs.url || imgElement.attrs.src;
       }
     }
     try {
       if (imageUrl) {
         let distance = 0.25;
         if (params) {
-          if (!isNaN(Number(params))) {
-            distance = Number(params);
+          const parsed = Number(params);
+          if (!isNaN(parsed)) {
+            distance = parsed;
           } else {
             return "图片搜索仅支持数字参数，表示相似度距离（distance）。";
           }
@@ -232,22 +208,23 @@ function apply(ctx, config) {
         if (images.length > 10) {
           return `搜索到过多图片 (${images.length} 张)，请尝试减小距离参数。`;
         }
-        let result = "";
-        result += (0, import_koishi.h)("at", { id: session.userId }).toString();
-        result += `
-distance: ${distance}`;
+        if (images.length === 0) return "未找到匹配的图片";
+        const result = [(0, import_koishi.h)("at", { id: session.userId }), import_koishi.h.text(`
+distance: ${distance}
+`)];
         for (const img of images) {
-          result += getMediaElement(img).toString();
-          result += `
-id: ${img.id}
-score: ${img.score}`;
+          result.push(getMediaElement(img));
+          result.push(import_koishi.h.text(`
+id: ${img.id} | score: ${img.score}
+`));
         }
         return result;
       } else {
         const userTags = params ? params.split(",").map((t) => t.trim()).filter((t) => t) : [];
         const globalTags = groupConfig.useGlobalTags ? config.defaultTags : [];
         const groupTags = groupConfig.customTags;
-        const allTags = [.../* @__PURE__ */ new Set([...groupTags, ...globalTags, ...userTags])];
+        const allTags = Array.from(/* @__PURE__ */ new Set([...groupTags, ...globalTags, ...userTags]));
+        if (allTags.length === 0 && !params) return "请输入搜索标签。";
         const queryParams = {
           q: allTags.join(", "),
           key: config.apiKey,
@@ -261,40 +238,33 @@ score: ${img.score}`;
         let index = options.i;
         let additionalMsg = "";
         if (index < 0 || index >= images.length) {
-          if (index >= 0) {
-            additionalMsg = `索引 ${index} 超出单页范围，已随机选择图片`;
-          }
+          if (index >= 0) additionalMsg = `索引 ${index} 超出单页范围，已随机选择图片`;
           index = Math.floor(Math.random() * images.length);
         }
         const selected = images[index];
-        let result = "";
-        result += (0, import_koishi.h)("at", { id: session.userId }).toString();
-        result += getMediaElement(selected).toString();
-        result += `
-id: ${selected.id}
-score: ${selected.score}`;
-        result += `
-tags: ${queryParams.q}`;
-        if (additionalMsg) result += `
-提示：${additionalMsg}`;
-        return result;
+        return [
+          (0, import_koishi.h)("at", { id: session.userId }),
+          getMediaElement(selected),
+          import_koishi.h.text(`
+id: ${selected.id} | score: ${selected.score}`),
+          import_koishi.h.text(`
+tags: ${queryParams.q}`),
+          additionalMsg ? import_koishi.h.text(`
+提示：${additionalMsg}`) : null
+        ];
       }
     } catch (error) {
       return error instanceof Error ? error.message : String(error);
     }
   });
   const confirmOffGlobal = /* @__PURE__ */ new Set();
-  ctx.command("搜图-c", "配置搜图功能", { authority: 3 }).option("on", "--on 开启当前群聊的搜图功能").option("off", "--off 关闭当前群聊的搜图功能").option("onglobal", "--onglobal 启用全局标签").option("offglobal", "--offglobal 关闭全局标签").option("add", "--add <tags:string> 添加标签，多个标签用逗号分隔").option("rm", "--rm <tags:string> 删除标签，多个标签用逗号分隔").action(async ({ session, options }) => {
+  ctx.command("搜图-c", "配置搜图功能", { authority: 3 }).option("on", "--on").option("off", "--off").option("onglobal", "--onglobal").option("offglobal", "--offglobal").option("add", "--add <tags:string>").option("rm", "--rm <tags:string>").action(async ({ session, options }) => {
     if (!session?.guildId) return "搜图配置仅限群聊使用。";
-    if (Object.keys(options).length === 0) {
-      return configHelp;
-    }
+    if (Object.keys(options).length === 0) return configHelp;
     const groupId = session.guildId;
     const groupConfig = await getGroupConfig(groupId);
     let response = "";
-    if (options.on && options.off) {
-      return "不能同时开启和关闭搜图功能，请选择一个操作";
-    }
+    if (options.on && options.off) return "不能同时开启和关闭搜图功能";
     if (options.on) {
       await updateGroupConfig(groupId, { enabled: true });
       response += "搜图功能已在本群开启\n";
@@ -302,9 +272,7 @@ tags: ${queryParams.q}`;
       await updateGroupConfig(groupId, { enabled: false });
       response += "搜图功能已在本群关闭\n";
     }
-    if (options.onglobal && options.offglobal) {
-      return "不能同时开启和关闭全局标签，请选择一个操作";
-    }
+    if (options.onglobal && options.offglobal) return "不能同时开启和关闭全局标签";
     if (options.onglobal) {
       await updateGroupConfig(groupId, { useGlobalTags: true });
       response += "全局标签已启用\n";
@@ -313,28 +281,27 @@ tags: ${queryParams.q}`;
       if (!confirmOffGlobal.has(confirmKey)) {
         confirmOffGlobal.add(confirmKey);
         ctx.setTimeout(() => confirmOffGlobal.delete(confirmKey), 6e4);
-        return "关闭全局标签，机器人将会搜出非safe图片\n请自行承担可能的炸群风险\n再输入一遍指令确认关闭";
+        return "关闭全局标签，机器人将会搜出非safe图片\n请自行承担风险，再次输入指令确认关闭";
       }
       confirmOffGlobal.delete(confirmKey);
       await updateGroupConfig(groupId, { useGlobalTags: false });
       response += "全局标签已禁用\n";
     }
     if (options.add || options.rm) {
-      const currentTags = groupConfig.customTags;
-      let newTags = [...currentTags];
+      let newTags = [...groupConfig.customTags];
       if (options.add) {
-        const tagsToAdd = translateText(options.add).split(",").map((t) => t.trim()).filter((t) => t);
+        const tagsToAdd = translateText(options.add).split(/[,，]/).map((t) => t.trim()).filter((t) => t);
         newTags = [.../* @__PURE__ */ new Set([...newTags, ...tagsToAdd])];
       }
       if (options.rm) {
-        const tagsToRm = translateText(options.rm).split(",").map((t) => t.trim()).filter((t) => t);
+        const tagsToRm = translateText(options.rm).split(/[,，]/).map((t) => t.trim()).filter((t) => t);
         newTags = newTags.filter((t) => !tagsToRm.includes(t));
       }
       await updateGroupConfig(groupId, { customTags: newTags });
       response += `修改成功，本群标签现为: ${newTags.join(", ") || "无"}
 `;
     }
-    return response.trim() || '请输入有效的配置选项。可用 "搜图-c --help" 查看帮助。';
+    return response.trim();
   });
 }
 __name(apply, "apply");
@@ -342,5 +309,6 @@ __name(apply, "apply");
 0 && (module.exports = {
   Config,
   apply,
-  inject
+  inject,
+  name
 });

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-phimg2",
-  "description": "该插件可以让机器人通过Philomena API在使用Philomena搭建的图站上使用标签（tags）搜图",
-  "version": "1.0.3",
+  "description": "Koishi 插件，让机器人通过Philomena API在使用Philomena搭建的图站上使用标签（tags）搜图或图搜图",
+  "version": "1.0.5",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [