st/lang
1
0
Fork 0
mirror of https://codeberg.org/ProgramSnail/lang.git synced 2025-12-09 08:28:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

symbol docs module implemented, part of file tree implemented

Browse source
This commit is contained in:
ProgramSnail 2023-07-17 23:15:12 +03:00
parent b56b72c98e
commit 100779d2d4
6 changed files with 232 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

86
include/tree_sitter_wrapper.hpp
View file

@ -5,7 +5,7 @@
// for clangd
#include "tree_sitter/api.h"
extern "C" const TSLanguage* tree_sitter_lang();
extern "C" const TSLanguage *tree_sitter_lang();
namespace parser {
@ -13,25 +13,22 @@ class ParseTree {
public:
class Node {
public:
Node(const TSNode& node, const std::string* source) : node_(node), source_(source) {}
Node(const TSNode &node, const std::string *source)
: node_(node), source_(source) {}
std::string GetType() {
return ts_node_type(node_);
}
std::string GetType() { return ts_node_type(node_); }
std::pair<size_t, size_t> GetStartPoint() {
TSPoint point = ts_node_start_point(node_);
return {point.row, point.column};
TSPoint point = ts_node_start_point(node_);
return {point.row, point.column};
}
std::pair<size_t, size_t> GetEndPoint() {
TSPoint point = ts_node_end_point(node_);
return {point.row, point.column};
TSPoint point = ts_node_end_point(node_);
return {point.row, point.column};
}
std::string GetAsSExpression() {
return ts_node_string(node_);
}
std::string GetAsSExpression() { return ts_node_string(node_); }
std::string GetValue() { // from source
size_t start = ts_node_start_byte(node_);
@ -39,43 +36,30 @@ public:
return source_->substr(start, end - start);
}
bool IsNull() {
return ts_node_is_null(node_);
}
bool IsNull() { return ts_node_is_null(node_); }
bool IsNamed() {
return ts_node_is_named(node_);
}
bool IsNamed() { return ts_node_is_named(node_); }
bool IsMissing() {
return ts_node_is_missing(node_);
}
bool IsMissing() { return ts_node_is_missing(node_); }
bool IsExtra() { // comments, etc.
return ts_node_is_extra(node_);
}
bool HasError() {
return ts_node_has_error(node_);
}
bool HasError() { return ts_node_has_error(node_); }
Node NthChild(size_t n) {
return Node(ts_node_child(node_, n), source_);
}
Node NthChild(size_t n) { return Node(ts_node_child(node_, n), source_); }
size_t ChildCount() {
return ts_node_child_count(node_);
}
size_t ChildCount() { return ts_node_child_count(node_); }
Node NthNamedChild(size_t n) {
return Node(ts_node_named_child(node_, n), source_);
}
size_t NamedChildCount() {
return ts_node_named_child_count(node_);
}
size_t NamedChildCount() { return ts_node_named_child_count(node_); }
Node ChildByFieldName(const std::string& name) {
return Node(ts_node_child_by_field_name(node_, name.c_str(), name.size()), source_);
Node ChildByFieldName(const std::string &name) {
return Node(ts_node_child_by_field_name(node_, name.c_str(), name.size()),
source_);
}
Node PreviousSibling() {
@ -86,46 +70,40 @@ public:
return Node(ts_node_prev_named_sibling(node_), source_);
}
Node NextSibling() {
return Node(ts_node_next_sibling(node_), source_);
}
Node NextSibling() { return Node(ts_node_next_sibling(node_), source_); }
Node NextNamedSibling() {
return Node(ts_node_next_named_sibling(node_), source_);
}
private:
TSNode node_;
const std::string* source_;
const std::string *source_;
};
ParseTree(const std::string& source) : source_(source) {
TSParser* parser = ts_parser_new();
ParseTree(const std::string &source) : source_(source) {
TSParser *parser = ts_parser_new();
ts_parser_set_language(parser, tree_sitter_lang());
tree_ = ts_parser_parse_string(
parser,
nullptr,
source_.c_str(),
source_.size());
tree_ = ts_parser_parse_string(parser, nullptr, source_.c_str(),
source_.size());
ts_parser_delete(parser);
}
ParseTree(const ParseTree& parse_tree) : tree_(ts_tree_copy(parse_tree.tree_)), source_(parse_tree.source_) {}
ParseTree(const ParseTree &parse_tree)
: tree_(ts_tree_copy(parse_tree.tree_)), source_(parse_tree.source_) {}
Node GetRoot() const {
return Node(ts_tree_root_node(tree_), &source_);
}
Node GetRoot() const { return Node(ts_tree_root_node(tree_), &source_); }
~ParseTree() {
ts_tree_delete(tree_);
}
~ParseTree() { ts_tree_delete(tree_); }
bool IsProperlyParsed() { // TODO: find place
return !GetRoot().HasError();
}
private:
TSTree* tree_;
TSTree *tree_;
std::string source_; // for token value extraction
};