willbooster-tree-sitter-cpp 25.0.0

#ifndef TREE_SITTER_CPP_INITIALIZER_CONDITIONAL_H_
#define TREE_SITTER_CPP_INITIALIZER_CONDITIONAL_H_

#include "tree_sitter/alloc.h"
#include "tree_sitter/parser.h"
#include "pragma.h"
#include "cppTrivia.h"

#include <stdint.h>
#include <string.h>
#include <wctype.h>

typedef enum { INITIALIZER_UNKNOWN, INITIALIZER_EMPTY, INITIALIZER_NONEMPTY } InitializerConditionalKind;

typedef struct {
    uint32_t children[2];
    InitializerConditionalKind kind;
} InitializerConditionalNode;

typedef struct {
    uint32_t node;
    bool has_else;
} InitializerConditionalFrame;

typedef struct {
    InitializerConditionalNode *nodes;
    uint32_t node_count;
    uint32_t node_capacity;
    InitializerConditionalFrame *frames;
    uint32_t frame_count;
    uint32_t frame_capacity;
    bool invalid_directive_order;
} InitializerConditionals;

typedef enum { INITIALIZER_CONDITION, INITIALIZER_HEADER, INITIALIZER_REPLACEMENT, INITIALIZER_DIRECTIVE_TEXT } InitializerHeaderMode;

typedef struct {
    InitializerHeaderMode mode;
    bool after_identifier;
    bool argument_start;
    uint32_t call_depth;
} InitializerHeaderContext;

typedef enum { INITIALIZER_IF, INITIALIZER_ELSE, INITIALIZER_ELIF, INITIALIZER_ENDIF, INITIALIZER_INCLUDE, INITIALIZER_DEFINE, INITIALIZER_OTHER, INITIALIZER_INVALID } InitializerDirective;

typedef enum { INITIALIZER_START_NONE, INITIALIZER_START_EMPTY, INITIALIZER_START_DIRECTIVE } InitializerStartKind;

static bool scan_empty_initializer_tail(InitializerConditionals *state, TSLexer *lexer);
static bool initializer_conditional_accept_header(InitializerConditionals *state, TSLexer *lexer, InitializerDirective directive);
static bool initializer_conditional_trivia(TSLexer *lexer);
static bool initializer_conditional_splice(TSLexer *lexer);
static uint32_t initializer_conditional_node(InitializerConditionals *state, uint32_t position);
static void initializer_conditional_push(InitializerConditionals *state, uint32_t node);
static bool initializer_conditional_nonempty(InitializerConditionals *state);
static InitializerDirective initializer_conditional_directive(TSLexer *lexer);
static bool initializer_conditional_word(TSLexer *lexer, MacroName *word);
static bool initializer_conditional_header(TSLexer *lexer, InitializerHeaderMode mode);
static bool initializer_conditional_header_operand(TSLexer *lexer, InitializerHeaderContext *context);
static void initializer_conditional_header_identifier(TSLexer *lexer, InitializerHeaderContext *context);
static void initializer_conditional_header_punctuation(InitializerHeaderContext *context, int32_t c);
static void initializer_conditional_pp_number(TSLexer *lexer);
static bool initializer_conditional_separator(TSLexer *lexer);
static void initializer_conditional_literal(TSLexer *lexer, int32_t closing);
static void initializer_conditional_literal_body(TSLexer *lexer, int32_t closing);

static InitializerStartKind scan_initializer_start(InitializerConditionals *state, TSLexer *lexer,
                                                   bool allow_empty, bool allow_directive) {
    TSLexerPosition position;
    lexer->get_position(lexer, &position);
    uint32_t first = 0;
    if (allow_empty) {
        first = initializer_conditional_node(state, position.bytes);
        InitializerConditionalKind cached = state->nodes[first].kind;
        if (cached != INITIALIZER_UNKNOWN) {
            return cached == INITIALIZER_EMPTY ? INITIALIZER_START_EMPTY : INITIALIZER_START_NONE;
        }
    }
    InitializerDirective directive = initializer_conditional_directive(lexer);
    if (allow_directive && directive == INITIALIZER_OTHER) return INITIALIZER_START_DIRECTIVE;
    if (!allow_empty || directive != INITIALIZER_IF || state->invalid_directive_order) return INITIALIZER_START_NONE;
    state->frame_count = 0;
    initializer_conditional_push(state, first);
    if (!initializer_conditional_accept_header(state, lexer, directive)) {
        initializer_conditional_nonempty(state);
        return INITIALIZER_START_NONE;
    }
    return scan_empty_initializer_tail(state, lexer) ? INITIALIZER_START_EMPTY : INITIALIZER_START_NONE;
}

