1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
// Copyright (c) 2017 Fabian Schuiki

use std;
use grind::Grinder;

pub struct Peekable<T: Grinder> {
	inner: T,
	peek: T::Item,
}

impl<T> Peekable<T> where T: Grinder {
	pub fn new(mut inner: T) -> Peekable<T> {
		let peek = inner.next();
		Peekable { inner: inner, peek: peek }
	}

	pub fn peek(&self) -> &T::Item {
		&self.peek
	}
}

impl<T> Grinder for Peekable<T> where T: Grinder {
	type Item = T::Item;
	type Error = T::Error;

	fn emit(&mut self, err: Self::Error) {
		self.inner.emit(err)
	}

	fn next(&mut self) -> Self::Item {
		let mut n = self.inner.next();
		std::mem::swap(&mut self.peek, &mut n);
		n
	}
}

impl<T> From<T> for Peekable<T> where T: Grinder {
	fn from(inner: T) -> Peekable<T> {
		Peekable::new(inner)
	}
}