fix(torii-graphql): option enums in arrays (#3068)

dojoengine · Feb 26, 2025 · fdf6ff0 · fdf6ff0
1 parent 1dfd801
commit fdf6ff0
Showing 1 changed file with 17 additions and 15 deletions.
diff --git a/crates/torii/graphql/src/query/mod.rs b/crates/torii/graphql/src/query/mod.rs
@@ -189,21 +189,23 @@ pub fn value_mapping_from_row(
                         ) {
                             match value {
                                 Value::Object(obj) => {
-                                    for (field_name, field_value) in obj.iter_mut() {
-                                        populate_value(
-                                            field_value,
-                                            &type_data.type_mapping().unwrap()[field_name],
-                                            entity_id,
-                                        );
-                                    }
-
-                                    if type_data.type_mapping().map_or(false, |mapping| {
-                                        mapping.contains_key(&Name::new("option"))
-                                    }) {
-                                        obj.insert(
-                                            Name::new("option"),
-                                            Value::String(obj.keys().next().unwrap().to_string()),
-                                        );
+                                    for (field_name, field_type) in
+                                        type_data.type_mapping().unwrap().iter()
+                                    {
+                                        if let Some(field_value) = obj.get_mut(field_name) {
+                                            populate_value(field_value, field_type, entity_id);
+                                        } else {
+                                            obj.insert(
+                                                field_name.clone(),
+                                                if field_name == "option" {
+                                                    Value::String(
+                                                        obj.keys().next().unwrap().to_string(),
+                                                    )
+                                                } else {
+                                                    Value::Null
+                                                },
+                                            );
+                                        }
                                     }
 
                                     // insert $entity_id$ relation