cairo-native 0.2.6

A compiler to convert Cairo's intermediate representation Sierra code to MLIR.
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

use anyhow::Context;
use cairo_lang_sierra::program::VersionedProgram;
use cairo_native::context::NativeContext;
use melior::ir::operation::OperationPrintingFlags;
use scarb_metadata::{MetadataCommand, ScarbCommand};
use std::{env, fs};

mod utils;

/// Compiles all packages from a Scarb project on the current directory.
fn main() -> anyhow::Result<()> {
    let metadata = MetadataCommand::new().inherit_stderr().exec()?;

    // Build only the filtered packages.
    ScarbCommand::new().arg("build").run()?;

    // Get `target` directory.
    let profile = env::var("SCARB_PROFILE").unwrap_or("dev".into());
    let default_target_dir = metadata.runtime_manifest.join("target");
    let target_dir = metadata
        .target_dir
        .clone()
        .unwrap_or(default_target_dir)
        .join(profile);

    let native_context = NativeContext::new();
    for package in metadata.packages.iter() {
        for target in &package.targets {
            let file_path = target_dir.join(format!("{}.sierra.json", target.name.clone()));

            if file_path.exists() {
                let compiled = serde_json::from_str::<VersionedProgram>(
                    &fs::read_to_string(file_path.clone())
                        .with_context(|| format!("failed to read file: {file_path}"))?,
                )
                .with_context(|| format!("failed to deserialize compiled file: {file_path}"))?;

                // Compile the sierra program into a MLIR module.
                let native_module = native_context
                    .compile(
                        &compiled.into_v1().unwrap().program,
                        false,
                        Some(Default::default()),
                    )
                    .unwrap();

                // Write the output.
                let output_str = native_module.module().as_operation().to_string_with_flags(
                    OperationPrintingFlags::new().enable_debug_info(true, false),
                )?;

                fs::write(
                    target_dir.join(format!("{}.mlir", target.name.clone())),
                    &output_str,
                )?;
            }
        }
    }

    Ok(())
}