tree-sitter/include/tree_sitter/parser.h

#ifndef TREE_SITTER_PARSER_H_
#define TREE_SITTER_PARSER_H_

#ifdef __cplusplus
extern "C" {
#endif

//#define TS_DEBUG_PARSE
// #define TS_DEBUG_LEX

#include "tree_sitter/runtime.h"
#include "tree_sitter/parser/lexer.h"
#include "tree_sitter/parser/stack.h"
#include "tree_sitter/parser/state_machine.h"

#define SYMBOL_NAMES \
static const char *ts_symbol_names[]

#define HIDDEN_SYMBOLS \
static const int hidden_symbol_flags[SYMBOL_COUNT]

#define LEX_STATES \
static TSStateId ts_lex_states[STATE_COUNT]

#define PARSE_TABLE \
static const TSParseAction ts_parse_actions[STATE_COUNT][SYMBOL_COUNT]

#define LEX_FN() \
static TSTree * ts_lex(TSLexer *lexer, TSStateId lex_state)

#ifdef TS_DEBUG_LEX
#include <stdio.h>
#define DEBUG_LEX(...) fprintf(stderr, "\n" __VA_ARGS__)
#else
#define DEBUG_LEX(...)
#endif

#define START_LEXER() \
DEBUG_LEX("LEX %d", lex_state); \
char lookahead; \
next_state: \
lookahead = ts_lexer_lookahead_char(lexer); \
DEBUG_LEX("CHAR '%c'", lookahead);

#define START_TOKEN() \
ts_lexer_start_token(lexer);

#define ADVANCE(state_index) \
{ \
    DEBUG_LEX("ADVANCE %d", state_index); \
    if (!ts_lexer_advance(lexer)) ACCEPT_TOKEN(ts_builtin_sym_end); \
    lex_state = state_index; goto next_state; \
}

#define ACCEPT_TOKEN(symbol) \
{ \
    DEBUG_LEX("TOKEN %s", ts_symbol_names[symbol]); \
    return ts_lexer_build_node(lexer, symbol); \
}

#define LEX_ERROR() \
{ \
    DEBUG_LEX("ERROR"); \
    return ts_lexer_build_node(lexer, ts_builtin_sym_error); \
}

#define LEX_PANIC() \
{ \
    DEBUG_LEX("LEX ERROR: unexpected state %d", lex_state); \
    return NULL; \
}

SYMBOL_NAMES;

static const TSTree * ts_parse(void *data, TSInput input, TSInputEdit *edit) {
    TSStateMachine *parser = (TSStateMachine *)data;
    ts_state_machine_initialize(parser, input, edit);
    for (;;) {
        const TSTree *tree = ts_state_machine_parse(parser, ts_symbol_names);
        if (tree) return tree;
    }
}

#define EXPORT_PARSER(constructor_name) \
TSParser constructor_name() { \
    return (TSParser) { \
        .parse_fn = ts_parse, \
        .free_fn = ts_state_machine_free, \
        .symbol_names = ts_symbol_names, \
        .data = ts_state_machine_make( \
            SYMBOL_COUNT, \
            (const TSParseAction *)ts_parse_actions, \
            ts_lex_states, \
            ts_lex, \
            hidden_symbol_flags \
        ), \
    }; \
}

#ifdef __cplusplus
}
#endif

#endif  // TREE_SITTER_PARSER_H_