From c09c6fc97c1d553dd348383eb98fc7a4788030cb Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Wed, 30 Jan 2019 23:58:52 +0300
Subject: start tt convertions boilerplate

---
 crates/ra_hir/src/macros.rs | 38 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 36 insertions(+), 2 deletions(-)

(limited to 'crates/ra_hir/src/macros.rs')

diff --git a/crates/ra_hir/src/macros.rs b/crates/ra_hir/src/macros.rs
index 647df4260..9ba381685 100644
--- a/crates/ra_hir/src/macros.rs
+++ b/crates/ra_hir/src/macros.rs
@@ -16,6 +16,7 @@ use std::sync::Arc;
 
 use ra_syntax::{
     TextRange, TextUnit, SourceFile, AstNode, SyntaxNode, TreeArc, SyntaxNodePtr,
+    SyntaxKind::*,
     ast::{self, NameOwner},
 };
 
@@ -201,6 +202,39 @@ pub(crate) fn expand_macro_invocation(
     def.expand(input).map(Arc::new)
 }
 
-fn macro_call_to_tt(call: &ast::MacroCall) -> Option<tt::TokenTree> {
-    None
+fn macro_call_to_tt(call: &ast::MacroCall) -> Option<tt::Subtree> {
+    let tt = call.token_tree()?;
+    convert_tt(tt.syntax())
+}
+
+fn convert_tt(tt: &SyntaxNode) -> Option<tt::Subtree> {
+    let first_child = tt.first_child()?;
+    let last_child = tt.last_child()?;
+    let delimiter = match (first_child.kind(), last_child.kind()) {
+        (L_PAREN, R_PAREN) => tt::Delimiter::Parenthesis,
+        (L_CURLY, R_CURLY) => tt::Delimiter::Brace,
+        (L_BRACK, R_BRACK) => tt::Delimiter::Bracket,
+        _ => return None,
+    };
+    let mut token_trees = Vec::new();
+    for child in tt.children().skip(1) {
+        if child == first_child || child == last_child || child.kind().is_trivia() {
+            continue;
+        }
+        let child = if child.kind() == TOKEN_TREE {
+            convert_tt(child)?.into()
+        } else if child.kind().is_keyword() {
+            let text = child.leaf_text().unwrap().clone();
+            tt::Leaf::from(tt::Ident { text }).into()
+        } else {
+            return None;
+        };
+        token_trees.push(child)
+    }
+
+    let res = tt::Subtree {
+        delimiter,
+        token_trees,
+    };
+    Some(res)
 }
-- 
cgit v1.2.3