A peekable lexer

Use the chars() iterator on &str, with the Peekable wrapper on Iterators, to iterate the input, rather than needing to hold the whole input and do iteration by indexes.
This commit is contained in:
Eryn Wells 2017-04-30 17:46:42 -07:00
parent a4282e7760
commit c5b769ff45
3 changed files with 93 additions and 1 deletions

View file

@ -4,4 +4,3 @@ version = "0.1.0"
authors = ["Eryn Wells <eryn@erynwells.me>"]
[dependencies]
sibiltypes = { path = "../types" }

View file

@ -0,0 +1,79 @@
/* lexer/src/lib.rs
* Eryn Wells <eryn@erynwells.me>
*/
use std::iter::Peekable;
#[derive(Debug)]
pub enum Token { LeftParen, RightParen, Id(String), }
enum Resume { Here, AtNext }
enum IterationResult {
Continue,
Emit(Token, Resume),
Error(String),
}
pub struct Lexer<T> where T: Iterator<Item=char> {
input: Peekable<T>,
}
impl<T> Lexer<T> where T: Iterator<Item=char> {
pub fn new(input: T) -> Lexer<T> {
Lexer { input: input.peekable() }
}
fn emit(&self, token: Token, resume: Resume) -> IterationResult {
IterationResult::Emit(token, resume)
}
fn fail(&self, msg: String) -> IterationResult {
IterationResult::Error(msg)
}
}
impl<T> Iterator for Lexer<T> where T: Iterator<Item=char> {
type Item = Result<Token, String>;
fn next(&mut self) -> Option<Self::Item> {
let mut buffer = String::new();
while let Some(peek) = self.input.peek().map(char::clone) {
let result = if buffer.is_empty() {
match peek {
'(' => self.emit(Token::LeftParen, Resume::AtNext),
')' => self.emit(Token::RightParen, Resume::AtNext),
c if c.is_whitespace() => IterationResult::Continue,
c if c.is_alphabetic() => {
buffer.push(c);
IterationResult::Continue
},
c => self.fail(format!("Invalid character: {}", c)),
}
}
else {
match peek {
c if c.is_alphabetic() => {
buffer.push(c);
IterationResult::Continue
}
c if c == '(' || c == ')' || c.is_whitespace() =>
self.emit(Token::Id(buffer.clone()), Resume::Here),
c => self.fail(format!("Invalid character: {}", c)),
}
};
match result {
IterationResult::Continue => self.input.next(),
IterationResult::Emit(token, resume) => {
match resume {
Resume::AtNext => self.input.next(),
Resume::Here => None,
};
return Some(Ok(token))
},
IterationResult::Error(msg) => return Some(Err(msg)),
};
}
None
}
}

14
lexer/src/main.rs Normal file
View file

@ -0,0 +1,14 @@
/* lexer/src/main.rs
* Eryn Wells <eryn@erynwells.me>
*/
extern crate sibillexer;
use sibillexer::Lexer;
fn main() {
let lexer = Lexer::new("(ab (cd) ef)".chars());
for tok in lexer {
println!("found {:?}", tok.unwrap());
}
}