use crate::error::{DirectiveError, DirectiveErrors};
use crate::spectest::spectest_importobject;
use anyhow::{anyhow, bail, Result};
use std::collections::{HashMap, HashSet};
use std::path::Path;
use std::str;
use wasmer::*;
use wast::core::{AbstractHeapType, HeapType, WastArgCore, WastRetCore};
use wast::token::{F32, F64};
use wast::{lexer::Lexer, parser};
use wast::{QuoteWat, Wast as WWast, WastArg};

/// The wast test script language allows modules to be defined and actions
/// to be performed on them.
#[allow(dead_code)]
pub struct Wast {
    /// Wast files have a concept of a "current" module, which is the most
    /// recently defined.
    current: Option<Instance>,
    /// The Import Object that all wast tests will have
    import_object: Imports,
    /// The instances in the test
    instances: HashMap<String, Instance>,
    /// Allowed failures (ideally this should be empty)
    allowed_instantiation_failures: HashSet<String>,
    /// If the (expected from .wast, actual) message pair is in this list,
    /// treat the strings as matching.
    match_trap_messages: HashMap<String, String>,
    /// If the current module was an allowed failure, we allow test to fail
    current_is_allowed_failure: bool,
    /// The store in which the tests are executing.
    store: Store,
    /// A flag indicating if Wast tests should stop as soon as one test fails.
    pub fail_fast: bool,
    /// A flag indicating that assert_trap and assert_exhaustion should be skipped.
    /// See https://github.com/wasmerio/wasmer/issues/1550 for more info
    disable_assert_trap_exhaustion: bool,

    /// A flag indicating that assert_exception should be skipped.
    disable_assert_exception: bool,
}

impl Wast {
    /// Construct a new instance of `Wast` with a given imports.
    pub fn new(store: Store, import_object: Imports) -> Self {
        Self {
            current: None,
            store,
            import_object,
            allowed_instantiation_failures: HashSet::new(),
            match_trap_messages: HashMap::new(),
            current_is_allowed_failure: false,
            instances: HashMap::new(),
            fail_fast: true,
            disable_assert_trap_exhaustion: false,
            disable_assert_exception: false,
        }
    }

    /// A list of instantiation failures to allow.
    pub fn allow_instantiation_failures(&mut self, failures: &[&str]) {
        for &failure_str in failures.iter() {
            self.allowed_instantiation_failures
                .insert(failure_str.to_string());
        }
    }

    /// A list of alternative messages to permit for a trap failure.
    pub fn allow_trap_message(&mut self, expected: &str, allowed: &str) {
        self.match_trap_messages
            .insert(expected.into(), allowed.into());
    }

    /// Do not run any code in assert_trap or assert_exhaustion.
    pub fn disable_assert_and_exhaustion(&mut self) {
        self.disable_assert_trap_exhaustion = true;
    }

    /// Do not run any code in assert_exception.
    pub fn disable_assert_exception(&mut self) {
        self.disable_assert_exception = true;
    }

    /// Construct a new instance of `Wast` with the spectests imports.
    pub fn new_with_spectest(mut store: Store) -> Self {
        let import_object = spectest_importobject(&mut store);
        Self::new(store, import_object)
    }

    fn get_instance(&self, instance_name: Option<&str>) -> Result<Instance> {
        match instance_name {
            Some(name) => self
                .instances
                .get(name)
                .cloned()
                .ok_or_else(|| anyhow!("failed to find instance named `{}`", name)),
            None => self
                .current
                .clone()
                .ok_or_else(|| anyhow!("no previous instance found")),
        }
    }

    /// Perform the action portion of a command.
    fn perform_execute(&mut self, exec: wast::WastExecute<'_>) -> Result<Vec<Value>> {
        match exec {
            wast::WastExecute::Invoke(invoke) => self.perform_invoke(invoke),
            wast::WastExecute::Wat(mut module) => {
                let binary = module.encode()?;
                let result = self.instantiate(&binary);
                result.map(|_| Vec::new())
            }
            wast::WastExecute::Get { module, global, .. } => {
                self.get(module.map(|s| s.name()), global)
            }
        }
    }

