NetworkTool/server/cmd/networktool-server/main.go

package main

import (
	"context"
	"os"
	"os/signal"
	"syscall"

	"networktool/internal/config"
	"networktool/internal/deviceinfo"
	"networktool/internal/discovery"
	"networktool/internal/httpserver"
	"networktool/internal/logger"
	applyexecsvc "networktool/internal/network/applyexec"
	configreadersvc "networktool/internal/network/configreader"
	interfacesvc "networktool/internal/network/interfaces"
	netplansvc "networktool/internal/network/netplan"
	validatorsvc "networktool/internal/network/validator"
	"networktool/internal/systemaction"
	"networktool/internal/tasks"
)

func main() {
	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
	defer stop()

	log := logger.New()
	cfg := config.Load(os.Args[1:])
	log.Info("server starting", "version", cfg.ServerVersion)
	if os.Geteuid() != 0 {
		log.Warn("server is not running as root; netplan write/apply and system actions will fail")
	}
	deviceSvc := deviceinfo.New(cfg)
	interfaceSvc := interfacesvc.New(cfg)
	configSvc := configreadersvc.New()
	validatorSvc := validatorsvc.New()
	netplanSvc := netplansvc.New()
	applySvc := applyexecsvc.New()
	taskSvc := tasks.New()
	systemSvc := systemaction.New()

	if err := interfaceSvc.EnsureMaintenanceAddress(); err != nil {
		log.Error("failed to ensure maintenance address", "error", err.Error())
		return
	}

	httpSrv := httpserver.New(cfg, log, deviceSvc, interfaceSvc, configSvc, validatorSvc, netplanSvc, applySvc, taskSvc, systemSvc)
	udpSrv := discovery.New(cfg, log, deviceSvc)

	errCh := make(chan error, 2)
	go func() { errCh <- httpSrv.Run(ctx) }()
	go func() { errCh <- udpSrv.Run(ctx) }()

	select {
	case <-ctx.Done():
		log.Info("server shutting down")
	case err := <-errCh:
		if err != nil {
			log.Error("server stopped with error", "error", err.Error())
		}
	}
}