From 0c1cb981820c55127c3c09d93868814a1df98246 Mon Sep 17 00:00:00 2001 From: Aleksey Kladov Date: Sat, 23 Feb 2019 16:07:29 +0300 Subject: rename --- crates/ra_syntax/src/parsing/text_token_source.rs | 67 +++++++++++++++++++++++ 1 file changed, 67 insertions(+) create mode 100644 crates/ra_syntax/src/parsing/text_token_source.rs (limited to 'crates/ra_syntax/src/parsing/text_token_source.rs') diff --git a/crates/ra_syntax/src/parsing/text_token_source.rs b/crates/ra_syntax/src/parsing/text_token_source.rs new file mode 100644 index 000000000..a6277f66f --- /dev/null +++ b/crates/ra_syntax/src/parsing/text_token_source.rs @@ -0,0 +1,67 @@ +use ra_parser::TokenSource; + +use crate::{ + SyntaxKind, SyntaxKind::EOF, TextRange, TextUnit, + parsing::lexer::Token, +}; + +pub(crate) struct TextTokenSource<'t> { + text: &'t str, + /// start position of each token(expect whitespace and comment) + /// ```non-rust + /// struct Foo; + /// ^------^--- + /// | | ^- + /// 0 7 10 + /// ``` + /// (token, start_offset): `[(struct, 0), (Foo, 7), (;, 10)]` + start_offsets: Vec, + /// non-whitespace/comment tokens + /// ```non-rust + /// struct Foo {} + /// ^^^^^^ ^^^ ^^ + /// ``` + /// tokens: `[struct, Foo, {, }]` + tokens: Vec, +} + +impl<'t> TokenSource for TextTokenSource<'t> { + fn token_kind(&self, pos: usize) -> SyntaxKind { + if !(pos < self.tokens.len()) { + return EOF; + } + self.tokens[pos].kind + } + fn is_token_joint_to_next(&self, pos: usize) -> bool { + if !(pos + 1 < self.tokens.len()) { + return true; + } + self.start_offsets[pos] + self.tokens[pos].len == self.start_offsets[pos + 1] + } + fn is_keyword(&self, pos: usize, kw: &str) -> bool { + if !(pos < self.tokens.len()) { + return false; + } + let range = TextRange::offset_len(self.start_offsets[pos], self.tokens[pos].len); + + self.text[range] == *kw + } +} + +impl<'t> TextTokenSource<'t> { + /// Generate input from tokens(expect comment and whitespace). + pub fn new(text: &'t str, raw_tokens: &'t [Token]) -> TextTokenSource<'t> { + let mut tokens = Vec::new(); + let mut start_offsets = Vec::new(); + let mut len = 0.into(); + for &token in raw_tokens.iter() { + if !token.kind.is_trivia() { + tokens.push(token); + start_offsets.push(len); + } + len += token.len; + } + + TextTokenSource { text, start_offsets, tokens } + } +} -- cgit v1.2.3