Skip to main content

oak_typescript/builder/
mod.rs

1use crate::{ast::*, language::TypeScriptLanguage, lexer::TypeScriptLexer, parser::TypeScriptParser};
2use oak_core::{Builder, BuilderCache, Lexer, OakDiagnostics, Parser, SourceText, TextEdit, source::Source};
3
4mod build_class;
5mod build_expression;
6mod build_jsx;
7mod build_misc;
8mod build_root;
9mod build_statement;
10mod build_type;
11
12/// TypeScript 语言的 AST 构建器
13#[derive(Clone)]
14pub struct TypeScriptBuilder<'config> {
15    config: &'config TypeScriptLanguage,
16}
17
18impl<'config> TypeScriptBuilder<'config> {
19    pub fn new(config: &'config TypeScriptLanguage) -> Self {
20        Self { config }
21    }
22}
23
24impl<'config> Builder<TypeScriptLanguage> for TypeScriptBuilder<'config> {
25    fn build<'a, S: Source + ?Sized>(&self, source: &'a S, edits: &[TextEdit], _cache: &'a mut impl BuilderCache<TypeScriptLanguage>) -> OakDiagnostics<TypeScriptRoot> {
26        let parser = TypeScriptParser::new(self.config);
27        let lexer = TypeScriptLexer::new(&self.config);
28
29        let mut session = oak_core::parser::session::ParseSession::<TypeScriptLanguage>::default();
30        lexer.lex(source, edits, &mut session);
31        let parse_result = parser.parse(source, edits, &mut session);
32
33        match parse_result.result {
34            Ok(green_tree) => {
35                let source_text = SourceText::new(source.get_text_in((0..source.length()).into()).into_owned());
36                match self.build_root(green_tree, &source_text) {
37                    Ok(ast_root) => OakDiagnostics { result: Ok(ast_root), diagnostics: parse_result.diagnostics },
38                    Err(build_error) => {
39                        let mut diagnostics = parse_result.diagnostics;
40                        diagnostics.push(build_error.clone());
41                        OakDiagnostics { result: Err(build_error), diagnostics }
42                    }
43                }
44            }
45            Err(parse_error) => OakDiagnostics { result: Err(parse_error), diagnostics: parse_result.diagnostics },
46        }
47    }
48}