use crate::error::Error;
use crate::private::Marker;
use crate::private::Sealed;
use crate::token::Token;
use crate::ParseBuffer;
use std::collections::HashSet;
use std::sync::Mutex;
pub struct Lookahead<'a> {
stream: ParseBuffer<'a>,
comparisons: Mutex<HashSet<String>>,
}
impl<'a> Lookahead<'a> {
pub(crate) fn new(stream: ParseBuffer<'a>) -> Lookahead<'a> {
Lookahead {
stream,
comparisons: Mutex::new(HashSet::new()),
}
}
pub fn peek<T: Peek>(&self, token: T) -> bool {
if self.stream.peek(token) {
true
} else {
self.comparisons.lock().unwrap().insert(T::Token::display());
false
}
}
pub fn error(self) -> Error {
self.stream
.unexpected_token(self.comparisons.into_inner().unwrap())
}
}
pub trait Peek: Sealed {
#[doc(hidden)]
type Token: Token;
}
impl<F: FnOnce(Marker) -> T, T: Token> Sealed for F {}
impl<F: FnOnce(Marker) -> T, T: Token> Peek for F {
type Token = T;
}