chunker ui redo

core/api/graphql.py

@@ -15,7 +15,9 @@ from strawberry.schema.config import StrawberryConfig
 from strawberry.types import Info
 
 from core.api.schema.graphql import (
+    CancelResultType,
     ChunkJobType,
+    ChunkOutputFileType,
     CreateChunkJobInput,
     CreateJobInput,
     DeleteResultType,
@@ -26,7 +28,7 @@ from core.api.schema.graphql import (
     TranscodePresetType,
     UpdateAssetInput,
 )
-from core.storage import BUCKET_IN, list_objects
+from core.storage import BUCKET_IN, list_objects, upload_file
 
 VIDEO_EXTS = {".mp4", ".mkv", ".avi", ".mov", ".webm", ".flv", ".wmv", ".m4v"}
 AUDIO_EXTS = {".mp3", ".wav", ".flac", ".aac", ".ogg", ".m4a"}
@@ -90,6 +92,25 @@ class Query:
     def system_status(self, info: Info) -> SystemStatusType:
         return SystemStatusType(status="ok", version="0.1.0")
 
+    @strawberry.field
+    def chunk_output_files(self, info: Info, job_id: str) -> List[ChunkOutputFileType]:
+        """List output chunk files for a completed job from media/out/."""
+        from pathlib import Path
+
+        media_out = os.environ.get("MEDIA_OUT_DIR", "/app/media/out")
+        output_dir = Path(media_out) / "chunks" / job_id
+        if not output_dir.is_dir():
+            return []
+        return [
+            ChunkOutputFileType(
+                key=f.name,
+                size=f.stat().st_size,
+                url=f"/media/out/chunks/{job_id}/{f.name}",
+            )
+            for f in sorted(output_dir.iterdir())
+            if f.is_file()
+        ]
+
 
 # ---------------------------------------------------------------------------
 # Mutations
@@ -100,8 +121,26 @@ class Query:
 class Mutation:
     @strawberry.mutation
     def scan_media_folder(self, info: Info) -> ScanResultType:
+        import logging
+        from pathlib import Path
+
         from core.db import create_asset, get_asset_filenames
 
+        logger = logging.getLogger(__name__)
+
+        # Sync local media/in/ files to MinIO (handles fresh installs / pruned volumes)
+        local_media = Path("/app/media/in")
+        if local_media.is_dir():
+            existing_keys = {o["key"] for o in list_objects(BUCKET_IN)}
+            for f in local_media.iterdir():
+                if f.is_file() and f.suffix.lower() in MEDIA_EXTS:
+                    if f.name not in existing_keys:
+                        try:
+                            upload_file(str(f), BUCKET_IN, f.name)
+                            logger.info("Uploaded %s to MinIO", f.name)
+                        except Exception as e:
+                            logger.warning("Failed to upload %s: %s", f.name, e)
+
         objects = list_objects(BUCKET_IN, extensions=MEDIA_EXTS)
         existing = get_asset_filenames()
 
@@ -284,6 +323,8 @@ class Mutation:
             "num_workers": input.num_workers,
             "max_retries": input.max_retries,
             "processor_type": input.processor_type,
+            "start_time": input.start_time,
+            "end_time": input.end_time,
         }
 
         executor_mode = os.environ.get("MPR_EXECUTOR", "local")
@@ -320,6 +361,17 @@ class Mutation:
             celery_task_id=celery_task_id,
         )
 
+    @strawberry.mutation
+    def cancel_chunk_job(self, info: Info, celery_task_id: str) -> CancelResultType:
+        """Cancel a running chunk job by revoking its Celery task."""
+        try:
+            from admin.mpr.celery import app as celery_app
+
+            celery_app.control.revoke(celery_task_id, terminate=True, signal="SIGTERM")
+            return CancelResultType(ok=True, message="Task revoked")
+        except Exception as e:
+            return CancelResultType(ok=False, message=str(e))
+
 
 # ---------------------------------------------------------------------------
 # Schema