package server

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

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

func (s *Server) ListLeadSessions(w http.ResponseWriter, r *http.Request) {
	sessions, err := s.agents.LeadSessions(r.Context(), actor(r))
	if err != nil {
		writeStoreError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, sessions)
}

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

	images, ok := s.images(w, r, req.Images)
	if !ok {
		return
	}

	if strings.TrimSpace(req.Text) == "" && len(images) == 0 {
		writeError(w, http.StatusBadRequest, errors.New("text or images are required"))
		return
	}

	u, err := s.store.GetUser(r.Context(), actor(r))
	if err != nil {
		writeStoreError(w, err)
		return
	}

	model := ""
	if req.Model != nil {
		model = *req.Model
	}

	sess, err := s.agents.MessageLead(r.Context(), u, req.Text, model, images)
	if err != nil {
		writeAgentError(w, err)
		return
	}
	writeJSON(w, http.StatusAccepted, sess)
}

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

func (s *Server) SetFocus(w http.ResponseWriter, r *http.Request) {
	f := api.Focus{}
	if !readJSON(w, r, &f) {
		return
	}

	err := s.agents.SetFocus(r.Context(), actor(r), f, actor(r))
	if err != nil {
		writeAgentError(w, err)
		return
	}

	f, err = s.store.GetFocus(r.Context(), actor(r))
	if err != nil {
		writeStoreError(w, err)
		return
	}
	writeJSON(w, http.StatusOK, f)
}