aver-lang 0.18.0

//! Effectful builtin tracking — `Console.print`, `Http.get`, etc.
//!
//! These don't get bodies in the user module. Instead, the codegen
//! emits `(import "aver" "<name>" (func ...))` and the host (browser
//! / workerd / wasmtime+wasi) supplies the implementation. Same shape
//! the legacy backend uses for effects, just without the
//! `aver_runtime.wasm` middleman.
//!
//! Imports take the lowest fn indices in wasm — `0..K` for K
//! registered effects. User fn indices and builtin helper fn
//! indices shift up by K.

use std::collections::HashMap;

use wasm_encoder::ValType;

use super::WasmGcError;
use super::types::TypeRegistry;

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub(super) enum EffectName {
    /// `Console.print(String) -> Unit`. On `TargetMode::AverBridge`,
    /// imported as `aver.console_print` (host writes to stdout). On
    /// `TargetMode::Wasip2` the same source-side effect lowers to
    /// canonical-ABI `wasi:cli/stdout.get-stdout` (cached) plus
    /// `wasi:io/streams.[method]output-stream.blocking-write-and-flush`
    /// in the per-effect glue at the call site — see
    /// `EffectName::lowers_on_wasip2` and `Wasip2ImportSlot`.
    ConsolePrint,
    ConsoleError,
    ConsoleWarn,
    /// `Time.unixMs() -> Int`. Imported as `aver.time_unix_ms` — host
    /// supplies the current unix timestamp in milliseconds.
    TimeUnixMs,
    // ── Fetch bridge — used by `--handler X` on `--target wasm-gc`.
    //   The synthesised `aver_http_handle` wrapper reads request
    //   fields through these and dispatches the user's HttpResponse
    //   via `Response.text` / `Response.setHeader`. Native wasm-gc
    //   shapes — refs to `(array i8)` carriers, not `(ptr, len)`
    //   pairs. Map<String, List<String>> for headers carries the
    //   per-instance struct ref the registry's Map slot allocates.
    /// `() -> String` — HTTP method (`"GET"`, `"POST"`, …).
    RequestMethod,
    /// `() -> String` — pathname only (no query string).
    RequestUrl,
    /// `() -> String` — query string after `?`, no leading `?`.
    RequestQuery,
    /// `() -> String` — request body (empty string when absent).
    RequestBody,
    /// `() -> Map<String, List<String>>` — host-supplied request
    /// headers map. Empty map when the request has no headers.
    RequestHeadersLoad,
    /// `(status: Int, body: String) -> Unit` — finalize the
    /// pending response. Header writes (`Response.setHeader`) must
    /// happen BEFORE this call.
    ResponseText,
    /// `(name: String, value: String) -> Unit` — append a header
    /// to the pending response. Multi-value headers are produced
    /// by repeated calls with the same name.
    ResponseSetHeader,
    // ── Outgoing HTTP (Http.* surface) — the aver_http_handle
    //   wrapper doesn't use these, but user code reachable from
    //   the handler may; register them so the typechecker's
    //   `! [Http.send]` declaration has a wasm import to resolve.
    /// `(method: String, url: String, body: String, contentType: String)
    /// -> (status: Int, body: String, headers: Map<String, List<String>>, err: String)`.
    HttpSend,
    /// `(name: String, value: String) -> Unit`.
    HttpAddRequestHeader,
    /// `() -> Unit`.
    HttpClearRequestHeaders,
    /// `(name: String) -> String` — Workers env binding lookup;
    /// returns empty string when the binding is absent.
    EnvGet,
    /// `(name: String, value: String) -> Unit` — no-op on
    /// Workers (env is read-only).
    EnvSet,
    // ── CLI / runtime side effects covered by the legacy
    //    `abi.rs` table; ported one-for-one so wasm-gc has the
    //    same surface area.
    /// `() -> String` — read one line from stdin.
    ConsoleReadLine,
    /// `() -> Int` — number of CLI args.
    ArgsLen,
    /// `(i: Int) -> String` — i-th CLI arg.
    ArgsGet,
    /// `() -> Float` — uniform [0, 1).
    RandomFloat,
    /// `(lo: Int, hi: Int) -> Int` — uniform inclusive on both ends.
    RandomInt,
    /// `(ms: Int) -> Unit` — block (or no-op when the host can't).
    TimeSleep,
    /// `() -> String` — RFC 3339 timestamp from the host clock.
    TimeNow,
    /// libm bridges — wasm has no native sin/cos/atan2/pow, so the
    /// host (Math.sin / `f64::sin`) supplies them.
    FloatSin,
    FloatCos,
    FloatAtan2,
    FloatPow,
    // ── Terminal raw-mode TUI surface — same set the legacy `abi.rs`
    //   ports. Hosts that don't implement Terminal (workers, browser)
    //   should treat these as no-ops; Aver `! [Terminal.foo]` effect
    //   declarations drive when imports are emitted.
    /// `() -> Unit` — switch the host terminal to raw mode (no line
    /// buffering, no echo). Pairs with `Terminal.disableRawMode`.
    TerminalEnableRawMode,
    /// `() -> Unit` — restore the host terminal's pre-raw mode.
    TerminalDisableRawMode,
    /// `() -> Unit` — clear the entire terminal screen.
    TerminalClear,
    /// `(row: Int, col: Int) -> Unit` — move the cursor to the given
    /// 1-indexed row/col.
    TerminalMoveTo,
    /// `(s: String) -> Unit` — write `s` to the terminal at the
    /// current cursor position. Doesn't append a newline.
    TerminalPrint,
    /// `(color: Int) -> Unit` — set the foreground colour to the
    /// given ANSI 256 colour code.
    TerminalSetColor,
    /// `() -> Unit` — reset foreground colour to default.
    TerminalResetColor,
    /// `() -> Option<String>` — read one keypress non-blocking. None
    /// if no input is available; Some(byte_string) for printable +
    /// special-key sequences.
    TerminalReadKey,
    /// `() -> Terminal.Size` — current terminal dimensions in
    /// (width, height) cells.
    TerminalSize,
    /// `() -> Unit` — hide the cursor.
    TerminalHideCursor,
    /// `() -> Unit` — show the cursor.
    TerminalShowCursor,
    /// `() -> Unit` — flush any buffered terminal output to the host.
    TerminalFlush,
    // ── Disk surface — POSIX-shaped file I/O wired through aver_rt::*.
    //   `Disk.exists` returns Bool; everything else returns
    //   `Result<…, String>`. `listDir` produces `Result<List<String>, String>`,
    //   the rest a `Result<Unit, String>` on the failure side and either
    //   `Result<String, String>` (read) or `Result<Unit, String>` (write/
    //   append/delete/mkdir) on success.
    DiskReadText,
    DiskWriteText,
    DiskAppendText,
    DiskExists,
    DiskDelete,
    DiskDeleteDir,
    DiskListDir,
    DiskMakeDir,
    // ── TCP surface — sessions backed by a thread-local socket pool
    //   in `aver_rt::tcp`. Connection records cross the boundary as
    //   `Tcp.Connection` struct refs.
    TcpConnect,
    TcpWriteLine,
    TcpReadLine,
    TcpClose,
    TcpSend,
    TcpPing,
    // ── Outgoing HTTP verbs. `Http.send` already exists for the
    //   Workers handler path; the surface verbs (`Http.get` etc.)
    //   route through the same `aver_rt::http` impl.
    HttpGet,
    HttpHead,
    HttpDelete,
    HttpPost,
    HttpPut,
    HttpPatch,
    // ── Independent-product structural-scope markers. Emitted by the
    //    wasm-gc body lowering wherever `?!` (or `!`) appears, so the
    //    recorder annotates each contained effect with its
    //    `(group_id, branch_path, effect_occurrence)` tuple. Replay
    //    matches these instead of strict sequence position. Without
    //    these calls, wasm-gc traces would be flat and cross-backend
    //    replay against a VM-recorded `?!` trace would break.
    /// `() -> Unit` — open a fresh independent-product scope.
    RecordEnterGroup,
    /// `(branch: Int) -> Unit` — switch the recorder's active branch
    /// inside the current group.
    RecordSetBranch,
    /// `() -> Unit` — close the most recently opened scope.
    RecordExitGroup,
}

