use anyhow::bail;
use quick_xml::events::{BytesStart, Event};
use std::io::Read;
use crate::{
excel::XmlReader,
raw::spreadsheet::stylesheet::color::{XlsxBackgroundColor, XlsxColor, XlsxForegroundColor},
};
#[derive(Debug, Clone, PartialEq)]
pub struct XlsxPatternFill {
pub pattern_type: Option<String>,
pub foreground_color: Option<XlsxForegroundColor>,
pub background_color: Option<XlsxBackgroundColor>,
}
impl XlsxPatternFill {
pub(crate) fn load(reader: &mut XmlReader<impl Read>, e: &BytesStart) -> anyhow::Result<Self> {
let attributes = e.attributes();
let mut fill = Self {
pattern_type: None,
foreground_color: None,
background_color: None,
};
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"patternType" => {
fill.pattern_type = Some(string_value);
break;
}
_ => {}
}
}
Err(error) => {
bail!(error.to_string())
}
}
}
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"fgColor" => {
let color = XlsxColor::load(e)?;
fill.foreground_color = Some(color);
}
Ok(Event::Start(ref e)) if e.local_name().as_ref() == b"bgColor" => {
let color = XlsxColor::load(e)?;
fill.background_color = Some(color);
}
Ok(Event::End(ref e)) if e.local_name().as_ref() == b"patternFill" => break,
Ok(Event::Eof) => bail!("unexpected end of file."),
Err(e) => bail!(e.to_string()),
_ => (),
}
}
Ok(fill)
}
}