"""Endpoint for splitting with a tracklist file upload.""" import json from pathlib import Path from fastapi import APIRouter, UploadFile, File, Form, HTTPException, BackgroundTasks from backend.config import settings from backend.services.task_manager import task_manager from backend.services.splitter import run_split_task from backend.models.request import TracklistEntry from backend.models.response import SplitResponse, TaskStatus router = APIRouter(prefix="/api", tags=["split"]) # Import CLI tracklist parsing functions from audio_splitter.tracklist import parse_format, read_tracklist @router.post("/split-file", response_model=SplitResponse) async def split_from_file( background_tasks: BackgroundTasks, task_id: str = Form(...), tracklist_file: UploadFile = File(...), options: str = Form("{}"), tracklist_format: str = Form("%ts %tn - %an"), ): """ Start a splitting task using a tracklist file. This endpoint accepts a plain text tracklist file (like the CLI does) and parses it using the same logic. Args: task_id: Task ID from upload. tracklist_file: Tracklist file (text/plain). options: JSON string of all CLI options. tracklist_format: Format string for parsing (default: "%ts %tn - %an"). Returns: SplitResponse with task_id and status. """ # Validate task exists if not task_manager.has_task(task_id): raise HTTPException(status_code=404, detail=f"Task {task_id} not found") # Check if task is already processing status = task_manager.get_status(task_id) if status and status["status"] in (TaskStatus.PROCESSING, TaskStatus.DONE): raise HTTPException( status_code=409, detail=f"Task {task_id} is already {status['status']}" ) # Validate file type if not tracklist_file.filename.endswith(('.txt', '.text')): raise HTTPException( status_code=400, detail="Tracklist file must be a text file (.txt or .text)" ) # Read and parse the tracklist file try: content = await tracklist_file.read() text = content.decode('utf-8') except UnicodeDecodeError: raise HTTPException( status_code=400, detail="Tracklist file must be UTF-8 encoded" ) if not text.strip(): raise HTTPException( status_code=400, detail="Tracklist file is empty" ) # Parse the tracklist using CLI logic try: # Parse the format string into tokens tokens = parse_format(tracklist_format) # Write the content to a temporary file (read_tracklist expects a file path) temp_tracklist_path = settings.temp_dir / task_id / "tracklist.txt" temp_tracklist_path.parent.mkdir(parents=True, exist_ok=True) temp_tracklist_path.write_text(text, encoding='utf-8') # Parse the tracklist file using CLI logic tracklist_dicts = read_tracklist(str(temp_tracklist_path), tokens) except Exception as e: raise HTTPException( status_code=400, detail=f"Failed to parse tracklist: {str(e)}" ) if not tracklist_dicts: raise HTTPException( status_code=400, detail="No valid tracks found in tracklist file" ) # Convert dicts to TracklistEntry objects (same as JSON endpoint) try: tracklist_entries = [TracklistEntry(**entry) for entry in tracklist_dicts] except Exception as e: raise HTTPException( status_code=400, detail=f"Invalid tracklist data: {str(e)}" ) # Parse options JSON try: options_dict = json.loads(options) except json.JSONDecodeError: raise HTTPException( status_code=400, detail="Invalid JSON in options field" ) # Update task status task_manager.update_task( task_id, status=TaskStatus.PROCESSING, progress=0, message="Preparing to split..." ) # Start background task background_tasks.add_task( run_split_task, task_id, tracklist_entries, # now TracklistEntry objects options_dict ) return SplitResponse( task_id=task_id, status=TaskStatus.PROCESSING )