pax-runtime 0.39.1

use std::collections::{HashMap, HashSet};
use std::iter;
use std::rc::Rc;
use_RefCell!();

use pax_language::Computable;
use pax_manifest::ExpressionInfo;
use pax_runtime_api::CoercionRules;
use pax_runtime_api::{
    borrow, borrow_mut, use_RefCell, ImplToFromPaxAny, LocalProperty, PaxValue, ToPaxValue,
    Variable,
};

use crate::api::Layer;
use crate::{
    BaseInstance, ExpandedNode, InstanceFlags, InstanceNode, InstantiationArgs, RuntimeContext,
    RuntimePropertiesStackFrame,
};

mod value_eq;

fn update_repeat_bindings(
    index: &LocalProperty<usize>,
    item: &LocalProperty<PaxValue>,
    i: usize,
    value: PaxValue,
) {
    index.set_if_neq(i);
    // PaxValue's language equality is approximate. Invalidation requires exact
    // representation equality, otherwise small animation/data edits disappear.
    if !item.read(|previous| value_eq::same_value(previous, &value)) {
        item.set(value);
    }
}

const INTERNAL_REPEAT_INDEX_SYMBOL: &str = "$repeat_index";
const INTERNAL_REPEAT_ELEMENT_SYMBOL: &str = "$repeat_element";
const INTERNAL_REPEAT_KEY_SYMBOL: &str = "$repeat_key";

/// A special "control-flow" primitive associated with the `for` statement.
/// Repeat allows for nodes to be rendered dynamically per data specified in `source_expression`.
/// That is: for a `source_expression` of length `n`, `Repeat` will render its
/// template `n` times, each with an embedded component context (`RepeatItem`)
/// with an index `i` and a pointer to that relevant datum `source_expression[i]`
pub struct RepeatInstance {
    pub base: BaseInstance,
}

impl ImplToFromPaxAny for RepeatProperties {}
///Contains modal _vec_ and _range_ variants, describing whether the Repeat source
///is encoded as a `Vec<T>` (where T is a `PaxValue` properties type) or as a `Range<isize>`
#[derive(Default)]
pub struct RepeatProperties {
    pub source_expression: LocalProperty<PaxValue>,
    pub iterator_i_symbol: LocalProperty<Option<String>>,
    pub iterator_elem_symbol: LocalProperty<Option<String>>,
    pub repeat_key_expression: Option<ExpressionInfo>,
}

