Add MATLAB, Lean 4, FORM, and Magma language support (59 -> 63)

Custom tree-sitter grammars for FORM and Magma (no upstream exists).
MATLAB uses acristoffers/tree-sitter-matlab, Lean uses leanprover/tree-sitter-lean4.
Includes .m file disambiguation (Objective-C vs MATLAB) via Linguist heuristics.

Author: DeusData

internal/cbm/cbm.go

@@ -205,6 +205,11 @@ var languageToC = map[lang.Language]C.CBMLanguage{
 	lang.Meson:      C.CBM_LANG_MESON,
 	lang.GLSL:       C.CBM_LANG_GLSL,
 	lang.INI:        C.CBM_LANG_INI,
+	// Scientific/math languages
+	lang.MATLAB: C.CBM_LANG_MATLAB,
+	lang.Lean:   C.CBM_LANG_LEAN,
+	lang.FORM:   C.CBM_LANG_FORM,
+	lang.Magma:  C.CBM_LANG_MAGMA,
 }
 
 // ParseTimeoutMicros is the default per-file parse timeout (10 seconds).

internal/cbm/cbm.h

@@ -69,6 +69,11 @@ typedef enum {
     CBM_LANG_MESON,
     CBM_LANG_GLSL,
     CBM_LANG_INI,
+    // Scientific/math languages
+    CBM_LANG_MATLAB,
+    CBM_LANG_LEAN,
+    CBM_LANG_FORM,
+    CBM_LANG_MAGMA,
     CBM_LANG_COUNT
 } CBMLanguage;
 

internal/cbm/extract_defs.c

@@ -207,6 +207,36 @@ static TSNode resolve_func_name(TSNode node, CBMLanguage lang, const char* sourc
         return null_node;
     }
 
