feat: 完善视频任务与 KOC 资源库

lib/task-service.ts

@@ -10,6 +10,8 @@ export type CreateDistributionTaskInput = {
   name: string;
   brand: string;
   dueAt: string;
+  platform?: "小红书" | "抖音";
+  contentFormat?: "image_text" | "video";
 };
 
 export type CreateScreenshotTaskInput = {
@@ -33,6 +35,8 @@ export type DistributionTaskCreation = {
   sheetId: string;
   sheetName: string;
   sourceUrl: string;
+  platform: "小红书" | "抖音";
+  contentFormat: "image_text" | "video";
 };
 
 export type ScreenshotTaskCreation = {
@@ -55,6 +59,8 @@ type TaskRow = {
   source_url: string;
   source_sheet_id: string;
   source_sheet_name: string;
+  platform: "小红书" | "抖音";
+  content_format: "image_text" | "video";
 };
 
 function normalizedValue(value: string) {
@@ -99,17 +105,27 @@ async function findExistingTask(
   return db
     .prepare(
       `SELECT id, share_token, name, brand, due_at, quantity,
-              source_url, source_sheet_id, source_sheet_name
+              source_url, source_sheet_id, source_sheet_name,
+              platform, content_format
        FROM tasks
        WHERE name = ?
          AND brand = ?
          AND due_at = ?
          AND source_url = ?
+         AND platform = ?
+         AND content_format = ?
          AND status IN ('active', 'importing')
        ORDER BY created_at DESC
        LIMIT 1`,
     )
-    .bind(input.name, input.brand, input.dueAt, sourceUrl)
+    .bind(
+      input.name,
+      input.brand,
+      input.dueAt,
+      sourceUrl,
+      input.platform ?? "小红书",
+      input.contentFormat ?? "image_text",
+    )
     .first<TaskRow>();
 }
 
@@ -124,9 +140,10 @@ async function insertTaskFromSource(
     .prepare(
       `INSERT INTO tasks
         (id, name, brand, quantity, claimed_quantity, due_at, status,
+         platform, content_format,
          source_url, source_sheet_id, source_sheet_name, source_synced_at,
          share_token)
-       VALUES (?, ?, ?, ?, 0, ?, 'importing', ?, ?, ?, ?, ?)`,
+       VALUES (?, ?, ?, ?, 0, ?, 'importing', ?, ?, ?, ?, ?, ?, ?)`,
     )
     .bind(
       taskId,
@@ -134,6 +151,8 @@ async function insertTaskFromSource(
       input.brand,
       source.rows.length,
       input.dueAt,
+      input.platform ?? "小红书",
+      input.contentFormat ?? "image_text",
       source.url,
       source.sheetId,
       source.sheetName,
@@ -149,11 +168,15 @@ async function insertTaskFromSource(
         ...image,
         key: `content-assets/${taskId}/${contentId}/${image.index}`,
       }));
+      const videoAssets = row.videos.map((video) => ({
+        ...video,
+        key: `content-videos/${taskId}/${contentId}/${video.index}`,
+      }));
       return db
         .prepare(
           `INSERT INTO contents
-            (id, task_id, title, body, image_assets, status, source, source_row)
-           VALUES (?, ?, ?, ?, ?, 'available', ?, ?)`,
+            (id, task_id, title, body, image_assets, video_assets, status, source, source_row)
+           VALUES (?, ?, ?, ?, ?, ?, 'available', ?, ?)`,
         )
         .bind(
           contentId,
@@ -161,6 +184,7 @@ async function insertTaskFromSource(
           row.title,
           row.body,
           JSON.stringify(imageAssets),
+          JSON.stringify(videoAssets),
           `飞书 · ${source.sheetName}`,
           row.sourceRow,
         );
@@ -189,11 +213,13 @@ export async function createDistributionTask(
   options: { deduplicate?: boolean } = {},
 ): Promise<DistributionTaskCreation> {
   await ensureSchema();
-  const input = {
+  const input: Required<CreateDistributionTaskInput> = {
     feishuUrl: normalizedFeishuUrl(rawInput.feishuUrl),
     name: normalizedValue(rawInput.name),
     brand: normalizedValue(rawInput.brand),
     dueAt: normalizedDueDate(rawInput.dueAt),
+    platform: rawInput.platform === "抖音" ? "抖音" : "小红书",
+    contentFormat: rawInput.contentFormat === "video" ? "video" : "image_text",
   };
   if (!input.feishuUrl || !input.name || !input.brand) {
     throw new Error("请补全飞书链接、任务名称和品牌/项目");
@@ -213,11 +239,19 @@ export async function createDistributionTask(
         sheetId: existing.source_sheet_id,
         sheetName: existing.source_sheet_name,
         sourceUrl: existing.source_url,
+        platform: existing.platform,
+        contentFormat: existing.content_format,
       };
     }
   }
 
   const source = await readFeishuSource(input.feishuUrl, bindings);
+  if (
+    input.contentFormat === "video" &&
+    source.rows.some((row) => row.videos.length === 0)
+  ) {
+    throw new Error("视频任务中存在未识别到视频的内容行，请检查飞书“视频”列");
+  }
   const inserted = await insertTaskFromSource(source, input);
   return {
     created: true,
@@ -230,6 +264,8 @@ export async function createDistributionTask(
     sheetId: source.sheetId,
     sheetName: source.sheetName,
     sourceUrl: source.url,
+    platform: input.platform,
+    contentFormat: input.contentFormat,
   };
 }
 
@@ -287,8 +323,8 @@ export async function createScreenshotTask(
       db
         .prepare(
           `INSERT INTO contents
-            (id, task_id, title, body, image_assets, status, source, source_row)
-           VALUES (?, ?, ?, ?, ?, 'available', '截图回收任务', ?)`,
+            (id, task_id, title, body, image_assets, video_assets, status, source, source_row)
+           VALUES (?, ?, ?, ?, ?, '[]', 'available', '截图回收任务', ?)`,
         )
         .bind(
           uid("content"),