impl ToPaxValue for RepeatProperties {
    fn to_pax_value(self) -> PaxValue {
        PaxValue::Object(
            vec![
                (
                    "source_expression".to_string(),
                    self.source_expression.to_pax_value(),
                ),
                (
                    "iterator_i_symbol".to_string(),
                    self.iterator_i_symbol.to_pax_value(),
                ),
                (
                    "iterator_elem_symbol".to_string(),
                    self.iterator_elem_symbol.to_pax_value(),
                ),
            ]
            .into_iter()
            .collect(),
        )
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::api::math::Transform2;
    use crate::api::CommonProperties;
    use crate::{ComponentInstance, ExpandedNode, Globals, RouteLocation, TransformAndBounds};
    use pax_language::interpreter::property_resolution::IdentifierResolver;
    use pax_language::parse_pax_expression;
    use pax_manifest::cartridge_generation::{
        ComponentTransitionConfig, TRANSITION_PHASE_ENTER, TRANSITION_PHASE_EXIT,
        TRANSITION_PHASE_IDLE,
    };
    use pax_runtime_api::pax_value::{PaxAny, ToFromPaxAny};
    use pax_runtime_api::{borrow, CoercionRules, Numeric, Platform, Size, TargetInfo, OS};
    use std::cell::RefCell;

    fn test_globals() -> Globals {
        Globals {
            elapsed_frames: LocalProperty::new(0),
            elapsed_millis: LocalProperty::new(0),
            viewport: LocalProperty::new(TransformAndBounds {
                transform: Transform2::identity(),
                bounds: (100.0, 100.0),
            }),
            gyro: LocalProperty::new(Default::default()),
            accel: LocalProperty::new(Default::default()),
            route_location: LocalProperty::new(RouteLocation::root()),
            browser_allows_scroller_vector_layers: LocalProperty::new(true),
            browser_allows_nested_scroller_vector_layers: LocalProperty::new(true),
            platform: Platform::Unknown,
            os: OS::Unknown,
            target: TargetInfo::new(Platform::Unknown, OS::Unknown),
            #[cfg(feature = "designtime")]
            designtime: crate::test_support::designtime(),
            get_elapsed_millis: Rc::new(|| 0),
        }
    }

    fn default_properties_factory() -> Box<
        dyn Fn(
            Rc<RuntimePropertiesStackFrame>,
            Option<Rc<ExpandedNode>>,
        ) -> Option<Rc<RefCell<PaxAny>>>,
    > {
        Box::new(|_, expanded_node| {
            expanded_node
                .is_none()
                .then(|| Rc::new(RefCell::new(PaxAny::Builtin(PaxValue::default()))))
        })
    }

    fn default_common_properties_factory() -> Box<
        dyn Fn(
            Rc<RuntimePropertiesStackFrame>,
            Option<Rc<ExpandedNode>>,
        ) -> Option<Rc<RefCell<CommonProperties>>>,
    > {
        Box::new(|_, expanded_node| {
            expanded_node
                .is_none()
                .then(|| Rc::new(RefCell::new(CommonProperties::default())))
        })
    }

    fn component_args(template: Option<Vec<Rc<dyn InstanceNode>>>) -> InstantiationArgs {
        InstantiationArgs {
            prototypical_common_properties: crate::CommonPropertiesInit::Factory(
                default_common_properties_factory(),
            ),
            prototypical_properties: crate::PropertiesInit::Factory(default_properties_factory()),
            handler_registry: None,
            children: None,
            component_template: template.map(RefCell::new),
            component_settings: None,
            template_node_identifier: None,
            template_node_type_id: None,
            template_node_selector_info: None,
            transition_config: Default::default(),
            properties_scope: crate::PropertiesScopeInit::None,
        }
    }

    fn leaf_args(transition_config: ComponentTransitionConfig) -> InstantiationArgs {
        InstantiationArgs {
            prototypical_common_properties: crate::CommonPropertiesInit::Factory(
                default_common_properties_factory(),
            ),
            prototypical_properties: crate::PropertiesInit::Factory(default_properties_factory()),
            handler_registry: None,
            children: None,
            component_template: Some(RefCell::new(Vec::new())),
            component_settings: None,
            template_node_identifier: None,
            template_node_type_id: None,
            template_node_selector_info: None,
            transition_config,
            properties_scope: crate::PropertiesScopeInit::None,
        }
    }

    fn repeat_args(
        source: LocalProperty<PaxValue>,
        children: Vec<Rc<dyn InstanceNode>>,
    ) -> InstantiationArgs {
        repeat_args_with_key(source, children, true)
    }

    fn unkeyed_repeat_args(
        source: LocalProperty<PaxValue>,
        children: Vec<Rc<dyn InstanceNode>>,
    ) -> InstantiationArgs {
        repeat_args_with_key(source, children, false)
    }

    fn repeat_args_with_key(
        source: LocalProperty<PaxValue>,
        children: Vec<Rc<dyn InstanceNode>>,
        use_key: bool,
    ) -> InstantiationArgs {
        let key_expression = ExpressionInfo::new(parse_pax_expression("item.id").unwrap());
        let source_for_factory = source.clone();
        InstantiationArgs {
            prototypical_common_properties: crate::CommonPropertiesInit::Factory(
                default_common_properties_factory(),
            ),
            prototypical_properties: crate::PropertiesInit::Factory(Box::new(
                move |_, expanded_node| {
                    expanded_node.is_none().then(|| {
                        Rc::new(RefCell::new(
                            RepeatProperties {
                                source_expression: source_for_factory.clone(),
                                iterator_i_symbol: LocalProperty::new(Some("i".to_string())),
                                iterator_elem_symbol: LocalProperty::new(Some("item".to_string())),
                                repeat_key_expression: use_key.then(|| key_expression.clone()),
                            }
                            .to_pax_any(),
                        ))
                    })
                },
            )),
            handler_registry: None,
            children: Some(RefCell::new(children)),
            component_template: None,
            component_settings: None,
            template_node_identifier: None,
            template_node_type_id: None,
            template_node_selector_info: None,
            transition_config: Default::default(),
            properties_scope: crate::PropertiesScopeInit::None,
        }
    }

    fn positioned_leaf_args() -> InstantiationArgs {
        InstantiationArgs {
            prototypical_common_properties: crate::CommonPropertiesInit::Factory(Box::new(
                |env, expanded_node| {
                    if expanded_node.is_some() {
                        return None;
                    }
                    let i_untyped = env.resolve_symbol_as_erased_property("i").unwrap();
                    let i = LocalProperty::<usize>::new_from_untyped(i_untyped.clone());
                    let deps = [i_untyped];

                    let mut cp = CommonProperties::default();
                    cp.x = LocalProperty::computed(
                        move || Some(Size::Pixels((i.get() as f64 * 10.0).into())),
                        &deps,
                    );
                    cp.width = LocalProperty::new(Some(Size::Pixels(10.into())));
                    cp.height = LocalProperty::new(Some(Size::Pixels(10.into())));
                    Some(Rc::new(RefCell::new(cp)))
                },
            )),
            prototypical_properties: crate::PropertiesInit::Factory(default_properties_factory()),
            handler_registry: None,
            children: None,
            component_template: Some(RefCell::new(Vec::new())),
            component_settings: None,
            template_node_identifier: None,
            template_node_type_id: None,
            template_node_selector_info: None,
            transition_config: Default::default(),
            properties_scope: crate::PropertiesScopeInit::None,
        }
    }

    fn transitioned_positioned_leaf_args() -> InstantiationArgs {
        let mut args = positioned_leaf_args();
        args.transition_config = ComponentTransitionConfig {
            has_enter: true,
            enter_frame_count: 10,
            has_exit: true,
            exit_frame_count: 10,
            timeout_ms: 5_000,
            ..Default::default()
        };
        args
    }

    fn expression_positioned_leaf_args(expr: &str) -> InstantiationArgs {
        let expression = ExpressionInfo::new(parse_pax_expression(expr).unwrap());
        InstantiationArgs {
            prototypical_common_properties: crate::CommonPropertiesInit::Factory(Box::new(
                move |env, expanded_node| {
                    if expanded_node.is_some() {
                        return None;
                    }
                    let mut deps = Vec::new();
                    for dependency in &expression.dependencies {
                        let property = env
                            .resolve_symbol_as_erased_property(dependency)
                            .unwrap_or_else(|| panic!("missing dependency: {dependency}"));
                        deps.push(property);
                    }

                    let expression_for_eval = expression.clone();
                    let env_for_eval = Rc::clone(&env);
                    let mut cp = CommonProperties::default();
                    cp.x = LocalProperty::computed(
                        move || {
                            let env_for_eval: Rc<dyn IdentifierResolver> = env_for_eval.clone();
                            expression_for_eval
                                .expression
                                .compute(env_for_eval)
                                .ok()
                                .and_then(|value| Size::try_coerce(value).ok())
                        },
                        &deps,
                    );
                    cp.width = LocalProperty::new(Some(Size::Pixels(10.into())));
                    cp.height = LocalProperty::new(Some(Size::Pixels(10.into())));
                    Some(Rc::new(RefCell::new(cp)))
                },
            )),
            prototypical_properties: crate::PropertiesInit::Factory(default_properties_factory()),
            handler_registry: None,
            children: None,
            component_template: Some(RefCell::new(Vec::new())),
            component_settings: None,
            template_node_identifier: None,
            template_node_type_id: None,
            template_node_selector_info: None,
            transition_config: Default::default(),
            properties_scope: crate::PropertiesScopeInit::None,
        }
    }

    fn transitioned_expression_positioned_leaf_args(expr: &str) -> InstantiationArgs {
        let mut args = expression_positioned_leaf_args(expr);
        args.transition_config = ComponentTransitionConfig {
            has_enter: true,
            enter_frame_count: 10,
            has_exit: true,
            exit_frame_count: 10,
            timeout_ms: 5_000,
            ..Default::default()
        };
        args
    }

    fn item(id: &str) -> PaxValue {
        item_with_x(id, 0.0)
    }

    fn item_with_x(id: &str, x: f64) -> PaxValue {
        PaxValue::Object(
            vec![
                ("id".to_string(), PaxValue::String(id.to_string())),
                ("x".to_string(), x.to_pax_value()),
            ]
            .into_iter()
            .collect(),
        )
    }

    fn source(ids: &[&str]) -> PaxValue {
        PaxValue::Vec(ids.iter().map(|id| item(id)).collect())
    }

    fn source_with_x(entries: &[(&str, f64)]) -> PaxValue {
        PaxValue::Vec(entries.iter().map(|(id, x)| item_with_x(id, *x)).collect())
    }

    fn ids(nodes: &[Rc<ExpandedNode>]) -> Vec<u32> {
        nodes.iter().map(|node| node.id.0).collect()
    }

    fn mounted_repeat(source: LocalProperty<PaxValue>) -> (Rc<ExpandedNode>, Rc<RuntimeContext>) {
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(leaf_args(Default::default()));
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source, vec![leaf]));
        let root = ComponentInstance::instantiate(component_args(Some(vec![repeat])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root, &context);
        root.recurse_update(&context);
        context.drain_node_effects();
        (root, context)
    }

    fn binding_probe(
        child: &ExpandedNode,
        symbol: &str,
    ) -> (LocalProperty<()>, Rc<std::cell::Cell<usize>>) {
        let property = child
            .stack
            .resolve_symbol_as_erased_property(symbol)
            .unwrap();
        let runs = Rc::new(std::cell::Cell::new(0));
        let count = Rc::clone(&runs);
        let probe = LocalProperty::computed(move || count.set(count.get() + 1), &[property]);
        probe.get();
        (probe, runs)
    }

    #[test]
    fn keyed_repeat_only_notifies_changed_items_and_indices() {
        let source_property = LocalProperty::new(source_with_x(&[("a", 1.0), ("b", 2.0)]));
        let (root, context) = mounted_repeat(source_property.clone());
        let repeat = root.children.get().remove(0);
        let children = repeat.children.get();
        let probes: Vec<_> = children
            .iter()
            .flat_map(|child| [binding_probe(child, "item"), binding_probe(child, "i")])
            .collect();
        let counts = || {
            probes
                .iter()
                .map(|(probe, runs)| {
                    probe.get();
                    runs.get()
                })
                .collect::<Vec<_>>()
        };

        source_property.set(source_with_x(&[("a", 1.0), ("b", 2.0)]));
        context.drain_node_effects();
        assert_eq!(ids(&repeat.children.get()), ids(&children));
        assert_eq!(
            counts(),
            [1, 1, 1, 1],
            "identical source must not notify retained items"
        );

        source_property.set(source_with_x(&[("a", 1.0 + 1e-8), ("b", 2.0)]));
        context.drain_node_effects();
        assert_eq!(
            counts(),
            [2, 1, 1, 1],
            "even sub-epsilon edits must propagate"
        );

        source_property.set(source_with_x(&[("b", 2.0), ("a", 1.0 + 1e-8)]));
        context.drain_node_effects();
        assert_eq!(
            ids(&repeat.children.get()),
            [children[1].id.0, children[0].id.0]
        );
        assert_eq!(counts(), [2, 2, 1, 2], "reordering only changes indices");
    }

