umya_spreadsheet/structs/office2010/excel/
data_validations.rs

1// x14:dataValidations
2use crate::helper::const_str::*;
3use crate::reader::driver::*;
4use crate::structs::office2010::excel::DataValidation;
5use crate::writer::driver::*;
6use quick_xml::events::{BytesStart, Event};
7use quick_xml::Reader;
8use quick_xml::Writer;
9use std::io::Cursor;
10use thin_vec::ThinVec;
11
12#[derive(Default, Debug, Clone)]
13pub struct DataValidations {
14    data_validation_list: ThinVec<DataValidation>,
15}
16
17impl DataValidations {
18    #[inline]
19    pub fn get_data_validation_list(&self) -> &[DataValidation] {
20        &self.data_validation_list
21    }
22
23    #[inline]
24    pub fn get_data_validation_list_mut(&mut self) -> &mut ThinVec<DataValidation> {
25        &mut self.data_validation_list
26    }
27
28    #[inline]
29    pub fn set_data_validation_list(
30        &mut self,
31        value: impl Into<ThinVec<DataValidation>>,
32    ) -> &mut Self {
33        self.data_validation_list = value.into();
34        self
35    }
36
37    #[inline]
38    pub fn add_data_validation_list(&mut self, value: DataValidation) -> &mut Self {
39        self.data_validation_list.push(value);
40        self
41    }
42
43    pub(crate) fn set_attributes<R: std::io::BufRead>(
44        &mut self,
45        reader: &mut Reader<R>,
46        _e: &BytesStart,
47    ) {
48        xml_read_loop!(
49            reader,
50            Event::Empty(ref e) => {
51                if e.name().into_inner() == b"x14:dataValidation" {
52                    let mut obj = DataValidation::default();
53                    obj.set_attributes(reader, e, true);
54                    self.add_data_validation_list(obj);
55                }
56            },
57            Event::Start(ref e) => {
58                if e.name().into_inner() == b"x14:dataValidation" {
59                    let mut obj = DataValidation::default();
60                    obj.set_attributes(reader, e, false);
61                    self.add_data_validation_list(obj);
62                }
63            },
64            Event::End(ref e) => {
65                if e.name().into_inner() == b"x14:dataValidations" {
66                    return
67                }
68            },
69            Event::Eof => panic!("Error: Could not find {} end element", "x14:dataValidations")
70        );
71    }
72
73    pub(crate) fn write_to(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
74        // ext
75        let mut attributes: Vec<(&str, &str)> = Vec::new();
76        attributes.push(("uri", "{CCE6A557-97BC-4b89-ADB6-D9C93CAAB3DF}"));
77        attributes.push(("xmlns:x14", SHEET_MS_MAIN_NS));
78        write_start_tag(writer, "ext", attributes, false);
79
80        // dataValidations
81        let mut attributes: Vec<(&str, &str)> = Vec::new();
82
83        let count = self.data_validation_list.len().to_string();
84        attributes.push(("count", &count));
85        attributes.push(("xmlns:xm", EXCEL_MAIN_NS));
86
87        write_start_tag(writer, "x14:dataValidations", attributes, false);
88
89        for obj in &self.data_validation_list {
90            obj.write_to(writer);
91        }
92
93        write_end_tag(writer, "x14:dataValidations");
94        write_end_tag(writer, "ext");
95    }
96}