88 lines
1.8 KiB
Go
88 lines
1.8 KiB
Go
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)
|
|
}
|