    #[test]
    fn keyed_repeat_reuses_reordered_children_and_exits_removed_keys() {
        let source_property = LocalProperty::new(source(&["a", "b", "c"]));
        let leaf_transition_config = ComponentTransitionConfig {
            has_exit: true,
            exit_frame_count: 30,
            timeout_ms: 5_000,
            ..Default::default()
        };
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(leaf_args(leaf_transition_config));
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property.clone(), vec![leaf]));
        let root_component =
            ComponentInstance::instantiate(component_args(Some(vec![Rc::clone(&repeat)])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root_component, &context);

        root.recurse_update(&context);
        let repeat_node = root.children.get().remove(0);
        let initial = repeat_node.children.get();
        assert_eq!(initial.len(), 3);
        let initial_ids = ids(&initial);

        source_property.set(source(&["c", "a", "b"]));
        root.recurse_update(&context);
        let reordered_active = borrow!(repeat_node.active_children).clone();
        assert_eq!(
            ids(&reordered_active),
            vec![initial_ids[2], initial_ids[0], initial_ids[1]]
        );
        assert!(borrow!(repeat_node.exiting_children).is_empty());

        source_property.set(source(&["c", "d", "a"]));
        root.recurse_update(&context);
        let active = borrow!(repeat_node.active_children).clone();
        let exiting = borrow!(repeat_node.exiting_children).clone();

        assert_eq!(active.len(), 3);
        assert_eq!(exiting.len(), 1);
        assert_eq!(active[0].id.0, initial_ids[2]);
        assert_ne!(active[1].id.0, initial_ids[0]);
        assert_ne!(active[1].id.0, initial_ids[1]);
        assert_ne!(active[1].id.0, initial_ids[2]);
        assert_eq!(active[2].id.0, initial_ids[0]);
        assert_eq!(exiting[0].id.0, initial_ids[1]);
        assert_eq!(exiting[0].transition_phase.get(), TRANSITION_PHASE_EXIT);

        source_property.set(source(&["c", "b", "a"]));
        root.recurse_update(&context);
        let active = borrow!(repeat_node.active_children).clone();
        assert_eq!(active[1].id.0, initial_ids[1]);
        assert_eq!(borrow!(repeat_node.mounted_children).len(), 4);
        assert_eq!(active[1].transition_phase.get(), TRANSITION_PHASE_IDLE);
    }

    #[test]
    fn retained_children_keep_parent_bindings_on_data_updates_and_reorder() {
        let source_property = LocalProperty::new(source_with_x(&[("a", 1.0), ("b", 2.0)]));
        let (root, context) = mounted_repeat(source_property.clone());
        let repeat = root.children.get().remove(0);
        let children = repeat.children.get();
        let initial: Vec<_> = children
            .iter()
            .map(|c| c.parent_binding_rebuilds.get())
            .collect();
        for value in [
            source_with_x(&[("a", 3.0), ("b", 2.0)]),
            source_with_x(&[("b", 2.0), ("a", 3.0)]),
        ] {
            source_property.set(value);
            context.drain_node_effects();
            assert_eq!(
                children
                    .iter()
                    .map(|c| c.parent_binding_rebuilds.get())
                    .collect::<Vec<_>>(),
                initial
            );
        }
        source_property.set(source_with_x(&[("b", 2.0), ("a", 3.0), ("c", 4.0)]));
        context.drain_node_effects();
        assert_eq!(
            children
                .iter()
                .map(|c| c.parent_binding_rebuilds.get())
                .collect::<Vec<_>>(),
            initial
        );
        assert_eq!(repeat.children.get().len(), 3);
    }

    #[test]
    fn equal_valued_parent_sources_are_not_interchangeable() {
        let (root, context) = mounted_repeat(LocalProperty::new(source(&["a"])));
        let repeat = root.children.get().remove(0);
        let children = repeat.children.get();
        let child = &children[0];
        let first = LocalProperty::new(None);
        let second = LocalProperty::new(None);
        repeat.attach_children(children.clone(), &context, &first);
        let n = child.parent_binding_rebuilds.get();
        repeat.attach_children(children.clone(), &context, &first);
        assert_eq!(child.parent_binding_rebuilds.get(), n);
        repeat.attach_children(children.clone(), &context, &second);
        assert_eq!(child.parent_binding_rebuilds.get(), n + 1);
        first.set(Some(root.id));
        assert_eq!(
            child.parent_frame.get(),
            None,
            "old frame source must be disconnected"
        );
        second.set(Some(repeat.id));
        assert_eq!(child.parent_frame.get(), Some(repeat.id));

        // Existing subscriptions must follow parent values without reattachment.
        let cp = repeat.get_common_properties();
        borrow!(cp)
            .opacity
            .set(Some(pax_runtime_api::Opacity::from(0.25)));
        assert_eq!(child.computed_opacity.get(), 0.25);
        assert_eq!(child.parent_binding_rebuilds.get(), n + 1);
    }

    #[test]
    fn data_only_reconciliation_skips_structure_and_churn_releases_properties() {
        let source_property = LocalProperty::new(source_with_x(&[("a", 1.0), ("b", 2.0)]));
        let (root, context) = mounted_repeat(source_property.clone());
        let repeat = root.children.get().remove(0);
        let counts = || {
            (
                repeat.layout_hull_rebuilds.get(),
                repeat.structural_children_updates.get(),
            )
        };
        let initial = counts();
        source_property.set(source_with_x(&[("a", 8.0), ("b", 2.0)]));
        context.drain_node_effects();
        assert_eq!(counts(), initial);
        let settled = pax_runtime_api::properties::property_table_total_properties_count();
        for _ in 0..40 {
            source_property.set(source_with_x(&[("a", 8.0), ("b", 2.0), ("c", 3.0)]));
            context.drain_node_effects();
            source_property.set(source_with_x(&[("a", 8.0), ("b", 2.0)]));
            context.drain_node_effects();
            assert_eq!(
                pax_runtime_api::properties::property_table_total_properties_count(),
                settled
            );
        }
    }

    #[test]
    fn nested_repeats_do_not_reconcile_unchanged_sibling_payloads() {
        let data = |x| {
            PaxValue::Vec(vec![
                PaxValue::Object(vec![
                    ("id".into(), PaxValue::String("a".into())),
                    ("parts".into(), source_with_x(&[("part", x)])),
                ]),
                PaxValue::Object(vec![
                    ("id".into(), PaxValue::String("b".into())),
                    ("parts".into(), source_with_x(&[("part", 2.0)])),
                ]),
            ])
        };
        let source_property = LocalProperty::new(data(1.0));
        let source_evaluations = Rc::new(std::cell::Cell::new(0));
        let evaluations = source_evaluations.clone();
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(leaf_args(Default::default()));
        let mut inner_args = repeat_args(LocalProperty::new(PaxValue::default()), vec![leaf]);
        inner_args.prototypical_properties =
            crate::PropertiesInit::Factory(Box::new(move |env, node| {
                if node.is_some() {
                    return None;
                }
                let outer_item = LocalProperty::<PaxValue>::new_from_untyped(
                    env.resolve_symbol_as_erased_property("item").unwrap(),
                );
                let dep = outer_item.untyped();
                let evaluations = evaluations.clone();
                let source_expression = LocalProperty::computed(
                    move || {
                        evaluations.set(evaluations.get() + 1);
                        let PaxValue::Object(fields) = outer_item.get() else {
                            panic!("expected item object")
                        };
                        fields
                            .into_iter()
                            .find(|(name, _)| name == "parts")
                            .unwrap()
                            .1
                    },
                    &[dep],
                );
                Some(Rc::new(RefCell::new(
                    RepeatProperties {
                        source_expression,
                        iterator_i_symbol: LocalProperty::new(Some("i".into())),
                        iterator_elem_symbol: LocalProperty::new(Some("item".into())),
                        repeat_key_expression: Some(ExpressionInfo::new(
                            parse_pax_expression("item.id").unwrap(),
                        )),
                    }
                    .to_pax_any(),
                )))
            }));
        let inner: Rc<dyn InstanceNode> = RepeatInstance::instantiate(inner_args);
        let outer: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property.clone(), vec![inner]));
        let root = ComponentInstance::instantiate(component_args(Some(vec![outer])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root, &context);
        root.recurse_update(&context);
        context.drain_node_effects();
        let outer = root.children.get().remove(0);
        let inners = outer.children.get();
        let first = inners[0].children.get()[0].clone();
        let other = inners[1].children.get()[0].clone();
        let (first_probe, first_runs) = binding_probe(&first, "item");
        let (other_probe, other_runs) = binding_probe(&other, "item");
        let evaluations = source_evaluations.get();
        source_property.set(data(1.0));
        context.drain_node_effects();
        assert_eq!(source_evaluations.get(), evaluations);
        source_property.set(data(1.0 + 1e-8));
        context.drain_node_effects();
        first_probe.get();
        other_probe.get();
        assert_eq!(source_evaluations.get(), evaluations + 1);
        assert_eq!((first_runs.get(), other_runs.get()), (2, 1));
        assert_eq!(inners[0].children.get()[0].id, first.id);
        assert_eq!(inners[1].children.get()[0].id, other.id);
    }

    #[test]
    fn parent_values_resize_reactively_and_replaced_sources_rebind() {
        let (root, context) = mounted_repeat(LocalProperty::new(source(&["a"])));
        let repeat = root.children.get().remove(0);
        let children = repeat.children.get();
        let child = &children[0];
        let parent_cp = repeat.get_common_properties();
        let child_cp = child.get_common_properties();
        borrow!(parent_cp).width.set(Some(Size::Pixels(100.into())));
        borrow!(child_cp).width.set(Some(Size::Percent(50.into())));
        assert_eq!(child.transform_and_bounds.get().bounds.0, 50.0);
        let rebuilds = child.parent_binding_rebuilds.get();
        borrow!(parent_cp).width.set(Some(Size::Pixels(200.into())));
        assert_eq!(child.transform_and_bounds.get().bounds.0, 100.0);
        repeat.attach_children(children.clone(), &context, &repeat.parent_frame);
        assert_eq!(child.parent_binding_rebuilds.get(), rebuilds);

        let old_opacity = borrow!(child_cp).opacity.clone();
        let new_opacity = LocalProperty::new(old_opacity.get());
        borrow_mut!(child_cp).opacity = new_opacity.clone();
        repeat.attach_children(children.clone(), &context, &repeat.parent_frame);
        assert_eq!(child.parent_binding_rebuilds.get(), rebuilds + 1);
        old_opacity.set(Some(0.25.into()));
        assert_eq!(child.computed_opacity.get(), 1.0);
        new_opacity.set(Some(0.75.into()));
        assert_eq!(child.computed_opacity.get(), 0.75);

        let template = borrow!(child.instance_node).clone();
        child.recreate_with_new_data(template, &context);
        context.drain_node_effects();
        let rebuilt = child.parent_binding_rebuilds.get();
        repeat.attach_children(children.clone(), &context, &repeat.parent_frame);
        assert_eq!(
            child.parent_binding_rebuilds.get(),
            rebuilt + 1,
            "reload must invalidate the binding fast path"
        );
    }

    #[test]
    fn repeat_exit_partition_and_final_removal_are_structural_changes() {
        let source_property = LocalProperty::new(source(&["a"]));
        let config = ComponentTransitionConfig {
            has_exit: true,
            exit_frame_count: 5,
            timeout_ms: 5_000,
            ..Default::default()
        };
        let leaf: Rc<dyn InstanceNode> = ComponentInstance::instantiate(leaf_args(config));
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property.clone(), vec![leaf]));
        let root = ComponentInstance::instantiate(component_args(Some(vec![repeat])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root, &context);
        root.recurse_update(&context);
        context.drain_node_effects();
        let repeat = root.children.get().remove(0);
        let id = repeat.children.get()[0].id;
        let before_exit = repeat.structural_children_updates.get();
        source_property.set(source(&[]));
        context.drain_node_effects();
        assert_eq!(repeat.children.get()[0].id, id);
        assert!(
            repeat.structural_children_updates.get() > before_exit,
            "active-to-exiting affects projection even with identical rendered IDs"
        );
        let before_prune = repeat.structural_children_updates.get();
        context.globals().elapsed_frames.set(10);
        context.drain_node_effects();
        assert!(repeat.children.get().is_empty());
        assert!(repeat.structural_children_updates.get() > before_prune);

        // Cleanup and a new selection in the same tick must both be observed.
        source_property.set(source(&["a"]));
        context.drain_node_effects();
        source_property.set(source(&[]));
        context.drain_node_effects();
        context.globals().elapsed_frames.set(20);
        source_property.set(source(&["b"]));
        context.drain_node_effects();
        assert_eq!(repeat.children.get().len(), 1);
        assert!(borrow!(repeat.exiting_children).is_empty());
    }

    #[test]
    fn external_key_dependencies_and_invalid_key_fallback_remain_reactive() {
        let source_property = LocalProperty::new(source(&["a"]));
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(leaf_args(Default::default()));
        let mut args = repeat_args(source_property.clone(), vec![leaf]);
        let source_copy = source_property.clone();
        args.prototypical_properties = crate::PropertiesInit::Factory(Box::new(move |_, node| {
            node.is_none().then(|| {
                Rc::new(RefCell::new(
                    RepeatProperties {
                        source_expression: source_copy.clone(),
                        iterator_i_symbol: LocalProperty::new(Some("i".into())),
                        iterator_elem_symbol: LocalProperty::new(Some("item".into())),
                        repeat_key_expression: Some(ExpressionInfo::new(
                            parse_pax_expression("$frames").unwrap(),
                        )),
                    }
                    .to_pax_any(),
                ))
            })
        }));
        let repeat: Rc<dyn InstanceNode> = RepeatInstance::instantiate(args);
        let root = ComponentInstance::instantiate(component_args(Some(vec![repeat])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root, &context);
        root.recurse_update(&context);
        context.drain_node_effects();
        let repeat = root.children.get().remove(0);
        let first_id = repeat.children.get()[0].id;
        context.globals().elapsed_frames.set(1);
        context.drain_node_effects();
        assert_ne!(repeat.children.get()[0].id, first_id);

        let invalid = PaxValue::Object(vec![("id".into(), PaxValue::Bool(true))]);
        let data = PaxValue::Vec(vec![item("same"), item("same"), invalid]);
        let source_property = LocalProperty::new(data.clone());
        let (root, context) = mounted_repeat(source_property.clone());
        let repeat = root.children.get().remove(0);
        let first_ids = ids(&repeat.children.get());
        assert_eq!(first_ids.iter().collect::<HashSet<_>>().len(), 3);
        source_property.set(data);
        context.drain_node_effects();
        assert_eq!(ids(&repeat.children.get()), first_ids);
    }

    #[test]
    fn keyed_groups_keep_all_template_roots_together_on_reorder() {
        let source_property = LocalProperty::new(source(&["a", "b"]));
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(leaf_args(Default::default()));
        let repeat: Rc<dyn InstanceNode> = RepeatInstance::instantiate(repeat_args(
            source_property.clone(),
            vec![leaf.clone(), leaf],
        ));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(
            ComponentInstance::instantiate(component_args(Some(vec![repeat]))),
            &context,
        );
        root.recurse_update(&context);
        context.drain_node_effects();
        let repeat = root.children.get().remove(0);
        let initial = ids(&repeat.children.get());
        assert_eq!(initial.len(), 4);
        source_property.set(source(&["b", "a"]));
        context.drain_node_effects();
        assert_eq!(
            ids(&repeat.children.get()),
            [initial[2], initial[3], initial[0], initial[1]]
        );
    }

    #[test]
    fn detached_mask_style_repeat_binds_inserted_children_and_releases_last_child() {
        let source_property = LocalProperty::new(source(&[]));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(
            ComponentInstance::instantiate(component_args(None)),
            &context,
        );
        let mut args = leaf_args(Default::default());
        args.prototypical_common_properties =
            crate::CommonPropertiesInit::Factory(Box::new(|_, node| {
                node.is_none().then(|| {
                    let mut cp = CommonProperties::default();
                    cp.width = LocalProperty::new(Some(Size::Percent(50.into())));
                    cp.height = LocalProperty::new(Some(Size::Percent(50.into())));
                    Rc::new(RefCell::new(cp))
                })
            }));
        let leaf: Rc<dyn InstanceNode> = ComponentInstance::instantiate(args);
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property.clone(), vec![leaf]));
        let sidecar = root.create_children_detached(
            iter::once((repeat, root.stack.clone())),
            &context,
            &Rc::downgrade(&root),
        );
        let sidecar = root.attach_sidecar_children(sidecar, &context, &root.parent_frame);
        let repeat = &sidecar[0];
        repeat.recurse_control_flow_expansion(&context);
        context.drain_node_effects();
        assert!(repeat.children.get().is_empty());
        source_property.set(source(&["ring"]));
        let children = repeat.children.get();
        repeat.attach_children(children.clone(), &context, &repeat.parent_frame);
        context.drain_node_effects();
        assert_eq!(
            children[0].attached.get(),
            0,
            "mask geometry must not acquire visible mount lifecycle"
        );
        assert_eq!(children[0].transform_and_bounds.get().bounds, (50.0, 50.0));
        context.globals().viewport.set(TransformAndBounds {
            transform: Transform2::identity(),
            bounds: (300.0, 200.0),
        });
        context.drain_node_effects();
        assert_eq!(
            children[0].transform_and_bounds.get().bounds,
            (150.0, 100.0)
        );
        let rebuilds = children[0].parent_binding_rebuilds.get();
        source_property.set(source(&["ring"]));
        repeat.attach_children(repeat.children.get(), &context, &repeat.parent_frame);
        assert_eq!(children[0].parent_binding_rebuilds.get(), rebuilds);
        source_property.set(source(&[]));
        repeat.attach_children(repeat.children.get(), &context, &repeat.parent_frame);
        context.drain_node_effects();
        assert!(repeat.children.get().is_empty());
        assert!(borrow!(repeat.mounted_children).is_empty());
    }

    #[test]
    fn unkeyed_repeat_rescues_a_position_readded_during_exit() {
        let source_property = LocalProperty::new(source(&["a", "b"]));
        let leaf_transition_config = ComponentTransitionConfig {
            has_enter: true,
            enter_frame_count: 10,
            has_exit: true,
            exit_frame_count: 30,
            timeout_ms: 5_000,
            ..Default::default()
        };
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(leaf_args(leaf_transition_config));
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(unkeyed_repeat_args(source_property.clone(), vec![leaf]));
        let root_component =
            ComponentInstance::instantiate(component_args(Some(vec![Rc::clone(&repeat)])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root_component, &context);

        root.recurse_update(&context);
        let repeat_node = root.children.get().remove(0);
        let second_id = borrow!(repeat_node.active_children)[1].id;

        source_property.set(source(&["a"]));
        root.recurse_update(&context);
        assert_eq!(borrow!(repeat_node.exiting_children)[0].id, second_id);

        source_property.set(source(&["a", "b"]));
        root.recurse_update(&context);
        assert_eq!(borrow!(repeat_node.active_children)[1].id, second_id);
        assert!(borrow!(repeat_node.exiting_children).is_empty());
        assert_eq!(
            borrow!(repeat_node.active_children)[1]
                .transition_phase
                .get(),
            TRANSITION_PHASE_ENTER
        );
    }

    #[test]
    fn keyed_repeat_assigns_distinct_scope_bindings_per_child() {
        let source_property = LocalProperty::new(source(&["a", "b", "c"]));
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(leaf_args(Default::default()));
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property, vec![leaf]));
        let root_component =
            ComponentInstance::instantiate(component_args(Some(vec![Rc::clone(&repeat)])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root_component, &context);

        root.recurse_update(&context);
        let repeat_node = root.children.get().remove(0);
        let children = repeat_node.children.get();
        assert_eq!(children.len(), 3);

        let indices: Vec<_> = children
            .iter()
            .map(|child| {
                child
                    .stack
                    .resolve_symbol("i")
                    .and_then(|variable| Numeric::try_coerce(variable.get_as_pax_value()).ok())
                    .map(|value| value.to_int())
            })
            .collect();
        assert_eq!(indices, vec![Some(0), Some(1), Some(2)]);
    }

    #[test]
    fn keyed_repeat_assigns_distinct_element_bindings_per_child() {
        let source_property =
            LocalProperty::new(source_with_x(&[("a", 0.0), ("b", 10.0), ("c", 20.0)]));
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(leaf_args(Default::default()));
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property, vec![leaf]));
        let root_component =
            ComponentInstance::instantiate(component_args(Some(vec![Rc::clone(&repeat)])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root_component, &context);

        root.recurse_update(&context);
        let repeat_node = root.children.get().remove(0);
        let children = repeat_node.children.get();
        assert_eq!(children.len(), 3);

        let xs: Vec<_> = children
            .iter()
            .map(|child| {
                child
                    .stack
                    .resolve_symbol("item")
                    .map(|variable| variable.get_as_pax_value())
                    .and_then(|value| match value {
                        PaxValue::Object(fields) => fields
                            .into_iter()
                            .find_map(|(name, value)| (name == "x").then_some(value)),
                        _ => None,
                    })
                    .and_then(|value| Numeric::try_coerce(value).ok())
                    .map(|value| value.to_float())
            })
            .collect();

        assert_eq!(xs, vec![Some(0.0), Some(10.0), Some(20.0)]);
    }

    #[test]
    fn keyed_repeat_resolves_distinct_common_properties_per_child() {
        let source_property = LocalProperty::new(source(&["a", "b", "c"]));
        let leaf: Rc<dyn InstanceNode> = ComponentInstance::instantiate(positioned_leaf_args());
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property, vec![leaf]));
        let root_component =
            ComponentInstance::instantiate(component_args(Some(vec![Rc::clone(&repeat)])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root_component, &context);

        root.recurse_update(&context);
        let repeat_node = root.children.get().remove(0);
        let children = repeat_node.children.get();
        assert_eq!(children.len(), 3);

        let xs: Vec<_> = children
            .iter()
            .map(|child| {
                borrow!(child.get_common_properties())
                    .x
                    .get()
                    .map(|size| size.expect_pixels().to_float())
            })
            .collect();

        assert_eq!(xs, vec![Some(0.0), Some(10.0), Some(20.0)]);
    }

    #[test]
    fn keyed_repeat_resolves_distinct_expression_common_properties_from_element_binding() {
        let source_property =
            LocalProperty::new(source_with_x(&[("a", 0.0), ("b", 10.0), ("c", 20.0)]));
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(expression_positioned_leaf_args("(item.x)px"));
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property, vec![leaf]));
        let root_component =
            ComponentInstance::instantiate(component_args(Some(vec![Rc::clone(&repeat)])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root_component, &context);

        root.recurse_update(&context);
        let repeat_node = root.children.get().remove(0);
        let children = repeat_node.children.get();
        assert_eq!(children.len(), 3);

        let xs: Vec<_> = children
            .iter()
            .map(|child| {
                borrow!(child.get_common_properties())
                    .x
                    .get()
                    .map(|size| size.expect_pixels().to_float())
            })
            .collect();

        assert_eq!(xs, vec![Some(0.0), Some(10.0), Some(20.0)]);
    }

    #[test]
    fn keyed_repeat_resolves_distinct_common_properties_for_transition_components() {
        let source_property = LocalProperty::new(source(&["a", "b", "c"]));
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(transitioned_positioned_leaf_args());
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property, vec![leaf]));
        let root_component =
            ComponentInstance::instantiate(component_args(Some(vec![Rc::clone(&repeat)])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root_component, &context);

        root.recurse_update(&context);
        let repeat_node = root.children.get().remove(0);
        let children = repeat_node.children.get();
        assert_eq!(children.len(), 3);

        let xs: Vec<_> = children
            .iter()
            .map(|child| {
                borrow!(child.get_common_properties())
                    .x
                    .get()
                    .map(|size| size.expect_pixels().to_float())
            })
            .collect();

        assert_eq!(xs, vec![Some(0.0), Some(10.0), Some(20.0)]);
    }

    #[test]
    fn keyed_repeat_resolves_expression_common_properties_for_transition_components() {
        let source_property =
            LocalProperty::new(source_with_x(&[("a", 0.0), ("b", 10.0), ("c", 20.0)]));
        let leaf: Rc<dyn InstanceNode> = ComponentInstance::instantiate(
            transitioned_expression_positioned_leaf_args("(item.x)px"),
        );
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(repeat_args(source_property, vec![leaf]));
        let root_component =
            ComponentInstance::instantiate(component_args(Some(vec![Rc::clone(&repeat)])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root_component, &context);

        root.recurse_update(&context);
        let repeat_node = root.children.get().remove(0);
        let children = repeat_node.children.get();
        assert_eq!(children.len(), 3);

        let xs: Vec<_> = children
            .iter()
            .map(|child| {
                borrow!(child.get_common_properties())
                    .x
                    .get()
                    .map(|size| size.expect_pixels().to_float())
            })
            .collect();

        assert_eq!(xs, vec![Some(0.0), Some(10.0), Some(20.0)]);
    }

    #[test]
    fn unkeyed_repeat_stale_element_binding_defaults_after_source_shrinks() {
        let source_property = LocalProperty::new(source_with_x(&[("a", 0.0), ("b", 10.0)]));
        let leaf: Rc<dyn InstanceNode> =
            ComponentInstance::instantiate(leaf_args(Default::default()));
        let repeat: Rc<dyn InstanceNode> =
            RepeatInstance::instantiate(unkeyed_repeat_args(source_property.clone(), vec![leaf]));
        let root_component =
            ComponentInstance::instantiate(component_args(Some(vec![Rc::clone(&repeat)])));
        let context = Rc::new(crate::test_support::runtime_context(test_globals()));
        let root = ExpandedNode::initialize_root(root_component, &context);

        root.recurse_update(&context);
        let repeat_node = root.children.get().remove(0);
        let children = repeat_node.children.get();
        assert_eq!(children.len(), 2);

        let stale_item = children[1].stack.resolve_symbol("item").unwrap();
        source_property.set(source_with_x(&[("a", 0.0)]));

        assert_eq!(stale_item.get_as_pax_value(), PaxValue::default());
    }
}

