rustledger-importer 0.16.2

//! Host loader for WASM-implemented importers (wave 2.3b).
//!
//! A [`WasmImporter`] wraps a `.wasm` module and implements the
//! [`crate::Importer`] trait by serializing inputs to `MessagePack`,
//! calling into the module via wasmtime, and deserializing outputs.
//!
//! # Sandbox model
//!
//! Mirrors the existing directive-plugin runtime in
//! `rustledger-plugin/src/runtime.rs`:
//!
//! - No imports allowed (rejected at load time)
//! - No WASI / filesystem / network / env / syscalls
//! - Memory limit enforced (default 256 MiB)
//! - Fuel-based execution time limit (default 30 s)
//!
//! The host reads the source file into memory and passes the bytes
//! via [`ImporterInput::content`]; the WASM importer never opens the
//! file itself.
//!
//! # Required WASM exports
//!
//! A WASM importer module must export:
//!
//! - `memory` — the standard linear memory
//! - `alloc(size: u32) -> u32` — allocates `size` bytes, returns pointer
//! - `metadata() -> u64` — packed `(ptr << 32) | len` of `MessagePack`
//!   [`MetadataOutput`]. Called once at load.
//! - `identify(ptr: u32, len: u32) -> u64` — input is msgpack
//!   [`IdentifyInput`], output is msgpack [`IdentifyOutput`].
//! - `extract(ptr: u32, len: u32) -> u64` — input is msgpack
//!   [`ImporterInput`], output is msgpack [`ImporterOutput`].
//! - `extract_enriched(ptr: u32, len: u32) -> u64` — input is msgpack
//!   [`ImporterInput`], output is msgpack [`EnrichedImporterOutput`].

use std::path::{Path, PathBuf};
use std::sync::Arc;

use rustledger_ops::fingerprint::Fingerprint;
use rustledger_plugin::sandbox::{self, StoreState};
use rustledger_plugin_types::{
    EnrichedImporterOutput, IdentifyInput, IdentifyOutput, ImporterInput, ImporterOutput,
    MetadataOutput, PluginError, PluginErrorSeverity,
};
use serde::{Serialize, de::DeserializeOwned};
use wasmtime::{Engine, Linker, Module, Store};

use crate::config::{CsvConfig, ImporterType};
use crate::{EnrichedImportResult, ImportResult, Importer, ImporterConfig};

// NOTE on hardcoded caps below: `MAX_OUTPUT_BYTES` and `MAX_INPUT_BYTES`
// are per-process constants, not per-importer config. They're sized
// generously (64 MiB each) for any realistic bank-statement import.
// Per-importer tunability is a v1.0 surface decision; for v0.16-pre the
// caps are intentionally fixed so the security contract is uniform
// across all loaded importers regardless of who configured them.

/// Hard cap on the byte length a WASM importer can return from any
/// entry point. Prevents a malicious or buggy module from triggering a
/// 4 GiB host allocation by returning `(any_ptr, u32::MAX)`. 64 MiB is
/// well above any realistic importer output for a single statement.
const MAX_OUTPUT_BYTES: usize = 64 * 1024 * 1024;

/// Hard cap on the byte length of input the host will marshal into the
/// WASM module. Mirrors `MAX_OUTPUT_BYTES` on the input side:
/// `wasm32` memory is `u32`-addressed, so anything over 4 GiB is
/// fundamentally not addressable, but we cap much lower to avoid
/// runaway allocations from accidentally-huge source files.
const MAX_INPUT_BYTES: usize = 64 * 1024 * 1024;

/// Configuration for the WASM importer runtime.
#[derive(Debug, Clone, Copy)]
pub struct WasmRuntimeConfig {
    /// Maximum memory in bytes (default 256 MiB).
    pub max_memory: usize,
    /// Maximum execution time in seconds (default 30). Converted to a
    /// fuel budget at roughly 1M instructions per second.
    pub max_time_secs: u64,
}

impl Default for WasmRuntimeConfig {
    fn default() -> Self {
        Self {
            // Both fields are aliases of the workspace-wide sandbox
            // defaults so the importer, the regular plugin path, and
            // the Python plugin runtime (memory only -- Python opts
            // out of the time default) can't drift apart silently.
            max_memory: sandbox::DEFAULT_SANDBOX_MAX_MEMORY,
            max_time_secs: sandbox::DEFAULT_SANDBOX_MAX_TIME_SECS,
        }
    }
}

/// Errors that can occur loading or invoking a WASM importer.
#[derive(Debug, thiserror::Error)]
pub enum WasmImporterError {
    /// Failed to read the `.wasm` file from disk.
    #[error("failed to read WASM file {path}: {source}")]
    Io {
        /// Path the host tried to read.
        path: PathBuf,
        /// Underlying I/O error.
        source: std::io::Error,
    },
    /// Failed to enumerate an entry while scanning a directory for
    /// `.wasm` files. Distinct from [`Self::Io`] because the entry's
    /// name is unknown when read fails — only the dir is named.
    /// Typically permission-denied on a single inode or a broken
    /// symlink.
    #[error("failed to enumerate entry in WASM importer directory {dir}: {source}")]
    DirEntry {
        /// Directory being scanned.
        dir: PathBuf,
        /// Underlying I/O error from `read_dir().next()`.
        source: std::io::Error,
    },
    /// The WASM module is malformed or uses unsupported features.
    #[error("failed to compile WASM module {path}: {source}")]
    Compile {
        /// Path of the module that failed to compile.
        path: PathBuf,
        /// Underlying wasmtime compile error.
        source: anyhow::Error,
    },
    /// The WASM module has imports — they're forbidden in the importer
    /// sandbox. Importers must be self-contained.
    #[error(
        "WASM importer has forbidden import {module}::{name} — importers must be self-contained"
    )]
    ForbiddenImport {
        /// Import module namespace (e.g. `env`, `wasi_snapshot_preview1`).
        module: String,
        /// Import item name within the module.
        name: String,
    },
    /// A required export is missing.
    #[error("WASM importer missing required export `{0}`")]
    MissingExport(&'static str),
    /// Runtime error during a wasmtime call (trap, fuel exhausted,
    /// memory limit, etc.).
    #[error("WASM importer runtime error: {0}")]
    Runtime(#[source] anyhow::Error),
    /// `MessagePack` decode error on the WASM-returned bytes.
    #[error("WASM importer returned malformed MessagePack: {0}")]
    Decode(#[source] rmp_serde::decode::Error),
    /// `MessagePack` encode error on the input being sent to the WASM
    /// importer. Practically only happens if `ImporterConfig` carries
    /// non-serializable state, which shouldn't.
    #[error("failed to encode input for WASM importer: {0}")]
    Encode(#[source] rmp_serde::encode::Error),
    /// The WASM importer returned an `out_len` larger than the host's
    /// allocation cap (`MAX_OUTPUT_BYTES`, currently 64 MiB). Either
    /// the module is buggy/malicious or the cap needs raising for a
    /// genuinely huge import.
    #[error("WASM importer returned output of {len} bytes, exceeds cap of {max} bytes")]
    OutputTooLarge {
        /// Length the module reported.
        len: usize,
        /// Host's enforced cap (`MAX_OUTPUT_BYTES`).
        max: usize,
    },
    /// The input the host tried to marshal exceeds the host's input
    /// cap (`MAX_INPUT_BYTES`, currently 64 MiB). The host caps
    /// before a lossy `as u32` cast (wasm32 memory is `u32`-addressed,
    /// so >4 GiB input would silently truncate).
    #[error("input of {len} bytes exceeds cap of {max} bytes for WASM importer")]
    InputTooLarge {
        /// Length the host attempted to send.
        len: usize,
        /// Host's enforced cap (`MAX_INPUT_BYTES`).
        max: usize,
    },
    /// A required export exists but has the wrong signature. Distinct
    /// from [`Self::MissingExport`] because `validate_module` already
    /// proved presence at load time — a `get_typed_func` failure
    /// thereafter is always a type mismatch, not absence.
    #[error("WASM importer export `{name}` has wrong signature: {source}")]
    ExportSignatureMismatch {
        /// Name of the export.
        name: &'static str,
        /// Underlying wasmtime type-mismatch error.
        source: anyhow::Error,
    },
    /// The importer does not advertise an ABI version (no
    /// `__rustledger_abi_version` export). It was built without the
    /// `wasm_importer_main!` macro or against a `plugin-types` from
    /// before the ABI handshake existed; the host can't confirm wire
    /// compatibility and refuses to run it (issue #1234).
    #[error(
        "WASM importer has a missing or invalid `{export}` export (expected signature \
         `() -> u32`): it was built against an incompatible rustledger-plugin-types, or the \
         export is absent, mistyped, or traps. Host requires ABI v{expected}. Rebuild against \
         a matching rustledger-plugin-types."
    )]
    AbiVersionMissing {
        /// The export symbol the host looked up.
        export: &'static str,
        /// ABI version the host speaks.
        expected: u32,
    },
    /// The importer advertises a different ABI version than the host.
    /// Running it would risk an opaque trap from a misread wire
    /// message, so the host rejects it at load (issue #1234).
    #[error(
        "WASM importer ABI version mismatch: importer declares v{found}, host requires \
         v{expected}. Rebuild against a matching rustledger-plugin-types."
    )]
    AbiVersionMismatch {
        /// Version the importer reported.
        found: u32,
        /// Version the host requires.
        expected: u32,
    },
}

