use harper_core::parsers::{self, Parser, PlainEnglish};
use harper_core::{Token, TokenKind};
use harper_tree_sitter::TreeSitterMasker;
use tree_sitter::Node;
pub struct InkParser {
inner: parsers::Mask<TreeSitterMasker, PlainEnglish>,
}
impl InkParser {
fn node_condition(n: &Node) -> bool {
matches!(n.kind(), "contentText" | "blockComment" | "lineComment")
}
}
impl Default for InkParser {
fn default() -> Self {
Self {
inner: parsers::Mask::new(
TreeSitterMasker::new(tree_sitter_ink_lbz::LANGUAGE.into(), Self::node_condition),
PlainEnglish,
),
}
}
}
impl Parser for InkParser {
fn parse(&self, source: &[char]) -> Vec<Token> {
let mut tokens = self.inner.parse(source);
for token in &mut tokens {
if let TokenKind::Space(v) = &mut token.kind {
*v = (*v).clamp(0, 1);
}
}
tokens
}
}