chunker ui redo

ui/chunker/src/api.ts

@@ -1,55 +1,13 @@
 /**
- * GraphQL API client for the chunker UI.
+ * Chunker-specific API functions.
+ * Shared functions (getAssets, scanMediaFolder) come from common.
  */
 
-import type { MediaAsset } from "./types";
+import { gql } from "../../common/api/graphql";
+import type { ChunkOutputFile } from "../../common/types/generated";
 
-const GRAPHQL_URL = "/api/graphql";
-
-async function gql<T>(query: string, variables?: Record<string, unknown>): Promise<T> {
-  const response = await fetch(GRAPHQL_URL, {
-    method: "POST",
-    headers: { "Content-Type": "application/json" },
-    body: JSON.stringify({ query, variables }),
-  });
-
-  const json = await response.json();
-
-  if (json.errors?.length) {
-    throw new Error(json.errors[0].message);
-  }
-
-  return json.data as T;
-}
-
-/** Fetch all media assets. */
-export async function getAssets(): Promise<MediaAsset[]> {
-  const data = await gql<{ assets: MediaAsset[] }>(`
-    query {
-      assets {
-        id filename file_path status error_message file_size duration
-        video_codec audio_codec width height framerate bitrate
-        properties comments tags created_at updated_at
-      }
-    }
-  `);
-  return data.assets;
-}
-
-/** Scan media/in/ folder for new files. */
-export async function scanMediaFolder(): Promise<{
-  found: number;
-  registered: number;
-  skipped: number;
-  files: string[];
-}> {
-  const data = await gql<{ scan_media_folder: { found: number; registered: number; skipped: number; files: string[] } }>(`
-    mutation {
-      scan_media_folder { found registered skipped files }
-    }
-  `);
-  return data.scan_media_folder;
-}
+// Re-export shared functions
+export { getAssets, scanMediaFolder } from "../../common/api/media";
 
 /** Create a chunk job via GraphQL mutation. */
 export async function createChunkJob(config: {
@@ -58,15 +16,70 @@ export async function createChunkJob(config: {
   num_workers: number;
   max_retries: number;
   processor_type: string;
-}): Promise<{ id: string }> {
-  const data = await gql<{ create_chunk_job: { id: string; status: string } }>(`
+  start_time?: number | null;
+  end_time?: number | null;
+}): Promise<{ id: string; celery_task_id: string | null }> {
+  const data = await gql<{
+    create_chunk_job: {
+      id: string;
+      status: string;
+      celery_task_id: string | null;
+    };
+  }>(
+    `
     mutation CreateChunkJob($input: CreateChunkJobInput!) {
       create_chunk_job(input: $input) {
         id
         status
+        celery_task_id
       }
     }
-  `, { input: config });
+  `,
+    { input: config },
+  );
 
   return data.create_chunk_job;
 }
+
+/** Cancel a running chunk job. */
+export async function cancelChunkJob(
+  celeryTaskId: string,
+): Promise<{ ok: boolean; message: string | null }> {
+  const data = await gql<{
+    cancel_chunk_job: { ok: boolean; message: string | null };
+  }>(
+    `
+    mutation CancelChunkJob($celery_task_id: String!) {
+      cancel_chunk_job(celery_task_id: $celery_task_id) {
+        ok
+        message
+      }
+    }
+  `,
+    { celery_task_id: celeryTaskId },
+  );
+
+  return data.cancel_chunk_job;
+}
+
+/** Fetch output chunk files for a completed job. */
+export async function getChunkOutputFiles(
+  jobId: string,
+): Promise<ChunkOutputFile[]> {
+  const data = await gql<{
+    chunk_output_files: ChunkOutputFile[];
+  }>(
+    `
+    query ChunkOutputFiles($job_id: String!) {
+      chunk_output_files(job_id: $job_id) {
+        key
+        size
+        url
+      }
+    }
+  `,
+    { job_id: jobId },
+  );
+
+  return data.chunk_output_files;
+}