Output enum value for sozo model get (#961)

dojoengine · Oct 3, 2023 · da8b1d9 · da8b1d9
1 parent ea19c4b
commit da8b1d9
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/crates/dojo-types/src/schema.rs b/crates/dojo-types/src/schema.rs
@@ -210,7 +210,7 @@ pub struct Enum {
 }
 
 impl Enum {
-    pub fn to_sql_value(&self) -> Result<String, EnumError> {
+    pub fn option(&self) -> Result<String, EnumError> {
         let option: usize = if let Some(option) = self.option {
             option as usize
         } else {
@@ -221,7 +221,11 @@ impl Enum {
             return Err(EnumError::OptionInvalid);
         }
 
-        Ok(format!("'{}'", self.options[option].0))
+        Ok(self.options[option].0.clone())
+    }
+
+    pub fn to_sql_value(&self) -> Result<String, EnumError> {
+        Ok(format!("'{}'", self.option()?))
     }
 }
 
@@ -290,6 +294,11 @@ fn format_member(m: &Member) -> String {
                 }
             }
         }
+    } else if let Ty::Enum(e) = &m.ty {
+        match e.option() {
+            Ok(option) => str.push_str(&format!(" = {option}")),
+            Err(_) => str.push_str(" = Invalid Option"),
+        }
     }
 
     str