feat(sozo-bindgen): add i8,i16,i32,i64 support (#2934)

feat: add i8,i16,i32,i64 support
dojoengine · Jan 20, 2025 · af46cb0 · af46cb0
1 parent 2dd3ead
commit af46cb0
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/crates/dojo/bindgen/src/plugins/typescript/generator/constants.rs b/crates/dojo/bindgen/src/plugins/typescript/generator/constants.rs
@@ -8,6 +8,10 @@ pub const CAIRO_U64: &str = "u64";
 pub const CAIRO_U128: &str = "u128";
 pub const CAIRO_U256: &str = "u256";
 pub const CAIRO_U256_STRUCT: &str = "U256";
+pub const CAIRO_I8: &str = "i8";
+pub const CAIRO_I16: &str = "i16";
+pub const CAIRO_I32: &str = "i32";
+pub const CAIRO_I64: &str = "i64";
 pub const CAIRO_I128: &str = "i128";
 pub const CAIRO_BOOL: &str = "bool";
 pub const CAIRO_OPTION: &str = "Option";

diff --git a/crates/dojo/bindgen/src/plugins/typescript/generator/mod.rs b/crates/dojo/bindgen/src/plugins/typescript/generator/mod.rs
@@ -1,8 +1,9 @@
 use cainome::parser::tokens::{Composite, CompositeInner, CompositeType, Token};
 use constants::{
-    CAIRO_BOOL, CAIRO_BYTE_ARRAY, CAIRO_CONTRACT_ADDRESS, CAIRO_FELT252, CAIRO_I128, CAIRO_OPTION,
-    CAIRO_OPTION_DEFAULT_VALUE, CAIRO_U128, CAIRO_U16, CAIRO_U256, CAIRO_U256_STRUCT, CAIRO_U32,
-    CAIRO_U64, CAIRO_U8, CAIRO_UNIT_TYPE, JS_BIGNUMBERISH, JS_BOOLEAN, JS_STRING,
+    CAIRO_BOOL, CAIRO_BYTE_ARRAY, CAIRO_CONTRACT_ADDRESS, CAIRO_FELT252, CAIRO_I128, CAIRO_I16,
+    CAIRO_I32, CAIRO_I64, CAIRO_I8, CAIRO_OPTION, CAIRO_OPTION_DEFAULT_VALUE, CAIRO_U128,
+    CAIRO_U16, CAIRO_U256, CAIRO_U256_STRUCT, CAIRO_U32, CAIRO_U64, CAIRO_U8, CAIRO_UNIT_TYPE,
+    JS_BIGNUMBERISH, JS_BOOLEAN, JS_STRING,
 };
 use convert_case::{Case, Casing};
 
@@ -72,6 +73,10 @@ impl From<&str> for JsPrimitiveType {
             CAIRO_U128 => JsPrimitiveType(JS_BIGNUMBERISH.to_owned()),
             CAIRO_U256 => JsPrimitiveType(JS_BIGNUMBERISH.to_owned()),
             CAIRO_U256_STRUCT => JsPrimitiveType(JS_BIGNUMBERISH.to_owned()),
+            CAIRO_I8 => JsPrimitiveType(JS_BIGNUMBERISH.to_owned()),
+            CAIRO_I16 => JsPrimitiveType(JS_BIGNUMBERISH.to_owned()),
+            CAIRO_I32 => JsPrimitiveType(JS_BIGNUMBERISH.to_owned()),
+            CAIRO_I64 => JsPrimitiveType(JS_BIGNUMBERISH.to_owned()),
             CAIRO_I128 => JsPrimitiveType(JS_BIGNUMBERISH.to_owned()),
             CAIRO_BOOL => JsPrimitiveType(JS_BOOLEAN.to_owned()),
             CAIRO_UNIT_TYPE => JsPrimitiveType(JS_STRING.to_owned()),
@@ -150,6 +155,10 @@ impl From<&str> for JsPrimitiveInputType {
             CAIRO_U128 => JsPrimitiveInputType(JS_BIGNUMBERISH.to_owned()),
             CAIRO_U256 => JsPrimitiveInputType(JS_BIGNUMBERISH.to_owned()),
             CAIRO_U256_STRUCT => JsPrimitiveInputType(JS_BIGNUMBERISH.to_owned()),
+            CAIRO_I8 => JsPrimitiveInputType(JS_BIGNUMBERISH.to_owned()),
+            CAIRO_I16 => JsPrimitiveInputType(JS_BIGNUMBERISH.to_owned()),
+            CAIRO_I32 => JsPrimitiveInputType(JS_BIGNUMBERISH.to_owned()),
+            CAIRO_I64 => JsPrimitiveInputType(JS_BIGNUMBERISH.to_owned()),
             CAIRO_I128 => JsPrimitiveInputType(JS_BIGNUMBERISH.to_owned()),
             CAIRO_BOOL => JsPrimitiveInputType(JS_BOOLEAN.to_owned()),
             _ => JsPrimitiveInputType(value.to_owned()),
@@ -216,6 +225,10 @@ impl From<&str> for JsPrimitiveDefaultValue {
             CAIRO_U128 => JsPrimitiveDefaultValue("0".to_string()),
             CAIRO_U256 => JsPrimitiveDefaultValue("0".to_string()),
             CAIRO_U256_STRUCT => JsPrimitiveDefaultValue("0".to_string()),
+            CAIRO_I8 => JsPrimitiveDefaultValue("0".to_string()),
+            CAIRO_I16 => JsPrimitiveDefaultValue("0".to_string()),
+            CAIRO_I32 => JsPrimitiveDefaultValue("0".to_string()),
+            CAIRO_I64 => JsPrimitiveDefaultValue("0".to_string()),
             CAIRO_I128 => JsPrimitiveDefaultValue("0".to_string()),
             CAIRO_BOOL => JsPrimitiveDefaultValue("false".to_string()),
             CAIRO_OPTION => JsPrimitiveDefaultValue(CAIRO_OPTION_DEFAULT_VALUE.to_string()),