zed/crates/editor2/src/highlight_matching_bracket.rs

use gpui::ViewContext;

use crate::{Editor, RangeToAnchorExt};

enum MatchingBracketHighlight {}

pub fn refresh_matching_bracket_highlights(editor: &mut Editor, cx: &mut ViewContext<Editor>) {
    // editor.clear_background_highlights::<MatchingBracketHighlight>(cx);

    let newest_selection = editor.selections.newest::<usize>(cx);
    // Don't highlight brackets if the selection isn't empty
    if !newest_selection.is_empty() {
        return;
    }

    let head = newest_selection.head();
    let snapshot = editor.snapshot(cx);
    if let Some((opening_range, closing_range)) = snapshot
        .buffer_snapshot
        .innermost_enclosing_bracket_ranges(head..head)
    {
        editor.highlight_background::<MatchingBracketHighlight>(
            vec![
                opening_range.to_anchors(&snapshot.buffer_snapshot),
                closing_range.to_anchors(&snapshot.buffer_snapshot),
            ],
            |theme| theme.editor_document_highlight_read_background,
            cx,
        )
    }
}

// #[cfg(test)]
// mod tests {
//     use super::*;
//     use crate::{editor_tests::init_test, test::editor_lsp_test_context::EditorLspTestContext};
//     use indoc::indoc;
//     use language::{BracketPair, BracketPairConfig, Language, LanguageConfig};

//     #[gpui::test]
//     async fn test_matching_bracket_highlights(cx: &mut gpui::TestAppContext) {
//         init_test(cx, |_| {});

//         let mut cx = EditorLspTestContext::new(
//             Language::new(
//                 LanguageConfig {
//                     name: "Rust".into(),
//                     path_suffixes: vec!["rs".to_string()],
//                     brackets: BracketPairConfig {
//                         pairs: vec![
//                             BracketPair {
//                                 start: "{".to_string(),
//                                 end: "}".to_string(),
//                                 close: false,
//                                 newline: true,
//                             },
//                             BracketPair {
//                                 start: "(".to_string(),
//                                 end: ")".to_string(),
//                                 close: false,
//                                 newline: true,
//                             },
//                         ],
//                         ..Default::default()
//                     },
//                     ..Default::default()
//                 },
//                 Some(tree_sitter_rust::language()),
//             )
//             .with_brackets_query(indoc! {r#"
//                 ("{" @open "}" @close)
//                 ("(" @open ")" @close)
//                 "#})
//             .unwrap(),
//             Default::default(),
//             cx,
//         )
//         .await;

//         // positioning cursor inside bracket highlights both
//         cx.set_state(indoc! {r#"
//             pub fn test("Test ˇargument") {
//                 another_test(1, 2, 3);
//             }
//         "#});
//         cx.assert_editor_background_highlights::<MatchingBracketHighlight>(indoc! {r#"
//             pub fn test«(»"Test argument"«)» {
//                 another_test(1, 2, 3);
//             }
//         "#});

//         cx.set_state(indoc! {r#"
//             pub fn test("Test argument") {
//                 another_test(1, ˇ2, 3);
//             }
//         "#});
//         cx.assert_editor_background_highlights::<MatchingBracketHighlight>(indoc! {r#"
//             pub fn test("Test argument") {
//                 another_test«(»1, 2, 3«)»;
//             }
//         "#});

//         cx.set_state(indoc! {r#"
//             pub fn test("Test argument") {
//                 anotherˇ_test(1, 2, 3);
//             }
//         "#});
//         cx.assert_editor_background_highlights::<MatchingBracketHighlight>(indoc! {r#"
//             pub fn test("Test argument") «{»
//                 another_test(1, 2, 3);
//             «}»
//         "#});

//         // positioning outside of brackets removes highlight
//         cx.set_state(indoc! {r#"
//             pub fˇn test("Test argument") {
//                 another_test(1, 2, 3);
//             }
//         "#});
//         cx.assert_editor_background_highlights::<MatchingBracketHighlight>(indoc! {r#"
//             pub fn test("Test argument") {
//                 another_test(1, 2, 3);
//             }
//         "#});

//         // non empty selection dismisses highlight
//         cx.set_state(indoc! {r#"
//             pub fn test("Te«st argˇ»ument") {
//                 another_test(1, 2, 3);
//             }
//         "#});
//         cx.assert_editor_background_highlights::<MatchingBracketHighlight>(indoc! {r#"
//             pub fn test("Test argument") {
//                 another_test(1, 2, 3);
//             }
//         "#});
//     }
// }