use std::io::Cursor;
use md5::Digest;
use quick_xml::{
Reader,
Writer,
events::{
BytesStart,
Event,
},
};
use crate::{
reader::driver::{
get_attribute,
set_string_from_xml,
xml_read_loop,
},
structs::{
EnumValue,
SourceValues,
WorksheetSource,
},
writer::driver::{
write_end_tag,
write_start_tag,
},
};
#[derive(Clone, Default, Debug)]
pub struct CacheSource {
r#type: EnumValue<SourceValues>,
worksheet_source: Option<WorksheetSource>,
}
impl CacheSource {
#[inline]
#[must_use]
pub fn get_type(&self) -> &SourceValues {
self.r#type.value()
}
#[inline]
pub fn set_type(&mut self, value: SourceValues) -> &mut Self {
self.r#type.set_value(value);
self
}
#[inline]
#[must_use]
pub fn worksheet_source(&self) -> Option<&WorksheetSource> {
self.worksheet_source.as_ref()
}
#[inline]
#[must_use]
#[deprecated(since = "3.0.0", note = "Use worksheet_source()")]
pub fn get_worksheet_source(&self) -> Option<&WorksheetSource> {
self.worksheet_source()
}
#[inline]
pub fn worksheet_source_mut(&mut self) -> Option<&mut WorksheetSource> {
self.worksheet_source.as_mut()
}
#[inline]
#[deprecated(since = "3.0.0", note = "Use worksheet_source_mut()")]
pub fn get_worksheet_source_mut(&mut self) -> Option<&mut WorksheetSource> {
self.worksheet_source_mut()
}
#[inline]
pub fn set_worksheet_source_mut(&mut self, value: WorksheetSource) -> &mut Self {
self.worksheet_source = Some(value);
self
}
#[must_use]
pub fn new_worksheet(worksheet_source: WorksheetSource) -> Self {
let mut cache_source = Self::default();
cache_source.set_type(SourceValues::Worksheet);
cache_source.set_worksheet_source_mut(worksheet_source);
cache_source
}
#[inline]
pub(crate) fn hash_code(&self) -> String {
format!(
"{:x}",
md5::Md5::digest(format!(
"{}{}",
self.r#type.value_string(),
match &self.worksheet_source {
Some(v) => {
v.hash_code()
}
None => {
"None".into()
}
},
))
)
}
#[inline]
pub(crate) fn set_attributes<R: std::io::BufRead>(
&mut self,
reader: &mut Reader<R>,
e: &BytesStart,
empty_flg: bool,
) {
set_string_from_xml!(self, e, r#type, "type");
if empty_flg {
return;
}
xml_read_loop!(
reader,
Event::Empty(ref e) => {
if e.name().into_inner() == b"worksheetSource" {
let mut obj = WorksheetSource::default();
obj.set_attributes(reader, e);
self.set_worksheet_source_mut(obj);
}
},
Event::End(ref e) => {
if e.name().into_inner() == b"cacheSource" {
return
}
},
Event::Eof => panic!("Error: Could not find {} end element", "cacheSource")
);
}
#[inline]
pub(crate) fn write_to(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
let empty_flg = self.worksheet_source.is_none();
let attributes: crate::structs::AttrCollection =
vec![("type", self.r#type.value_string()).into()];
write_start_tag(writer, "cacheSource", attributes, empty_flg);
if !empty_flg {
if let Some(v) = &self.worksheet_source {
v.write_to(writer);
}
write_end_tag(writer, "cacheSource");
}
}
}