massive changes and it works

internal/api/metadata.go

@@ -7,7 +7,7 @@ import (
 	"log"
 	"net/http"
 
-	"fuego/pkg/types"
+	"jiggablend/pkg/types"
 )
 
 // handleSubmitMetadata handles metadata submission from runner
@@ -19,7 +19,7 @@ func (s *Server) handleSubmitMetadata(w http.ResponseWriter, r *http.Request) {
 	}
 
 	// Get runner ID from context (set by runnerAuthMiddleware)
-	runnerID, ok := r.Context().Value("runner_id").(int64)
+	runnerID, ok := r.Context().Value(runnerIDContextKey).(int64)
 	if !ok {
 		s.respondError(w, http.StatusUnauthorized, "runner_id not found in context")
 		return
@@ -44,16 +44,32 @@ func (s *Server) handleSubmitMetadata(w http.ResponseWriter, r *http.Request) {
 	}
 
 	// Find the metadata extraction task for this job
+	// First try to find task assigned to this runner, then fall back to any metadata task for this job
 	var taskID int64
 	err = s.db.QueryRow(
 		`SELECT id FROM tasks WHERE job_id = ? AND task_type = ? AND runner_id = ?`,
 		jobID, types.TaskTypeMetadata, runnerID,
 	).Scan(&taskID)
 	if err == sql.ErrNoRows {
-		s.respondError(w, http.StatusNotFound, "Metadata extraction task not found")
-		return
-	}
-	if err != nil {
+		// Fall back to any metadata task for this job (in case assignment changed)
+		err = s.db.QueryRow(
+			`SELECT id FROM tasks WHERE job_id = ? AND task_type = ? ORDER BY created_at DESC LIMIT 1`,
+			jobID, types.TaskTypeMetadata,
+		).Scan(&taskID)
+		if err == sql.ErrNoRows {
+			s.respondError(w, http.StatusNotFound, "Metadata extraction task not found")
+			return
+		}
+		if err != nil {
+			s.respondError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to find task: %v", err))
+			return
+		}
+		// Update the task to be assigned to this runner if it wasn't already
+		s.db.Exec(
+			`UPDATE tasks SET runner_id = ? WHERE id = ? AND runner_id IS NULL`,
+			runnerID, taskID,
+		)
+	} else if err != nil {
 		s.respondError(w, http.StatusInternalServerError, fmt.Sprintf("Failed to find task: %v", err))
 		return
 	}
@@ -82,6 +98,9 @@ func (s *Server) handleSubmitMetadata(w http.ResponseWriter, r *http.Request) {
 	)
 	if err != nil {
 		log.Printf("Failed to mark metadata task as completed: %v", err)
+	} else {
+		// Update job status and progress after metadata task completes
+		s.updateJobStatusFromTasks(jobID)
 	}
 
 	log.Printf("Metadata extracted for job %d: frame_start=%d, frame_end=%d", jobID, metadata.FrameStart, metadata.FrameEnd)