autobrr/internal/http/feed.go

package http

import (
	"context"
	"encoding/json"
	"net/http"
	"strconv"

	"github.com/autobrr/autobrr/internal/domain"

	"github.com/go-chi/chi"
)

type feedService interface {
	Find(ctx context.Context) ([]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
	ToggleEnabled(ctx context.Context, id int, enabled bool) error
}

type feedHandler struct {
	encoder encoder
	service feedService
}

func newFeedHandler(encoder encoder, service feedService) *feedHandler {
	return &feedHandler{
		encoder: encoder,
		service: service,
	}
}

func (h feedHandler) Routes(r chi.Router) {
	r.Get("/", h.find)
	r.Post("/", h.store)
	r.Put("/{feedID}", h.update)
	r.Patch("/{feedID}/enabled", h.toggleEnabled)
	r.Delete("/{feedID}", h.delete)
}

func (h feedHandler) find(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()

	feeds, err := h.service.Find(ctx)
	if err != nil {
		h.encoder.StatusNotFound(ctx, w)
		return
	}

	h.encoder.StatusResponse(ctx, w, feeds, http.StatusOK)
}

func (h feedHandler) store(w http.ResponseWriter, r *http.Request) {
	var (
		ctx  = r.Context()
		data *domain.Feed
	)

	if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
		// encode error
		h.encoder.StatusNotFound(ctx, w)
		return
	}

	err := h.service.Store(ctx, data)
	if err != nil {
		// encode error
		h.encoder.StatusInternalError(w)
		return
	}

	h.encoder.StatusResponse(ctx, w, data, http.StatusCreated)
}

func (h feedHandler) update(w http.ResponseWriter, r *http.Request) {
	var (
		ctx  = r.Context()
		data *domain.Feed
	)

	if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
		// encode error
		h.encoder.StatusInternalError(w)
		return
	}

	err := h.service.Update(ctx, data)
	if err != nil {
		// encode error
		h.encoder.StatusInternalError(w)
		return
	}

	h.encoder.StatusResponse(ctx, w, data, http.StatusCreated)
}

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, _ := strconv.Atoi(filterID)

	if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
		// encode error
		h.encoder.StatusInternalError(w)
		return
	}

	err := h.service.ToggleEnabled(ctx, id, data.Enabled)
	if err != nil {
		// encode error
		h.encoder.StatusInternalError(w)
		return
	}

	h.encoder.StatusResponse(ctx, w, nil, http.StatusNoContent)
}

func (h feedHandler) delete(w http.ResponseWriter, r *http.Request) {
	var (
		ctx      = r.Context()
		filterID = chi.URLParam(r, "feedID")
	)

	id, _ := strconv.Atoi(filterID)

	if err := h.service.Delete(ctx, id); err != nil {
		h.encoder.StatusInternalError(w)
		return
	}

	h.encoder.StatusResponse(ctx, w, nil, http.StatusNoContent)
}