1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

/* 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 std::process::Command;

use anyhow::Result;
use camino::Utf8Path;
use fs_err as fs;

pub mod gen_python;
mod test;
use super::super::interface::ComponentInterface;
pub use gen_python::{generate_python_bindings, Config};
pub use test::{run_script, run_test};

// Generate python bindings for the given ComponentInterface, in the given output directory.
pub fn write_bindings(
    config: &Config,
    ci: &ComponentInterface,
    out_dir: &Utf8Path,
    try_format_code: bool,
) -> Result<()> {
    let py_file = out_dir.join(format!("{}.py", ci.namespace()));
    fs::write(&py_file, generate_python_bindings(config, ci)?)?;

    if try_format_code {
        if let Err(e) = Command::new("yapf").arg(&py_file).output() {
            println!(
                "Warning: Unable to auto-format {} using yapf: {e:?}",
                py_file.file_name().unwrap(),
            )
        }
    }

    Ok(())
}