silex 0.1.0-beta.1

use silex_core::reactivity::Effect;
use silex_core::traits::Get;
use silex_dom::View;
use std::cell::RefCell;
use std::rc::Rc;
use web_sys::Node;

/// Helper trait to deduce View type from a closure
pub trait ShowContent {
    type View: View;
    fn render(&self) -> Self::View;
}

impl<F, V> ShowContent for F
where
    F: Fn() -> V,
    V: View,
{
    type View = V;
    fn render(&self) -> Self::View {
        (self)()
    }
}

/// Show 组件：根据条件渲染不同的视图
///
/// 使用 Builder 模式构建：
/// ```rust
/// use silex::prelude::*;
///
/// let (condition, set_condition) = signal(true);
/// let view = "Content";
/// let fallback_view = "Fallback";
///
/// Show::new(condition, move || view)
///     .fallback(move || fallback_view);
/// ```
#[derive(Clone)]
pub struct Show<Cond, ViewFn, FalsyViewFn> {
    condition: Cond,
    view: ViewFn,
    fallback: FalsyViewFn,
}

// 默认无 fallback 的构造函数
impl<Cond, ViewFn> Show<Cond, ViewFn, fn() -> ()> {
    pub fn new<V>(condition: Cond, view: ViewFn) -> Self
    where
        Cond: Get<Value = bool> + 'static,
        ViewFn: Fn() -> V + 'static,
        V: View,
    {
        Self {
            condition,
            view,
            fallback: || (),
        }
    }
}

// Builder 方法
impl<Cond, ViewFn, FalsyViewFn> Show<Cond, ViewFn, FalsyViewFn>
where
    Cond: Get<Value = bool> + 'static,
    ViewFn: ShowContent + 'static,
    FalsyViewFn: ShowContent + 'static,
{
    /// 设置当条件为 false 时的 fallback 视图 (Else 分支)
    pub fn fallback<NewFalsyFn>(self, fallback: NewFalsyFn) -> Show<Cond, ViewFn, NewFalsyFn>
    where
        NewFalsyFn: ShowContent + 'static,
    {
        Show {
            condition: self.condition,
            view: self.view,
            fallback,
        }
    }
}

impl<Cond, ViewFn, FalsyViewFn> View for Show<Cond, ViewFn, FalsyViewFn>
where
    Cond: Get<Value = bool> + 'static,
    ViewFn: ShowContent + Clone + 'static,
    FalsyViewFn: ShowContent + Clone + 'static,
{
    fn mount(self, parent: &Node) {
        let document = silex_dom::document();

        // 1. Create Anchors (Start & End Markers)
        let start_marker = document.create_comment("show-start");
        let start_node: Node = start_marker.into();

        if let Err(e) = parent
            .append_child(&start_node)
            .map_err(crate::SilexError::from)
        {
            silex_core::error::handle_error(e);
            return;
        }

        let end_marker = document.create_comment("show-end");
        let end_node: Node = end_marker.into();

        if let Err(e) = parent
            .append_child(&end_node)
            .map_err(crate::SilexError::from)
        {
            silex_core::error::handle_error(e);
            return;
        }

        let cond = self.condition;
        let view_fn = self.view;
        let fallback_fn = self.fallback;
        let prev_state = Rc::new(RefCell::new(None::<bool>));

        Effect::new(move |_| {
            let val = cond.get();

            let mut state = prev_state.borrow_mut();

            if *state == Some(val) {
                return;
            }

            // 清理旧内容 (Clean up between markers)
            if let Some(parent) = start_node.parent_node() {
                while let Some(sibling) = start_node.next_sibling() {
                    if sibling == end_node {
                        break;
                    }
                    let _ = parent.remove_child(&sibling);
                }
            }

            // 准备新内容 (Prepare new content)
            let fragment = document.create_document_fragment();
            let fragment_node: Node = fragment.clone().into();

            if val {
                view_fn.render().mount(&fragment_node);
            } else {
                fallback_fn.render().mount(&fragment_node);
            }

            // 插入新内容 (Insert new content)
            if let Some(parent) = end_node.parent_node() {
                let _ = parent.insert_before(&fragment_node, Some(&end_node));
            }

            *state = Some(val);
        });
    }
}

// --- Signal 扩展 ---

use silex_core::traits::IntoSignal;

/// Signal 扩展特质，提供 .when() 语法糖
pub trait SignalShowExt {
    type Cond: Get<Value = bool> + 'static;

    fn when<V, F>(self, view: F) -> Show<Self::Cond, F, fn() -> ()>
    where
        V: View,
        F: Fn() -> V + 'static;
}

// 为所有 IntoSignal<Value = bool> 的类型实现扩展
impl<S> SignalShowExt for S
where
    S: IntoSignal<Value = bool>,
    S::Signal: Clone + 'static,
{
    type Cond = S::Signal;

    fn when<V, F>(self, view: F) -> Show<Self::Cond, F, fn() -> ()>
    where
        V: View,
        F: Fn() -> V + 'static,
    {
        Show::new(self.into_signal(), view)
    }
}