mecha10_templates/services/

init_service.rs

//! Init service for project initialization
//!
//! Handles business logic for creating new Mecha10 projects including:
//! - Directory structure creation
//! - Component addition

use crate::paths;
use anyhow::Result;
use std::path::Path;

/// Service for initializing new Mecha10 projects
///
/// # Examples
///
/// ```rust,ignore
/// use mecha10_cli::services::InitService;
///
/// let service = InitService::new();
///
/// // Create project structure
/// service.create_project_directories(&project_path).await?;
///
/// // Add example nodes
/// service.add_example_node(&project_path, "speaker").await?;
/// ```
pub struct InitService;

impl InitService {
    /// Create a new InitService
    pub fn new() -> Self {
        Self
    }

    /// Create project directory structure
    ///
    /// Creates all the necessary directories for a new Mecha10 project:
    /// - nodes/ - Custom node implementations
    /// - drivers/ - Hardware driver implementations
    /// - types/ - Shared type definitions
    /// - configs/behaviors/ - Behavior tree definitions (LAB-2072: nested under configs/ so
    ///   `mecha10 config push` syncs them)
    /// - config/ - Configuration files
    /// - logs/ - Runtime logs
    /// - models/ - ML/AI models (ONNX, etc.) for remote nodes
    /// - configs/simulation/ - Simulation environments
    /// - configs/simulation/models/ - Robot physical models
    /// - configs/rl/tasks/ - RL task-curriculum configs (LAB-2429 - not under configs/simulation/,
    ///   see `paths::project::RL_TASKS_DIR`'s doc comment)
    /// - configs/simulation/images/ - Billboard image assets for simulation environments/tasks
    pub async fn create_project_directories(&self, path: &Path) -> Result<()> {
        let dirs = vec![
            paths::project::NODES_DIR,
            paths::project::DRIVERS_DIR,
            paths::project::TYPES_DIR,
            paths::project::BEHAVIORS_DIR,
            paths::project::LOGS_DIR,
            paths::project::MODELS_DIR,
            paths::project::SIMULATION_DIR,
            paths::project::SIMULATION_MODELS_DIR,
            paths::project::SIMULATION_ENVIRONMENTS_DIR,
            paths::project::SIMULATION_IMAGES_DIR,
            paths::project::RL_TASKS_DIR,
            paths::config::NODES_DIR,
        ];

        for dir in dirs {
            let dir_path = path.join(dir);
            tokio::fs::create_dir_all(&dir_path).await?;

            tokio::fs::write(dir_path.join(".gitkeep"), "").await?;
        }

        Ok(())
    }

    /// Add an example node from the catalog to the project
    ///
    /// This method:
    /// 1. Searches the component catalog for the component
    /// 2. Generates component files from templates (if any)
    /// 3. Updates Cargo.toml (workspace members or dependencies)
    /// 4. Updates mecha10.json configuration
    ///
    /// # Arguments
    ///
    /// * `project_root` - Root path of the project
    /// * `component_name` - Name of the component to add (e.g., "speaker", "listener")
    pub async fn add_example_node(&self, project_root: &Path, component_name: &str) -> Result<()> {
        use crate::component_catalog::ComponentCatalog;

        let catalog = ComponentCatalog::new();
        let components = catalog.search(component_name);

        if components.is_empty() {
            return Err(anyhow::anyhow!("Component '{}' not found in catalog", component_name));
        }

        let component = &components[0];

        for file_template in &component.files {
            let file_path = project_root.join(&file_template.path);

            if let Some(parent) = file_path.parent() {
                tokio::fs::create_dir_all(parent).await?;
            }

            tokio::fs::write(&file_path, &file_template.content).await?;
        }

        if !component.files.is_empty() {
            self.add_workspace_member(project_root, component_name).await?;
        }

        let is_monorepo_node = component.files.is_empty();
        if is_monorepo_node {
            self.add_dependency(project_root, component).await?;
        }

        self.add_node_to_config(project_root, component_name, component, is_monorepo_node)
            .await?;

        Ok(())
    }