// `MemoryLimiter`, `StoreState`, `MAX_TABLE_ELEMENTS`, and the
// `make_sandboxed_store` helper live in `rustledger_plugin::sandbox`
// so the per-call enforcement is identical between the WASM importer
// host and the directive-plugin runtime. See sandbox.rs for the
// rationale + tests.

// Note: no manual `impl From<WasmImporterError> for anyhow::Error` — `anyhow`
// has a blanket impl for any `std::error::Error + Send + Sync + 'static`,
// which thiserror's derive already satisfies. Adding our own would conflict.

/// Wrap a `wasmtime::Error` in `WasmImporterError::Runtime`. Function form
/// (not closure) so call sites stay terse: `.map_err(runtime_err)`.
#[inline]
fn runtime_err(e: wasmtime::Error) -> WasmImporterError {
    WasmImporterError::Runtime(anyhow::Error::from(e))
}

/// A WASM-loaded importer. Implements [`Importer`] by dispatching to
/// the loaded module's `extract` / `extract_enriched` entry points.
///
/// Cheap to clone — the [`Module`] is shared via `Arc` and the
/// [`Engine`] is process-wide (see [`rustledger_plugin::sandbox`]).
/// A fresh wasmtime [`Store`] is created per call, so concurrent
/// extract calls don't share state.
#[derive(Clone)]
pub struct WasmImporter {
    /// Filesystem path the module was loaded from (for diagnostics).
    path: PathBuf,
    /// Module's declared name (from the cached `metadata` call).
    name: String,
    /// Module's declared description (from the cached `metadata` call).
    description: String,
    /// Compiled module.
    module: Arc<Module>,
    /// Shared wasmtime engine — one per process, sourced from the
    /// workspace's shared sandbox config in `rustledger_plugin`.
    engine: Arc<Engine>,
    /// Per-call runtime limits.
    config: WasmRuntimeConfig,
}

impl std::fmt::Debug for WasmImporter {
    /// Hand-rolled to avoid wasmtime's `Module`/`Engine` (whose `Debug`
    /// outputs are noisy and version-dependent). Prints just the
    /// host-side metadata that's useful for assertions and logging.
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("WasmImporter")
            .field("path", &self.path)
            .field("name", &self.name)
            .field("description", &self.description)
            .field("config", &self.config)
            .finish_non_exhaustive()
    }
}

impl WasmImporter {
    /// Load a WASM importer from a `.wasm` file with default runtime
    /// limits.
    pub fn load(path: impl Into<PathBuf>) -> Result<Self, WasmImporterError> {
        Self::load_with_config(path, WasmRuntimeConfig::default())
    }

    /// Load a WASM importer with custom runtime limits.
    pub fn load_with_config(
        path: impl Into<PathBuf>,
        config: WasmRuntimeConfig,
    ) -> Result<Self, WasmImporterError> {
        let path = path.into();
        let bytes = std::fs::read(&path).map_err(|source| WasmImporterError::Io {
            path: path.clone(),
            source,
        })?;
        Self::load_from_bytes(path, &bytes, config)
    }

    /// Load a WASM importer from in-memory bytes that aren't backed by
    /// a real file. Use this when shipping `.wasm` modules embedded in
    /// a binary or generated at runtime — `name_for_diagnostics`
    /// surfaces in error messages and [`Self::path`] but doesn't have
    /// to correspond to anything on disk.
    pub fn load_embedded(
        name_for_diagnostics: &str,
        bytes: &[u8],
    ) -> Result<Self, WasmImporterError> {
        Self::load_from_bytes(
            PathBuf::from(name_for_diagnostics),
            bytes,
            WasmRuntimeConfig::default(),
        )
    }

    /// Load from in-memory WASM bytes — useful for tests and embedders
    /// that ship the module inside their binary. The `path` is used
    /// only for diagnostics; see [`Self::load_embedded`] for an
    /// embedder-friendly wrapper.
    pub fn load_from_bytes(
        path: impl Into<PathBuf>,
        bytes: &[u8],
        config: WasmRuntimeConfig,
    ) -> Result<Self, WasmImporterError> {
        let path = path.into();

        // Process-wide shared engine — amortizes the JIT/cache cost
        // across all WASM-loaded modules in the workspace, and
        // applies the same security-locked-down `Config` as the
        // directive-plugin runtime.
        let engine = sandbox::shared_engine();

        let module = Module::new(&engine, bytes).map_err(|e| WasmImporterError::Compile {
            path: path.clone(),
            source: anyhow::Error::from(e),
        })?;

        Self::validate_module(&module)?;

        let module = Arc::new(module);

        // Call `metadata` once and cache the result. Importers don't
        // change name/description across calls; this avoids paying the
        // wasmtime instantiation cost on every `name()` / `description()`.
        let metadata = call_metadata(&engine, &module, config)?;

        Ok(Self {
            path,
            name: metadata.name,
            description: metadata.description,
            module,
            engine,
            config,
        })
    }

    /// The path the module was loaded from (or the
    /// `name_for_diagnostics` passed to [`Self::load_embedded`]).
    #[must_use]
    pub fn path(&self) -> &Path {
        &self.path
    }

    /// The per-call runtime caps this importer was loaded with. Useful
    /// for diagnostics ("did I hit the host's cap?") — the values
    /// surfaced in error variants like
    /// [`WasmImporterError::InputTooLarge::max`] are the same ones
    /// returned here.
    #[must_use]
    pub const fn runtime_config(&self) -> WasmRuntimeConfig {
        self.config
    }

    /// Reject imports (sandbox requirement) and check required exports.
    fn validate_module(module: &Module) -> Result<(), WasmImporterError> {
        if let Some(import) = module.imports().next() {
            return Err(WasmImporterError::ForbiddenImport {
                module: import.module().to_string(),
                name: import.name().to_string(),
            });
        }

        let exports: Vec<_> = module.exports().map(|e| e.name().to_string()).collect();
        for required in &[
            "memory",
            "alloc",
            "metadata",
            "identify",
            "extract",
            "extract_enriched",
        ] {
            if !exports.iter().any(|n| n == required) {
                return Err(WasmImporterError::MissingExport(required));
            }
        }
        Ok(())
    }

