video-workflow/cloud-functions/06-voiceManager.js

/**
 * 云函数：voiceManager（音色档案 + speaker_id 池）
 * 替代：
 *   POST /api/voice/auto-speaker-id   → action=autoSpeakerId
 *   POST /api/voice-profiles/sync     → action=syncProfile
 * 新增：
 *   action=listProfiles  列出所有音色档案
 *   action=initPool      初始化 speaker_id 池（首次部署用）
 *
 * 注意：speaker_id 池原本在 docs/音色创建/speaker_id.md 文件里。
 * 迁移到云函数后改为存储在 "VoiceSpeakerPool" 表。
 * 首次部署后请用 action=initPool 把现有池一次性灌入。
 */
async function handler(request, response) {
  try {
    await Psql.query(`
      CREATE TABLE IF NOT EXISTS "VoiceProfile" (
        "objectId"   VARCHAR(50) PRIMARY KEY,
        "timbreId"   VARCHAR(255),
        "speakerId"  VARCHAR(255),
        "name"       VARCHAR(255) DEFAULT '',
        "data"       JSONB NOT NULL DEFAULT '{}',
        "userId"     VARCHAR(255) DEFAULT '',
        "occupied"   BOOLEAN DEFAULT TRUE,
        "createdAt"  TIMESTAMPTZ DEFAULT NOW(),
        "updatedAt"  TIMESTAMPTZ DEFAULT NOW()
      )
    `);
    await Psql.query(`CREATE UNIQUE INDEX IF NOT EXISTS idx_voice_timbre ON "VoiceProfile" ("timbreId") WHERE "timbreId" IS NOT NULL AND "timbreId" <> ''`);
    await Psql.query(`CREATE INDEX IF NOT EXISTS idx_voice_speaker ON "VoiceProfile" ("speakerId")`);

    await Psql.query(`
      CREATE TABLE IF NOT EXISTS "VoiceSpeakerPool" (
        "speakerId"  VARCHAR(255) PRIMARY KEY,
        "trained"    BOOLEAN DEFAULT FALSE,
        "createdAt"  TIMESTAMPTZ DEFAULT NOW()
      )
    `);

    const action = pickParam(request, 'action') || 'listProfiles';
    const userId = pickParam(request, 'userId') || '';

    if (action === 'autoSpeakerId') {
      const used = await Psql.query(`
        SELECT s."speakerId" FROM "VoiceSpeakerPool" s
        LEFT JOIN "VoiceProfile" v ON v."speakerId" = s."speakerId"
        WHERE v."speakerId" IS NULL AND s."trained" = FALSE
        ORDER BY s."createdAt" ASC
        LIMIT 1
      `);
      if (!used.length) {
        return response.json({ code: 500, success: false, error: 'speaker_id 池已耗尽，请先 initPool 添加' });
      }
      response.json({ code: 200, success: true, data: { speakerId: used[0].speakerId } });
      return;
    }

    if (action === 'syncProfile') {
      const body = pickParam(request, 'profile', 'data') || request.body || {};
      const timbreId = body.timbre_id || body.timbreId || '';
      const speakerId = body.speaker_id || body.speakerId || '';
      if (!timbreId && !speakerId) {
        return response.json({ code: 400, success: false, error: '缺少 timbre_id 或 speaker_id' });
      }
      const now = new Date().toISOString();
      const merged = { ...body, timbre_id: timbreId, speaker_id: speakerId, updated_at: now };

      let existing = [];
      if (timbreId) {
        existing = await Psql.query(`SELECT * FROM "VoiceProfile" WHERE "timbreId" = $1 LIMIT 1`, [timbreId]);
      }
      if (!existing.length && speakerId) {
        existing = await Psql.query(`SELECT * FROM "VoiceProfile" WHERE "speakerId" = $1 LIMIT 1`, [speakerId]);
      }

      if (existing.length) {
        const old = typeof existing[0].data === 'string' ? JSON.parse(existing[0].data) : (existing[0].data || {});
        const merged2 = { ...old, ...merged };
        await Psql.query(
          `UPDATE "VoiceProfile" SET "timbreId"=$1, "speakerId"=$2, "name"=$3, "data"=$4, "occupied"=$5, "updatedAt"=NOW() WHERE "objectId"=$6`,
          [timbreId, speakerId, body.name || merged2.name || '', JSON.stringify(merged2), body.occupied !== false, existing[0].objectId]
        );
        response.json({ code: 200, success: true, data: { ...merged2, objectId: existing[0].objectId } });
      } else {
        merged.created_at = now;
        const objectId = generateId();
        await Psql.query(
          `INSERT INTO "VoiceProfile" ("objectId","timbreId","speakerId","name","data","userId","occupied")
           VALUES ($1,$2,$3,$4,$5,$6,$7)`,
          [objectId, timbreId, speakerId, body.name || '', JSON.stringify(merged), userId, body.occupied !== false]
        );
        response.json({ code: 200, success: true, data: { ...merged, objectId } });
      }
      return;
    }

    if (action === 'listProfiles') {
      const conds = [], params = [];
      if (userId) { params.push(userId); conds.push(`"userId" = $${params.length}`); }
      const where = conds.length ? `WHERE ${conds.join(' AND ')}` : '';
      const rows = await Psql.query(
        `SELECT * FROM "VoiceProfile" ${where} ORDER BY "createdAt" DESC LIMIT 1000`,
        params
      );
      response.json({ code: 200, success: true, data: rows.map(rowToObj) });
      return;
    }

    if (action === 'initPool') {
      const ids = pickParam(request, 'speakerIds') || [];
      const list = Array.isArray(ids) ? ids : String(ids || '').split(/[\s,]+/).filter(Boolean);
      if (!list.length) {
        return response.json({ code: 400, success: false, error: '缺少 speakerIds（数组或逗号分隔字符串）' });
      }
      let inserted = 0;
      for (const sid of list) {
        const r = await Psql.query(
          `INSERT INTO "VoiceSpeakerPool" ("speakerId") VALUES ($1) ON CONFLICT DO NOTHING RETURNING "speakerId"`,
          [String(sid).trim()]
        );
        if (r.length) inserted++;
      }
      response.json({ code: 200, success: true, data: { total: list.length, inserted } });
      return;
    }

    if (action === 'deleteProfile') {
      const objectId = pickParam(request, 'objectId');
      const timbreId = pickParam(request, 'timbreId');
      if (!objectId && !timbreId) {
        return response.json({ code: 400, success: false, error: '缺少 objectId 或 timbreId' });
      }
      const r = objectId
        ? await Psql.query(`DELETE FROM "VoiceProfile" WHERE "objectId" = $1 RETURNING "objectId"`, [objectId])
        : await Psql.query(`DELETE FROM "VoiceProfile" WHERE "timbreId" = $1 RETURNING "objectId"`, [timbreId]);
      response.json({ code: 200, success: true, data: { deleted: r.length > 0 } });
      return;
    }

    response.json({ code: 400, success: false, error: `未知 action: ${action}` });
  } catch (error) {
    console.error('❌ voiceManager 失败:', error.message);
    response.json({ code: 500, success: false, error: error.message });
  }
}

function rowToObj(row) {
  const data = typeof row.data === 'string' ? JSON.parse(row.data) : (row.data || {});
  return { ...data, objectId: row.objectId, occupied: row.occupied, createdAt: row.createdAt, updatedAt: row.updatedAt };
}

function pickParam(request, ...names) {
  const sources = [request.params, request.body, request];
  for (const src of sources) {
    if (!src || typeof src !== 'object') continue;
    for (const n of names) {
      const v = src[n];
      if (v !== undefined && v !== null && v !== '') return v;
    }
  }
  return null;
}

function generateId() {
  const chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
  let s = '';
  for (let i = 0; i < 10; i++) s += chars.charAt(Math.floor(Math.random() * chars.length));
  return s;
}