hypen_engine/ir/

component.rs

use super::Element;
use indexmap::IndexMap;
use std::sync::Arc;

/// Result from component resolution
/// Contains the source code and the resolved path for the component
pub struct ResolvedComponent {
    pub source: String,
    pub path: String,
    pub passthrough: bool,
    pub lazy: bool,
}

/// Callback type for resolving component source code
/// Takes component name and optional path context, returns resolved component
/// Path context is the file path where this component is being referenced from
/// Resolver should return (source_code, resolved_path) where resolved_path is the
/// absolute path to the component file (used for resolving nested components)
pub type ComponentResolver =
    Arc<dyn Fn(&str, Option<&str>) -> Option<ResolvedComponent> + Send + Sync>;

/// A component definition - a template that can be instantiated
#[derive(Clone)]
pub struct Component {
    /// Component name
    pub name: String,

    /// Template function that produces element tree
    /// Takes props and returns expanded element tree
    pub template: Arc<dyn Fn(IndexMap<String, serde_json::Value>) -> Element + Send + Sync>,

    /// Default props
    pub default_props: IndexMap<String, serde_json::Value>,

    /// Source path where this component was loaded from (optional)
    pub source_path: Option<String>,

    /// If true, this component acts as a passthrough container
    /// It preserves its children and props without template expansion
    pub passthrough: bool,

    /// If true, this component's children are NOT expanded during initial pass
    /// Children remain as component references for lazy rendering
    pub lazy: bool,
}

impl Component {
    pub fn new(
        name: impl Into<String>,
        template: impl Fn(IndexMap<String, serde_json::Value>) -> Element + Send + Sync + 'static,
    ) -> Self {
        Self {
            name: name.into(),
            template: Arc::new(template),
            default_props: IndexMap::new(),
            source_path: None,
            passthrough: false,
            lazy: false,
        }
    }

    pub fn with_defaults(mut self, defaults: IndexMap<String, serde_json::Value>) -> Self {
        self.default_props = defaults;
        self
    }

    pub fn with_source_path(mut self, path: impl Into<String>) -> Self {
        self.source_path = Some(path.into());
        self
    }

    pub fn with_passthrough(mut self, passthrough: bool) -> Self {
        self.passthrough = passthrough;
        self
    }

    pub fn with_lazy(mut self, lazy: bool) -> Self {
        self.lazy = lazy;
        self
    }

    /// Instantiate this component with given props
    pub fn instantiate(&self, props: IndexMap<String, serde_json::Value>) -> Element {
        let mut merged_props = self.default_props.clone();
        merged_props.extend(props);
        (self.template)(merged_props)
    }
}

/// Registry of all available components
pub struct ComponentRegistry {
    /// Components indexed by fully qualified key (path:name or just name)
    components: IndexMap<String, Component>,
    /// Optional resolver for dynamically loading components
    resolver: Option<ComponentResolver>,
    /// Cache of resolved component paths to prevent re-resolving
    /// Key format: "path:component_name" or "component_name" if no path
    resolved_cache: IndexMap<String, bool>,
}

impl ComponentRegistry {
    pub fn new() -> Self {
        Self {
            components: IndexMap::new(),
            resolver: None,
            resolved_cache: IndexMap::new(),
        }
    }

    /// Register a primitive element name to skip component resolution
    /// Called by the renderer to mark built-in DOM elements
    pub fn register_primitive(&mut self, name: &str) {
        self.resolved_cache.insert(name.to_string(), false);
    }

    /// Set the component resolver callback
    pub fn set_resolver(&mut self, resolver: ComponentResolver) {
        self.resolver = Some(resolver);
    }

    pub fn register(&mut self, component: Component) {
        // Register with qualified key if source path is available
        if let Some(ref path) = component.source_path {
            let qualified_key = format!("{}:{}", path, component.name);
            self.components.insert(qualified_key, component.clone());
        }

        // Always register with unqualified name as fallback
        self.components.insert(component.name.clone(), component);
    }

