koishi-plugin-chatsound 0.0.2 → 0.0.3

package/lib/index.d.ts

@@ -1,10 +1,20 @@
 import { Context, Schema } from 'koishi';
+import SilkService from 'koishi-plugin-silk';
+declare module 'koishi' {
+    interface Context {
+        silk: SilkService;
+    }
+}
 export declare const name = "chatsound";
+export declare const inject: {
+    required: string[];
+};
 export interface Config {
     soundPath: string[];
     defaultPitch: number;
     minPitch: number;
     maxPitch: number;
+    audioType: "mp3" | "silk";
 }
 export declare const Config: Schema<Config>;
 export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -22,6 +22,7 @@ var src_exports = {};
 __export(src_exports, {
   Config: () => Config,
   apply: () => apply,
+  inject: () => inject,
   name: () => name
 });
 module.exports = __toCommonJS(src_exports);
@@ -32,11 +33,15 @@ 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"]
+};
 var Config = import_koishi.Schema.object({
   soundPath: import_koishi.Schema.array(import_koishi.Schema.string()).description("用于搜索的音频路径"),
   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")
+  maxPitch: import_koishi.Schema.number().default(200).description("最大Pitch"),
+  audioType: import_koishi.Schema.union(["mp3", "silk"]).default("mp3").description("最终发送的类型，QQ及微信选择SILK")
 });
 var AUDIO_EXTENSIONS = [".mp3", ".wav", ".ogg", ".flac", ".m4a", ".aac"];
 async function findAudioFile(trigger, soundPaths) {
@@ -59,11 +64,63 @@ async function findAudioFile(trigger, soundPaths) {
   return null;
 }
 __name(findAudioFile, "findAudioFile");
-async function applyPitch(inputPath, pitch) {
+async function runFFmpeg(commandArgs) {
+  return new Promise((resolve, reject) => {
+    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));
+      } else {
+        reject(new Error(`FFmpeg exited with code ${code}`));
+      }
+    });
+    ffmpeg.on("error", reject);
+  });
+}
+__name(runFFmpeg, "runFFmpeg");
+async function applyPitch(inputPath, pitch, audioType) {
   const pitchFactor = pitch / 100;
-  const command = `ffmpeg -i "${inputPath}" -filter:a "asetrate=44100*${pitchFactor},aresample=44100" -f mp3 -`;
-  const { stdout } = await execAsync(command);
-  return Buffer.from(stdout);
+  if (audioType === "mp3") {
+    const args = [
+      "-i",
+      inputPath,
+      "-vn",
+      "-sn",
+      "-dn",
+      "-filter:a",
+      `rubberband=pitch=${pitchFactor}`,
+      "-f",
+      "mp3",
+      "-y",
+      "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"
+    ];
+    const data = await runFFmpeg(args);
+    return { data, mimeType: "audio/pcm" };
+  }
 }
 __name(applyPitch, "applyPitch");
 function apply(ctx, config) {
@@ -77,8 +134,13 @@ function apply(ctx, config) {
       return `没有这种音频`;
     }
     try {
-      const audioBuffer = await applyPitch(audioPath, actualPitch);
-      await session.send(import_koishi.h.audio(audioBuffer, "audio/mp3"));
+      const { data, mimeType } = await applyPitch(audioPath, actualPitch, config.audioType);
+      if (config.audioType === "silk") {
+        const silkResult = await ctx.silk.encode(data, 24e3);
+        await session.send(import_koishi.h.audio(silkResult.data, "audio/silk"));
+      } else {
+        await session.send(import_koishi.h.audio(data, mimeType));
+      }
     } catch (error) {
       ctx.logger.error("发送音频失败:", error);
       return "发送音频失败";
@@ -90,5 +152,6 @@ __name(apply, "apply");
 0 && (module.exports = {
   Config,
   apply,
+  inject,
   name
 });

package/package.json

@@ -4,7 +4,7 @@
   "contributors": [
     "Dr.Abc <me@drabc.net>"
   ],
-  "version": "0.0.2",
+  "version": "0.0.3",
   "homepage": "https://github.com/DrAbcOfficial/koishi-plugin-chatsound",
   "repository": {
     "type": "git",
@@ -13,7 +13,8 @@
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
-    "lib"
+    "lib",
+    "dist"
   ],
   "license": "MIT",
   "scripts": {},
@@ -31,6 +32,9 @@
       "en": "发送预定的噪音，可以变调，依赖ffmpeg",
       "zh": "发送预定的噪音，可以变调，依赖ffmpeg"
     },
+    "service":{
+      "required": ["silk"]
+    },
     "preview": true
   }
 }