use syntax::{algo::non_trivia_sibling, Direction, SyntaxKind, T};
use crate::{AssistContext, AssistId, AssistKind, Assists};
pub(crate) fn flip_comma(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
let comma = ctx.find_token_syntax_at_offset(T![,])?;
let prev = non_trivia_sibling(comma.clone().into(), Direction::Prev)?;
let next = non_trivia_sibling(comma.clone().into(), Direction::Next)?;
if next.kind().is_punct() {
return None;
}
if comma.ancestors().any(|it| it.kind() == SyntaxKind::MACRO_CALL) {
return None;
}
acc.add(
AssistId("flip_comma", AssistKind::RefactorRewrite),
"Flip comma",
comma.text_range(),
|edit| {
edit.replace(prev.text_range(), next.to_string());
edit.replace(next.text_range(), prev.to_string());
},
)
}
#[cfg(test)]
mod tests {
use super::*;
use crate::tests::{check_assist, check_assist_not_applicable, check_assist_target};
#[test]
fn flip_comma_works_for_function_parameters() {
check_assist(
flip_comma,
r#"fn foo(x: i32,$0 y: Result<(), ()>) {}"#,
r#"fn foo(y: Result<(), ()>, x: i32) {}"#,
)
}
#[test]
fn flip_comma_target() {
check_assist_target(flip_comma, r#"fn foo(x: i32,$0 y: Result<(), ()>) {}"#, ",")
}
#[test]
fn flip_comma_before_punct() {
check_assist_not_applicable(flip_comma, "pub enum Test { A,$0 }");
check_assist_not_applicable(flip_comma, "pub struct Test { foo: usize,$0 }");
}
#[test]
fn flip_comma_works() {
check_assist(
flip_comma,
r#"fn main() {((1, 2),$0 (3, 4));}"#,
r#"fn main() {((3, 4), (1, 2));}"#,
)
}
#[test]
fn flip_comma_not_applicable_for_macro_input() {
check_assist_not_applicable(flip_comma, r#"bar!(a,$0 b)"#);
}
}