tenant.openagents/omega

1use collections::HashSet;
2use std::{ops::Range, sync::LazyLock};
3use tree_sitter::{Query, QueryMatch};
4
5use crate::MigrationPatterns;
6use crate::patterns::KEYMAP_ACTION_ARRAY_ARGUMENT_AS_OBJECT_PATTERN;
7
8pub const KEYMAP_PATTERNS: MigrationPatterns = &[(
9    KEYMAP_ACTION_ARRAY_ARGUMENT_AS_OBJECT_PATTERN,
10    replace_array_with_single_string,
11)];
12
13fn replace_array_with_single_string(
14    contents: &str,
15    mat: &QueryMatch,
16    query: &Query,
17) -> Option<(Range<usize>, String)> {
18    let array_ix = query.capture_index_for_name("array")?;
19    let action_name_ix = query.capture_index_for_name("action_name")?;
20
21    let action_name = contents.get(
22        mat.nodes_for_capture_index(action_name_ix)
23            .next()?
24            .byte_range(),
25    )?;
26
27    if TRANSFORM_ARRAY.contains(&action_name) {
28        let replacement_as_string = format!("\"{action_name}\"");
29        let range_to_replace = mat.nodes_for_capture_index(array_ix).next()?.byte_range();
30        return Some((range_to_replace, replacement_as_string));
31    }
32
33    None
34}
35
36/// ["editor::GoToPreviousHunk", { "center_cursor": true }] -> "editor::GoToPreviousHunk"
37static TRANSFORM_ARRAY: LazyLock<HashSet<&str>> =
38    LazyLock::new(|| HashSet::from_iter(["editor::GoToHunk", "editor::GoToPreviousHunk"]));
39