#[derive(Clone, Debug, Eq, Hash, PartialEq)]
enum RepeatKey {
    String(String),
    Int(i64),
    Synthetic(usize),
}

#[derive(Clone)]
struct RepeatChildGroup {
    key: RepeatKey,
    elem: LocalProperty<PaxValue>,
    i: LocalProperty<usize>,
    children: Vec<Rc<ExpandedNode>>,
}

/// Per-iteration bindings exposed inside a `for` template body.
pub struct RepeatItem {
    pub elem: LocalProperty<PaxValue>,
    pub i: LocalProperty<usize>,
}

impl ToPaxValue for RepeatItem {
    fn to_pax_value(self) -> PaxValue {
        PaxValue::Object(
            vec![
                ("elem".to_string(), self.elem.get().to_pax_value()),
                ("i".to_string(), self.i.get().to_pax_value()),
            ]
            .into_iter()
            .collect(),
        )
    }
}

impl InstanceNode for RepeatInstance {
    fn instantiate(args: InstantiationArgs) -> Rc<Self>
    where
        Self: Sized,
    {
        Rc::new(Self {
            base: BaseInstance::new(
                args,
                InstanceFlags {
                    invisible_to_slot: true,
                    invisible_to_raycasting: true,
                    layer: Layer::DontCare,
                    is_component: false,
                    is_slot: false,
                },
            ),
        })
    }