    /// Get a component by name and optional context path
    pub fn get(&self, name: &str, context_path: Option<&str>) -> Option<&Component> {
        // Try with context path first
        if let Some(path) = context_path {
            let qualified_key = format!("{}:{}", path, name);
            if let Some(component) = self.components.get(&qualified_key) {
                return Some(component);
            }
        }

        // Fall back to just name (for globally registered components)
        self.components.get(name)
    }

    /// Try to resolve and register a component by name and context path
    fn try_resolve(&mut self, name: &str, context_path: Option<&str>) -> bool {
        // Build cache key
        let cache_key = if let Some(path) = context_path {
            format!("{}:{}", path, name)
        } else {
            name.to_string()
        };

        // Check cache first
        if let Some(&cached) = self.resolved_cache.get(&cache_key) {
            return cached;
        }

        // Try to resolve
        if let Some(ref resolver) = self.resolver {
            if let Some(resolved) = resolver(name, context_path) {
                // For lazy components, don't parse the template
                // Children won't be expanded until explicitly requested
                if resolved.lazy {
                    #[cfg(all(target_arch = "wasm32", feature = "js"))]
                    web_sys::console::log_1(
                        &format!("Registering lazy component: {}", name).into(),
                    );

                    // Create a dummy component - children won't be expanded
                    let dummy_element = Element::new(name);
                    let component = Component::new(name, move |_props| dummy_element.clone())
                        .with_source_path(resolved.path.clone())
                        .with_lazy(true);

                    self.register(component);
                    self.resolved_cache.insert(cache_key, true);
                    return true;
                }

                // For passthrough components, don't parse the template
                // They act as transparent containers
                if resolved.passthrough {
                    #[cfg(all(target_arch = "wasm32", feature = "js"))]
                    web_sys::console::log_1(
                        &format!("Registering passthrough component: {}", name).into(),
                    );

                    // Create a dummy component - the template won't be used for passthrough
                    let dummy_element = Element::new(name);
                    let component = Component::new(name, move |_props| dummy_element.clone())
                        .with_source_path(resolved.path.clone())
                        .with_passthrough(true);

                    self.register(component);
                    self.resolved_cache.insert(cache_key, true);
                    return true;
                }

                // Parse the component source for non-passthrough components
                match hypen_parser::parse_component(&resolved.source) {
                    Ok(component_spec) => {
                        // Convert to IR, preserving control-flow children
                        let ir_node = super::expand::ast_to_ir_node(&component_spec);
                        let element = match ir_node {
                            super::IRNode::Element(e) => e,
                            _ => {
                                #[cfg(all(target_arch = "wasm32", feature = "js"))]
                                web_sys::console::error_1(
                                    &format!("Component {} root must be an element", name).into(),
                                );
                                return false;
                            }
                        };

                        // Create a component that returns the parsed element
                        let component = Component::new(name, move |_props| element.clone())
                            .with_source_path(resolved.path.clone())
                            .with_passthrough(false);

                        self.register(component);
                        self.resolved_cache.insert(cache_key, true);
                        return true;
                    }
                    Err(e) => {
                        #[cfg(all(target_arch = "wasm32", feature = "js"))]
                        web_sys::console::error_1(
                            &format!("Failed to parse component {}: {:?}", name, e).into(),
                        );

                        #[cfg(not(all(target_arch = "wasm32", feature = "js")))]
                        eprintln!("Failed to parse component {}: {:?}", name, e);

                        self.resolved_cache.insert(cache_key, false);
                        return false;
                    }
                }
            }
        }

        self.resolved_cache.insert(cache_key, false);
        false
    }

    pub fn expand(&mut self, element: &Element) -> Element {
        self.expand_with_context(element, None)
    }

    /// Force expand an element's children (used for lazy components)
    /// This will expand children that were previously kept unexpanded
    pub fn expand_children(
        &mut self,
        element: &Element,
        context_path: Option<&str>,
    ) -> Vec<Element> {
        element
            .children
            .iter()
            .map(|child| self.expand_with_context(child, context_path))
            .collect()
    }