impl EffectName {
    pub(super) fn from_dotted(s: &str) -> Option<Self> {
        match s {
            "Console.print" => Some(Self::ConsolePrint),
            "Console.error" => Some(Self::ConsoleError),
            "Console.warn" => Some(Self::ConsoleWarn),
            "Time.unixMs" => Some(Self::TimeUnixMs),
            "Request.method" => Some(Self::RequestMethod),
            "Request.url" | "Request.path" => Some(Self::RequestUrl),
            "Request.query" => Some(Self::RequestQuery),
            "Request.body" => Some(Self::RequestBody),
            "Request.headersLoad" | "Request.headers" => Some(Self::RequestHeadersLoad),
            "Response.text" => Some(Self::ResponseText),
            "Response.setHeader" => Some(Self::ResponseSetHeader),
            "Http.send" => Some(Self::HttpSend),
            "Http.addRequestHeader" => Some(Self::HttpAddRequestHeader),
            "Http.clearRequestHeaders" => Some(Self::HttpClearRequestHeaders),
            "Env.get" => Some(Self::EnvGet),
            "Env.set" => Some(Self::EnvSet),
            "Console.readLine" => Some(Self::ConsoleReadLine),
            "Args._len" | "Args.len" => Some(Self::ArgsLen),
            // `Args._get` is the low-level host import (i64 idx → String).
            // The user-facing `Args.get()` (no args, returns List<String>)
            // doesn't map to a single effect — `body/builtins.rs::emit_args_get_inline`
            // expands it to `args_len + loop args_get(i) cons`. The
            // discovery walker in `module.rs` registers ArgsLen + ArgsGet
            // separately whenever it sees `Args.get()` so both host
            // imports are linked.
            "Args._get" => Some(Self::ArgsGet),
            "Random.float" => Some(Self::RandomFloat),
            "Random.int" => Some(Self::RandomInt),
            "Time.sleep" => Some(Self::TimeSleep),
            "Time.now" => Some(Self::TimeNow),
            "Float.sin" => Some(Self::FloatSin),
            "Float.cos" => Some(Self::FloatCos),
            "Float.atan2" => Some(Self::FloatAtan2),
            "Float.pow" => Some(Self::FloatPow),
            "Terminal.enableRawMode" => Some(Self::TerminalEnableRawMode),
            "Terminal.disableRawMode" => Some(Self::TerminalDisableRawMode),
            "Terminal.clear" => Some(Self::TerminalClear),
            "Terminal.moveTo" => Some(Self::TerminalMoveTo),
            "Terminal.print" => Some(Self::TerminalPrint),
            "Terminal.setColor" => Some(Self::TerminalSetColor),
            "Terminal.resetColor" => Some(Self::TerminalResetColor),
            "Terminal.readKey" => Some(Self::TerminalReadKey),
            "Terminal.size" => Some(Self::TerminalSize),
            "Terminal.hideCursor" => Some(Self::TerminalHideCursor),
            "Terminal.showCursor" => Some(Self::TerminalShowCursor),
            "Terminal.flush" => Some(Self::TerminalFlush),
            "Disk.readText" => Some(Self::DiskReadText),
            "Disk.writeText" => Some(Self::DiskWriteText),
            "Disk.appendText" => Some(Self::DiskAppendText),
            "Disk.exists" => Some(Self::DiskExists),
            "Disk.delete" => Some(Self::DiskDelete),
            "Disk.deleteDir" => Some(Self::DiskDeleteDir),
            "Disk.listDir" => Some(Self::DiskListDir),
            "Disk.makeDir" => Some(Self::DiskMakeDir),
            "Tcp.connect" => Some(Self::TcpConnect),
            "Tcp.writeLine" => Some(Self::TcpWriteLine),
            "Tcp.readLine" => Some(Self::TcpReadLine),
            "Tcp.close" => Some(Self::TcpClose),
            "Tcp.send" => Some(Self::TcpSend),
            "Tcp.ping" => Some(Self::TcpPing),
            "Http.get" => Some(Self::HttpGet),
            "Http.head" => Some(Self::HttpHead),
            "Http.delete" => Some(Self::HttpDelete),
            "Http.post" => Some(Self::HttpPost),
            "Http.put" => Some(Self::HttpPut),
            "Http.patch" => Some(Self::HttpPatch),
            _ => None,
        }
    }

