kaspad/httpserverutils/request.go

package httpserverutils

import (
	"fmt"
	"github.com/gorilla/mux"
	"io/ioutil"
	"net/http"
)

// HandlerFunc is a handler function that is passed to the
// MakeHandler wrapper and gets the relevant request fields
// from it.
type HandlerFunc func(ctx *ServerContext, r *http.Request, routeParams map[string]string, queryParams map[string]string, requestBody []byte) (
	interface{}, *HandlerError)

// MakeHandler is a wrapper function that takes a handler in the form of HandlerFunc
// and returns a function that can be used as a handler in mux.Router.HandleFunc.
func MakeHandler(handler HandlerFunc) func(http.ResponseWriter, *http.Request) {
	return func(w http.ResponseWriter, r *http.Request) {
		ctx := ToServerContext(r.Context())

		var requestBody []byte
		if r.Method == "POST" {
			var err error
			requestBody, err = ioutil.ReadAll(r.Body)
			if err != nil {
				SendErr(ctx, w, NewInternalServerHandlerError("Error reading POST data"))
			}
		}

		flattenedQueryParams, hErr := flattenQueryParams(r.URL.Query())
		if hErr != nil {
			SendErr(ctx, w, hErr)
			return
		}

		response, hErr := handler(ctx, r, mux.Vars(r), flattenedQueryParams, requestBody)
		if hErr != nil {
			SendErr(ctx, w, hErr)
			return
		}
		if response != nil {
			SendJSONResponse(w, response)
		}
	}
}

func flattenQueryParams(queryParams map[string][]string) (map[string]string, *HandlerError) {
	flattenedMap := make(map[string]string)
	for param, valuesSlice := range queryParams {
		if len(valuesSlice) > 1 {
			return nil, NewHandlerError(http.StatusUnprocessableEntity, fmt.Sprintf("Couldn't parse the '%s' query parameter:"+
				" expected a single value but got multiple values", param))
		}
		flattenedMap[param] = valuesSlice[0]
	}
	return flattenedMap, nil
}