feat(http): implement missing findByID methods (#1635)

* feat(http): implement missing methods

* general cleanup
* unify param handling
* handle not found errors
* unify err handlers

* fix(http): fmt type
This commit is contained in:
ze0s 2024-08-29 12:22:03 +02:00 committed by GitHub
parent accc875960
commit acb91e8709
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 379 additions and 422 deletions

View file

@ -6,6 +6,7 @@ package http
import (
"context"
"encoding/json"
"github.com/autobrr/autobrr/pkg/errors"
"net/http"
"strconv"
@ -16,6 +17,7 @@ import (
type feedService interface {
Find(ctx context.Context) ([]domain.Feed, error)
FindByID(ctx context.Context, id int) (*domain.Feed, error)
Store(ctx context.Context, feed *domain.Feed) error
Update(ctx context.Context, feed *domain.Feed) error
Delete(ctx context.Context, id int) error
@ -44,6 +46,7 @@ func (h feedHandler) Routes(r chi.Router) {
r.Post("/test", h.test)
r.Route("/{feedID}", func(r chi.Router) {
r.Get("/", h.findByID)
r.Put("/", h.update)
r.Delete("/", h.delete)
r.Delete("/cache", h.deleteCache)
@ -54,29 +57,44 @@ func (h feedHandler) Routes(r chi.Router) {
}
func (h feedHandler) find(w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
feeds, err := h.service.Find(ctx)
feeds, err := h.service.Find(r.Context())
if err != nil {
h.encoder.StatusNotFound(w)
h.encoder.Error(w, err)
return
}
h.encoder.StatusResponse(w, http.StatusOK, feeds)
}
func (h feedHandler) store(w http.ResponseWriter, r *http.Request) {
var (
ctx = r.Context()
data *domain.Feed
)
func (h feedHandler) findByID(w http.ResponseWriter, r *http.Request) {
feedID, err := strconv.Atoi(chi.URLParam(r, "feedID"))
if err != nil {
h.encoder.Error(w, err)
return
}
feed, err := h.service.FindByID(r.Context(), feedID)
if err != nil {
if errors.Is(err, domain.ErrRecordNotFound) {
h.encoder.NotFoundErr(w, errors.New("could not find feed with id %d", feedID))
return
}
h.encoder.Error(w, err)
return
}
h.encoder.StatusResponse(w, http.StatusOK, feed)
}
func (h feedHandler) store(w http.ResponseWriter, r *http.Request) {
var data *domain.Feed
if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
h.encoder.Error(w, err)
return
}
err := h.service.Store(ctx, data)
err := h.service.Store(r.Context(), data)
if err != nil {
h.encoder.Error(w, err)
return
@ -86,17 +104,13 @@ func (h feedHandler) store(w http.ResponseWriter, r *http.Request) {
}
func (h feedHandler) test(w http.ResponseWriter, r *http.Request) {
var (
ctx = r.Context()
data *domain.Feed
)
var data *domain.Feed
if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
h.encoder.Error(w, err)
return
}
if err := h.service.Test(ctx, data); err != nil {
if err := h.service.Test(r.Context(), data); err != nil {
h.encoder.Error(w, err)
return
}
@ -105,17 +119,13 @@ func (h feedHandler) test(w http.ResponseWriter, r *http.Request) {
}
func (h feedHandler) update(w http.ResponseWriter, r *http.Request) {
var (
ctx = r.Context()
data *domain.Feed
)
var data *domain.Feed
if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
h.encoder.Error(w, err)
return
}
err := h.service.Update(ctx, data)
err := h.service.Update(r.Context(), data)
if err != nil {
h.encoder.Error(w, err)
return
@ -125,16 +135,18 @@ func (h feedHandler) update(w http.ResponseWriter, r *http.Request) {
}
func (h feedHandler) forceRun(w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
feedID := chi.URLParam(r, "feedID")
id, err := strconv.Atoi(feedID)
feedID, err := strconv.Atoi(chi.URLParam(r, "feedID"))
if err != nil {
h.encoder.Error(w, err)
return
}
if err := h.service.ForceRun(ctx, id); err != nil {
if err := h.service.ForceRun(r.Context(), feedID); err != nil {
if errors.Is(err, domain.ErrRecordNotFound) {
h.encoder.NotFoundErr(w, errors.New("could not find feed with id %d", feedID))
return
}
h.encoder.Error(w, err)
return
}
@ -143,26 +155,27 @@ func (h feedHandler) forceRun(w http.ResponseWriter, r *http.Request) {
}
func (h feedHandler) toggleEnabled(w http.ResponseWriter, r *http.Request) {
var (
ctx = r.Context()
filterID = chi.URLParam(r, "feedID")
data struct {
Enabled bool `json:"enabled"`
}
)
id, err := strconv.Atoi(filterID)
feedID, err := strconv.Atoi(chi.URLParam(r, "feedID"))
if err != nil {
h.encoder.Error(w, err)
return
}
var data struct {
Enabled bool `json:"enabled"`
}
if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
h.encoder.Error(w, err)
return
}
if err := h.service.ToggleEnabled(ctx, id, data.Enabled); err != nil {
if err := h.service.ToggleEnabled(r.Context(), feedID, data.Enabled); err != nil {
if errors.Is(err, domain.ErrRecordNotFound) {
h.encoder.NotFoundErr(w, errors.New("could not find feed with id %d", feedID))
return
}
h.encoder.Error(w, err)
return
}
@ -171,18 +184,18 @@ func (h feedHandler) toggleEnabled(w http.ResponseWriter, r *http.Request) {
}
func (h feedHandler) delete(w http.ResponseWriter, r *http.Request) {
var (
ctx = r.Context()
filterID = chi.URLParam(r, "feedID")
)
id, err := strconv.Atoi(filterID)
feedID, err := strconv.Atoi(chi.URLParam(r, "feedID"))
if err != nil {
h.encoder.Error(w, err)
return
}
if err := h.service.Delete(ctx, id); err != nil {
if err := h.service.Delete(r.Context(), feedID); err != nil {
if errors.Is(err, domain.ErrRecordNotFound) {
h.encoder.NotFoundErr(w, errors.New("could not find feed with id %d", feedID))
return
}
h.encoder.Error(w, err)
return
}
@ -191,18 +204,13 @@ func (h feedHandler) delete(w http.ResponseWriter, r *http.Request) {
}
func (h feedHandler) deleteCache(w http.ResponseWriter, r *http.Request) {
var (
ctx = r.Context()
filterID = chi.URLParam(r, "feedID")
)
id, err := strconv.Atoi(filterID)
feedID, err := strconv.Atoi(chi.URLParam(r, "feedID"))
if err != nil {
h.encoder.Error(w, err)
return
}
if err := h.service.DeleteFeedCache(ctx, id); err != nil {
if err := h.service.DeleteFeedCache(r.Context(), feedID); err != nil {
h.encoder.Error(w, err)
return
}
@ -211,18 +219,13 @@ func (h feedHandler) deleteCache(w http.ResponseWriter, r *http.Request) {
}
func (h feedHandler) latestRun(w http.ResponseWriter, r *http.Request) {
var (
ctx = r.Context()
filterID = chi.URLParam(r, "feedID")
)
id, err := strconv.Atoi(filterID)
feedID, err := strconv.Atoi(chi.URLParam(r, "feedID"))
if err != nil {
h.encoder.Error(w, err)
return
}
feed, err := h.service.GetLastRunData(ctx, id)
feed, err := h.service.GetLastRunData(r.Context(), feedID)
if err != nil {
h.encoder.Error(w, err)
return