koishi-plugin-chatsound 0.0.3 → 1.0.1

package/lib/index.d.ts

@@ -7,13 +7,14 @@ declare module 'koishi' {
 }
 export declare const name = "chatsound";
 export declare const inject: {
-    required: string[];
+    optional: string[];
 };
 export interface Config {
     soundPath: string[];
     defaultPitch: number;
     minPitch: number;
     maxPitch: number;
+    keepSoundLength: boolean;
     audioType: "mp3" | "silk";
 }
 export declare const Config: Schema<Config>;

package/lib/index.js

@@ -30,20 +30,62 @@ var import_koishi = require("koishi");
 var import_promises = require("fs/promises");
 var import_path = require("path");
 var import_child_process = require("child_process");
-var import_util = require("util");
-var execAsync = (0, import_util.promisify)(import_child_process.exec);
 var name = "chatsound";
 var inject = {
-  required: ["silk"]
+  optional: ["silk"]
 };
 var Config = import_koishi.Schema.object({
-  soundPath: import_koishi.Schema.array(import_koishi.Schema.string()).description("用于搜索的音频路径"),
+  soundPath: import_koishi.Schema.array(import_koishi.Schema.string()).description("用于搜索音频的*绝对*路径，支持搜索mp3 wav ogg flac m4a aac格式"),
   defaultPitch: import_koishi.Schema.number().default(100).description("默认的语音音调（百分比）"),
   minPitch: import_koishi.Schema.number().default(80).min(0).description("最小Pitch，不小于0"),
   maxPitch: import_koishi.Schema.number().default(200).description("最大Pitch"),
+  keepSoundLength: import_koishi.Schema.boolean().default(false).description("保持音频播放长度，仅进行  变调"),
   audioType: import_koishi.Schema.union(["mp3", "silk"]).default("mp3").description("最终发送的类型，QQ及微信选择SILK")
 });
 var AUDIO_EXTENSIONS = [".mp3", ".wav", ".ogg", ".flac", ".m4a", ".aac"];
+var ffmpegAvailable = null;
+var ffmpegCheckPromise = null;
+async function checkFFmpeg() {
+  if (ffmpegAvailable !== null)
+    return ffmpegAvailable;
+  if (ffmpegCheckPromise)
+    return ffmpegCheckPromise;
+  ffmpegCheckPromise = new Promise((resolve) => {
+    const { spawn: spawn2 } = require("child_process");
+    let resolved = false;
+    const timeout = setTimeout(() => {
+      if (!resolved) {
+        resolved = true;
+        resolve(false);
+      }
+    }, 3e3);
+    try {
+      const ffmpeg = spawn2("ffmpeg", ["-version"]);
+      ffmpeg.on("error", () => {
+        if (!resolved) {
+          resolved = true;
+          clearTimeout(timeout);
+          resolve(false);
+        }
+      });
+      ffmpeg.on("close", (code) => {
+        if (!resolved) {
+          resolved = true;
+          clearTimeout(timeout);
+          resolve(code === 0);
+        }
+      });
+    } catch {
+      clearTimeout(timeout);
+      resolve(false);
+    }
+  }).then((result) => {
+    ffmpegAvailable = result;
+    return result;
+  });
+  return ffmpegCheckPromise;
+}
+__name(checkFFmpeg, "checkFFmpeg");
 async function findAudioFile(trigger, soundPaths) {
   for (const basePath of soundPaths) {
     try {
@@ -69,8 +111,6 @@ async function runFFmpeg(commandArgs) {
     const ffmpeg = (0, import_child_process.spawn)("ffmpeg", commandArgs);
     const chunks = [];
     ffmpeg.stdout.on("data", (chunk) => chunks.push(chunk));
-    ffmpeg.stderr.on("data", (chunk) => {
-    });
     ffmpeg.on("close", (code) => {
       if (code === 0) {
         resolve(Buffer.concat(chunks));
@@ -82,49 +122,41 @@ async function runFFmpeg(commandArgs) {
   });
 }
 __name(runFFmpeg, "runFFmpeg");
-async function applyPitch(inputPath, pitch, audioType) {
+async function applyPitch(inputPath, pitch, audioType, keepSoundLength) {
   const pitchFactor = pitch / 100;
+  let args = [
+    "-i",
+    inputPath,
+    "-vn",
+    "-sn",
+    "-dn",
+    "-y"
+  ];
+  if (keepSoundLength) {
+    args.push("-af", `rubberband=pitch=${pitchFactor}`);
+  } else {
+    args.push("-af", `asetrate=sample_rate*${pitchFactor}`);
+  }
   if (audioType === "mp3") {
-    const args = [
-      "-i",
-      inputPath,
-      "-vn",
-      "-sn",
-      "-dn",
-      "-filter:a",
-      `rubberband=pitch=${pitchFactor}`,
-      "-f",
-      "mp3",
-      "-y",
-      "pipe:1"
-    ];
+    args.push("-f", "mp3", "pipe:1");
     const data = await runFFmpeg(args);
     return { data, mimeType: "audio/mp3" };
   } else {
-    const args = [
-      "-i",
-      inputPath,
-      "-vn",
-      "-sn",
-      "-dn",
-      "-filter:a",
-      `rubberband=pitch=${pitchFactor}`,
-      "-f",
-      "s16le",
-      "-ac",
-      "1",
-      "-ar",
-      "24000",
-      "-y",
-      "pipe:1"
-    ];
+    args.push("-f", "s16le", "-ac", "1", "-ar", "24000", "pipe:1");
     const data = await runFFmpeg(args);
     return { data, mimeType: "audio/pcm" };
   }
 }
 __name(applyPitch, "applyPitch");
 function apply(ctx, config) {
-  ctx.command("v <trigger:string> [pitch:number]").action(async ({ session }, trigger, pitch) => {
+  checkFFmpeg().then((available) => {
+    if (!available) {
+      ctx.logger.warn("FFmpeg not found! Audio processing will fail.");
+    } else {
+      ctx.logger.debug("FFmpeg is available.");
+    }
+  });
+  ctx.command("v <trigger:string> [pitch:number] 发送一个噪音").action(async ({ session }, trigger, pitch) => {
     if (!trigger) {
       return "至少给个名字吧大王";
     }
@@ -134,10 +166,14 @@ function apply(ctx, config) {
       return `没有这种音频`;
     }
     try {
-      const { data, mimeType } = await applyPitch(audioPath, actualPitch, config.audioType);
+      const { data, mimeType } = await applyPitch(audioPath, actualPitch, config.audioType, config.keepSoundLength);
       if (config.audioType === "silk") {
-        const silkResult = await ctx.silk.encode(data, 24e3);
-        await session.send(import_koishi.h.audio(silkResult.data, "audio/silk"));
+        if (ctx.silk) {
+          const silkResult = await ctx.silk.encode(data, 24e3);
+          await session.send(import_koishi.h.audio(silkResult.data, "audio/silk"));
+        } else {
+          return "没有安装必要的SILK插件";
+        }
       } else {
         await session.send(import_koishi.h.audio(data, mimeType));
       }

package/package.json

@@ -4,7 +4,7 @@
   "contributors": [
     "Dr.Abc <me@drabc.net>"
   ],
-  "version": "0.0.3",
+  "version": "1.0.1",
   "homepage": "https://github.com/DrAbcOfficial/koishi-plugin-chatsound",
   "repository": {
     "type": "git",
@@ -33,7 +33,7 @@
       "zh": "发送预定的噪音，可以变调，依赖ffmpeg"
     },
     "service":{
-      "required": ["silk"]
+      "optional": ["silk"]
     },
     "preview": true
   }