initial working disaster

2023-05-01 01:22:25 +01:00
commit 4822ded5a9
10 changed files with 682 additions and 0 deletions
@@ -0,0 +1,99 @@
+package main
+
+import (
+	"encoding/json"
+	"flag"
+	"io"
+	"net/http"
+	"os"
+
+	"github.com/ProtonMail/gopenpgp/v2/crypto"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+	log "github.com/sirupsen/logrus"
+	"github.com/wlcx/debanator"
+)
+
+type responseRecorder struct {
+	http.ResponseWriter
+	status int
+}
+
+func (r responseRecorder) WriteHeader(statusCode int) {
+	r.status = statusCode
+	r.ResponseWriter.WriteHeader(statusCode)
+}
+
+func logMiddleware(h http.HandlerFunc) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		rec := responseRecorder{ResponseWriter: w}
+		h.ServeHTTP(rec, r)
+		log.WithFields(log.Fields{
+			"remote": r.RemoteAddr,
+			"method": r.Method,
+			"url": r.URL,
+			"status": rec.status,
+		}).Info("Got request")
+	}
+}
+
+func md(err error) {
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+func unwrap[T any](val T, err error) T {
+	md(err)
+	return val
+}
+
+func main() {
+	listenAddr := *flag.String("listen", ":1612", "HTTP listen address")
+	debPath := *flag.String("debpath", "debs", "Path to directory containing deb files.")
+	flag.Parse()
+	log.Info("Starting...")
+	var ecKey *crypto.Key
+	kb, err := os.ReadFile("privkey.gpg")
+	if err != nil {
+		log.Infof("Generating new key...")
+		ecKey = unwrap(crypto.GenerateKey("Debanator", "packager@example.com", "x25519", 0))
+		f := unwrap(os.Create("privkey.gpg"))
+		defer f.Close()
+		armored := unwrap(ecKey.Armor())
+		f.WriteString(armored)
+	} else {
+		log.Infof("Using existing key...")
+		ecKey = unwrap(crypto.NewKeyFromArmored(string(kb)))
+	}
+
+	signingKeyRing, err := crypto.NewKeyRing(ecKey)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	repo := debanator.ScanDebs(debPath)
+	if err := repo.GenerateFiles(); err != nil {
+		log.Fatal(err)
+	}
+	log.Infof("Listening on %s", listenAddr)
+	r := chi.NewRouter()
+	r.Use(middleware.Logger)
+	r.Get("/", func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("Content-Type", "application/json")
+		enc := json.NewEncoder(w)
+		if err := enc.Encode(repo); err != nil {
+			log.Errorf("encoding json: %v", err)
+		}
+	})
+	r.Get("/pubkey.gpg", func(w http.ResponseWriter, r *http.Request) {
+		pub, err := ecKey.GetArmoredPublicKey()
+		if err != nil {
+			log.Fatal(err)
+		}
+		io.WriteString(w, pub)
+	})
+	r.Mount("/dists/stable", repo.GetHandler(signingKeyRing))
+	r.Get("/pool/main/*", http.StripPrefix("/pool/main/", http.FileServer(http.Dir(debPath))).ServeHTTP)
+	http.ListenAndServe(listenAddr, r)
+}