6 files changed, 255 insertions, 15 deletions

diff --git a/crates/ra_assists/src/flip_comma.rs b/crates/ra_assists/src/flip_comma.rs
index 34489329c..5ee7561bc 100644
--- a/crates/ra_assists/src/flip_comma.rs
+++ b/crates/ra_assists/src/flip_comma.rs
@@ -7,6 +7,13 @@ pub(crate) fn flip_comma(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist>
     let comma = ctx.token_at_offset().find(|leaf| leaf.kind() == T![,])?;
     let prev = non_trivia_sibling(comma.clone().into(), Direction::Prev)?;
     let next = non_trivia_sibling(comma.clone().into(), Direction::Next)?;
+
+    // Don't apply a "flip" in case of a last comma
+    // that typically comes before punctuation
+    if next.kind().is_punct() {
+        return None;
+    }
+
     ctx.add_action(AssistId("flip_comma"), "flip comma", |edit| {
         edit.target(comma.text_range());
         edit.replace(prev.text_range(), next.to_string());
@@ -35,4 +42,27 @@ mod tests {
     fn flip_comma_target() {
         check_assist_target(flip_comma, "fn foo(x: i32,<|> y: Result<(), ()>) {}", ",")
     }
+
+    #[test]
+    #[should_panic]
+    fn flip_comma_before_punct() {
+        // See https://github.com/rust-analyzer/rust-analyzer/issues/1619
+        // "Flip comma" assist shouldn't be applicable to the last comma in enum or struct
+        // declaration body.
+        check_assist_target(
+            flip_comma,
+            "pub enum Test { \
+             A,<|> \
+             }",
+            ",",
+        );
+
+        check_assist_target(
+            flip_comma,
+            "pub struct Test { \
+             foo: usize,<|> \
+             }",
+            ",",
+        );
+    }
 }
diff --git a/crates/ra_assists/src/lib.rs b/crates/ra_assists/src/lib.rs
index 0d848629d..03eec73ad 100644
--- a/crates/ra_assists/src/lib.rs
+++ b/crates/ra_assists/src/lib.rs
@@ -93,6 +93,7 @@ mod flip_comma;
 mod flip_binexpr;
 mod change_visibility;
 mod fill_match_arms;
+mod merge_match_arms;
 mod introduce_variable;
 mod inline_local_variable;
 mod replace_if_let_with_match;
@@ -109,6 +110,7 @@ fn all_assists<DB: HirDatabase>() -> &'static [fn(AssistCtx<DB>) -> Option<Assis
         add_impl::add_impl,
         change_visibility::change_visibility,
         fill_match_arms::fill_match_arms,
+        merge_match_arms::merge_match_arms,
         flip_comma::flip_comma,
         flip_binexpr::flip_binexpr,
         introduce_variable::introduce_variable,
diff --git a/crates/ra_assists/src/merge_match_arms.rs b/crates/ra_assists/src/merge_match_arms.rs
new file mode 100644
index 000000000..bc5f6f17c
--- /dev/null
+++ b/crates/ra_assists/src/merge_match_arms.rs
@@ -0,0 +1,188 @@
+use crate::{Assist, AssistCtx, AssistId, TextRange, TextUnit};
+use hir::db::HirDatabase;
+use ra_syntax::ast::{AstNode, MatchArm};
+
+pub(crate) fn merge_match_arms(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+    let current_arm = ctx.node_at_offset::<MatchArm>()?;
+
+    // We check if the following match arm matches this one. We could, but don't,
+    // compare to the previous match arm as well.
+    let next = current_arm.syntax().next_sibling();
+    let next_arm = MatchArm::cast(next?.clone())?;
+
+    // Don't try to handle arms with guards for now - can add support for this later
+    if current_arm.guard().is_some() || next_arm.guard().is_some() {
+        return None;
+    }
+
+    let current_expr = current_arm.expr()?;
+    let next_expr = next_arm.expr()?;
+
+    // Check for match arm equality by comparing lengths and then string contents
+    if current_expr.syntax().text_range().len() != next_expr.syntax().text_range().len() {
+        return None;
+    }
+    if current_expr.syntax().text() != next_expr.syntax().text() {
+        return None;
+    }
+
+    let cursor_to_end = current_arm.syntax().text_range().end() - ctx.frange.range.start();
+
+    ctx.add_action(AssistId("merge_match_arms"), "merge match arms", |edit| {
+        fn contains_placeholder(a: &MatchArm) -> bool {
+            a.pats().any(|x| match x.kind() {
+                ra_syntax::ast::PatKind::PlaceholderPat(..) => true,
+                _ => false,
+            })
+        }
+
+        let pats = if contains_placeholder(&current_arm) || contains_placeholder(&next_arm) {
+            "_".into()
+        } else {
+            let ps: Vec<String> = current_arm
+                .pats()
+                .map(|x| x.syntax().to_string())
+                .chain(next_arm.pats().map(|x| x.syntax().to_string()))
+                .collect();
+            ps.join(" | ")
+        };
+
+        let arm = format!("{} => {}", pats, current_expr.syntax().text());
+        let offset = TextUnit::from_usize(arm.len()) - cursor_to_end;
+
+        let start = current_arm.syntax().text_range().start();
+        let end = next_arm.syntax().text_range().end();
+
+        edit.target(current_arm.syntax().text_range());
+        edit.replace(TextRange::from_to(start, end), arm);
+        edit.set_cursor(start + offset);
+    });
+
+    ctx.build()
+}
+
+#[cfg(test)]
+mod tests {
+    use super::merge_match_arms;
+    use crate::helpers::{check_assist, check_assist_not_applicable};
+
+    #[test]
+    fn merge_match_arms_single_patterns() {
+        check_assist(
+            merge_match_arms,
+            r#"
+            #[derive(Debug)]
+            enum X { A, B, C }
+
+            fn main() {
+                let x = X::A;
+                let y = match x {
+                    X::A => { 1i32<|> }
+                    X::B => { 1i32 }
+                    X::C => { 2i32 }
+                }
+            }
+            "#,
+            r#"
+            #[derive(Debug)]
+            enum X { A, B, C }
+
+            fn main() {
+                let x = X::A;
+                let y = match x {
+                    X::A | X::B => { 1i32<|> }
+                    X::C => { 2i32 }
+                }
+            }
+            "#,
+        );
+    }
+
+    #[test]
+    fn merge_match_arms_multiple_patterns() {
+        check_assist(
+            merge_match_arms,
+            r#"
+            #[derive(Debug)]
+            enum X { A, B, C, D, E }
+
+            fn main() {
+                let x = X::A;
+                let y = match x {
+                    X::A | X::B => {<|> 1i32 },
+                    X::C | X::D => { 1i32 },
+                    X::E => { 2i32 },
+                }
+            }
+            "#,
+            r#"
+            #[derive(Debug)]
+            enum X { A, B, C, D, E }
+
+            fn main() {
+                let x = X::A;
+                let y = match x {
+                    X::A | X::B | X::C | X::D => {<|> 1i32 },
+                    X::E => { 2i32 },
+                }
+            }
+            "#,
+        );
+    }
+
+    #[test]
+    fn merge_match_arms_placeholder_pattern() {
+        check_assist(
+            merge_match_arms,
+            r#"
+            #[derive(Debug)]
+            enum X { A, B, C, D, E }
+
+            fn main() {
+                let x = X::A;
+                let y = match x {
+                    X::A => { 1i32 },
+                    X::B => { 2i<|>32 },
+                    _ => { 2i32 }
+                }
+            }
+            "#,
+            r#"
+            #[derive(Debug)]
+            enum X { A, B, C, D, E }
+
+            fn main() {
+                let x = X::A;
+                let y = match x {
+                    X::A => { 1i32 },
+                    _ => { 2i<|>32 }
+                }
+            }
+            "#,
+        );
+    }
+
+    #[test]
+    fn merge_match_arms_rejects_guards() {
+        check_assist_not_applicable(
+            merge_match_arms,
+            r#"
+            #[derive(Debug)]
+            enum X {
+                A(i32),
+                B,
+                C
+            }
+
+            fn main() {
+                let x = X::A;
+                let y = match x {
+                    X::A(a) if a > 5 => { <|>1i32 },
+                    X::B => { 1i32 },
+                    X::C => { 2i32 }
+                }
+            }
+            "#,
+        );
+    }
+}
diff --git a/crates/ra_hir/src/ty/tests.rs b/crates/ra_hir/src/ty/tests.rs
index 676711d0a..36dea17a3 100644
--- a/crates/ra_hir/src/ty/tests.rs
+++ b/crates/ra_hir/src/ty/tests.rs
@@ -334,6 +334,8 @@ fn infer_literals() {
         infer(r##"
 fn test() {
     5i32;
+    5f32;
+    5f64;
     "hello";
     b"bytes";
     'c';
@@ -351,18 +353,20 @@ fn test() {
 }
 "##),
         @r###"
-[11; 201) '{     ...o"#; }': ()
+[11; 221) '{     ...o"#; }': ()
 [17; 21) '5i32': i32
-[27; 34) '"hello"': &str
-[40; 48) 'b"bytes"': &[u8]
-[54; 57) ''c'': char
-[63; 67) 'b'b'': u8
-[73; 77) '3.14': f64
-[83; 87) '5000': i32
-[93; 98) 'false': bool
-[104; 108) 'true': bool
-[114; 182) 'r#"   ...    "#': &str
-[188; 198) 'br#"yolo"#': &[u8]"###
+[27; 31) '5f32': f32
+[37; 41) '5f64': f64
+[47; 54) '"hello"': &str
+[60; 68) 'b"bytes"': &[u8]
+[74; 77) ''c'': char
+[83; 87) 'b'b'': u8
+[93; 97) '3.14': f64
+[103; 107) '5000': i32
+[113; 118) 'false': bool
+[124; 128) 'true': bool
+[134; 202) 'r#"   ...    "#': &str
+[208; 218) 'br#"yolo"#': &[u8]"###
     );
 }
 
diff --git a/crates/ra_syntax/Cargo.toml b/crates/ra_syntax/Cargo.toml
index 40d63ef7a..bc1c88070 100644
--- a/crates/ra_syntax/Cargo.toml
+++ b/crates/ra_syntax/Cargo.toml
@@ -10,7 +10,7 @@ repository = "https://github.com/rust-analyzer/rust-analyzer"
 [dependencies]
 unicode-xid = "0.1.0"
 itertools = "0.8.0"
-rowan = "0.6.0"
+rowan = "0.6.1"
 ra_rustc_lexer = { version = "0.1.0-pre.2" }
 
 # ideally, `serde` should be enabled by `ra_lsp_server`, but we enable it here
diff --git a/crates/ra_syntax/src/ast/expr_extensions.rs b/crates/ra_syntax/src/ast/expr_extensions.rs
index f9190d877..8284f1b25 100644
--- a/crates/ra_syntax/src/ast/expr_extensions.rs
+++ b/crates/ra_syntax/src/ast/expr_extensions.rs
@@ -239,16 +239,32 @@ impl ast::Literal {
     pub fn kind(&self) -> LiteralKind {
         match self.token().kind() {
             INT_NUMBER => {
-                let allowed_suffix_list = [
+                let int_suffix_list = [
                     "isize", "i128", "i64", "i32", "i16", "i8", "usize", "u128", "u64", "u32",
                     "u16", "u8",
                 ];
+
+                // The lexer treats e.g. `1f64` as an integer literal. See
+                // https://github.com/rust-analyzer/rust-analyzer/issues/1592
+                // and the comments on the linked PR.
+                let float_suffix_list = ["f32", "f64"];
+
                 let text = self.token().text().to_string();
-                let suffix = allowed_suffix_list
+
+                let float_suffix = float_suffix_list
                     .iter()
                     .find(|&s| text.ends_with(s))
                     .map(|&suf| SmolStr::new(suf));
-                LiteralKind::IntNumber { suffix }
+
+                if float_suffix.is_some() {
+                    LiteralKind::FloatNumber { suffix: float_suffix }
+                } else {
+                    let suffix = int_suffix_list
+                        .iter()
+                        .find(|&s| text.ends_with(s))
+                        .map(|&suf| SmolStr::new(suf));
+                    LiteralKind::IntNumber { suffix }
+                }
             }
             FLOAT_NUMBER => {
                 let allowed_suffix_list = ["f64", "f32"];