mirror of https://git.sr.ht/~garritfra/sabre
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.9 KiB
87 lines
2.9 KiB
use crate::generator; |
|
use crate::lexer; |
|
use crate::parser; |
|
use crate::Lib; |
|
use crate::PathBuf; |
|
use parser::node_type::Module; |
|
/** |
|
* Copyright 2021 Garrit Franke |
|
* |
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
* you may not use this file except in compliance with the License. |
|
* You may obtain a copy of the License at |
|
* |
|
* https://www.apache.org/licenses/LICENSE-2.0 |
|
* |
|
* Unless required by applicable law or agreed to in writing, software |
|
* distributed under the License is distributed on an "AS IS" BASIS, |
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
* See the License for the specific language governing permissions and |
|
* limitations under the License. |
|
*/ |
|
use std::fs::File; |
|
use std::io::Read; |
|
use std::io::Write; |
|
|
|
pub struct Builder { |
|
in_file: PathBuf, |
|
modules: Vec<Module>, |
|
} |
|
|
|
impl Builder { |
|
pub fn new(entrypoint: PathBuf) -> Self { |
|
Self { |
|
in_file: entrypoint, |
|
modules: Vec::new(), |
|
} |
|
} |
|
|
|
pub fn build(&mut self) -> Result<(), String> { |
|
self.build_module(self.in_file.clone())?; |
|
|
|
// Append standard library |
|
self.modules.push(build_stdlib()); |
|
Ok(()) |
|
} |
|
|
|
fn build_module(&mut self, file_path: PathBuf) -> Result<Module, String> { |
|
let mut file = File::open(&file_path) |
|
.map_err(|_| format!("Could not open file: {}", file_path.display()))?; |
|
let mut contents = String::new(); |
|
|
|
file.read_to_string(&mut contents) |
|
.expect("Could not read file"); |
|
let tokens = lexer::tokenize(&contents); |
|
let module = parser::parse(tokens, Some(contents), file_path.display().to_string())?; |
|
for import in &module.imports { |
|
self.build_module(PathBuf::from(import))?; |
|
} |
|
self.modules.push(module.clone()); |
|
Ok(module) |
|
} |
|
|
|
pub(crate) fn generate(&mut self, out_file: PathBuf) -> Result<(), String> { |
|
let mut mod_iter = self.modules.iter(); |
|
|
|
// TODO: We shouldn't clone here |
|
let mut condensed = mod_iter.next().ok_or("No module specified")?.clone(); |
|
while let Some(module) = mod_iter.next() { |
|
condensed.merge_with(module.clone()); |
|
} |
|
let output = generator::generate(condensed); |
|
let mut file = std::fs::File::create(out_file).expect("create failed"); |
|
file.write_all(output.as_bytes()).expect("write failed"); |
|
Ok(file.flush().expect("Could not flush file")) |
|
} |
|
} |
|
|
|
fn build_stdlib() -> parser::node_type::Module { |
|
let stdlib_raw = |
|
Lib::get("stdio.sb").expect("Standard library not found. This should not occur."); |
|
let stblib_str = |
|
std::str::from_utf8(&stdlib_raw).expect("Could not interpret standard library."); |
|
let stdlib_tokens = lexer::tokenize(&stblib_str); |
|
|
|
parser::parse(stdlib_tokens, Some(stblib_str.into()), "stdio".to_string()) |
|
.expect("Could not parse stdlib") |
|
}
|
|
|