TS imports, special handling for @angular-like refs

Author: rimutaka

stackmuncher_lib/src/kwc.rs

@@ -94,8 +94,8 @@ impl KeywordCounter {
 
         // loop through the characters to find the first boundary
         for (i, c) in kwc.k.as_bytes().iter().enumerate() {
-            // keep iterating until the first separator (not -._"')
-            if c.is_ascii_alphanumeric() || *c == 45u8 || *c == 46u8 || *c == 95u8 {
+            // keep iterating until the first separator (not -._"'@)
+            if c.is_ascii_alphanumeric() || *c == 45u8 || *c == 46u8 || *c == 95u8 || *c == 64u8 {
                 continue;
             }
 

stackmuncher_lib/src/tech.rs

@@ -291,6 +291,23 @@ impl Tech {
             tech.refs.remove(&local_import);
         }
 
+        // some TypeScript refs start with @, e.g. @angular/core
+        // it's a valid name, but @ will get in the way of users searching for "angular"
+        tech.refs = tech
+            .refs
+            .into_iter()
+            .map(|kwc| {
+                if kwc.k.starts_with("@") {
+                    KeywordCounter {
+                        k: kwc.k.trim_start_matches("@").to_string(),
+                        ..kwc
+                    }
+                } else {
+                    kwc
+                }
+            })
+            .collect::<HashSet<KeywordCounter>>();
+
         tech
     }
 }

stackmuncher_lib/stm_rules/munchers/ts.json

@@ -16,5 +16,8 @@
   ],
   "block_comments_end": [
     "(?i)\\*/\\s*$"
+  ],
+  "refs": [
+    "^[[:blank:]]*import[[:blank:]]+.*[[:blank:]]*['\"]([@A-Za-z0-9_][@A-Za-z0-9_\\-\\.]+).*['\"][[:blank:]]*;"
   ]
 }