@lumiastream/wakeword 1.0.1-alpha.6 → 1.0.1-alpha.8

package/bin/wakeword

@@ -24,7 +24,7 @@ const child = spawn(
 			path.dirname(fileURLToPath(import.meta.url)),
 			"..",
 			"lib",
-			"voice.mjs"
+			"voice.js"
 		),
 		soxPath,
 		...process.argv.slice(2),

package/lib/record.js

@@ -0,0 +1,120 @@
+"use strict";
+
+import assert from "assert";
+import debug from "debug";
+import { spawn } from "child_process";
+import recorders from "./recorders/index.js";
+
+class Recording {
+	constructor(options = {}) {
+		const defaults = {
+			sampleRate: 16000,
+			channels: 1,
+			compress: false,
+			threshold: 0.5,
+			thresholdStart: null,
+			thresholdEnd: null,
+			silence: "1.0",
+			recorder: "sox",
+			endOnSilence: false,
+			audioType: "wav",
+			binPath: null,
+			bufferSize: null,
+			arguments: [],
+		};
+
+		this.options = Object.assign(defaults, options);
+
+		const recorder = recorders.load(this.options.recorder);
+		console.log(recorder);
+		const { cmd, args, spawnOptions = {} } = recorder(this.options);
+
+		this.cmd = cmd;
+		this.args = args;
+		this.cmdOptions = Object.assign(
+			{ encoding: "binary", stdio: "pipe" },
+			spawnOptions
+		);
+
+		debug(`Started recording`);
+		debug(this.options);
+
+		const command = ` ${this.cmd} ${this.args.join(" ")}`;
+		debug(command);
+
+		return this.start();
+	}
+
+	start() {
+		const { cmd, args, cmdOptions } = this;
+
+		const cp = spawn(cmd, args, cmdOptions);
+		const rec = cp.stdout;
+		const err = cp.stderr;
+
+		this.process = cp; // expose child process
+		this._stream = rec; // expose output stream
+
+		cp.on("close", (code) => {
+			if (code === 0) return;
+			rec.emit(
+				"error",
+				`${this.cmd} has exited with error code ${code}.
+
+Enable debugging with the environment variable DEBUG=record.`
+			);
+		});
+
+		err.on("data", (chunk) => {
+			debug(`STDERR: ${chunk}`);
+		});
+
+		rec.on("data", (chunk) => {
+			debug(`Recording ${chunk.length} bytes`);
+		});
+
+		rec.on("end", () => {
+			debug("Recording ended");
+		});
+
+		return this;
+	}
+
+	stop() {
+		assert(this.process, "Recording not yet started");
+
+		this.process.kill();
+	}
+
+	pause() {
+		assert(this.process, "Recording not yet started");
+
+		this.process.kill("SIGSTOP");
+		this._stream.pause();
+		debug("Paused recording");
+	}
+
+	resume() {
+		assert(this.process, "Recording not yet started");
+
+		this.process.kill("SIGCONT");
+		this._stream.resume();
+		debug("Resumed recording");
+	}
+
+	isPaused() {
+		assert(this.process, "Recording not yet started");
+
+		return this._stream.isPaused();
+	}
+
+	stream() {
+		assert(this._stream, "Recording not yet started");
+
+		return this._stream;
+	}
+}
+
+export default {
+	record: (...args) => new Recording(...args),
+};

package/lib/recorders/arecord.js

@@ -0,0 +1,27 @@
+// On some systems (RasPi), arecord is the prefered recording binary
+export default (options) => {
+	let cmd = "arecord";
+
+	if (options.binPath) {
+		cmd = options.binPath;
+	}
+
+	const args = [
+		"-q", // show no progress
+		"-r",
+		options.sampleRate, // sample rate
+		"-c",
+		options.channels, // channels
+		"-t",
+		options.audioType, // audio type
+		"-f",
+		"S16_LE", // Sample format
+		"-", // pipe
+	];
+
+	if (options.device) {
+		args.unshift("-D", options.device);
+	}
+
+	return { cmd, args };
+};

package/lib/recorders/index.js