    fn perform_invoke(&mut self, exec: wast::WastInvoke<'_>) -> Result<Vec<Value>> {
        let values = exec
            .args
            .iter()
            .map(|v| match v {
                WastArg::Core(v) => self.runtime_value(v),
                WastArg::Component(_) => bail!("expected component function, found core"),
                _ => todo!(),
            })
            .collect::<Result<Vec<_>>>()?;
        self.invoke(exec.module.map(|i| i.name()), exec.name, &values)
    }

    fn assert_return(
        &self,
        result: Result<Vec<Value>>,
        results: &[wast::WastRet<'_>],
    ) -> Result<()> {
        let values = result?;
        for (v, e) in values.iter().zip(results) {
            match e {
                wast::WastRet::Core(e) => {
                    if self.val_matches(v, e)? {
                        continue;
                    }
                }
                wast::WastRet::Component(_) => anyhow::bail!("Components not supported yet!"),
                _ => todo!(),
            }

            if let Value::V128(bits) = v {
                if let wast::WastRet::Core(WastRetCore::V128(pattern)) = e {
                    bail!(
                        "expected {:?}, got {:?} (v128 bits: {})",
                        e,
                        v128_format(*bits, pattern),
                        bits
                    );
                }
            }
            if let Some(f) = v.f64() {
                if let wast::WastRet::Core(WastRetCore::F64(wast::core::NanPattern::Value(f1))) = e
                {
                    let f = f64::from_bits(f1.bits);
                    bail!("expected {f:?} ({:?}), got {v:?} ({})", e, f.to_bits())
                } else {
                    bail!("expected {:?}, got {:?} ({})", e, v, f.to_bits())
                }
            } else if let Some(f) = v.f32() {
                if let wast::WastRet::Core(WastRetCore::F32(wast::core::NanPattern::Value(f1))) = e
                {
                    let f = f32::from_bits(f1.bits);
                    bail!("expected {f:?} ({:?}), got {v:?} ({})", e, f.to_bits())
                } else {
                    bail!("expected {:?}, got {:?} ({})", e, v, f.to_bits())
                }
            } else {
                bail!("expected {:?}, got {:?}", e, v)
            }
        }
        Ok(())
    }
    /// Define a module and register it.
    fn wat(&mut self, mut wat: QuoteWat<'_>) -> Result<()> {
        let (is_module, name) = match &wat {
            QuoteWat::Wat(wast::Wat::Module(m)) => (true, m.id.map(|v| v.name())),
            QuoteWat::QuoteModule(..) => (true, None),
            QuoteWat::Wat(wast::Wat::Component(m)) => (false, m.id.map(|v| v.name())),
            QuoteWat::QuoteComponent(..) => (false, None),
        };
        let bytes = wat.encode()?;
        if is_module {
            self.module(name, &bytes)?;
        } else {
            bail!("component-model support not enabled");
        }
        Ok(())
    }

    fn assert_trap(&self, result: Result<Vec<Value>>, expected: &str) -> Result<()> {
        let actual = match result {
            Ok(values) => bail!("expected trap, got {:?}", values),
            Err(t) => format!("{t}"),
        };
        if self.matches_message_assert_trap(expected, &actual) {
            return Ok(());
        }
        bail!("expected '{}', got '{}'", expected, actual)
    }

    fn run_directive(&mut self, _test: &Path, directive: wast::WastDirective) -> Result<()> {
        use wast::WastDirective::*;

        match directive {
            ModuleDefinition(module) => self.wat(module)?,
            Module(module) => self.wat(module)?,
            Register {
                span: _,
                name,
                module,
            } => {
                self.register(module.map(|s| s.name()), name)?;
            }
            Invoke(i) => {
                self.perform_invoke(i)?;
            }
            AssertReturn {
                span: _,
                exec,
                results,
            } => {
                let result = self.perform_execute(exec);
                self.assert_return(result, &results)?;
            }
            AssertTrap {
                span: _,
                exec,
                message,
            } => {
                if !self.disable_assert_trap_exhaustion {
                    let result = self.perform_execute(exec);
                    self.assert_trap(result, message)?;
                }
            }
            AssertExhaustion {
                span: _,
                call,
                message,
            } => {
                if !self.disable_assert_trap_exhaustion {
                    let result = self.perform_invoke(call);
                    self.assert_trap(result, message)?;
                }
            }
            AssertInvalid {
                span: _,
                module,
                message,
            } => {
                let err = match self.wat(module) {
                    Ok(()) => bail!("expected module to fail to build"),
                    Err(e) => e,
                };
                let error_message = format!("{err:?}");
                if !Self::matches_message_assert_invalid(message, &error_message) {
                    bail!(
                        "assert_invalid: expected \"{}\", got \"{}\"",
                        message,
                        error_message
                    )
                }
            }
            AssertException { span: _, exec } => {
                if !self.disable_assert_exception {
                    let result = self.perform_execute(exec);
                    self.assert_exception(result)?;
                }
            }
            AssertMalformed {
                module,
                span: _,
                message: _,
            } => {
                let mut module = match module {
                    wast::QuoteWat::Wat(m) => m,
                    // This is a `*.wat` parser test which we're not
                    // interested in.
                    wast::QuoteWat::QuoteModule(_, _) => return Ok(()),
                    wast::QuoteWat::QuoteComponent(_, _) => {
                        anyhow::bail!("Components not supported!")
                    }
                };
                let bytes = module.encode()?;
                if self.module(None, &bytes).is_ok() {
                    bail!("expected malformed module to fail to instantiate");
                }
            }
            AssertUnlinkable {
                span: _,
                mut module,
                message,
            } => {
                let bytes = module.encode()?;
                let err = match self.module(None, &bytes) {
                    Ok(()) => bail!("expected module to fail to link"),
                    Err(e) => e,
                };
                let error_message = format!("{err:?}");
                if !Self::matches_message_assert_unlinkable(message, &error_message) {
                    bail!(
                        "assert_unlinkable: expected {}, got {}",
                        message,
                        error_message
                    )
                }
            }
            Thread(_) => anyhow::bail!("`thread` directives not implemented yet!"),
            Wait { .. } => anyhow::bail!("`wait` directives not implemented yet!"),
            ModuleInstance { .. } => {
                anyhow::bail!("module instance directive not implemented yet!")
            }
            AssertSuspension { .. } => {
                anyhow::bail!("`assert suspension` directive not implemented yet!")
            }
        }

        Ok(())
    }

    /// Run a wast script from a byte buffer.
    pub fn run_buffer(&mut self, test: &Path, wast: &[u8]) -> Result<()> {
        let wast = str::from_utf8(wast)?;
        let filename = test.to_str().unwrap();
        let adjust_wast = |mut err: wast::Error| {
            err.set_path(filename.as_ref());
            err.set_text(wast);
            err
        };

        let mut lexer = Lexer::new(wast);
        lexer.allow_confusing_unicode(filename.ends_with("names.wast"));
        let buf = wast::parser::ParseBuffer::new_with_lexer(lexer).map_err(adjust_wast)?;
        let ast = parser::parse::<WWast>(&buf).map_err(adjust_wast)?;

        let mut errors = Vec::with_capacity(ast.directives.len());
        for directive in ast.directives {
            let sp = directive.span();
            if let Err(e) = self.run_directive(test, directive) {
                let message = format!("{e}");
                // If depends on an instance that doesn't exist
                if message.contains("no previous instance found") {
                    continue;
                }
                // We don't compute it, comes from instantiating an instance
                // that we expected to fail.
                if self.current.is_none() && self.current_is_allowed_failure {
                    continue;
                }
                let (line, col) = sp.linecol_in(wast);
                errors.push(DirectiveError {
                    line: line + 1,
                    col,
                    message,
                });
                if self.fail_fast {
                    break;
                }
            }
        }
        if !errors.is_empty() {
            return Err(DirectiveErrors {
                filename: filename.to_string(),
                errors,
            }
            .into());
        }
        Ok(())
    }

    //fn parse_quote_module(&self, test: &Path, source: &[&[u8]]) -> Result<Vec<u8>> {
    //    let mut ret = String::new();
    //    for src in source {
    //        match str::from_utf8(src) {
    //            Ok(s) => ret.push_str(s),
    //            Err(_) => bail!("malformed UTF-8 encoding"),
    //        }
    //        ret.push(' ');
    //    }
    //    let buf = wast::parser::ParseBuffer::new(&ret)?;
    //    let mut wat = wast::parser::parse::<wast::Wat>(&buf)?;

    //    // TODO: when memory64 merges into the proper spec then this should be
    //    // removed since it will presumably no longer be a text-format error but
    //    // rather a validation error. Currently all non-memory64 proposals
    //    // assert that this offset is a text-parser error, whereas with memory64
    //    // support that error is deferred until later.
    //    if ret.contains("offset=4294967296") && !test.iter().any(|t| t == "memory64") {
    //        bail!("i32 constant out of bounds");
    //    }
    //    Ok(wat.encode()?)
    //}

    /// Run a wast script from a file.
    pub fn run_file(&mut self, path: &Path) -> Result<()> {
        let bytes = std::fs::read(path)?;
        self.run_buffer(path, &bytes)
    }
}

// This is the implementation specific to the Runtime
impl Wast {
    /// Define a module and register it.
    fn module(&mut self, instance_name: Option<&str>, module: &[u8]) -> Result<()> {
        let instance = match self.instantiate(module) {
            Ok(i) => i,
            Err(e) => {
                // We set the current to None to allow running other
                // spectests when `fail_fast` is `false`.
                self.current = None;
                let error_message = format!("{e}");
                self.current_is_allowed_failure = false;
                for allowed_failure in self.allowed_instantiation_failures.iter() {
                    if error_message.contains(allowed_failure) {
                        self.current_is_allowed_failure = true;
                        break;
                    }
                }
                bail!("instantiation failed with: {}", e)
            }
        };
        if let Some(name) = instance_name {
            self.instances.insert(name.to_string(), instance.clone());
        }
        self.current = Some(instance);
        self.current_is_allowed_failure = false;
        Ok(())
    }

