koishi-plugin-music-to-voice 1.0.0

package/dist/index.mjs

@@ -0,0 +1,366 @@
+// src/index.ts
+import { Schema, Logger, h } from "koishi";
+import axios from "axios";
+import fs from "fs";
+import path from "path";
+import os from "os";
+import crypto from "crypto";
+var name = "music-voice-pro";
+var logger = new Logger(name);
+var Config = Schema.object({
+  commandName: Schema.string().default("\u542C\u6B4C").description("\u6307\u4EE4\u540D\u79F0"),
+  commandAlias: Schema.string().default("music").description("\u6307\u4EE4\u522B\u540D"),
+  apiBase: Schema.string().default("https://music-api.gdstudio.xyz/api.php").description("\u97F3\u4E50 API \u5730\u5740\uFF08GD\u97F3\u4E50\u53F0 API\uFF09"),
+  source: Schema.union([
+    Schema.const("netease").description("\u7F51\u6613\u4E91"),
+    Schema.const("tencent").description("QQ\u97F3\u4E50"),
+    Schema.const("kugou").description("\u9177\u72D7"),
+    Schema.const("kuwo").description("\u9177\u6211"),
+    Schema.const("migu").description("\u54AA\u5495"),
+    Schema.const("baidu").description("\u767E\u5EA6")
+  ]).default("netease").description("\u97F3\u6E90\uFF08\u4E0B\u62C9\u9009\u62E9\uFF09"),
+  searchListCount: Schema.number().min(5).max(50).default(20).description("\u641C\u7D22\u5217\u8868\u6570\u91CF"),
+  waitForTimeout: Schema.number().min(10).max(180).default(45).description("\u7B49\u5F85\u8F93\u5165\u5E8F\u53F7\u8D85\u65F6\uFF08\u79D2\uFF09"),
+  nextPageCommand: Schema.string().default("\u4E0B\u4E00\u9875").description("\u4E0B\u4E00\u9875\u6307\u4EE4"),
+  prevPageCommand: Schema.string().default("\u4E0A\u4E00\u9875").description("\u4E0A\u4E00\u9875\u6307\u4EE4"),
+  exitCommandList: Schema.array(String).default(["0", "\u4E0D\u542C\u4E86", "\u9000\u51FA"]).description("\u9000\u51FA\u6307\u4EE4\u5217\u8868"),
+  menuExitCommandTip: Schema.boolean().default(false).description("\u662F\u5426\u5728\u6B4C\u5355\u672B\u5C3E\u63D0\u793A\u9000\u51FA\u6307\u4EE4"),
+  imageMode: Schema.boolean().default(false).description("\u56FE\u7247\u6B4C\u5355\u6A21\u5F0F\uFF08\u53EF\u9009\uFF1A\u9700\u8981 puppeteer \u63D2\u4EF6\uFF0C\u5F53\u524D\u4EC5\u4FDD\u7559\u5F00\u5173\uFF09"),
+  sendAs: Schema.union([
+    Schema.const("record").description("\u8BED\u97F3 record"),
+    Schema.const("audio").description("\u97F3\u9891 audio")
+  ]).default("record").description("\u53D1\u9001\u7C7B\u578B"),
+  forceTranscode: Schema.boolean().default(true).description("\u662F\u5426\u5F3A\u5236\u8F6C\u7801\u4E3A silk\uFF08\u9700\u8981 ffmpeg + silk \u63D2\u4EF6\uFF09"),
+  tempDir: Schema.string().default(path.join(os.tmpdir(), "koishi-music-voice")).description("\u4E34\u65F6\u76EE\u5F55"),
+  cacheMinutes: Schema.number().min(0).max(1440).default(120).description("\u7F13\u5B58\u65F6\u957F\uFF08\u5206\u949F\uFF0C0=\u4E0D\u7F13\u5B58\uFF09"),
+  generationTip: Schema.string().default("\u751F\u6210\u8BED\u97F3\u4E2D...").description("\u7528\u6237\u9009\u6B4C\u540E\u63D0\u793A"),
+  recallSearchMenuMessage: Schema.boolean().default(true).description("\u64A4\u56DE\uFF1A\u6B4C\u5355\u6D88\u606F"),
+  recallTipMessage: Schema.boolean().default(true).description("\u64A4\u56DE\uFF1A\u751F\u6210\u63D0\u793A\u6D88\u606F"),
+  recallUserSelectMessage: Schema.boolean().default(true).description("\u64A4\u56DE\uFF1A\u7528\u6237\u8F93\u5165\u7684\u5E8F\u53F7\u6D88\u606F"),
+  recallVoiceMessage: Schema.boolean().default(false).description("\u64A4\u56DE\uFF1A\u8BED\u97F3\u6D88\u606F"),
+  loggerinfo: Schema.boolean().default(false).description("\u65E5\u5FD7\u8C03\u8BD5\u6A21\u5F0F")
+}).description("\u70B9\u6B4C\u8BED\u97F3\uFF08\u652F\u6301\u7FFB\u9875 + \u53EF\u9009 silk/ffmpeg\uFF09");
+function safeText(x) {
+  return typeof x === "string" ? x : x == null ? "" : String(x);
+}
+function normalizeSearchList(data) {
+  const arr = Array.isArray(data) ? data : Array.isArray(data?.result) ? data.result : Array.isArray(data?.data) ? data.data : Array.isArray(data?.songs) ? data.songs : [];
+  return arr.map((it) => {
+    const id = safeText(it?.id ?? it?.songid ?? it?.rid ?? it?.hash ?? it?.mid);
+    const name2 = safeText(it?.name ?? it?.songname ?? it?.title);
+    const artist = safeText(it?.artist) || safeText(it?.singer) || safeText(it?.author) || (Array.isArray(it?.artists) ? it.artists.map((a) => safeText(a?.name)).filter(Boolean).join("/") : "");
+    const album = safeText(it?.album ?? it?.albummid ?? it?.albumname);
+    return { id, name: name2, artist, album };
+  }).filter((x) => x.id && x.name);
+}
+function normalizeUrl(data) {
+  return safeText(data?.url) || safeText(data?.data?.url) || safeText(data?.result?.url) || safeText(data?.data) || "";
+}
+function md5(s) {
+  return crypto.createHash("md5").update(s).digest("hex");
+}
+function ensureDir(p) {
+  fs.mkdirSync(p, { recursive: true });
+}
+async function tryRecall(session, messageId) {
+  if (!messageId) return;
+  try {
+    await session.bot.deleteMessage(session.channelId, messageId);
+  } catch {
+  }
+}
+function hRecord(src) {
+  return h("record", { src });
+}
+function hAudio(src) {
+  return h("audio", { src });
+}
+function buildMenuText(config, keyword, list, page) {
+  const lines = [];
+  lines.push(`NetEase Music:`);
+  lines.push(`\u5173\u952E\u8BCD\uFF1A${keyword}`);
+  lines.push(`\u97F3\u6E90\uFF1A${config.source}  \u7B2C ${page} \u9875`);
+  lines.push("");
+  for (let i = 0; i < list.length; i++) {
+    const it = list[i];
+    const meta = [it.artist, it.album].filter(Boolean).join(" - ");
+    lines.push(`${i + 1}. ${it.name}${meta ? ` -- ${meta}` : ""}`);
+  }
+  lines.push("");
+  lines.push(`\u8BF7\u5728 ${config.waitForTimeout} \u79D2\u5185\u8F93\u5165\u5E8F\u53F7\uFF081-${list.length}\uFF09`);
+  lines.push(`\u7FFB\u9875\uFF1A${config.prevPageCommand} / ${config.nextPageCommand}`);
+  if (config.menuExitCommandTip) {
+    lines.push(`\u9000\u51FA\uFF1A${config.exitCommandList.join(" / ")}`);
+  }
+  lines.push("");
+  lines.push("\u6570\u636E\u6765\u6E90\uFF1AGD\u97F3\u4E50\u53F0 API");
+  return lines.join("\n");
+}
+async function apiSearch(config, keyword, page = 1) {
+  const params = {
+    types: "search",
+    source: config.source,
+    name: keyword,
+    count: config.searchListCount,
+    pages: page
+  };
+  const { data } = await axios.get(config.apiBase, { params, timeout: 15e3 });
+  return normalizeSearchList(data);
+}
+async function apiGetSongUrl(config, id) {
+  const params = { types: "url", source: config.source, id };
+  const { data } = await axios.get(config.apiBase, { params, timeout: 15e3 });
+  return normalizeUrl(data);
+}
+async function downloadToFile(url, filePath) {
+  const res = await axios.get(url, { responseType: "stream", timeout: 3e4 });
+  await new Promise((resolve, reject) => {
+    const ws = fs.createWriteStream(filePath);
+    res.data.pipe(ws);
+    ws.on("finish", () => resolve());
+    ws.on("error", reject);
+  });
+}
+async function sendVoiceByUrl(session, config, audioUrl) {
+  const seg = config.sendAs === "record" ? hRecord(audioUrl) : hAudio(audioUrl);
+  const ids = await session.send(seg);
+  return Array.isArray(ids) ? ids[0] : ids;
+}
+async function sendVoiceByFile(session, config, absPath) {
+  const url = `file://${absPath.replace(/\\/g, "/")}`;
+  const seg = config.sendAs === "record" ? hRecord(url) : hAudio(url);
+  const ids = await session.send(seg);
+  return Array.isArray(ids) ? ids[0] : ids;
+}
+async function buildSilkIfPossible(ctx, config, audioUrl, cacheKey) {
+  const hasFfmpeg = !!ctx.ffmpeg;
+  const hasSilk = !!ctx.silk;
+  if (!hasFfmpeg || !hasSilk) return null;
+  ensureDir(config.tempDir);
+  const silkPath = path.join(config.tempDir, `${cacheKey}.silk`);
+  if (config.cacheMinutes > 0 && fs.existsSync(silkPath)) {
+    return silkPath;
+  }
+  const rawPath = path.join(config.tempDir, `${cacheKey}.src`);
+  await downloadToFile(audioUrl, rawPath);
+  const wavPath = path.join(config.tempDir, `${cacheKey}.wav`);
+  const ffmpeg = ctx.ffmpeg;
+  try {
+    if (typeof ffmpeg.convert === "function") {
+      await ffmpeg.convert(rawPath, wavPath, {
+        format: "wav",
+        audioChannels: 1,
+        audioFrequency: 24e3
+      });
+    } else if (typeof ffmpeg.exec === "function") {
+      await ffmpeg.exec(["-y", "-i", rawPath, "-ac", "1", "-ar", "24000", wavPath]);
+    } else {
+      throw new Error("ffmpeg service API not recognized");
+    }
+  } catch (e) {
+    throw new Error(`ffmpeg \u8F6C\u7801\u5931\u8D25\uFF1A${e?.message || String(e)}`);
+  }
+  const silk = ctx.silk;
+  try {
+    if (typeof silk.encode === "function") {
+      await silk.encode(wavPath, silkPath, { rate: 24e3 });
+    } else if (typeof silk.encodeWav === "function") {
+      await silk.encodeWav(wavPath, silkPath, { rate: 24e3 });
+    } else {
+      throw new Error("silk service API not recognized");
+    }
+  } catch (e) {
+    throw new Error(`silk \u7F16\u7801\u5931\u8D25\uFF1A${e?.message || String(e)}`);
+  } finally {
+    try {
+      fs.unlinkSync(rawPath);
+    } catch {
+    }
+    try {
+      fs.unlinkSync(wavPath);
+    } catch {
+    }
+  }
+  return silkPath;
+}
+function logDepsHint(ctx) {
+  const hasPuppeteer = !!ctx.puppeteer;
+  const hasFfmpeg = !!ctx.ffmpeg;
+  const hasSilk = !!ctx.silk;
+  const hasDownloads = !!ctx.downloads;
+  logger.info("\u5F00\u542F\u63D2\u4EF6\u524D\uFF0C\u8BF7\u786E\u4FDD\u4EE5\u4E0B\u670D\u52A1\u5DF2\u7ECF\u542F\u7528\uFF08\u53EF\u9009\u5B89\u88C5\uFF09\uFF1A");
+  logger.info(`- puppeteer\u670D\u52A1\uFF08\u53EF\u9009\u5B89\u88C5\uFF09\uFF1A${hasPuppeteer ? "\u5DF2\u68C0\u6D4B\u5230" : "\u672A\u68C0\u6D4B\u5230"}`);
+  logger.info("\u6B64\u5916\u53EF\u80FD\u8FD8\u9700\u8981\u8FD9\u4E9B\u670D\u52A1\u624D\u80FD\u53D1\u9001\u8BED\u97F3\uFF1A");
+  logger.info(`- ffmpeg\u670D\u52A1\uFF08\u53EF\u9009\u5B89\u88C5\uFF09\uFF08\u6B64\u670D\u52A1\u53EF\u80FD\u989D\u5916\u4F9D\u8D56downloads\u670D\u52A1\uFF09\uFF1A${hasFfmpeg ? "\u5DF2\u68C0\u6D4B\u5230" : "\u672A\u68C0\u6D4B\u5230"}`);
+  logger.info(`- silk\u670D\u52A1\uFF08\u53EF\u9009\u5B89\u88C5\uFF09\uFF1A${hasSilk ? "\u5DF2\u68C0\u6D4B\u5230" : "\u672A\u68C0\u6D4B\u5230"}`);
+  logger.info(`- downloads\u670D\u52A1\uFF08\u53EF\u9009\u5B89\u88C5\uFF09\uFF1A${hasDownloads ? "\u5DF2\u68C0\u6D4B\u5230" : "\u672A\u68C0\u6D4B\u5230"}`);
+  logger.info("Music API \u51FA\u5904\uFF1AGD\u97F3\u4E50\u53F0 API\uFF08https://music-api.gdstudio.xyz/api.php\uFF09");
+}
+function apply(ctx, config) {
+  if (config.loggerinfo) logger.level = Logger.DEBUG;
+  logDepsHint(ctx);
+  const pending = /* @__PURE__ */ new Map();
+  function getKey(session) {
+    return String(session?.channelId || "");
+  }
+  function isExit(input) {
+    const t = input.trim();
+    return config.exitCommandList.map((s) => s.trim()).includes(t);
+  }
+  function isExpired(state) {
+    return Date.now() > state.expiresAt;
+  }
+  async function refreshMenu(session, state) {
+    const list = await apiSearch(config, state.keyword, state.page);
+    state.list = list;
+    state.expiresAt = Date.now() + config.waitForTimeout * 1e3;
+    if (config.recallSearchMenuMessage) {
+      await tryRecall(session, state.menuMessageId);
+      state.menuMessageId = void 0;
+    }
+    if (!config.recallSearchMenuMessage) {
+      const text = buildMenuText(config, state.keyword, list, state.page);
+      const ids = await session.send(text);
+      state.menuMessageId = Array.isArray(ids) ? ids[0] : ids;
+    } else {
+      await session.send(`\u5DF2\u7FFB\u5230\u7B2C ${state.page} \u9875\uFF0C\u8BF7\u76F4\u63A5\u53D1\u9001\u5E8F\u53F7\uFF081-${list.length}\uFF09`);
+    }
+  }
+  async function handlePick(session, state, pickIndex) {
+    const item = state.list[pickIndex];
+    if (!item) {
+      await session.send(`\u5E8F\u53F7\u65E0\u6548\uFF0C\u8BF7\u8F93\u5165 1-${state.list.length}\uFF0C\u6216\u8F93\u5165 ${config.exitCommandList.join("/")} \u9000\u51FA\u3002`);
+      return;
+    }
+    let tipId;
+    if (!config.recallTipMessage && config.generationTip?.trim()) {
+      const ids = await session.send(config.generationTip);
+      tipId = Array.isArray(ids) ? ids[0] : ids;
+    }
+    let songUrl = "";
+    try {
+      songUrl = await apiGetSongUrl(config, item.id);
+      if (!songUrl) throw new Error("empty url");
+    } catch {
+      await session.send("\u83B7\u53D6\u6B4C\u66F2\u76F4\u94FE\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u518D\u8BD5\uFF0C\u6216\u66F4\u6362\u97F3\u6E90\u3002");
+      return;
+    }
+    const cacheKey = md5(`${config.source}:${item.id}`);
+    let voiceId;
+    try {
+      const silkPath = await buildSilkIfPossible(ctx, config, songUrl, cacheKey);
+      if (silkPath) {
+        voiceId = await sendVoiceByFile(session, config, silkPath);
+      } else {
+        if (config.forceTranscode) {
+          await session.send(
+            `\u5F53\u524D\u914D\u7F6E\u4E3A\u3010\u5F3A\u5236 silk \u8F6C\u7801\u3011\u4F46\u672A\u68C0\u6D4B\u5230 ffmpeg/silk \u670D\u52A1\u3002
+\u8BF7\u5728 Koishi \u63D2\u4EF6\u5E02\u573A\u5B89\u88C5\u5E76\u542F\u7528\uFF1Affmpeg\u3001silk\uFF08\u53EF\u80FD\u8FD8\u9700\u8981 downloads\uFF09\u3002`
+          );
+          return;
+        }
+        voiceId = await sendVoiceByUrl(session, config, songUrl);
+      }
+    } catch (e) {
+      await session.send(`\u751F\u6210\u8BED\u97F3\u5931\u8D25\uFF1A${e?.message || String(e)}
+\u8BF7\u68C0\u67E5 ffmpeg/silk \u63D2\u4EF6\u662F\u5426\u542F\u7528\uFF0C\u6216\u5173\u95ED\u201C\u5F3A\u5236\u8F6C\u7801\u201D\u3002`);
+      return;
+    } finally {
+      state.tipMessageId = tipId;
+      state.voiceMessageId = voiceId;
+    }
+    if (config.recallSearchMenuMessage) await tryRecall(session, state.menuMessageId);
+    if (config.recallTipMessage) await tryRecall(session, state.tipMessageId);
+    if (config.recallVoiceMessage) await tryRecall(session, state.voiceMessageId);
+    pending.delete(getKey(session));
+  }
+  ctx.command(`${config.commandName} <keyword:text>`, "\u70B9\u6B4C\u5E76\u53D1\u9001\u8BED\u97F3\uFF08GD\u97F3\u4E50\u53F0 API\uFF09").alias(config.commandAlias).action(async ({ session }, keyword) => {
+    if (!session) return;
+    keyword = (keyword || "").trim();
+    if (!keyword) return `\u7528\u6CD5\uFF1A${config.commandName} \u6B4C\u66F2\u540D`;
+    const userId = session.userId;
+    const channelId = session.channelId;
+    if (!userId || !channelId) {
+      await session.send("\u5F53\u524D\u9002\u914D\u5668\u672A\u63D0\u4F9B userId/channelId\uFF0C\u65E0\u6CD5\u8FDB\u5165\u9009\u6B4C\u6A21\u5F0F\u3002");
+      return;
+    }
+    let list = [];
+    try {
+      list = await apiSearch(config, keyword, 1);
+    } catch {
+      return "\u641C\u7D22\u5931\u8D25\uFF08API \u4E0D\u53EF\u7528\u6216\u8D85\u65F6\uFF09\uFF0C\u8BF7\u7A0D\u540E\u518D\u8BD5\u3002";
+    }
+    if (!list.length) return "\u6CA1\u6709\u641C\u5230\u7ED3\u679C\uFF0C\u6362\u4E2A\u5173\u952E\u8BCD\u8BD5\u8BD5\u3002";
+    const state = {
+      userId,
+      channelId,
+      guildId: session.guildId,
+      keyword,
+      page: 1,
+      list,
+      expiresAt: Date.now() + config.waitForTimeout * 1e3
+    };
+    if (!config.recallSearchMenuMessage) {
+      const text = buildMenuText(config, keyword, list, 1);
+      const ids = await session.send(text);
+      state.menuMessageId = Array.isArray(ids) ? ids[0] : ids;
+    } else {
+      await session.send(`\u5DF2\u8FDB\u5165\u9009\u6B4C\u6A21\u5F0F\uFF0C\u8BF7\u76F4\u63A5\u53D1\u9001\u5E8F\u53F7\uFF081-${list.length}\uFF09\uFF0C\u6216\u53D1\u9001\u201C${config.nextPageCommand}/${config.prevPageCommand}\u201D\u7FFB\u9875\u3002`);
+    }
+    pending.set(channelId, state);
+  });
+  ctx.middleware(async (session, next) => {
+    const key = getKey(session);
+    if (!key) return next();
+    const state = pending.get(key);
+    if (!state) return next();
+    if (session.userId !== state.userId) return next();
+    if (isExpired(state)) {
+      pending.delete(key);
+      return next();
+    }
+    const content = (session.content || "").trim();
+    if (!content) return next();
+    if (isExit(content)) {
+      if (config.recallSearchMenuMessage) await tryRecall(session, state.menuMessageId);
+      pending.delete(key);
+      if (!config.recallUserSelectMessage) await session.send("\u5DF2\u9000\u51FA\u9009\u6B4C\u3002");
+      return;
+    }
+    if (content === config.nextPageCommand) {
+      if (config.recallUserSelectMessage) await tryRecall(session, session.messageId);
+      state.page += 1;
+      try {
+        await refreshMenu(session, state);
+      } catch {
+        state.page -= 1;
+        await session.send("\u7FFB\u9875\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u518D\u8BD5\u3002");
+      }
+      return;
+    }
+    if (content === config.prevPageCommand) {
+      if (config.recallUserSelectMessage) await tryRecall(session, session.messageId);
+      if (state.page <= 1) {
+        await session.send("\u5DF2\u7ECF\u662F\u7B2C\u4E00\u9875\u3002");
+        return;
+      }
+      state.page -= 1;
+      try {
+        await refreshMenu(session, state);
+      } catch {
+        state.page += 1;
+        await session.send("\u7FFB\u9875\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u518D\u8BD5\u3002");
+      }
+      return;
+    }
+    const n = Number(content);
+    if (!Number.isInteger(n) || n < 1 || n > state.list.length) return next();
+    if (config.recallUserSelectMessage) await tryRecall(session, session.messageId);
+    await handlePick(session, state, n - 1);
+  });
+}
+export {
+  Config,
+  apply,
+  name
+};

