koishi-plugin-maple-drift-bottle 0.0.3 → 0.1.0

package/lib/index.d.ts

@@ -0,0 +1,33 @@
+import { Context, Schema } from 'koishi';
+export declare const name = "maple-drift-bottle";
+export declare const using: readonly ["database"];
+interface DriftBottle {
+    id: number;
+    author: string;
+    authorName: string;
+    anonymous: boolean;
+    content: string;
+    images: string[];
+    created: Date;
+}
+declare module 'koishi' {
+    interface Tables {
+        'maple-drift-bottles': DriftBottle;
+    }
+}
+export interface Config {
+    anonymousByDefault: boolean;
+    maxContentLength: number;
+    maxPreviewLength: number;
+    maxImages: number;
+    saveImagesLocally: boolean;
+    imageSavePath: string;
+    compressImages: boolean;
+    maxImageWidth: number;
+    maxImageHeight: number;
+    imageQuality: number;
+    adminQQ: string[];
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;
+export {};

package/lib/index.js

@@ -1,6 +1,8 @@
+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) => {
@@ -15,6 +17,14 @@ 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
@@ -27,12 +37,24 @@ __export(src_exports, {
 });
 module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
+var import_promises = require("fs/promises");
+var import_path = require("path");
+var import_crypto = require("crypto");
+var import_fs = require("fs");
+var import_sharp = __toESM(require("sharp"));
 var name = "maple-drift-bottle";
 var using = ["database"];
 var Config = import_koishi.Schema.object({
   anonymousByDefault: import_koishi.Schema.boolean().default(false).description("默认是否匿名（true: 匿名，false: 不匿名）"),
-  maxContentLength: import_koishi.Schema.number().default(500).min(50).max(2e3).description("漂流瓶内容最大长度（字）（50-2000）"),
-  maxPreviewLength: import_koishi.Schema.number().default(8).min(3).max(50).description('指令"我的漂流瓶"最大预览长度（字）（3-50）'),
+  maxContentLength: import_koishi.Schema.number().default(500).min(50).max(2e3).description("漂流瓶文本内容最大长度（字符）"),
+  maxPreviewLength: import_koishi.Schema.number().default(8).min(3).max(50).description('指令"我的漂流瓶"最大预览字数'),
+  maxImages: import_koishi.Schema.number().default(3).min(0).max(10).description("每一条漂流瓶最多可包含的图片数量（0为不允许发送图片）"),
+  saveImagesLocally: import_koishi.Schema.boolean().default(true).description("是否本地保存图片（启用后图片永久保存，不依赖外部URL）"),
+  imageSavePath: import_koishi.Schema.string().default("./data/maple-drift-bottle/images").description("图片本地保存路径（相对Koishi根目录）"),
+  compressImages: import_koishi.Schema.boolean().default(true).description("是否压缩图片以节省空间"),
+  maxImageWidth: import_koishi.Schema.number().default(1200).min(100).max(3840).description("图片最大宽度（像素）"),
+  maxImageHeight: import_koishi.Schema.number().default(1200).min(100).max(3840).description("图片最大高度（像素）"),
+  imageQuality: import_koishi.Schema.number().default(85).min(10).max(100).description("图片质量 (0-100，越高图片质量越好但文件越大)"),
   adminQQ: import_koishi.Schema.array(String).role("table").default([]).description("管理人QQ号（可添加多个）")
 });
 function getSenderDisplay(bottle) {
@@ -55,12 +77,90 @@ function formatTime(date) {
 }
 __name(formatTime, "formatTime");
 function getContentPreview(content, maxLength) {
+  if (!content || content.length === 0) {
+    return "[无文本内容]";
+  }
   if (content.length <= maxLength) {
     return content;
   }
   return content.substring(0, maxLength) + "...";
 }
 __name(getContentPreview, "getContentPreview");
+function extractContentAndImages(elements) {
+  let text = "";
+  const images = [];
+  for (const element of elements) {
+    if (element.type === "text") {
+      text += element.attrs?.content || "";
+    } else if (element.type === "image" || element.type === "img") {
+      const url = element.attrs?.url || element.attrs?.src;
+      if (url) {
+        images.push(url);
+      }
+    } else if (element.type === "at") {
+      continue;
+    } else if (element.type === "face") {
+      continue;
+    } else {
+      text += `[${element.type}]`;
+    }
+  }
+  return { text: text.trim(), images };
+}
+__name(extractContentAndImages, "extractContentAndImages");
+async function downloadAndSaveImage(ctx, url, savePath, compress, maxWidth, maxHeight, quality) {
+  try {
+    if (!(0, import_fs.existsSync)(savePath)) {
+      await (0, import_promises.mkdir)(savePath, { recursive: true });
+    }
+    const hash = (0, import_crypto.createHash)("md5").update(url + Date.now()).digest("hex");
+    const originalExt = (0, import_path.extname)(new URL(url).pathname) || ".jpg";
+    const fileName = `${hash}${originalExt}`;
+    const filePath = (0, import_path.join)(savePath, fileName);
+    const response = await ctx.http.get(url, { responseType: "arraybuffer" });
+    const imageBuffer = Buffer.from(response);
+    if (compress) {
+      try {
+        let sharpInstance = (0, import_sharp.default)(imageBuffer);
+        const metadata = await sharpInstance.metadata();
+        if (metadata.width > maxWidth || metadata.height > maxHeight) {
+          sharpInstance = sharpInstance.resize({
+            width: maxWidth,
+            height: maxHeight,
+            fit: "inside",
+            // 保持宽高比，不超过指定尺寸
+            withoutEnlargement: true
+            // 不放大比指定尺寸小的图片
+          });
+        }
+        await sharpInstance.jpeg({ quality }).toFile(filePath);
+        ctx.logger("maple-drift-bottle").info(`图片压缩保存: ${fileName} (${imageBuffer.length}字节 -> ${(await (0, import_promises.readFile)(filePath)).length}字节)`);
+      } catch (sharpError) {
+        ctx.logger("maple-drift-bottle").warn("sharp压缩失败，保存原始图片:", sharpError);
+        await (0, import_promises.writeFile)(filePath, imageBuffer);
+      }
+    } else {
+      await (0, import_promises.writeFile)(filePath, imageBuffer);
+    }
+    return fileName;
+  } catch (error) {
+    ctx.logger("maple-drift-bottle").error("下载或保存图片失败:", error);
+    throw new Error(`图片保存失败: ${error.message}`);
+  }
+}
+__name(downloadAndSaveImage, "downloadAndSaveImage");
+function getLocalImagePath(fileName, ctx, config) {
+  if (!fileName || !config.saveImagesLocally) {
+    return fileName;
+  }
+  const fullPath = (0, import_path.join)(config.imageSavePath, fileName);
+  if (!(0, import_fs.existsSync)(fullPath)) {
+    ctx.logger("maple-drift-bottle").warn(`图片文件不存在: ${fullPath}`);
+    return fileName;
+  }
+  return fullPath;
+}
+__name(getLocalImagePath, "getLocalImagePath");
 function apply(ctx, config) {
   console.log("maple-drift-bottle 漂流瓶插件加载中...");
   ctx.model.extend("maple-drift-bottles", {
@@ -72,7 +172,9 @@ function apply(ctx, config) {
     anonymous: "boolean",
     // 是否匿名
     content: "text",
-    // 漂流瓶内容（text类型，支持长文本）
+    // 文本内容
+    images: "json",
+    // 图片本地文件名数组（JSON格式存储）
     created: "timestamp"
     // 创建时间
   }, {
@@ -92,24 +194,50 @@ function apply(ctx, config) {
       const bottle = allBottles[randomIndex];
       const senderDisplay = getSenderDisplay(bottle);
       const timeDisplay = formatTime(bottle.created);
-      return `【漂流瓶 #${bottle.id}】
-发送者：${senderDisplay}
-时间：${timeDisplay}
-内容：
-${bottle.content}`;
+      const message = [
+        `【漂流瓶 #${bottle.id}】`,
+        `发送者：${senderDisplay}`,
+        `时间：${timeDisplay}`
+      ];
+      if (bottle.content) {
+        message.push(`内容：
+${bottle.content}`);
+      }
+      if (bottle.images && bottle.images.length > 0) {
+        await session.send(message.join("\n"));
+        for (const imageName of bottle.images) {
+          try {
+            const imagePath = getLocalImagePath(imageName, ctx, config);
+            await session.send(import_koishi.h.image(imagePath));
+          } catch (error) {
+            ctx.logger("maple-drift-bottle").warn(`发送图片失败: ${imageName}`, error);
+            await session.send(`[图片加载失败]`);
+          }
+        }
+        return "";
+      } else {
+        return message.join("\n");
+      }
     } catch (error) {
       ctx.logger("maple-drift-bottle").error("捞漂流瓶时出错:", error);
       return "捞取漂流瓶时出错了，请稍后再试。";
     }
   });
-  ctx.command("漂流瓶/扔漂流瓶 <content:text>", "扔一个漂流瓶到海里").alias("扔漂流瓶").alias("丢漂流瓶").option("invisible", "-i  匿名发送（不显示发送者）").option("visible", "-v  公开发送（显示发送者）").example("扔漂流瓶 这是一条漂流瓶内容").example("扔漂流瓶 -i 这是一条匿名漂流瓶内容").action(async ({ session, options }, content) => {
+  ctx.command("漂流瓶/扔漂流瓶", "扔一个漂流瓶到海里（支持文本和图片）").alias("扔漂流瓶").alias("丢漂流瓶").option("invisible", "-i  匿名发送（不显示发送者）").option("visible", "-v  公开发送（显示发送者）").example("扔漂流瓶 这是一条漂流瓶内容").example("扔漂流瓶 -i 这是一条匿名漂流瓶内容").action(async ({ session, options }) => {
     try {
-      if (!content || content.trim().length === 0) {
-        return "漂流瓶内容不能为空！\n格式：扔漂流瓶 [-i|-v] 内容\n示例：扔漂流瓶 -i 这是一条匿名漂流瓶";
+      const elements = session.elements;
+      if (elements.length === 0) {
+        return "漂流瓶内容不能为空！请发送文本或图片。";
+      }
+      const { text, images: imageUrls } = extractContentAndImages(elements);
+      if (!text && imageUrls.length === 0) {
+        return "漂流瓶内容不能为空！请发送文本或图片。";
+      }
+      if (text.length > config.maxContentLength) {
+        return `漂流瓶文本内容太长了！最多只能输入${config.maxContentLength}个字符（当前: ${text.length}）。`;
       }
-      const trimmedContent = content.trim();
-      if (trimmedContent.length > config.maxContentLength) {
-        return `漂流瓶内容太长了！最多只能输入${config.maxContentLength}个字（当前: ${trimmedContent.length}）。`;
+      if (imageUrls.length > config.maxImages) {
+        return `图片数量太多了！最多只能发送${config.maxImages}张图片（当前: ${imageUrls.length}）。`;
       }
       if (options.invisible && options.visible) {
         return "选项冲突：-i（匿名）和-v（公开）不能同时使用";
@@ -122,17 +250,40 @@ ${bottle.content}`;
       }
       const userInfo = await session.getUser(session.userId);
       const authorName = userInfo?.name || session.username || `用户${session.userId}`;
+      const savedImageNames = [];
+      if (config.saveImagesLocally && imageUrls.length > 0) {
+        for (const imageUrl of imageUrls) {
+          try {
+            const savedName = await downloadAndSaveImage(
+              ctx,
+              imageUrl,
+              config.imageSavePath,
+              config.compressImages,
+              config.maxImageWidth,
+              config.maxImageHeight,
+              config.imageQuality
+            );
+            savedImageNames.push(savedName);
+          } catch (error) {
+            ctx.logger("maple-drift-bottle").error("图片处理失败:", error);
+            return `图片处理失败: ${error.message}`;
+          }
+        }
+      }
       const newBottle = await ctx.database.create("maple-drift-bottles", {
         author: session.userId,
         authorName,
         anonymous,
-        content: trimmedContent,
+        content: text,
+        images: config.saveImagesLocally ? savedImageNames : imageUrls,
         created: /* @__PURE__ */ new Date()
       });
       const anonymousText = anonymous ? "匿名" : "不匿名";
+      const imageText = savedImageNames.length > 0 ? `，包含${savedImageNames.length}张图片` : "";
+      const saveText = config.saveImagesLocally ? "（已本地保存）" : "";
       return `成功扔出一个漂流瓶！
 ID: #${newBottle.id}
-状态: ${anonymousText}
+状态: ${anonymousText}${imageText}${saveText}
 内容已保存到海中，等待有缘人捞取。`;
     } catch (error) {
       ctx.logger("maple-drift-bottle").error("扔漂流瓶时出错:", error);
@@ -167,8 +318,9 @@ ID: #${newBottle.id}
         const timeDisplay = formatTime(bottle.created);
         const anonymousText = bottle.anonymous ? "（匿名）" : "";
         const contentPreview = getContentPreview(bottle.content, config.maxPreviewLength);
+        const imageText = bottle.images && bottle.images.length > 0 ? ` [${bottle.images.length}图]` : "";
         output += `${displayIndex}. 漂流瓶 #${bottle.id} ${anonymousText} - ${timeDisplay}
-   预览: ${contentPreview}
+   预览: ${contentPreview}${imageText}
 `;
       });
       output += "\n──────────\n";
@@ -204,12 +356,31 @@ ID: #${newBottle.id}
       }
       const senderDisplay = getSenderDisplay(bottle);
       const timeDisplay = formatTime(bottle.created);
-      return `【漂流瓶 #${bottle.id}】
-发送者：${senderDisplay}
-时间：${timeDisplay}
-状态：${bottle.anonymous ? "匿名" : "公开"}
-内容：
-${bottle.content}`;
+      const message = [
+        `【漂流瓶 #${bottle.id}】`,
+        `发送者：${senderDisplay}`,
+        `时间：${timeDisplay}`,
+        `状态：${bottle.anonymous ? "匿名" : "公开"}`
+      ];
+      if (bottle.content) {
+        message.push(`内容：
+${bottle.content}`);
+      }
+      if (bottle.images && bottle.images.length > 0) {
+        await session.send(message.join("\n"));
+        for (const imageName of bottle.images) {
+          try {
+            const imagePath = getLocalImagePath(imageName, ctx, config);
+            await session.send(import_koishi.h.image(imagePath));
+          } catch (error) {
+            ctx.logger("maple-drift-bottle").warn(`发送图片失败: ${imageName}`, error);
+            await session.send(`[图片加载失败]`);
+          }
+        }
+        return "";
+      } else {
+        return message.join("\n");
+      }
     } catch (error) {
       ctx.logger("maple-drift-bottle").error("查看漂流瓶时出错:", error);
       return "查看漂流瓶时出错了，请稍后再试。";
@@ -232,12 +403,26 @@ ${bottle.content}`;
       }
       const senderDisplay = getSenderDisplay(bottle);
       const timeDisplay = formatTime(bottle.created);
-      const contentPreview = bottle.content.length > 50 ? bottle.content.substring(0, 50) + "..." : bottle.content;
+      const contentPreview = bottle.content ? bottle.content.length > 50 ? bottle.content.substring(0, 50) + "..." : bottle.content : "[无文本内容]";
+      const imageText = bottle.images && bottle.images.length > 0 ? ` (${bottle.images.length}张图片)` : "";
+      if (config.saveImagesLocally && bottle.images && bottle.images.length > 0) {
+        for (const imageName of bottle.images) {
+          try {
+            const imagePath = (0, import_path.join)(config.imageSavePath, imageName);
+            if ((0, import_fs.existsSync)(imagePath)) {
+              await (0, import_promises.rm)(imagePath);
+              ctx.logger("maple-drift-bottle").info(`删除图片文件: ${imagePath}`);
+            }
+          } catch (error) {
+            ctx.logger("maple-drift-bottle").warn(`删除图片文件失败: ${imageName}`, error);
+          }
+        }
+      }
       await ctx.database.remove("maple-drift-bottles", { id });
       return `已成功删除漂流瓶 #${id}
 发送者：${senderDisplay}
 时间：${timeDisplay}
-预览：${contentPreview}`;
+预览：${contentPreview}${imageText}`;
     } catch (error) {
       ctx.logger("maple-drift-bottle").error("删除漂流瓶时出错:", error);
       return "删除漂流瓶时出错了，请稍后再试。";
@@ -273,6 +458,27 @@ ${bottle.content}`;
       if (count === 0) {
         return `没有找到${description}漂流瓶。`;
       }
+      if (config.saveImagesLocally) {
+        let deletedImages = 0;
+        for (const bottle of bottles) {
+          if (bottle.images && bottle.images.length > 0) {
+            for (const imageName of bottle.images) {
+              try {
+                const imagePath = (0, import_path.join)(config.imageSavePath, imageName);
+                if ((0, import_fs.existsSync)(imagePath)) {
+                  await (0, import_promises.rm)(imagePath);
+                  deletedImages++;
+                }
+              } catch (error) {
+                ctx.logger("maple-drift-bottle").warn(`删除图片文件失败: ${imageName}`, error);
+              }
+            }
+          }
+        }
+        if (deletedImages > 0) {
+          ctx.logger("maple-drift-bottle").info(`清空漂流瓶时删除了 ${deletedImages} 张图片文件`);
+        }
+      }
       await ctx.database.remove("maple-drift-bottles", query);
       return `已成功清空 ${count} 条${description}漂流瓶记录。`;
     } catch (error) {
@@ -327,11 +533,27 @@ ${failDetails.join("\n")}`;
       return "联系漂流瓶管理人时出错了，请稍后再试。";
     }
   });
-  ctx.on("ready", () => {
+  ctx.on("ready", async () => {
     console.log("maple-drift-bottle 漂流瓶插件已加载完成");
+    if (config.saveImagesLocally) {
+      if (!(0, import_fs.existsSync)(config.imageSavePath)) {
+        await (0, import_promises.mkdir)(config.imageSavePath, { recursive: true });
+        ctx.logger("maple-drift-bottle").info(`创建图片保存目录: ${config.imageSavePath}`);
+      }
+    }
     ctx.logger("maple-drift-bottle").info(`默认匿名设置: ${config.anonymousByDefault ? "是" : "否"}`);
-    ctx.logger("maple-drift-bottle").info(`内容最大长度: ${config.maxContentLength} 字`);
+    ctx.logger("maple-drift-bottle").info(`内容最大长度: ${config.maxContentLength} 字符`);
     ctx.logger("maple-drift-bottle").info(`我的漂流瓶最大预览字数: ${config.maxPreviewLength} 字`);
+    ctx.logger("maple-drift-bottle").info(`最大图片数量: ${config.maxImages} 张`);
+    ctx.logger("maple-drift-bottle").info(`本地保存图片: ${config.saveImagesLocally ? "是" : "否"}`);
+    if (config.saveImagesLocally) {
+      ctx.logger("maple-drift-bottle").info(`图片保存路径: ${config.imageSavePath}`);
+      ctx.logger("maple-drift-bottle").info(`压缩图片: ${config.compressImages ? "是" : "否"}`);
+      if (config.compressImages) {
+        ctx.logger("maple-drift-bottle").info(`图片最大尺寸: ${config.maxImageWidth}x${config.maxImageHeight} 像素`);
+        ctx.logger("maple-drift-bottle").info(`图片质量: ${config.imageQuality}%`);
+      }
+    }
     ctx.logger("maple-drift-bottle").info(`管理人QQ: ${config.adminQQ.length > 0 ? config.adminQQ.join(", ") : "未配置"}`);
   });
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-maple-drift-bottle",
   "description": "-",
-  "version": "0.0.3",
+  "version": "0.1.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -14,5 +14,8 @@
   ],
   "peerDependencies": {
     "koishi": "^4.18.7"
+  },
+  "dependencies": {
+    "sharp": "^0.33.0"
   }
 }