mirror of https://github.com/Wilfred/difftastic/
746 lines
25 KiB
C
746 lines
25 KiB
C
#include <assert.h>
|
|
#include <ctype.h>
|
|
#include <string.h>
|
|
#include <tree_sitter/parser.h>
|
|
#include <wctype.h>
|
|
|
|
#define MAX(a, b) ((a) > (b) ? (a) : (b))
|
|
|
|
#define MIN(a, b) ((a) < (b) ? (a) : (b))
|
|
|
|
#define STRING_RESIZE(vec, _cap) \
|
|
void *tmp = realloc((vec).data, ((_cap) + 1) * sizeof((vec).data[0])); \
|
|
assert(tmp != NULL); \
|
|
(vec).data = tmp; \
|
|
memset((vec).data + (vec).len, 0, \
|
|
(((_cap) + 1) - (vec).len) * sizeof((vec).data[0])); \
|
|
(vec).cap = (_cap);
|
|
|
|
#define STRING_GROW(vec, _cap) \
|
|
if ((vec).cap < (_cap)) { \
|
|
STRING_RESIZE((vec), (_cap)); \
|
|
}
|
|
|
|
#define STRING_PUSH(vec, el) \
|
|
if ((vec).cap == (vec).len) { \
|
|
STRING_RESIZE((vec), MAX(16, (vec).len * 2)); \
|
|
} \
|
|
(vec).data[(vec).len++] = (el);
|
|
|
|
#define STRING_FREE(vec) \
|
|
if ((vec).data != NULL) \
|
|
free((vec).data);
|
|
|
|
#define STRING_CLEAR(vec) \
|
|
{ \
|
|
(vec).len = 0; \
|
|
memset((vec).data, 0, (vec).cap * sizeof(char)); \
|
|
}
|
|
|
|
enum TokenType {
|
|
HEREDOC_START,
|
|
SIMPLE_HEREDOC_BODY,
|
|
HEREDOC_BODY_BEGINNING,
|
|
HEREDOC_BODY_MIDDLE,
|
|
HEREDOC_END,
|
|
FILE_DESCRIPTOR,
|
|
EMPTY_VALUE,
|
|
CONCAT,
|
|
VARIABLE_NAME,
|
|
REGEX,
|
|
REGEX_NO_SLASH,
|
|
REGEX_NO_SPACE,
|
|
EXTGLOB_PATTERN,
|
|
BARE_DOLLAR,
|
|
BRACE_START,
|
|
CLOSING_BRACE,
|
|
CLOSING_BRACKET,
|
|
HEREDOC_ARROW,
|
|
HEREDOC_ARROW_DASH,
|
|
NEWLINE,
|
|
};
|
|
|
|
typedef struct {
|
|
uint32_t cap;
|
|
uint32_t len;
|
|
char *data;
|
|
} String;
|
|
|
|
static String string_new() {
|
|
return (String){.cap = 16, .len = 0, .data = calloc(1, sizeof(char) * 17)};
|
|
}
|
|
|
|
typedef struct {
|
|
bool heredoc_is_raw;
|
|
bool started_heredoc;
|
|
bool heredoc_allows_indent;
|
|
String heredoc_delimiter;
|
|
String current_leading_word;
|
|
} Scanner;
|
|
|
|
static inline void advance(TSLexer *lexer) { lexer->advance(lexer, false); }
|
|
|
|
static inline void skip(TSLexer *lexer) { lexer->advance(lexer, true); }
|
|
|
|
static inline bool in_error_recovery(const bool *valid_symbols) {
|
|
return valid_symbols[HEREDOC_START] && valid_symbols[HEREDOC_END] &&
|
|
valid_symbols[FILE_DESCRIPTOR] && valid_symbols[EMPTY_VALUE] &&
|
|
valid_symbols[CONCAT] && valid_symbols[REGEX];
|
|
}
|
|
|
|
static inline void reset(Scanner *scanner) {
|
|
scanner->heredoc_is_raw = false;
|
|
scanner->started_heredoc = false;
|
|
scanner->heredoc_allows_indent = false;
|
|
STRING_CLEAR(scanner->heredoc_delimiter);
|
|
}
|
|
|
|
static unsigned serialize(Scanner *scanner, char *buffer) {
|
|
if (scanner->heredoc_delimiter.len + 3 >=
|
|
TREE_SITTER_SERIALIZATION_BUFFER_SIZE) {
|
|
return 0;
|
|
}
|
|
buffer[0] = (char)scanner->heredoc_is_raw;
|
|
buffer[1] = (char)scanner->started_heredoc;
|
|
buffer[2] = (char)scanner->heredoc_allows_indent;
|
|
memcpy(&buffer[3], scanner->heredoc_delimiter.data,
|
|
scanner->heredoc_delimiter.len);
|
|
return scanner->heredoc_delimiter.len + 3;
|
|
}
|
|
|
|
static void deserialize(Scanner *scanner, const char *buffer, unsigned length) {
|
|
if (length == 0) {
|
|
reset(scanner);
|
|
} else {
|
|
scanner->heredoc_is_raw = buffer[0];
|
|
scanner->started_heredoc = buffer[1];
|
|
scanner->heredoc_allows_indent = buffer[2];
|
|
scanner->heredoc_delimiter.len = length - 3;
|
|
STRING_GROW(scanner->heredoc_delimiter, scanner->heredoc_delimiter.len);
|
|
memcpy(scanner->heredoc_delimiter.data, &buffer[3],
|
|
scanner->heredoc_delimiter.len);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Consume a "word" in POSIX parlance, and returns it unquoted.
|
|
*
|
|
* This is an approximate implementation that doesn't deal with any
|
|
* POSIX-mandated substitution, and assumes the default value for
|
|
* IFS.
|
|
*/
|
|
static bool advance_word(TSLexer *lexer, String *unquoted_word) {
|
|
bool empty = true;
|
|
|
|
int32_t quote = 0;
|
|
if (lexer->lookahead == '\'' || lexer->lookahead == '"') {
|
|
quote = lexer->lookahead;
|
|
advance(lexer);
|
|
}
|
|
|
|
while (lexer->lookahead &&
|
|
!(quote ? lexer->lookahead == quote : iswspace(lexer->lookahead))) {
|
|
if (lexer->lookahead == '\\') {
|
|
advance(lexer);
|
|
if (!lexer->lookahead) {
|
|
return false;
|
|
}
|
|
}
|
|
empty = false;
|
|
STRING_PUSH(*unquoted_word, lexer->lookahead);
|
|
advance(lexer);
|
|
}
|
|
|
|
if (quote && lexer->lookahead == quote) {
|
|
advance(lexer);
|
|
}
|
|
|
|
return !empty;
|
|
}
|
|
|
|
static bool scan_heredoc_start(Scanner *scanner, TSLexer *lexer) {
|
|
while (iswspace(lexer->lookahead)) {
|
|
skip(lexer);
|
|
}
|
|
|
|
lexer->result_symbol = HEREDOC_START;
|
|
scanner->heredoc_is_raw = lexer->lookahead == '\'' ||
|
|
lexer->lookahead == '"' ||
|
|
lexer->lookahead == '\\';
|
|
scanner->started_heredoc = false;
|
|
STRING_CLEAR(scanner->heredoc_delimiter);
|
|
|
|
bool found_delimiter = advance_word(lexer, &scanner->heredoc_delimiter);
|
|
if (!found_delimiter)
|
|
STRING_CLEAR(scanner->heredoc_delimiter);
|
|
return found_delimiter;
|
|
}
|
|
|
|
static bool scan_heredoc_end_identifier(Scanner *scanner, TSLexer *lexer) {
|
|
STRING_CLEAR(scanner->current_leading_word);
|
|
// Scan the first 'n' characters on this line, to see if they match the
|
|
// heredoc delimiter
|
|
int32_t size = 0;
|
|
while (lexer->lookahead != '\0' && lexer->lookahead != '\n' &&
|
|
((int32_t)scanner->heredoc_delimiter.data[size++]) ==
|
|
lexer->lookahead &&
|
|
scanner->current_leading_word.len < scanner->heredoc_delimiter.len) {
|
|
STRING_PUSH(scanner->current_leading_word, lexer->lookahead);
|
|
advance(lexer);
|
|
}
|
|
return strcmp(scanner->current_leading_word.data,
|
|
scanner->heredoc_delimiter.data) == 0;
|
|
}
|
|
|
|
static bool scan_heredoc_content(Scanner *scanner, TSLexer *lexer,
|
|
enum TokenType middle_type,
|
|
enum TokenType end_type) {
|
|
bool did_advance = false;
|
|
|
|
for (;;) {
|
|
switch (lexer->lookahead) {
|
|
case '\0': {
|
|
if (lexer->eof(lexer) && did_advance) {
|
|
reset(scanner);
|
|
lexer->result_symbol = end_type;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
case '\\': {
|
|
did_advance = true;
|
|
advance(lexer);
|
|
advance(lexer);
|
|
break;
|
|
}
|
|
|
|
case '$': {
|
|
if (scanner->heredoc_is_raw) {
|
|
did_advance = true;
|
|
advance(lexer);
|
|
break;
|
|
}
|
|
if (did_advance) {
|
|
lexer->result_symbol = middle_type;
|
|
scanner->started_heredoc = true;
|
|
return true;
|
|
}
|
|
if (middle_type == HEREDOC_BODY_BEGINNING &&
|
|
lexer->get_column(lexer) == 0) {
|
|
lexer->result_symbol = middle_type;
|
|
scanner->started_heredoc = true;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
case '\n': {
|
|
if (!did_advance) {
|
|
skip(lexer);
|
|
} else {
|
|
advance(lexer);
|
|
}
|
|
did_advance = true;
|
|
if (scanner->heredoc_allows_indent) {
|
|
while (iswspace(lexer->lookahead)) {
|
|
advance(lexer);
|
|
}
|
|
}
|
|
lexer->result_symbol =
|
|
scanner->started_heredoc ? middle_type : end_type;
|
|
lexer->mark_end(lexer);
|
|
if (scan_heredoc_end_identifier(scanner, lexer)) {
|
|
return true;
|
|
}
|
|
break;
|
|
}
|
|
|
|
default: {
|
|
if (lexer->get_column(lexer) == 0) {
|
|
// an alternative is to check the starting column of the
|
|
// heredoc body and track that statefully
|
|
while (iswspace(lexer->lookahead)) {
|
|
skip(lexer);
|
|
}
|
|
if (end_type != SIMPLE_HEREDOC_BODY) {
|
|
lexer->result_symbol = middle_type;
|
|
if (scan_heredoc_end_identifier(scanner, lexer)) {
|
|
return true;
|
|
}
|
|
}
|
|
if (end_type == SIMPLE_HEREDOC_BODY) {
|
|
lexer->result_symbol = end_type;
|
|
lexer->mark_end(lexer);
|
|
if (scan_heredoc_end_identifier(scanner, lexer)) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
did_advance = true;
|
|
advance(lexer);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static bool scan(Scanner *scanner, TSLexer *lexer, const bool *valid_symbols) {
|
|
if (valid_symbols[CONCAT]) {
|
|
if (!(lexer->lookahead == 0 || iswspace(lexer->lookahead) ||
|
|
lexer->lookahead == '\\' || lexer->lookahead == '>' ||
|
|
lexer->lookahead == '<' || lexer->lookahead == ')' ||
|
|
lexer->lookahead == '(' || lexer->lookahead == ';' ||
|
|
lexer->lookahead == '&' || lexer->lookahead == '|' ||
|
|
(lexer->lookahead == '}' && valid_symbols[CLOSING_BRACE]) ||
|
|
(lexer->lookahead == ']' && valid_symbols[CLOSING_BRACKET]))) {
|
|
lexer->result_symbol = CONCAT;
|
|
// This sucks
|
|
if (lexer->lookahead == '`') {
|
|
lexer->mark_end(lexer);
|
|
advance(lexer);
|
|
while (lexer->lookahead != '`' && !lexer->eof(lexer)) {
|
|
advance(lexer);
|
|
}
|
|
if (lexer->eof(lexer)) {
|
|
return false;
|
|
}
|
|
if (lexer->lookahead == '`') {
|
|
advance(lexer);
|
|
}
|
|
return iswspace(lexer->lookahead) || lexer->eof(lexer);
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
|
|
if (valid_symbols[BARE_DOLLAR] && !in_error_recovery(valid_symbols)) {
|
|
while (iswspace(lexer->lookahead)) {
|
|
skip(lexer);
|
|
}
|
|
|
|
if (lexer->lookahead == '$') {
|
|
advance(lexer);
|
|
lexer->result_symbol = BARE_DOLLAR;
|
|
return iswspace(lexer->lookahead) || lexer->eof(lexer);
|
|
}
|
|
}
|
|
|
|
if (valid_symbols[EMPTY_VALUE]) {
|
|
if (iswspace(lexer->lookahead) || lexer->eof(lexer) ||
|
|
lexer->lookahead == ';' || lexer->lookahead == '&') {
|
|
lexer->result_symbol = EMPTY_VALUE;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
if ((valid_symbols[HEREDOC_BODY_BEGINNING] ||
|
|
valid_symbols[SIMPLE_HEREDOC_BODY]) &&
|
|
scanner->heredoc_delimiter.len > 0 && !scanner->started_heredoc &&
|
|
!in_error_recovery(valid_symbols)) {
|
|
return scan_heredoc_content(scanner, lexer, HEREDOC_BODY_BEGINNING,
|
|
SIMPLE_HEREDOC_BODY);
|
|
}
|
|
|
|
if (valid_symbols[HEREDOC_END]) {
|
|
if (scan_heredoc_end_identifier(scanner, lexer)) {
|
|
reset(scanner);
|
|
lexer->result_symbol = HEREDOC_END;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
if (valid_symbols[HEREDOC_BODY_MIDDLE] &&
|
|
scanner->heredoc_delimiter.len > 0 && scanner->started_heredoc &&
|
|
!in_error_recovery(valid_symbols)) {
|
|
return scan_heredoc_content(scanner, lexer, HEREDOC_BODY_MIDDLE,
|
|
HEREDOC_END);
|
|
}
|
|
|
|
if (valid_symbols[HEREDOC_START] && !in_error_recovery(valid_symbols)) {
|
|
return scan_heredoc_start(scanner, lexer);
|
|
}
|
|
|
|
if ((valid_symbols[VARIABLE_NAME] || valid_symbols[FILE_DESCRIPTOR] ||
|
|
valid_symbols[HEREDOC_ARROW]) &&
|
|
!valid_symbols[REGEX_NO_SLASH]) {
|
|
for (;;) {
|
|
if (lexer->lookahead == ' ' || lexer->lookahead == '\t' ||
|
|
lexer->lookahead == '\r' ||
|
|
(lexer->lookahead == '\n' && !valid_symbols[NEWLINE])) {
|
|
skip(lexer);
|
|
} else if (lexer->lookahead == '\\') {
|
|
skip(lexer);
|
|
if (lexer->lookahead == '\r') {
|
|
skip(lexer);
|
|
}
|
|
if (lexer->lookahead == '\n') {
|
|
skip(lexer);
|
|
} else {
|
|
return false;
|
|
}
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
|
|
// no '*', '@', '?', '-', '$', '0', '_'
|
|
if (lexer->lookahead == '*' || lexer->lookahead == '@' ||
|
|
lexer->lookahead == '?' || lexer->lookahead == '-' ||
|
|
lexer->lookahead == '$' || lexer->lookahead == '0' ||
|
|
lexer->lookahead == '_') {
|
|
lexer->mark_end(lexer);
|
|
advance(lexer);
|
|
if (lexer->lookahead == '=' || lexer->lookahead == '[' ||
|
|
lexer->lookahead == ':' || lexer->lookahead == '-' ||
|
|
lexer->lookahead == '%' || lexer->lookahead == '#' ||
|
|
lexer->lookahead == '/')
|
|
return false;
|
|
}
|
|
|
|
if (valid_symbols[HEREDOC_ARROW] && lexer->lookahead == '<') {
|
|
advance(lexer);
|
|
if (lexer->lookahead == '<') {
|
|
advance(lexer);
|
|
if (lexer->lookahead == '-') {
|
|
advance(lexer);
|
|
scanner->heredoc_allows_indent = true;
|
|
lexer->result_symbol = HEREDOC_ARROW_DASH;
|
|
} else if (lexer->lookahead == '<' || lexer->lookahead == '=') {
|
|
return false;
|
|
} else {
|
|
scanner->heredoc_allows_indent = false;
|
|
lexer->result_symbol = HEREDOC_ARROW;
|
|
}
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool is_number = true;
|
|
if (iswdigit(lexer->lookahead)) {
|
|
advance(lexer);
|
|
} else if (iswalpha(lexer->lookahead) || lexer->lookahead == '_') {
|
|
is_number = false;
|
|
advance(lexer);
|
|
} else {
|
|
if (lexer->lookahead == '{') {
|
|
goto brace_start;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
for (;;) {
|
|
if (iswdigit(lexer->lookahead)) {
|
|
advance(lexer);
|
|
} else if (iswalpha(lexer->lookahead) || lexer->lookahead == '_') {
|
|
is_number = false;
|
|
advance(lexer);
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (is_number && valid_symbols[FILE_DESCRIPTOR] &&
|
|
(lexer->lookahead == '>' || lexer->lookahead == '<')) {
|
|
lexer->result_symbol = FILE_DESCRIPTOR;
|
|
return true;
|
|
}
|
|
|
|
if (valid_symbols[VARIABLE_NAME]) {
|
|
if (lexer->lookahead == '+') {
|
|
lexer->mark_end(lexer);
|
|
advance(lexer);
|
|
if (lexer->lookahead == '=' || lexer->lookahead == ':' ||
|
|
valid_symbols[CLOSING_BRACE]) {
|
|
lexer->result_symbol = VARIABLE_NAME;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
if (lexer->lookahead == '=' || lexer->lookahead == '[' ||
|
|
lexer->lookahead == ':' || lexer->lookahead == '%' ||
|
|
(lexer->lookahead == '#' && !is_number) ||
|
|
lexer->lookahead == '/') {
|
|
lexer->mark_end(lexer);
|
|
lexer->result_symbol = VARIABLE_NAME;
|
|
return true;
|
|
}
|
|
|
|
if (lexer->lookahead == '?') {
|
|
lexer->mark_end(lexer);
|
|
advance(lexer);
|
|
lexer->result_symbol = VARIABLE_NAME;
|
|
return isalpha(lexer->lookahead);
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
if (valid_symbols[REGEX] || valid_symbols[REGEX_NO_SLASH] ||
|
|
valid_symbols[REGEX_NO_SPACE] && !in_error_recovery(valid_symbols)) {
|
|
if (valid_symbols[REGEX] || valid_symbols[REGEX_NO_SPACE]) {
|
|
while (iswspace(lexer->lookahead)) {
|
|
skip(lexer);
|
|
}
|
|
}
|
|
|
|
if (lexer->lookahead != '"' && lexer->lookahead != '\'' ||
|
|
(lexer->lookahead == '$' && valid_symbols[REGEX_NO_SLASH])) {
|
|
typedef struct {
|
|
bool done;
|
|
bool advanced_once;
|
|
bool found_non_alnumdollarunderdash;
|
|
uint32_t paren_depth;
|
|
uint32_t bracket_depth;
|
|
uint32_t brace_depth;
|
|
} State;
|
|
|
|
lexer->mark_end(lexer);
|
|
|
|
State state = {false, false, false, 0, 0, 0};
|
|
while (!state.done) {
|
|
switch (lexer->lookahead) {
|
|
case '\0':
|
|
return false;
|
|
case '(':
|
|
state.paren_depth++;
|
|
break;
|
|
case '[':
|
|
state.bracket_depth++;
|
|
break;
|
|
case '{':
|
|
state.brace_depth++;
|
|
break;
|
|
case ')':
|
|
if (state.paren_depth == 0) {
|
|
state.done = true;
|
|
}
|
|
state.paren_depth--;
|
|
break;
|
|
case ']':
|
|
if (state.bracket_depth == 0) {
|
|
state.done = true;
|
|
}
|
|
state.bracket_depth--;
|
|
break;
|
|
case '}':
|
|
if (state.brace_depth == 0) {
|
|
state.done = true;
|
|
}
|
|
state.brace_depth--;
|
|
break;
|
|
}
|
|
|
|
if (!state.done) {
|
|
if (valid_symbols[REGEX]) {
|
|
bool was_space = iswspace(lexer->lookahead);
|
|
advance(lexer);
|
|
state.advanced_once = true;
|
|
if (!was_space) {
|
|
lexer->mark_end(lexer);
|
|
}
|
|
} else if (valid_symbols[REGEX_NO_SLASH]) {
|
|
if (lexer->lookahead == '/') {
|
|
lexer->mark_end(lexer);
|
|
lexer->result_symbol = REGEX_NO_SLASH;
|
|
return true;
|
|
}
|
|
if (lexer->lookahead == '\\') {
|
|
advance(lexer);
|
|
if (!lexer->eof(lexer)) {
|
|
advance(lexer);
|
|
}
|
|
} else {
|
|
bool was_space = iswspace(lexer->lookahead);
|
|
advance(lexer);
|
|
if (!was_space) {
|
|
lexer->mark_end(lexer);
|
|
}
|
|
}
|
|
} else if (valid_symbols[REGEX_NO_SPACE]) {
|
|
if (lexer->lookahead == '\\') {
|
|
state.found_non_alnumdollarunderdash = true;
|
|
advance(lexer);
|
|
if (!lexer->eof(lexer)) {
|
|
advance(lexer);
|
|
}
|
|
} else {
|
|
if (iswspace(lexer->lookahead)) {
|
|
lexer->mark_end(lexer);
|
|
lexer->result_symbol = REGEX_NO_SPACE;
|
|
return state.found_non_alnumdollarunderdash;
|
|
}
|
|
/* state. = true; */
|
|
if (!iswalnum(lexer->lookahead) &&
|
|
lexer->lookahead != '$' &&
|
|
lexer->lookahead != '-' &&
|
|
lexer->lookahead != '_') {
|
|
state.found_non_alnumdollarunderdash = true;
|
|
}
|
|
advance(lexer);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
lexer->result_symbol =
|
|
valid_symbols[REGEX_NO_SLASH] ? REGEX_NO_SLASH
|
|
: valid_symbols[REGEX_NO_SPACE] ? REGEX_NO_SPACE
|
|
: REGEX;
|
|
if (valid_symbols[REGEX] && !state.advanced_once) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
|
|
if (valid_symbols[EXTGLOB_PATTERN]) {
|
|
// first skip ws, then check for ? * + @ !
|
|
while (iswspace(lexer->lookahead)) {
|
|
skip(lexer);
|
|
}
|
|
|
|
if (lexer->lookahead == '?' || lexer->lookahead == '*' ||
|
|
lexer->lookahead == '+' || lexer->lookahead == '@' ||
|
|
lexer->lookahead == '!') {
|
|
lexer->mark_end(lexer);
|
|
advance(lexer);
|
|
|
|
if (lexer->lookahead != '(') {
|
|
return false;
|
|
}
|
|
|
|
typedef struct {
|
|
bool done;
|
|
uint32_t paren_depth;
|
|
uint32_t bracket_depth;
|
|
uint32_t brace_depth;
|
|
} State;
|
|
|
|
State state = {false, 0, 0, 0};
|
|
while (!state.done) {
|
|
switch (lexer->lookahead) {
|
|
case '\0':
|
|
return false;
|
|
case '(':
|
|
state.paren_depth++;
|
|
break;
|
|
case '[':
|
|
state.bracket_depth++;
|
|
break;
|
|
case '{':
|
|
state.brace_depth++;
|
|
break;
|
|
case ')':
|
|
if (state.paren_depth == 0) {
|
|
state.done = true;
|
|
}
|
|
state.paren_depth--;
|
|
break;
|
|
case ']':
|
|
if (state.bracket_depth == 0) {
|
|
state.done = true;
|
|
}
|
|
state.bracket_depth--;
|
|
break;
|
|
case '}':
|
|
if (state.brace_depth == 0) {
|
|
state.done = true;
|
|
}
|
|
state.brace_depth--;
|
|
break;
|
|
}
|
|
|
|
if (!state.done) {
|
|
bool was_space = iswspace(lexer->lookahead);
|
|
advance(lexer);
|
|
if (!was_space) {
|
|
lexer->mark_end(lexer);
|
|
}
|
|
}
|
|
}
|
|
|
|
lexer->result_symbol = EXTGLOB_PATTERN;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
brace_start:
|
|
if (valid_symbols[BRACE_START] && !in_error_recovery(valid_symbols)) {
|
|
while (iswspace(lexer->lookahead)) {
|
|
skip(lexer);
|
|
}
|
|
|
|
if (lexer->lookahead != '{') {
|
|
return false;
|
|
}
|
|
|
|
advance(lexer);
|
|
lexer->mark_end(lexer);
|
|
|
|
while (isdigit(lexer->lookahead)) {
|
|
advance(lexer);
|
|
}
|
|
|
|
if (lexer->lookahead != '.') {
|
|
return false;
|
|
}
|
|
advance(lexer);
|
|
|
|
if (lexer->lookahead != '.') {
|
|
return false;
|
|
}
|
|
advance(lexer);
|
|
|
|
while (isdigit(lexer->lookahead)) {
|
|
advance(lexer);
|
|
}
|
|
|
|
if (lexer->lookahead != '}') {
|
|
return false;
|
|
}
|
|
|
|
lexer->result_symbol = BRACE_START;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void *tree_sitter_bash_external_scanner_create() {
|
|
Scanner *scanner = calloc(1, sizeof(Scanner));
|
|
scanner->heredoc_delimiter = string_new();
|
|
scanner->current_leading_word = string_new();
|
|
return scanner;
|
|
}
|
|
|
|
bool tree_sitter_bash_external_scanner_scan(void *payload, TSLexer *lexer,
|
|
const bool *valid_symbols) {
|
|
Scanner *scanner = (Scanner *)payload;
|
|
return scan(scanner, lexer, valid_symbols);
|
|
}
|
|
|
|
unsigned tree_sitter_bash_external_scanner_serialize(void *payload,
|
|
char *state) {
|
|
Scanner *scanner = (Scanner *)payload;
|
|
return serialize(scanner, state);
|
|
}
|
|
|
|
void tree_sitter_bash_external_scanner_deserialize(void *payload,
|
|
const char *state,
|
|
unsigned length) {
|
|
Scanner *scanner = (Scanner *)payload;
|
|
deserialize(scanner, state, length);
|
|
}
|
|
|
|
void tree_sitter_bash_external_scanner_destroy(void *payload) {
|
|
Scanner *scanner = (Scanner *)payload;
|
|
STRING_FREE(scanner->heredoc_delimiter);
|
|
STRING_FREE(scanner->current_leading_word);
|
|
free(scanner);
|
|
}
|