    fn resolve_debug(
        &self,
        f: &mut std::fmt::Formatter,
        _expanded_node: Option<&ExpandedNode>,
    ) -> std::fmt::Result {
        f.debug_struct("Repeat").finish()
    }

    fn base(&self) -> &BaseInstance {
        &self.base
    }

    fn handle_mount(
        self: Rc<Self>,
        expanded_node: &Rc<ExpandedNode>,
        context: &Rc<RuntimeContext>,
    ) {
        self.handle_setup(expanded_node, context, true);
    }

    fn handle_control_flow_node_expansion(
        self: Rc<Self>,
        expanded_node: &Rc<ExpandedNode>,
        context: &Rc<RuntimeContext>,
    ) {
        self.handle_setup(expanded_node, context, false);
    }
}

impl RepeatInstance {
    fn handle_setup(
        self: Rc<Self>,
        expanded_node: &Rc<ExpandedNode>,
        context: &Rc<RuntimeContext>,
        is_mount: bool,
    ) {
        // No-op: wait with creating child-nodes until update tick, since the
        // condition has then been evaluated
        let weak_ref_self = Rc::downgrade(expanded_node);
        let cloned_self = Rc::clone(&self);
        let cloned_context = Rc::clone(context);
        let source_expression =
            expanded_node.with_properties_unwrapped(|properties: &mut RepeatProperties| {
                properties.source_expression.clone()
            });

        let i_symbol =
            expanded_node.with_properties_unwrapped(|properties: &mut RepeatProperties| {
                properties.iterator_i_symbol.clone()
            });
        let elem_symbol =
            expanded_node.with_properties_unwrapped(|properties: &mut RepeatProperties| {
                properties.iterator_elem_symbol.clone()
            });
        let repeat_key_expression =
            expanded_node.with_properties_unwrapped(|properties: &mut RepeatProperties| {
                properties.repeat_key_expression.clone()
            });

        let mut deps = vec![
            source_expression.untyped(),
            i_symbol.untyped(),
            elem_symbol.untyped(),
        ];
        if let Some(repeat_key_expression) = &repeat_key_expression {
            for dependency in &repeat_key_expression.dependencies {
                if let Some(property) = expanded_node
                    .stack
                    .resolve_symbol_as_erased_property(dependency)
                {
                    deps.push(property);
                }
            }
        }

        let last_length = Rc::new(RefCell::new(0));
        let last_elem_sym = Rc::new(RefCell::new(None));
        let last_i_sym = Rc::new(RefCell::new(None));
        let cached_children: Rc<RefCell<Vec<Rc<ExpandedNode>>>> = Default::default();
        let keyed_groups: Rc<RefCell<Vec<RepeatChildGroup>>> = Default::default();

        let children = LocalProperty::computed_with_name(
            move || {
                let Some(cloned_expanded_node) = weak_ref_self.upgrade() else {
                    panic!("ran evaluator after expanded node dropped (repeat elem)")
                };
                let source_len = source_expression.read(Self::source_len);
                if let Some(repeat_key_expression) = &repeat_key_expression {
                    let i_symbol_value = i_symbol.get();
                    let elem_symbol_value = elem_symbol.get();
                    let symbols_changed = i_symbol_value != *borrow!(last_i_sym)
                        || elem_symbol_value != *borrow!(last_elem_sym);
                    *borrow_mut!(last_i_sym) = i_symbol_value.clone();
                    *borrow_mut!(last_elem_sym) = elem_symbol_value.clone();
                    if symbols_changed {
                        borrow_mut!(keyed_groups).clear();
                    }
                    return Self::compute_keyed_children(
                        &cloned_expanded_node,
                        &cloned_context,
                        Rc::clone(&cloned_self),
                        &source_expression,
                        i_symbol_value,
                        elem_symbol_value,
                        repeat_key_expression,
                        Rc::clone(&keyed_groups),
                        source_len,
                        is_mount,
                        !symbols_changed,
                    );
                }

                if source_len == *borrow!(last_length)
                    && i_symbol.read(|i| i == &*borrow!(last_i_sym))
                    && elem_symbol.read(|e| e == &*borrow!(last_elem_sym))
                {
                    return if cloned_expanded_node.attached.get() > 0 {
                        cloned_expanded_node.current_attached_children()
                    } else {
                        borrow!(cached_children).clone()
                    };
                }
                let i_symbol_value = i_symbol.get();
                let elem_symbol_value = elem_symbol.get();
                let symbols_unchanged = i_symbol_value == *borrow!(last_i_sym)
                    && elem_symbol_value == *borrow!(last_elem_sym);
                *borrow_mut!(last_length) = source_len;
                *borrow_mut!(last_i_sym) = i_symbol_value.clone();
                *borrow_mut!(last_elem_sym) = elem_symbol_value.clone();

                let template_children = borrow!(cloned_self.base().get_instance_children()).clone();
                let active_children = borrow!(cloned_expanded_node.active_children).clone();
                let mut selected_children = Vec::new();
                for i in 0..source_len {
                    let property_i = LocalProperty::new(i);
                    let cp_source_expression = source_expression.clone();
                    let property_elem = LocalProperty::computed_with_name(
                        move || cp_source_expression.read(|source| Self::source_elem(source, i)),
                        &[source_expression.untyped()],
                        "repeat elem",
                    );
                    let scope = Self::repeat_scope(
                        i_symbol_value.clone(),
                        elem_symbol_value.clone(),
                        property_i,
                        property_elem,
                    );
                    let new_env = cloned_expanded_node.stack.push(scope);
                    for (template_offset, template) in template_children.iter().enumerate() {
                        let flat_index = i * template_children.len() + template_offset;
                        let active = symbols_unchanged
                            .then(|| active_children.get(flat_index))
                            .flatten()
                            .filter(|child| {
                                let child_template = borrow!(child.instance_node).clone();
                                Rc::ptr_eq(&child_template, template)
                            })
                            .cloned();
                        let rescued = active.or_else(|| {
                            (symbols_unchanged
                                && is_mount
                                && cloned_expanded_node.attached.get() > 0)
                                .then(|| {
                                    cloned_expanded_node.rescue_exiting_child_matching(|child| {
                                        let child_template = borrow!(child.instance_node).clone();
                                        let template_matches =
                                            Rc::ptr_eq(&child_template, template);
                                        let index_matches = child
                                            .stack
                                            .resolve_symbol_as_erased_property(
                                                INTERNAL_REPEAT_INDEX_SYMBOL,
                                            )
                                            .map(LocalProperty::<usize>::new_from_untyped)
                                            .map(|index| index.get() == i)
                                            .unwrap_or(true);
                                        template_matches && index_matches
                                    })
                                })
                                .flatten()
                        });
                        let child = rescued.unwrap_or_else(|| {
                            cloned_expanded_node
                                .create_children_detached(
                                    iter::once((Rc::clone(template), Rc::clone(&new_env))),
                                    &cloned_context,
                                    &Rc::downgrade(&cloned_expanded_node),
                                )
                                .pop()
                                .expect("repeat child creation returned no child")
                        });
                        if !is_mount && child.attached.get() == 0 {
                            child.recurse_control_flow_expansion(&cloned_context);
                        }
                        selected_children.push(child);
                    }
                }
                let ret = if is_mount {
                    cloned_expanded_node.attach_children(
                        selected_children,
                        &cloned_context,
                        &cloned_expanded_node.parent_frame,
                    )
                } else {
                    selected_children
                };
                *borrow_mut!(cached_children) = ret.clone();
                ret
            },
            &deps,
            &format!("repeat_children (node id: {})", expanded_node.id.0),
        );
        expanded_node.children.replace_with(children);
    }