+    // MATLAB: function_definition — name via "name" field
+    if (lang == CBM_LANG_MATLAB && strcmp(kind, "function_definition") == 0) {
+        // MATLAB grammar: function [ret] = name(args)
+        // The "name" field should be set; fallback to first identifier child
+        if (!ts_node_is_null(name)) return name;
+        return cbm_find_child_by_kind(node, "identifier");
+    }
+
+    // Lean: def/theorem/instance/abbrev — name via declId field
+    if (lang == CBM_LANG_LEAN) {
+        // Lean grammar uses "declId" field for the name (6 chars)
+        TSNode decl_id = ts_node_child_by_field_name(node, "declId", 6);
+        if (!ts_node_is_null(decl_id)) {
+            // declId contains an identifier
+            TSNode id = cbm_find_child_by_kind(decl_id, "ident");
+            if (!ts_node_is_null(id)) return id;
+            // Fallback: first named child of declId
+            if (ts_node_named_child_count(decl_id) > 0)
+                return ts_node_named_child(decl_id, 0);
+            return decl_id;
+        }
+        // Fallback: look for "name" field or first identifier
+        if (!ts_node_is_null(name)) return name;
+        return cbm_find_child_by_kind(node, "ident");
+    }
+
+    // FORM: procedure_definition — name via "name" field (standard)
+    // Magma: function/procedure/intrinsic_definition — name via "name" field (standard)
+    // Both use standard "name" field which is handled above at line 33
+
     // C/C++/CUDA/GLSL: function_definition — name is inside the declarator chain
     // C grammar: function_definition{declarator:function_declarator{declarator:identifier}}
     if ((lang == CBM_LANG_C || lang == CBM_LANG_CPP || lang == CBM_LANG_CUDA ||

internal/cbm/extract_imports.c

@@ -609,6 +609,15 @@ void cbm_extract_imports(CBMExtractCtx* ctx) {
         case CBM_LANG_DART:
             parse_generic_imports(ctx, "import_declaration");
             break;
+        case CBM_LANG_LEAN:
+            parse_generic_imports(ctx, "import");
+            break;
+        case CBM_LANG_FORM:
+            parse_generic_imports(ctx, "include_directive");
+            break;
+        case CBM_LANG_MAGMA:
+            parse_generic_imports(ctx, "load_statement");
+            break;
         default:
             break;
     }

internal/cbm/grammar_form.c

@@ -0,0 +1,2 @@
+// Vendored tree-sitter grammar: form
+#include "vendored/grammars/form/parser.c"

internal/cbm/grammar_lean.c

@@ -0,0 +1,3 @@
+// Vendored tree-sitter grammar: lean
+#include "vendored/grammars/lean/parser.c"
+#include "vendored/grammars/lean/scanner.c"

internal/cbm/grammar_magma.c

@@ -0,0 +1,2 @@
+// Vendored tree-sitter grammar: magma
+#include "vendored/grammars/magma/parser.c"

internal/cbm/grammar_matlab.c

@@ -0,0 +1,3 @@
+// Vendored tree-sitter grammar: matlab
+#include "vendored/grammars/matlab/parser.c"
+#include "vendored/grammars/matlab/scanner.c"

internal/cbm/helpers.c

@@ -202,6 +202,8 @@ bool cbm_is_test_file(const char* rel_path, CBMLanguage lang) {
         case CBM_LANG_C:
             return has_suffix(base, "_test.c") || has_suffix(base, "_test.cc") ||
                    has_suffix(base, "_test.cpp") || has_prefix(base, "test_");
+        case CBM_LANG_MATLAB:
+            return has_prefix(base, "test_") || has_prefix(base, "Test");
         default:
             return false;
     }
@@ -284,6 +286,10 @@ static const char* func_kinds_zig[] = {"function_declaration","test_declaration"
 static const char* func_kinds_bash[] = {"function_definition",NULL};
 static const char* func_kinds_erlang[] = {"function_clause",NULL};
 static const char* func_kinds_csharp[] = {"method_declaration","constructor_declaration",NULL};
+static const char* func_kinds_matlab[] = {"function_definition",NULL};
+static const char* func_kinds_lean[] = {"def","theorem","instance","abbrev",NULL};
+static const char* func_kinds_form[] = {"procedure_definition",NULL};
+static const char* func_kinds_magma[] = {"function_definition","procedure_definition","intrinsic_definition",NULL};
 static const char* func_kinds_generic[] = {"function_declaration","function_definition","method_declaration","method_definition",NULL};
 
 static const char** func_kinds_for_lang(CBMLanguage lang) {
@@ -309,6 +315,10 @@ static const char** func_kinds_for_lang(CBMLanguage lang) {
         case CBM_LANG_BASH:       return func_kinds_bash;
         case CBM_LANG_ERLANG:     return func_kinds_erlang;
         case CBM_LANG_CSHARP:     return func_kinds_csharp;
+        case CBM_LANG_MATLAB:     return func_kinds_matlab;
+        case CBM_LANG_LEAN:       return func_kinds_lean;
+        case CBM_LANG_FORM:       return func_kinds_form;
+        case CBM_LANG_MAGMA:      return func_kinds_magma;
         default:                  return func_kinds_generic;
     }
 }
@@ -439,6 +449,9 @@ static const char* module_parents_config[] = {"document","table","table_array_el
 static const char* module_parents_hcl[] = {"config_file",NULL};
 static const char* module_parents_makefile[] = {"makefile",NULL};
 static const char* module_parents_commonlisp[] = {"source",NULL};
+static const char* module_parents_matlab[] = {"source_file",NULL};
+static const char* module_parents_form[] = {"source_file",NULL};
+static const char* module_parents_magma[] = {"source_file",NULL};
 
 bool cbm_is_module_level(TSNode node, CBMLanguage lang) {
     TSNode parent = ts_node_parent(node);
@@ -519,6 +532,10 @@ bool cbm_is_module_level(TSNode node, CBMLanguage lang) {
         case CBM_LANG_R:        parents = module_parents_php; break;  // program
         case CBM_LANG_MAKEFILE: parents = module_parents_makefile; break;
         case CBM_LANG_COMMONLISP: parents = module_parents_commonlisp; break;
+        case CBM_LANG_MATLAB:     parents = module_parents_matlab; break;
+        case CBM_LANG_LEAN:       parents = module_parents_zig; break;  // source_file
+        case CBM_LANG_FORM:       parents = module_parents_form; break;
+        case CBM_LANG_MAGMA:      parents = module_parents_magma; break;
         default:                return false;
     }
     if (parents) {

internal/cbm/lang_specs.c

@@ -62,6 +62,11 @@ extern const TSLanguage* tree_sitter_svelte(void);
 extern const TSLanguage* tree_sitter_meson(void);
 extern const TSLanguage* tree_sitter_glsl(void);
 extern const TSLanguage* tree_sitter_ini(void);
+// Scientific/math languages
+extern const TSLanguage* tree_sitter_matlab(void);
+extern const TSLanguage* tree_sitter_lean(void);
+extern const TSLanguage* tree_sitter_form(void);
+extern const TSLanguage* tree_sitter_magma(void);
 
 // -- Empty sentinel --
 static const char* empty_types[] = {NULL};
@@ -545,6 +550,37 @@ static const char* ini_module_types[] = {"document",NULL};
 static const char* ini_class_types[] = {"section",NULL};
 static const char* ini_var_types[] = {"setting",NULL};
 
+// ==================== MATLAB ====================
+static const char* matlab_func_types[] = {"function_definition",NULL};
+static const char* matlab_class_types[] = {"class_definition",NULL};
+static const char* matlab_module_types[] = {"source_file",NULL};
+static const char* matlab_branch_types[] = {"if_statement","for_statement","while_statement","switch_statement","try_statement",NULL};
+static const char* matlab_var_types[] = {"assignment",NULL};
+
+// ==================== LEAN ====================
+static const char* lean_func_types[] = {"def","theorem","instance","abbrev",NULL};
+static const char* lean_class_types[] = {"structure","class_inductive","inductive",NULL};
+static const char* lean_module_types[] = {"module",NULL};
+static const char* lean_import_types[] = {"import",NULL};
+static const char* lean_branch_types[] = {"if","match","do",NULL};
+
+// ==================== FORM ====================
+static const char* form_func_types[] = {"procedure_definition",NULL};
+static const char* form_module_types[] = {"source_file",NULL};
+static const char* form_call_types[] = {"call_statement",NULL};
+static const char* form_import_types[] = {"include_directive",NULL};
+static const char* form_branch_types[] = {"if_statement","repeat_statement","do_loop",NULL};
+static const char* form_var_types[] = {"declaration_statement",NULL};
+static const char* form_assign_types[] = {"substitution_statement",NULL};
+
+// ==================== MAGMA ====================
+static const char* magma_func_types[] = {"function_definition","procedure_definition","intrinsic_definition",NULL};
+static const char* magma_module_types[] = {"source_file",NULL};
+static const char* magma_call_types[] = {"call_expression",NULL};
+static const char* magma_import_types[] = {"load_statement",NULL};
+static const char* magma_branch_types[] = {"if_statement","for_statement","while_statement","repeat_statement","case_statement",NULL};
+static const char* magma_var_types[] = {"assignment_statement",NULL};
+
 // ==================== NEW LANG ENV ACCESS ====================
 static const char* julia_env_funcs[] = {"ENV",NULL};
 static const char* nix_env_funcs[] = {"builtins.getEnv",NULL};
@@ -848,6 +884,26 @@ static const CBMLangSpec lang_specs[CBM_LANG_COUNT] = {
     {CBM_LANG_INI, empty_types, ini_class_types, empty_types, ini_module_types, empty_types,
      empty_types, empty_types, empty_types, ini_var_types, empty_types,
      empty_types, NULL, empty_types, NULL, NULL},
+
+    // CBM_LANG_MATLAB (definitions-only, no call graph due to A(1) ambiguity)
+    {CBM_LANG_MATLAB, matlab_func_types, matlab_class_types, empty_types, matlab_module_types, empty_types,
+     empty_types, empty_types, matlab_branch_types, matlab_var_types, matlab_var_types,
+     empty_types, NULL, empty_types, NULL, NULL},
+
+    // CBM_LANG_LEAN (definitions-only, Lean 4 syntax is runtime-extensible)
+    {CBM_LANG_LEAN, lean_func_types, lean_class_types, empty_types, lean_module_types, empty_types,
+     lean_import_types, empty_types, lean_branch_types, empty_types, empty_types,
+     empty_types, NULL, empty_types, NULL, NULL},
+
+    // CBM_LANG_FORM
+    {CBM_LANG_FORM, form_func_types, empty_types, empty_types, form_module_types, form_call_types,
+     form_import_types, empty_types, form_branch_types, form_var_types, form_assign_types,
+     empty_types, NULL, empty_types, NULL, NULL},
+
+    // CBM_LANG_MAGMA
+    {CBM_LANG_MAGMA, magma_func_types, empty_types, empty_types, magma_module_types, magma_call_types,
+     magma_import_types, empty_types, magma_branch_types, magma_var_types, magma_var_types,
+     empty_types, NULL, empty_types, NULL, NULL},
 };
 
 const CBMLangSpec* cbm_lang_spec(CBMLanguage lang) {
@@ -916,6 +972,10 @@ const TSLanguage* cbm_ts_language(CBMLanguage lang) {
         case CBM_LANG_MESON:      return tree_sitter_meson();
         case CBM_LANG_GLSL:       return tree_sitter_glsl();
         case CBM_LANG_INI:        return tree_sitter_ini();
+        case CBM_LANG_MATLAB:     return tree_sitter_matlab();
+        case CBM_LANG_LEAN:       return tree_sitter_lean();
+        case CBM_LANG_FORM:       return tree_sitter_form();
+        case CBM_LANG_MAGMA:      return tree_sitter_magma();
         default:                  return NULL;
     }
 }