koishi-plugin-watermark-remover 0.0.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/lib/index.d.ts

@@ -0,0 +1,13 @@
+import { Context, Schema } from 'koishi';
+export declare const name = "watermark-remover";
+export declare const inject: string[];
+export interface Config {
+    model: 'Standard V1' | 'Auto V2' | 'Auto V2.3' | 'Manual';
+    removeText: boolean;
+    removeLogo: boolean;
+    enhanceQuality: boolean;
+    debug: boolean;
+    timeout: number;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -0,0 +1,208 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+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 __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 import_path = require("path");
+var import_fs = require("fs");
+var name = "watermark-remover";
+var inject = ["puppeteer"];
+var Config = import_koishi.Schema.object({
+  model: import_koishi.Schema.union(["Standard V1", "Auto V2", "Auto V2.3", "Manual"]).default("Auto V2.3").description("选择使用的模型。"),
+  removeText: import_koishi.Schema.boolean().default(false).description("移除图片中的文字。"),
+  removeLogo: import_koishi.Schema.boolean().default(false).description("移除图片中的 Logo。"),
+  enhanceQuality: import_koishi.Schema.boolean().default(false).description("增强图片画质。"),
+  debug: import_koishi.Schema.boolean().default(false).description("开启调试模式（显示浏览器窗口和详细日志）。"),
+  timeout: import_koishi.Schema.number().default(18e4).description("处理超时时间（毫秒）。")
+});
+function apply(ctx, config) {
+  const logger = ctx.logger("watermark-remover");
+  ctx.command("wmremove [image:image]", "去除图片水印").alias("去水印").action(async ({ session }, image) => {
+    let imgUrl = image?.src;
+    if (!imgUrl) {
+      const imgElement = import_koishi.h.select(session.elements, "img")[0];
+      imgUrl = imgElement?.attrs.src;
+    }
+    if (!imgUrl) {
+      await session.send("请发送一张图片。");
+      const input = await session.prompt(3e4);
+      if (!input) return "输入超时。";
+      const img = import_koishi.h.select(input, "img")[0];
+      if (!img) return "未检测到图片。";
+      imgUrl = img.attrs.src;
+    }
+    const processingMsg = await session.send("正在处理图片，请稍候...");
+    const processingMsgId = processingMsg[0];
+    const page = await ctx.puppeteer.page({
+      args: ["--no-sandbox", "--disable-setuid-sandbox"]
+    });
+    const root = ctx.baseDir || process.cwd();
+    const tempDir = (0, import_path.resolve)(root, "data", "watermark-remover");
+    await import_fs.promises.mkdir(tempDir, { recursive: true });
+    const tempFile = (0, import_path.resolve)(tempDir, `unwatermark_${Date.now()}.png`);
+    try {
+      if (config.debug) logger.info("正在下载图片...");
+      const buffer = await ctx.http.get(imgUrl, { responseType: "arraybuffer" });
+      await import_fs.promises.writeFile(tempFile, Buffer.from(buffer));
+      if (config.debug) logger.info(`图片已保存至临时文件: ${tempFile}`);
+      if (config.debug) logger.info("正在打开目标网站...");
+      await page.goto("https://unwatermark.ai/", { waitUntil: "networkidle2" });
+      if (config.model !== "Auto V2.3") {
+        const modelSelector = `::-p-text(${config.model})`;
+        try {
+          if (config.debug) logger.info(`正在选择模型: ${config.model}`);
+          await page.waitForSelector(modelSelector, { timeout: 5e3 });
+          await page.click(modelSelector);
+        } catch (e) {
+          logger.warn(`无法选择模型 ${config.model}`);
+        }
+      }
+      const toggle = /* @__PURE__ */ __name(async (text, enable) => {
+        if (!enable) return;
+        const selector = `::-p-text(${text})`;
+        try {
+          if (config.debug) logger.info(`正在开启选项: ${text}`);
+          await page.waitForSelector(selector, { timeout: 2e3 });
+          await page.click(selector);
+        } catch (e) {
+          logger.warn(`无法开启选项 ${text}`);
+        }
+      }, "toggle");
+      await toggle("Remove Text", config.removeText);
+      await toggle("Remove Logo", config.removeLogo);
+      await toggle("Enhance Quality", config.enhanceQuality);
+      if (config.debug) logger.info("正在查找上传入口...");
+      const fileInput = await page.waitForSelector('input[type="file"]', { timeout: 1e4 });
+      if (config.debug) logger.info("找到上传入口，开始上传...");
+      await fileInput.uploadFile(tempFile);
+      if (config.debug) logger.info("等待处理结果（Download 按钮出现）...");
+      const downloadButtonSelector = "::-p-text(Download)";
+      await page.waitForSelector(downloadButtonSelector, { timeout: config.timeout, visible: true });
+      let resultData = null;
+      let resultBuffer = null;
+      let resultMimeType = "image/jpeg";
+      let resultSource = "unknown";
+      try {
+        if (config.debug) logger.info("处理完成，尝试通过 Download 按钮获取结果图...");
+        const responsePromise = page.waitForResponse((response2) => {
+          const url = response2.url();
+          const contentType = response2.headers()["content-type"] || "";
+          const isOutput = url.includes("/output/");
+          const isImageType = contentType.includes("image/");
+          const isImageExt = /\.(png|jpe?g|webp|bmp|gif)(\?|$)/i.test(url);
+          return isOutput && (isImageType || isImageExt);
+        }, { timeout: Math.min(config.timeout, 15e3) });
+        await page.click(downloadButtonSelector);
+        const response = await responsePromise;
+        const downloadUrl = response.url();
+        resultMimeType = (response.headers()["content-type"] || "image/jpeg").split(";")[0];
+        const downloadBuffer = await ctx.http.get(downloadUrl, { responseType: "arraybuffer" });
+        resultBuffer = Buffer.from(downloadBuffer);
+        resultSource = "download-response";
+        if (config.debug) logger.info(`通过 Download 响应获取结果图片: ${downloadUrl}`);
+      } catch (e) {
+        if (config.debug) logger.warn("通过 Download 按钮获取结果失败，回退到页面结果图解析。");
+      }
+      if (!resultBuffer) {
+        if (config.debug) logger.info("开始解析页面结果图片...");
+        await page.waitForFunction(() => {
+          const imgs = Array.from(document.querySelectorAll("img.comparison-image"));
+          return imgs.some((img) => {
+            const src = img.src || "";
+            if (src.includes("watermark_remove_before")) return false;
+            return src.includes("watermark_remove_after") || src.includes("/output/") || src.startsWith("blob:") || src.startsWith("data:image/");
+          });
+        }, { timeout: config.timeout });
+        resultData = await page.evaluate(async () => {
+          const images = Array.from(document.querySelectorAll("img.comparison-image"));
+          const getScore = /* @__PURE__ */ __name((src) => {
+            if (!src || src.includes("watermark_remove_before")) return -1;
+            if (src.includes("watermark_remove_after")) return 4;
+            if (src.includes("/output/")) return 3;
+            if (src.startsWith("blob:")) return 2;
+            if (src.startsWith("data:image/")) return 2;
+            if (src.startsWith("http://") || src.startsWith("https://")) return 1;
+            return 0;
+          }, "getScore");
+          const candidate = images.map((img) => img.src).sort((a, b) => getScore(b) - getScore(a))[0];
+          if (!candidate || getScore(candidate) < 0) return null;
+          if (candidate.startsWith("blob:")) {
+            const response = await fetch(candidate);
+            const blob = await response.blob();
+            return new Promise((resolve2, reject) => {
+              const reader = new FileReader();
+              reader.onloadend = () => resolve2(reader.result);
+              reader.onerror = reject;
+              reader.readAsDataURL(blob);
+            });
+          }
+          return candidate;
+        });
+        if (!resultData) {
+          throw new Error("未找到处理后的图片结果");
+        }
+        resultSource = typeof resultData === "string" && resultData.startsWith("data:") ? "dom-blob-fallback" : "dom-after";
+      }
+      if (config.debug) {
+        const resultDisplay = resultBuffer ? `[Buffer ${resultMimeType}]` : typeof resultData === "string" && resultData.startsWith("data:") ? "Base64 Data" : resultData;
+        logger.info(`找到结果图片地址/数据(${resultSource}): ${resultDisplay}`);
+      }
+      try {
+        if (session.bot.deleteMessage) {
+          await session.bot.deleteMessage(session.channelId, processingMsgId);
+        }
+      } catch (e) {
+        logger.warn("撤回消息失败", e);
+      }
+      if (resultBuffer) {
+        return import_koishi.h.image(resultBuffer, resultMimeType);
+      }
+      if (typeof resultData === "string" && resultData.startsWith("data:")) {
+        const base64Data = resultData.split(",")[1];
+        const buffer2 = Buffer.from(base64Data, "base64");
+        const mimeType = resultData.match(/data:([^;]+);/)?.[1] || "image/png";
+        return import_koishi.h.image(buffer2, mimeType);
+      }
+      return import_koishi.h.image(resultData);
+    } catch (e) {
+      logger.error("处理失败:", e);
+      return `处理失败: ${e.message || "未知错误"}`;
+    } finally {
+      await page.close();
+      await import_fs.promises.unlink(tempFile).catch(() => {
+      });
+    }
+  });
+}
+__name(apply, "apply");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  inject,
+  name
+});

