Skip to main content

oak_powershell/builder/
mod.rs

1//! PowerShell AST builder
2
3use crate::{ast::*, language::PowerShellLanguage, parser::PowerShellParser};
4use oak_core::{Builder, BuilderCache, GreenNode, OakDiagnostics, Parser, RedNode, SourceText, TextEdit, source::Source};
5
6/// Builder for the PowerShell language.
7pub struct PowerShellBuilder<'config> {
8    /// The language configuration.
9    pub config: &'config PowerShellLanguage,
10}
11
12impl<'config> PowerShellBuilder<'config> {
13    /// Creates a new `PowerShellBuilder`.
14    pub fn new(config: &'config PowerShellLanguage) -> Self {
15        Self { config }
16    }
17
18    /// Builds the root node of the AST.
19    pub(crate) fn build_root<'a>(&self, green_tree: &'a GreenNode<'a, PowerShellLanguage>, _source: &SourceText) -> Result<PowerShellRoot, oak_core::OakError> {
20        let _red_root = RedNode::new(green_tree, 0);
21        // Minimal implementation
22        Ok(PowerShellRoot { items: Vec::new() })
23    }
24}
25
26impl<'config> Builder<PowerShellLanguage> for PowerShellBuilder<'config> {
27    fn build<'a, S: Source + ?Sized>(&self, source: &S, edits: &[TextEdit], _cache: &'a mut impl BuilderCache<PowerShellLanguage>) -> OakDiagnostics<PowerShellRoot> {
28        let parser = PowerShellParser::new(self.config);
29
30        let mut parse_cache = oak_core::parser::session::ParseSession::<PowerShellLanguage>::default();
31        let parse_result = parser.parse(source, edits, &mut parse_cache);
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}