tree-sitter/src/runtime/document.c

63 lines
1.7 KiB
C

#include "tree_sitter/parser.h"
#include "runtime/node.h"
#include "runtime/parser.h"
#include "runtime/string_input.h"
struct TSDocument {
TSParser parser;
TSInput input;
TSNode *node;
};
TSDocument *ts_document_make() {
return calloc(sizeof(TSDocument), 1);
}
void ts_document_free(TSDocument *document) {
ts_parser_destroy(&document->parser);
if (document->input.release_fn)
document->input.release_fn(document->input.data);
if (document->node)
ts_node_release(document->node);
free(document);
}
static void reparse(TSDocument *document, TSInputEdit *edit) {
if (document->input.read_fn && document->parser.language) {
const TSTree *tree =
ts_parser_parse(&document->parser, document->input, edit);
if (document->node)
ts_node_release(document->node);
document->node =
ts_node_make_root(tree, document->parser.language->symbol_names);
}
}
void ts_document_set_language(TSDocument *document,
const TSLanguage *language) {
ts_parser_destroy(&document->parser);
document->parser = ts_parser_make(language);
reparse(document, NULL);
}
void ts_document_set_input(TSDocument *document, TSInput input) {
document->input = input;
reparse(document, NULL);
}
void ts_document_edit(TSDocument *document, TSInputEdit edit) {
reparse(document, &edit);
}
const char *ts_document_symbol_name(const TSDocument *document,
const TSTree *tree) {
return document->parser.language->symbol_names[tree->symbol];
}
void ts_document_set_input_string(TSDocument *document, const char *text) {
ts_document_set_input(document, ts_string_input_make(text));
}
TSNode *ts_document_root_node(const TSDocument *document) {
return document->node;
}