    pub(super) fn canonical(self) -> &'static str {
        match self {
            Self::ConsolePrint => "Console.print",
            Self::ConsoleError => "Console.error",
            Self::ConsoleWarn => "Console.warn",
            Self::TimeUnixMs => "Time.unixMs",
            Self::RequestMethod => "Request.method",
            Self::RequestUrl => "Request.url",
            Self::RequestQuery => "Request.query",
            Self::RequestBody => "Request.body",
            Self::RequestHeadersLoad => "Request.headersLoad",
            Self::ResponseText => "Response.text",
            Self::ResponseSetHeader => "Response.setHeader",
            Self::HttpSend => "Http.send",
            Self::HttpAddRequestHeader => "Http.addRequestHeader",
            Self::HttpClearRequestHeaders => "Http.clearRequestHeaders",
            Self::EnvGet => "Env.get",
            Self::EnvSet => "Env.set",
            Self::ConsoleReadLine => "Console.readLine",
            Self::ArgsLen => "Args.len",
            // `Args.get` is the user-facing name; the `Args._get`
            // alias exists for legacy reasons but the dotted dispatch
            // in `emit_dotted_builtin` looks up by the source-level
            // name. Use the user-facing form as the canonical so
            // `fn_map.effects.get("Args.get")` succeeds.
            Self::ArgsGet => "Args.get",
            Self::RandomFloat => "Random.float",
            Self::RandomInt => "Random.int",
            Self::TimeSleep => "Time.sleep",
            Self::TimeNow => "Time.now",
            Self::FloatSin => "Float.sin",
            Self::FloatCos => "Float.cos",
            Self::FloatAtan2 => "Float.atan2",
            Self::FloatPow => "Float.pow",
            Self::TerminalEnableRawMode => "Terminal.enableRawMode",
            Self::TerminalDisableRawMode => "Terminal.disableRawMode",
            Self::TerminalClear => "Terminal.clear",
            Self::TerminalMoveTo => "Terminal.moveTo",
            Self::TerminalPrint => "Terminal.print",
            Self::TerminalSetColor => "Terminal.setColor",
            Self::TerminalResetColor => "Terminal.resetColor",
            Self::TerminalReadKey => "Terminal.readKey",
            Self::TerminalSize => "Terminal.size",
            Self::TerminalHideCursor => "Terminal.hideCursor",
            Self::TerminalShowCursor => "Terminal.showCursor",
            Self::TerminalFlush => "Terminal.flush",
            Self::DiskReadText => "Disk.readText",
            Self::DiskWriteText => "Disk.writeText",
            Self::DiskAppendText => "Disk.appendText",
            Self::DiskExists => "Disk.exists",
            Self::DiskDelete => "Disk.delete",
            Self::DiskDeleteDir => "Disk.deleteDir",
            Self::DiskListDir => "Disk.listDir",
            Self::DiskMakeDir => "Disk.makeDir",
            Self::TcpConnect => "Tcp.connect",
            Self::TcpWriteLine => "Tcp.writeLine",
            Self::TcpReadLine => "Tcp.readLine",
            Self::TcpClose => "Tcp.close",
            Self::TcpSend => "Tcp.send",
            Self::TcpPing => "Tcp.ping",
            Self::HttpGet => "Http.get",
            Self::HttpHead => "Http.head",
            Self::HttpDelete => "Http.delete",
            Self::HttpPost => "Http.post",
            Self::HttpPut => "Http.put",
            Self::HttpPatch => "Http.patch",
            // Group markers aren't user-facing effects; the dotted
            // form is internal-only. Used by `effect_idx_lookup`
            // string keying so the body emit can resolve them.
            Self::RecordEnterGroup => "__record_enter_group",
            Self::RecordSetBranch => "__record_set_branch",
            Self::RecordExitGroup => "__record_exit_group",
        }
    }

    /// Wasm import (module, field) pair. Module is always `aver` for
    /// our effects — host supplies a single namespace.
    pub(super) fn import_pair(self) -> (&'static str, &'static str) {
        match self {
            Self::ConsolePrint => ("aver", "console_print"),
            Self::ConsoleError => ("aver", "console_error"),
            Self::ConsoleWarn => ("aver", "console_warn"),
            Self::TimeUnixMs => ("aver", "time_unix_ms"),
            Self::RequestMethod => ("aver", "request_method"),
            Self::RequestUrl => ("aver", "request_url"),
            Self::RequestQuery => ("aver", "request_query"),
            Self::RequestBody => ("aver", "request_body"),
            Self::RequestHeadersLoad => ("aver", "request_headers_load"),
            Self::ResponseText => ("aver", "response_text"),
            Self::ResponseSetHeader => ("aver", "response_set_header"),
            Self::HttpSend => ("aver", "http_send"),
            Self::HttpAddRequestHeader => ("aver", "http_add_request_header"),
            Self::HttpClearRequestHeaders => ("aver", "http_clear_request_headers"),
            Self::EnvGet => ("aver", "env_get"),
            Self::EnvSet => ("aver", "env_set"),
            Self::ConsoleReadLine => ("aver", "console_read_line"),
            Self::ArgsLen => ("aver", "args_len"),
            Self::ArgsGet => ("aver", "args_get"),
            Self::RandomFloat => ("aver", "random_float"),
            Self::RandomInt => ("aver", "random_int"),
            Self::TimeSleep => ("aver", "time_sleep"),
            Self::TimeNow => ("aver", "time_now"),
            Self::FloatSin => ("aver", "float_sin"),
            Self::FloatCos => ("aver", "float_cos"),
            Self::FloatAtan2 => ("aver", "float_atan2"),
            Self::FloatPow => ("aver", "float_pow"),
            Self::TerminalEnableRawMode => ("aver", "terminal_enable_raw_mode"),
            Self::TerminalDisableRawMode => ("aver", "terminal_disable_raw_mode"),
            Self::TerminalClear => ("aver", "terminal_clear"),
            Self::TerminalMoveTo => ("aver", "terminal_move_to"),
            Self::TerminalPrint => ("aver", "terminal_print"),
            Self::TerminalSetColor => ("aver", "terminal_set_color"),
            Self::TerminalResetColor => ("aver", "terminal_reset_color"),
            Self::TerminalReadKey => ("aver", "terminal_read_key"),
            Self::TerminalSize => ("aver", "terminal_size"),
            Self::TerminalHideCursor => ("aver", "terminal_hide_cursor"),
            Self::TerminalShowCursor => ("aver", "terminal_show_cursor"),
            Self::TerminalFlush => ("aver", "terminal_flush"),
            Self::DiskReadText => ("aver", "disk_read_text"),
            Self::DiskWriteText => ("aver", "disk_write_text"),
            Self::DiskAppendText => ("aver", "disk_append_text"),
            Self::DiskExists => ("aver", "disk_exists"),
            Self::DiskDelete => ("aver", "disk_delete"),
            Self::DiskDeleteDir => ("aver", "disk_delete_dir"),
            Self::DiskListDir => ("aver", "disk_list_dir"),
            Self::DiskMakeDir => ("aver", "disk_make_dir"),
            Self::TcpConnect => ("aver", "tcp_connect"),
            Self::TcpWriteLine => ("aver", "tcp_write_line"),
            Self::TcpReadLine => ("aver", "tcp_read_line"),
            Self::TcpClose => ("aver", "tcp_close"),
            Self::TcpSend => ("aver", "tcp_send"),
            Self::TcpPing => ("aver", "tcp_ping"),
            Self::HttpGet => ("aver", "http_get"),
            Self::HttpHead => ("aver", "http_head"),
            Self::HttpDelete => ("aver", "http_delete"),
            Self::HttpPost => ("aver", "http_post"),
            Self::HttpPut => ("aver", "http_put"),
            Self::HttpPatch => ("aver", "http_patch"),
            Self::RecordEnterGroup => ("aver", "record_enter_group"),
            Self::RecordSetBranch => ("aver", "record_set_branch"),
            Self::RecordExitGroup => ("aver", "record_exit_group"),
        }
    }

    /// Param types declared in the wasm import. Strings are passed as
    /// `(ref null any)` — engine subtyping accepts our `(ref null
    /// $string)` and the host doesn't have to know the concrete type
    /// idx. The headers Map crossing uses the registered concrete
    /// `Map<String, List<String>>` ref so the host bridge has a
    /// type-safe handle.
    ///
    /// **Trailing `caller_fn: any_ref`.** Every host import gains a
    /// trailing String-ref param identifying the Aver fn that
    /// emitted the call. The codegen pushes the current fn name
    /// right before `call`, the host pulls it via `lm_string_to_host`
    /// and stamps the resulting record's `caller_fn` field with it.
    /// Without this surface, `caller_fn` was always `"main"` because
    /// wasm-gc executes natively and the host has no other way to
    /// recover the originating Aver fn — VM/self-host get it from
    /// their interpreter stack frames, wasm-gc has to be told.
    pub(super) fn params(self, registry: &TypeRegistry) -> Result<Vec<ValType>, WasmGcError> {
        let mut p = self.params_without_caller(registry)?;
        // Trailing caller_fn arg is now an `i32` index into the
        // exported caller-fn name table (built by the host at
        // instantiation via `__caller_fn_count` + `__caller_fn_name`).
        // 0.16.2 used `(ref null $string)` per-call — replaced to
        // drop N globals + N start-fn allocs and to give the host
        // a vector-index lookup instead of a per-call LM round-trip.
        p.push(ValType::I32);
        Ok(p)
    }

    fn params_without_caller(self, registry: &TypeRegistry) -> Result<Vec<ValType>, WasmGcError> {
        match self {
            Self::ConsolePrint | Self::ConsoleError | Self::ConsoleWarn => Ok(vec![any_ref_ty()]),
            Self::TimeUnixMs => Ok(vec![]),
            Self::RequestMethod
            | Self::RequestUrl
            | Self::RequestQuery
            | Self::RequestBody
            | Self::RequestHeadersLoad
            | Self::HttpClearRequestHeaders => Ok(vec![]),
            Self::ResponseText => Ok(vec![ValType::I64, any_ref_ty()]),
            Self::ResponseSetHeader => Ok(vec![any_ref_ty(), any_ref_ty()]),
            Self::HttpSend => Ok(vec![any_ref_ty(), any_ref_ty(), any_ref_ty(), any_ref_ty()]),
            Self::HttpAddRequestHeader => Ok(vec![any_ref_ty(), any_ref_ty()]),
            Self::EnvGet => Ok(vec![any_ref_ty()]),
            Self::EnvSet => Ok(vec![any_ref_ty(), any_ref_ty()]),
            Self::ConsoleReadLine | Self::ArgsLen | Self::RandomFloat | Self::TimeNow => Ok(vec![]),
            Self::ArgsGet | Self::TimeSleep => Ok(vec![ValType::I64]),
            Self::RandomInt => Ok(vec![ValType::I64, ValType::I64]),
            Self::FloatSin | Self::FloatCos => Ok(vec![ValType::F64]),
            Self::FloatAtan2 | Self::FloatPow => Ok(vec![ValType::F64, ValType::F64]),
            Self::TerminalEnableRawMode
            | Self::TerminalDisableRawMode
            | Self::TerminalClear
            | Self::TerminalResetColor
            | Self::TerminalReadKey
            | Self::TerminalSize
            | Self::TerminalHideCursor
            | Self::TerminalShowCursor
            | Self::TerminalFlush => Ok(vec![]),
            Self::TerminalMoveTo => Ok(vec![ValType::I64, ValType::I64]),
            Self::TerminalPrint => Ok(vec![any_ref_ty()]),
            // Terminal.setColor takes `color: String`, not Int — same
            // shape as Terminal.print. Was previously typed as i64 by
            // mistake; emit pushes a String ref (`array.new_data`),
            // engine validator rejects with i64 vs ref mismatch as
            // soon as a setColor call site appears.
            Self::TerminalSetColor => Ok(vec![any_ref_ty()]),
            Self::DiskReadText
            | Self::DiskExists
            | Self::DiskDelete
            | Self::DiskDeleteDir
            | Self::DiskListDir
            | Self::DiskMakeDir => Ok(vec![any_ref_ty()]),
            Self::DiskWriteText | Self::DiskAppendText => Ok(vec![any_ref_ty(), any_ref_ty()]),
            // Tcp.connect: (host, port). Returns Result<Tcp.Connection, String>.
            Self::TcpConnect => Ok(vec![any_ref_ty(), ValType::I64]),
            // Tcp.writeLine / readLine / close all take a Connection as
            // first arg. Carrying it as `any_ref` lets the host
            // `struct.get` the id field directly without us having to
            // mention the concrete record type idx in the import sig.
            Self::TcpClose | Self::TcpReadLine => Ok(vec![any_ref_ty()]),
            Self::TcpWriteLine => Ok(vec![any_ref_ty(), any_ref_ty()]),
            Self::TcpSend => Ok(vec![any_ref_ty(), ValType::I64, any_ref_ty()]),
            Self::TcpPing => Ok(vec![any_ref_ty(), ValType::I64]),
            // HTTP verbs all take a URL. `Http.get` / `Http.head` /
            // `Http.delete` are arity-1; the body verbs add (body,
            // contentType, headersMap). The headers map crosses as
            // the registered `Map<String, List<String>>` ref so the
            // host can iterate (key, values) pairs.
            Self::HttpGet | Self::HttpHead | Self::HttpDelete => Ok(vec![any_ref_ty()]),
            Self::HttpPost | Self::HttpPut | Self::HttpPatch => Ok(vec![
                any_ref_ty(),
                any_ref_ty(),
                any_ref_ty(),
                map_string_list_string_ref_ty(registry)?,
            ]),
            Self::RecordEnterGroup | Self::RecordExitGroup => Ok(vec![]),
            Self::RecordSetBranch => Ok(vec![ValType::I64]),
        }
    }

    pub(super) fn results(self, registry: &TypeRegistry) -> Result<Vec<ValType>, WasmGcError> {
        match self {
            Self::ConsolePrint | Self::ConsoleError | Self::ConsoleWarn => Ok(vec![]),
            Self::TimeUnixMs => Ok(vec![ValType::I64]),
            Self::RequestMethod
            | Self::RequestUrl
            | Self::RequestQuery
            | Self::RequestBody
            | Self::EnvGet => Ok(vec![string_ref_ty(registry)?]),
            Self::RequestHeadersLoad => Ok(vec![map_string_list_string_ref_ty(registry)?]),
            Self::ResponseText
            | Self::ResponseSetHeader
            | Self::HttpAddRequestHeader
            | Self::HttpClearRequestHeaders
            | Self::EnvSet => Ok(vec![]),
            Self::HttpSend => Ok(vec![
                ValType::I64,
                string_ref_ty(registry)?,
                map_string_list_string_ref_ty(registry)?,
                string_ref_ty(registry)?,
            ]),
            Self::ConsoleReadLine => {
                // Aver-level signature is `Result<String, String>`. The
                // host returns a Result struct ref directly so the
                // caller's `?` (ErrorProp) can read the tag without an
                // intermediate wrap step.
                let idx = registry.result_type_idx("Result<String,String>").ok_or(
                    WasmGcError::Validation(
                        "Console.readLine: Result<String,String> slot not registered".into(),
                    ),
                )?;
                Ok(vec![ValType::Ref(wasm_encoder::RefType {
                    nullable: true,
                    heap_type: wasm_encoder::HeapType::Concrete(idx),
                })])
            }
            Self::ArgsGet | Self::TimeNow => Ok(vec![string_ref_ty(registry)?]),
            Self::ArgsLen | Self::RandomInt => Ok(vec![ValType::I64]),
            Self::TimeSleep => Ok(vec![]),
            Self::RandomFloat
            | Self::FloatSin
            | Self::FloatCos
            | Self::FloatAtan2
            | Self::FloatPow => Ok(vec![ValType::F64]),
            Self::TerminalEnableRawMode
            | Self::TerminalDisableRawMode
            | Self::TerminalClear
            | Self::TerminalMoveTo
            | Self::TerminalPrint
            | Self::TerminalSetColor
            | Self::TerminalResetColor
            | Self::TerminalHideCursor
            | Self::TerminalShowCursor
            | Self::TerminalFlush => Ok(vec![]),
            Self::TerminalReadKey => {
                // `Option<String>` ref — eager-registered when any
                // call site reaches for `String.charAt` / `Char.fromCode`,
                // or now also when `Terminal.readKey` shows up.
                let opt_idx =
                    registry
                        .option_type_idx("Option<String>")
                        .ok_or(WasmGcError::Validation(
                            "Terminal.readKey: Option<String> slot not registered".into(),
                        ))?;
                Ok(vec![ValType::Ref(wasm_encoder::RefType {
                    nullable: true,
                    heap_type: wasm_encoder::HeapType::Concrete(opt_idx),
                })])
            }
            Self::TerminalSize => {
                let idx =
                    registry
                        .record_type_idx("Terminal.Size")
                        .ok_or(WasmGcError::Validation(
                            "Terminal.size: Terminal.Size record slot not registered \
                         (did you call Terminal.size in user code?)"
                                .into(),
                        ))?;
                Ok(vec![ValType::Ref(wasm_encoder::RefType {
                    nullable: true,
                    heap_type: wasm_encoder::HeapType::Concrete(idx),
                })])
            }
            Self::DiskReadText => Ok(vec![result_ref_ty(registry, "Result<String,String>")?]),
            Self::DiskWriteText
            | Self::DiskAppendText
            | Self::DiskDelete
            | Self::DiskDeleteDir
            | Self::DiskMakeDir => Ok(vec![result_ref_ty(registry, "Result<Unit,String>")?]),
            Self::DiskListDir => Ok(vec![result_ref_ty(
                registry,
                "Result<List<String>,String>",
            )?]),
            Self::DiskExists => Ok(vec![ValType::I32]),
            Self::TcpConnect => Ok(vec![result_ref_ty(
                registry,
                "Result<Tcp.Connection,String>",
            )?]),
            Self::TcpReadLine | Self::TcpSend => {
                Ok(vec![result_ref_ty(registry, "Result<String,String>")?])
            }
            Self::TcpWriteLine | Self::TcpClose | Self::TcpPing => {
                Ok(vec![result_ref_ty(registry, "Result<Unit,String>")?])
            }
            Self::HttpGet
            | Self::HttpHead
            | Self::HttpDelete
            | Self::HttpPost
            | Self::HttpPut
            | Self::HttpPatch => Ok(vec![result_ref_ty(
                registry,
                "Result<HttpResponse,String>",
            )?]),
            Self::RecordEnterGroup | Self::RecordSetBranch | Self::RecordExitGroup => Ok(vec![]),
        }
    }
}