    /// Expand an element with a context path for component resolution
    fn expand_with_context(&mut self, element: &Element, context_path: Option<&str>) -> Element {
        // First check if component exists, if not try to resolve it
        let component_exists = self.get(&element.element_type, context_path).is_some();

        if !component_exists {
            // Try to resolve the component dynamically
            self.try_resolve(&element.element_type, context_path);
        }

        // If this element references a registered component, expand it
        if let Some(component) = self.get(&element.element_type, context_path) {
            // Check if this is a lazy component (children NOT expanded until explicitly requested)
            if component.lazy {
                // Lazy component: keep element and children, but DON'T expand children yet
                let mut element = element.clone();

                // Mark as lazy so reconciler knows to skip children
                element.props.insert(
                    "__lazy".to_string(),
                    super::Value::Static(serde_json::json!(true)),
                );

                #[cfg(all(target_arch = "wasm32", feature = "js"))]
                web_sys::console::log_1(
                    &format!(
                        "Lazy {} (props: {:?}): {} children kept unexpanded",
                        element.element_type,
                        element.props.keys().collect::<Vec<_>>(),
                        element.children.len()
                    )
                    .into(),
                );

                return element;
            }

            // Check if this is a passthrough component
            if component.passthrough {
                // Passthrough component: keep the original element but expand its children
                let mut element = element.clone();

                #[cfg(all(target_arch = "wasm32", feature = "js"))]
                {
                    let props_str = element
                        .props
                        .iter()
                        .map(|(k, v)| format!("{}={:?}", k, v))
                        .collect::<Vec<_>>()
                        .join(", ");
                    web_sys::console::log_1(
                        &format!(
                            "Passthrough {} (props: [{}]): {} children before expansion",
                            element.element_type,
                            props_str,
                            element.children.len()
                        )
                        .into(),
                    );
                }

                // Get the source path for child context
                let child_context = component.source_path.clone();
                let child_context_ref = child_context.as_deref();

                // Recursively expand children
                element.children = element
                    .children
                    .into_iter()
                    .map(|child| Arc::new(self.expand_with_context(&child, child_context_ref)))
                    .collect();

                #[cfg(all(target_arch = "wasm32", feature = "js"))]
                web_sys::console::log_1(
                    &format!(
                        "Passthrough {}: {} children after expansion",
                        element.element_type,
                        element.children.len()
                    )
                    .into(),
                );

                element
            } else {
                // Regular component: instantiate template and replace
                // Convert Value props to serde_json::Value (resolve only static values here)
                let mut props = IndexMap::new();
                for (k, v) in &element.props {
                    if let super::Value::Static(val) = v {
                        props.insert(k.clone(), val.clone());
                    }
                }

                let mut expanded = component.instantiate(props);

                // Preserve bindings and actions from the original element
                for (k, v) in &element.props {
                    match v {
                        super::Value::Binding(_) | super::Value::Action(_) => {
                            expanded.props.insert(k.clone(), v.clone());
                        }
                        _ => {}
                    }
                }

                // Get the source path of this component for resolving its children
                // Clone it to avoid holding a borrow
                let child_context = component.source_path.clone();

                // Replace Children() placeholders with actual children from the caller
                expanded.children = self.replace_children_slots(
                    &expanded.children,
                    &element.children,
                    context_path,
                );

                // Recursively expand children with the new context
                let child_context_ref = child_context.as_deref();
                expanded.children = expanded
                    .children
                    .into_iter()
                    .map(|child| Arc::new(self.expand_with_context(&child, child_context_ref)))
                    .collect();
                // Also expand ir_children (control-flow children from ast_to_ir_node)
                if !expanded.ir_children.is_empty() {
                    expanded.ir_children = expanded
                        .ir_children
                        .iter()
                        .map(|child| self.expand_ir_node_with_context(child, child_context_ref))
                        .collect();
                }

                expanded
            }
        } else {
            // Not a component, just expand children
            let mut element = element.clone();
            element.children = element
                .children
                .into_iter()
                .map(|child| Arc::new(self.expand_with_context(&child, context_path)))
                .collect();
            // Also expand ir_children (control-flow children from ast_to_ir_node)
            if !element.ir_children.is_empty() {
                element.ir_children = element
                    .ir_children
                    .iter()
                    .map(|child| self.expand_ir_node_with_context(child, context_path))
                    .collect();
            }
            element
        }
    }

