use crate::expression_tree::Expression;
use crate::object_tree::*;
use std::rc::Rc;

pub fn collect_resources(component: &Rc<Component>) {
    recurse_elem(&component.root_element, &(), &mut |elem, _| {
        let bindings = &elem.borrow().bindings;
        for e in bindings.values() {
            collect_resources_from_expression(&e, component);
        }
        if let Some(e) = &elem.borrow().repeated {
            collect_resources_from_expression(&e.model, component);
        }
    })
}

fn collect_resources_from_expression(e: &Expression, component: &Rc<Component>) {
    match e {
        Expression::ResourceReference { absolute_source_path } => {
            let mut resources = component.embedded_file_resources.borrow_mut();
            let maybe_id = resources.len();
            resources.entry(absolute_source_path.clone()).or_insert(maybe_id);
        }
        _ => {}
    };

    e.visit(|e| collect_resources_from_expression(e, component));
}