use super::DataValidation;
use crate::reader::driver::*;
use crate::writer::driver::*;
use quick_xml::events::{BytesStart, Event};
use quick_xml::Reader;
use quick_xml::Writer;
use std::io::Cursor;
use thin_vec::ThinVec;
#[derive(Default, Debug, Clone)]
pub struct DataValidations {
data_validation_list: ThinVec<DataValidation>,
}
impl DataValidations {
#[inline]
pub fn get_data_validation_list(&self) -> &[DataValidation] {
&self.data_validation_list
}
#[inline]
pub fn get_data_validation_list_mut(&mut self) -> &mut ThinVec<DataValidation> {
&mut self.data_validation_list
}
#[inline]
pub fn set_data_validation_list(
&mut self,
value: impl Into<ThinVec<DataValidation>>,
) -> &mut Self {
self.data_validation_list = value.into();
self
}
#[inline]
pub fn add_data_validation_list(&mut self, value: DataValidation) -> &mut Self {
self.data_validation_list.push(value);
self
}
pub(crate) fn set_attributes<R: std::io::BufRead>(
&mut self,
reader: &mut Reader<R>,
_e: &BytesStart,
) {
xml_read_loop!(
reader,
Event::Empty(ref e) => {
if e.name().into_inner() == b"dataValidation" {
let mut obj = DataValidation::default();
obj.set_attributes(reader, e, true);
self.add_data_validation_list(obj);
}
},
Event::Start(ref e) => {
if e.name().into_inner() == b"dataValidation" {
let mut obj = DataValidation::default();
obj.set_attributes(reader, e, false);
self.add_data_validation_list(obj);
}
},
Event::End(ref e) => {
if e.name().into_inner() == b"dataValidations" {
return
}
},
Event::Eof => panic!("Error: Could not find {} end element", "dataValidations")
);
}
pub(crate) fn write_to(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
let mut attributes: Vec<(&str, &str)> = Vec::new();
let count = self.data_validation_list.len().to_string();
attributes.push(("count", &count));
write_start_tag(writer, "dataValidations", attributes, false);
for obj in &self.data_validation_list {
obj.write_to(writer);
}
write_end_tag(writer, "dataValidations");
}
}