mecha10_runtime/

lifecycle.rs

//! Lifecycle management for mode-based dynamic node control
//!
//! This module provides the lifecycle manager that orchestrates node
//! start/stop based on operational mode changes.
//!
//! # Architecture
//!
//! ```text
//! CLI → Redis (mode/request) → LifecycleManager → Supervisor → Nodes
//! ```
//!
//! The lifecycle manager:
//! - Subscribes to mode change requests
//! - Calculates which nodes need to start/stop
//! - Uses Supervisor to execute changes
//! - Publishes mode changed events
//!
//! # Example
//!
//! ```rust,ignore
//! use mecha10_runtime::lifecycle::LifecycleManager;
//!
//! let lifecycle = LifecycleManager::new(
//!     context,
//!     supervisor,
//!     mode_config,
//!     "startup".to_string(),
//! );
//!
//! // Run in background
//! tokio::spawn(async move {
//!     lifecycle.run().await
//! });
//! ```

use anyhow::Result;
use std::collections::{HashMap, HashSet};
use std::sync::Arc;
use tracing::{debug, error, info, warn};

// Note: mecha10-core types will be imported once we add the dependency
// For now, we'll define placeholder types that will be replaced

/// Lifecycle manager for mode-based node orchestration
///
/// Manages transitions between operational modes by starting and stopping
/// nodes via the Supervisor.
pub struct LifecycleManager {
    /// Current operational mode
    current_mode: String,

    /// Set of currently running node names
    pub(crate) running_nodes: HashSet<String>,

    /// Mode configurations (mode_name -> ModeConfig)
    mode_config: HashMap<String, ModeConfig>,

    /// Supervisor for node management
    supervisor: Arc<dyn SupervisorTrait>,
}

/// Configuration for a single mode
///
/// This is a simplified version that will be replaced with the actual
/// type from mecha10-cli once we integrate.
#[derive(Debug, Clone)]
pub struct ModeConfig {
    /// Nodes that should run in this mode
    pub nodes: Vec<String>,

    /// Nodes that should be stopped when entering this mode
    pub stop_nodes: Vec<String>,

    /// Description of this mode
    pub description: Option<String>,
}

/// Trait for supervisor operations
///
/// This allows us to mock the supervisor for testing while
/// using the real Supervisor in production.
#[async_trait::async_trait]
pub trait SupervisorTrait: Send + Sync {
    /// Start a node by name
    async fn start_node(&self, name: &str) -> Result<()>;

    /// Stop a node by name
    async fn stop_node(&self, name: &str) -> Result<()>;

    /// Get list of running nodes
    fn get_running_nodes(&self) -> Vec<String>;
}

impl LifecycleManager {
    /// Create a new lifecycle manager
    ///
    /// # Arguments
    ///
    /// * `supervisor` - Supervisor instance for node management
    /// * `mode_config` - Map of mode name to mode configuration
    /// * `default_mode` - Initial mode to start in
    pub fn new(
        supervisor: Arc<dyn SupervisorTrait>,
        mode_config: HashMap<String, ModeConfig>,
        default_mode: String,
    ) -> Self {
        Self {
            current_mode: default_mode,
            running_nodes: HashSet::new(),
            mode_config,
            supervisor,
        }
    }

    /// Get the current mode
    pub fn current_mode(&self) -> &str {
        &self.current_mode
    }

    /// Get running nodes
    pub fn running_nodes(&self) -> &HashSet<String> {
        &self.running_nodes
    }

    /// Request a mode change
    ///
    /// This is the main entry point for mode transitions.
    /// It calculates the diff and executes the transition.
    ///
    /// # Arguments
    ///
    /// * `target_mode` - The mode to transition to
    ///
    /// # Returns
    ///
    /// Ok with nodes started/stopped, or Err if mode doesn't exist
    pub async fn change_mode(&mut self, target_mode: &str) -> Result<ModeTransitionResult> {
        info!("Mode change requested: {} -> {}", self.current_mode, target_mode);

        // Validate mode exists
        if !self.mode_config.contains_key(target_mode) {
            let available: Vec<_> = self.mode_config.keys().map(|k| k.as_str()).collect();
            return Err(anyhow::anyhow!(
                "Mode '{}' not found. Available modes: {}",
                target_mode,
                available.join(", ")
            ));
        }

        // Calculate diff
        let diff = self.calculate_mode_diff(target_mode)?;

        info!("Mode transition plan: stop {:?}, start {:?}", diff.stop, diff.start);

        // Execute transition
        self.execute_mode_transition(target_mode, diff).await
    }

