use anyhow::bail;
use quick_xml::events::{BytesStart, Event};
use std::io::Read;
use crate::{
excel::XmlReader,
helper::{string_to_bool, string_to_int, string_to_unsignedint},
};
pub type XlsxWorkbookViews = Vec<XlsxWorkbookView>;
pub(crate) fn load_bookviews(
reader: &mut XmlReader<impl Read>,
) -> anyhow::Result<XlsxWorkbookViews> {
let mut views: XlsxWorkbookViews = 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"workbookView" => {
views.push(XlsxWorkbookView::load(reader, e)?);
}
Ok(Event::End(ref e)) if e.local_name().as_ref() == b"bookViews" => break,
Ok(Event::Eof) => bail!("unexpected end of file."),
Err(e) => bail!(e.to_string()),
_ => (),
}
}
Ok(views)
}
#[derive(Debug, Clone, PartialEq)]
pub struct XlsxWorkbookView {
pub active_tab: Option<u64>,
pub auto_filter_date_grouping: Option<bool>,
pub first_sheet: Option<u64>,
pub minimized: Option<bool>,
pub show_horizontal_scroll: Option<bool>,
pub show_sheet_tabs: Option<bool>,
pub show_vertical_scroll: Option<bool>,
pub tab_ratio: Option<u64>,
pub visibility: Option<String>,
pub window_height: Option<u64>,
pub window_width: Option<u64>,
pub x_window: Option<i64>,
pub y_window: Option<i64>,
}
impl XlsxWorkbookView {
pub(crate) fn load(reader: &mut XmlReader<impl Read>, e: &BytesStart) -> anyhow::Result<Self> {
let attributes = e.attributes();
let mut view = Self {
active_tab: None,
auto_filter_date_grouping: None,
first_sheet: None,
minimized: None,
show_horizontal_scroll: None,
show_sheet_tabs: None,
show_vertical_scroll: None,
tab_ratio: None,
visibility: None,
window_height: None,
window_width: None,
x_window: None,
y_window: 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"activeTab" => {
view.active_tab = string_to_unsignedint(&string_value);
}
b"autoFilterDateGrouping" => {
view.auto_filter_date_grouping = string_to_bool(&string_value);
}
b"firstSheet" => {
view.first_sheet = string_to_unsignedint(&string_value);
}
b"minimized" => {
view.minimized = string_to_bool(&string_value);
}
b"showHorizontalScroll" => {
view.show_horizontal_scroll = string_to_bool(&string_value);
}
b"showSheetTabs" => {
view.show_sheet_tabs = string_to_bool(&string_value);
}
b"showVerticalScroll" => {
view.show_vertical_scroll = string_to_bool(&string_value);
}
b"tabRatio" => {
view.tab_ratio = string_to_unsignedint(&string_value);
}
b"visibility" => {
view.visibility = Some(string_value);
}
b"windowHeight" => {
view.window_height = string_to_unsignedint(&string_value);
}
b"windowWidth" => {
view.window_width = string_to_unsignedint(&string_value);
}
b"xWindow" => {
view.x_window = string_to_int(&string_value);
}
b"yWindow" => {
view.y_window = string_to_int(&string_value);
}
_ => {}
}
}
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"extLst" => {
let _ = reader.read_to_end_into(e.to_end().to_owned().name(), &mut Vec::new());
}
Ok(Event::End(ref e)) if e.local_name().as_ref() == b"workbookView" => break,
Ok(Event::Eof) => bail!("unexpected end of file."),
Err(e) => bail!(e.to_string()),
_ => (),
}
}
Ok(view)
}
}