flatrender/services/render/internal/handlers/internal.go

package handlers

import (
	"context"
	"fmt"
	"net/http"
	"time"

	"github.com/flatrender/render-svc/internal/db"
	"github.com/flatrender/render-svc/internal/models"
	"github.com/flatrender/render-svc/internal/notifier"
	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
	"github.com/minio/minio-go/v7"
)

type InternalHandler struct {
	store           *db.Store
	notifier        *notifier.Client // may be nil — notifications are best-effort
	minio           *minio.Client
	templatesBucket string // bucket that holds .aep project files
	exportsBucket   string // bucket that receives rendered MP4 outputs
}

func NewInternalHandler(store *db.Store, n *notifier.Client, mc *minio.Client, templatesBucket, exportsBucket string) *InternalHandler {
	return &InternalHandler{
		store:           store,
		notifier:        n,
		minio:           mc,
		templatesBucket: templatesBucket,
		exportsBucket:   exportsBucket,
	}
}

// completeRequest is the body for POST .../complete
type completeRequest struct {
	ExportID *uuid.UUID `json:"export_id"`
}

// failRequest is the body for POST .../fail
type failRequest struct {
	Reason string `json:"reason" binding:"required"`
	AtStep string `json:"at_step"` // optional: which render step failed
}

// POST /v1/internal/render/jobs/:job_id/complete
func (h *InternalHandler) Complete(c *gin.Context) {
	jobID, err := uuid.Parse(c.Param("job_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid job_id"})
		return
	}
	var req completeRequest
	_ = c.ShouldBindJSON(&req) // export_id is optional

	job, err := h.store.CompleteJob(c.Request.Context(), jobID, req.ExportID)
	if err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}

	// Fire notification if the user requested it (tell_me_when_done)
	if h.notifier != nil && job.TellMeWhenDone {
		jobName := ""
		if job.Name != nil {
			jobName = *job.Name
		} else if job.Title != nil {
			jobName = *job.Title
		}
		h.notifier.NotifyRenderDone(c.Request.Context(),
			job.UserID, job.TenantID, job.ID, job.ExportID, jobName)
	}

	c.JSON(http.StatusOK, gin.H{"status": "done", "job_id": job.ID})
}

// POST /v1/internal/render/jobs/:job_id/fail
func (h *InternalHandler) Fail(c *gin.Context) {
	jobID, err := uuid.Parse(c.Param("job_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid job_id"})
		return
	}
	var req failRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: err.Error()})
		return
	}
	atStep := req.AtStep
	if atStep == "" {
		atStep = "Rendering"
	}

	job, err := h.store.FailJob(c.Request.Context(), jobID, req.Reason, atStep)
	if err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}

	// Notify user of failure
	if h.notifier != nil {
		jobName := ""
		if job.Name != nil {
			jobName = *job.Name
		} else if job.Title != nil {
			jobName = *job.Title
		}
		h.notifier.NotifyRenderFailed(c.Request.Context(),
			job.UserID, job.TenantID, job.ID, jobName, req.Reason)
	}

	c.JSON(http.StatusOK, gin.H{"status": "failed", "job_id": job.ID})
}

// POST /v1/internal/nodes/:node_id/heartbeat
func (h *InternalHandler) Heartbeat(c *gin.Context) {
	nodeID, err := uuid.Parse(c.Param("node_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid node_id"})
		return
	}
	var req models.NodeHeartbeatRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: err.Error()})
		return
	}
	req.NodeID = nodeID
	if err := h.store.UpdateNodeHeartbeat(c.Request.Context(), nodeID, &req); err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}
	c.JSON(http.StatusOK, gin.H{
		"next_heartbeat_in_sec": 5,
		"pending_commands":      []any{},
	})
}

// POST /v1/internal/nodes/:node_id/online
func (h *InternalHandler) Online(c *gin.Context) {
	nodeID, err := uuid.Parse(c.Param("node_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid node_id"})
		return
	}
	var req models.NodeOnlineRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: err.Error()})
		return
	}
	if err := h.store.UpdateNodeOnline(c.Request.Context(), nodeID, &req); err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}
	c.Status(http.StatusOK)
}

// POST /v1/internal/render/jobs/:job_id/frames
func (h *InternalHandler) FrameProgress(c *gin.Context) {
	jobID, err := uuid.Parse(c.Param("job_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid job_id"})
		return
	}
	var req models.FrameProgressRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: err.Error()})
		return
	}
	if err := h.store.UpdateFrameProgress(c.Request.Context(), jobID, &req); err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}
	c.Status(http.StatusNoContent)
}

