darkfi/bin/zkas/src/main.rs

use std::{
    fs::{read_to_string, File},
    io::Write,
};

use clap::Parser as ClapParser;

use darkfi::{
    zkas::{
        analyzer::Analyzer, compiler::Compiler, decoder::ZkBinary, lexer::Lexer, parser::Parser,
    },
    Result,
};

#[derive(clap::Parser)]
#[clap(name = "zkas", version)]
struct Args {
    /// Place the output into <FILE>
    #[clap(short, value_name = "FILE")]
    output: Option<String>,

    /// Strip debug symbols
    #[clap(short)]
    strip: bool,

    /// Preprocess only; do not compile
    #[clap(short)]
    evaluate: bool,

    /// Interactive semantic analysis
    #[clap(short)]
    interactive: bool,

    /// Examine decoded bytecode
    #[clap(long)]
    examine: bool,

    /// ZK script to compile
    input: String,
}

fn main() -> Result<()> {
    let args = Args::parse();

    let filename = args.input.as_str();
    let source = read_to_string(filename)?;

    let lexer = Lexer::new(filename, source.chars());
    let tokens = lexer.lex();

    let parser = Parser::new(filename, source.chars(), tokens);
    let (constants, witnesses, statements) = parser.parse();

    let mut analyzer = Analyzer::new(filename, source.chars(), constants, witnesses, statements);
    analyzer.analyze_types();

    if args.interactive {
        analyzer.analyze_semantic();
    }

    if args.evaluate {
        println!("{:#?}", analyzer.constants);
        println!("{:#?}", analyzer.witnesses);
        println!("{:#?}", analyzer.statements);
        println!("{:#?}", analyzer.stack);
        return Ok(())
    }

    let compiler = Compiler::new(
        filename,
        source.chars(),
        analyzer.constants,
        analyzer.witnesses,
        analyzer.statements,
        !args.strip,
    );

    let bincode = compiler.compile();

    let output = match args.output {
        Some(o) => o,
        None => format!("{}.bin", args.input),
    };

    let mut file = File::create(&output)?;
    file.write_all(&bincode)?;
    println!("Wrote output to {}", &output);

    if args.examine {
        let zkbin = ZkBinary::decode(&bincode)?;
        println!("{:#?}", zkbin);
    }

    Ok(())
}