static bool scan_empty_initializer_tail(InitializerConditionals *state, TSLexer *lexer) {
    TSLexerPosition position;
    for (;;) {
        if (!initializer_conditional_trivia(lexer) || lexer->lookahead != '#') {
            return initializer_conditional_nonempty(state);
        }
        lexer->get_position(lexer, &position);
        InitializerDirective directive = initializer_conditional_directive(lexer);
        if (directive == INITIALIZER_IF) {
            uint32_t node = initializer_conditional_node(state, position.bytes);
            if (state->nodes[node].kind == INITIALIZER_NONEMPTY) return initializer_conditional_nonempty(state);
            initializer_conditional_push(state, node);
        }
        if (!initializer_conditional_accept_header(state, lexer, directive)) {
            return initializer_conditional_nonempty(state);
        }
        if (!state->frame_count) return true;
    }
}

static bool initializer_conditional_accept_header(InitializerConditionals *state, TSLexer *lexer, InitializerDirective directive) {
    if (directive == INITIALIZER_INVALID) return false;
    if (directive == INITIALIZER_OTHER || directive == INITIALIZER_INCLUDE || directive == INITIALIZER_DEFINE) {
        InitializerHeaderMode mode = directive == INITIALIZER_INCLUDE ? INITIALIZER_HEADER :
            directive == INITIALIZER_DEFINE ? INITIALIZER_REPLACEMENT : INITIALIZER_DIRECTIVE_TEXT;
        initializer_conditional_header(lexer, mode);
        return true;
    }
    if (directive == INITIALIZER_IF) return initializer_conditional_header(lexer, INITIALIZER_CONDITION);
    if (directive == INITIALIZER_ENDIF) {
        if (initializer_conditional_header(lexer, INITIALIZER_CONDITION)) return false;
        state->nodes[state->frames[--state->frame_count].node].kind = INITIALIZER_EMPTY;
        return true;
    }
    InitializerConditionalFrame *frame = &state->frames[state->frame_count - 1];
    if (frame->has_else) {
        state->invalid_directive_order = true;
        return false;
    }
    frame->has_else = directive == INITIALIZER_ELSE;
    return initializer_conditional_header(lexer, INITIALIZER_CONDITION) != frame->has_else;
}

static bool initializer_conditional_trivia(TSLexer *lexer) {
    for (;;) {
        bool newline = false;
        if (!skip_cpp_trivia(lexer, &newline)) return false;
        if (lexer->lookahead != '_') return true;
        if (!scan_pragma(lexer)) return false;
    }
}

static bool initializer_conditional_splice(TSLexer *lexer) {
    lexer->advance(lexer, false);
    return scan_preproc_newline(lexer, false);
}

static uint32_t initializer_conditional_node(InitializerConditionals *state, uint32_t position) {
    if (!state->node_count) {
        state->node_capacity = 64;
        state->nodes = ts_calloc(state->node_capacity, sizeof(*state->nodes));
        state->node_count = 1;
    }
    uint32_t node = 0;
    for (int bit = 31; bit >= 0; bit--) {
        uint32_t branch = (position >> bit) & 1;
        uint32_t child = state->nodes[node].children[branch];
        if (!child) {
            if (state->node_count == state->node_capacity) {
                uint32_t previous_capacity = state->node_capacity;
                state->node_capacity *= 2;
                state->nodes = ts_realloc(state->nodes, state->node_capacity * sizeof(*state->nodes));
                memset(state->nodes + previous_capacity, 0,
                       (state->node_capacity - previous_capacity) * sizeof(*state->nodes));
            }
            child = state->node_count++;
            state->nodes[node].children[branch] = child;
        }
        node = child;
    }
    return node;
}

