elko/
build.rs

1//! Command `Build`.
2use crate::utils::WasmBuilder;
3use anyhow::{anyhow, Result};
4use ccli::clap::{self, Parser};
5use etc::{Etc, FileSystem};
6use std::{env, fs, path::PathBuf};
7use zinkc::{Compiler, Config};
8
9/// Build zink project to EVM bytecode.
10#[derive(Debug, Parser)]
11#[command(name = "build", version)]
12pub struct Build {
13    /// The path of the cargo project.
14    pub input: Option<PathBuf>,
15    /// Write output to <filename>
16    #[clap(short, long, value_name = "filename")]
17    pub output: Option<PathBuf>,
18    /// Write output to compiler-chosen filename in <dir>
19    #[clap(long, value_name = "dir")]
20    pub out_dir: Option<PathBuf>,
21    /// Compiler configuration
22    #[clap(flatten)]
23    pub config: Config,
24}
25
26impl Build {
27    /// Run build
28    pub fn run(&self) -> Result<()> {
29        // Get and check the input.
30        let input = if let Some(input) = self.input.as_ref() {
31            input.clone()
32        } else {
33            env::current_dir()?
34        };
35        {
36            if Etc::new(&input)?.find("Cargo.toml").is_err() {
37                return Ok(());
38            }
39
40            if !input.is_dir() {
41                return Err(anyhow!(
42                    "Only support rust project directory as input for now"
43                ));
44            }
45        }
46
47        // Build the wasm.
48        let mut builder = WasmBuilder::new(input)?;
49        {
50            if let Some(out_dir) = self.out_dir.as_ref() {
51                builder.with_out_dir(out_dir);
52            }
53
54            if let Some(output) = self.output.as_ref() {
55                builder.with_output(output);
56            }
57
58            builder.build()?;
59        }
60
61        // Copy the WASM file to target/zink/
62        let wasm_path = builder.output()?;
63        let wasm_dest = wasm_path
64            .with_extension("wasm")
65            .parent()
66            .unwrap()
67            .join("zink")
68            .join(wasm_path.file_name().unwrap());
69        fs::create_dir_all(wasm_dest.parent().unwrap())?;
70        fs::copy(&wasm_path, &wasm_dest)?;
71
72        // Compile the wasm to evm bytecode.
73        let wasm = fs::read(&wasm_path)?;
74        let config = Config::default().dispatcher(self.config.dispatcher);
75        let artifact = Compiler::new(config).compile(&wasm)?;
76        let dst = wasm_path.with_extension("bin");
77
78        fs::write(dst, artifact.runtime_bytecode)?;
79        Ok(())
80    }
81}
elko/build.rs

elko/
build.rs
