Function extract_class_methods

internal/cbm/extract_defs.c:4753–4854

Extract methods inside a class body

// Extract methods inside a class body
static void extract_class_methods(CBMExtractCtx *ctx, TSNode class_node, const char *class_qn,
                                  const CBMLangSpec *spec) {
    TSNode body = find_class_member_body(class_node, ctx->language);
    if (ts_node_is_null(body)) {
        return;
    }

    uint32_t count = ts_node_child_count(body);
    for (uint32_t i = 0; i < count; i++) {
        TSNode child = ts_node_child(body, i);
        if (ts_node_is_null(child)) {
            continue;
        }

        if (ctx->language == CBM_LANG_OBJC &&
            strcmp(ts_node_type(child), "implementation_definition") == 0) {
            extract_objc_impl_methods(ctx, child, class_qn, spec);
            continue;
        }

        // Squirrel wraps each class member in a member_declaration node; the
        // method is the inner function_declaration. Peek through the wrapper.
        if (ctx->language == CBM_LANG_SQUIRREL &&
            strcmp(ts_node_type(child), "member_declaration") == 0) {
            TSNode inner = cbm_find_child_by_kind(child, "function_declaration");
            if (!ts_node_is_null(inner)) {
                child = inner;
            }
        }

        // Python wraps @classmethod / @staticmethod / @property methods in
        // a decorated_definition node. Peek through it to find the inner
        // function_definition so we still emit a Method entry.
        TSNode method_node = child;
        if (strcmp(ts_node_type(child), "decorated_definition") == 0) {
            TSNode def = ts_node_child_by_field_name(child, TS_FIELD("definition"));
            if (ts_node_is_null(def) || !cbm_kind_in_set(def, spec->function_node_types)) {
                continue;
            }
            method_node = def;
        }

        // TS/JS class-field arrow functions: `handleClick = () => {...}` is a
        // public_field_definition whose `value` is an arrow_function (a common
        // React event-handler pattern). It is not in function_node_types, so it
        // would otherwise be dropped. Peek through to the inner arrow and take
        // the method name from the field's `name` child (#new_ts_class_field_arrow).
        if (strcmp(ts_node_type(child), "public_field_definition") == 0) {
            TSNode value = ts_node_child_by_field_name(child, TS_FIELD("value"));
            if (ts_node_is_null(value) || !cbm_kind_in_set(value, spec->function_node_types)) {
                continue;
            }
            TSNode fname = ts_node_child_by_field_name(child, TS_FIELD("name"));
            if (ts_node_is_null(fname)) {
                continue;
            }
            push_method_def(ctx, value, class_node, class_qn, spec, fname);
            continue;