handlers.create.go

package server

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"

	"github.com/go-chi/chi"
	"github.com/gofrs/uuid"
	log "github.com/sirupsen/logrus"

	"go.jlucktay.dev/rest-api/pkg/storage"
)

func (s *Server) createPayments() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if r.ContentLength == 0 {
			log.Debug("Request body was empty, returning.")
			http.Error(w, "Empty request body.", http.StatusBadRequest) // 400

			return
		}

		bodyBytes, errRead := ioutil.ReadAll(r.Body)
		if errRead != nil {
			log.Fatal(errRead)
		}
		defer r.Body.Close()

		var p storage.Payment

		errUm := json.Unmarshal(bodyBytes, &p)
		if errUm != nil {
			log.Fatal(errUm)
		}

		id, errCreate := s.Storage.Create(p)
		if errCreate != nil {
			log.Fatal(errCreate)
		}

		w.Header().Set("Location", fmt.Sprintf("/v1/payments/%s", id))
		w.WriteHeader(http.StatusCreated) // 201
	}
}

func (s *Server) createPaymentByID() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		id := uuid.FromStringOrNil(chi.URLParam(r, "id"))

		if id == uuid.Nil {
			log.Debug("ID was invalid, returning.")
			http.Error(w, "Invalid ID.", http.StatusNotFound) // 404

			return
		}

		_, errRead := s.Storage.Read(id)
		if errRead == nil {
			log.Debug("ID already existed and was found, returning.")
			http.Error(w, (&storage.AlreadyExistsError{ID: id}).Error(), http.StatusConflict) // 409

			return
		}

		log.Debug("ID was given but should not have been, returning.")
		http.Error(w, `Cannot specify an ID for payment creation.
One will be generated for you.`, http.StatusNotFound) // 404
	}
}