use std::io;
use std::iter::Iterator;
pub mod newc;
pub use newc::Builder as NewcBuilder;
pub use newc::Reader as NewcReader;
pub fn write_cpio<I, RS, W>(inputs: I, output: W) -> io::Result<W>
where
I: Iterator<Item = (NewcBuilder, RS)> + Sized,
RS: io::Read + io::Seek,
W: io::Write,
{
let output = inputs
.enumerate()
.try_fold(output, |output, (idx, (builder, mut input))| {
let len = input.seek(io::SeekFrom::End(0))?;
input.seek(io::SeekFrom::Start(0))?;
let mut fp = builder.ino(idx as u32).write(output, len as u32);
io::copy(&mut input, &mut fp)?;
fp.finish()
})?;
newc::trailer(output)
}
#[cfg(test)]
mod tests {
use super::*;
use std::io::Cursor;
#[test]
fn test_multi_file() {
let mut input = vec![
(
NewcBuilder::new("./hello_world")
.uid(1000)
.gid(1000)
.mode(0o100644),
Cursor::new("Hello, World".to_string()),
),
(
NewcBuilder::new("./hello_world2")
.uid(1000)
.gid(1000)
.mode(0o100644),
Cursor::new("Hello, World 2".to_string()),
),
];
let output = Cursor::new(vec![]);
let _ = write_cpio(input.drain(..), output).unwrap();
}
}