    fn instantiate(&mut self, module: &[u8]) -> Result<Instance> {
        let module = Module::new(&self.store, module)?;
        let mut imports = self.import_object.clone();

        for import in module.imports() {
            let module_name = import.module();
            if imports.contains_namespace(module_name) {
                continue;
            }
            let instance = self
                .instances
                .get(module_name)
                .ok_or_else(|| anyhow!("constant expression required"))?;
            imports.register_namespace(module_name, instance.exports.clone());
        }

        let instance = Instance::new(&mut self.store, &module, &imports)?;
        Ok(instance)
    }

    /// Register an instance to make it available for performing actions.
    fn register(&mut self, name: Option<&str>, as_name: &str) -> Result<()> {
        let instance = self.get_instance(name)?;
        self.instances.insert(as_name.to_string(), instance);
        Ok(())
    }

    /// Invoke an exported function from an instance.
    fn invoke(
        &mut self,
        instance_name: Option<&str>,
        field: &str,
        args: &[Value],
    ) -> Result<Vec<Value>> {
        let instance = self.get_instance(instance_name)?;
        let func: &Function = instance.exports.get(field)?;
        match func.call(&mut self.store, args) {
            Ok(result) => Ok(result.into()),
            Err(e) => Err(e.into()),
        }
    }

