koishi-plugin-cocoyyy-console 1.1.2-beta.0 → 1.1.2-beta.2

package/lib/index.js

@@ -38,7 +38,7 @@ __export(src_exports, {
   savePath: () => savePath
 });
 module.exports = __toCommonJS(src_exports);
-var import_koishi13 = require("koishi");
+var import_koishi14 = require("koishi");
 
 // src/utils/configs/function_config.ts
 var import_koishi = require("koishi");
@@ -49,6 +49,7 @@ var FunctionConfigSchema = import_koishi.Schema.object({
   repeat_flag: import_koishi.Schema.boolean().default(true).description("复读功能是否启用"),
   rbq_flag: import_koishi.Schema.boolean().default(true).description("*人功能是否启用"),
   son_flag: import_koishi.Schema.boolean().default(true).description("鉴别史消息功能是否启用"),
+  setu_flag: import_koishi.Schema.boolean().default(true).description("随机涩图功能是否启用"),
   game_flag: import_koishi.Schema.boolean().default(true).description("游戏功能是否启用"),
   kuro_flag: import_koishi.Schema.boolean().default(true).description("鸣潮功能是否启用")
 });
@@ -342,6 +343,11 @@ var menuList = [
     description: "shit_or_not功能，用于鉴别史消息",
     command: "无"
   },
+  {
+    name: "setu",
+    description: "随机涩图功能，用于获取涩图",
+    command: "无"
+  },
   {
     name: "game",
     description: "游戏功能，用于游玩游戏",
@@ -374,6 +380,10 @@ function getMenuList(command = null, functionConfig) {
       if (!functionConfig.son_flag)
         return "son功能已关闭";
       return getSonFuncMenu(command);
+    case "setu":
+      if (!functionConfig.setu_flag)
+        return "son功能已关闭";
+      return getSetuFuncMenu(command);
     case "game":
       if (!functionConfig.game_flag)
         return "游戏功能已关闭";
@@ -487,6 +497,20 @@ son功能菜单:
 输入对应指令使用son功能，如果遇到问题请联系开发人员。`;
 }
 __name(getSonFuncMenu, "getSonFuncMenu");
+var setuFuncMenuList = [
+  {
+    name: "setu",
+    description: "随机一个涩图",
+    command: "setu [是否r-18] [标签1] [标签2]"
+  }
+];
+function getSetuFuncMenu(command = null) {
+  return `[所有命令都需要@bot]
+setu功能菜单:
+	${setuFuncMenuList.map((item) => item.command + ": " + item.description).join("\n	")}
+输入对应指令使用setu功能，如果遇到问题请联系开发人员。`;
+}
+__name(getSetuFuncMenu, "getSetuFuncMenu");
 var gameFuncMenuList = [
   {
     name: "gamelist",
@@ -1571,12 +1595,6 @@ __name(extractImageAndText, "extractImageAndText");
 async function checkShitOrNot(session, parttern_msg, config) {
   chatHistory = [];
   chatHistory.push({ role: "system", content: [{ type: "text", text: parttern_msg }] });
-  const { err, resp } = await callOpenRouter(config.api_url, config.api_key, config.api_model, chatHistory);
-  if (err != null) {
-    logger.error("[callOpenRouter Error 1]: " + err);
-    return null;
-  }
-  chatHistory.push({ role: "assistant", content: [{ type: "text", text: resp }] });
   const quoteContent = session.quote.content;
   logger.info("[checkShitOrNot Info]: 引用消息内容: " + quoteContent);
   const { imageUrls, text } = extractImageAndText(quoteContent);
@@ -1592,7 +1610,7 @@ async function checkShitOrNot(session, parttern_msg, config) {
   chatHistory.push({ role: "user", content: targetContent });
   const { err: err2, resp: resp2 } = await callOpenRouter(config.api_url, config.api_key, config.api_model, chatHistory);
   if (err2 != null) {
-    logger.error("[callOpenRouter Error 2]: " + err2);
+    logger.error("[checkShitOrNot Error]: " + err2);
     return null;
   }
   return resp2;
@@ -1627,8 +1645,6 @@ son判断模式列表:
   ctx.command("son <参数>", "判定是否为史").action(async ({ session }, ...args) => {
     if (!dev_mode) {
       if (!is_at_bot_quote(session)) return "请提供正确格式，如： [引用消息] @bot son [判断模式]";
-    } else {
-      if (!session.quote) return "请引用消息后使用功能";
     }
     if (think_flag) return "请等待思考结果";
     const parttern = args?.[0];
@@ -1669,8 +1685,66 @@ function is_at_bot_quote(session) {
 }
 __name(is_at_bot_quote, "is_at_bot_quote");
 
-// src/services/kuro_func/kuro_service.ts
+// src/services/himg_func/himg_service.ts
 var import_axios2 = __toESM(require("axios"));
+var import_koishi12 = require("koishi");
+async function randomHImg(keywords) {
+  try {
+    const r18 = keywords?.[0];
+    const tags = keywords?.slice(1);
+    const params = new URLSearchParams();
+    const random = new import_koishi12.Random(() => Math.random());
+    let level = random.int(1, 4);
+    if (r18) {
+      params.append("r18", "1");
+      level = random.int(4, 7);
+    }
+    params.append("level", level.toString());
+    tags.forEach((tag) => params.append("tag", tag));
+    const res = await import_axios2.default.get("https://lolisuki.cn/api/setu/v1", {
+      params,
+      paramsSerializer: /* @__PURE__ */ __name((p) => p.toString(), "paramsSerializer")
+    });
+    if (res.data) {
+      let author = res.data?.data[0].author;
+      let url = res.data?.data[0].urls.regular;
+      return {
+        result: "作者: " + author + "\n" + import_koishi12.h.image(url),
+        error: null
+      };
+    }
+    logger.error(`[randomHImg Error]: 获取${r18 ? "r-18" : "非r-18"}图片返回为空，请求url: https://lolisuki.cn/api/setu/v1?level=` + level.toString() + "&tag=" + tags.join("&tag="));
+    return {
+      result: null,
+      error: "获取图片返回为空"
+    };
+  } catch (err) {
+    return {
+      result: null,
+      error: err?.message ?? "获取图片请求错误"
+    };
+  }
+}
+__name(randomHImg, "randomHImg");
+
+// src/services/himg_func/himg_commands.ts
+function registerHImgCommands(ctx) {
+  ctx.command("setu <参数>", "随机一个涩图").action(async ({ session }, ...args) => {
+    if (!dev_mode) {
+      if (!is_at_bot(session)) return;
+    }
+    const tag = args;
+    if (!tag) return "请提供正确格式，如：@bot setu [是否r-18] [标签1] [标签2]";
+    let exec = await randomHImg(tag);
+    if (!exec.result)
+      return `获取图片失败：${exec.error}`;
+    return exec.result;
+  });
+}
+__name(registerHImgCommands, "registerHImgCommands");
+
+// src/services/kuro_func/kuro_service.ts
+var import_axios3 = __toESM(require("axios"));
 var import_qs = __toESM(require("qs"));
 
 // src/models/kuro_user.ts
