trying embedded caddy reverse proxy

2026-09-25 08:09:07 +02:00 · 2026-01-14 17:16:42 +01:00
parent d9118eb239
commit 626e892e3b
30 changed files with 6978 additions and 0 deletions
@@ -0,0 +1,120 @@
+package cmd
+
+import (
+	"context"
+	"os"
+	"os/signal"
+	"syscall"
+
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+
+	"github.com/netbirdio/netbird/proxy/pkg/proxy"
+	"github.com/netbirdio/netbird/proxy/pkg/version"
+)
+
+var (
+	configFile string
+	rootCmd    = &cobra.Command{
+		Use:           "proxy",
+		Short:         "Netbird Reverse Proxy Server",
+		Long:          "A lightweight, configurable reverse proxy server.",
+		SilenceUsage:  true,
+		SilenceErrors: true,
+		RunE:          run,
+	}
+)
+
+func init() {
+	rootCmd.PersistentFlags().StringVarP(&configFile, "config", "c", "", "path to JSON configuration file (optional, can use env vars instead)")
+
+	// Set version information
+	rootCmd.Version = version.Short()
+	rootCmd.SetVersionTemplate("{{.Version}}\n")
+}
+
+// Execute runs the root command
+func Execute() error {
+	return rootCmd.Execute()
+}
+
+func run(cmd *cobra.Command, args []string) error {
+	// Load configuration from file or environment variables
+	config, err := proxy.LoadFromFileOrEnv(configFile)
+	if err != nil {
+		log.Fatalf("Failed to load configuration: %v", err)
+		return err
+	}
+
+	// Set log level
+	setupLogging(config.LogLevel)
+
+	log.Infof("Starting Netbird Proxy - %s", version.Short())
+	log.Debugf("Full version info: %s", version.String())
+	log.Info("Configuration loaded successfully")
+	log.Infof("Listen Address: %s", config.ListenAddress)
+	log.Infof("Log Level: %s", config.LogLevel)
+
+	// Create server instance
+	server, err := proxy.NewServer(config)
+	if err != nil {
+		log.Fatalf("Failed to create server: %v", err)
+		return err
+	}
+
+	// Start server in a goroutine
+	serverErrors := make(chan error, 1)
+	go func() {
+		if err := server.Start(); err != nil {
+			serverErrors <- err
+		}
+	}()
+
+	// Set up signal handler for graceful shutdown
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+
+	// Wait for either an error or shutdown signal
+	select {
+	case err := <-serverErrors:
+		log.Fatalf("Server error: %v", err)
+		return err
+	case sig := <-quit:
+		log.Infof("Received signal: %v", sig)
+	}
+
+	// Create shutdown context with timeout
+	ctx, cancel := context.WithTimeout(context.Background(), config.ShutdownTimeout)
+	defer cancel()
+
+	// Gracefully stop the server
+	if err := server.Stop(ctx); err != nil {
+		log.Fatalf("Failed to stop server gracefully: %v", err)
+		return err
+	}
+
+	log.Info("Server exited successfully")
+	return nil
+}
+
+func setupLogging(level string) {
+	// Set log format
+	log.SetFormatter(&log.TextFormatter{
+		FullTimestamp:   true,
+		TimestampFormat: "2006-01-02 15:04:05",
+	})
+
+	// Set log level
+	switch level {
+	case "debug":
+		log.SetLevel(log.DebugLevel)
+	case "info":
+		log.SetLevel(log.InfoLevel)
+	case "warn":
+		log.SetLevel(log.WarnLevel)
+	case "error":
+		log.SetLevel(log.ErrorLevel)
+	default:
+		log.SetLevel(log.InfoLevel)
+	}
+}