Skip to main content

oak_typescript/parser/
mod.rs

1/// Element types for TypeScript AST.
2pub mod element_type;
3mod parse_class;
4mod parse_expression;
5mod parse_jsx;
6mod parse_misc;
7mod parse_statement;
8
9use crate::{language::TypeScriptLanguage, lexer::TypeScriptLexer};
10
11use oak_core::{
12    GreenNode, TextEdit,
13    parser::{ParseCache, ParseOutput, Parser, ParserState, parse_with_lexer, pratt::Pratt},
14    source::Source,
15};
16
17pub(crate) type State<'a, S> = ParserState<'a, TypeScriptLanguage, S>;
18
19/// Parser for the TypeScript language.
20pub struct TypeScriptParser<'config> {
21    pub(crate) config: &'config TypeScriptLanguage,
22}
23
24impl<'config> TypeScriptParser<'config> {
25    /// Creates a new `TypeScriptParser` with the given language configuration.
26    pub fn new(config: &'config TypeScriptLanguage) -> Self {
27        Self { config }
28    }
29}
30
31impl<'config> Pratt<TypeScriptLanguage> for TypeScriptParser<'config> {
32    fn primary<'a, S: Source + ?Sized>(&self, state: &mut State<'a, S>) -> &'a GreenNode<'a, TypeScriptLanguage> {
33        self.primary(state)
34    }
35
36    fn prefix<'a, S: Source + ?Sized>(&self, state: &mut State<'a, S>) -> &'a GreenNode<'a, TypeScriptLanguage> {
37        self.prefix(state)
38    }
39
40    fn infix<'a, S: Source + ?Sized>(&self, state: &mut State<'a, S>, left: &'a GreenNode<'a, TypeScriptLanguage>, min_precedence: u8) -> Option<&'a GreenNode<'a, TypeScriptLanguage>> {
41        self.infix(state, left, min_precedence)
42    }
43}
44
45impl<'config> Parser<TypeScriptLanguage> for TypeScriptParser<'config> {
46    fn parse<'a, S: Source + ?Sized>(&self, text: &'a S, edits: &[TextEdit], cache: &'a mut impl ParseCache<TypeScriptLanguage>) -> ParseOutput<'a, TypeScriptLanguage> {
47        let lexer = TypeScriptLexer::new(self.config);
48        parse_with_lexer(&lexer, text, edits, cache, |state| {
49            let checkpoint = state.checkpoint();
50            while state.not_at_end() {
51                self.skip_trivia(state);
52                if state.not_at_end() {
53                    self.parse_statement(state).ok();
54                }
55            }
56            Ok(state.finish_at(checkpoint, crate::parser::element_type::TypeScriptElementType::SourceFile))
57        })
58    }
59}