fn result_ref_ty(registry: &TypeRegistry, canonical: &str) -> Result<ValType, WasmGcError> {
    let idx = registry
        .result_type_idx(canonical)
        .ok_or(WasmGcError::Validation(format!(
            "effect requires `{canonical}` slot but none was registered"
        )))?;
    Ok(ValType::Ref(wasm_encoder::RefType {
        nullable: true,
        heap_type: wasm_encoder::HeapType::Concrete(idx),
    }))
}

fn map_string_list_string_ref_ty(registry: &TypeRegistry) -> Result<ValType, WasmGcError> {
    let slots = registry
        .map_slots("Map<String,List<String>>")
        .ok_or(WasmGcError::Validation(
            "fetch effect requires `Map<String, List<String>>` slot but none was registered".into(),
        ))?;
    Ok(ValType::Ref(wasm_encoder::RefType {
        nullable: true,
        heap_type: wasm_encoder::HeapType::Concrete(slots.map),
    }))
}

fn any_ref_ty() -> ValType {
    ValType::Ref(wasm_encoder::RefType {
        nullable: true,
        heap_type: wasm_encoder::HeapType::Abstract {
            shared: false,
            ty: wasm_encoder::AbstractHeapType::Any,
        },
    })
}

/// Per-module registry of used effects. Allocates wasm fn indices
/// starting at 0 (imports come first); the offset returned via
/// `import_count` is the value that gets added to every other fn
/// index in the module.
#[derive(Default)]
pub(super) struct EffectRegistry {
    order: Vec<EffectName>,
    wasm_fn_idx: HashMap<EffectName, u32>,
    wasm_type_idx: HashMap<EffectName, u32>,
}