    /// Add a workspace member to Cargo.toml
    async fn add_workspace_member(&self, project_root: &Path, component_name: &str) -> Result<()> {
        let cargo_toml_path = project_root.join(paths::rust::CARGO_TOML);
        let content = tokio::fs::read_to_string(&cargo_toml_path).await?;

        let new_member = format!("nodes/{}", component_name);
        let updated_content = if content.contains("members = []") {
            content.replace("members = []", &format!("members = [\n    \"{}\",\n]", new_member))
        } else if let Some(start) = content.find("members = [") {
            if let Some(end) = content[start..].find(']') {
                let insert_pos = start + end;
                let members_section = &content[start..insert_pos];
                if members_section.contains('\"') {
                    format!(
                        "{}\n    \"{}\",{}",
                        &content[..insert_pos],
                        new_member,
                        &content[insert_pos..]
                    )
                } else {
                    format!(
                        "{}\n    \"{}\",\n{}",
                        &content[..insert_pos],
                        new_member,
                        &content[insert_pos..]
                    )
                }
            } else {
                content
            }
        } else {
            content
        };

        tokio::fs::write(&cargo_toml_path, updated_content).await?;
        Ok(())
    }

    /// Add a dependency to Cargo.toml
    async fn add_dependency(&self, project_root: &Path, component: &crate::component_catalog::Component) -> Result<()> {
        let cargo_toml_path = project_root.join(paths::rust::CARGO_TOML);
        let content = tokio::fs::read_to_string(&cargo_toml_path).await?;

        let package_name = if let Some(dep) = component.cargo_dependencies.first() {
            dep.name.clone()
        } else {
            format!("mecha10-nodes-{}", component.id)
        };

        if !content.contains(&package_name) {
            let updated_content = if let Some(pos) = content.find("mecha10-core = \"0.1\"") {
                if let Some(newline_pos) = content[pos..].find('\n') {
                    let insert_pos = pos + newline_pos + 1;
                    format!(
                        "{}{} = \"0.1\"\n{}",
                        &content[..insert_pos],
                        package_name,
                        &content[insert_pos..]
                    )
                } else {
                    content
                }
            } else {
                content
            };

            tokio::fs::write(&cargo_toml_path, updated_content).await?;
        }

        Ok(())
    }

    /// Add node entry to mecha10.json
    async fn add_node_to_config(
        &self,
        project_root: &Path,
        component_name: &str,
        component: &crate::component_catalog::Component,
        is_monorepo_node: bool,
    ) -> Result<()> {
        use crate::services::ConfigService;

        let config_path = project_root.join(paths::PROJECT_CONFIG);
        let mut config = ConfigService::load_from(&config_path).await?;

        let (node_identifier, version) = if is_monorepo_node {
            (
                format!("@mecha10/{}", component_name),
                env!("CARGO_PKG_VERSION").to_string(),
            )
        } else {
            (component_name.to_string(), "0.1.0".to_string())
        };

        config.nodes.add_node(&node_identifier, &version);

        for key in component.relevant_target_keys() {
            let entry = target_entry_mut(&mut config.targets, key);
            if !entry.nodes.contains(&node_identifier) {
                entry.nodes.push(node_identifier.clone());
            }
        }

        let config_json = serde_json::to_string_pretty(&config)?;
        tokio::fs::write(&config_path, config_json).await?;

        Ok(())
    }
}

/// Mutable access to the `targets.<key>` entry, creating it with an empty node list if absent.
fn target_entry_mut<'a>(
    targets: &'a mut mecha10_cli_core::types::project::TargetsConfig,
    key: &str,
) -> &'a mut mecha10_cli_core::types::project::TargetEntry {
    match key {
        "dev" => targets.dev.get_or_insert_with(Default::default),
        "simulation" => targets.simulation.get_or_insert_with(Default::default),
        "robot" => targets.robot.get_or_insert_with(Default::default),
        "remote" => targets.remote.get_or_insert_with(Default::default),
        _ => unreachable!("target_entry_mut only called with \"dev\"/\"simulation\"/\"robot\"/\"remote\""),
    }
}

impl Default for InitService {
    fn default() -> Self {
        Self::new()
    }
}

#[cfg(test)]
mod add_node_to_config_tests {
    use super::*;
    use crate::component_catalog::{Component, ComponentType, Mecha10Config};
    use mecha10_cli_core::services::ConfigService;

