bzzz/internal/common/runtime/runtime.go

package runtime

import (
	"context"
	"fmt"
	"os"
	"time"

	"chorus.services/bzzz/logging"
	"chorus.services/bzzz/pkg/config"
	"chorus.services/bzzz/pkg/health"
)

// StandardRuntime implements the Runtime interface
type StandardRuntime struct {
	services *RuntimeServices
	logger   logging.Logger
	config   RuntimeConfig
}

// NewRuntime creates a new runtime instance
func NewRuntime(logger logging.Logger) Runtime {
	return &StandardRuntime{
		logger: logger,
	}
}

// Initialize sets up all runtime services according to the configuration
func (r *StandardRuntime) Initialize(ctx context.Context, cfg RuntimeConfig) (*RuntimeServices, error) {
	r.config = cfg
	r.logger.Info("🚀 Initializing BZZZ runtime (%s mode)", cfg.BinaryType.String())

	services := &RuntimeServices{
		Logger: r.logger,
	}

	// Phase 1: Configuration loading and validation
	if err := r.initializeConfig(cfg.ConfigPath, services); err != nil {
		return nil, NewRuntimeError(ErrConfigInvalid, "config", cfg.BinaryType, 
			fmt.Sprintf("config initialization failed: %v", err), err)
	}
	r.logger.Info("✅ Configuration loaded and validated")

	// Phase 2: P2P Infrastructure
	if err := r.initializeP2P(ctx, services); err != nil {
		return nil, NewRuntimeError(ErrP2PInitFailed, "p2p", cfg.BinaryType,
			fmt.Sprintf("P2P initialization failed: %v", err), err)
	}
	r.logger.Info("✅ P2P infrastructure initialized")

	// Phase 3: Core Services (PubSub, DHT, etc.)
	if err := r.initializeCoreServices(ctx, services); err != nil {
		return nil, NewRuntimeError(ErrServiceStartFailed, "core", cfg.BinaryType,
			fmt.Sprintf("core services initialization failed: %v", err), err)
	}
	r.logger.Info("✅ Core services initialized")

	// Phase 4: Binary-specific configuration
	if err := r.applyBinarySpecificConfig(cfg.BinaryType, services); err != nil {
		return nil, NewRuntimeError(ErrConfigInvalid, "binary-specific", cfg.BinaryType,
			fmt.Sprintf("binary-specific config failed: %v", err), err)
	}
	r.logger.Info("✅ Binary-specific configuration applied")

	// Phase 5: Health and Monitoring
	if err := r.initializeMonitoring(services); err != nil {
		return nil, NewRuntimeError(ErrServiceStartFailed, "monitoring", cfg.BinaryType,
			fmt.Sprintf("monitoring initialization failed: %v", err), err)
	}
	r.logger.Info("✅ Health monitoring initialized")

	r.services = services
	r.logger.Info("🎉 Runtime initialization completed successfully")
	return services, nil
}

