X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/8fa166777cbcd92e871e937d9557f0f1a732c04d..d9bc63a10d0978d9a5222cf7be74044e55f409b7:/cmd/mobius-hotline-server/main.go

diff --git a/cmd/mobius-hotline-server/main.go b/cmd/mobius-hotline-server/main.go
index 1e9b5b6..878b425 100644
--- a/cmd/mobius-hotline-server/main.go
+++ b/cmd/mobius-hotline-server/main.go
@@ -7,14 +7,17 @@ import (
 	"flag"
 	"fmt"
 	"github.com/jhalter/mobius/hotline"
+	"github.com/jhalter/mobius/internal/mobius"
 	"gopkg.in/natefinch/lumberjack.v2"
 	"io"
 	"log"
 	"log/slog"
 	"net/http"
 	"os"
+	"os/signal"
 	"path"
 	"runtime"
+	"syscall"
 )
 
 //go:embed mobius/config
@@ -36,28 +39,16 @@ var (
 )
 
 func main() {
-	ctx, _ := context.WithCancel(context.Background())
-
-	// TODO: implement graceful shutdown by closing context
-	//c := make(chan os.Signal, 1)
-	//signal.Notify(c, os.Interrupt)
-	//defer func() {
-	//	signal.Stop(c)
-	//	cancel()
-	//}()
-	//go func() {
-	//	select {
-	//	case <-c:
-	//		cancel()
-	//	case <-ctx.Done():
-	//	}
-	//}()
+	ctx, cancel := context.WithCancel(context.Background())
+
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGINT, os.Interrupt)
 
 	netInterface := flag.String("interface", "", "IP addr of interface to listen on.  Defaults to all interfaces.")
 	basePort := flag.Int("bind", defaultPort, "Base Hotline server port.  File transfer port is base port + 1.")
 	statsPort := flag.String("stats-port", "", "Enable stats HTTP endpoint on address and port")
 	configDir := flag.String("config", defaultConfigPath(), "Path to config root")
-	printVersion := flag.Bool("version", false, "print version and exit")
+	printVersion := flag.Bool("version", false, "Print version and exit")
 	logLevel := flag.String("log-level", "info", "Log level")
 	logFile := flag.String("log-file", "", "Path to log file")
 
@@ -105,12 +96,36 @@ func main() {
 		os.Exit(1)
 	}
 
-	srv, err := hotline.NewServer(*configDir, *netInterface, *basePort, slogger, &hotline.OSFileStore{})
+	config, err := mobius.LoadConfig(path.Join(*configDir, "config.yaml"))
+	if err != nil {
+		slogger.Error(fmt.Sprintf("Error loading config: %v", err))
+		os.Exit(1)
+	}
+
+	srv, err := hotline.NewServer(*config, *configDir, *netInterface, *basePort, slogger, &hotline.OSFileStore{})
 	if err != nil {
 		slogger.Error(fmt.Sprintf("Error starting server: %s", err))
 		os.Exit(1)
 	}
 
+	srv.MessageBoard, err = mobius.NewFlatNews(path.Join(*configDir, "MessageBoard.txt"))
+	if err != nil {
+		slogger.Error(fmt.Sprintf("Error loading message board: %v", err))
+		os.Exit(1)
+	}
+
+	srv.BanList, err = mobius.NewBanFile(path.Join(*configDir, "Banlist.yaml"))
+	if err != nil {
+		slogger.Error(fmt.Sprintf("Error loading ban list: %v", err))
+		os.Exit(1)
+	}
+
+	srv.ThreadedNewsMgr, err = mobius.NewThreadedNewsYAML(path.Join(*configDir, "ThreadedNews.yaml"))
+	if err != nil {
+		slogger.Error(fmt.Sprintf("Error loading news: %v", err))
+		os.Exit(1)
+	}
+
 	sh := statHandler{hlServer: srv}
 	if *statsPort != "" {
 		http.HandleFunc("/", sh.RenderStats)
@@ -124,6 +139,33 @@ func main() {
 		}(srv)
 	}
 
+	go func() {
+		for {
+			sig := <-sigChan
+			switch sig {
+			case syscall.SIGHUP:
+				slogger.Info("SIGHUP received.  Reloading configuration.")
+
+				if err := srv.MessageBoard.(*mobius.FlatNews).Reload(); err != nil {
+					slogger.Error("Error reloading news", "err", err)
+				}
+
+				if err := srv.BanList.(*mobius.BanFile).Load(); err != nil {
+					slogger.Error("Error reloading ban list", "err", err)
+				}
+
+				if err := srv.ThreadedNewsMgr.(*mobius.ThreadedNewsYAML).Load(); err != nil {
+					slogger.Error("Error reloading threaded news list", "err", err)
+				}
+			default:
+				signal.Stop(sigChan)
+				cancel()
+				os.Exit(0)
+			}
+
+		}
+	}()
+
 	slogger.Info("Hotline server started",
 		"version", version,
 		"API port", fmt.Sprintf("%s:%v", *netInterface, *basePort),
@@ -168,24 +210,6 @@ func defaultConfigPath() string {
 	return cfgPath
 }
 
-// copyFile copies a file from src to dst. If dst does not exist, it is created.
-func copyFile(src, dst string) error {
-	sourceFile, err := os.Open(src)
-	if err != nil {
-		return err
-	}
-	defer sourceFile.Close()
-
-	destinationFile, err := os.Create(dst)
-	if err != nil {
-		return err
-	}
-	defer destinationFile.Close()
-
-	_, err = io.Copy(destinationFile, sourceFile)
-	return err
-}
-
 // copyDir recursively copies a directory tree, attempting to preserve permissions.
 func copyDir(src, dst string) error {
 	entries, err := cfgTemplate.ReadDir(src)