Skip to main content

oak_scala/builder/
mod.rs

1use crate::{ast::ScalaRoot, language::ScalaLanguage, parser::ScalaParser};
2use oak_core::{Builder, BuilderCache, Lexer, OakDiagnostics, Parser, RedNode, TextEdit, source::Source};
3
4/// Scala 语言的 AST 构建器
5#[derive(Clone)]
6pub struct ScalaBuilder<'config> {
7    config: &'config ScalaLanguage,
8}
9
10impl<'config> ScalaBuilder<'config> {
11    pub fn new(config: &'config ScalaLanguage) -> Self {
12        Self { config }
13    }
14}
15
16impl<'config> Builder<ScalaLanguage> for ScalaBuilder<'config> {
17    fn build<'a, S: Source + ?Sized>(&self, source: &'a S, edits: &[TextEdit], cache: &'a mut impl BuilderCache<ScalaLanguage>) -> OakDiagnostics<ScalaRoot> {
18        let parser = ScalaParser::new(self.config);
19        let lexer = crate::lexer::ScalaLexer::new(&self.config);
20
21        lexer.lex(source, edits, cache);
22        let parse_result = parser.parse(source, edits, cache);
23
24        match parse_result.result {
25            Ok(green_tree) => {
26                let ast_root = ScalaRoot::new(RedNode::new(green_tree, 0).span());
27                OakDiagnostics { result: Ok(ast_root), diagnostics: parse_result.diagnostics }
28            }
29            Err(parse_error) => OakDiagnostics { result: Err(parse_error), diagnostics: parse_result.diagnostics },
30        }
31    }
32}