Skip to main content

oak_vampire/builder/
mod.rs

1use crate::{ast::*, language::VampireLanguage, parser::VampireParser};
2use oak_core::{Builder, BuilderCache, GreenNode, OakDiagnostics, OakError, Parser, RedNode, TextEdit, source::Source};
3
4/// AST builder for Vampire language
5#[derive(Clone)]
6pub struct VampireBuilder<'config> {
7    config: &'config VampireLanguage,
8}
9
10impl<'config> VampireBuilder<'config> {
11    /// Creates a new VampireBuilder with the given language configuration.
12    pub fn new(config: &'config VampireLanguage) -> Self {
13        Self { config }
14    }
15}
16
17impl<'config> Builder<VampireLanguage> for VampireBuilder<'config> {
18    fn build<'a, S: Source + ?Sized>(&self, text: &'a S, edits: &[TextEdit], cache: &'a mut impl BuilderCache<VampireLanguage>) -> oak_core::builder::BuildOutput<VampireLanguage> {
19        let parser = VampireParser::new(self.config);
20
21        let parse_result = parser.parse(text, edits, cache);
22
23        match parse_result.result {
24            Ok(green_tree) => match self.build_root(green_tree) {
25                Ok(ast_root) => OakDiagnostics { result: Ok(ast_root), diagnostics: parse_result.diagnostics },
26                Err(build_error) => {
27                    let mut diagnostics = parse_result.diagnostics;
28                    diagnostics.push(build_error.clone());
29                    OakDiagnostics { result: Err(build_error), diagnostics }
30                }
31            },
32            Err(parse_error) => OakDiagnostics { result: Err(parse_error), diagnostics: parse_result.diagnostics },
33        }
34    }
35}
36
37impl<'config> VampireBuilder<'config> {
38    pub(crate) fn build_root(&self, green_tree: &GreenNode<VampireLanguage>) -> Result<VampireRoot, OakError> {
39        let red_root = RedNode::new(green_tree, 0);
40        let span = red_root.span();
41        Ok(VampireRoot { span, formulas: Vec::new() })
42    }
43}