@@ -0,0 +1,23 @@
+// import { fileURLToPath } from "node:url";
+// import path from "node:path";
+// const __dirname = path.dirname(fileURLToPath(import.meta.url));
+import rec from "./sox.js";
+
+function load(recorderName) {
+	try {
+		// const recoderPath = path.resolve(__dirname, recorderName);
+		// const module = await import(recoderPath);
+		// return module.default;
+		return rec;
+	} catch (err) {
+		if (err.code === "MODULE_NOT_FOUND") {
+			throw new Error(`No such recorder found: ${recorderName}`);
+		}
+
+		throw err;
+	}
+}
+
+export default {
+	load,
+};

package/lib/recorders/rec.js

@@ -0,0 +1,36 @@
+export default (options) => {
+	let cmd = "rec";
+
+	if (options.binPath) {
+		cmd = options.binPath;
+	}
+
+	let args = [
+		"-q", // show no progress
+		"-r",
+		options.sampleRate, // sample rate
+		"-c",
+		options.channels, // channels
+		"-e",
+		"signed-integer", // sample encoding
+		"-b",
+		"16", // precision (bits)
+		"-t",
+		options.audioType, // audio type
+		"-", // pipe
+	];
+
+	if (options.endOnSilence) {
+		args = args.concat([
+			"silence",
+			"1",
+			"0.1",
+			options.thresholdStart || options.threshold + "%",
+			"1",
+			options.silence,
+			options.thresholdEnd || options.threshold + "%",
+		]);
+	}
+
+	return { cmd, args };
+};

package/lib/recorders/sox.js

@@ -0,0 +1,51 @@
+export default (options) => {
+	let cmd = "sox";
+
+	if (options.binPath) {
+		cmd = options.binPath;
+	}
+
+	let args = [
+		"--default-device",
+		"--no-show-progress", // show no progress
+		"--rate",
+		options.sampleRate, // sample rate
+		"--channels",
+		options.channels, // channels
+		"--encoding",
+		"signed-integer", // sample encoding
+		"--bits",
+		"16", // precision (bits)
+		"--type",
+		options.audioType, // audio type
+		"-", // pipe
+	];
+
+	if (options.bufferSize) {
+		args.push("--buffer", options.bufferSize);
+	}
+
+	if (options.endOnSilence) {
+		args = args.concat([
+			"silence",
+			"1",
+			"0.1",
+			options.thresholdStart || options.threshold + "%",
+			"1",
+			options.silence,
+			options.thresholdEnd || options.threshold + "%",
+		]);
+	}
+
+	if (options.arguments) {
+		args = args.concat(options.arguments);
+	}
+
+	const spawnOptions = {};
+
+	if (options.device) {
+		spawnOptions.env = { ...process.env, AUDIODEV: options.device };
+	}
+
+	return { cmd, args, spawnOptions };
+};

package/lib/{voice.mjs → voice.js}

@@ -1,5 +1,5 @@
-import { Model, Recognizer, setLogLevel } from "vosk";
-import record from "@lumiastream/record";
+import { Model, Recognizer, setLogLevel } from "vosk-koffi";
+import record from "./record.js";
 import { join } from "node:path";
 
 const binPath = join(
@@ -14,6 +14,7 @@ const binPath = join(
 console.log(binPath);
 
 let COMMANDS = [
+	"blue",
 	"[unk]", // always keep an [unk] fallback!
 ];
 

package/package.json

@@ -1,8 +1,8 @@
 {
 	"name": "@lumiastream/wakeword",
-	"version": "1.0.1-alpha.6",
+	"version": "1.0.1-alpha.8",
 	"type": "module",
-	"main": "lib/voice.mjs",
+	"main": "lib/voice.js",
 	"bin": {
 		"wakeword": "bin/wakeword"
 	},
@@ -16,7 +16,6 @@
 		"postinstall": "chmod +x binaries/soxmac binaries/soxlinux binaries/sox.exe || true"
 	},
 	"dependencies": {
-		"@lumiastream/record": "^1.0.1",
-		"vosk": "^0.3.39"
+		"vosk-koffi": "^1.1.1"
 	}
 }