    /// Expand an IRNode recursively, expanding any Element nodes via the component registry
    pub fn expand_ir_node(&mut self, node: &super::IRNode) -> super::IRNode {
        self.expand_ir_node_with_context(node, None)
    }

    /// Expand an IRNode with context path for component resolution
    fn expand_ir_node_with_context(
        &mut self,
        node: &super::IRNode,
        context_path: Option<&str>,
    ) -> super::IRNode {
        match node {
            super::IRNode::Element(element) => {
                // Expand the element through the registry
                let mut expanded = self.expand_with_context(element, context_path);

                // Also expand ir_children if present (control-flow children)
                if !expanded.ir_children.is_empty() {
                    expanded.ir_children = expanded
                        .ir_children
                        .iter()
                        .map(|child| self.expand_ir_node_with_context(child, context_path))
                        .collect();
                }

                super::IRNode::Element(expanded)
            }
            super::IRNode::ForEach {
                source,
                item_name,
                key_path,
                template,
                props,
            } => {
                // Recursively expand template children
                let expanded_template: Vec<super::IRNode> = template
                    .iter()
                    .map(|child| self.expand_ir_node_with_context(child, context_path))
                    .collect();

                super::IRNode::ForEach {
                    source: source.clone(),
                    item_name: item_name.clone(),
                    key_path: key_path.clone(),
                    template: expanded_template,
                    props: props.clone(),
                }
            }
            super::IRNode::Conditional {
                value,
                branches,
                fallback,
            } => {
                // Expand branch children
                let expanded_branches: Vec<super::ConditionalBranch> = branches
                    .iter()
                    .map(|branch| super::ConditionalBranch {
                        pattern: branch.pattern.clone(),
                        children: branch
                            .children
                            .iter()
                            .map(|child| self.expand_ir_node_with_context(child, context_path))
                            .collect(),
                    })
                    .collect();

                // Expand fallback children if present
                let expanded_fallback = fallback.as_ref().map(|fb| {
                    fb.iter()
                        .map(|child| self.expand_ir_node_with_context(child, context_path))
                        .collect()
                });

                super::IRNode::Conditional {
                    value: value.clone(),
                    branches: expanded_branches,
                    fallback: expanded_fallback,
                }
            }
        }
    }

    /// Replace Children() placeholders with actual children
    /// Supports named slots via Children().slot("header")
    fn replace_children_slots(
        &self,
        template_children: &im::Vector<Arc<Element>>,
        actual_children: &im::Vector<Arc<Element>>,
        _context_path: Option<&str>,
    ) -> im::Vector<Arc<Element>> {
        let mut result = im::Vector::new();

        for child in template_children {
            if child.element_type == "Children" {
                // Check if this is a named slot via .slot() applicator
                // Applicators are stored as props with format "applicatorName.argIndex"
                let slot_name = self.get_slot_name(&child.props);

                // If named slot, filter children by slot applicator
                // Otherwise, include all children that don't have a slot applicator
                if let Some(slot) = slot_name {
                    for c in actual_children.iter() {
                        if self.get_slot_name(&c.props) == Some(slot) {
                            result.push_back(Arc::clone(c));
                        }
                    }
                } else {
                    // Default slot - children without slot applicator
                    for c in actual_children.iter() {
                        if self.get_slot_name(&c.props).is_none() {
                            result.push_back(Arc::clone(c));
                        }
                    }
                }
            } else {
                // Not a Children() placeholder - keep as is but recurse into its children
                let mut new_child = (**child).clone();
                new_child.children =
                    self.replace_children_slots(&child.children, actual_children, _context_path);
                result.push_back(Arc::new(new_child));
            }
        }

        result
    }

