use std::io::Read;
use anyhow::bail;
use quick_xml::events::{BytesStart, Event};
use crate::excel::XmlReader;
pub type XlsxTableParts = Vec<XlsxTablePart>;
pub(crate) fn load_table_parts(reader: &mut XmlReader<impl Read>) -> anyhow::Result<XlsxTableParts> {
let mut parts: XlsxTableParts = vec![];
let mut buf = Vec::new();
loop {
buf.clear();
match reader.read_event_into(&mut buf) {
Ok(Event::Start(ref e)) if e.local_name().as_ref() == b"tablePart" => {
parts.push(XlsxTablePart::load(e)?);
}
Ok(Event::End(ref e)) if e.local_name().as_ref() == b"tableParts" => break,
Ok(Event::Eof) => bail!("unexpected end of file."),
Err(e) => bail!(e.to_string()),
_ => (),
}
}
Ok(parts)
}
#[derive(Debug, Clone, PartialEq)]
pub struct XlsxTablePart {
pub id: String,
}
impl XlsxTablePart {
pub(crate) fn load(e: &BytesStart) -> anyhow::Result<Self> {
let attributes = e.attributes();
for a in attributes {
match a {
Ok(a) => {
let string_value = String::from_utf8(a.value.to_vec())?;
match a.key.local_name().as_ref() {
b"id" => return Ok(Self { id: string_value }),
_ => {}
}
}
Err(error) => {
bail!(error.to_string())
}
}
}
bail!("id not specified for the table part.")
}
}