//! Translation of multi example

use anyhow::{ensure, format_err, Context as _, Result};
use std::rc::Rc;
use wasmtime::*;

struct Callback;

impl Callable for Callback {
    fn call(&self, args: &[Val], results: &mut [Val]) -> Result<(), Trap> {
        println!("Calling back...");
        println!("> {} {}", args[0].unwrap_i32(), args[1].unwrap_i64());

        results[0] = Val::I64(args[1].unwrap_i64() + 1);
        results[1] = Val::I32(args[0].unwrap_i32() + 1);
        Ok(())
    }
}

const WAT: &str = r#"
(module
  (func $f (import "" "f") (param i32 i64) (result i64 i32))

  (func $g (export "g") (param i32 i64) (result i64 i32)
    (call $f (local.get 0) (local.get 1))
  )

  (func $round_trip_many
        (export "round_trip_many")
        (param i64 i64 i64 i64 i64 i64 i64 i64 i64 i64)
        (result i64 i64 i64 i64 i64 i64 i64 i64 i64 i64)
    local.get 0
    local.get 1
    local.get 2
    local.get 3
    local.get 4
    local.get 5
    local.get 6
    local.get 7
    local.get 8
    local.get 9)
)
"#;

fn main() -> Result<()> {
    // Initialize.
    println!("Initializing...");
    let mut cfg = Config::new();
    cfg.features(wasmtime_jit::Features {
        multi_value: true,
        ..Default::default()
    });
    let engine = HostRef::new(Engine::new(&cfg));
    let store = HostRef::new(Store::new(&engine));

    // Load binary.
    println!("Loading binary...");
    let binary = wat::parse_str(WAT)?;

    // Compile.
    println!("Compiling module...");
    let module = HostRef::new(Module::new(&store, &binary).context("Error compiling module!")?);

    // Create external print functions.
    println!("Creating callback...");
    let callback_type = FuncType::new(
        Box::new([ValType::I32, ValType::I64]),
        Box::new([ValType::I64, ValType::I32]),
    );
    let callback_func = HostRef::new(Func::new(&store, callback_type, Rc::new(Callback)));

    // Instantiate.
    println!("Instantiating module...");
    let imports = vec![callback_func.into()];
    let instance = Instance::new(&store, &module, imports.as_slice())
        .context("Error instantiating module!")?;

    // Extract exports.
    println!("Extracting export...");
    let exports = instance.exports();
    ensure!(!exports.is_empty(), "Error accessing exports!");
    let g = exports[0].func().context("> Error accessing export $g!")?;
    let round_trip_many = exports[1]
        .func()
        .context("> Error accessing export $round_trip_many")?;

    // Call `$g`.
    println!("Calling export \"g\"...");
    let args = vec![Val::I32(1), Val::I64(3)];
    let results = g
        .borrow()
        .call(&args)
        .map_err(|e| format_err!("> Error calling g! {:?}", e))?;

    println!("Printing result...");
    println!("> {} {}", results[0].unwrap_i64(), results[1].unwrap_i32());

    debug_assert_eq!(results[0].unwrap_i64(), 4);
    debug_assert_eq!(results[1].unwrap_i32(), 2);

    // Call `$round_trip_many`.
    println!("Calling export \"round_trip_many\"...");
    let args = vec![
        Val::I64(0),
        Val::I64(1),
        Val::I64(2),
        Val::I64(3),
        Val::I64(4),
        Val::I64(5),
        Val::I64(6),
        Val::I64(7),
        Val::I64(8),
        Val::I64(9),
    ];
    let results = round_trip_many
        .borrow()
        .call(&args)
        .map_err(|e| format_err!("> Error calling round_trip_many! {:?}", e))?;

    println!("Printing result...");
    print!(">");
    for r in results.iter() {
        print!(" {}", r.unwrap_i64());
    }
    println!();

    debug_assert_eq!(results.len(), 10);
    debug_assert!(args
        .iter()
        .zip(results.iter())
        .all(|(a, r)| a.i64() == r.i64()));

    // Shut down.
    println!("Shutting down...");
    drop(store);

    // All done.
    println!("Done.");
    Ok(())
}