@goodandready/dsh-tts 0.3.22 → 0.4.0

package/lib/engines/types.js

@@ -1,46 +1,59 @@
-import { execSync } from 'node:child_process'
+import { execFile } from 'node:child_process'
+import { promisify } from 'node:util'
 
-export function detectGPU() {
+const execFileAsync = promisify(execFile)
+
+let gpuCache
+
+export async function detectGPU() {
+  if (typeof gpuCache === 'boolean') return gpuCache
   try {
-    const out = execSync('nvidia-smi --query-gpu=name --format=csv,noheader', {
-      stdio: ['ignore', 'pipe', 'ignore'],
-      timeout: 2000,
-      encoding: 'utf8',
-    })
-    return !!(out && out.trim())
+    const { stdout } = await execFileAsync(
+      'nvidia-smi',
+      ['--query-gpu=name', '--format=csv,noheader'],
+      { timeout: 2000 },
+    )
+    gpuCache = !!(stdout && stdout.trim())
   } catch {
-    return false
+    gpuCache = false
   }
+  return gpuCache
 }
 
 export class EngineRegistry {
-  constructor({ gpu, installed, manager } = {}) {
-    this.gpu = typeof gpu === 'boolean' ? gpu : detectGPU()
+  constructor({ gpu = false, installed = [], manager = null } = {}) {
+    this.gpu = !!gpu
     this.installed = Array.isArray(installed) ? installed : []
     this.manager = manager || null
   }
 
+  static async create({ installed, manager } = {}) {
+    const gpu = await detectGPU()
+    return new EngineRegistry({ gpu, installed, manager })
+  }
+
   isInstalled(engineId) {
     return this.installed.includes(engineId)
   }
 
   list() {
     const list = []
-    // Kokoro works on CPU everywhere
+    // Kokoro weights may be present, but inference is not bundled.
     if (this.installed.includes('kokoro')) {
       list.push({
         id: 'kokoro',
         name: 'Kokoro-82M',
         type: 'local-cpu',
+        runtimeAvailable: false,
         supports: (lang) => !lang || ['en', 'ru', 'ja', 'zh'].includes(String(lang).slice(0, 2).toLowerCase()),
       })
     }
-    // F5-TTS requires GPU
     if (this.gpu && this.installed.includes('f5')) {
       list.push({
         id: 'f5',
         name: 'F5-TTS',
         type: 'local-gpu',
+        runtimeAvailable: false,
         supports: () => true,
       })
     }

package/lib/http-util.js

@@ -0,0 +1,27 @@
+// Shared HTTP helpers for dsh-tts host routes.
+
+export function writeJson(res, code, body) {
+  try {
+    res.writeHead(code, { 'Content-Type': 'application/json', 'Cache-Control': 'no-store' })
+    res.end(JSON.stringify(body))
+  } catch { /* socket closed */ }
+}
+
+export function readBody(req, maxBytes) {
+  return new Promise((resolve, reject) => {
+    const chunks = []
+    let size = 0
+    req.on('data', (c) => {
+      size += c.length
+      if (size > maxBytes) { reject(new Error('body too large')); req.destroy(); return }
+      chunks.push(c)
+    })
+    req.on('end', () => resolve(Buffer.concat(chunks)))
+    req.on('error', reject)
+  })
+}
+
+/** Reject cross-site POSTs. Do not require loopback or Origin==Host: the Web UI is used over LAN and reverse proxies. */
+export function isTrustedSettingsRequest(request) {
+  return request.headers['sec-fetch-site'] !== 'cross-site'
+}