package api

import (
	"net/http"

	"forge.cadoles.com/Cadoles/emissary/internal/datastore"
	"github.com/pkg/errors"
	"gitlab.com/wpetit/goweb/api"
	"gitlab.com/wpetit/goweb/logger"
)

type updateAgentRequest struct {
	Status *datastore.AgentStatus `json:"status" validate:"omitempty,oneof=0 1 2 3"`
	Label  *string                `json:"label" validate:"omitempty"`
}

func (m *Mount) updateAgent(w http.ResponseWriter, r *http.Request) {
	agentID, ok := getAgentID(w, r)
	if !ok {
		return
	}

	ctx := r.Context()

	updateAgentReq := &updateAgentRequest{}
	if ok := api.Bind(w, r, updateAgentReq); !ok {
		return
	}

	options := make([]datastore.AgentUpdateOptionFunc, 0)

	if updateAgentReq.Status != nil {
		options = append(options, datastore.WithAgentUpdateStatus(*updateAgentReq.Status))
	}

	if updateAgentReq.Label != nil {
		options = append(options, datastore.WithAgentUpdateLabel(*updateAgentReq.Label))
	}

	agent, err := m.agentRepo.Update(
		ctx,
		datastore.AgentID(agentID),
		options...,
	)
	if err != nil {
		if errors.Is(err, datastore.ErrNotFound) {
			api.ErrorResponse(w, http.StatusNotFound, ErrCodeNotFound, nil)

			return
		}

		err = errors.WithStack(err)
		logger.Error(ctx, "could not update agent", logger.CapturedE(err))
		api.ErrorResponse(w, http.StatusInternalServerError, ErrCodeUnknownError, nil)

		return
	}

	api.DataResponse(w, http.StatusOK, struct {
		Agent *datastore.Agent `json:"agent"`
	}{
		Agent: agent,
	})
}