use std::{
collections::VecDeque,
io::{self, Write}, str::FromStr,
};
use cid::Cid;
use crate::{error::CarError, reader::CarReader, unixfs::UnixFs, Ipld};
pub fn ipld_write(
reader: &mut impl CarReader,
cid: Cid,
output: &mut impl Write,
) -> Result<(), CarError>
{
let mut vecq = VecDeque::new();
vecq.push_back(cid);
ipld_write_inner(reader, &mut vecq, output)
}
fn ipld_write_inner(
reader: &mut impl CarReader,
vecq: &mut VecDeque<Cid>,
output: &mut impl Write,
) -> Result<(), CarError> {
while let Some(file_cid) = vecq.pop_front() {
let file_ipld: Ipld = reader.ipld(&file_cid).unwrap();
match file_ipld {
Ipld::Bytes(b) => {
output.write_all(&b[..])?;
}
m @ Ipld::Map(_) => {
let unix_fs: Result<UnixFs, CarError> = (file_cid, m).try_into();
let ufs = unix_fs?;
for link in ufs.links().iter() {
vecq.push_back(link.hash);
}
}
_ => {}
};
}
Ok(())
}
#[inline(always)]
pub fn cat_ipld_str(reader: &mut impl CarReader, cid: &str) -> Result<(), CarError> {
let cid = Cid::from_str(cid).map_err(|e| CarError::Parsing(e.to_string()))?;
cat_ipld(reader, cid)
}
pub fn cat_ipld(reader: &mut impl CarReader, file_cid: Cid) -> Result<(), CarError> {
let mut stdout = io::stdout();
let mut vecq = VecDeque::new();
vecq.push_back(file_cid);
ipld_write_inner(reader, &mut vecq, &mut stdout)
}