package server

import (
	"fmt"
	"io"
	"net/http"
	"net/url"

	"g.fc.run/misc/byre/internal/workflow"
)

const maxWorkflowFileBytes = 1 << 20

func (s *Server) ListWorkflows(w http.ResponseWriter, r *http.Request) {
	workflows, err := s.store.ListWorkflows(r.Context())
	if err != nil {
		writeStoreError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, workflows)
}

func (s *Server) GetWorkflow(w http.ResponseWriter, r *http.Request, name string) {
	file, err := s.agents.WorkflowFile(r.Context(), name)
	if err != nil {
		writeAgentError(w, err)
		return
	}
	w.Header().Set("Content-Type", "application/yaml")
	w.Header().Set("Content-Disposition", "attachment; filename*=UTF-8''"+url.PathEscape(workflow.Filename(name)))
	w.WriteHeader(http.StatusOK)
	w.Write(file)
}

func (s *Server) PutWorkflow(w http.ResponseWriter, r *http.Request, name string) {
	file, err := io.ReadAll(http.MaxBytesReader(w, r.Body, maxWorkflowFileBytes))
	if err != nil {
		writeError(w, http.StatusBadRequest, fmt.Errorf("read workflow file: %w", err))
		return
	}
	summary, err := s.agents.SaveWorkflow(r.Context(), name, file, actor(r))
	if err != nil {
		writeAgentError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, summary)
}

func (s *Server) DeleteWorkflow(w http.ResponseWriter, r *http.Request, name string) {
	err := s.store.DeleteWorkflow(r.Context(), name, actor(r))
	if err != nil {
		writeStoreError(w, err)
		return
	}
	w.WriteHeader(http.StatusNoContent)
}

func (s *Server) GetTaskWorkflow(w http.ResponseWriter, r *http.Request, taskID string) {
	tw, err := s.agents.TaskWorkflow(r.Context(), taskID)
	if err != nil {
		writeStoreError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, tw)
}

func (s *Server) GetWorkflowRun(w http.ResponseWriter, r *http.Request, taskID string) {
	run, err := s.agents.WorkflowRun(r.Context(), taskID)
	if err != nil {
		writeAgentError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, run)
}