    /// Calculate which nodes need to start/stop for a mode transition
    pub fn calculate_mode_diff(&self, target_mode: &str) -> Result<ModeDiff> {
        let target_config = self
            .mode_config
            .get(target_mode)
            .ok_or_else(|| anyhow::anyhow!("Mode not found"))?;

        // Nodes that should run in target mode
        let target_nodes: HashSet<_> = target_config.nodes.iter().map(|s| s.as_str()).collect();

        // Nodes explicitly marked to stop
        let stop_nodes: HashSet<_> = target_config.stop_nodes.iter().map(|s| s.as_str()).collect();

        // Nodes to start: in target but not currently running
        let start: Vec<_> = target_nodes
            .iter()
            .filter(|n| !self.running_nodes.contains(**n))
            .map(|s| s.to_string())
            .collect();

        // Nodes to stop: running but not in target, or explicitly in stop_nodes
        let stop: Vec<_> = self
            .running_nodes
            .iter()
            .filter(|n| !target_nodes.contains(n.as_str()) || stop_nodes.contains(n.as_str()))
            .cloned()
            .collect();

        Ok(ModeDiff { start, stop })
    }

    /// Execute a mode transition
    async fn execute_mode_transition(&mut self, target_mode: &str, diff: ModeDiff) -> Result<ModeTransitionResult> {
        debug!(
            "Executing mode transition: {} -> {} (stop: {:?}, start: {:?})",
            self.current_mode, target_mode, diff.stop, diff.start
        );

        let mut stopped = Vec::new();
        let mut started = Vec::new();
        let mut errors = Vec::new();

        // Stop nodes first
        for node in &diff.stop {
            info!("Stopping node: {}", node);
            match self.supervisor.stop_node(node).await {
                Ok(()) => {
                    self.running_nodes.remove(node);
                    stopped.push(node.clone());
                    debug!("✓ Stopped: {}", node);
                }
                Err(e) => {
                    warn!("Failed to stop node {}: {}", node, e);
                    errors.push(format!("stop {}: {}", node, e));
                }
            }
        }

        // Start new nodes
        for node in &diff.start {
            info!("Starting node: {}", node);
            match self.supervisor.start_node(node).await {
                Ok(()) => {
                    self.running_nodes.insert(node.clone());
                    started.push(node.clone());
                    debug!("✓ Started: {}", node);
                }
                Err(e) => {
                    error!("Failed to start node {}: {}", node, e);
                    errors.push(format!("start {}: {}", node, e));
                }
            }
        }

        // Update current mode
        let previous_mode = self.current_mode.clone();
        self.current_mode = target_mode.to_string();

        info!(
            "Mode transition complete: {} -> {} (started: {}, stopped: {})",
            previous_mode,
            target_mode,
            started.len(),
            stopped.len()
        );

        Ok(ModeTransitionResult {
            previous_mode,
            current_mode: target_mode.to_string(),
            nodes_started: started,
            nodes_stopped: stopped,
            errors,
        })
    }
}

/// Difference between current and target mode
#[derive(Debug)]
pub struct ModeDiff {
    /// Nodes that need to be started
    pub start: Vec<String>,

    /// Nodes that need to be stopped
    pub stop: Vec<String>,
}

/// Result of a mode transition
#[derive(Debug, Clone)]
pub struct ModeTransitionResult {
    /// Previous mode
    pub previous_mode: String,

    /// New current mode
    pub current_mode: String,

    /// Nodes that were started
    pub nodes_started: Vec<String>,

    /// Nodes that were stopped
    pub nodes_stopped: Vec<String>,

    /// Errors encountered (non-fatal)
    pub errors: Vec<String>,
}