Tree-sitter 高亮

用法

将此crate以及您想要解析的任何语言的特定语言crate添加到您的Cargo.toml

[dependencies]
tree-sitter-highlight = "^0.20"
tree-sitter-javascript = "0.19"

定义您将识别的高亮名称列表

let highlight_names = [
    "attribute",
    "constant",
    "function.builtin",
    "function",
    "keyword",
    "operator",
    "property",
    "punctuation",
    "punctuation.bracket",
    "punctuation.delimiter",
    "string",
    "string.special",
    "tag",
    "type",
    "type.builtin",
    "variable",
    "variable.builtin",
    "variable.parameter",
];

创建一个高亮器。您需要为每个用于语法高亮的线程使用一个

use tree_sitter_highlight::Highlighter;

let mut highlighter = Highlighter::new();

从语言存储库的queries目录中加载一些高亮查询

use tree_sitter_highlight::HighlightConfiguration;

let javascript_language = tree_sitter_javascript::language();

let mut javascript_config = HighlightConfiguration::new(
    javascript_language,
    tree_sitter_javascript::HIGHLIGHT_QUERY,
    tree_sitter_javascript::INJECTION_QUERY,
    tree_sitter_javascript::LOCALS_QUERY,
).unwrap();

配置识别的名称

javascript_config.configure(&highlight_names);

高亮一些代码

use tree_sitter_highlight::HighlightEvent;

let highlights = highlighter.highlight(
    &javascript_config,
    b"const x = new Y();",
    None,
    |_| None
).unwrap();

for event in highlights {
    match event.unwrap() {
        HighlightEvent::Source {start, end} => {
            eprintln!("source: {}-{}", start, end);
        },
        HighlightEvent::HighlightStart(s) => {
            eprintln!("highlight style started: {:?}", s);
        },
        HighlightEvent::HighlightEnd => {
            eprintln!("highlight style ended");
        },
    }
}

highlight的最后一个参数是一个语言注入回调。当Tree-sitter检测到嵌入式文档（例如，HTML中的script标签内的JavaScript代码）时，这允许检索其他语言。