zenno, cmd/horsebot: host website in horsebot

cmd/horsebot/main.go

@@ -3,11 +3,14 @@ package main
 import (
 	"bytes"
 	"context"
+	"encoding/hex"
 	"encoding/json"
 	"errors"
 	"flag"
 	"fmt"
 	"log/slog"
+	"net"
+	"net/http"
 	"os"
 	"os/signal"
 	"path/filepath"
@@ -26,20 +29,23 @@ import (
 
 func main() {
 	var (
-		dataDir   string
+		// public site
+		addr    string
+		dataDir string
+		public  string
+		// discord
 		tokenFile string
-		// http api options
-		addr   string
-		route  string
-		pubkey string
-		// logging options
+		apiRoute  string
+		pubkey    string
+		// logging
 		level   slog.Level
 		textfmt string
 	)
+	flag.StringVar(&addr, "http", ":80", "`address` to bind HTTP server")
 	flag.StringVar(&dataDir, "data", "", "`dir`ectory containing exported json data")
+	flag.StringVar(&public, "public", "", "`dir`ectory containing the website to serve")
 	flag.StringVar(&tokenFile, "token", "", "`file` containing the Discord bot token")
-	flag.StringVar(&addr, "http", "", "`address` to bind HTTP API server")
-	flag.StringVar(&route, "route", "/interactions/callback", "`path` to serve HTTP API calls")
+	flag.StringVar(&apiRoute, "route", "/interactions/callback", "`path` to serve Discord HTTP API calls")
 	flag.StringVar(&pubkey, "key", "", "Discord public key")
 	flag.TextVar(&level, "log", slog.LevelInfo, "slog logging `level`")
 	flag.StringVar(&textfmt, "log-format", "text", "slog logging `format`, text or json")
@@ -57,6 +63,16 @@ func main() {
 	}
 	slog.SetDefault(slog.New(lh))
 
+	stat, err := os.Stat(public)
+	if err != nil {
+		slog.Error("public", slog.Any("err", err))
+		os.Exit(1)
+	}
+	if !stat.IsDir() {
+		slog.Error("public", slog.String("err", "not a directory"))
+		os.Exit(1)
+	}
+
 	skills, err := loadSkills(filepath.Join(dataDir, "skill.json"))
 	slog.Info("loaded skills", slog.Int("count", len(skills)))
 	groups, err2 := loadSkillGroups(filepath.Join(dataDir, "skill-group.json"))
@@ -87,33 +103,52 @@ func main() {
 		r.SelectMenuComponent("/swap", skillSrv.menu)
 		r.ButtonComponent("/swap/{id}", skillSrv.button)
 	})
-
-	opts := []bot.ConfigOpt{bot.WithDefaultGateway(), bot.WithEventListeners(r)}
-	if addr != "" {
-		if pubkey == "" {
-			slog.Error("Discord public key must be provided when using HTTP API")
-			os.Exit(1)
-		}
-		opts = append(opts, bot.WithHTTPServerConfigOpts(pubkey,
-			httpserver.WithAddress(addr),
-			httpserver.WithURL(route),
-		))
-	}
-
 	slog.Info("connect", slog.String("disgo", disgo.Version))
-	client, err := disgo.New(string(token), opts...)
+	client, err := disgo.New(string(token), bot.WithDefaultGateway(), bot.WithEventListeners(r))
 	if err != nil {
 		slog.Error("building bot", slog.Any("err", err))
 		os.Exit(1)
 	}
 
+	mux := http.NewServeMux()
+	mux.Handle("GET /", http.FileServerFS(os.DirFS(public)))
+	if pubkey != "" {
+		pk, err := hex.DecodeString(pubkey)
+		if err != nil {
+			slog.Error("pubkey", slog.Any("err", err))
+			os.Exit(1)
+		}
+		mux.Handle(apiRoute, httpserver.HandleInteraction(httpserver.DefaultVerifier{}, pk, slog.Default(), client.EventManager.HandleHTTPEvent))
+		slog.Info("Discord HTTP API enabled", slog.String("pubkey", pubkey))
+	}
+	l, err := net.Listen("tcp", addr)
+	if err != nil {
+		slog.Error("listen", slog.String("addr", addr), slog.Any("err", err))
+		os.Exit(1)
+	}
+	srv := http.Server{
+		Addr:         addr,
+		Handler:      mux,
+		ReadTimeout:  15 * time.Second,
+		WriteTimeout: 15 * time.Second,
+		BaseContext:  func(l net.Listener) context.Context { return ctx },
+	}
+	go func() {
+		slog.Info("HTTP", slog.Any("addr", l.Addr()))
+		err := srv.Serve(l)
+		if err == http.ErrServerClosed {
+			return
+		}
+		slog.Error("HTTP server closed", slog.Any("err", err))
+	}()
+
 	if err := handler.SyncCommands(client, commands, nil, rest.WithCtx(ctx)); err != nil {
 		slog.Error("syncing commands", slog.Any("err", err))
 		os.Exit(1)
 	}
 
-	if addr != "" {
-		slog.Info("start HTTP server", slog.String("address", addr), slog.String("route", route))
+	if pubkey != "" {
+		slog.Info("start HTTP server", slog.String("address", addr), slog.String("route", apiRoute))
 		if err := client.OpenHTTPServer(); err != nil {
 			slog.Error("starting HTTP server", slog.Any("err", err))
 			stop()
@@ -131,6 +166,10 @@ func main() {
 	ctx, stop = context.WithTimeout(context.Background(), 5*time.Second)
 	defer stop()
 	client.Close(ctx)
+	if err := srv.Shutdown(ctx); err != nil {
+		slog.Error("HTTP API shutdown", slog.Any("err", err))
+		os.Exit(1)
+	}
 }
 
 var commands = []discord.ApplicationCommandCreate{