package/package.json

@@ -0,0 +1,66 @@
+{
+  "name": "koishi-plugin-music-to-voice",
+  "version": "1.0.0",
+  "description": "音乐聚合点播语音插件：搜索歌曲、翻页选择、序号点播，可选转码发送语音。数据来源：GD音乐台 API。",
+  "main": "dist/index.js",
+  "module": "dist/index.mjs",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "src",
+    "README.md",
+    "LICENSE"
+  ],
+  "keywords": [
+    "koishi",
+    "koishi-plugin",
+    "music",
+    "voice",
+    "aggregate"
+  ],
+  "license": "MIT",
+  "author": "sdxcvbot <zhufuzhi1@gmail.com>",
+  "homepage": "https://github.com/sdxcvbot/koishi-plugin-music-to-voice#readme",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/sdxcvbot/koishi-plugin-music-to-voice.git"
+  },
+  "bugs": {
+    "url": "https://github.com/sdxcvbot/koishi-plugin-music-to-voice/issues"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "engines": {
+    "node": ">=16"
+  },
+  "peerDependencies": {
+    "koishi": "^4.18.0"
+  },
+  "dependencies": {
+    "axios": "^1.7.7"
+  },
+  "devDependencies": {
+    "koishi": "^4.18.0",
+    "tsup": "^8.2.4",
+    "typescript": "^5.4.5"
+  },
+  "scripts": {
+    "build": "tsup src/index.ts --format cjs,esm --dts",
+    "dev": "tsup src/index.ts --watch --format cjs,esm --dts",
+    "prepublishOnly": "npm run build"
+  },
+  "koishi": {
+    "description": {
+      "zh": "音乐聚合点播语音插件：搜索→列表→翻页→序号点播；可选 ffmpeg+silk 转码发送语音。数据来源：GD音乐台 API。"
+    },
+    "service": {
+      "optional": [
+        "puppeteer",
+        "downloads",
+        "ffmpeg",
+        "silk"
+      ]
+    }
+  }
+}