@dgck81lnn/koishi-plugin-music 0.1.1 → 0.1.3

package/browser/synth.js

@@ -1,6 +1,6 @@
 const sampleRate = 44100
 
-async function synth(notes) {
+async function synth(notes, { noise } = {}) {
   const seconds = Math.max(...notes.map(i => i.end))
 
   const ctx = new OfflineAudioContext(1, seconds * sampleRate, sampleRate)
@@ -26,6 +26,21 @@ async function synth(notes) {
     gain.gain.linearRampToValueAtTime(0, note.end)
   }
 
+  if (noise) {
+    // Add some white noise to avoid QQ's voice message encoding issues
+    const noiseBuf = ctx.createBuffer(1, Math.min(seconds, 10) * sampleRate, sampleRate)
+    for (let data = noiseBuf.getChannelData(0), i = 0; i < data.length; i++)
+      data[i] = Math.random() * 2 - 1
+    const noiseGain = ctx.createGain()
+    noiseGain.gain.value = 0.005
+    const noiseSrc = ctx.createBufferSource()
+    noiseSrc.buffer = noiseBuf
+    noiseSrc.loop = true
+    noiseSrc.connect(noiseGain)
+    noiseGain.connect(ctx.destination)
+    noiseSrc.start(0)
+  }
+
   const buf = await ctx.startRendering()
   return buf.getChannelData(0)
 }
@@ -58,40 +73,41 @@ function encodeWav(samples) {
   writeInt16(16) // sample depth
   writeBytes("data")
   writeInt32(samples.length * 2) // size of data chunk
-  for (let i = 0; i < samples.length; i++) writeInt16(samples[i] * 0x7fff)
+  for (let i = 0; i < samples.length; i++)
+    writeInt16(Math.min(Math.max(samples[i] * 0x7fff, -32768), 32767))
 
   return buffer
 }
 
 // https://gist.github.com/jonleighton/958841
-window.arrayBufferToBase64 = function (arrayBuffer) {
-  var base64    = ''
-  var encodings = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/'
-  var bytes         = new Uint8Array(arrayBuffer)
-  var byteLength    = bytes.byteLength
+function arrayBufferToBase64(arrayBuffer) {
+  var base64 = ""
+  var encodings = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
+  var bytes = new Uint8Array(arrayBuffer)
+  var byteLength = bytes.byteLength
   var byteRemainder = byteLength % 3
-  var mainLength    = byteLength - byteRemainder
+  var mainLength = byteLength - byteRemainder
   var a, b, c, d
   var chunk
   for (var i = 0; i < mainLength; i = i + 3) {
     chunk = (bytes[i] << 16) | (bytes[i + 1] << 8) | bytes[i + 2]
     a = (chunk & 16515072) >> 18
-    b = (chunk & 258048)   >> 12
-    c = (chunk & 4032)     >>  6
+    b = (chunk & 258048) >> 12
+    c = (chunk & 4032) >> 6
     d = chunk & 63
     base64 += encodings[a] + encodings[b] + encodings[c] + encodings[d]
   }
   if (byteRemainder == 1) {
     chunk = bytes[mainLength]
     a = (chunk & 252) >> 2
-    b = (chunk & 3)   << 4
-    base64 += encodings[a] + encodings[b] + '=='
+    b = (chunk & 3) << 4
+    base64 += encodings[a] + encodings[b] + "=="
   } else if (byteRemainder == 2) {
     chunk = (bytes[mainLength] << 8) | bytes[mainLength + 1]
     a = (chunk & 64512) >> 10
-    b = (chunk & 1008)  >>  4
-    c = (chunk & 15)    <<  2
-    base64 += encodings[a] + encodings[b] + encodings[c] + '='
+    b = (chunk & 1008) >> 4
+    c = (chunk & 15) << 2
+    base64 += encodings[a] + encodings[b] + encodings[c] + "="
   }
   return base64
 }

package/lib/index.d.ts

@@ -1,8 +1,9 @@
-import { Context, Schema } from "koishi";
+import { Computed, Context, Schema } from "koishi";
 export declare const name = "music";
 export declare const inject: string[];
 export interface Config {
     evalCommand: "glot" | "eval";
+    noise: Computed<boolean>;
 }
 export declare const Config: Schema<Config>;
 export interface Note {

package/lib/index.js

@@ -11,6 +11,9 @@ exports.Config = koishi_1.Schema.object({
     evalCommand: koishi_1.Schema.union(["glot", "eval"])
         .default("glot")
         .description("用于安全执行 js 代码的指令。"),
+    noise: koishi_1.Schema.computed(Boolean)
+        .default(true)
+        .description("是否添加白噪音来尝试规避 QQ 的语音编码杂音问题。"),
 });
 const gutterFunc = (f) => {
     function createCompatibilityBpmFunction(v) {
@@ -118,7 +121,10 @@ function apply(ctx, config) {
         }
         const page = await ctx.puppeteer.page();
         await page.goto((0, url_1.pathToFileURL)((0, path_1.resolve)(__dirname, "../browser/index.html")).href);
-        const base64 = (await page.evaluate(`synth(${data}).then(encodeWav).then(arrayBufferToBase64)`));
+        const opt = {
+            noise: session.resolve(config.noise),
+        };
+        const base64 = (await page.evaluate(`synth(${data}, ${JSON.stringify(opt)}).then(encodeWav).then(arrayBufferToBase64)`));
         page.close().catch(() => { });
         return koishi_1.h.audio("data:audio/wav;base64," + base64);
     });

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@dgck81lnn/koishi-plugin-music",
   "description": "Synthesize melodies in Koishi",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [