mirror of https://github.com/artifacthub/hub.git
108 lines
3.5 KiB
Go
108 lines
3.5 KiB
Go
package apikey
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/artifacthub/hub/internal/handlers/helpers"
|
|
"github.com/artifacthub/hub/internal/hub"
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
// Handlers represents a group of http handlers in charge of handling api keys
|
|
// operations.
|
|
type Handlers struct {
|
|
apiKeyManager hub.APIKeyManager
|
|
logger zerolog.Logger
|
|
}
|
|
|
|
// NewHandlers creates a new Handlers instance.
|
|
func NewHandlers(apiKeyManager hub.APIKeyManager) *Handlers {
|
|
return &Handlers{
|
|
apiKeyManager: apiKeyManager,
|
|
logger: log.With().Str("handlers", "apikey").Logger(),
|
|
}
|
|
}
|
|
|
|
// Add is an http handler that adds the provided api key to the database.
|
|
func (h *Handlers) Add(w http.ResponseWriter, r *http.Request) {
|
|
akIN := &hub.APIKey{}
|
|
if err := json.NewDecoder(r.Body).Decode(&akIN); err != nil {
|
|
h.logger.Error().Err(err).Str("method", "Add").Msg(hub.ErrInvalidInput.Error())
|
|
helpers.RenderErrorJSON(w, hub.ErrInvalidInput)
|
|
return
|
|
}
|
|
akOUT, err := h.apiKeyManager.Add(r.Context(), akIN)
|
|
if err != nil {
|
|
h.logger.Error().Err(err).Str("method", "Add").Send()
|
|
helpers.RenderErrorJSON(w, err)
|
|
return
|
|
}
|
|
akOUTJSON, _ := json.Marshal(akOUT)
|
|
helpers.RenderJSON(w, akOUTJSON, 0, http.StatusCreated)
|
|
}
|
|
|
|
// Delete is an http handler that deletes the provided api key from the database.
|
|
func (h *Handlers) Delete(w http.ResponseWriter, r *http.Request) {
|
|
apiKeyID := chi.URLParam(r, "apiKeyID")
|
|
if err := h.apiKeyManager.Delete(r.Context(), apiKeyID); err != nil {
|
|
h.logger.Error().Err(err).Str("method", "Delete").Send()
|
|
helpers.RenderErrorJSON(w, err)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusNoContent)
|
|
}
|
|
|
|
// Get is an http handler that returns the requested api key.
|
|
func (h *Handlers) Get(w http.ResponseWriter, r *http.Request) {
|
|
apiKeyID := chi.URLParam(r, "apiKeyID")
|
|
dataJSON, err := h.apiKeyManager.GetJSON(r.Context(), apiKeyID)
|
|
if err != nil {
|
|
h.logger.Error().Err(err).Str("method", "Get").Send()
|
|
helpers.RenderErrorJSON(w, err)
|
|
return
|
|
}
|
|
helpers.RenderJSON(w, dataJSON, 0, http.StatusOK)
|
|
}
|
|
|
|
// GetOwnedByUser is an http handler that returns the api keys owned by the
|
|
// user doing the request.
|
|
func (h *Handlers) GetOwnedByUser(w http.ResponseWriter, r *http.Request) {
|
|
p, err := helpers.GetPagination(r.URL.Query(), helpers.PaginationDefaultLimit, helpers.PaginationMaxLimit)
|
|
if err != nil {
|
|
err = fmt.Errorf("%w: %w", hub.ErrInvalidInput, err)
|
|
h.logger.Error().Err(err).Str("query", r.URL.RawQuery).Str("method", "GetOwnedByUser").Send()
|
|
helpers.RenderErrorJSON(w, err)
|
|
return
|
|
}
|
|
result, err := h.apiKeyManager.GetOwnedByUserJSON(r.Context(), p)
|
|
if err != nil {
|
|
h.logger.Error().Err(err).Str("method", "GetOwnedByUser").Send()
|
|
helpers.RenderErrorJSON(w, err)
|
|
return
|
|
}
|
|
w.Header().Set(helpers.PaginationTotalCount, strconv.Itoa(result.TotalCount))
|
|
helpers.RenderJSON(w, result.Data, 0, http.StatusOK)
|
|
}
|
|
|
|
// Update is an http handler that updates the provided api key in the database.
|
|
func (h *Handlers) Update(w http.ResponseWriter, r *http.Request) {
|
|
ak := &hub.APIKey{}
|
|
if err := json.NewDecoder(r.Body).Decode(&ak); err != nil {
|
|
h.logger.Error().Err(err).Str("method", "Update").Msg(hub.ErrInvalidInput.Error())
|
|
helpers.RenderErrorJSON(w, hub.ErrInvalidInput)
|
|
return
|
|
}
|
|
ak.APIKeyID = chi.URLParam(r, "apiKeyID")
|
|
if err := h.apiKeyManager.Update(r.Context(), ak); err != nil {
|
|
h.logger.Error().Err(err).Str("method", "Update").Send()
|
|
helpers.RenderErrorJSON(w, err)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusNoContent)
|
|
}
|