65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
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,
|
|
})
|
|
}
|