Organization for stupid-backend

stupid.go

@@ -0,0 +1,118 @@
+package main
+
+import (
+	"context"
+	"io"
+	"log"
+	"net/http"
+	"os"
+
+	"github.com/CalebQ42/stupid-backend"
+	"github.com/CalebQ42/stupid-backend/pkg/db"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+func setupStupid(keyPath, mongoStr string) error {
+	client, err := mongo.NewClient(options.Client().ApplyURI(mongoStr))
+	if err != nil {
+		log.Println("Issues connecting to mongo:", err)
+		return err
+	}
+	err = client.Connect(context.TODO())
+	if err != nil {
+		log.Println("Issues connecting to mongo:", err)
+		return err
+	}
+	swApp := &stupid.App{
+		Logs:    db.NewMongoTable(client.Database("swassistant").Collection("log")),
+		Crashes: db.NewMongoTable(client.Database("swassistant").Collection("crash")),
+	}
+	cdrApp := &stupid.App{
+		Logs:    db.NewMongoTable(client.Database("cdr").Collection("log")),
+		Crashes: db.NewMongoTable(client.Database("cdr").Collection("crash")),
+	}
+	websiteApp := &stupid.App{
+		Logs:    db.NewMongoTable(client.Database("darkstormtech").Collection("log")),
+		Crashes: db.NewMongoTable(client.Database("darkstormtech").Collection("crash")),
+		Extension: func(r *stupid.Request) bool {
+			return websiteRequest(*client.Database("darkstormtech"), r)
+		},
+	}
+	stupid := stupid.NewStupidBackend(db.NewMongoTable(client.Database("stupid").Collection("keys")), func(app string) *stupid.App {
+		switch app {
+		case "swassistant":
+			return swApp
+		case "cdr":
+			return cdrApp
+		case "darkstormtech":
+			return websiteApp
+		}
+		return nil
+	})
+	users := true
+	var pub, priv []byte
+	stupidPubFil, err := os.Open(keyPath + "/stupid-pub.key")
+	if err != nil {
+		log.Println("Disabling API users:", err)
+		users = false
+	} else {
+		pub, err = io.ReadAll(stupidPubFil)
+		if err != nil {
+			log.Println("Disabling API users:", err)
+			users = false
+		}
+	}
+	stupidPrivFil, err := os.Open(keyPath + "/stupid-pub.key")
+	if err != nil {
+		log.Println("Disabling API users:", err)
+		users = false
+	} else {
+		priv, err = io.ReadAll(stupidPrivFil)
+		if err != nil {
+			log.Println("Disabling API users:", err)
+			users = false
+		}
+	}
+	if users {
+		stupid.EnableUserAuth(db.NewMongoTable(client.Database("stupid").Collection("keys")), pub, priv)
+	}
+	http.Handle("api.darkstorm.tech/", stupid)
+	return nil
+}
+
+type page struct {
+	Page     string
+	Contents string
+}
+
+func websiteRequest(d mongo.Database, r *stupid.Request) bool {
+	if len(r.Path) > 0 {
+		r.Resp.WriteHeader(http.StatusBadRequest)
+		return true
+	}
+	if pag, ok := r.Query["page"]; ok {
+		res := d.Collection("pages").FindOne(context.TODO(), bson.M{"page": pag})
+		if res.Err() == mongo.ErrNoDocuments {
+			r.Resp.WriteHeader(http.StatusNotFound)
+			return true
+		}
+		var p page
+		err := res.Decode(&p)
+		if err != nil {
+			r.Resp.WriteHeader(http.StatusInternalServerError)
+			log.Print("page decode:", err)
+			return true
+		}
+		_, err = r.Resp.Write([]byte(p.Contents))
+		if err != nil {
+			r.Resp.WriteHeader(http.StatusInternalServerError)
+			log.Print("content send:", err)
+			return true
+		}
+		return true
+	}
+	r.Resp.WriteHeader(http.StatusBadRequest)
+	return true
+}