emissary/internal/server/spec_api.go

package server

import (
	"net/http"
	"strconv"

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

const (
	ErrCodeUnexpectedRevision api.ErrorCode = "unexpected-revision"
)

type updateSpecRequest struct {
	Name     string         `json:"name"`
	Revision int            `json:"revision"`
	Data     map[string]any `json:"data"`
}

func (s *Server) updateSpec(w http.ResponseWriter, r *http.Request) {
	agentID, ok := getAgentID(w, r)
	if !ok {
		return
	}

	ctx := r.Context()

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

	spec, err := s.agentRepo.UpdateSpec(
		ctx,
		datastore.AgentID(agentID),
		updateSpecReq.Name,
		updateSpecReq.Revision,
		updateSpecReq.Data,
	)
	if err != nil {
		if errors.Is(err, datastore.ErrUnexpectedRevision) {
			api.ErrorResponse(w, http.StatusConflict, ErrCodeUnexpectedRevision, nil)

			return
		}

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

		return
	}

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

func (s *Server) getAgentSpecs(w http.ResponseWriter, r *http.Request) {
	agentID, ok := getAgentID(w, r)
	if !ok {
		return
	}

	ctx := r.Context()

	specs, err := s.agentRepo.GetSpecs(ctx, agentID)
	if err != nil {
		logger.Error(ctx, "could not list specs", logger.E(errors.WithStack(err)))
		api.ErrorResponse(w, http.StatusInternalServerError, ErrCodeUnknownError, nil)

		return
	}

	api.DataResponse(w, http.StatusOK, struct {
		Specs []*datastore.Spec `json:"specs"`
	}{
		Specs: specs,
	})
}

type deleteSpecRequest struct {
	Name string `json:"name"`
}

func (s *Server) deleteSpec(w http.ResponseWriter, r *http.Request) {
	agentID, ok := getAgentID(w, r)
	if !ok {
		return
	}

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

	ctx := r.Context()

	err := s.agentRepo.DeleteSpec(
		ctx,
		agentID,
		deleteSpecReq.Name,
	)
	if err != nil {
		if errors.Is(err, datastore.ErrNotFound) {
			api.ErrorResponse(w, http.StatusNotFound, ErrCodeNotFound, nil)

			return
		}

		logger.Error(ctx, "could not delete spec", logger.E(errors.WithStack(err)))
		api.ErrorResponse(w, http.StatusInternalServerError, ErrCodeUnknownError, nil)

		return
	}

	api.DataResponse(w, http.StatusOK, struct {
		Name string `json:"name"`
	}{
		Name: deleteSpecReq.Name,
	})
}

func getSpecID(w http.ResponseWriter, r *http.Request) (datastore.SpecID, bool) {
	rawSpecID := chi.URLParam(r, "")

	specID, err := strconv.ParseInt(rawSpecID, 10, 64)
	if err != nil {
		logger.Error(r.Context(), "could not parse spec id", logger.E(errors.WithStack(err)))
		api.ErrorResponse(w, http.StatusBadRequest, api.ErrCodeMalformedRequest, nil)

		return 0, false
	}

	return datastore.SpecID(specID), true
}