// POST /v1/internal/render/jobs/:job_id/crash
func (h *InternalHandler) Crash(c *gin.Context) {
	jobID, err := uuid.Parse(c.Param("job_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid job_id"})
		return
	}
	var req models.CrashReportRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: err.Error()})
		return
	}
	if err := h.store.InsertCrash(c.Request.Context(), req.NodeID, jobID, &req); err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}
	c.JSON(http.StatusOK, gin.H{
		"action_recommended":     "ResetAndRestart",
		"reassigned_to_node_id": nil,
	})
}

// POST /v1/internal/render/jobs/:job_id/replica-ready
func (h *InternalHandler) ReplicaReady(c *gin.Context) {
	_, err := uuid.Parse(c.Param("job_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid job_id"})
		return
	}
	var req models.ReplicaReadyRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: err.Error()})
		return
	}
	// In production: update job step to TemplateCache → JsxGen, signal next pipeline phase
	c.Status(http.StatusNoContent)
}

// POST /v1/internal/render/jobs/:job_id/preview
// Node agent pushes a base64-encoded frame image so the frontend can show
// a live preview while the job is rendering.
func (h *InternalHandler) Preview(c *gin.Context) {
	jobID, err := uuid.Parse(c.Param("job_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid job_id"})
		return
	}
	var req struct {
		ImageB64 string `json:"image_b64" binding:"required"`
	}
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: err.Error()})
		return
	}
	if err := h.store.UpdateJobPreview(c.Request.Context(), jobID, req.ImageB64); err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}
	c.Status(http.StatusNoContent)
}

// POST /v1/internal/render/jobs/claim
// Node agent calls this to atomically claim the next queued job.
// Returns 204 when there is nothing queued (agent should back off and retry).
func (h *InternalHandler) Claim(c *gin.Context) {
	var req models.ClaimJobRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: err.Error()})
		return
	}

	job, err := h.store.ClaimJob(c.Request.Context(), req.NodeID, req.Region)
	if err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}
	if job == nil {
		c.Status(http.StatusNoContent) // nothing queued
		return
	}

	// Generate presigned AEP download URL. AEP files are stored at
	// templates/{original_project_id}/template.aep in the templates bucket.
	// Errors are non-fatal — node agent falls back to mock render when URL is empty.
	aepURL := ""
	if h.minio != nil {
		objectKey := fmt.Sprintf("templates/%s/template.aep", job.OriginalProjectID)
		purl, perr := h.minio.PresignedGetObject(
			context.Background(), h.templatesBucket, objectKey,
			2*time.Hour, nil,
		)
		if perr == nil {
			aepURL = purl.String()
		}
	}

	c.JSON(http.StatusOK, models.ClaimedJob{
		JobID:          job.ID,
		SavedProjectID: job.SavedProjectID,
		Quality:        job.Quality,
		Resolution:     job.Resolution,
		FrameRate:      job.FrameRate,
		HasMusic:       job.HasMusic,
		HasVoiceover:   job.HasVoiceover,
		AEPDownloadURL: aepURL,
	})
}

// POST /v1/internal/render/jobs/:job_id/output-upload-url
// Node agent calls this after rendering to get a presigned MinIO PUT URL.
// Creates an Export record in the DB and returns the export_id + upload URL.
func (h *InternalHandler) OutputUploadURL(c *gin.Context) {
	jobID, err := uuid.Parse(c.Param("job_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid job_id"})
		return
	}

	export, err := h.store.CreateExportForJob(c.Request.Context(), jobID)
	if err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}

	expiry := 2 * time.Hour
	purl, err := h.minio.PresignedPutObject(
		context.Background(), h.exportsBucket, export.Path, expiry,
	)
	if err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{
			Code:    "presign_error",
			Message: "could not generate upload URL",
		})
		return
	}

	c.JSON(http.StatusOK, models.OutputUploadURLResponse{
		ExportID:  export.ID,
		UploadURL: purl.String(),
		ObjectKey: export.Path,
		ExpiresAt: time.Now().Add(expiry),
	})
}

// POST /v1/internal/nodes/:node_id/cache-update
func (h *InternalHandler) CacheUpdate(c *gin.Context) {
	nodeID, err := uuid.Parse(c.Param("node_id"))
	if err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: "invalid node_id"})
		return
	}
	var req models.CacheUpdateRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, models.APIError{Code: "bad_request", Message: err.Error()})
		return
	}
	if err := h.store.UpdateNodeCache(c.Request.Context(), nodeID, &req); err != nil {
		c.JSON(http.StatusInternalServerError, models.APIError{Code: "internal_error", Message: err.Error()})
		return
	}
	c.Status(http.StatusNoContent)
}