初版功能完成

cmd/site/main.go

@@ -0,0 +1,78 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"os"
+	"os/signal"
+	"path/filepath"
+	"syscall"
+	"time"
+
+	"remlink/internal/appdir"
+	"remlink/internal/config"
+	"remlink/internal/identity"
+	"remlink/internal/logging"
+	"remlink/internal/model"
+	"remlink/internal/nodeagent"
+	"remlink/internal/version"
+)
+
+const joinTokenEnvironment = "REMLINK_JOIN_TOKEN"
+
+func main() {
+	if err := run(os.Args[1:]); err != nil {
+		fmt.Fprintf(os.Stderr, "remlink-site: %v\n", err)
+		os.Exit(1)
+	}
+}
+
+func run(arguments []string) error {
+	defaultConfigPath, err := appdir.Join("site.yaml")
+	if err != nil {
+		return err
+	}
+	flags := flag.NewFlagSet("remlink-site", flag.ContinueOnError)
+	configPath := flags.String("config", defaultConfigPath, "Site YAML configuration path")
+	joinToken := flags.String("join-token", os.Getenv(joinTokenEnvironment), "first-registration Join Token (CLI/environment override YAML)")
+	identityPath := flags.String("identity", "", "override DPAPI identity path")
+	if err := flags.Parse(arguments); err != nil {
+		return err
+	}
+	clientConfig, err := config.LoadSite(*configPath)
+	if err != nil {
+		return err
+	}
+	resolvedJoinToken := config.ResolveJoinToken(*joinToken, clientConfig.JoinToken)
+	if *identityPath == "" {
+		*identityPath, err = identity.DefaultPath(model.NodeTypeSite)
+		if err != nil {
+			return err
+		}
+	}
+	applicationLogger, err := logging.New(logging.DefaultConfig(filepath.Join(filepath.Dir(*identityPath), "logs", "site.jsonl")))
+	if err != nil {
+		return err
+	}
+	defer applicationLogger.Close()
+	logger, _ := applicationLogger.For(logging.ModuleCore)
+	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
+	defer stop()
+	logger.Info("RemLink Site 正在启动", "version", version.String())
+	console := newSiteConsole(os.Stdout)
+	console.Header(clientConfig.Server, clientConfig.NodeName, version.String())
+	return nodeagent.Run(ctx, nodeagent.Options{
+		NodeType: model.NodeTypeSite, NodeName: clientConfig.NodeName,
+		ServerURL: clientConfig.Server, JoinToken: resolvedJoinToken, IdentityPath: *identityPath,
+		Version: version.String(), Logger: logger, ApplicationLogger: applicationLogger,
+		TCPFlowLimit:   clientConfig.Netstack.TCPFlowLimit,
+		UDPFlowLimit:   clientConfig.Netstack.UDPFlowLimit,
+		UDPIdleTimeout: time.Duration(clientConfig.Netstack.UDPIdleSeconds) * time.Second,
+		OnOverlayReady: console.OverlayReady,
+		OnControlState: console.ControlState,
+		OnSiteReady:    console.SiteReady,
+		OnSession:      console.Session,
+		OnRoute:        console.Route,
+	})
+}