把@y68的wuyu-plugin的fanqie下载搬过来

https://github.com/wuyubot/wuyu-plugin/tree/master

apps/fanqie.js

@@ -0,0 +1,138 @@
+import path from 'path';
+import fs from 'node:fs';
+import Fanqie from './fanqie.js';
+import ConfigControl from '../lib/config/configControl.js';
+
+let redis = global.redis;
+
+const outDir = await ConfigControl.get('fanqieConfig')?.outDir;
+const apiurl = await ConfigControl.get('fanqieConfig')?.url;
+let fq = new Fanqie(apiurl);
+
+function clearCache(e, isTask = false, sessionId = null) {
+  const dir = path.join(outDir, 'fanqie');
+  if (sessionId) {
+    const cdir = path.join(outDir, 'files', sessionId);
+    if (fs.existsSync(cdir)) {
+      fs.rmSync
+        ? fs.rmSync(cdir, { recursive: true, force: true })
+        : fs.rmdirSync(cdir, { recursive: true });
+    }
+    return true;
+  }
+
+  if (fs.existsSync(dir)) {
+    const files = fs.readdirSync(dir);
+    for (const file of files) {
+      const filePath = path.join(dir, file);
+      const stat = fs.statSync(filePath);
+      if (stat.isDirectory()) {
+        fs.rmSync
+          ? fs.rmSync(filePath, { recursive: true, force: true })
+          : fs.rmdirSync(filePath, { recursive: true });
+      } else {
+        fs.unlinkSync(filePath);
+      }
+    }
+  }
+  if (!isTask && e) e.reply('缓存已清理', true);
+  return true;
+}
+
+async function upload(e, filePath) {
+  try {
+    let res;
+    if (e.isGroup) {
+      res = e.group.sendFile
+        ? await e.group.sendFile(filePath)
+        : await e.group.fs?.upload?.(filePath);
+    } else {
+      res = await e.friend?.sendFile?.(filePath);
+    }
+    return !!res;
+  } catch (err) {
+    logger.error(`上传失败：${err.stack}`);
+    e.reply(`上传失败：${err.stack}`);
+    return false;
+  }
+}
+
+export default {
+  name: 'fanqie',
+  desc: '番茄小说下载',
+  event: 'message',
+  priority: -114,
+  rule: [
+    {
+      reg: '(changdunovel.com/wap/share-v2.html|fanqienovel.com/page)',
+      fnc: 'handleJX',
+    },
+    {
+      reg: '#?fq下载(.*)',
+      fnc: 'handleID',
+    },
+    {
+      reg: '^fq(允许|禁止)(群|用户)使用',
+      fnc: 'handleControl',
+    },
+    {
+      reg: '^fq清(理|除|空)缓存$',
+      fnc: 'handleClear',
+    },
+  ],
+  async handleJX(e) {
+    const msg = e.msg.trim();
+    const book_id = msg.includes('changdunovel')
+      ? msg.match(/book_id=(\d+)/)?.[1]
+      : msg.match(/page\/(\d+)/)?.[1];
+    if (!book_id) return e.reply('无法识别ID');
+    return this.handleDownload(e, book_id);
+  },
+  async handleID(e) {
+    const id = e.msg
+      .trim()
+      .replace(/^#?fq下载/, '')
+      .trim();
+    return this.handleDownload(e, id);
+  },
+  async handleControl(e) {
+    if (!e.isMaster) return e.reply('你没有权限使用此功能');
+    let id = e.msg.trim().replace(/^fq(允许|禁止)(群|用户)使用/, '') || e.group_id;
+    const key = e.msg.includes('群') ? `fqxzq:g:${id}` : `fqxzq:u:${id}`;
+    redis.set(key, e.msg.includes('允许'));
+    e.reply(
+      `已${e.msg.includes('允许') ? '允许' : '禁止'}${e.msg.includes('群') ? '群' : '用户'} ${id} 使用`,
+      true
+    );
+  },
+  async handleClear(e) {
+    if (!e.isMaster) return e.reply('你没有权限使用此功能');
+    clearCache(e);
+  },
+  async handleDownload(e, book_id) {
+    const book_info = await fq.get_info(book_id);
+    if (!book_info) return e.reply('获取信息失败');
+    e.reply(
+      `识别: [番茄小说]《${book_info.book_name}》\n作者: ${book_info.author}\n原名: ${book_info.original_book_name}`,
+      true
+    );
+
+    if (!e.isMaster) {
+      const groupAllow = await redis.get(`fqxzq:g:${e.group_id}`);
+      const userDeny = await redis.get(`fqxzq:u:${e.user_id}`);
+      if (!groupAllow || userDeny) return;
+    }
+
+    e.reply('开始下载，请稍等...');
+    const ok = await fq.down(book_id, e.message_id);
+    if (!ok) return e.reply('下载失败');
+
+    const file = await fq.listen_outdir(path.join(outDir, 'files', String(e.message_id)));
+    if (!file) return e.reply('下载超时');
+
+    const success = await upload(e, file);
+    clearCache(null, true, String(e.message_id));
+    if (!success) return e.reply('上传失败');
+    e.reply(`《${book_info.book_name}》上传成功！`, true);
+  },
+};

config/default.json

@@ -6,7 +6,12 @@
     "wsUrl": "",
     "wsClientId": "",
     "wsSecret": "",
-    "wsReConnectInterval": "5000"
+    "wsReConnectInterval": "5000",
+    "token": ""
+  },
+  "fanqieConfig": {
+    "url": "http://127.0.0.1:6868",
+    "outDir": "/home/user/debian/cache/Downloads"
   },
   "mode": "deepseek",
   "modelType": "deepseek-ai/DeepSeek-V3",

models/apps/fanqie/fanqie.js

@@ -0,0 +1,58 @@
+import axios from 'axios';
+import chokidar from 'chokidar';
+
+class Fanqie {
+  constructor(apiurl) {
+    this.apiurl = apiurl;
+  }
+
+  async get_info(book_id) {
+    try {
+      const url = `${this.apiurl}/api/info?book_id=${book_id}&source=fanqie`;
+      const res = await axios.get(url);
+      if (res.status !== 200 || !res.data) throw new Error('请求失败或无数据');
+      const result = res.data['data'];
+      if (!result) throw new Error('data 字段不存在');
+      return {
+        author: result.author,
+        book_name: result.book_name,
+        original_book_name: result.original_book_name,
+      };
+    } catch (e) {
+      logger.error(e);
+      return false;
+    }
+  }
+
+  async down(book_id, msg_id) {
+    try {
+      const url = `${this.apiurl}/api/down?book_id=${book_id}&source=fanqie&type=txt&user_id=${msg_id}`;
+      await axios.get(url);
+      return true;
+    } catch (e) {
+      logger.error(e);
+      return false;
+    }
+  }
+
+  listen_outdir(dir, timeout = 30000) {
+    if (!dir) return false;
+    return new Promise((resolve) => {
+      const watcher = chokidar.watch(dir, {
+        persistent: true,
+        ignoreInitial: true,
+      });
+      const timer = setTimeout(() => {
+        watcher.close();
+        resolve(false);
+      }, timeout);
+      watcher.on('add', (filePath) => {
+        clearTimeout(timer);
+        watcher.close();
+        resolve(filePath);
+      });
+    });
+  }
+}
+
+export default Fanqie;