    fn source_len(source: &PaxValue) -> usize {
        if let PaxValue::Range(start, end) = source {
            (isize::try_coerce(*end.clone()).unwrap() - isize::try_coerce(*start.clone()).unwrap())
                as usize
        } else if let PaxValue::Vec(v) = source {
            v.len()
        } else {
            log::warn!("source is not a vec");
            0
        }
    }

    fn source_elem(source: &PaxValue, i: usize) -> PaxValue {
        if let PaxValue::Range(start, _) = source {
            let start = isize::try_coerce(*start.clone()).unwrap();
            (start + i as isize).to_pax_value()
        } else if let PaxValue::Vec(v) = source {
            v.get(i).cloned().unwrap_or_default()
        } else {
            log::warn!("source is not a vec");
            Default::default()
        }
    }

    fn repeat_scope(
        i_symbol: Option<String>,
        elem_symbol: Option<String>,
        property_i: LocalProperty<usize>,
        property_elem: LocalProperty<PaxValue>,
    ) -> HashMap<String, Variable> {
        let mut scope = HashMap::new();
        if let Some(i_symbol) = i_symbol {
            scope.insert(
                i_symbol,
                Variable::new_from_typed_property(property_i.clone()),
            );
        }
        if let Some(elem_symbol) = elem_symbol {
            scope.insert(
                elem_symbol,
                Variable::new_from_typed_property(property_elem.clone()),
            );
        }
        scope.insert(
            INTERNAL_REPEAT_INDEX_SYMBOL.to_string(),
            Variable::new_from_typed_property(property_i),
        );
        scope.insert(
            INTERNAL_REPEAT_ELEMENT_SYMBOL.to_string(),
            Variable::new_from_typed_property(property_elem),
        );
        scope
    }