static void initializer_conditional_push(InitializerConditionals *state, uint32_t node) {
    if (state->frame_count == state->frame_capacity) {
        state->frame_capacity = state->frame_capacity ? state->frame_capacity * 2 : 16;
        state->frames = ts_realloc(state->frames, state->frame_capacity * sizeof(*state->frames));
    }
    state->frames[state->frame_count++] = (InitializerConditionalFrame){.node = node};
}

static bool initializer_conditional_nonempty(InitializerConditionals *state) {
    for (uint32_t i = 0; i < state->frame_count; i++) {
        state->nodes[state->frames[i].node].kind = INITIALIZER_NONEMPTY;
    }
    return false;
}

static InitializerDirective initializer_conditional_directive(TSLexer *lexer) {
    lexer->advance(lexer, false);
    while (lexer->lookahead == ' ' || lexer->lookahead == '\t') lexer->advance(lexer, false);
    char name[9] = {0};
    MacroName word = {.word = name, .capacity = sizeof(name) - 1, .ascii = true, .lookahead_only = true};
    if (!initializer_conditional_word(lexer, &word)) return INITIALIZER_INVALID;
    if (!word.ascii) return INITIALIZER_OTHER;
    if (!strcmp(name, "if") || !strcmp(name, "ifdef") || !strcmp(name, "ifndef")) return INITIALIZER_IF;
    if (!strcmp(name, "else")) return INITIALIZER_ELSE;
    if (!strcmp(name, "elif") || !strcmp(name, "elifdef") || !strcmp(name, "elifndef")) return INITIALIZER_ELIF;
    if (!strcmp(name, "endif")) return INITIALIZER_ENDIF;
    if (!strcmp(name, "include")) return INITIALIZER_INCLUDE;
    if (!strcmp(name, "define")) return INITIALIZER_DEFINE;
    return INITIALIZER_OTHER;
}

static bool initializer_conditional_word(TSLexer *lexer, MacroName *word) {
    bool has_name = false;
    for (;;) {
        MacroName next = *word;
        MacroNameCharacter character = scan_macro_name_character(lexer, true, &next);
        if (character == MACRO_NAME_SPLICE) continue;
        if (character != MACRO_NAME_CHARACTER) return has_name;
        *word = next;
        has_name = true;
        lexer->mark_end(lexer);
    }
}

// code-gauge-allow functionCognitiveComplexity 16 -- The header loop keeps newline termination and token-role transitions together.
static bool initializer_conditional_header(TSLexer *lexer, InitializerHeaderMode mode) {
    bool content = false;
    InitializerHeaderContext context = {.mode = mode, .argument_start = mode == INITIALIZER_HEADER};
    while (!lexer->eof(lexer)) {
        if (lexer->lookahead == '\n' || lexer->lookahead == '\r') {
            lexer->advance(lexer, false);
            return content;
        }
        if (lexer->lookahead == '\\') {
            content |= !initializer_conditional_splice(lexer);
        } else if (lexer->lookahead == '/') {
            bool newline = false;
            if (!skip_cpp_comment(lexer, &newline)) {
                content = true;
                initializer_conditional_header_punctuation(&context, '/');
            }
        } else if (initializer_conditional_header_operand(lexer, &context)) {
            content = true;
        } else if (preproc_word(lexer->lookahead, false)) {
            initializer_conditional_header_identifier(lexer, &context);
            content = true;
        } else {
            bool token = !iswspace(lexer->lookahead);
            content |= token;
            if (token) initializer_conditional_header_punctuation(&context, lexer->lookahead);
            lexer->advance(lexer, false);
        }
    }
    return content;
}

