1use std::ops::Range;
 2
 3use tree_sitter::{Query, QueryMatch};
 4
 5use crate::MigrationPatterns;
 6
 7pub const SETTINGS_PATTERNS: MigrationPatterns = &[(
 8    SETTINGS_CONTEXT_SERVER_PATTERN,
 9    migrate_context_server_settings,
10)];
11
12const SETTINGS_CONTEXT_SERVER_PATTERN: &str = r#"(document
13    (object
14        (pair
15            key: (string (string_content) @context-servers)
16            value: (object
17                (pair
18                    key: (string (string_content) @server-name)
19                    value: (object) @server-settings
20                )
21            )
22        )
23    )
24    (#eq? @context-servers "context_servers")
25)"#;
26
27fn migrate_context_server_settings(
28    contents: &str,
29    mat: &QueryMatch,
30    query: &Query,
31) -> Option<(Range<usize>, String)> {
32    let server_settings_index = query.capture_index_for_name("server-settings")?;
33    let server_settings = mat.nodes_for_capture_index(server_settings_index).next()?;
34
35    let mut has_command = false;
36    let mut has_settings = false;
37    let mut other_keys = 0;
38    let mut column = None;
39
40    // Parse the server settings to check what keys it contains
41    let mut cursor = server_settings.walk();
42    for child in server_settings.children(&mut cursor) {
43        if child.kind() == "pair"
44            && let Some(key_node) = child.child_by_field_name("key") {
45                if let (None, Some(quote_content)) = (column, key_node.child(0)) {
46                    column = Some(quote_content.start_position().column);
47                }
48                if let Some(string_content) = key_node.child(1) {
49                    let key = &contents[string_content.byte_range()];
50                    match key {
51                        // If it already has a source key, don't modify it
52                        "source" => return None,
53                        "command" => has_command = true,
54                        "settings" => has_settings = true,
55                        _ => other_keys += 1,
56                    }
57                }
58            }
59    }
60
61    let source_type = if has_command { "custom" } else { "extension" };
62
63    // Insert the source key at the beginning of the object
64    let start = server_settings.start_byte() + 1;
65    let indent = " ".repeat(column.unwrap_or(12));
66
67    if !has_command && !has_settings {
68        return Some((
69            start..start,
70            format!(
71                r#"
72{indent}"source": "{}",
73{indent}"settings": {{}}{}
74        "#,
75                source_type,
76                if other_keys > 0 { "," } else { "" }
77            ),
78        ));
79    }
80
81    Some((
82        start..start,
83        format!(
84            r#"
85{indent}"source": "{}","#,
86            source_type
87        ),
88    ))
89}