    /// Minimal valid mecha10.json content for a freshly-initialized project.
    fn minimal_project_config_json() -> &'static str {
        r#"{
            "name": "test-project",
            "version": "0.1.0",
            "robot_id": "test-robot"
        }"#
    }

    /// Build a placeholder `Component` sufficient for `add_node_to_config`. Plain (untagged)
    /// components have no `relevant_lifecycle_modes()`, so they exercise only the
    /// `nodes` bookkeeping, not the `targets.simulation.nodes` sync.
    fn placeholder_component(name: &str) -> Component {
        Component {
            id: name.to_string(),
            name: name.to_string(),
            description: String::new(),
            component_type: ComponentType::Package,
            version: "0.1.0".to_string(),
            category: "custom".to_string(),
            tags: Vec::new(),
            cargo_dependencies: Vec::new(),
            files: Vec::new(),
            mecha10_config: Mecha10Config {
                driver: None,
                custom_node: None,
                behavior: None,
            },
        }
    }

    /// Same as [`placeholder_component`] but tagged as an AI/perception node (e.g.
    /// `image-classifier`), so `relevant_lifecycle_modes()` returns `simulation` (see
    /// `Component::relevant_lifecycle_modes` in `mecha10-cli-core/src/component_catalog/types.rs`).
    fn ai_perception_component(name: &str) -> Component {
        Component {
            tags: vec!["vision".to_string()],
            ..placeholder_component(name)
        }
    }

    /// Same as [`placeholder_component`] but hardware-backed with `run_target: "robot"`, so
    /// `relevant_target()` returns `Some("robot")` (see `Component::relevant_target`).
    fn robot_hardware_component(name: &str) -> Component {
        use crate::component_catalog::{CustomNodeConfig, Mecha10Config};

        Component {
            mecha10_config: Mecha10Config {
                driver: None,
                custom_node: Some(CustomNodeConfig {
                    name: name.to_string(),
                    path: format!("mecha10-nodes-{}", name),
                    config: None,
                    description: String::new(),
                    run_target: "robot".to_string(),
                    enabled: true,
                }),
                behavior: None,
            },
            ..placeholder_component(name)
        }
    }

    #[tokio::test]
    async fn mints_project_nodes_with_plain_identifier() {
        let temp_dir = tempfile::tempdir().expect("create temp dir");
        let project_root = temp_dir.path();
        tokio::fs::write(project_root.join(paths::PROJECT_CONFIG), minimal_project_config_json())
            .await
            .expect("write mecha10.json");

        let service = InitService::new();
        let component = placeholder_component("my-custom");

        service
            .add_node_to_config(project_root, "my-custom", &component, false)
            .await
            .expect("add_node_to_config should succeed");

        let config = ConfigService::load_from(&project_root.join(paths::PROJECT_CONFIG))
            .await
            .expect("reload config");

        assert!(
            config.nodes.identifiers().contains(&"my-custom"),
            "expected plain identifier 'my-custom' in nodes, got: {:?}",
            config.nodes.identifiers()
        );
        assert!(
            !config.nodes.identifiers().contains(&"@local/my-custom"),
            "new project nodes must not be minted with the legacy @local/ prefix, got: {:?}",
            config.nodes.identifiers()
        );
    }

    #[tokio::test]
    async fn mints_framework_nodes_with_mecha10_scope() {
        let temp_dir = tempfile::tempdir().expect("create temp dir");
        let project_root = temp_dir.path();
        tokio::fs::write(project_root.join(paths::PROJECT_CONFIG), minimal_project_config_json())
            .await
            .expect("write mecha10.json");

        let service = InitService::new();
        let component = placeholder_component("speaker");

        service
            .add_node_to_config(project_root, "speaker", &component, true)
            .await
            .expect("add_node_to_config should succeed");

        let config = ConfigService::load_from(&project_root.join(paths::PROJECT_CONFIG))
            .await
            .expect("reload config");

        assert!(
            config.nodes.identifiers().contains(&"@mecha10/speaker"),
            "expected framework identifier '@mecha10/speaker' in nodes, got: {:?}",
            config.nodes.identifiers()
        );
    }

    #[tokio::test]
    async fn appends_ai_perception_node_to_simulation_target_when_targets_absent() {
        let temp_dir = tempfile::tempdir().expect("create temp dir");
        let project_root = temp_dir.path();
        tokio::fs::write(project_root.join(paths::PROJECT_CONFIG), minimal_project_config_json())
            .await
            .expect("write mecha10.json");

        let service = InitService::new();
        let component = ai_perception_component("image-classifier");

        service
            .add_node_to_config(project_root, "image-classifier", &component, true)
            .await
            .expect("add_node_to_config should succeed");

        let config = ConfigService::load_from(&project_root.join(paths::PROJECT_CONFIG))
            .await
            .expect("reload config");

        let simulation = config
            .targets
            .simulation
            .expect("targets.simulation should be created when a relevant node is added");
        assert!(
            simulation.nodes.contains(&"@mecha10/image-classifier".to_string()),
            "expected @mecha10/image-classifier in targets.simulation.nodes, got {:?}",
            simulation.nodes
        );
    }

    #[tokio::test]
    async fn appends_ai_perception_node_to_existing_target_lists_without_duplicating() {
        let temp_dir = tempfile::tempdir().expect("create temp dir");
        let project_root = temp_dir.path();
        let config_with_targets = r#"{
            "name": "test-project",
            "version": "0.1.0",
            "robot_id": "test-robot",
            "targets": {
                "simulation": { "nodes": ["@mecha10/simulator"] }
            }
        }"#;
        tokio::fs::write(project_root.join(paths::PROJECT_CONFIG), config_with_targets)
            .await
            .expect("write mecha10.json");

        let service = InitService::new();
        let component = ai_perception_component("image-classifier");

        service
            .add_node_to_config(project_root, "image-classifier", &component, true)
            .await
            .expect("add_node_to_config should succeed");

        let config = ConfigService::load_from(&project_root.join(paths::PROJECT_CONFIG))
            .await
            .expect("reload config");
        let simulation = config
            .targets
            .simulation
            .expect("targets.simulation should be preserved");

        assert_eq!(
            simulation.nodes,
            vec![
                "@mecha10/simulator".to_string(),
                "@mecha10/image-classifier".to_string()
            ],
            "expected image-classifier appended to the existing simulation target nodes"
        );
    }

    #[tokio::test]
    async fn plain_component_does_not_create_simulation_target() {
        let temp_dir = tempfile::tempdir().expect("create temp dir");
        let project_root = temp_dir.path();
        tokio::fs::write(project_root.join(paths::PROJECT_CONFIG), minimal_project_config_json())
            .await
            .expect("write mecha10.json");

        let service = InitService::new();
        let component = placeholder_component("my-custom");

        service
            .add_node_to_config(project_root, "my-custom", &component, false)
            .await
            .expect("add_node_to_config should succeed");

        let config = ConfigService::load_from(&project_root.join(paths::PROJECT_CONFIG))
            .await
            .expect("reload config");

        assert!(
            config.targets.simulation.is_none(),
            "a component with no relevant lifecycle modes must not synthesize a targets.simulation entry"
        );
    }

    #[tokio::test]
    async fn syncs_hardware_backed_node_into_targets_robot() {
        let temp_dir = tempfile::tempdir().expect("create temp dir");
        let project_root = temp_dir.path();
        tokio::fs::write(project_root.join(paths::PROJECT_CONFIG), minimal_project_config_json())
            .await
            .expect("write mecha10.json");

        let service = InitService::new();
        let component = robot_hardware_component("motor");

        service
            .add_node_to_config(project_root, "motor", &component, true)
            .await
            .expect("add_node_to_config should succeed");

        let config = ConfigService::load_from(&project_root.join(paths::PROJECT_CONFIG))
            .await
            .expect("reload config");

        let robot = config
            .targets
            .robot
            .expect("targets.robot should be created for a robot-targeted node");
        assert!(
            robot.nodes.contains(&"@mecha10/motor".to_string()),
            "expected @mecha10/motor in targets.robot.nodes, got {:?}",
            robot.nodes
        );
        assert!(
            config.targets.remote.is_none() || config.targets.remote.unwrap().nodes.is_empty(),
            "a robot-targeted node must not also be added to targets.remote"
        );
    }

    #[tokio::test]
    async fn does_not_duplicate_hardware_node_already_present_in_targets_robot() {
        let temp_dir = tempfile::tempdir().expect("create temp dir");
        let project_root = temp_dir.path();
        let config_with_targets = r#"{
            "name": "test-project",
            "version": "0.1.0",
            "robot_id": "test-robot",
            "targets": {
                "robot": { "nodes": ["@mecha10/motor"] }
            }
        }"#;
        tokio::fs::write(project_root.join(paths::PROJECT_CONFIG), config_with_targets)
            .await
            .expect("write mecha10.json");

        let service = InitService::new();
        let component = robot_hardware_component("motor");

        service
            .add_node_to_config(project_root, "motor", &component, true)
            .await
            .expect("add_node_to_config should succeed");

        let config = ConfigService::load_from(&project_root.join(paths::PROJECT_CONFIG))
            .await
            .expect("reload config");
        let robot = config.targets.robot.expect("targets.robot should be preserved");
        let occurrences = robot.nodes.iter().filter(|n| n.as_str() == "@mecha10/motor").count();
        assert_eq!(
            occurrences, 1,
            "expected exactly one @mecha10/motor entry in targets.robot.nodes, got {:?}",
            robot.nodes
        );
    }
}