byre/internal/server/workflows.go

package server

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

	"g.fc.run/ai/byre/internal/api"
	"g.fc.run/ai/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) {
	accept := r.Header.Get("Accept")
	if strings.Contains(accept, "application/json") && !strings.Contains(accept, "yaml") {
		def, err := s.store.GetWorkflow(r.Context(), name)
		if err != nil {
			writeStoreError(w, err)
			return
		}
		writeJSON(w, http.StatusOK, def)
		return
	}
	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, params api.PutWorkflowParams) {
	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
	}
	from := ""
	if params.From != nil {
		from = *params.From
	}
	summary, err := s.agents.SaveWorkflow(r.Context(), name, from, file, actor(r))
	if err != nil {
		writeAgentError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, summary)
}

func (s *Server) SetReviewWorkflow(w http.ResponseWriter, r *http.Request, projectID string) {
	req := api.ReviewWorkflow{}
	if !readJSON(w, r, &req) {
		return
	}
	p, err := s.agents.SetReviewWorkflow(r.Context(), projectID, req.Workflow, actor(r))
	if err != nil {
		writeAgentError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, p)
}

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)
}