    /// Wraps a wasmtime call that takes msgpack input and returns
    /// msgpack output. The WASM module's entry-point convention:
    /// `fn (ptr: u32, len: u32) -> u64` where the return packs
    /// `(out_ptr << 32) | out_len`.
    fn call_msgpack<I: Serialize, O: DeserializeOwned>(
        &self,
        entry: &'static str,
        input: &I,
    ) -> Result<O, WasmImporterError> {
        call_msgpack_with(&self.engine, &self.module, self.config, entry, input)
    }
}

/// Cap input length before the lossy `as u32` cast — wasm32 memory
/// is u32-addressed, so >4 GiB input would silently truncate and
/// corrupt the import. Returns the validated length as `u32` so
/// callers don't need to repeat the cast.
const fn validate_input_size(len: usize) -> Result<u32, WasmImporterError> {
    if len > MAX_INPUT_BYTES {
        return Err(WasmImporterError::InputTooLarge {
            len,
            max: MAX_INPUT_BYTES,
        });
    }
    // Safe: `MAX_INPUT_BYTES` (64 MiB) fits in u32, and `len <= MAX_INPUT_BYTES`.
    Ok(len as u32)
}

/// Read a packed `(out_ptr, out_len)` u64 from a WASM entry-point
/// return, validate `out_len` against [`MAX_OUTPUT_BYTES`], and copy
/// the bytes out of WASM memory.
///
/// Centralized so the cap is enforced uniformly across `metadata`,
/// `identify`, `extract`, and `extract_enriched`.
fn read_packed_output(
    store: &Store<StoreState>,
    memory: &wasmtime::Memory,
    packed: u64,
) -> Result<Vec<u8>, WasmImporterError> {
    let out_ptr = (packed >> 32) as u32;
    let out_len = (packed & 0xFFFF_FFFF) as u32 as usize;
    if out_len > MAX_OUTPUT_BYTES {
        return Err(WasmImporterError::OutputTooLarge {
            len: out_len,
            max: MAX_OUTPUT_BYTES,
        });
    }
    let mut out_bytes = vec![0u8; out_len];
    memory
        .read(store, out_ptr as usize, &mut out_bytes)
        .map_err(|e| WasmImporterError::Runtime(e.into()))?;
    Ok(out_bytes)
}

/// Free-form wasmtime call helper. Extracted from `WasmImporter`'s
/// methods so the load-time `metadata` call can use it before `self`
/// is fully constructed.
fn call_msgpack_with<I: Serialize, O: DeserializeOwned>(
    engine: &Engine,
    module: &Module,
    config: WasmRuntimeConfig,
    entry: &'static str,
    input: &I,
) -> Result<O, WasmImporterError> {
    let input_bytes = rmp_serde::to_vec(input).map_err(WasmImporterError::Encode)?;
    let input_len = validate_input_size(input_bytes.len())?;

    let mut store = sandbox::make_sandboxed_store(engine, config.max_memory, config.max_time_secs)
        .map_err(runtime_err)?;

    // No imports at all — full sandbox.
    let linker = Linker::new(engine);
    let instance = linker
        .instantiate(&mut store, module)
        .map_err(runtime_err)?;

    // For each `get_typed_func` below: `validate_module` already
    // verified that the export exists at load time, so any error here
    // is necessarily a signature mismatch (not absence). Surfacing it
    // as `ExportSignatureMismatch` rather than `MissingExport` saves
    // guest authors from chasing a misleading "export not found"
    // error message.
    // `validate_module` proved `memory` export presence at load time,
    // so this `expect` documents an invariant rather than guarding a
    // real failure path. (The variant `MissingExport("memory")` is
    // reachable only via `validate_module` itself.)
    let memory = instance
        .get_memory(&mut store, "memory")
        .expect("validate_module verified `memory` export at load");

    let alloc = instance
        .get_typed_func::<u32, u32>(&mut store, "alloc")
        .map_err(|e| WasmImporterError::ExportSignatureMismatch {
            name: "alloc",
            source: anyhow::Error::from(e),
        })?;

    let input_ptr = alloc.call(&mut store, input_len).map_err(runtime_err)?;
    memory
        .write(&mut store, input_ptr as usize, &input_bytes)
        .map_err(|e| WasmImporterError::Runtime(e.into()))?;

    let func = instance
        .get_typed_func::<(u32, u32), u64>(&mut store, entry)
        .map_err(|e| WasmImporterError::ExportSignatureMismatch {
            name: entry,
            source: anyhow::Error::from(e),
        })?;

    let packed = func
        .call(&mut store, (input_ptr, input_len))
        .map_err(runtime_err)?;

    let out_bytes = read_packed_output(&store, &memory, packed)?;
    rmp_serde::from_slice(&out_bytes).map_err(WasmImporterError::Decode)
}

/// Special-case helper for the no-input `metadata` entry point. The
/// WASM convention is `fn metadata() -> u64` returning the packed
/// `(ptr, len)` of msgpack-encoded [`MetadataOutput`].
fn call_metadata(
    engine: &Engine,
    module: &Module,
    config: WasmRuntimeConfig,
) -> Result<MetadataOutput, WasmImporterError> {
    let mut store = sandbox::make_sandboxed_store(engine, config.max_memory, config.max_time_secs)
        .map_err(runtime_err)?;

    let linker = Linker::new(engine);
    let instance = linker
        .instantiate(&mut store, module)
        .map_err(runtime_err)?;

    // ABI handshake. `metadata` is the first thing the host calls on a
    // freshly loaded importer, so this is the natural load-time gate:
    // an importer built against an incompatible plugin-types is
    // rejected here with a clear error instead of trapping opaquely
    // inside a later `extract` (issue #1234). Subsequent identify /
    // extract calls re-instantiate the same already-verified module,
    // so they don't repeat the check.
    match sandbox::check_guest_abi(&instance, &mut store) {
        sandbox::AbiCheck::Match => {}
        sandbox::AbiCheck::Missing => {
            return Err(WasmImporterError::AbiVersionMissing {
                export: rustledger_plugin_types::ABI_VERSION_EXPORT,
                expected: sandbox::HOST_ABI_VERSION,
            });
        }
        sandbox::AbiCheck::Mismatch { found } => {
            return Err(WasmImporterError::AbiVersionMismatch {
                found,
                expected: sandbox::HOST_ABI_VERSION,
            });
        }
    }

    // Invariant: `validate_module` verified `memory` at load time.
    let memory = instance
        .get_memory(&mut store, "memory")
        .expect("validate_module verified `memory` export at load");

    // Same reasoning as in `call_msgpack_with`: validate_module
    // proved presence, so a typed_func error is a signature mismatch.
    let metadata = instance
        .get_typed_func::<(), u64>(&mut store, "metadata")
        .map_err(|e| WasmImporterError::ExportSignatureMismatch {
            name: "metadata",
            source: anyhow::Error::from(e),
        })?;

    let packed = metadata.call(&mut store, ()).map_err(runtime_err)?;
    let out_bytes = read_packed_output(&store, &memory, packed)?;
    rmp_serde::from_slice(&out_bytes).map_err(WasmImporterError::Decode)
}

/// Flatten the host's [`ImporterConfig`] into the wire-format
/// [`ImporterInput`] expected by the WASM module. A *subset* of
/// CSV-specific config fields is serialized into the free-form
/// `options` map — see [`project_csv_config_into_options`] for the
/// list and what's deferred.
fn build_wasm_input(path: &Path, content: Vec<u8>, config: &ImporterConfig) -> ImporterInput {
    let mut options = std::collections::HashMap::new();
    let ImporterType::Csv(csv) = &config.importer_type;
    project_csv_config_into_options(csv, &mut options);
    ImporterInput {
        path: path.to_string_lossy().into_owned(),
        content,
        account: config.account.clone(),
        currency: config.currency.clone(),
        options,
    }
}

