101 lines
4.9 KiB
C
101 lines
4.9 KiB
C
|
#pragma once
|
||
|
|
// SyntaxHighlighterCpp.h helpers for SyntaxHighlighter.
|
||
|
|
|
||
|
|
static void walkCpp(TSNode node, const std::string& source,
|
||
|
|
std::vector<HighlightSpan>& spans) {
|
||
|
|
if (ts_node_is_null(node)) return;
|
||
|
|
std::string type = nodeType(node);
|
||
|
|
std::string text = nodeText(node, source);
|
||
|
|
|
||
|
|
bool descend = true;
|
||
|
|
|
||
|
|
if (type == "comment") {
|
||
|
|
addSpan(spans, node, TokenCategory::Comment);
|
||
|
|
descend = false;
|
||
|
|
} else if (type == "string_literal" || type == "raw_string_literal" ||
|
||
|
|
type == "char_literal" || type == "string_content") {
|
||
|
|
addSpan(spans, node, TokenCategory::String);
|
||
|
|
descend = false;
|
||
|
|
} else if (type == "number_literal") {
|
||
|
|
addSpan(spans, node, TokenCategory::Number);
|
||
|
|
descend = false;
|
||
|
|
} else if (type == "primitive_type" || type == "sized_type_specifier" ||
|
||
|
|
type == "type_identifier") {
|
||
|
|
addSpan(spans, node, TokenCategory::Type);
|
||
|
|
descend = false;
|
||
|
|
} else if (type == "identifier") {
|
||
|
|
TSNode parent = ts_node_parent(node);
|
||
|
|
std::string parentType = ts_node_is_null(parent) ? "" : nodeType(parent);
|
||
|
|
|
||
|
|
if (parentType == "function_declarator") {
|
||
|
|
TSNode declField = ts_node_child_by_field_name(parent, "declarator", 10);
|
||
|
|
if (!ts_node_is_null(declField) &&
|
||
|
|
ts_node_start_byte(declField) == ts_node_start_byte(node)) {
|
||
|
|
addSpan(spans, node, TokenCategory::Function);
|
||
|
|
} else {
|
||
|
|
addSpan(spans, node, TokenCategory::Identifier);
|
||
|
|
}
|
||
|
|
} else if (parentType == "call_expression") {
|
||
|
|
TSNode funcField = ts_node_child_by_field_name(parent, "function", 8);
|
||
|
|
if (!ts_node_is_null(funcField) &&
|
||
|
|
ts_node_start_byte(funcField) == ts_node_start_byte(node)) {
|
||
|
|
addSpan(spans, node, TokenCategory::Function);
|
||
|
|
} else {
|
||
|
|
addSpan(spans, node, TokenCategory::Identifier);
|
||
|
|
}
|
||
|
|
} else if (parentType == "parameter_declaration") {
|
||
|
|
TSNode declField = ts_node_child_by_field_name(parent, "declarator", 10);
|
||
|
|
if (!ts_node_is_null(declField) &&
|
||
|
|
ts_node_start_byte(declField) == ts_node_start_byte(node)) {
|
||
|
|
addSpan(spans, node, TokenCategory::Parameter);
|
||
|
|
} else if (isCppType(text)) {
|
||
|
|
addSpan(spans, node, TokenCategory::Type);
|
||
|
|
} else {
|
||
|
|
addSpan(spans, node, TokenCategory::Identifier);
|
||
|
|
}
|
||
|
|
} else if (isCppType(text)) {
|
||
|
|
addSpan(spans, node, TokenCategory::Type);
|
||
|
|
} else {
|
||
|
|
addSpan(spans, node, TokenCategory::Identifier);
|
||
|
|
}
|
||
|
|
descend = false;
|
||
|
|
} else if (type == "true" || type == "false" || type == "nullptr") {
|
||
|
|
addSpan(spans, node, TokenCategory::Builtin);
|
||
|
|
descend = false;
|
||
|
|
} else if (type == "preproc_include" || type == "preproc_def" ||
|
||
|
|
type == "preproc_ifdef" || type == "preproc_else" ||
|
||
|
|
type == "preproc_endif" || type == "preproc_call") {
|
||
|
|
addSpan(spans, node, TokenCategory::Keyword);
|
||
|
|
descend = false;
|
||
|
|
} else if (type == "system_lib_string") {
|
||
|
|
addSpan(spans, node, TokenCategory::String);
|
||
|
|
descend = false;
|
||
|
|
} else if (!ts_node_is_named(node)) {
|
||
|
|
if (isCppKeyword(text)) {
|
||
|
|
addSpan(spans, node, TokenCategory::Keyword);
|
||
|
|
} else if (text == "(" || text == ")" || text == "[" || text == "]" ||
|
||
|
|
text == "{" || text == "}" || text == ";" || text == "," ||
|
||
|
|
text == "." || text == "::" || text == ":" || text == "->") {
|
||
|
|
addSpan(spans, node, TokenCategory::Punctuation);
|
||
|
|
} else if (text == "+" || text == "-" || text == "*" || text == "/" ||
|
||
|
|
text == "%" || text == "=" || text == "==" || text == "!=" ||
|
||
|
|
text == "<" || text == ">" || text == "<=" || text == ">=" ||
|
||
|
|
text == "+=" || text == "-=" || text == "*=" || text == "/=" ||
|
||
|
|
text == "&&" || text == "||" || text == "!" || text == "&" ||
|
||
|
|
text == "|" || text == "^" || text == "~" || text == "<<" ||
|
||
|
|
text == ">>" || text == "++" || text == "--") {
|
||
|
|
addSpan(spans, node, TokenCategory::Operator);
|
||
|
|
} else if (isCppType(text)) {
|
||
|
|
addSpan(spans, node, TokenCategory::Type);
|
||
|
|
}
|
||
|
|
descend = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (descend) {
|
||
|
|
uint32_t count = ts_node_child_count(node);
|
||
|
|
for (uint32_t i = 0; i < count; ++i) {
|
||
|
|
walkCpp(ts_node_child(node, i), source, spans);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|