package/package.json

@@ -0,0 +1,23 @@
+{
+  "name": "koishi-plugin-watermark-remover",
+  "description": "",
+  "version": "0.0.1",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "scripts": {},
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "devDependencies": {},
+  "peerDependencies": {
+    "koishi": "4.18.9",
+    "koishi-plugin-puppeteer": "*"
+  }
+}

package/readme.md

@@ -0,0 +1,59 @@
+# koishi-plugin-watermark-remover
+
+一个用于 Koishi 的去水印插件，基于 Puppeteer 自动操作 `https://unwatermark.ai/`。
+
+## 功能
+
+- 命令：`wmremove [image:image]`
+- 支持从参数或消息内容中读取图片
+- 可选项：`Remove Text`、`Remove Logo`、`Enhance Quality`
+- 处理完成后自动返回结果图片
+
+## 安装
+
+确保已安装并启用以下依赖：
+
+- `koishi`
+- `koishi-plugin-puppeteer`
+
+在 Koishi 中安装本插件后启用即可。
+
+## 配置
+
+- `model`：模型选择（默认 `Auto V2.3`）
+- `removeText`：是否移除文字
+- `removeLogo`：是否移除 Logo
+- `enhanceQuality`：是否增强画质
+- `debug`：是否输出调试日志
+- `timeout`：处理超时时间（毫秒）
+
+## 使用
+
+发送命令并附带图片：
+
+```text
+wmremove
+```
+
+或：
+
+```text
+wmremove [图片]
+```
+
+## 注意
+unwatermark 每日有免费次数上限，请适当使用。
+
+## 免责声明
+
+本插件通过自动化方式调用第三方网站服务。使用者需自行承担以下责任：
+
+- 遵守目标网站的服务条款、使用政策及法律法规
+- 确认上传图片的版权、隐私和数据合规性
+- 自行评估第三方服务稳定性、可用性和数据安全风险
+
+作者与贡献者不对因使用本插件导致的服务中断、数据泄露、侵权纠纷或其他损失承担责任。
+
+## License
+
+MIT