36 lines
863 B
Rust
36 lines
863 B
Rust
use {
|
|
crate::{Error, Node},
|
|
std::{
|
|
io::{ErrorKind, Read},
|
|
iter::Iterator,
|
|
},
|
|
};
|
|
|
|
/// An iterator over a series of archive `Node`'s. This struct is generic over any
|
|
/// type which implements `Read`, such as a file or a network stream.
|
|
#[derive(Debug)]
|
|
pub struct Stream<T: Read> {
|
|
reader: T,
|
|
}
|
|
|
|
impl<T: Read> From<T> for Stream<T> {
|
|
fn from(value: T) -> Self {
|
|
Self { reader: value }
|
|
}
|
|
}
|
|
|
|
impl<T: Read> Iterator for Stream<T> {
|
|
type Item = Result<Node, Error>;
|
|
|
|
fn next(&mut self) -> Option<Self::Item> {
|
|
match Node::read(&mut self.reader) {
|
|
Err(Error::Io(e)) if e.kind() == ErrorKind::UnexpectedEof => None,
|
|
Ok(f) => match f.filetype {
|
|
crate::FileType::Eof => None,
|
|
_ => Some(Ok(f)),
|
|
},
|
|
x => Some(x),
|
|
}
|
|
}
|
|
}
|