feat: add C, C++, Kotlin, Swift, Scala, Bash language support

crates/codegraph-core/Cargo.toml

@@ -22,6 +22,12 @@ tree-sitter-java = "0.23"
 tree-sitter-c-sharp = "0.23"
 tree-sitter-ruby = "0.23"
 tree-sitter-php = "0.23"
+tree-sitter-c = "0.23"
+tree-sitter-cpp = "0.23"
+tree-sitter-kotlin = "0.3"
+tree-sitter-swift = "0.7"
+tree-sitter-scala = "0.25"
+tree-sitter-bash = "0.23"
 tree-sitter-hcl = "1"
 rayon = "1"
 # `bundled` embeds a second SQLite copy (better-sqlite3 already bundles one).

crates/codegraph-core/src/cfg.rs

@@ -330,6 +330,216 @@ pub static PHP_CFG: CfgRules = CfgRules {
     labeled_node: None,
 };
 
+pub static C_CFG: CfgRules = CfgRules {
+    if_node: Some("if_statement"),
+    if_nodes: &[],
+    elif_node: None,
+    else_clause: None,
+    else_via_alternative: true,
+    if_consequent_field: None,
+    for_nodes: &["for_statement"],
+    condition_field: Some("condition"),
+    while_node: Some("while_statement"),
+    while_nodes: &[],
+    do_node: Some("do_statement"),
+    infinite_loop_node: None,
+    unless_node: None,
+    until_node: None,
+    switch_node: Some("switch_statement"),
+    switch_nodes: &[],
+    case_node: Some("case_statement"),
+    case_nodes: &[],
+    default_node: Some("default_statement"),
+    wildcard_pattern_node: None,
+    try_node: None,
+    try_nodes: &[],
+    catch_node: None,
+    finally_node: None,
+    else_node: None,
+    return_node: Some("return_statement"),
+    throw_node: None,
+    break_node: Some("break_statement"),
+    continue_node: Some("continue_statement"),
+    block_node: Some("compound_statement"),
+    block_nodes: &[],
+    labeled_node: Some("labeled_statement"),
+};
+
+pub static CPP_CFG: CfgRules = CfgRules {
+    if_node: Some("if_statement"),
+    if_nodes: &[],
+    elif_node: None,
+    else_clause: None,
+    else_via_alternative: true,
+    if_consequent_field: None,
+    for_nodes: &["for_statement", "for_range_loop"],
+    condition_field: Some("condition"),
+    while_node: Some("while_statement"),
+    while_nodes: &[],
+    do_node: Some("do_statement"),
+    infinite_loop_node: None,
+    unless_node: None,
+    until_node: None,
+    switch_node: Some("switch_statement"),
+    switch_nodes: &[],
+    case_node: Some("case_statement"),
+    case_nodes: &[],
+    default_node: Some("default_statement"),
+    wildcard_pattern_node: None,
+    try_node: Some("try_statement"),
+    try_nodes: &[],
+    catch_node: Some("catch_clause"),
+    finally_node: None,
+    else_node: None,
+    return_node: Some("return_statement"),
+    throw_node: Some("throw_statement"),
+    break_node: Some("break_statement"),
+    continue_node: Some("continue_statement"),
+    block_node: Some("compound_statement"),
+    block_nodes: &[],
+    labeled_node: Some("labeled_statement"),
+};
+
+pub static KOTLIN_CFG: CfgRules = CfgRules {
+    if_node: Some("if_expression"),
+    if_nodes: &[],
+    elif_node: None,
+    else_clause: None,
+    else_via_alternative: true,
+    if_consequent_field: None,
+    for_nodes: &["for_statement"],
+    condition_field: Some("condition"),
+    while_node: Some("while_statement"),
+    while_nodes: &[],
+    do_node: Some("do_while_statement"),
+    infinite_loop_node: None,
+    unless_node: None,
+    until_node: None,
+    switch_node: Some("when_expression"),
+    switch_nodes: &[],
+    case_node: Some("when_entry"),
+    case_nodes: &[],
+    default_node: None,
+    wildcard_pattern_node: None,
+    try_node: Some("try_expression"),
+    try_nodes: &[],
+    catch_node: Some("catch_block"),
+    finally_node: Some("finally_block"),
+    else_node: None,
+    return_node: Some("jump_expression"),
+    throw_node: Some("throw_expression"),
+    break_node: Some("jump_expression"),
+    continue_node: Some("jump_expression"),
+    block_node: None,
+    block_nodes: &["statements"],
+    labeled_node: None,
+};
+
+pub static SWIFT_CFG: CfgRules = CfgRules {
+    if_node: Some("if_statement"),
+    if_nodes: &[],
+    elif_node: None,
+    else_clause: None,
+    else_via_alternative: true,
+    if_consequent_field: None,
+    for_nodes: &["for_in_statement"],
+    condition_field: Some("condition"),
+    while_node: Some("while_statement"),
+    while_nodes: &[],
+    do_node: Some("repeat_while_statement"),
+    infinite_loop_node: None,
+    unless_node: None,
+    until_node: None,
+    switch_node: Some("switch_statement"),
+    switch_nodes: &[],
+    case_node: Some("switch_entry"),
+    case_nodes: &[],
+    default_node: None,
+    wildcard_pattern_node: Some("wildcard_pattern"),
+    try_node: Some("do_statement"),
+    try_nodes: &[],
+    catch_node: Some("catch_clause"),
+    finally_node: None,
+    else_node: None,
+    return_node: Some("return_statement"),
+    throw_node: Some("throw_statement"),
+    break_node: Some("break_statement"),
+    continue_node: Some("continue_statement"),
+    block_node: None,
+    block_nodes: &["code_block", "class_body"],
+    labeled_node: Some("labeled_statement"),
+};
+
+pub static SCALA_CFG: CfgRules = CfgRules {
+    if_node: Some("if_expression"),
+    if_nodes: &[],
+    elif_node: None,
+    else_clause: None,
+    else_via_alternative: true,
+    if_consequent_field: None,
+    for_nodes: &["for_expression"],
+    condition_field: Some("condition"),
+    while_node: Some("while_expression"),
+    while_nodes: &[],
+    do_node: None,
+    infinite_loop_node: None,
+    unless_node: None,
+    until_node: None,
+    switch_node: Some("match_expression"),
+    switch_nodes: &[],
+    case_node: Some("case_clause"),
+    case_nodes: &[],
+    default_node: None,
+    wildcard_pattern_node: Some("wildcard"),
+    try_node: Some("try_expression"),
+    try_nodes: &[],
+    catch_node: Some("catch_clause"),
+    finally_node: Some("finally_clause"),
+    else_node: None,
+    return_node: Some("return_expression"),
+    throw_node: Some("throw_expression"),
+    break_node: None,
+    continue_node: None,
+    block_node: Some("block"),
+    block_nodes: &["template_body"],
+    labeled_node: None,
+};
+
+pub static BASH_CFG: CfgRules = CfgRules {
+    if_node: Some("if_statement"),
+    if_nodes: &[],
+    elif_node: Some("elif_clause"),
+    else_clause: Some("else_clause"),
+    else_via_alternative: false,
+    if_consequent_field: None,
+    for_nodes: &["for_statement", "c_style_for_statement"],
+    condition_field: Some("condition"),
+    while_node: Some("while_statement"),
+    while_nodes: &[],
+    do_node: None,
+    infinite_loop_node: None,
+    unless_node: None,
+    until_node: Some("until_statement"),
+    switch_node: Some("case_statement"),
+    switch_nodes: &[],
+    case_node: Some("case_item"),
+    case_nodes: &[],
+    default_node: None,
+    wildcard_pattern_node: None,
+    try_node: None,
+    try_nodes: &[],
+    catch_node: None,
+    finally_node: None,
+    else_node: Some("else_clause"),
+    return_node: Some("return_statement"),
+    throw_node: None,
+    break_node: Some("break_statement"),
+    continue_node: Some("continue_statement"),
+    block_node: Some("compound_statement"),
+    block_nodes: &[],
+    labeled_node: None,
+};
+
 /// Get CFG rules for a language ID.
 pub fn get_cfg_rules(lang_id: &str) -> Option<&'static CfgRules> {
     match lang_id {
@@ -341,6 +551,12 @@ pub fn get_cfg_rules(lang_id: &str) -> Option<&'static CfgRules> {
         "csharp" => Some(&CSHARP_CFG),
         "ruby" => Some(&RUBY_CFG),
         "php" => Some(&PHP_CFG),
+        "c" => Some(&C_CFG),
+        "cpp" => Some(&CPP_CFG),
+        "kotlin" => Some(&KOTLIN_CFG),
+        "swift" => Some(&SWIFT_CFG),
+        "scala" => Some(&SCALA_CFG),
+        "bash" => Some(&BASH_CFG),
         _ => None,
     }
 }