Further changes

trustwallet · Jan 23, 2025 · 163f578 · 163f578
1 parent 72f3cbb
commit 163f578
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/codegen-v2/src/codegen/cpp/code_gen.rs b/codegen-v2/src/codegen/cpp/code_gen.rs
@@ -32,8 +32,12 @@ pub struct TWArg {
 
 fn convert_rust_type_to_cpp(ty: &str) -> String {
     match ty.trim() {
-        s if s.starts_with("* const") => format!("{} *_Nonnull", &s[8..]),
-        s if s.starts_with("* mut") => format!("{} *_Nullable", &s[6..]),
+        s if s.starts_with("* const ") => {
+            format!("{} *_Nonnull", s.strip_prefix("* const ").expect("Checked"))
+        }
+        s if s.starts_with("* mut ") => {
+            format!("{} *_Nullable", s.strip_prefix("* mut ").expect("Checked"))
+        }
         "bool" => "bool".to_string(),
         _ => ty.to_string(),
     }