use eyre::Result;
use tinywasm::{Module, Store};

// WebAssembly module defining and exporting an `add` function.
const WASM_ADD: &str = r#"
(module
  (func $add (param $lhs i32) (param $rhs i32) (result i32)
    local.get $lhs
    local.get $rhs
    i32.add)
  (export "add" (func $add)))
"#;

// WebAssembly module importing an `add` function and using it.
const WASM_IMPORT: &str = r#"
(module
  (import "adder" "add" (func $add (param i32 i32) (result i32)))
  (func $main (result i32)
    i32.const 1
    i32.const 2
    call $add)
  (export "main" (func $main))
)
"#;

fn main() -> Result<()> {
    let wasm_add = wat::parse_str(WASM_ADD).expect("failed to parse wat");
    let wasm_import = wat::parse_str(WASM_IMPORT).expect("failed to parse wat");

    let add_module = Module::parse_bytes(&wasm_add)?;
    let import_module = Module::parse_bytes(&wasm_import)?;

    let mut store = Store::default();

    // Instantiate the `add` module.
    let add_instance = add_module.instantiate(&mut store, None)?;

    // Link the `adder` namespace to the `add` module's instance.
    let mut imports = tinywasm::Imports::new();
    imports.link_module("adder", add_instance)?;

    // Instantiate the `import` module with the linked imports.
    let import_instance = import_module.instantiate(&mut store, Some(imports))?;

    // Call the `main` function, which uses the imported `add` function.
    let main = import_instance.func_typed::<(), i32>(&store, "main")?;
    assert_eq!(main.call(&mut store, ())?, 3);

    Ok(())
}