/// Project a *subset* of [`CsvConfig`] into the wire-format `options`
/// map. String-encoded per the ABI's String→String contract.
///
/// # Currently projected
///
/// - `date_format`, `delimiter`, `has_header`, `skip_rows`,
///   `invert_sign`, `skip_zero_amounts` — simple String/bool/number
/// - `default_expense`, `default_income` — `Option<String>`
///
/// # Deferred to wave 2.3e+
///
/// The richer fields — `date_column` / `narration_column` /
/// `payee_column` / `amount_column` / `debit_column` /
/// `credit_column` (`ColumnSpec` enum: name OR index), `amount_locale`
/// / `amount_format`, `mappings` / `regex_mappings` (`Vec<(String,
/// String)>`), `use_merchant_dict` — are not yet projected. Encoding
/// them in a String→String map needs design decisions (key prefixes,
/// JSON-in-string, parallel collections?) that are best driven by a
/// real WASM CSV importer in wave 2.3e rather than guessed now.
///
/// A WASM importer in 2.3b can still extract from CSV files; it just
/// has to implement its own column-spec discovery rather than
/// inheriting the host's. Most non-CSV importers (OFX, MT940, …)
/// don't need any of the deferred fields.
fn project_csv_config_into_options(
    csv: &CsvConfig,
    options: &mut std::collections::HashMap<String, String>,
) {
    options.insert("date_format".to_string(), csv.date_format.clone());
    options.insert("delimiter".to_string(), csv.delimiter.to_string());
    options.insert("has_header".to_string(), csv.has_header.to_string());
    options.insert("skip_rows".to_string(), csv.skip_rows.to_string());
    options.insert("invert_sign".to_string(), csv.invert_sign.to_string());
    options.insert(
        "skip_zero_amounts".to_string(),
        csv.skip_zero_amounts.to_string(),
    );
    if let Some(de) = &csv.default_expense {
        options.insert("default_expense".to_string(), de.clone());
    }
    if let Some(di) = &csv.default_income {
        options.insert("default_income".to_string(), di.clone());
    }
}

/// Format a [`PluginError`] into a single human-readable line that
/// preserves the severity ("error" vs "warning") and avoids orphan
/// colons when location fields are absent.
///
/// Examples:
/// - severity=Error, file="foo.csv", line=42 → `"error foo.csv:42: bad row"`
/// - severity=Warning, file="foo.csv", line=None → `"warning foo.csv: weird value"`
/// - severity=Warning, file=None, line=Some(7) → `"warning line 7: weird value"`
/// - severity=Error, file=None, line=None → `"error: parser bug"`
fn format_plugin_error(e: &PluginError) -> String {
    let severity = match e.severity {
        PluginErrorSeverity::Error => "error",
        PluginErrorSeverity::Warning => "warning",
    };
    let location = match (&e.source_file, e.line_number) {
        (Some(f), Some(n)) => format!(" {f}:{n}"),
        (Some(f), None) => format!(" {f}"),
        (None, Some(n)) => format!(" line {n}"),
        (None, None) => String::new(),
    };
    format!("{severity}{location}: {}", e.message)
}

/// Materialize an [`ImporterOutput`] wire-format value back to the
/// host-side [`ImportResult`]. Delegates wrapper→directive conversion
/// to `rustledger_plugin::convert::wrapper_to_directive` so the WASM
/// importer path and the directive-plugin path share a single
/// converter — improvements there land here for free.
///
/// # Warning ordering
///
/// Warnings are appended in this order:
///
/// 1. **Output warnings** — `output.warnings` forwarded verbatim.
/// 2. **Output errors** — `output.errors`, formatted via
///    [`format_plugin_error`] so the severity prefix is preserved.
///
/// (The enriched analogue [`bridge_enriched_output`] additionally
/// emits *bridge warnings* first, for per-entry lossy paths that have
/// no analogue here.)
fn output_to_import_result(out: ImporterOutput) -> anyhow::Result<ImportResult> {
    let mut directives = Vec::with_capacity(out.directives.len());
    for w in out.directives {
        let d = rustledger_plugin::convert::wrapper_to_directive(&w)
            .map_err(|e| anyhow::anyhow!("WASM importer returned invalid directive: {e:?}"))?;
        directives.push(d);
    }
    let mut result = ImportResult::new(directives);
    for w in out.warnings {
        result = result.with_warning(w);
    }
    // Errors and warnings flow through the same `warnings` channel,
    // but the formatted string preserves the severity prefix so a
    // fatal-but-recoverable importer error is still distinguishable
    // from informational chatter. The structured error path
    // (`LedgerError::location`) is reserved for the loader layer.
    for e in &out.errors {
        result = result.with_warning(format_plugin_error(e));
    }
    Ok(result)
}

impl Importer for WasmImporter {
    fn name(&self) -> &str {
        &self.name
    }

    fn description(&self) -> &str {
        &self.description
    }

    fn identify(&self, path: &Path) -> bool {
        let input = IdentifyInput {
            path: path.to_string_lossy().into_owned(),
        };
        // The trait contract is `-> bool` (matches OFX/CSV), so we
        // can't surface a structured error. But "wrong signature on
        // `identify`" or "module trapped" are real bugs the guest
        // author needs to see — emit to stderr so they get a signal
        // instead of silently never matching. Successful identify
        // calls are quiet.
        match self.call_msgpack::<_, IdentifyOutput>("identify", &input) {
            Ok(out) => out.matches,
            Err(e) => {
                eprintln!(
                    "warning: WASM importer `{}` identify({}) failed: {e}",
                    self.name,
                    path.display()
                );
                false
            }
        }
    }

    fn extract(&self, path: &Path, config: &ImporterConfig) -> anyhow::Result<ImportResult> {
        // Use the typed `Io` variant before erasing to anyhow at the
        // trait boundary — keeps load and extract symmetric on file-
        // read failures, even though only the typed-error name is
        // observable to crate-internal callers.
        let content = std::fs::read(path).map_err(|source| WasmImporterError::Io {
            path: path.to_path_buf(),
            source,
        })?;
        let input = build_wasm_input(path, content, config);
        let output: ImporterOutput = self.call_msgpack("extract", &input)?;
        output_to_import_result(output)
    }

    fn extract_enriched(
        &self,
        path: &Path,
        config: &ImporterConfig,
    ) -> anyhow::Result<EnrichedImportResult> {
        let content = std::fs::read(path).map_err(|source| WasmImporterError::Io {
            path: path.to_path_buf(),
            source,
        })?;
        let input = build_wasm_input(path, content, config);
        let output: EnrichedImporterOutput = self.call_msgpack("extract_enriched", &input)?;
        bridge_enriched_output(output)
    }
}

