npm - koishi-plugin-meme-upload-114 - Versions diffs - 0.0.1 - Mend

koishi-plugin-meme-upload-114 0.0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (4) hide show

package/lib/index.d.ts ADDED Viewed

@@ -0,0 +1,16 @@
+import { Context, Schema } from 'koishi';
+declare module 'koishi' {
+    interface Context {
+        puppeteer: any;
+    }
+}
+export declare const name = "local-image-manager";
+export declare const inject: string[];
+export interface Config {
+    rootDir: string;
+    defaultTimeout: number;
+    autoWithdrawSuccess: boolean;
+    successWithdrawTime: number;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js ADDED Viewed

@@ -0,0 +1,218 @@
+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 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 __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
+var src_exports = {};
+__export(src_exports, {
+  Config: () => Config,
+  apply: () => apply,
+  inject: () => inject,
+  name: () => name
+});
+module.exports = __toCommonJS(src_exports);
+var import_koishi = require("koishi");
+var fs = __toESM(require("fs"));
+var path = __toESM(require("path"));
+var name = "local-image-manager";
+var inject = ["puppeteer"];
+var Config = import_koishi.Schema.object({
+  rootDir: import_koishi.Schema.path({ filters: ["directory"], allowCreate: true }).default("data/images").description("图片存储的根目录"),
+  defaultTimeout: import_koishi.Schema.number().default(1e4).description("创建文件夹确认的等待时间(毫秒)"),
+  autoWithdrawSuccess: import_koishi.Schema.boolean().default(true).description("是否自动撤回“上传成功”的提示"),
+  successWithdrawTime: import_koishi.Schema.number().default(1e4).description("上传成功提示的撤回延迟(毫秒)")
+});
+function apply(ctx, config) {
+  const root = path.resolve(ctx.baseDir, config.rootDir);
+  if (!fs.existsSync(root)) {
+    fs.mkdirSync(root, { recursive: true });
+  }
+  const generateQuoteHtml = /* @__PURE__ */ __name((avatar, name2, content) => {
+    return `
+    <html>
+      <head>
+        <style>
+          body { font-family: sans-serif; padding: 20px; background: transparent; width: fit-content; }
+          .container {
+            display: flex;
+            align-items: flex-start;
+            max-width: 400px;
+            background: #ffffff;
+            padding: 15px;
+            border-radius: 12px;
+            box-shadow: 0 4px 10px rgba(0,0,0,0.1);
+          }
+          .avatar {
+            width: 50px;
+            height: 50px;
+            border-radius: 50%;
+            margin-right: 15px;
+            border: 1px solid #eee;
+            flex-shrink: 0;
+          }
+          .content-box { display: flex; flex-direction: column; }
+          .name { font-size: 14px; color: #888; margin-bottom: 6px; font-weight: bold; }
+          .text { font-size: 17px; color: #333; line-height: 1.5; word-wrap: break-word; }
+        </style>
+      </head>
+      <body>
+        <div class="container">
+          <img class="avatar" src="${avatar}" />
+          <div class="content-box">
+            <div class="name">${name2}</div>
+            <div class="text">${content}</div>
+          </div>
+        </div>
+      </body>
+    </html>
+    `;
+  }, "generateQuoteHtml");
+  const getImageBuffer = /* @__PURE__ */ __name(async (session) => {
+    let targetImgUrl = null;
+    const currImg = session.elements.find((e) => e.type === "img" || e.type === "image");
+    if (currImg) {
+      targetImgUrl = currImg.attrs.src || currImg.attrs.url;
+    }
+    if (!targetImgUrl && session.quote) {
+      const quoteElements = session.quote.elements;
+      const quoteImg = quoteElements.find((e) => e.type === "img" || e.type === "image");
+      if (quoteImg) {
+        targetImgUrl = quoteImg.attrs.src || quoteImg.attrs.url;
+      } else {
+        const author = session.quote.author;
+        const avatarUrl = author?.avatar || "https://koishi.chat/logo.png";
+        const username = author?.nickname || author?.username || "用户";
+        const textContent = session.quote.content || " ";
+        const html = generateQuoteHtml(avatarUrl, username, textContent);
+        try {
+          const page = await ctx.puppeteer.page();
+          await page.setContent(html);
+          const element = await page.$(".container");
+          const buffer = await element?.screenshot({ type: "png" });
+          await page.close();
+          return { buffer, ext: ".png" };
+        } catch (e) {
+          ctx.logger.error("Puppeteer 生成图片失败", e);
+          return { buffer: null, ext: "" };
+        }
+      }
+    }
+    if (targetImgUrl) {
+      try {
+        const buffer = await ctx.http.get(targetImgUrl, { responseType: "arraybuffer" });
+        let ext = ".jpg";
+        if (targetImgUrl.includes(".png")) ext = ".png";
+        else if (targetImgUrl.includes(".gif")) ext = ".gif";
+        else if (targetImgUrl.includes(".webp")) ext = ".webp";
+        return { buffer: Buffer.from(buffer), ext };
+      } catch (e) {
+        ctx.logger.error("下载图片失败", e);
+        return { buffer: null, ext: "" };
+      }
+    }
+    return { buffer: null, ext: "" };
+  }, "getImageBuffer");
+  const withdrawLater = /* @__PURE__ */ __name(async (session, messageId, delay) => {
+    await (0, import_koishi.sleep)(delay);
+    try {
+      await session.bot.deleteMessage(session.channelId, messageId);
+    } catch (e) {
+    }
+  }, "withdrawLater");
+  ctx.command("上传 <folderName>", "上传图片或引用转换").action(async ({ session }, folderName) => {
+    if (!folderName) return "请输入文件夹名称。";
+    const { buffer, ext } = await getImageBuffer(session);
+    if (!buffer) {
+      return "无法获取图片。请直接附带图片，或引用含有图片/文字的消息。";
+    }
+    const targetDir = path.join(root, folderName);
+    let shouldUpload = false;
+    if (!fs.existsSync(targetDir)) {
+      const promptMsgIds = await session.send(`文件夹 [${folderName}] 不存在，是否创建？请发送 y 确认。`);
+      const promptMsgId = promptMsgIds[0];
+      const timeoutTimer = setTimeout(() => {
+        session.bot.deleteMessage(session.channelId, promptMsgId).catch(() => {
+        });
+      }, config.defaultTimeout);
+      const input = await session.prompt(config.defaultTimeout);
+      clearTimeout(timeoutTimer);
+      if (input && input.toLowerCase() === "y") {
+        fs.mkdirSync(targetDir, { recursive: true });
+        shouldUpload = true;
+        try {
+          await session.bot.deleteMessage(session.channelId, promptMsgId);
+        } catch (e) {
+        }
+      } else {
+        try {
+          await session.bot.deleteMessage(session.channelId, promptMsgId);
+        } catch (e) {
+        }
+        return "已取消。";
+      }
+    } else {
+      shouldUpload = true;
+    }
+    if (shouldUpload) {
+      try {
+        const filename = `${Date.now()}_${Math.floor(Math.random() * 1e3)}${ext}`;
+        const filePath = path.join(targetDir, filename);
+        fs.writeFileSync(filePath, buffer);
+        const successMsgIds = await session.send(`已成功上传至 ${folderName}`);
+        if (config.autoWithdrawSuccess && successMsgIds.length > 0) {
+          withdrawLater(session, successMsgIds[0], config.successWithdrawTime);
+        }
+      } catch (error) {
+        ctx.logger.error(error);
+        return "保存文件出错。";
+      }
+    }
+  });
+  ctx.middleware(async (session, next) => {
+    const content = session.content.trim();
+    if (!content) return next();
+    const targetDir = path.join(root, content);
+    if (!targetDir.startsWith(root)) return next();
+    if (fs.existsSync(targetDir) && fs.statSync(targetDir).isDirectory()) {
+      const files = fs.readdirSync(targetDir).filter((file) => /\.(jpg|jpeg|png|gif|webp)$/i.test(file));
+      if (files.length === 0) return next();
+      const randomFile = files[Math.floor(Math.random() * files.length)];
+      const absPath = path.join(targetDir, randomFile);
+      return session.send(import_koishi.h.image("file://" + absPath));
+    }
+    return next();
+  });
+}
+__name(apply, "apply");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  inject,
+  name
+});

package/package.json ADDED Viewed

@@ -0,0 +1,23 @@
+{
+  "name": "koishi-plugin-meme-upload-114",
+  "description": "图片上传发送",
+  "version": "0.0.1",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "peerDependencies": {
+    "koishi": "^4.18.7"
+  },
+  "dependencies": {
+    "puppeteer": "^24.36.0"
+  }
+}

package/readme.md ADDED Viewed

@@ -0,0 +1,5 @@
+# koishi-plugin-meme-upload-114
+[![npm](https://img.shields.io/npm/v/koishi-plugin-meme-upload-114?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-meme-upload-114)
+图片上传发送