slint/sixtyfps_compiler/passes/embed_images.rs

/* LICENSE BEGIN
    This file is part of the SixtyFPS Project -- https://sixtyfps.io
    Copyright (c) 2021 Olivier Goffart <olivier.goffart@sixtyfps.io>
    Copyright (c) 2021 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::diagnostics::BuildDiagnostics;
use crate::expression_tree::{Expression, ImageReference};
use crate::object_tree::*;
use std::cell::RefCell;
use std::collections::HashMap;
use std::rc::Rc;

pub fn embed_images(
    component: &Rc<Component>,
    embed_files_by_default: bool,
    diag: &mut BuildDiagnostics,
) {
    let global_embedded_resources = &component.embedded_file_resources;

    for component in
        component.used_types.borrow().sub_components.iter().chain(std::iter::once(component))
    {
        visit_all_expressions(component, |e, _| {
            embed_images_from_expression(e, global_embedded_resources, embed_files_by_default, diag)
        });
    }
}

fn embed_images_from_expression(
    e: &mut Expression,
    global_embedded_resources: &RefCell<HashMap<String, usize>>,
    embed_files_by_default: bool,
    diag: &mut BuildDiagnostics,
) {
    if let Expression::ImageReference { ref mut resource_ref, source_location } = e {
        match resource_ref {
            ImageReference::AbsolutePath(path)
                if embed_files_by_default || path.starts_with("builtin:/") =>
            {
                // Check that the file exists, so that later we can unwrap safely in the generators, etc.
                if crate::fileaccess::load_file(std::path::Path::new(path)).is_some() {
                    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 = ImageReference::EmbeddedData {
                        resource_id,
                        extension: std::path::Path::new(path)
                            .extension()
                            .and_then(|e| e.to_str())
                            .map(|x| x.to_string())
                            .unwrap_or_default(),
                    }
                } else {
                    diag.push_warning(format!("Cannot find image file {}", path), source_location);
                    *resource_ref = ImageReference::None;
                }
            }
            _ => {}
        }
    };

    e.visit_mut(|mut e| {
        embed_images_from_expression(
            &mut e,
            global_embedded_resources,
            embed_files_by_default,
            diag,
        )
    });
}