    /// Get the value of an exported global from an instance.
    fn get(&mut self, instance_name: Option<&str>, field: &str) -> Result<Vec<Value>> {
        let instance = self.get_instance(instance_name)?;
        let global: &Global = instance.exports.get(field)?;
        Ok(vec![global.get(&mut self.store)])
    }

    /// Translate from a `script::Value` to a `Value`.
    fn runtime_value(&mut self, v: &WastArgCore) -> Result<Value> {
        use wast::core::WastArgCore::*;

        Ok(match v {
            I32(x) => Value::I32(*x),
            I64(x) => Value::I64(*x),
            F32(x) => Value::F32(f32::from_bits(x.bits)),
            F64(x) => Value::F64(f64::from_bits(x.bits)),
            V128(x) => Value::V128(u128::from_le_bytes(x.to_le_bytes())),
            RefNull(HeapType::Abstract {
                ty: AbstractHeapType::Func,
                ..
            }) => Value::FuncRef(None),
            RefNull(HeapType::Abstract {
                ty: AbstractHeapType::Extern,
                ..
            }) => Value::null(),
            RefExtern(number) => Value::ExternRef(Some(ExternRef::new(&mut self.store, *number))),
            other => bail!("couldn't convert {:?} to a runtime value", other),
        })
    }

