handlers.go

package oauth2

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

	"github.com/hellofresh/janus/pkg/errors"
	"github.com/hellofresh/janus/pkg/render"
	"github.com/hellofresh/janus/pkg/router"
	"go.opencensus.io/trace"
)

// Controller is the api rest controller
type Controller struct {
	repo Repository
}

// NewController creates a new instance of Controller
func NewController(repo Repository) *Controller {
	return &Controller{repo}
}

// Get is the find all handler
func (c *Controller) Get() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		_, span := trace.StartSpan(r.Context(), "repo.FindAll")
		data, err := c.repo.FindAll()
		span.End()

		if err != nil {
			errors.Handler(w, r, err)
			return
		}

		render.JSON(w, http.StatusOK, data)
	}
}

// GetBy is the find by handler
func (c *Controller) GetBy() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		name := router.URLParam(r, "name")
		_, span := trace.StartSpan(r.Context(), "repo.FindByName")
		data, err := c.repo.FindByName(name)
		span.End()

		if err != nil {
			errors.Handler(w, r, err)
			return
		}

		render.JSON(w, http.StatusOK, data)
	}
}

// PutBy is the update handler
func (c *Controller) PutBy() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		var err error
		name := router.URLParam(r, "name")

		_, span := trace.StartSpan(r.Context(), "repo.FindByName")
		oauth, err := c.repo.FindByName(name)
		span.End()

		if oauth.Name == "" {
			errors.Handler(w, r, ErrOauthServerNotFound)
			return
		}

		if err != nil {
			errors.Handler(w, r, err)
			return
		}

		err = json.NewDecoder(r.Body).Decode(oauth)
		if err != nil {
			errors.Handler(w, r, err)
			return
		}

		_, span = trace.StartSpan(r.Context(), "repo.Save")
		err = c.repo.Save(oauth)
		span.End()

		if err != nil {
			errors.Handler(w, r, errors.New(http.StatusBadRequest, err.Error()))
			return
		}

		w.WriteHeader(http.StatusOK)
	}
}

// Post is the create handler
func (c *Controller) Post() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		oauth := NewOAuth()
		err := json.NewDecoder(r.Body).Decode(oauth)
		if nil != err {
			errors.Handler(w, r, err)
			return
		}

		_, span := trace.StartSpan(r.Context(), "repo.Add")
		err = c.repo.Add(oauth)
		span.End()

		if nil != err {
			errors.Handler(w, r, err)
			return
		}

		w.WriteHeader(http.StatusCreated)
	}
}

// DeleteBy is the delete handler
func (c *Controller) DeleteBy() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		name := router.URLParam(r, "name")

		_, span := trace.StartSpan(r.Context(), "repo.Remove")
		err := c.repo.Remove(name)
		span.End()

		if err != nil {
			errors.Handler(w, r, err)
			return
		}

		w.WriteHeader(http.StatusNoContent)
	}
}