impl EffectRegistry {
    pub(super) fn new() -> Self {
        Self::default()
    }

    pub(super) fn register(&mut self, name: EffectName) {
        if !self.order.contains(&name) {
            self.order.push(name);
        }
    }

    pub(super) fn iter(&self) -> impl Iterator<Item = EffectName> + '_ {
        self.order.iter().copied()
    }

    pub(super) fn import_count(&self) -> u32 {
        self.order.len() as u32
    }

    /// Reserve type and fn-idx slots for each registered effect.
    /// Imports occupy fn-idx 0..K; type indices come from the same
    /// counter the user-fn types use, deferred by the caller.
    pub(super) fn assign_slots(&mut self, next_type_idx: &mut u32) {
        for (i, name) in self.order.iter().copied().enumerate() {
            self.wasm_fn_idx.insert(name, i as u32);
            self.wasm_type_idx.insert(name, *next_type_idx);
            *next_type_idx += 1;
        }
    }

    pub(super) fn lookup_wasm_fn_idx(&self, name: EffectName) -> Option<u32> {
        self.wasm_fn_idx.get(&name).copied()
    }

    pub(super) fn lookup_wasm_type_idx(&self, name: EffectName) -> Option<u32> {
        self.wasm_type_idx.get(&name).copied()
    }
}