/// Bridge a wire-format [`EnrichedImporterOutput`] into the host's
/// [`EnrichedImportResult`]. Extracted as a free function so the lossy
/// paths (unknown method strings, malformed fingerprint hex) can be
/// unit-tested without standing up wasmtime.
///
/// # Warning ordering
///
/// Warnings are emitted in this order, which is part of the contract
/// for any downstream consumer that filters or surfaces them:
///
/// 1. **Bridge warnings** (per-entry lossy paths: unknown method,
///    malformed fingerprint hex) — host-side issues with the wire
///    data, surface first so the importer author sees them prominently.
/// 2. **Output warnings** (importer's own informational warnings),
///    forwarded verbatim from `output.warnings`.
/// 3. **Output errors** (importer's structured errors), formatted via
///    [`format_plugin_error`] which preserves severity prefix.
fn bridge_enriched_output(output: EnrichedImporterOutput) -> anyhow::Result<EnrichedImportResult> {
    let mut entries = Vec::with_capacity(output.entries.len());
    let mut bridge_warnings: Vec<String> = Vec::new();
    for (wrapper, enr) in output.entries {
        let dir = rustledger_plugin::convert::wrapper_to_directive(&wrapper)
            .map_err(|e| anyhow::anyhow!("WASM importer returned invalid directive: {e:?}"))?;
        let method = parse_method(&enr.method).unwrap_or_else(|unknown| {
            bridge_warnings.push(format!(
                "warning: WASM importer used unknown categorization method `{unknown}`, falling back to Default"
            ));
            rustledger_ops::enrichment::CategorizationMethod::Default
        });
        let alternatives = enr
            .alternatives
            .into_iter()
            .map(|a| {
                let alt_method = parse_method(&a.method).unwrap_or_else(|unknown| {
                    bridge_warnings.push(format!(
                        "warning: WASM importer used unknown categorization method `{unknown}` in alternative, falling back to Default"
                    ));
                    rustledger_ops::enrichment::CategorizationMethod::Default
                });
                rustledger_ops::enrichment::Alternative {
                    account: a.account,
                    confidence: a.confidence,
                    method: alt_method,
                }
            })
            .collect();
        let fingerprint = match enr.fingerprint {
            Some(hex) => match Fingerprint::from_hex(&hex) {
                Ok(fp) => Some(fp),
                Err(e) => {
                    bridge_warnings.push(format!(
                        "warning: WASM importer returned malformed fingerprint hex `{hex}`: {e}"
                    ));
                    None
                }
            },
            None => None,
        };
        let enrichment = rustledger_ops::enrichment::Enrichment {
            directive_index: enr.directive_index,
            confidence: enr.confidence,
            method,
            alternatives,
            fingerprint,
        };
        entries.push((dir, enrichment));
    }
    let mut enriched = EnrichedImportResult::new(entries);
    for w in bridge_warnings {
        enriched = enriched.with_warning(w);
    }
    for w in output.warnings {
        enriched = enriched.with_warning(w);
    }
    for e in &output.errors {
        enriched = enriched.with_warning(format_plugin_error(e));
    }
    Ok(enriched)
}

