slint/sixtyfps_compiler/passes/embed_resources.rs

/* LICENSE BEGIN
    This file is part of the SixtyFPS Project -- https://sixtyfps.io
    Copyright (c) 2020 Olivier Goffart <olivier.goffart@sixtyfps.io>
    Copyright (c) 2020 Simon Hausmann <simon.hausmann@sixtyfps.io>

    SPDX-License-Identifier: GPL-3.0-only
    This file is also available under commercial licensing terms.
    Please contact info@sixtyfps.io for more information.
LICENSE END */
use crate::expression_tree::{Expression, ResourceReference};
use crate::object_tree::*;
use std::cell::RefCell;
use std::collections::HashMap;
use std::rc::Rc;

pub fn embed_resources(component: &Rc<Component>) {
    let global_embedded_resources = &component.embedded_file_resources;

    visit_all_expressions(component, |e, _| {
        embed_resources_from_expression(e, component, global_embedded_resources)
    });
}

fn embed_resources_from_expression(
    e: &mut Expression,
    component: &Rc<Component>,
    global_embedded_resources: &RefCell<HashMap<String, usize>>,
) {
    if let Expression::ResourceReference(ref mut resource_ref) = e {
        match resource_ref {
            ResourceReference::AbsolutePath(path) => {
                let mut resources = global_embedded_resources.borrow_mut();
                let maybe_id = resources.len();
                let resource_id = *resources.entry(path.clone()).or_insert(maybe_id);
                *resource_ref = ResourceReference::EmbeddedData(resource_id)
            }
            ResourceReference::EmbeddedData(_) => {}
        }
    };

    e.visit_mut(|mut e| {
        embed_resources_from_expression(&mut e, component, global_embedded_resources)
    });
}