use std::fs::File;
use std::io::Read;

use clap::{App, Arg};

use prov_cosmwasm_vm::features_from_csv;
use prov_cosmwasm_vm::internals::{check_wasm, compile};

const DEFAULT_SUPPORTED_FEATURES: &str = "iterator,staking,stargate";

pub fn main() {
    let matches = App::new("Contract checking")
        .version("0.1.0")
        .long_about("Checks the given wasm file (memories, exports, imports, supported features, and non-determinism).")
        .author("Mauro Lacy <mauro@lacy.com.es>")
        .arg(
            Arg::with_name("FEATURES")
                // `long` setting required to turn the position argument into an option 🤷
                .long("supported-features")
                .value_name("FEATURES")
                .help("Sets the supported features that the desired target chain supports")
                .takes_value(true)
        )
        .arg(
            Arg::with_name("WASM")
                .help("Wasm file to read and compile")
                .required(true)
                .index(1),
        )
        .get_matches();

    // Supported features
    let supported_features_csv = matches
        .value_of("FEATURES")
        .unwrap_or(DEFAULT_SUPPORTED_FEATURES);
    let supported_features = features_from_csv(supported_features_csv);
    println!("Supported features: {:?}", supported_features);

    // File
    let path = matches.value_of("WASM").expect("Error parsing file name");
    let mut file = File::open(path).unwrap();

    // Read wasm
    let mut wasm = Vec::<u8>::new();
    file.read_to_end(&mut wasm).unwrap();

    // Check wasm
    check_wasm(&wasm, &supported_features).unwrap();

    // Compile module
    compile(&wasm, None, &[]).unwrap();
    println!("contract checks passed.")
}