netris-nestri/packages/relay/main.go

package main

import (
	"context"
	"log"
	"log/slog"
	"os"
	"os/signal"
	"relay/internal"
	"relay/internal/common"
	"syscall"
)

func main() {
	// Setup main context and stopper
	mainCtx, mainStopper := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)

	// Get flags and log them
	common.InitFlags()
	common.GetFlags().DebugLog()

	logLevel := slog.LevelInfo
	if common.GetFlags().Verbose {
		logLevel = slog.LevelDebug
	}

	// Create the base handler with debug level
	baseHandler := slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
		Level: logLevel,
	})
	customHandler := &common.CustomHandler{Handler: baseHandler}
	logger := slog.New(customHandler)
	slog.SetDefault(logger)

	// Start relay
	err := internal.InitRelay(mainCtx, mainStopper, common.GetFlags().MeshPort)
	if err != nil {
		slog.Error("Failed to initialize relay", "err", err)
		mainStopper()
		return
	}

	// Wait for exit signal
	<-mainCtx.Done()
	log.Println("Shutting down gracefully by signal...")
}