    /// Extract slot name from applicators
    /// Looks for .slot("name") which becomes prop "slot.0" = "name"
    fn get_slot_name<'a>(&self, props: &'a super::Props) -> Option<&'a str> {
        props.get("slot.0").and_then(|v| {
            if let super::Value::Static(serde_json::Value::String(s)) = v {
                Some(s.as_str())
            } else {
                None
            }
        })
    }
}

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

#[cfg(test)]
mod tests {
    use super::*;
    use crate::ir::Value;

    #[test]
    fn test_dynamic_component_resolution() {
        let mut registry = ComponentRegistry::new();

        // Set up a resolver that returns component source and path
        registry.set_resolver(Arc::new(|name: &str, _context: Option<&str>| {
            if name == "Header" {
                Some(ResolvedComponent {
                    source: r#"Row { Text("Header") }"#.to_string(),
                    path: "/components/Header.hypen".to_string(),
                    passthrough: false,
                    lazy: false,
                })
            } else {
                None
            }
        }));

        // Create an element that references the unregistered Header component
        let element = Element::new("Column").with_child(Element::new("Header"));

        // Expand should trigger resolution
        let expanded = registry.expand(&element);

        // Should have expanded Header into Row { Text }
        assert_eq!(expanded.element_type, "Column");
        assert_eq!(expanded.children.len(), 1);
        let row = &expanded.children[0];
        assert_eq!(row.element_type, "Row");
        // Text child is in ir_children (from ast_to_ir_node path)
        assert_eq!(row.ir_children.len(), 1);
        match &row.ir_children[0] {
            crate::ir::IRNode::Element(text) => assert_eq!(text.element_type, "Text"),
            other => panic!("Expected Element, got {:?}", other),
        }
    }

    #[test]
    fn test_component_resolution_with_path_context() {
        let mut registry = ComponentRegistry::new();

        // Set up a resolver that resolves based on context path
        registry.set_resolver(Arc::new(|name: &str, context: Option<&str>| {
            match (name, context) {
                ("Button", Some("/pages/Home.hypen")) => Some(ResolvedComponent {
                    source: r#"Text("Home Button")"#.to_string(),
                    path: "/components/buttons/HomeButton.hypen".to_string(),
                    passthrough: false,
                    lazy: false,
                }),
                ("Button", Some("/pages/About.hypen")) => Some(ResolvedComponent {
                    source: r#"Text("About Button")"#.to_string(),
                    path: "/components/buttons/AboutButton.hypen".to_string(),
                    passthrough: false,
                    lazy: false,
                }),
                _ => None,
            }
        }));

        // Register a component with a source path
        let home_element =
            Element::new("Text").with_prop("0", Value::Static(serde_json::json!("Home")));
        let home_component = Component::new("Home", move |_| home_element.clone())
            .with_source_path("/pages/Home.hypen");
        registry.register(home_component);

        // Create an element that uses Button from Home context
        let element = Element::new("Column")
            .with_child(Element::new("Home").with_child(Element::new("Button")));

        let expanded = registry.expand(&element);

        // Button should resolve differently based on its context
        assert_eq!(expanded.element_type, "Column");
    }

