darkstorm-server/main.go

package main

import (
	"context"
	"crypto/tls"
	"flag"
	"log"
	"net/http"
	"net/http/httputil"
	"net/url"
	"os"
	"path"
	"path/filepath"
	"strings"

	"github.com/CalebQ42/darkstorm-server/internal/backend"
	"github.com/CalebQ42/darkstorm-server/internal/backend/db"
	"github.com/CalebQ42/darkstorm-server/internal/blog"
	"github.com/CalebQ42/darkstorm-server/internal/cdr"
	"github.com/CalebQ42/darkstorm-server/internal/swassistant"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var (
	mongoClient *mongo.Client
	back        *backend.Backend
	blogApp     *blog.BlogApp
	webRoot     *string
	testing     *bool
)

func main() {
	mongoURL := flag.String("mongo", "", "Enables MongoDB usage for Darkstorm backend.")
	webRoot = flag.String("web-root", "", "Sets root directory of web server.")
	addr := flag.String("addr", ":443", "Set listen address. Defaults to \":443\"")
	testing = flag.Bool("testing", false, "Start in testing mode. If you don't know what this is, don't use it.")
	flag.Parse()
	if *testing {
		*addr = ":4242"
	}
	if !*testing && flag.NArg() != 1 {
		log.Fatal("You must specify key directory. ex: darkstorm-server /etc/web-keys")
	}
	if *mongoURL == "" || *webRoot == "" {
		log.Fatal("SPECIFY MONGO AND WEB-ROOT OR I WILL DIE, OH NO, THEY'RE COMING FOR ME.... **DEATH NOISES**")
	}
	if !*testing {
		go func() {
			log.Println("error redirecting http traffice:",
				http.ListenAndServe(":80", http.RedirectHandler("https://darkstorm.tech", http.StatusPermanentRedirect)))
		}()
	}
	mux := http.NewServeMux()
	setupMongo(*mongoURL)
	setupBackend(mux)
	setupWebsite(mux)
	serv := &http.Server{
		Addr:    *addr,
		Handler: mux,
	}
	var err error
	if *testing {
		err = serv.ListenAndServe()
	} else {
		err = serv.ListenAndServeTLS(filepath.Join(flag.Arg(0), "fullchain.pem"), filepath.Join(flag.Arg(0), "key.pem"))
	}
	log.Println("webserver closed:", err)
}

func setupMongo(uri string) {
	if !*testing {
		mongoCert, err := tls.LoadX509KeyPair(filepath.Join(flag.Arg(0), "mongo.pem"), filepath.Join(flag.Arg(0), "key.pem"))
		if err != nil {
			log.Fatal("error loading mongo keys:", err)
		}
		mongoClient, err = mongo.Connect(context.Background(), options.Client().ApplyURI(uri).SetTLSConfig(&tls.Config{
			Certificates: []tls.Certificate{mongoCert},
		}))
		if err != nil {
			log.Fatal("error connecting to mongo:", err)
		}
	} else {
		var err error
		mongoClient, err = mongo.Connect(context.Background(), options.Client().ApplyURI(uri))
		if err != nil {
			log.Fatal("error connecting to mongo:", err)
		}
	}
}

func setupBackend(mux *http.ServeMux) {
	blogApp = blog.NewBlogApp(mongoClient.Database("blog"))
	var err error
	back, err = backend.NewBackend(db.NewMongoTable[backend.ApiKey](
		mongoClient.Database("darkstorm").Collection("keys")),
		blogApp,
		swassistant.NewSWBackend(mongoClient.Database("swassistant")),
		cdr.NewBackend(mongoClient.Database("cdr")),
	)
	if !*testing {
		back.AddCorsAddress("https://darkstorm.tech")
	} else {
		back.AddCorsAddress("*")
	}
	if err != nil {
		log.Fatal("error setting up backend:", err)
	}
	if !*testing {
		mux.Handle("api.darkstorm.tech/", back)
	} else {
		go func() {
			http.ListenAndServe(":2323", back)
		}()
	}
}

func setupWebsite(mux *http.ServeMux) {
	if !*testing {
		url, _ := url.Parse("https://localhost:30000")
		mux.Handle("rpg.darkstorm.tech/", httputil.NewSingleHostReverseProxy(url))
	}
	mux.HandleFunc("GET /files/{w...}", filesRequest)
	mux.HandleFunc("GET /portfolio", portfolioRequest)
	mux.HandleFunc("GET /list", blogListHandle)
	mux.HandleFunc("/", mainHandle)
}

func mainHandle(w http.ResponseWriter, r *http.Request) {
	path := strings.TrimPrefix(path.Clean(r.URL.Path), "/")
	if path == "/" || path == "" {
		latestBlogsHandle(w, r)
		return
	}
	if path == "login" {
		sendContent(w, r, loginScreen, "", "")
		return
	}
	stat, err := os.Stat(filepath.Join(*webRoot, path))
	if err == nil && !stat.IsDir() {
		http.ServeFile(w, r, filepath.Join(*webRoot, path))
		return
	}
	spl := strings.Split(path, "/")
	ind := filepath.Join(*webRoot, spl[0], "index.html")
	_, err = os.Stat(ind)
	if err == nil {
		http.ServeFile(w, r, ind)
		return
	}
	blogHandle(w, r, path)
}

const loginScreen = `
<script src="https://unpkg.com/htmx-ext-json-enc@2.0.1/json-enc.js"></script>
<form id="loginForm" hx-ext='json-enc' hx-post="https://api.darkstorm.tech/users/login">
	<label for="username">Username:</label>
	<input name="username" id="usernameInput"></input>
	<label for="password">Password:</label>
	<input name="password" type="password" id="passwordInput"></input>
	<button id="loginButton" type="submit">Login</button>
</form>
`