fix: escape input strings for token alias (#48)

cartridge-gg · Aug 9, 2024 · 3ca83b5 · 3ca83b5
1 parent 7b08841
commit 3ca83b5
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/crates/rs-macro/src/macro_inputs.rs b/crates/rs-macro/src/macro_inputs.rs
@@ -169,20 +169,20 @@ pub(crate) struct TypeAlias {
 
 impl Parse for TypeAlias {
     fn parse(input: ParseStream) -> Result<Self> {
-        let abi = input
-            .parse::<Type>()?
-            .into_token_stream()
-            .to_string()
-            .replace(' ', "");
+        let abi = sanitize_str(&input.parse::<Type>()?.into_token_stream().to_string());
 
         input.parse::<Token![as]>()?;
 
-        let alias = input.parse::<Ident>()?.to_string();
+        let alias = sanitize_str(&input.parse::<Ident>()?.to_string());
 
         Ok(TypeAlias { abi, alias })
     }
 }
 
+fn sanitize_str(abi: &str) -> String {
+    abi.trim().replace([' ', '\n', '\t'], "").to_string()
+}
+
 fn open_json_file(file_path: &str) -> Result<File> {
     File::open(file_path).map_err(|e| {
         syn::Error::new(