78 lines
1.8 KiB
Go
78 lines
1.8 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 claimAgentRequest struct {
|
||
|
|
Thumbprint string `json:"thumbprint" validate:"required"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Mount) claimAgent(w http.ResponseWriter, r *http.Request) {
|
||
|
|
user, ok := assertRequestUser(w, r)
|
||
|
|
if !ok {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx := r.Context()
|
||
|
|
|
||
|
|
claimAgentReq := &claimAgentRequest{}
|
||
|
|
if ok := api.Bind(w, r, claimAgentReq); !ok {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
results, _, err := m.agentRepo.Query(
|
||
|
|
ctx,
|
||
|
|
datastore.WithAgentQueryThumbprints(claimAgentReq.Thumbprint),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
err = errors.WithStack(err)
|
||
|
|
logger.Error(ctx, "could not query agents", logger.CapturedE(err))
|
||
|
|
api.ErrorResponse(w, http.StatusInternalServerError, ErrCodeUnknownError, nil)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(results) == 0 {
|
||
|
|
api.ErrorResponse(w, http.StatusNotFound, ErrCodeNotFound, nil)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(results) > 1 {
|
||
|
|
logger.Error(ctx, "multiple results for agent thumbprint", logger.F("agentThumbprint", claimAgentReq.Thumbprint))
|
||
|
|
api.ErrorResponse(w, http.StatusInternalServerError, ErrCodeMultipleResults, nil)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
agent := results[0]
|
||
|
|
|
||
|
|
if agent.TenantID != nil {
|
||
|
|
logger.Error(ctx, "agent already claimed", logger.F("agentThumbprint", claimAgentReq.Thumbprint), logger.F("agentID", agent.ID), logger.F("tenantID", agent.TenantID))
|
||
|
|
api.ErrorResponse(w, http.StatusConflict, ErrCodeAlreadyClaimed, nil)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
agent, err = m.agentRepo.Attach(ctx, user.Tenant(), agent.ID)
|
||
|
|
if err != nil {
|
||
|
|
err = errors.WithStack(err)
|
||
|
|
logger.Error(ctx, "could not attach 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,
|
||
|
|
})
|
||
|
|
}
|