impl EffectName {
    /// True iff this effect is wired on `TargetMode::Wasip2` today.
    /// Drives the population of `Wasip2ImportRegistry` from the
    /// effects discovered by the per-fn walker. Stays `false` for
    /// every effect not yet lowered; those effects are rejected
    /// upstream by `wasip2::effect_check` before reaching the
    /// wasm-gc emitter.
    ///
    /// Set grows as each phase commit lands: 1.2b1 wired Console.*,
    /// 1.4 added Time.unixMs + Random.{int,float}, 1.3.2 added
    /// ArgsGet, 1.3.3 added EnvGet.
    pub(super) fn lowers_on_wasip2(self) -> bool {
        matches!(
            self,
            EffectName::ConsolePrint
                | EffectName::ConsoleError
                | EffectName::ConsoleWarn
                | EffectName::ConsoleReadLine
                | EffectName::TimeUnixMs
                | EffectName::TimeNow
                | EffectName::TimeSleep
                | EffectName::RandomInt
                | EffectName::RandomFloat
                | EffectName::ArgsGet
                | EffectName::EnvGet
                | EffectName::DiskExists
                | EffectName::DiskReadText
                | EffectName::DiskWriteText
                | EffectName::DiskAppendText
                | EffectName::DiskDelete
                | EffectName::DiskDeleteDir
                | EffectName::DiskMakeDir
                | EffectName::DiskListDir,
        )
    }
}

fn string_ref_ty(registry: &TypeRegistry) -> Result<ValType, WasmGcError> {
    let idx = registry
        .string_array_type_idx
        .ok_or(WasmGcError::Validation(
            "effect requires String repr but no string type slot was allocated".into(),
        ))?;
    Ok(ValType::Ref(wasm_encoder::RefType {
        nullable: true,
        heap_type: wasm_encoder::HeapType::Concrete(idx),
    }))
}