slint/tests/rustdriver/build.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 std::io::Write;
use std::path::Path;

fn main() -> std::io::Result<()> {
    let mut generated_file = std::fs::File::create(
        Path::new(&std::env::var_os("OUT_DIR").unwrap()).join("generated.rs"),
    )?;

    let mut test_dirs = std::collections::HashSet::new();

    for testcase in test_driver_lib::collect_test_cases()? {
        println!("cargo:rerun-if-changed={}", testcase.absolute_path.to_string_lossy());

        test_dirs.insert(testcase.absolute_path.parent().unwrap().to_owned());

        let module_name =
            testcase.relative_path.file_stem().unwrap().to_string_lossy().replace("/", "_");
        write!(generated_file, "#[path=\"{0}.rs\"] mod r#{0};\n", module_name)?;

        let source = std::fs::read_to_string(&testcase.absolute_path)?;
        let mut output = std::fs::File::create(
            Path::new(&std::env::var_os("OUT_DIR").unwrap()).join(format!("{}.rs", module_name)),
        )?;

        let include_paths = test_driver_lib::extract_include_paths(&source);

        output.write_all(b"sixtyfps::sixtyfps!{")?;

        for path in include_paths {
            let mut abs_path = testcase.absolute_path.clone();
            abs_path.pop();
            abs_path.push(path);

            output.write_all(b"#[include_path\r=r#\"")?;
            output.write_all(abs_path.to_string_lossy().as_bytes())?;
            output.write_all(b"\"#]")?;
            println!("cargo:rerun-if-changed={}", abs_path.to_string_lossy());
        }

        output.write_all(b"\n")?;
        output.write_all(source.as_bytes())?;
        output.write_all(b"}\n")?;

        for (i, x) in test_driver_lib::extract_test_functions(&source)
            .filter(|x| x.language_id == "rust")
            .enumerate()
        {
            write!(
                output,
                "#[test] fn t_{}() -> Result<(), Box<dyn std::error::Error>> {{\n    {}\n    Ok(()) }}",
                i,
                x.source.replace('\n', "\n    ")
            )?;
        }
    }

    test_dirs.iter().for_each(|dir| {
        println!("cargo:rerun-if-changed={}", dir.to_string_lossy());
    });

    Ok(())
}