59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log/slog"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
"tslink/core"
|
|
)
|
|
|
|
func main() {
|
|
useJsonFormatLogger := flag.Bool("json-format", false, "use json format logger")
|
|
logLevel := flag.String("level", "info", "log level (DEBUG|INFO|WARN|ERROR)")
|
|
configPath := flag.String("c", "config.toml", "path to config file")
|
|
flag.Parse()
|
|
|
|
logger := core.NewLogger(*logLevel, *useJsonFormatLogger)
|
|
|
|
logger.Info("Starting tslink server", "level", *logLevel, "configPath", *configPath)
|
|
cfg, err := core.LoadConfig(*configPath)
|
|
if err != nil {
|
|
logger.With(
|
|
slog.String("error", err.Error()),
|
|
).Error("Error loading config")
|
|
os.Exit(1)
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
logger.Info("initializing tsnet server")
|
|
srv, err := core.InitTsNet(ctx, &cfg.Core, logger)
|
|
if err != nil {
|
|
logger.With(
|
|
slog.String("error", err.Error())).Error("Error initializing tsnet")
|
|
os.Exit(1)
|
|
}
|
|
logger.Info("tsnet server initialized")
|
|
|
|
core.StartForwarders(ctx, srv, cfg.Forward)
|
|
core.StartConnectors(ctx, srv, cfg.Connect)
|
|
|
|
core.RunLanDiscoverService(ctx, cfg.Connect, logger)
|
|
|
|
sigHandler := make(chan os.Signal, 1)
|
|
signal.Notify(sigHandler, os.Interrupt, syscall.SIGTERM)
|
|
|
|
for {
|
|
select {
|
|
case <-sigHandler:
|
|
logger.Warn("Shutting down")
|
|
cancel()
|
|
time.Sleep(time.Second)
|
|
return
|
|
}
|
|
}
|
|
}
|