passed/routes/create_password.go

package routes

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

	"git.1e99.eu/1e99/passed/storage"
)

func CreatePassword(store storage.Store, maxLength int, encoding *base64.Encoding) http.HandlerFunc {
	return func(res http.ResponseWriter, req *http.Request) {
		var reqBody struct {
			// Go automatically decodes byte arrays using Base64
			Password  []byte        `json:"password"`
			ExpiresIn time.Duration `json:"expires-in"`
		}
		err := json.NewDecoder(req.Body).Decode(&reqBody)
		if err != nil {
			http.Error(res, "Malformed JSON body", http.StatusBadRequest)
			return
		}

		if len(reqBody.Password) > maxLength {
			http.Error(res, "Password too long", http.StatusRequestEntityTooLarge)
			return
		}

		expiresIn := reqBody.ExpiresIn * time.Second
		if expiresIn <= 0 {
			http.Error(res, "Too short expires-in", http.StatusBadRequest)
			return
		}

		if expiresIn > 2*7*24*time.Hour {
			http.Error(res, "Too long expires-in", http.StatusBadRequest)
			return
		}

		id, err := store.Create(
			reqBody.Password,
			time.Now().Add(expiresIn),
		)
		switch {
		case err == storage.ErrFull:
			http.Error(res, "Insufficient storage", http.StatusInsufficientStorage)
			return
		case err != nil:
			http.Error(res, "", http.StatusInternalServerError)
			return
		}

		resBody := struct {
			Id string `json:"id"`
		}{
			Id: id,
		}
		err = json.NewEncoder(res).Encode(&resBody)
		if err != nil {
			http.Error(res, "", http.StatusInternalServerError)
			return
		}
	}
}