Skip to main content

oak_hlsl/builder/
mod.rs

1use crate::{ast::*, language::HlslLanguage, parser::HlslParser};
2use oak_core::{Builder, BuilderCache, GreenNode, Lexer, OakDiagnostics, OakError, Parser, RedNode, RedTree, SourceText, TextEdit, source::Source};
3
4/// HLSL 语言的 AST 构建器
5pub struct HlslBuilder<'config> {
6    config: &'config HlslLanguage,
7}
8
9impl<'config> HlslBuilder<'config> {
10    pub fn new(config: &'config HlslLanguage) -> Self {
11        Self { config }
12    }
13}
14
15impl<'config> Builder<HlslLanguage> for HlslBuilder<'config> {
16    fn build<'a, S: Source + ?Sized>(&self, text: &S, edits: &[TextEdit], _cache: &'a mut impl BuilderCache<HlslLanguage>) -> oak_core::builder::BuildOutput<HlslLanguage> {
17        let parser = HlslParser::new(self.config);
18        let lexer = crate::lexer::HlslLexer::new(&self.config);
19
20        // For now, we don't have a proper incremental builder implementation here
21        // We just run the lexer and parser
22        let mut lexer_cache = oak_core::parser::session::ParseSession::<HlslLanguage>::default();
23        let _lex_output = lexer.lex(text, edits, &mut lexer_cache);
24
25        let mut parser_cache = oak_core::parser::session::ParseSession::<HlslLanguage>::default();
26        let parse_result = parser.parse(text, edits, &mut parser_cache);
27
28        let OakDiagnostics { result, diagnostics } = parse_result;
29
30        match result {
31            Ok(green_tree) => {
32                let source_text = SourceText::new(text.get_text_in((0..text.length()).into()).into_owned());
33                match self.build_root(green_tree, &source_text) {
34                    Ok(ast_root) => OakDiagnostics { result: Ok(ast_root), diagnostics },
35                    Err(build_error) => {
36                        let mut diagnostics = diagnostics;
37                        diagnostics.push(build_error.clone());
38                        OakDiagnostics { result: Err(build_error), diagnostics }
39                    }
40                }
41            }
42            Err(parse_error) => OakDiagnostics { result: Err(parse_error), diagnostics },
43        }
44    }
45}
46
47impl<'config> HlslBuilder<'config> {
48    fn build_root<'a>(&self, green_tree: &'a GreenNode<'a, HlslLanguage>, _source: &SourceText) -> Result<HlslRoot, OakError> {
49        let red_root = RedNode::<HlslLanguage>::new(green_tree, 0);
50        let declarations = Vec::new();
51
52        for child in red_root.children() {
53            if let RedTree::Node(_n) = child {
54                // HLSL parser logic would go here
55                // For now, just a placeholder
56            }
57        }
58
59        Ok(HlslRoot { declarations })
60    }
61}