uniffi_bindgen 0.32.0

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

use anyhow::{bail, Result};
use askama::Template;
use camino::Utf8Path;
use fs_err as fs;

use crate::{bindings::GenerateOptions, BindgenLoader};

pub mod filters;
mod pipeline;
pub use pipeline::{pipeline, Root};

#[cfg(feature = "bindgen-tests")]
pub mod test;

/// Generate Python bindings
pub fn generate(loader: &BindgenLoader, options: GenerateOptions) -> Result<()> {
    let metadata = loader.load_metadata(&options.source)?;
    if let Some(crate_filter) = &options.crate_filter {
        if !metadata.contains_key(crate_filter) {
            bail!("No UniFFI metadata found for crate {crate_filter}");
        }
    }
    let root = loader.load_pipeline_initial_root(&options.source, metadata)?;
    run_pipeline(root, &options.out_dir, options.crate_filter.as_deref())?;

    Ok(())
}

pub fn run_pipeline(
    initial_root: pipeline::initial::Root,
    out_dir: &Utf8Path,
    crate_filter: Option<&str>,
) -> Result<()> {
    let python_root = pipeline().execute(initial_root)?;
    println!("writing out {out_dir}");
    if !out_dir.exists() {
        fs::create_dir_all(out_dir)?;
    }
    for module in python_root.modules.values() {
        if let Some(crate_filter) = crate_filter {
            if module.crate_name != crate_filter {
                continue;
            }
        }
        let path = out_dir.join(format!("{}.py", module.name));
        let content = module.render()?;
        println!("writing {path}");
        fs::write(path, content)?;
    }
    Ok(())
}