tfstated/pkg/backend/post.go

package backend

import (
	"fmt"
	"io"
	"net/http"

	"git.adyxax.org/adyxax/tfstated/pkg/database"
	"git.adyxax.org/adyxax/tfstated/pkg/helpers"
	"git.adyxax.org/adyxax/tfstated/pkg/model"
)

func handlePost(db *database.DB) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path == "/" {
			helpers.ErrorResponse(w, http.StatusBadRequest,
				fmt.Errorf("no state path provided, cannot POST /"),
			)
			return
		}

		id := r.URL.Query().Get("ID")

		data, err := io.ReadAll(r.Body)
		if err != nil || len(data) == 0 {
			helpers.ErrorResponse(w, http.StatusBadRequest, err)
			return
		}
		account := r.Context().Value(model.AccountContextKey{}).(*model.Account)
		if idMismatch, err := db.SetState(r.URL.Path, account.Id, data, id); err != nil {
			if idMismatch {
				helpers.ErrorResponse(w, http.StatusConflict, err)
			} else {
				helpers.ErrorResponse(w, http.StatusInternalServerError, err)
			}
		} else {
			w.WriteHeader(http.StatusOK)
		}
	})
}