1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use crate::{
    helpers::{IntoWasm, WasmName},
    ParameterType, WasmSymbol, WasmType,
};
use wast::{
    core::{Import, ItemKind, ItemSig, TypeUse},
    token::Span,
};

mod codegen;

/// `@ffi("module", "field")`
pub struct ExternalType {
    pub module: WasmSymbol,
    pub field: WasmSymbol,
    pub alias: Option<WasmSymbol>,
    pub input: Vec<ParameterType>,
    pub output: Vec<WasmType>,
}

impl ExternalType {
    pub fn new(module: &str, field: &str) -> ExternalType {
        Self { module: WasmSymbol::new(module), field: WasmSymbol::new(field), alias: None, input: vec![], output: vec![] }
    }
    pub fn name(&self) -> &str {
        match &self.alias {
            None => self.field.as_ref(),
            Some(s) => s.as_ref(),
        }
    }
    pub fn with_alias<S: Into<WasmSymbol>>(self, alias: S) -> Self {
        Self { alias: Some(alias.into()), ..self }
    }

    pub fn with_input<I>(mut self, inputs: I) -> Self
    where
        I: IntoIterator<Item = ParameterType>,
    {
        self.input.extend(inputs);
        self
    }
    pub fn with_output<I>(mut self, outputs: I) -> Self
    where
        I: IntoIterator<Item = WasmType>,
    {
        self.output.extend(outputs);
        self
    }
}