feat(tfstated): implement DELETE method

2 files changed, 32 insertions, 0 deletions

diff --git a/cmd/tfstated/delete.go b/cmd/tfstated/delete.go
new file mode 100644
index 0000000..82811b0
--- /dev/null
+++ b/cmd/tfstated/delete.go
@@ -0,0 +1,31 @@
+package main
+
+import (
+	"database/sql"
+	"errors"
+	"fmt"
+	"net/http"
+
+	"git.adyxax.org/adyxax/tfstated/pkg/database"
+)
+
+func handleDelete(db *database.DB) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path == "/" {
+			_ = errorResponse(w, http.StatusBadRequest,
+				fmt.Errorf("no state path provided, cannot DELETE /"))
+			return
+		}
+
+		if err := db.DeleteState(r.URL.Path); err != nil {
+			if errors.Is(err, sql.ErrNoRows) {
+				_ = errorResponse(w, http.StatusNotFound,
+					fmt.Errorf("state path not found: %s", r.URL.Path))
+			} else {
+				_ = errorResponse(w, http.StatusInternalServerError, err)
+			}
+		} else {
+			w.WriteHeader(http.StatusOK)
+		}
+	})
+}
diff --git a/cmd/tfstated/routes.go b/cmd/tfstated/routes.go
index 853970a..da46078 100644
--- a/cmd/tfstated/routes.go
+++ b/cmd/tfstated/routes.go
@@ -12,6 +12,7 @@ func addRoutes(
 ) {
 	mux.Handle("GET /healthz", handleHealthz())
 
+	mux.Handle("DELETE /", handleDelete(db))
 	mux.Handle("GET /", handleGet(db))
 	mux.Handle("POST /", handlePost(db))
 }