    // Checks if the `assert_unlinkable` message matches the expected one
    fn matches_message_assert_unlinkable(expected: &str, actual: &str) -> bool {
        actual.contains(expected)
    }

    // Checks if the `assert_invalid` message matches the expected one
    fn matches_message_assert_invalid(expected: &str, actual: &str) -> bool {
        actual.contains(expected)
            // Waiting on https://github.com/WebAssembly/bulk-memory-operations/pull/137
            // to propagate to WebAssembly/testsuite.
            || (expected.contains("unknown table") && actual.contains("unknown elem"))
            // wasmparser return the wrong message
            || (expected.contains("unknown memory") && actual.contains("no linear memories are present"))
            // `elem.wast` and `proposals/bulk-memory-operations/elem.wast` disagree
            // on the expected error message for the same error.
            || (expected.contains("out of bounds") && actual.contains("does not fit"))
            // handle `unknown global $NUM` error messages that wasmparser doesn't return yet
            || (expected.contains("unknown global") && actual.contains("unknown global"))
            // handle `unknown memory $NUM` error messages that wasmparser doesn't return yet
            || (expected.contains("unknown memory") && actual.contains("unknown memory"))
            || (expected.contains("unknown memory") && actual.contains("Data segment extends past end of the data section"))
            || (expected.contains("unknown elem segment") && actual.contains("unknown element segment"))
            // The same test here is asserted to have one error message in
            // `memory.wast` and a different error message in
            // `memory64/memory.wast`, so we equate these two error messages to get
            // the memory64 tests to pass.
            || (expected.contains("memory size must be at most 65536 pages") && actual.contains("invalid u32 number"))
            // the spec test suite asserts a different error message than we print
            // for this scenario
            || (expected == "unknown global" && actual.contains("global.get of locally defined global"))
            || (expected == "immutable global" && actual.contains("global is immutable: cannot modify it with `global.set`"))
            || (expected.contains("type mismatch: instruction requires") && actual.contains("instantiation failed with: Validation error: type mismatch: expected"))
    }

    // Checks if the `assert_trap` message matches the expected one
    fn matches_message_assert_trap(&self, expected: &str, actual: &str) -> bool {
        actual.contains(expected)
            || self
                .match_trap_messages
                .get(expected)
                .map_or(false, |alternative| actual.contains(alternative))
    }

    fn assert_exception(&self, result: Result<Vec<Value>>) -> Result<()> {
        if result.is_ok() {
            anyhow::bail!("Expected exception to be thrown, returned {result:?} instead");
        }
        Ok(())
    }

