Properly serde enum

dojoengine · Oct 2, 2023 · c787896 · c787896
1 parent 1aef1a9
commit c787896
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/crates/dojo-types/src/schema.rs b/crates/dojo-types/src/schema.rs
@@ -68,6 +68,12 @@ impl Ty {
                     }
                 }
                 Ty::Enum(e) => {
+                    let option = e
+                        .option
+                        .map(|v| Ok(vec![FieldElement::from(v)]))
+                        .unwrap_or(Err(PrimitiveError::MissingFieldElement))?;
+                    felts.extend(option);
+
                     for (_, child) in &e.options {
                         serialize_inner(child, felts)?;
                     }
@@ -97,6 +103,8 @@ impl Ty {
                 }
             }
             Ty::Enum(e) => {
+                e.option =
+                    Some(felts.remove(0).try_into().map_err(PrimitiveError::ValueOutOfRange)?);
                 for (_, child) in &mut e.options {
                     child.deserialize(felts)?;
                 }