use erl_pp::{self, MacroDef};
use erl_tokenize::{Lexer, LexicalToken};
pub trait Preprocessor {
fn define_macro(&mut self, name: &str, replacement: Vec<LexicalToken>);
fn undef_macro(&mut self, name: &str);
}
impl<'a> Preprocessor for &'a mut Preprocessor {
fn define_macro(&mut self, name: &str, replacement: Vec<LexicalToken>) {
(*self).define_macro(name, replacement);
}
fn undef_macro(&mut self, name: &str) {
(*self).undef_macro(name);
}
}
impl<'a, T, E> Preprocessor for &'a mut erl_pp::Preprocessor<T, E> {
fn define_macro(&mut self, name: &str, replacement: Vec<LexicalToken>) {
(*self).define_macro(name, replacement);
}
fn undef_macro(&mut self, name: &str) {
(*self).undef_macro(name);
}
}
impl<T, E> Preprocessor for erl_pp::Preprocessor<T, E> {
fn define_macro(&mut self, name: &str, replacement: Vec<LexicalToken>) {
self.macros_mut()
.insert(name.to_string(), MacroDef::Dynamic(replacement));
}
fn undef_macro(&mut self, name: &str) {
self.macros_mut().remove(name);
}
}
impl<T> Preprocessor for Lexer<T> {
fn define_macro(&mut self, _name: &str, _replacement: Vec<LexicalToken>) {}
fn undef_macro(&mut self, _name: &str) {}
}