    fn val_matches(&self, actual: &Value, expected: &WastRetCore) -> Result<bool> {
        Ok(match (actual, expected) {
            (Value::I32(a), WastRetCore::I32(b)) => a == b,
            (Value::I64(a), WastRetCore::I64(b)) => a == b,
            // Note that these float comparisons are comparing bits, not float
            // values, so we're testing for bit-for-bit equivalence
            (Value::F32(a), WastRetCore::F32(b)) => f32_matches(*a, b),
            (Value::F64(a), WastRetCore::F64(b)) => f64_matches(*a, b),
            (Value::V128(a), WastRetCore::V128(b)) => v128_matches(*a, b),
            (
                Value::FuncRef(None),
                WastRetCore::RefNull(Some(wast::core::HeapType::Abstract {
                    ty: AbstractHeapType::Func,
                    ..
                })),
            ) => true,
            (Value::FuncRef(Some(_)), WastRetCore::RefNull(_)) => false,
            (Value::FuncRef(None), WastRetCore::RefFunc(None)) => true,
            (Value::FuncRef(None), WastRetCore::RefFunc(Some(_))) => false,
            (
                Value::ExternRef(None),
                WastRetCore::RefNull(Some(wast::core::HeapType::Abstract {
                    ty: AbstractHeapType::Extern,
                    ..
                })),
            ) => true,
            (Value::ExternRef(None), WastRetCore::RefExtern(_)) => false,
            (Value::ExceptionRef(None), WastRetCore::RefNull(_)) => true,
            (Value::ExternRef(Some(_)), WastRetCore::RefNull(_)) => false,
            (Value::ExternRef(Some(extern_ref)), WastRetCore::RefExtern(num)) => {
                let x = extern_ref.downcast::<u32>(&self.store).cloned();
                x == *num
            }
            _ => bail!(
                "don't know how to compare {:?} and {:?} yet",
                actual,
                expected
            ),
        })
    }
}

fn extract_lane_as_i8(bytes: u128, lane: usize) -> i8 {
    (bytes >> (lane * 8)) as i8
}

fn extract_lane_as_i16(bytes: u128, lane: usize) -> i16 {
    (bytes >> (lane * 16)) as i16
}

fn extract_lane_as_i32(bytes: u128, lane: usize) -> i32 {
    (bytes >> (lane * 32)) as i32
}

fn extract_lane_as_i64(bytes: u128, lane: usize) -> i64 {
    (bytes >> (lane * 64)) as i64
}

fn f32_matches(actual: f32, expected: &wast::core::NanPattern<F32>) -> bool {
    match expected {
        wast::core::NanPattern::CanonicalNan => actual.is_canonical_nan(),
        wast::core::NanPattern::ArithmeticNan => actual.is_arithmetic_nan(),
        wast::core::NanPattern::Value(expected_value) => actual.to_bits() == expected_value.bits,
    }
}

fn f64_matches(actual: f64, expected: &wast::core::NanPattern<F64>) -> bool {
    match expected {
        wast::core::NanPattern::CanonicalNan => actual.is_canonical_nan(),
        wast::core::NanPattern::ArithmeticNan => actual.is_arithmetic_nan(),
        wast::core::NanPattern::Value(expected_value) => actual.to_bits() == expected_value.bits,
    }
}

fn v128_matches(actual: u128, expected: &wast::core::V128Pattern) -> bool {
    match expected {
        wast::core::V128Pattern::I8x16(b) => b
            .iter()
            .enumerate()
            .all(|(i, b)| *b == extract_lane_as_i8(actual, i)),
        wast::core::V128Pattern::I16x8(b) => b
            .iter()
            .enumerate()
            .all(|(i, b)| *b == extract_lane_as_i16(actual, i)),
        wast::core::V128Pattern::I32x4(b) => b
            .iter()
            .enumerate()
            .all(|(i, b)| *b == extract_lane_as_i32(actual, i)),
        wast::core::V128Pattern::I64x2(b) => b
            .iter()
            .enumerate()
            .all(|(i, b)| *b == extract_lane_as_i64(actual, i)),
        wast::core::V128Pattern::F32x4(b) => b.iter().enumerate().all(|(i, b)| {
            let a = extract_lane_as_i32(actual, i) as u32;
            f32_matches(f32::from_bits(a), b)
        }),
        wast::core::V128Pattern::F64x2(b) => b.iter().enumerate().all(|(i, b)| {
            let a = extract_lane_as_i64(actual, i) as u64;
            f64_matches(f64::from_bits(a), b)
        }),
    }
}

fn v128_format(actual: u128, expected: &wast::core::V128Pattern) -> wast::core::V128Pattern {
    match expected {
        wast::core::V128Pattern::I8x16(_) => wast::core::V128Pattern::I8x16([
            extract_lane_as_i8(actual, 0),
            extract_lane_as_i8(actual, 1),
            extract_lane_as_i8(actual, 2),
            extract_lane_as_i8(actual, 3),
            extract_lane_as_i8(actual, 4),
            extract_lane_as_i8(actual, 5),
            extract_lane_as_i8(actual, 6),
            extract_lane_as_i8(actual, 7),
            extract_lane_as_i8(actual, 8),
            extract_lane_as_i8(actual, 9),
            extract_lane_as_i8(actual, 10),
            extract_lane_as_i8(actual, 11),
            extract_lane_as_i8(actual, 12),
            extract_lane_as_i8(actual, 13),
            extract_lane_as_i8(actual, 14),
            extract_lane_as_i8(actual, 15),
        ]),
        wast::core::V128Pattern::I16x8(_) => wast::core::V128Pattern::I16x8([
            extract_lane_as_i16(actual, 0),
            extract_lane_as_i16(actual, 1),
            extract_lane_as_i16(actual, 2),
            extract_lane_as_i16(actual, 3),
            extract_lane_as_i16(actual, 4),
            extract_lane_as_i16(actual, 5),
            extract_lane_as_i16(actual, 6),
            extract_lane_as_i16(actual, 7),
        ]),
        wast::core::V128Pattern::I32x4(_) => wast::core::V128Pattern::I32x4([
            extract_lane_as_i32(actual, 0),
            extract_lane_as_i32(actual, 1),
            extract_lane_as_i32(actual, 2),
            extract_lane_as_i32(actual, 3),
        ]),
        wast::core::V128Pattern::I64x2(_) => wast::core::V128Pattern::I64x2([
            extract_lane_as_i64(actual, 0),
            extract_lane_as_i64(actual, 1),
        ]),
        wast::core::V128Pattern::F32x4(_) => wast::core::V128Pattern::F32x4([
            wast::core::NanPattern::Value(F32 {
                bits: extract_lane_as_i32(actual, 0) as _,
            }),
            wast::core::NanPattern::Value(F32 {
                bits: extract_lane_as_i32(actual, 1) as _,
            }),
            wast::core::NanPattern::Value(F32 {
                bits: extract_lane_as_i32(actual, 2) as _,
            }),
            wast::core::NanPattern::Value(F32 {
                bits: extract_lane_as_i32(actual, 3) as _,
            }),
        ]),
        wast::core::V128Pattern::F64x2(_) => wast::core::V128Pattern::F64x2([
            wast::core::NanPattern::Value(F64 {
                bits: extract_lane_as_i64(actual, 0) as _,
            }),
            wast::core::NanPattern::Value(F64 {
                bits: extract_lane_as_i64(actual, 1) as _,
            }),
        ]),
    }
}

pub trait NaNCheck {
    fn is_arithmetic_nan(&self) -> bool;
    fn is_canonical_nan(&self) -> bool;
}

impl NaNCheck for f32 {
    fn is_arithmetic_nan(&self) -> bool {
        const AF32_NAN: u32 = 0x0040_0000;
        (self.to_bits() & AF32_NAN) == AF32_NAN
    }

    fn is_canonical_nan(&self) -> bool {
        (self.to_bits() & 0x7fff_ffff) == 0x7fc0_0000
    }
}

impl NaNCheck for f64 {
    fn is_arithmetic_nan(&self) -> bool {
        const AF64_NAN: u64 = 0x0008_0000_0000_0000;
        (self.to_bits() & AF64_NAN) == AF64_NAN
    }

    fn is_canonical_nan(&self) -> bool {
        (self.to_bits() & 0x7fff_ffff_ffff_ffff) == 0x7ff8_0000_0000_0000
    }
}