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

package/lib/index.js

@@ -1688,16 +1688,21 @@ __name(is_at_bot_quote, "is_at_bot_quote");
 // src/services/himg_func/himg_service.ts
 var import_axios2 = __toESM(require("axios"));
 var import_koishi12 = require("koishi");
-async function randomHImg(keywords) {
+async function randomHImg(keywords, ...args) {
   try {
     const r18 = keywords?.[0];
-    const tags = keywords?.slice(1);
+    let tags = keywords;
     const params = new URLSearchParams();
+    params.append("ai", "0");
     const random = new import_koishi12.Random(() => Math.random());
     let level = random.int(1, 4);
-    if (r18) {
+    if (r18.toLowerCase() === "r18") {
       params.append("r18", "1");
       level = random.int(4, 7);
+      tags = keywords?.slice(1);
+    }
+    if (args[0]) {
+      params.append("num", args[0].toString());
     }
     params.append("level", level.toString());
     tags.forEach((tag) => params.append("tag", tag));
@@ -1705,7 +1710,7 @@ async function randomHImg(keywords) {
       params,
       paramsSerializer: /* @__PURE__ */ __name((p) => p.toString(), "paramsSerializer")
     });
-    if (res.data) {
+    if (res.data?.data?.length > 0) {
       let author = res.data?.data[0].author;
       let url = res.data?.data[0].urls.regular;
       return {
@@ -1716,7 +1721,7 @@ async function randomHImg(keywords) {
     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: "获取图片返回为空"
+      error: "未获取到图片，请使用其他标签尝试"
     };
   } catch (err) {
     return {

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

@@ -2,5 +2,5 @@ type RandomHImgResp = {
     result: string;
     error: string | null;
 };
-declare function randomHImg(keywords: string[]): Promise<RandomHImgResp>;
+declare function randomHImg(keywords: string[], ...args: any[]): Promise<RandomHImgResp>;
 export { randomHImg };

package/package.json

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