    fn encoded_repeat_key(key: &RepeatKey) -> String {
        match key {
            RepeatKey::String(value) => format!("string:{value}"),
            RepeatKey::Int(value) => format!("int:{value}"),
            RepeatKey::Synthetic(value) => format!("synthetic:{value}"),
        }
    }

    fn repeat_scope_with_key(
        i_symbol: Option<String>,
        elem_symbol: Option<String>,
        property_i: LocalProperty<usize>,
        property_elem: LocalProperty<PaxValue>,
        key: &RepeatKey,
    ) -> HashMap<String, Variable> {
        let mut scope = Self::repeat_scope(i_symbol, elem_symbol, property_i, property_elem);
        scope.insert(
            INTERNAL_REPEAT_KEY_SYMBOL.to_string(),
            Variable::new_from_typed_property(LocalProperty::new(Self::encoded_repeat_key(key))),
        );
        scope
    }

    fn repeat_key_from_value(value: PaxValue, index: usize) -> Option<RepeatKey> {
        match value {
            PaxValue::String(value) => Some(RepeatKey::String(value)),
            PaxValue::Numeric(value) if !value.is_float() => Some(RepeatKey::Int(value.to_int())),
            _ => {
                log::warn!(
                    "repeat key for index {} must evaluate to a string or integer; using positional fallback",
                    index
                );
                None
            }
        }
    }