/// Convert the wire-format method string (as emitted by
/// `CategorizationMethod::as_meta_value`) back into the host enum.
///
/// Returns `Err(unknown)` for strings the host doesn't recognize — the
/// caller is expected to surface a warning and fall back to
/// `CategorizationMethod::Default`. We don't silently absorb unknown
/// strings here: a typo like `"merchant_dict"` vs `"merchant-dict"`
/// (the exact Copilot-flagged bug from #1130) would otherwise degrade
/// data without any signal to the user.
fn parse_method(s: &str) -> Result<rustledger_ops::enrichment::CategorizationMethod, &str> {
    use rustledger_ops::enrichment::CategorizationMethod;
    match s {
        "rule" => Ok(CategorizationMethod::Rule),
        "merchant-dict" => Ok(CategorizationMethod::MerchantDict),
        "ml" => Ok(CategorizationMethod::Ml),
        "llm" => Ok(CategorizationMethod::Llm),
        "manual" => Ok(CategorizationMethod::Manual),
        "default" => Ok(CategorizationMethod::Default),
        unknown => Err(unknown),
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn wasm_runtime_config_default_is_sensible() {
        let c = WasmRuntimeConfig::default();
        // Sourced from the workspace-wide sandbox constants. A future
        // bump propagates here without manual update; this test pins
        // that the importer continues to track the shared defaults
        // rather than drifting to a local literal.
        assert_eq!(c.max_memory, sandbox::DEFAULT_SANDBOX_MAX_MEMORY);
        assert_eq!(c.max_time_secs, sandbox::DEFAULT_SANDBOX_MAX_TIME_SECS);
    }

    #[test]
    fn validate_module_rejects_module_with_imports() {
        // A WAT module with a single import — should be rejected.
        let wat = r#"
            (module
                (import "env" "ext" (func $ext))
                (memory (export "memory") 1)
                (func (export "alloc") (param i32) (result i32) i32.const 0)
                (func (export "metadata") (result i64) i64.const 0)
                (func (export "identify") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract_enriched") (param i32 i32) (result i64) i64.const 0)
            )
        "#;
        let bytes = wat::parse_str(wat).expect("WAT parses");
        let engine = sandbox::shared_engine();
        let module = Module::new(&engine, &bytes).unwrap();
        let err = WasmImporter::validate_module(&module).unwrap_err();
        assert!(matches!(err, WasmImporterError::ForbiddenImport { .. }));
    }

    #[test]
    fn validate_module_rejects_missing_export() {
        // Has memory + alloc + metadata but missing identify/extract/extract_enriched.
        let wat = r#"
            (module
                (memory (export "memory") 1)
                (func (export "alloc") (param i32) (result i32) i32.const 0)
                (func (export "metadata") (result i64) i64.const 0)
            )
        "#;
        let bytes = wat::parse_str(wat).expect("WAT parses");
        let engine = sandbox::shared_engine();
        let module = Module::new(&engine, &bytes).unwrap();
        let err = WasmImporter::validate_module(&module).unwrap_err();
        assert!(matches!(err, WasmImporterError::MissingExport(_)));
    }

    #[test]
    fn parse_method_round_trips_known_values() {
        use rustledger_ops::enrichment::CategorizationMethod;
        assert!(matches!(
            parse_method("rule"),
            Ok(CategorizationMethod::Rule)
        ));
        assert!(matches!(
            parse_method("merchant-dict"),
            Ok(CategorizationMethod::MerchantDict)
        ));
        assert!(matches!(parse_method("ml"), Ok(CategorizationMethod::Ml)));
        assert!(matches!(parse_method("llm"), Ok(CategorizationMethod::Llm)));
        assert!(matches!(
            parse_method("manual"),
            Ok(CategorizationMethod::Manual)
        ));
        assert!(matches!(
            parse_method("default"),
            Ok(CategorizationMethod::Default)
        ));
    }

    #[test]
    fn parse_method_round_trips_via_as_meta_value() {
        // Pin the contract: every `CategorizationMethod` round-trips
        // through its `as_meta_value()` string. If a host variant is
        // added without updating `parse_method`, this test fails.
        use rustledger_ops::enrichment::CategorizationMethod;
        for m in [
            CategorizationMethod::Rule,
            CategorizationMethod::MerchantDict,
            CategorizationMethod::Ml,
            CategorizationMethod::Llm,
            CategorizationMethod::Manual,
            CategorizationMethod::Default,
        ] {
            let s = m.as_meta_value();
            let parsed = parse_method(s)
                .unwrap_or_else(|u| panic!("as_meta_value `{u}` not handled by parse_method"));
            assert_eq!(parsed, m, "round-trip failed for {m:?}");
        }
    }

    #[test]
    fn parse_method_unknown_surfaces_the_unknown_string() {
        // Previously: silently fell back to Default. Now: returns
        // Err(unknown) so the caller can warn — protects against
        // typos like `merchant_dict` (underscore) vs `merchant-dict`
        // (hyphen, the actual wire encoding from
        // `CategorizationMethod::as_meta_value`).
        assert_eq!(parse_method("future-method"), Err("future-method"));
        assert_eq!(parse_method("merchant_dict"), Err("merchant_dict"));
        assert_eq!(parse_method(""), Err(""));
    }

    #[test]
    fn format_plugin_error_with_full_location() {
        let e = PluginError::error("bad row").at("foo.csv", 42);
        assert_eq!(format_plugin_error(&e), "error foo.csv:42: bad row");
    }

    #[test]
    fn format_plugin_error_warning_severity() {
        let e = PluginError::warning("weird value").at("foo.csv", 42);
        assert_eq!(format_plugin_error(&e), "warning foo.csv:42: weird value");
    }

    #[test]
    fn format_plugin_error_no_location_no_orphan_colon() {
        let e = PluginError::error("parser bug");
        // Previously: ": parser bug" (orphan colon). Now: "error: parser bug".
        assert_eq!(format_plugin_error(&e), "error: parser bug");
    }

    #[test]
    fn format_plugin_error_file_only() {
        let e = PluginError::warning("weird value");
        let e = PluginError {
            source_file: Some("foo.csv".to_string()),
            ..e
        };
        assert_eq!(format_plugin_error(&e), "warning foo.csv: weird value");
    }

    #[test]
    fn format_plugin_error_line_only_uses_human_phrasing() {
        // Previously: ":42: weird" (orphan colon). Now: "warning line 42: weird".
        let e = PluginError::warning("weird");
        let e = PluginError {
            line_number: Some(42),
            ..e
        };
        assert_eq!(format_plugin_error(&e), "warning line 42: weird");
    }

    /// Build a WAT module that pre-loads `MessagePack` outputs for every
    /// entry point in low memory and returns hardcoded packed
    /// `(ptr, len)` u64s. `alloc` is a bump allocator starting at
    /// offset 1024, so host-allocated input never overlaps the
    /// pre-loaded data.
    ///
    /// Wire-format bytes are rmp-serde's default positional encoding
    /// (struct → fixarray-N, fields in declaration order).
    fn roundtrip_wat() -> &'static str {
        r#"
        (module
            (memory (export "memory") 1)

            ;; MetadataOutput { name: "tst", description: "tst" }
            ;; 0x92 fixarray-2, 0xa3 fixstr-3 "tst", 0xa3 fixstr-3 "tst"
            (data (i32.const 0) "\92\a3tst\a3tst")

            ;; IdentifyOutput { matches: true }
            ;; 0x91 fixarray-1, 0xc3 true
            (data (i32.const 16) "\91\c3")

            ;; ImporterOutput { directives: [], warnings: [], errors: [] }
            ;; 0x93 fixarray-3, then three 0x90 fixarray-0
            (data (i32.const 24) "\93\90\90\90")

            ;; EnrichedImporterOutput { entries: [], warnings: [], errors: [] }
            (data (i32.const 32) "\93\90\90\90")

            ;; bump allocator: hand out at $bump, advance by $size.
            ;; NOTE: real importers MUST bounds-check $bump+$size
            ;; against current memory and call `memory.grow` (subject
            ;; to MemoryLimiter approval). This test fixture skips
            ;; that — inputs in the test are small and we declare 1
            ;; full page (64 KiB), so the bump never crosses the
            ;; boundary.
            (global $bump (mut i32) (i32.const 1024))
            (func (export "alloc") (param $size i32) (result i32)
                (local $ret i32)
                global.get $bump
                local.set $ret
                global.get $bump
                local.get $size
                i32.add
                global.set $bump
                local.get $ret)

            ;; metadata: ptr=0, len=9 → (0<<32) | 9 = 9
            (func (export "metadata") (result i64)
                i64.const 9)

            ;; identify: ptr=16, len=2 → (16<<32) | 2
            (func (export "identify") (param i32 i32) (result i64)
                i64.const 0x10_0000_0002)

            ;; extract: ptr=24, len=4 → (24<<32) | 4
            (func (export "extract") (param i32 i32) (result i64)
                i64.const 0x18_0000_0004)

            ;; extract_enriched: ptr=32, len=4 → (32<<32) | 4
            (func (export "extract_enriched") (param i32 i32) (result i64)
                i64.const 0x20_0000_0004)

            ;; ABI handshake export. Must equal sandbox::HOST_ABI_VERSION
            ;; (rustledger_plugin_types::ABI_VERSION = 1). If the ABI
            ;; version is ever bumped, this literal moves in lockstep —
            ;; the deliberate test update that proves a real guest would
            ;; need rebuilding too.
            (func (export "__rustledger_abi_version") (result i32)
                i32.const 1)
        )
        "#
    }

    fn minimal_config() -> ImporterConfig {
        ImporterConfig {
            account: "Assets:Bank:Checking".to_string(),
            currency: Some("USD".to_string()),
            importer_type: ImporterType::Csv(CsvConfig::default()),
        }
    }

    /// A WAT importer with every required export present (so
    /// `validate_module` passes) but a configurable
    /// `__rustledger_abi_version`. `abi_section` is spliced in verbatim,
    /// so a caller can omit it entirely to model a pre-handshake guest.
    /// `metadata` returns junk on purpose — the ABI check runs before
    /// the host ever reads it, so these fixtures never need real
    /// `MessagePack`.
    fn importer_wat_with_abi(abi_section: &str) -> String {
        format!(
            r#"
            (module
                (memory (export "memory") 1)
                (func (export "alloc") (param i32) (result i32) i32.const 0)
                (func (export "metadata") (result i64) i64.const 0)
                (func (export "identify") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract_enriched") (param i32 i32) (result i64) i64.const 0)
                {abi_section}
            )
            "#
        )
    }

    /// Issue #1234: an importer that doesn't advertise an ABI version
    /// is rejected at load with a clear error, not an opaque trap on
    /// the first `extract`.
    #[test]
    fn load_rejects_importer_missing_abi_version() {
        let bytes = wat::parse_str(importer_wat_with_abi("")).expect("WAT parses");
        let err = WasmImporter::load_from_bytes(
            PathBuf::from("noabi.wasm"),
            &bytes,
            WasmRuntimeConfig::default(),
        )
        .expect_err("load must reject an importer with no ABI export");
        assert!(
            matches!(err, WasmImporterError::AbiVersionMissing { .. }),
            "expected AbiVersionMissing, got: {err:?}"
        );
    }

    /// Issue #1234: an importer built against a different ABI version
    /// is rejected at load, naming both versions.
    #[test]
    fn load_rejects_importer_with_mismatched_abi_version() {
        // 999 is deliberately not the host ABI version.
        let wat = importer_wat_with_abi(
            r#"(func (export "__rustledger_abi_version") (result i32) i32.const 999)"#,
        );
        let bytes = wat::parse_str(wat).expect("WAT parses");
        let err = WasmImporter::load_from_bytes(
            PathBuf::from("badabi.wasm"),
            &bytes,
            WasmRuntimeConfig::default(),
        )
        .expect_err("load must reject an ABI-mismatched importer");
        match err {
            WasmImporterError::AbiVersionMismatch { found, expected } => {
                assert_eq!(found, 999);
                assert_eq!(expected, sandbox::HOST_ABI_VERSION);
            }
            other => panic!("expected AbiVersionMismatch, got: {other:?}"),
        }
    }

    #[test]
    fn end_to_end_wat_module_round_trips_all_entry_points() {
        let bytes = wat::parse_str(roundtrip_wat()).expect("WAT parses");
        let importer = WasmImporter::load_from_bytes(
            PathBuf::from("test.wasm"),
            &bytes,
            WasmRuntimeConfig::default(),
        )
        .expect("module loads + metadata round-trips");

        // metadata was decoded once at load and cached for these
        // accessors — proves the MetadataOutput msgpack flowed end to
        // end through the host.
        assert_eq!(importer.name(), "tst");
        assert_eq!(importer.description(), "tst");

        // identify round-trip — input ignored, module hardcodes true.
        assert!(importer.identify(Path::new("anything.csv")));

        // extract + extract_enriched need a real file for std::fs::read.
        let tmp = tempfile::NamedTempFile::new().expect("tempfile");
        let config = minimal_config();

        let result = importer
            .extract(tmp.path(), &config)
            .expect("extract round-trip");
        assert!(result.directives.is_empty());
        assert!(result.warnings.is_empty());

        let enriched = importer
            .extract_enriched(tmp.path(), &config)
            .expect("extract_enriched round-trip");
        assert!(enriched.entries.is_empty());
        assert!(enriched.warnings.is_empty());
    }

    #[test]
    fn oversized_output_is_rejected_before_allocation() {
        // Module's metadata() returns out_len = u32::MAX. Without the
        // MAX_OUTPUT_BYTES check, the host would attempt a ~4 GiB Vec
        // allocation. The check should catch it during load.
        let wat = r#"
            (module
                (memory (export "memory") 1)
                (func (export "alloc") (param i32) (result i32) i32.const 0)
                ;; metadata: ptr=0, len=u32::MAX
                (func (export "metadata") (result i64)
                    i64.const 0x0000_0000_ffff_ffff)
                (func (export "identify") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract_enriched") (param i32 i32) (result i64) i64.const 0)
                ;; ABI handshake passes so the oversized-metadata check
                ;; downstream is what rejects this module (issue #1234).
                (func (export "__rustledger_abi_version") (result i32) i32.const 1)
            )
        "#;
        let bytes = wat::parse_str(wat).expect("WAT parses");
        // Can't use `.expect_err(...)` here — `WasmImporter` doesn't
        // implement `Debug` (the wasmtime `Module`/`Engine` it holds
        // aren't trivially debuggable), so we destructure manually.
        let Err(err) = WasmImporter::load_from_bytes(
            PathBuf::from("oversized.wasm"),
            &bytes,
            WasmRuntimeConfig::default(),
        ) else {
            panic!("oversized metadata output should have been rejected at load");
        };
        assert!(
            matches!(
                err,
                WasmImporterError::OutputTooLarge { len, max }
                    if len == u32::MAX as usize && max == MAX_OUTPUT_BYTES
            ),
            "expected OutputTooLarge, got {err:?}"
        );
    }

    // Note: `memory_limiter_rejects_grow_above_max` and
    // `table_limiter_rejects_grow_above_max` live in
    // `rustledger_plugin::sandbox::tests` now that the limiter
    // itself was hoisted there. The integration test below
    // (`initial_memory_above_cap_is_rejected_via_limiter_wiring`)
    // still proves the importer's load path wires it correctly.

    #[test]
    fn zero_max_time_secs_does_not_starve_fuel() {
        // Regression: previously fuel = 0 * 1_000_000 = 0, causing
        // immediate trap on first instruction. Now clamped via
        // .max(1) so a 0 config still gets enough fuel to complete a
        // trivial call.
        let config = WasmRuntimeConfig {
            max_memory: sandbox::DEFAULT_SANDBOX_MAX_MEMORY,
            max_time_secs: 0,
        };
        let bytes = wat::parse_str(roundtrip_wat()).expect("WAT parses");
        // Loading calls metadata(), which is a single i64.const +
        // return — well under 1M instructions.
        let importer = WasmImporter::load_from_bytes(PathBuf::from("test.wasm"), &bytes, config)
            .expect("zero max_time_secs is clamped, not starved");
        assert_eq!(importer.name(), "tst");
    }

    #[test]
    fn validate_input_size_accepts_at_cap_and_rejects_above() {
        // Exactly at the cap is fine.
        assert_eq!(
            validate_input_size(MAX_INPUT_BYTES).unwrap(),
            MAX_INPUT_BYTES as u32
        );
        // One byte over is rejected, with the offending length surfaced
        // in the error so the user can see how much they overshot.
        let err = validate_input_size(MAX_INPUT_BYTES + 1).unwrap_err();
        assert!(
            matches!(
                err,
                WasmImporterError::InputTooLarge { len, max }
                    if len == MAX_INPUT_BYTES + 1 && max == MAX_INPUT_BYTES
            ),
            "got: {err:?}"
        );
    }

    #[test]
    fn fuel_calc_saturates_instead_of_overflowing() {
        // Regression for Copilot #2: u64::MAX max_time_secs would have
        // overflowed in release (silent wrap to a tiny number ⇒ fuel
        // starvation) and panicked in debug. Saturating_mul caps at
        // u64::MAX which set_fuel accepts.
        let bytes = wat::parse_str(roundtrip_wat()).expect("WAT parses");
        let config = WasmRuntimeConfig {
            max_memory: sandbox::DEFAULT_SANDBOX_MAX_MEMORY,
            max_time_secs: u64::MAX,
        };
        // Successful load proves the fuel calc didn't panic and the
        // resulting saturated value is acceptable to set_fuel.
        let importer = WasmImporter::load_from_bytes(PathBuf::from("test.wasm"), &bytes, config)
            .expect("u64::MAX max_time_secs saturates, doesn't overflow");
        assert_eq!(importer.name(), "tst");
    }

    #[test]
    fn wrong_signature_export_surfaces_export_signature_mismatch() {
        // `metadata` is declared but with the wrong signature
        // (returns i32 instead of i64). `validate_module` checks
        // presence-by-name only, so this passes validate. The
        // signature error surfaces when `call_metadata` tries
        // `get_typed_func::<(), u64>`.
        let wat = r#"
            (module
                (memory (export "memory") 1)
                (func (export "alloc") (param i32) (result i32) i32.const 0)
                ;; WRONG: should be (result i64), declared as (result i32)
                (func (export "metadata") (result i32) i32.const 0)
                (func (export "identify") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract_enriched") (param i32 i32) (result i64) i64.const 0)
                ;; Correct ABI so the check passes and the metadata
                ;; signature mismatch is what surfaces (issue #1234).
                (func (export "__rustledger_abi_version") (result i32) i32.const 1)
            )
        "#;
        let bytes = wat::parse_str(wat).expect("WAT parses");
        let Err(err) = WasmImporter::load_from_bytes(
            PathBuf::from("badsig.wasm"),
            &bytes,
            WasmRuntimeConfig::default(),
        ) else {
            panic!("metadata with wrong signature should be rejected");
        };
        // Previously: silently surfaced as MissingExport("metadata"),
        // which is misleading because the export DOES exist. Now:
        // ExportSignatureMismatch names the export and includes the
        // wasmtime type-mismatch error in the source chain.
        assert!(
            matches!(
                err,
                WasmImporterError::ExportSignatureMismatch {
                    name: "metadata",
                    ..
                }
            ),
            "expected ExportSignatureMismatch for metadata, got {err:?}"
        );
    }

    #[test]
    fn initial_memory_above_cap_is_rejected_via_limiter_wiring() {
        // Pins the `store.limiter(|s| &mut s.limiter)` wiring against
        // refactor regression. wasmtime calls `memory_growing` for
        // both initial allocation and grow — a module declaring 5000
        // pages (320 MiB) initial memory with a 64 MiB cap should
        // fail to instantiate. If the limiter wiring breaks, this
        // test catches it (the direct trait-method test above does
        // not).
        let wat = r#"
            (module
                (memory (export "memory") 5000)
                (func (export "alloc") (param i32) (result i32) i32.const 0)
                (func (export "metadata") (result i64) i64.const 0)
                (func (export "identify") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract") (param i32 i32) (result i64) i64.const 0)
                (func (export "extract_enriched") (param i32 i32) (result i64) i64.const 0)
            )
        "#;
        let bytes = wat::parse_str(wat).expect("WAT parses");
        let config = WasmRuntimeConfig {
            max_memory: 64 * 1024 * 1024,
            max_time_secs: 30,
        };
        let Err(err) = WasmImporter::load_from_bytes(PathBuf::from("bigmem.wasm"), &bytes, config)
        else {
            panic!("module declaring 320 MiB initial memory should be rejected with 64 MiB cap");
        };
        // wasmtime turns Ok(false) at instantiation into an instantiate
        // error, which the host maps to Runtime.
        assert!(
            matches!(err, WasmImporterError::Runtime(_)),
            "expected Runtime (instantiate failed via limiter), got {err:?}"
        );
    }

    // ===== bridge_enriched_output direct tests =====
    //
    // These exercise the lossy paths (unknown method, malformed
    // fingerprint hex, valid fingerprint round-trip) without standing
    // up wasmtime — the bridge logic is the testable piece, the
    // wasmtime round-trip is covered by the end-to-end WAT test.

    use rustledger_plugin_types::{
        AlternativeWrapper, DirectiveData, DirectiveWrapper, EnrichmentWrapper, OpenData,
    };

    fn open_wrapper(account: &str) -> DirectiveWrapper {
        DirectiveWrapper {
            directive_type: String::new(),
            date: "2024-01-01".to_string(),
            filename: None,
            lineno: None,
            data: DirectiveData::Open(OpenData {
                account: account.to_string(),
                currencies: vec![],
                booking: None,
                metadata: vec![],
            }),
        }
    }

    fn enrichment_wrapper(method: &str, fingerprint: Option<String>) -> EnrichmentWrapper {
        EnrichmentWrapper {
            directive_index: 0,
            confidence: 1.0,
            method: method.to_string(),
            alternatives: vec![],
            fingerprint,
        }
    }

    #[test]
    fn bridge_round_trips_valid_fingerprint_hex() {
        let fp = Fingerprint::compute("2024-01-01", Some("100"), "coffee");
        let hex = fp.to_hex();
        let out = EnrichedImporterOutput {
            entries: vec![(
                open_wrapper("Assets:Bank"),
                enrichment_wrapper("rule", Some(hex)),
            )],
            warnings: vec![],
            errors: vec![],
        };
        let bridged = bridge_enriched_output(out).expect("bridge succeeds");
        assert_eq!(bridged.entries.len(), 1);
        assert_eq!(
            bridged.entries[0].1.fingerprint,
            Some(fp),
            "fingerprint should round-trip"
        );
        assert!(bridged.warnings.is_empty(), "no warnings expected");
    }

    #[test]
    fn bridge_warns_on_malformed_fingerprint_hex_and_drops_to_none() {
        let out = EnrichedImporterOutput {
            entries: vec![(
                open_wrapper("Assets:Bank"),
                enrichment_wrapper("rule", Some("not-a-valid-hex".to_string())),
            )],
            warnings: vec![],
            errors: vec![],
        };
        let bridged = bridge_enriched_output(out).expect("bridge succeeds");
        assert_eq!(bridged.entries.len(), 1);
        assert_eq!(bridged.entries[0].1.fingerprint, None);
        // Warning text names the bad hex so the importer author can
        // find the bug quickly.
        assert_eq!(bridged.warnings.len(), 1);
        assert!(
            bridged.warnings[0].contains("not-a-valid-hex"),
            "warning should name the bad hex: {}",
            bridged.warnings[0]
        );
    }

    #[test]
    fn bridge_warns_on_unknown_method_and_falls_back_to_default() {
        use rustledger_ops::enrichment::CategorizationMethod;
        let out = EnrichedImporterOutput {
            entries: vec![(
                open_wrapper("Assets:Bank"),
                enrichment_wrapper("merchant_dict", None), // underscore typo, exact #1130 bug shape
            )],
            warnings: vec![],
            errors: vec![],
        };
        let bridged = bridge_enriched_output(out).expect("bridge succeeds");
        assert_eq!(bridged.entries[0].1.method, CategorizationMethod::Default);
        assert_eq!(bridged.warnings.len(), 1);
        assert!(
            bridged.warnings[0].contains("merchant_dict"),
            "warning should name the unknown method: {}",
            bridged.warnings[0]
        );
    }

    #[test]
    fn bridge_warns_on_unknown_method_in_alternative() {
        use rustledger_ops::enrichment::CategorizationMethod;
        let mut enr = enrichment_wrapper("rule", None);
        enr.alternatives = vec![AlternativeWrapper {
            account: "Expenses:Other".to_string(),
            confidence: 0.3,
            method: "future-method".to_string(),
        }];
        let out = EnrichedImporterOutput {
            entries: vec![(open_wrapper("Assets:Bank"), enr)],
            warnings: vec![],
            errors: vec![],
        };
        let bridged = bridge_enriched_output(out).expect("bridge succeeds");
        let alt = &bridged.entries[0].1.alternatives[0];
        assert_eq!(alt.method, CategorizationMethod::Default);
        assert_eq!(bridged.warnings.len(), 1);
        assert!(bridged.warnings[0].contains("future-method"));
        assert!(
            bridged.warnings[0].contains("alternative"),
            "warning should distinguish the alternative slot: {}",
            bridged.warnings[0]
        );
    }

    #[test]
    fn bridge_warning_ordering_is_bridge_then_output_warnings_then_errors() {
        // Pins the warning-emission order documented on
        // `bridge_enriched_output`. Order matters for downstream
        // consumers that filter or surface them.
        let out = EnrichedImporterOutput {
            entries: vec![(
                open_wrapper("Assets:Bank"),
                enrichment_wrapper("nonsense", None),
            )],
            warnings: vec!["informational warning".to_string()],
            errors: vec![PluginError::error("structured error").at("foo.csv", 7)],
        };
        let bridged = bridge_enriched_output(out).expect("bridge succeeds");
        assert_eq!(bridged.warnings.len(), 3);
        assert!(
            bridged.warnings[0].contains("nonsense"),
            "first: bridge warning, got {}",
            bridged.warnings[0]
        );
        assert_eq!(
            bridged.warnings[1], "informational warning",
            "second: output.warnings forwarded verbatim"
        );
        assert_eq!(
            bridged.warnings[2], "error foo.csv:7: structured error",
            "third: output.errors via format_plugin_error"
        );
    }

    #[test]
    fn output_to_import_result_uses_severity_aware_formatter() {
        // Integration test: proves format_plugin_error is actually
        // wired into the production path, not just unit-tested in
        // isolation. A refactor that switches back to raw format!()
        // would regress this.
        let out = ImporterOutput {
            directives: vec![],
            warnings: vec!["plain warning".to_string()],
            errors: vec![
                PluginError::error("bad row").at("foo.csv", 42),
                PluginError::warning("weird value"),
            ],
        };
        let result = output_to_import_result(out).expect("succeeds");
        assert_eq!(
            result.warnings,
            vec![
                "plain warning".to_string(),
                "error foo.csv:42: bad row".to_string(),
                "warning: weird value".to_string(),
            ]
        );
    }

    // ===== Accessor / constructor tests =====

    #[test]
    fn load_embedded_uses_name_as_path_and_default_config() {
        let bytes = wat::parse_str(roundtrip_wat()).expect("WAT parses");
        let importer =
            WasmImporter::load_embedded("inline-test", &bytes).expect("embedded load succeeds");
        // The diagnostic name flows through to `path()` so error
        // messages and logs identify the embedded module.
        assert_eq!(importer.path(), Path::new("inline-test"));
        // Default config is used, caller didn't pass one.
        assert_eq!(
            importer.runtime_config().max_memory,
            sandbox::DEFAULT_SANDBOX_MAX_MEMORY
        );
        assert_eq!(
            importer.runtime_config().max_time_secs,
            sandbox::DEFAULT_SANDBOX_MAX_TIME_SECS
        );
        // Metadata still cached as in the standard load path.
        assert_eq!(importer.name(), "tst");
    }

    #[test]
    fn runtime_config_returns_the_loaded_config() {
        let custom = WasmRuntimeConfig {
            max_memory: 128 * 1024 * 1024,
            max_time_secs: 60,
        };
        let bytes = wat::parse_str(roundtrip_wat()).expect("WAT parses");
        let importer = WasmImporter::load_from_bytes(PathBuf::from("custom.wasm"), &bytes, custom)
            .expect("custom-config load succeeds");
        assert_eq!(importer.runtime_config().max_memory, custom.max_memory);
        assert_eq!(
            importer.runtime_config().max_time_secs,
            custom.max_time_secs
        );
    }

    #[test]
    fn debug_impl_does_not_panic_and_redacts_wasmtime_types() {
        let bytes = wat::parse_str(roundtrip_wat()).expect("WAT parses");
        let importer = WasmImporter::load_embedded("dbg-test", &bytes).expect("load succeeds");
        let s = format!("{importer:?}");
        // Includes host metadata...
        assert!(s.contains("WasmImporter"));
        assert!(s.contains("dbg-test"));
        assert!(s.contains("tst")); // name + description
        // ...but doesn't leak wasmtime Module/Engine internals.
        assert!(
            !s.contains("Module {"),
            "Debug should not expand the wasmtime Module: {s}"
        );
    }
}