Use injection to re-parse rust macro calls and defs as rust

atom · Jun 17, 2019 · 8988f87 · 8988f87
1 parent 6708932
commit 8988f87
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/packages/language-rust-bundled/grammars/tree-sitter-rust.cson b/packages/language-rust-bundled/grammars/tree-sitter-rust.cson
@@ -2,6 +2,7 @@ name: 'Rust'
 scopeName: 'source.rust'
 type: 'tree-sitter'
 parser: 'tree-sitter-rust'
+injectionRegex: 'rust'
 
 fileTypes: [
   'rs'

diff --git a/packages/language-rust-bundled/lib/main.js b/packages/language-rust-bundled/lib/main.js
@@ -0,0 +1,9 @@
+exports.activate = function() {
+  for (const nodeType of ['macro_invocation', 'macro_rule']) {
+    atom.grammars.addInjectionPoint('source.rust', {
+      type: nodeType,
+      language() { return 'rust'; },
+      content(node) { return node.lastChild; },
+    });
+  }
+};
diff --git a/packages/language-rust-bundled/package.json b/packages/language-rust-bundled/package.json
@@ -7,6 +7,7 @@
     "grammar",
     "rust"
   ],
+  "main": "lib/main.js",
   "repository": "https://github.com/atom/atom",
   "license": "MIT",
   "dependencies": {