Complete rewrite and re-architecture Osmedeus Engine in v5

2026-09-23 01:54:58 +02:00 · 2026-01-18 19:32:24 +08:00
commit 7a2c5a5dc9
743 changed files with 99767 additions and 0 deletions
@@ -0,0 +1,166 @@
+package cli
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/j3ssie/osmedeus/v5/internal/config"
+	"github.com/j3ssie/osmedeus/v5/internal/distributed"
+	"github.com/j3ssie/osmedeus/v5/internal/logger"
+	"github.com/j3ssie/osmedeus/v5/pkg/server"
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+var (
+	serverHost    string
+	serverPort    int
+	noAuth        bool
+	masterMode    bool
+	redisURLServe string
+)
+
+// serveCmd represents the serve command
+var serveCmd = &cobra.Command{
+	Use:     "serve",
+	Aliases: []string{"server"},
+	Short:   "Start the Osmedeus web server",
+	Long:    UsageServe(),
+	RunE:    runServer,
+}
+
+func init() {
+	serveCmd.Flags().StringVar(&serverHost, "host", "", "host to bind (default from config)")
+	serveCmd.Flags().IntVar(&serverPort, "port", 0, "port to listen on (default from config)")
+	serveCmd.Flags().BoolVarP(&noAuth, "no-auth", "A", false, "disable all authentication")
+	serveCmd.Flags().BoolVar(&masterMode, "master", false, "run as distributed master node (requires Redis)")
+	serveCmd.Flags().StringVar(&redisURLServe, "redis-url", "", "Redis connection URL for master mode (overrides settings)")
+}
+
+func runServer(cmd *cobra.Command, args []string) error {
+	log := logger.Get()
+	cfg := config.Get()
+
+	if cfg == nil {
+		return fmt.Errorf("configuration not loaded")
+	}
+
+	// Override with flags if provided
+	if serverHost != "" {
+		cfg.Server.Host = serverHost
+	}
+	if serverPort != 0 {
+		cfg.Server.Port = serverPort
+	}
+
+	// Show critical warning if running without authentication
+	if noAuth {
+		fmt.Println("\033[31m\033[1m")
+		fmt.Println("╔════════════════════════════════════════════════════════════════╗")
+		fmt.Println("║  ⚠️  CRITICAL WARNING: Server running WITHOUT authentication   ║")
+		fmt.Println("║  Anyone can access all API endpoints without credentials!      ║")
+		fmt.Println("║  Only use -A flag for development/testing purposes.            ║")
+		fmt.Println("╚════════════════════════════════════════════════════════════════╝")
+		fmt.Println("\033[0m")
+	}
+
+	// Handle shutdown signals
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
+
+	go func() {
+		<-sigChan
+		log.Info("Shutting down...")
+		cancel()
+	}()
+
+	// Start master node if --master flag is set
+	var master *distributed.Master
+	if masterMode {
+		// Override Redis config if --redis-url provided
+		if redisURLServe != "" {
+			redisCfg, err := distributed.ParseRedisURL(redisURLServe)
+			if err != nil {
+				return fmt.Errorf("invalid redis URL: %w", err)
+			}
+			cfg.Redis = *redisCfg
+		}
+
+		if !cfg.IsRedisConfigured() {
+			return fmt.Errorf("redis not configured. Add redis section to osm-settings.yaml or use --redis-url")
+		}
+
+		var err error
+		master, err = distributed.NewMaster(cfg)
+		if err != nil {
+			return fmt.Errorf("failed to create master: %w", err)
+		}
+
+		// Start master in background
+		go func() {
+			if err := master.Start(ctx); err != nil {
+				log.Error("Master error", zap.Error(err))
+			}
+		}()
+
+		log.Info("Started distributed master node")
+	}
+
+	// Create server with master reference for distributed endpoints
+	opts := &server.Options{
+		NoAuth: noAuth,
+		Master: master,
+		Debug:  debug,
+	}
+	srv, err := server.New(cfg, opts)
+	if err != nil {
+		log.Error("Failed to create server", zap.Error(err))
+		return err
+	}
+
+	// Log debug mode status
+	if debug {
+		log.Info("Debug mode enabled - request bodies and detailed errors will be logged")
+	}
+
+	// Start server in goroutine
+	serverErr := make(chan error, 1)
+	go func() {
+		addr := fmt.Sprintf("%s:%d", cfg.Server.Host, cfg.Server.Port)
+		log.Info("Starting Osmedeus server", zap.String("address", addr))
+
+		// Log swagger access URL
+		swaggerURL := fmt.Sprintf("http://%s:%d/swagger/", cfg.Server.Host, cfg.Server.Port)
+		log.Info("Access Swagger documentation", zap.String("url", swaggerURL))
+
+		serverErr <- srv.Start(addr)
+	}()
+
+	// Wait for shutdown or server error
+	select {
+	case <-ctx.Done():
+		log.Info("Shutting down server...")
+
+		// Give server 5 seconds to shutdown gracefully
+		shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer shutdownCancel()
+
+		if err := srv.ShutdownWithContext(shutdownCtx); err != nil {
+			log.Error("Error during shutdown", zap.Error(err))
+		}
+		log.Info("Server stopped")
+	case err := <-serverErr:
+		if err != nil {
+			return fmt.Errorf("server error: %w", err)
+		}
+	}
+
+	return nil
+}