pass fixed size lists on to encoder

Author: cpetig

crates/wac-graph/src/encoding.rs

@@ -243,6 +243,7 @@ impl<'a> TypeEncoder<'a> {
         let index = match ty {
             DefinedType::Tuple(types) => self.tuple(state, types),
             DefinedType::List(ty) => self.list(state, *ty),
+            DefinedType::FixedSizeList(ty, elements) => self.fixed_size_list(state, *ty, *elements),
             DefinedType::Option(ty) => self.option(state, *ty),
             DefinedType::Result { ok, err } => self.result(state, *ok, *err),
             DefinedType::Variant(v) => self.variant(state, v),
@@ -552,6 +553,18 @@ impl<'a> TypeEncoder<'a> {
         index
     }
 
+    fn fixed_size_list(&self, state: &mut State, ty: ValueType, elements: u32) -> u32 {
+        let ty = self.value_type(state, ty);
+        let index = state.current.encodable.type_count();
+        state
+            .current
+            .encodable
+            .ty()
+            .defined_type()
+            .fixed_size_list(ty, elements);
+        index
+    }
+
     fn stream(&self, state: &mut State, ty: Option<ValueType>) -> u32 {
         let ty = ty.map(|ty| self.value_type(state, ty));
         let index = state.current.encodable.type_count();

crates/wac-types/src/aggregator.rs

@@ -789,6 +789,9 @@ impl TypeAggregator {
                     .collect::<Result<_>>()?,
             ),
             DefinedType::List(ty) => DefinedType::List(self.remap_value_type(types, *ty, checker)?),
+            DefinedType::FixedSizeList(ty, elements) => {
+                DefinedType::FixedSizeList(self.remap_value_type(types, *ty, checker)?, *elements)
+            }
             DefinedType::Option(ty) => {
                 DefinedType::Option(self.remap_value_type(types, *ty, checker)?)
             }

crates/wac-types/src/checker.rs

@@ -551,6 +551,13 @@ impl<'a> SubtypeChecker<'a> {
             (DefinedType::List(a), DefinedType::List(b)) => self
                 .value_type(*a, at, *b, bt)
                 .context("mismatched type for list element"),
+            (DefinedType::FixedSizeList(a, asize), DefinedType::FixedSizeList(b, bsize)) => {
+                if asize != bsize {
+                    bail!("mismatched size for fixed size list element");
+                }
+                self.value_type(*a, at, *b, bt)
+                    .context("mismatched type for fixed size list element")
+            }
             (DefinedType::Future(a), DefinedType::Future(b)) => self
                 .payload(*a, at, *b, bt)
                 .context("mismatched type for future payload"),
@@ -583,6 +590,7 @@ impl<'a> SubtypeChecker<'a> {
 
             (DefinedType::Tuple(_), _)
             | (DefinedType::List(_), _)
+            | (DefinedType::FixedSizeList(_, _), _)
             | (DefinedType::Option(_), _)
             | (DefinedType::Result { .. }, _)
             | (DefinedType::Variant(_), _)

crates/wac-types/src/component.rs

@@ -659,6 +659,8 @@ pub enum DefinedType {
     Tuple(Vec<ValueType>),
     /// A list type.
     List(ValueType),
+    /// A fixed size array
+    FixedSizeList(ValueType, u32),
     /// An option type.
     Option(ValueType),
     /// A result type.
@@ -689,7 +691,7 @@ impl DefinedType {
     pub fn contains_borrow(&self, types: &Types) -> bool {
         match self {
             Self::Tuple(tys) => tys.iter().any(|ty| ty.contains_borrow(types)),
-            Self::List(ty) => ty.contains_borrow(types),
+            Self::List(ty) | Self::FixedSizeList(ty, _) => ty.contains_borrow(types),
             Self::Option(ty) => ty.contains_borrow(types),
             Self::Result { ok, err } => {
                 ok.map(|ty| ty.contains_borrow(types)).unwrap_or(false)
@@ -721,7 +723,7 @@ impl DefinedType {
 
                 Ok(())
             }
-            DefinedType::List(ty) | DefinedType::Option(ty) => {
+            DefinedType::List(ty) | DefinedType::Option(ty) | DefinedType::FixedSizeList(ty, _) => {
                 ty._visit_defined_types(types, visitor, false)
             }
             DefinedType::Result { ok, err } => {
@@ -765,6 +767,7 @@ impl DefinedType {
         match self {
             Self::Tuple(_) => "tuple",
             Self::List(_) => "list",
+            Self::FixedSizeList(_, _) => "list<,N>",
             Self::Option(_) => "option",
             Self::Result { .. } => "result",
             Self::Variant(_) => "variant",