static bool initializer_conditional_header_operand(TSLexer *lexer, InitializerHeaderContext *context) {
    if (context->mode != INITIALIZER_DIRECTIVE_TEXT &&
        ((lexer->lookahead >= '0' && lexer->lookahead <= '9') || lexer->lookahead == '.')) {
        initializer_conditional_pp_number(lexer);
    } else if (context->mode == INITIALIZER_DIRECTIVE_TEXT && lexer->lookahead == '"') {
        lexer->advance(lexer, false);
        scan_preproc_quote(lexer, '"', false, true);
    } else if ((context->mode != INITIALIZER_DIRECTIVE_TEXT && lexer->lookahead == '\'') ||
               lexer->lookahead == '"' || (context->argument_start && lexer->lookahead == '<')) {
        initializer_conditional_literal(lexer, lexer->lookahead == '<' ? '>' : lexer->lookahead);
    } else {
        return false;
    }
    context->after_identifier = false;
    context->argument_start = false;
    return true;
}

static void initializer_conditional_header_identifier(TSLexer *lexer, InitializerHeaderContext *context) {
    do {
        lexer->advance(lexer, false);
    } while (preproc_word(lexer->lookahead, true));
    context->after_identifier = true;
    context->argument_start = false;
}

static void initializer_conditional_header_punctuation(InitializerHeaderContext *context, int32_t c) {
    if (context->mode != INITIALIZER_CONDITION) {
        context->after_identifier = false;
        context->argument_start = false;
        return;
    }
    if (c == '(') {
        context->argument_start = context->after_identifier;
        if (context->after_identifier || context->call_depth) context->call_depth++;
    } else {
        context->argument_start = c == ',' && context->call_depth;
        if (c == ')' && context->call_depth) context->call_depth--;
    }
    context->after_identifier = false;
}

// code-gauge-allow functionCognitiveComplexity 20 -- The pp-number loop consumes continuations, separators and UCNs in one pass.
static void initializer_conditional_pp_number(TSLexer *lexer) {
    int32_t previous = 0;
    while (!lexer->eof(lexer)) {
        int32_t c = lexer->lookahead;
        bool exponent_sign = (c == '+' || c == '-') &&
            (previous == 'e' || previous == 'E' || previous == 'p' || previous == 'P');
        if (preproc_word(c, true) || c == '.' || exponent_sign) {
            previous = c;
            lexer->advance(lexer, false);
        } else if (c == '\'') {
            if (!initializer_conditional_separator(lexer)) return;
            previous = 0;
        } else if (c == '\\') {
            lexer->advance(lexer, false);
            if (scan_preproc_newline(lexer, false)) continue;
            bool in_number = true;
            if (!scan_preproc_ucn(lexer, &in_number) || !in_number) return;
            previous = 0;
        } else {
            return;
        }
    }
}

static bool initializer_conditional_separator(TSLexer *lexer) {
    lexer->advance(lexer, false);
    while (lexer->lookahead == '\\') {
        lexer->advance(lexer, false);
        if (scan_preproc_newline(lexer, false)) continue;
        bool in_number = true;
        if (scan_preproc_ucn(lexer, &in_number) && in_number) return true;
        if (!lexer->eof(lexer)) lexer->advance(lexer, false);
        initializer_conditional_literal_body(lexer, '\'');
        return false;
    }
    if (preproc_word(lexer->lookahead, true)) return true;
    initializer_conditional_literal_body(lexer, '\'');
    return false;
}

static void initializer_conditional_literal(TSLexer *lexer, int32_t closing) {
    lexer->advance(lexer, false);
    initializer_conditional_literal_body(lexer, closing);
}

static void initializer_conditional_literal_body(TSLexer *lexer, int32_t closing) {
    while (!lexer->eof(lexer) && lexer->lookahead != '\n' && lexer->lookahead != '\r') {
        if (lexer->lookahead == '\\') {
            bool splice = initializer_conditional_splice(lexer);
            if (!splice && closing != '>' && !lexer->eof(lexer)) lexer->advance(lexer, false);
        } else {
            int32_t c = lexer->lookahead;
            lexer->advance(lexer, false);
            if (c == closing) return;
        }
    }
}

static void delete_initializer_conditionals(InitializerConditionals *state) {
    ts_free(state->nodes);
    ts_free(state->frames);
}

#endif