@@ -1803,7 +1877,7 @@ var CONSTANTS = {
     "source": "ios"
   }
 };
-var wavesApi = import_axios2.default.create();
+var wavesApi = import_axios3.default.create();
 wavesApi.interceptors.request.use(
   async (config) => {
     if (config.url.startsWith("/")) {
@@ -2078,7 +2152,7 @@ function registerKuroCommands(ctx, connected) {
 __name(registerKuroCommands, "registerKuroCommands");
 
 // src/services/game_func/game_command.ts
-var import_koishi12 = require("koishi");
+var import_koishi13 = require("koishi");
 
 // src/services/game_func/game_service.ts
 var is_gaming = false;
@@ -2267,7 +2341,7 @@ async function situationPuzzleMiddleware(session, ai_config) {
       return checkResp(resp2);
     }
     const resp = await questSituationPuzzle(text, ai_config);
-    return (0, import_koishi12.h)("at", { id: userId }) + " " + checkResp(resp);
+    return (0, import_koishi13.h)("at", { id: userId }) + " " + checkResp(resp);
   } finally {
     is_thinking = false;
   }
@@ -2284,7 +2358,7 @@ __name(checkResp, "checkResp");
 // src/index.ts
 var name = "cocoyyy-console";
 var dev_mode;
-var Config = import_koishi13.Schema.object({
+var Config = import_koishi14.Schema.object({
   function_config: FunctionConfigSchema.description("功能开关配置"),
   ai_config: AiAPIConfigSchema.description("AI API配置"),
   mysql_config: MysqlConfigSchema.description("MySQL 数据库配置"),
@@ -2292,9 +2366,9 @@ var Config = import_koishi13.Schema.object({
   repeat_config: RepeatConfigSchema.description("复读配置"),
   son_config: ShitOrNotConfigSchema.description("SON配置"),
   game_config: GameConfigSchema.description("游戏配置"),
-  test: import_koishi13.Schema.string().description("测试")
+  test: import_koishi14.Schema.string().description("测试")
 });
-var logger = new import_koishi13.Logger(name);
+var logger = new import_koishi14.Logger(name);
 var savePath = null;
 async function apply(ctx, config) {
   ctx = ctx.guild();
@@ -2316,6 +2390,8 @@ async function apply(ctx, config) {
     registerRepeatMiddleware(ctx, config?.repeat_config);
   if (config.function_config.son_flag)
     registerShitOrNotCommands(ctx, config?.ai_config, config?.son_config);
+  if (config.function_config.setu_flag)
+    registerHImgCommands(ctx);
   if (config.function_config.game_flag) {
     registerGameCommands(ctx, config?.game_config, config?.ai_config);
     registerGameMiddleware(ctx, config?.ai_config);

package/lib/services/himg_func/himg_commands.d.ts

@@ -0,0 +1,3 @@
+import { Context } from 'koishi';
+declare function registerHImgCommands(ctx: Context): void;
+export { registerHImgCommands };

package/lib/services/himg_func/himg_service.d.ts

@@ -0,0 +1,6 @@
+type RandomHImgResp = {
+    result: string;
+    error: string | null;
+};
+declare function randomHImg(keywords: string[]): Promise<RandomHImgResp>;
+export { randomHImg };

package/lib/utils/configs/function_config.d.ts

@@ -6,6 +6,7 @@ interface FunctionConfig {
     repeat_flag: boolean;
     rbq_flag: boolean;
     son_flag: boolean;
+    setu_flag: boolean;
     game_flag: boolean;
     kuro_flag: boolean;
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-cocoyyy-console",
   "description": "自用koishi插件，功能包含复读，记录黑历史，*人等",
-  "version": "1.1.2-beta.0",
+  "version": "1.1.2-beta.2",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "contributors": [