diff --git a/src/direct/mod.rs b/src/direct/mod.rs index eb7e371..ad6144a 100644 --- a/src/direct/mod.rs +++ b/src/direct/mod.rs @@ -1,6 +1,6 @@ use std::{collections::VecDeque}; -use crate::{token::{Token}, builtin::modules::Module}; +use crate::{token::{Token, MessageType}, builtin::modules::Module}; #[derive(Default)] pub struct LangSpecs { @@ -8,7 +8,7 @@ pub struct LangSpecs { builtin_features: Vec, lang_version: u32, authors: Vec, - embedded_files: Vec, + embedded_files: Vec<(usize, String)>, } impl LangSpecs { @@ -16,24 +16,22 @@ impl LangSpecs { pub fn features(&self) -> &[crate::builtin::modules::Module] { &self.builtin_features } + + pub fn embedded_files(&self) -> &[(usize, String)] { + &self.embedded_files + } } pub fn resolve_directives(tokens: &mut VecDeque) -> LangSpecs { let mut specs = LangSpecs::default(); - for token in tokens.iter() { + for (idx, token) in tokens.iter().enumerate() { match token { - Token::CompilerDirective(text, _) => parse_directive(text, &mut specs), + Token::CompilerDirective(text, _) => parse_directive(text, idx, &mut specs), _ => () } } - // remove compiler directives from source - tokens.retain(|token| match token { - Token::CompilerDirective(_, _) => false, - _ => true - }); - specs } @@ -41,7 +39,7 @@ static DIRECTIVE_REGEX_SRC: &'static str = concat!( r"@feature\(((?:\s*[\w]+\s*,?)*)\)", r"|@version\(\s*([0-9]{3})\s*\)", r"|@author\((.*)\)", - r"|@embed\((.*)\)" + r"|@include\((.*)\)" ); lazy_static::lazy_static! { @@ -58,7 +56,7 @@ pub fn from_list(text: &str) -> Vec { vec } -fn parse_directive(text: &str, specs: &mut LangSpecs) { +fn parse_directive(text: &str, token_idx: usize, specs: &mut LangSpecs) { for cap in DIRECTIVE_REGEX.captures_iter(text) { let mut enumerator = cap.iter().enumerate(); @@ -91,8 +89,13 @@ fn parse_directive(text: &str, specs: &mut LangSpecs) { return; }, 4 => { - specs.embedded_files.append(&mut from_list(mat.as_str())); - crate::message(crate::token::MessageType::Warning, "Embed directive not working at current state"); + for path in from_list(mat.as_str()).iter() { + if let Ok(str) = std::fs::read_to_string(path) { + specs.embedded_files.push((token_idx, str)); + } else { + crate::message(MessageType::Warning, format!("Unable to read embedded file: {path}")); + } + } return; }, _ => crate::message(crate::token::MessageType::Warning, format!("unknown directive: `{}`", text)), diff --git a/src/main.rs b/src/main.rs index 298b3fa..d2f46cf 100644 --- a/src/main.rs +++ b/src/main.rs @@ -36,6 +36,15 @@ fn compile(settings: &Settings) -> Option<(Vec, Vec, Vec (), // valid whitespace + Token::LineBreak(_) | Token::Terminator(_) | Token::CompilerDirective(_,_) => (), // valid whitespace _ => { diagnostics.set_err(&top, crate::msg::ERR22, ""); return Err(()) ; diff --git a/src/token/mod.rs b/src/token/mod.rs index f8b0855..94c87ab 100644 --- a/src/token/mod.rs +++ b/src/token/mod.rs @@ -472,7 +472,7 @@ pub struct DebugInfo { /// index in source string where the token ends in the current line pub end: usize, /// line number where the line in which the token is begins - pub line: usize, + pub line: usize } #[derive(Debug)]