video-workflow/cloud-functions/02-taskManager.js

/**
 * 云函数：taskManager（任务管理）
 * 替代：GET/POST/PUT/DELETE /api/tasks[/:id]
 * actions: list | get | create | update | delete
 */
async function handler(request, response) {
  try {
    await Psql.query(`
      CREATE TABLE IF NOT EXISTS "VideoflowTask" (
        "objectId"  VARCHAR(50) PRIMARY KEY,
        "bizId"     VARCHAR(255),
        "data"      JSONB NOT NULL DEFAULT '{}',
        "userId"    VARCHAR(255) DEFAULT '',
        "type"      VARCHAR(50)  DEFAULT '',
        "status"    VARCHAR(50)  DEFAULT '',
        "createdAt" TIMESTAMPTZ  DEFAULT NOW(),
        "updatedAt" TIMESTAMPTZ  DEFAULT NOW()
      )
    `);
    await Psql.query(`CREATE UNIQUE INDEX IF NOT EXISTS idx_videoflow_task_biz ON "VideoflowTask" ("bizId")`);
    await Psql.query(`CREATE INDEX IF NOT EXISTS idx_videoflow_task_user ON "VideoflowTask" ("userId")`);
    await Psql.query(`CREATE INDEX IF NOT EXISTS idx_videoflow_task_status ON "VideoflowTask" ("status")`);

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

    if (action === 'list') {
      const conds = [], params = [];
      if (userId) { params.push(userId); conds.push(`"userId" = $${params.length}`); }
      const status = pickParam(request, 'status');
      if (status) { params.push(status); conds.push(`"status" = $${params.length}`); }
      const type = pickParam(request, 'type');
      if (type) { params.push(type); conds.push(`"type" = $${params.length}`); }
      const limit = Math.min(parseInt(pickParam(request, 'limit') || '500'), 1000);
      const where = conds.length ? `WHERE ${conds.join(' AND ')}` : '';
      params.push(limit);
      const rows = await Psql.query(
        `SELECT * FROM "VideoflowTask" ${where} ORDER BY "createdAt" DESC LIMIT $${params.length}`,
        params
      );
      response.json({ code: 200, success: true, data: rows.map(rowToObj) });
      return;
    }

    if (action === 'get') {
      const id = pickParam(request, 'id', 'taskId', 'bizId');
      if (!id) return response.json({ code: 400, success: false, error: '缺少 id' });
      const rows = await Psql.query(`SELECT * FROM "VideoflowTask" WHERE "bizId" = $1 LIMIT 1`, [id]);
      if (!rows.length) return response.json({ code: 404, success: false, error: '未找到任务' });
      response.json({ code: 200, success: true, data: rowToObj(rows[0]) });
      return;
    }

    if (action === 'create') {
      const body = pickParam(request, 'task', 'data') || request.body || {};
      const bizId = body.id || body.bizId || generateId();
      const merged = { ...body, id: bizId, created_at: body.created_at || new Date().toISOString(), updated_at: new Date().toISOString() };
      const objectId = generateId();
      await Psql.query(
        `INSERT INTO "VideoflowTask" ("objectId","bizId","data","userId","type","status")
         VALUES ($1,$2,$3,$4,$5,$6)
         ON CONFLICT ("bizId") DO UPDATE SET "data"=$3, "type"=$5, "status"=$6, "updatedAt"=NOW()`,
        [objectId, bizId, JSON.stringify(merged), userId, merged.type || '', merged.status || '']
      );
      response.json({ code: 200, success: true, data: merged });
      return;
    }

    if (action === 'update') {
      const id = pickParam(request, 'id', 'taskId', 'bizId');
      if (!id) return response.json({ code: 400, success: false, error: '缺少 id' });
      const patch = pickParam(request, 'task', 'data', 'patch') || request.body || {};
      const cur = await Psql.query(`SELECT * FROM "VideoflowTask" WHERE "bizId" = $1 LIMIT 1`, [id]);
      const now = new Date().toISOString();
      const previous = cur.length ? rowToObj(cur[0]) : {};
      const merged = {
        ...previous,
        ...patch,
        id,
        bizId: patch.bizId || patch.id || id,
        created_at: previous.created_at || patch.created_at || now,
        updated_at: now,
      };
      if (cur.length) {
        await Psql.query(
          `UPDATE "VideoflowTask" SET "data"=$1, "type"=$2, "status"=$3, "updatedAt"=NOW() WHERE "bizId"=$4`,
          [JSON.stringify(merged), merged.type || '', merged.status || '', id]
        );
      } else {
        await Psql.query(
          `INSERT INTO "VideoflowTask" ("objectId","bizId","data","userId","type","status")
           VALUES ($1,$2,$3,$4,$5,$6)
           ON CONFLICT ("bizId") DO UPDATE SET "data"=$3, "type"=$5, "status"=$6, "updatedAt"=NOW()`,
          [generateId(), id, JSON.stringify(merged), userId || merged.userId || '', merged.type || '', merged.status || '']
        );
      }
      response.json({ code: 200, success: true, data: merged });
      return;
    }

    if (action === 'update') {
      const id = pickParam(request, 'id', 'taskId', 'bizId');
      if (!id) return response.json({ code: 400, success: false, error: '缺少 id' });
      const patch = pickParam(request, 'task', 'data', 'patch') || request.body || {};
      const cur = await Psql.query(`SELECT * FROM "VideoflowTask" WHERE "bizId" = $1 LIMIT 1`, [id]);
      if (!cur.length) return response.json({ code: 404, success: false, error: '未找到任务' });
      const merged = { ...rowToObj(cur[0]), ...patch, updated_at: new Date().toISOString() };
      await Psql.query(
        `UPDATE "VideoflowTask" SET "data"=$1, "type"=$2, "status"=$3, "updatedAt"=NOW() WHERE "bizId"=$4`,
        [JSON.stringify(merged), merged.type || '', merged.status || '', id]
      );
      response.json({ code: 200, success: true, data: merged });
      return;
    }

    if (action === 'delete') {
      const id = pickParam(request, 'id', 'taskId', 'bizId');
      if (!id) return response.json({ code: 400, success: false, error: '缺少 id' });
      const r = await Psql.query(`DELETE FROM "VideoflowTask" WHERE "bizId" = $1 RETURNING "bizId"`, [id]);
      response.json({ code: 200, success: true, data: { id, deleted: r.length > 0 } });
      return;
    }

    response.json({ code: 400, success: false, error: `未知 action: ${action}` });
  } catch (error) {
    console.error('❌ taskManager 失败:', 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, 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;
}