    fn evaluate_key(
        key_expression: &ExpressionInfo,
        env: Rc<RuntimePropertiesStackFrame>,
        index: usize,
    ) -> Option<RepeatKey> {
        key_expression
            .expression
            .compute(env)
            .map(|value| Self::repeat_key_from_value(value, index))
            .unwrap_or_else(|err| {
                log::warn!(
                    "failed to compute repeat key for index {}: {}; using positional fallback",
                    index,
                    err
                );
                None
            })
    }

    fn compute_keyed_children(
        expanded_node: &Rc<ExpandedNode>,
        context: &Rc<RuntimeContext>,
        repeat: Rc<RepeatInstance>,
        source_expression: &LocalProperty<PaxValue>,
        i_symbol_value: Option<String>,
        elem_symbol_value: Option<String>,
        key_expression: &ExpressionInfo,
        keyed_groups: Rc<RefCell<Vec<RepeatChildGroup>>>,
        source_len: usize,
        is_mount: bool,
        allow_rescue: bool,
    ) -> Vec<Rc<ExpandedNode>> {
        let mut old_groups_by_key: HashMap<RepeatKey, RepeatChildGroup> = borrow_mut!(keyed_groups)
            .drain(..)
            .map(|group| (group.key.clone(), group))
            .collect();
        let mut new_groups = Vec::new();
        let mut new_children = Vec::new();
        let mut used_keys = HashSet::new();
        let template_children = repeat.base().get_instance_children();

        for i in 0..source_len {
            let elem = source_expression.read(|source| Self::source_elem(source, i));
            let key_probe_i = LocalProperty::new(i);
            let key_probe_elem = LocalProperty::new(elem.clone());
            let key_probe_scope = Self::repeat_scope(
                i_symbol_value.clone(),
                elem_symbol_value.clone(),
                key_probe_i,
                key_probe_elem,
            );
            let key_probe_env = expanded_node.stack.push(key_probe_scope);
            let evaluated_key = Self::evaluate_key(key_expression, key_probe_env, i);
            let key = match evaluated_key {
                Some(key) if used_keys.insert(key.clone()) => key,
                Some(key) => {
                    log::warn!(
                        "duplicate repeat key {:?} at index {}; using positional fallback",
                        key,
                        i
                    );
                    RepeatKey::Synthetic(i)
                }
                None => RepeatKey::Synthetic(i),
            };

            let group = if let Some(group) = old_groups_by_key.remove(&key) {
                update_repeat_bindings(&group.i, &group.elem, i, elem);
                group
            } else if allow_rescue && is_mount {
                Self::rescue_keyed_group(
                    expanded_node,
                    context,
                    &template_children,
                    i_symbol_value.clone(),
                    elem_symbol_value.clone(),
                    &key,
                    i,
                    elem.clone(),
                )
                .unwrap_or_else(|| {
                    Self::create_keyed_group(
                        expanded_node,
                        context,
                        &template_children,
                        i_symbol_value.clone(),
                        elem_symbol_value.clone(),
                        key.clone(),
                        i,
                        elem,
                        is_mount,
                    )
                })
            } else {
                Self::create_keyed_group(
                    expanded_node,
                    context,
                    &template_children,
                    i_symbol_value.clone(),
                    elem_symbol_value.clone(),
                    key.clone(),
                    i,
                    elem,
                    is_mount,
                )
            };

            new_children.extend(group.children.iter().cloned());
            new_groups.push(group);
        }

        *borrow_mut!(keyed_groups) = new_groups;
        if is_mount {
            expanded_node.attach_children(new_children, context, &expanded_node.parent_frame)
        } else {
            new_children
        }
    }

    fn create_keyed_group(
        expanded_node: &Rc<ExpandedNode>,
        context: &Rc<RuntimeContext>,
        template_children: &RefCell<Vec<Rc<dyn InstanceNode>>>,
        i_symbol: Option<String>,
        elem_symbol: Option<String>,
        key: RepeatKey,
        i: usize,
        elem: PaxValue,
        is_mount: bool,
    ) -> RepeatChildGroup {
        let property_i = LocalProperty::new(i);
        let property_elem = LocalProperty::new(elem);
        let scope = Self::repeat_scope_with_key(
            i_symbol,
            elem_symbol,
            property_i.clone(),
            property_elem.clone(),
            &key,
        );
        let new_env = expanded_node.stack.push(scope);
        let children = expanded_node.create_children_detached(
            borrow!(template_children)
                .clone()
                .into_iter()
                .zip(iter::repeat(new_env)),
            context,
            &Rc::downgrade(expanded_node),
        );
        if !is_mount {
            for child in &children {
                child.recurse_control_flow_expansion(context);
            }
        }
        RepeatChildGroup {
            key,
            elem: property_elem,
            i: property_i,
            children,
        }
    }

    fn rescue_keyed_group(
        expanded_node: &Rc<ExpandedNode>,
        context: &Rc<RuntimeContext>,
        template_children: &RefCell<Vec<Rc<dyn InstanceNode>>>,
        i_symbol: Option<String>,
        elem_symbol: Option<String>,
        key: &RepeatKey,
        i: usize,
        elem: PaxValue,
    ) -> Option<RepeatChildGroup> {
        let encoded_key = Self::encoded_repeat_key(key);
        let templates = borrow!(template_children).clone();
        let mut children = templates
            .iter()
            .map(|template| {
                expanded_node.rescue_exiting_child_matching(|child| {
                    let child_template = borrow!(child.instance_node).clone();
                    Rc::ptr_eq(&child_template, template)
                        && child
                            .stack
                            .resolve_symbol_as_erased_property(INTERNAL_REPEAT_KEY_SYMBOL)
                            .map(LocalProperty::<String>::new_from_untyped)
                            .map(|key| key.get() == encoded_key)
                            .unwrap_or(false)
                })
            })
            .collect::<Vec<_>>();
        let seed = children.iter().flatten().next()?.clone();
        let property_i = seed
            .stack
            .resolve_symbol_as_erased_property(INTERNAL_REPEAT_INDEX_SYMBOL)
            .map(LocalProperty::<usize>::new_from_untyped)?;
        let property_elem = seed
            .stack
            .resolve_symbol_as_erased_property(INTERNAL_REPEAT_ELEMENT_SYMBOL)
            .map(LocalProperty::<PaxValue>::new_from_untyped)?;
        update_repeat_bindings(&property_i, &property_elem, i, elem);
        let env = expanded_node.stack.push(Self::repeat_scope_with_key(
            i_symbol,
            elem_symbol,
            property_i.clone(),
            property_elem.clone(),
            key,
        ));
        let resolved_children = children
            .drain(..)
            .zip(templates)
            .map(|(child, template)| {
                child.unwrap_or_else(|| {
                    expanded_node
                        .create_children_detached(
                            iter::once((template, Rc::clone(&env))),
                            context,
                            &Rc::downgrade(expanded_node),
                        )
                        .pop()
                        .expect("repeat child creation returned no child")
                })
            })
            .collect();
        Some(RepeatChildGroup {
            key: key.clone(),
            elem: property_elem,
            i: property_i,
            children: resolved_children,
        })
    }
}