// Start begins all runtime services
func (r *StandardRuntime) Start(ctx context.Context, services *RuntimeServices) error {
	r.logger.Info("🚀 Starting BZZZ runtime services")

	// Start shutdown manager (begins listening for signals)
	services.ShutdownManager.Start()
	r.logger.Info("🛡️ Graceful shutdown manager started")

	// Start health manager
	if err := services.HealthManager.Start(); err != nil {
		return NewRuntimeError(ErrServiceStartFailed, "health", r.config.BinaryType,
			fmt.Sprintf("failed to start health manager: %v", err), err)
	}
	r.logger.Info("❤️ Health monitoring started")

	// Start health HTTP server
	healthPort := 8081
	if r.config.CustomPorts.HealthPort != 0 {
		healthPort = r.config.CustomPorts.HealthPort
	}
	
	if err := services.HealthManager.StartHTTPServer(healthPort); err != nil {
		r.logger.Warn("⚠️ Failed to start health HTTP server: %v", err)
	} else {
		r.logger.Info("🏥 Health endpoints available at http://localhost:%d/health", healthPort)
	}

	// Start HTTP API server
	httpPort := 8080
	if r.config.CustomPorts.HTTPPort != 0 {
		httpPort = r.config.CustomPorts.HTTPPort
	}
	
	go func() {
		if err := services.HTTPServer.Start(); err != nil {
			r.logger.Error("❌ HTTP server error: %v", err)
		}
	}()
	r.logger.Info("🌐 HTTP API server started on :%d", httpPort)

	// Start UCXI server if enabled
	if services.UCXIServer != nil {
		go func() {
			if err := services.UCXIServer.Start(); err != nil {
				r.logger.Error("❌ UCXI server error: %v", err)
			}
		}()
		ucxiPort := services.Config.UCXL.Server.Port
		if r.config.CustomPorts.UCXIPort != 0 {
			ucxiPort = r.config.CustomPorts.UCXIPort
		}
		r.logger.Info("🔗 UCXI server started on :%d", ucxiPort)
	}

	// Start task coordination
	if services.TaskCoordinator != nil {
		services.TaskCoordinator.Start()
		r.logger.Info("✅ Task coordination system active")
	}

	// Start election manager
	if services.ElectionManager != nil {
		if err := services.ElectionManager.Start(); err != nil {
			r.logger.Error("❌ Failed to start election manager: %v", err)
		} else {
			r.logger.Info("✅ Election manager started with automated heartbeat management")
		}
	}

	r.logger.Info("✅ All runtime services started successfully")
	return nil
}

// Stop gracefully shuts down all runtime services
func (r *StandardRuntime) Stop(ctx context.Context, services *RuntimeServices) error {
	r.logger.Info("🛑 Shutting down BZZZ runtime services")

	// Use the shutdown manager for graceful shutdown
	if services.ShutdownManager != nil {
		// The shutdown manager will handle the graceful shutdown of all registered components
		services.ShutdownManager.Wait()
		r.logger.Info("✅ Graceful shutdown completed")
	} else {
		// Fallback manual shutdown if shutdown manager is not available
		r.logger.Warn("⚠️ Shutdown manager not available, performing manual shutdown")
		r.manualShutdown(services)
	}

	return nil
}

// GetHealthStatus returns the current health status
func (r *StandardRuntime) GetHealthStatus() *health.Status {
	if r.services != nil && r.services.HealthManager != nil {
		status := r.services.HealthManager.GetOverallStatus()
		return &status
	}
	return &health.Status{
		Healthy:   false,
		Timestamp: time.Now(),
		Message:   "Runtime not initialized",
	}
}

// manualShutdown performs manual shutdown when shutdown manager is not available
func (r *StandardRuntime) manualShutdown(services *RuntimeServices) {
	// Stop services in reverse order of initialization
	
	if services.ElectionManager != nil {
		services.ElectionManager.Stop()
		r.logger.Info("🗳️ Election manager stopped")
	}

	if services.TaskCoordinator != nil {
		// TaskCoordinator.Stop() method needs to be implemented
		r.logger.Info("📋 Task coordinator stopped")
	}

	if services.UCXIServer != nil {
		services.UCXIServer.Stop()
		r.logger.Info("🔗 UCXI server stopped")
	}

	if services.HTTPServer != nil {
		services.HTTPServer.Stop()
		r.logger.Info("🌐 HTTP server stopped")
	}

	if services.HealthManager != nil {
		services.HealthManager.Stop()
		r.logger.Info("❤️ Health manager stopped")
	}

	if services.DHT != nil {
		services.DHT.Close()
		r.logger.Info("🕸️ DHT closed")
	}

	if services.PubSub != nil {
		services.PubSub.Close()
		r.logger.Info("📡 PubSub closed")
	}

	if services.MDNSDiscovery != nil {
		// MDNSDiscovery.Close() method needs to be called
		r.logger.Info("📡 mDNS discovery closed")
	}

	if services.Node != nil {
		services.Node.Close()
		r.logger.Info("🌐 P2P node closed")
	}
}