Skip to main content

oak_haskell/builder/
mod.rs

1use crate::{ast::HaskellRoot, language::HaskellLanguage, parser::HaskellParser};
2use oak_core::{Builder, BuilderCache, GreenNode, OakDiagnostics, OakError, RedNode, Source, SourceText, TextEdit};
3
4/// Haskell 语言的 AST 构建器
5#[derive(Clone)]
6pub struct HaskellBuilder<'config> {
7    config: &'config HaskellLanguage,
8}
9
10impl<'config> HaskellBuilder<'config> {
11    pub fn new(config: &'config HaskellLanguage) -> Self {
12        Self { config }
13    }
14}
15
16impl<'config> Builder<HaskellLanguage> for HaskellBuilder<'config> {
17    fn build<'a, S: Source + ?Sized>(&self, source: &S, edits: &[TextEdit], _cache: &'a mut impl BuilderCache<HaskellLanguage>) -> OakDiagnostics<HaskellRoot> {
18        let parser = HaskellParser::new(self.config);
19        let lexer = crate::lexer::HaskellLexer::new(&self.config);
20
21        let mut cache = oak_core::parser::session::ParseSession::<HaskellLanguage>::default();
22        let parse_result = oak_core::parser::parse(&parser, &lexer, source, edits, &mut cache);
23
24        match parse_result.result {
25            Ok(green_tree) => {
26                let source_text = SourceText::new(source.get_text_in((0..source.length()).into()).into_owned());
27                match self.build_root(green_tree.clone(), &source_text) {
28                    Ok(ast_root) => OakDiagnostics { result: Ok(ast_root), diagnostics: parse_result.diagnostics },
29                    Err(build_error) => {
30                        let mut diagnostics = parse_result.diagnostics;
31                        diagnostics.push(build_error.clone());
32                        OakDiagnostics { result: Err(build_error), diagnostics }
33                    }
34                }
35            }
36            Err(parse_error) => OakDiagnostics { result: Err(parse_error), diagnostics: parse_result.diagnostics },
37        }
38    }
39}
40
41impl<'config> HaskellBuilder<'config> {
42    pub(crate) fn build_root(&self, green_tree: GreenNode<HaskellLanguage>, _source: &SourceText) -> Result<HaskellRoot, OakError> {
43        let _red_root = RedNode::new(&green_tree, 0);
44        let items = Vec::new();
45        // Simplified for now
46        Ok(HaskellRoot { module_name: None, items })
47    }
48}