summaryrefslogtreecommitdiff
path: root/cmd
diff options
context:
space:
mode:
Diffstat (limited to 'cmd')
-rw-r--r--cmd/tfstated/delete.go28
-rw-r--r--cmd/tfstated/get.go28
-rw-r--r--cmd/tfstated/healthz.go12
-rw-r--r--cmd/tfstated/lock.go61
-rw-r--r--cmd/tfstated/main.go31
-rw-r--r--cmd/tfstated/post.go40
-rw-r--r--cmd/tfstated/routes.go22
-rw-r--r--cmd/tfstated/unlock.go32
8 files changed, 2 insertions, 252 deletions
diff --git a/cmd/tfstated/delete.go b/cmd/tfstated/delete.go
deleted file mode 100644
index d594073..0000000
--- a/cmd/tfstated/delete.go
+++ /dev/null
@@ -1,28 +0,0 @@
-package main
-
-import (
- "fmt"
- "net/http"
-
- "git.adyxax.org/adyxax/tfstated/pkg/database"
- "git.adyxax.org/adyxax/tfstated/pkg/helpers"
-)
-
-func handleDelete(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 DELETE /"))
- return
- }
-
- if success, err := db.DeleteState(r.URL.Path); err != nil {
- helpers.ErrorResponse(w, http.StatusInternalServerError, err)
- } else if success {
- w.WriteHeader(http.StatusOK)
- } else {
- helpers.ErrorResponse(w, http.StatusNotFound,
- fmt.Errorf("state path not found: %s", r.URL.Path))
- }
- })
-}
diff --git a/cmd/tfstated/get.go b/cmd/tfstated/get.go
deleted file mode 100644
index 3310560..0000000
--- a/cmd/tfstated/get.go
+++ /dev/null
@@ -1,28 +0,0 @@
-package main
-
-import (
- "fmt"
- "net/http"
-
- "git.adyxax.org/adyxax/tfstated/pkg/database"
- "git.adyxax.org/adyxax/tfstated/pkg/helpers"
-)
-
-func handleGet(db *database.DB) http.Handler {
- return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Cache-Control", "no-store, no-cache")
-
- if r.URL.Path == "/" {
- helpers.ErrorResponse(w, http.StatusBadRequest,
- fmt.Errorf("no state path provided, cannot GET /"))
- return
- }
-
- if data, err := db.GetState(r.URL.Path); err != nil {
- helpers.ErrorResponse(w, http.StatusInternalServerError, err)
- } else {
- w.WriteHeader(http.StatusOK)
- _, _ = w.Write(data)
- }
- })
-}
diff --git a/cmd/tfstated/healthz.go b/cmd/tfstated/healthz.go
deleted file mode 100644
index 20c72c9..0000000
--- a/cmd/tfstated/healthz.go
+++ /dev/null
@@ -1,12 +0,0 @@
-package main
-
-import "net/http"
-
-func handleHealthz() http.Handler {
- return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Cache-Control", "no-store, no-cache")
- w.Header().Set("Content-Type", "application/json")
- w.WriteHeader(http.StatusOK)
- _, _ = w.Write([]byte("{}"))
- })
-}
diff --git a/cmd/tfstated/lock.go b/cmd/tfstated/lock.go
deleted file mode 100644
index 80e3575..0000000
--- a/cmd/tfstated/lock.go
+++ /dev/null
@@ -1,61 +0,0 @@
-package main
-
-import (
- "fmt"
- "net/http"
- "regexp"
- "time"
-
- "git.adyxax.org/adyxax/tfstated/pkg/database"
- "git.adyxax.org/adyxax/tfstated/pkg/helpers"
-)
-
-type lockRequest struct {
- Created time.Time `json:"Created"`
- ID string `json:"ID"`
- Info string `json:"Info"`
- Operation string `json:"Operation"`
- Path string `json:"Path"`
- Version string `json:"Version"`
- Who string `json:"Who"`
-}
-
-var (
- validID = regexp.MustCompile("[a-f0-9]{8}-(?:[a-f0-9]{4}-){3}[a-f0-9]{12}")
-)
-
-func (l *lockRequest) valid() []error {
- err := make([]error, 0)
- if !validID.MatchString(l.ID) {
- err = append(err, fmt.Errorf("invalid ID"))
- }
- return err
-}
-
-func handleLock(db *database.DB) http.Handler {
- return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- if r.URL.Path == "/" {
- _ = helpers.Encode(w, http.StatusBadRequest,
- fmt.Errorf("no state path provided, cannot LOCK /"))
- return
- }
-
- var lock lockRequest
- if err := helpers.Decode(r, &lock); err != nil {
- _ = helpers.Encode(w, http.StatusBadRequest, err)
- return
- }
- if errs := lock.valid(); len(errs) > 0 {
- _ = helpers.Encode(w, http.StatusBadRequest,
- fmt.Errorf("invalid lock: %+v", errs))
- return
- }
- if success, err := db.SetLockOrGetExistingLock(r.URL.Path, &lock); err != nil {
- helpers.ErrorResponse(w, http.StatusInternalServerError, err)
- } else if success {
- w.WriteHeader(http.StatusOK)
- } else {
- _ = helpers.Encode(w, http.StatusConflict, lock)
- }
- })
-}
diff --git a/cmd/tfstated/main.go b/cmd/tfstated/main.go
index 79b64f7..e6e9a38 100644
--- a/cmd/tfstated/main.go
+++ b/cmd/tfstated/main.go
@@ -4,17 +4,14 @@ import (
"context"
"fmt"
"io"
- "log"
"log/slog"
- "net"
- "net/http"
"os"
"os/signal"
"sync"
"time"
+ "git.adyxax.org/adyxax/tfstated/pkg/backend"
"git.adyxax.org/adyxax/tfstated/pkg/database"
- "git.adyxax.org/adyxax/tfstated/pkg/logger"
)
func run(
@@ -33,31 +30,7 @@ func run(
return err
}
- mux := http.NewServeMux()
- addRoutes(
- mux,
- db,
- )
-
- host := getenv("TFSTATED_HOST")
- if host == "" {
- host = "127.0.0.1"
- }
- port := getenv("TFSTATED_PORT")
- if port == "" {
- port = "8080"
- }
-
- httpServer := &http.Server{
- Addr: net.JoinHostPort(host, port),
- Handler: logger.Middleware(mux, false),
- }
- go func() {
- log.Printf("listening on %s\n", httpServer.Addr)
- if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
- _, _ = fmt.Fprintf(stderr, "error listening and serving: %+v\n", err)
- }
- }()
+ httpServer := backend.Run(ctx, db, getenv, stderr)
var wg sync.WaitGroup
wg.Add(1)
go func() {
diff --git a/cmd/tfstated/post.go b/cmd/tfstated/post.go
deleted file mode 100644
index 86344b1..0000000
--- a/cmd/tfstated/post.go
+++ /dev/null
@@ -1,40 +0,0 @@
-package main
-
-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)
- }
- })
-}
diff --git a/cmd/tfstated/routes.go b/cmd/tfstated/routes.go
deleted file mode 100644
index 019bb76..0000000
--- a/cmd/tfstated/routes.go
+++ /dev/null
@@ -1,22 +0,0 @@
-package main
-
-import (
- "net/http"
-
- "git.adyxax.org/adyxax/tfstated/pkg/basic_auth"
- "git.adyxax.org/adyxax/tfstated/pkg/database"
-)
-
-func addRoutes(
- mux *http.ServeMux,
- db *database.DB,
-) {
- mux.Handle("GET /healthz", handleHealthz())
-
- basicAuth := basic_auth.Middleware(db)
- mux.Handle("DELETE /", basicAuth(handleDelete(db)))
- mux.Handle("GET /", basicAuth(handleGet(db)))
- mux.Handle("LOCK /", basicAuth(handleLock(db)))
- mux.Handle("POST /", basicAuth(handlePost(db)))
- mux.Handle("UNLOCK /", basicAuth(handleUnlock(db)))
-}
diff --git a/cmd/tfstated/unlock.go b/cmd/tfstated/unlock.go
deleted file mode 100644
index c003d8d..0000000
--- a/cmd/tfstated/unlock.go
+++ /dev/null
@@ -1,32 +0,0 @@
-package main
-
-import (
- "fmt"
- "net/http"
-
- "git.adyxax.org/adyxax/tfstated/pkg/database"
- "git.adyxax.org/adyxax/tfstated/pkg/helpers"
-)
-
-func handleUnlock(db *database.DB) http.Handler {
- return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- if r.URL.Path == "/" {
- _ = helpers.Encode(w, http.StatusBadRequest,
- fmt.Errorf("no state path provided, cannot LOCK /"))
- return
- }
-
- var lock lockRequest
- if err := helpers.Decode(r, &lock); err != nil {
- _ = helpers.Encode(w, http.StatusBadRequest, err)
- return
- }
- if success, err := db.Unlock(r.URL.Path, &lock); err != nil {
- helpers.ErrorResponse(w, http.StatusInternalServerError, err)
- } else if success {
- w.WriteHeader(http.StatusOK)
- } else {
- _ = helpers.Encode(w, http.StatusConflict, lock)
- }
- })
-}