Attempt to implement ranking of rules when none matches perfectly (wip)

2 files changed, 13 insertions, 10 deletions

diff --git a/crates/ra_mbe/src/mbe_expander.rs b/crates/ra_mbe/src/mbe_expander.rs
index da3952428..5083d5410 100644
--- a/crates/ra_mbe/src/mbe_expander.rs
+++ b/crates/ra_mbe/src/mbe_expander.rs
@@ -14,24 +14,27 @@ pub(crate) fn expand(
     rules: &crate::MacroRules,
     input: &tt::Subtree,
 ) -> ExpandResult<tt::Subtree> {
-    let (mut result, mut err) = (tt::Subtree::default(), Some(ExpandError::NoMatchingRule));
+    let (mut result, mut left_over, mut err) = (tt::Subtree::default(), usize::max_value(), Some(ExpandError::NoMatchingRule));
     for rule in &rules.rules {
-        let (res, e) = expand_rule(rule, input);
+        let ((res, left), e) = expand_rule(rule, input);
         if e.is_none() {
             // if we find a rule that applies without errors, we're done
             return (res, None);
         }
-        // TODO decide which result is better
-        result = res;
-        err = e;
+        // use the rule if we matched more tokens
+        if left < left_over {
+            result = res;
+            err = e;
+            left_over = left;
+        }
     }
     (result, err)
 }
 
-fn expand_rule(rule: &crate::Rule, input: &tt::Subtree) -> ExpandResult<tt::Subtree> {
-    let (bindings, bindings_err) = dbg!(matcher::match_(&rule.lhs, input));
+fn expand_rule(rule: &crate::Rule, input: &tt::Subtree) -> ExpandResult<(tt::Subtree, usize)> {
+    let ((bindings, left_over), bindings_err) = dbg!(matcher::match_(&rule.lhs, input));
     let (res, transcribe_err) = dbg!(transcriber::transcribe(&rule.rhs, &bindings));
-    (res, bindings_err.or(transcribe_err))
+    ((res, left_over), bindings_err.or(transcribe_err))
 }
 
 /// The actual algorithm for expansion is not too hard, but is pretty tricky.
diff --git a/crates/ra_mbe/src/mbe_expander/matcher.rs b/crates/ra_mbe/src/mbe_expander/matcher.rs
index f9d4952c6..52f1ac252 100644
--- a/crates/ra_mbe/src/mbe_expander/matcher.rs
+++ b/crates/ra_mbe/src/mbe_expander/matcher.rs
@@ -65,7 +65,7 @@ macro_rules! bail {
     };
 }
 
-pub(super) fn match_(pattern: &tt::Subtree, src: &tt::Subtree) -> ExpandResult<Bindings> {
+pub(super) fn match_(pattern: &tt::Subtree, src: &tt::Subtree) -> ExpandResult<(Bindings, usize)> {
     assert!(pattern.delimiter == None);
 
     let mut res = Bindings::default();
@@ -77,7 +77,7 @@ pub(super) fn match_(pattern: &tt::Subtree, src: &tt::Subtree) -> ExpandResult<B
         err = Some(err!("leftover tokens"));
     }
 
-    (res, err)
+    ((res, src.len()), err)
 }
 
 fn match_subtree(