package server

import (
	"errors"
	"net/http"
	"strings"

	"g.fc.run/ai/byre/internal/api"
)

func (s *Server) UpdateNotificationPrefs(w http.ResponseWriter, r *http.Request) {
	req := api.NotificationPrefs{}
	if !readJSON(w, r, &req) {
		return
	}

	u, err := s.store.UpdateNotificationPrefs(r.Context(), actor(r), req)
	if err != nil {
		writeStoreError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, u)
}

func (s *Server) UpdateDefaultModel(w http.ResponseWriter, r *http.Request) {
	req := api.DefaultModel{}
	if !readJSON(w, r, &req) {
		return
	}

	if strings.TrimSpace(req.Model) == "" {
		writeError(w, http.StatusBadRequest, errors.New("model is required"))
		return
	}

	u, err := s.store.UpdateDefaultModel(r.Context(), actor(r), req.Model)
	if err != nil {
		writeStoreError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, u)
}

func (s *Server) UpdateQuickRun(w http.ResponseWriter, r *http.Request) {
	req := api.QuickRun{}
	if !readJSON(w, r, &req) {
		return
	}

	if (req.ProjectId == "") != (req.Workflow == "") {
		writeError(w, http.StatusBadRequest, errors.New("project_id and workflow are set together"))
		return
	}
	if req.ProjectId != "" {
		_, err := s.store.GetProject(r.Context(), req.ProjectId)
		if err != nil {
			writeStoreError(w, err)
			return
		}
		_, err = s.store.GetWorkflow(r.Context(), req.Workflow)
		if err != nil {
			writeStoreError(w, err)
			return
		}
	}

	u, err := s.store.UpdateQuickRun(r.Context(), actor(r), req)
	if err != nil {
		writeStoreError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, u)
}

func (s *Server) UpdateAway(w http.ResponseWriter, r *http.Request) {
	req := api.Away{}
	if !readJSON(w, r, &req) {
		return
	}

	u, err := s.agents.SetAway(r.Context(), actor(r), req.Away)
	if err != nil {
		writeAgentError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, u)
}