    #[test]
    fn test_component_resolution_caching() {
        let mut registry = ComponentRegistry::new();
        let call_count = Arc::new(std::sync::Mutex::new(0));
        let call_count_clone = call_count.clone();

        // Resolver that tracks calls
        registry.set_resolver(Arc::new(move |name: &str, _context: Option<&str>| {
            if name == "Button" {
                *call_count_clone.lock().unwrap() += 1;
                Some(ResolvedComponent {
                    source: r#"Text("Click")"#.to_string(),
                    path: "/components/Button.hypen".to_string(),
                    passthrough: false,
                    lazy: false,
                })
            } else {
                None
            }
        }));

        // First expansion should call resolver
        let element1 = Element::new("Button");
        let _ = registry.expand(&element1);
        assert_eq!(*call_count.lock().unwrap(), 1);

        // Second expansion should use cache
        let element2 = Element::new("Button");
        let _ = registry.expand(&element2);
        assert_eq!(*call_count.lock().unwrap(), 1); // Still 1, not 2
    }

    #[test]
    fn test_failed_resolution_cached() {
        let mut registry = ComponentRegistry::new();
        let call_count = Arc::new(std::sync::Mutex::new(0));
        let call_count_clone = call_count.clone();

        // Resolver that returns None
        registry.set_resolver(Arc::new(move |_name: &str, _context: Option<&str>| {
            *call_count_clone.lock().unwrap() += 1;
            None
        }));

        // First expansion should call resolver
        let element1 = Element::new("Unknown");
        let _ = registry.expand(&element1);
        assert_eq!(*call_count.lock().unwrap(), 1);

        // Second expansion should use cached failure
        let element2 = Element::new("Unknown");
        let _ = registry.expand(&element2);
        assert_eq!(*call_count.lock().unwrap(), 1); // Cached
    }

    #[test]
    fn test_passthrough_component_preserves_props() {
        let mut registry = ComponentRegistry::new();

        // Register Router and Route as passthrough components
        registry.set_resolver(Arc::new(|name: &str, _context: Option<&str>| {
            if name == "Router" || name == "Route" {
                Some(ResolvedComponent {
                    source: String::new(), // Empty template for passthrough
                    path: name.to_string(),
                    passthrough: true,
                    lazy: false,
                })
            } else if name == "HomePage" {
                // Regular component with a simple template
                Some(ResolvedComponent {
                    source: "Text(\"Home\")".to_string(),
                    path: name.to_string(),
                    passthrough: false,
                    lazy: false,
                })
            } else {
                None
            }
        }));

        // Build a tree structure like:
        // Router {
        //   Route("/") { HomePage }
        //   Route("/about") { HomePage }
        // }
        let mut router = Element::new("Router");

        let mut route1 = Element::new("Route");
        route1
            .props
            .insert("0".to_string(), Value::Static(serde_json::json!("/")));
        route1
            .children
            .push_back(std::sync::Arc::new(Element::new("HomePage")));

        let mut route2 = Element::new("Route");
        route2
            .props
            .insert("0".to_string(), Value::Static(serde_json::json!("/about")));
        route2
            .children
            .push_back(std::sync::Arc::new(Element::new("HomePage")));

        router.children.push_back(std::sync::Arc::new(route1));
        router.children.push_back(std::sync::Arc::new(route2));

        // Expand the tree
        let expanded = registry.expand(&router);

        // Verify Router is preserved
        assert_eq!(expanded.element_type, "Router");
        assert_eq!(expanded.children.len(), 2);

        // Verify first Route preserves its path prop
        let expanded_route1 = &expanded.children[0];
        assert_eq!(expanded_route1.element_type, "Route");
        if let Some(Value::Static(path)) = expanded_route1.props.get("0") {
            assert_eq!(path.as_str().unwrap(), "/");
        } else {
            panic!("Route 1 missing path prop");
        }

        // Verify second Route preserves its path prop
        let expanded_route2 = &expanded.children[1];
        assert_eq!(expanded_route2.element_type, "Route");
        if let Some(Value::Static(path)) = expanded_route2.props.get("0") {
            assert_eq!(path.as_str().unwrap(), "/about");
        } else {
            panic!("Route 2 missing path prop");
        }

        // Verify children are expanded (HomePage should be replaced with Text)
        assert_eq!(expanded_route1.children.len(), 1);
        assert_eq!(expanded_route1.children[0].element_type, "Text");
    }
}