add commands for manipulating the "database"prod

1 files changed, 20 insertions, 0 deletions

diff --git a/main.go b/main.go
index eaba20e..9f62b38 100644
--- a/main.go
+++ b/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	_ "embed"
 	"encoding/gob"
+	"fmt"
 	"io"
 	"log/slog"
 	"net/http"
@@ -57,6 +58,25 @@ func main() {
 		dec.Decode(&migrated)
 	}
 
+	if len(os.Args) > 1 {
+		switch os.Args[1] {
+		case "migrate":
+			migrated[os.Args[2]] = struct{}{}
+			save()
+		case "unmigrate":
+			delete(migrated, os.Args[2])
+			save()
+		case "list":
+			for k := range migrated {
+				fmt.Println(k)
+			}
+		default:
+			fmt.Println("Usage:")
+			fmt.Printf("\t%s <[migrate|unmigrate|list] [padid]>", os.Args[0])
+		}
+		os.Exit(0)
+	}
+
 	http.HandleFunc("GET /p/{id}", func(w http.ResponseWriter, r *http.Request) {
 		id := r.PathValue("id")