1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
use super::TextElement;
use md5::Digest;
use quick_xml::events::{BytesStart, Event};
use quick_xml::Reader;
use quick_xml::Writer;
use std::borrow::Cow;
use std::fmt::Write;
use std::io::Cursor;
use writer::driver::*;

#[derive(Clone, Default, Debug, PartialEq, PartialOrd)]
pub struct RichText {
    rich_text_elements: Vec<TextElement>,
}
impl RichText {
    pub fn get_text(&self) -> Cow<'static, str> {
        let mut text = String::from("");
        for rich_text_elements in &self.rich_text_elements {
            text = format!("{}{}", text, rich_text_elements.get_text());
        }
        text.into()
    }

    pub fn set_text<S: Into<String>>(&mut self, value: S) -> &mut Self {
        self.rich_text_elements.clear();
        let mut text_element = TextElement::default();
        text_element.set_text(value);
        self.add_rich_text_elements(text_element);
        self
    }

    pub fn get_rich_text_elements(&self) -> &Vec<TextElement> {
        &self.rich_text_elements
    }

    pub fn get_rich_text_elements_mut(&mut self) -> &mut Vec<TextElement> {
        &mut self.rich_text_elements
    }

    pub fn set_rich_text_elements(&mut self, value: Vec<TextElement>) -> &mut Self {
        self.rich_text_elements = value;
        self
    }

    pub fn add_rich_text_elements(&mut self, value: TextElement) -> &mut Self {
        self.rich_text_elements.push(value);
        self
    }

    pub(crate) fn get_hash_code(&self) -> String {
        let mut value = String::from("");
        for ele in &self.rich_text_elements {
            write!(value, "{}", ele.get_hash_code().as_str()).unwrap();
        }
        format!("{:x}", md5::Md5::digest(&value))
    }

    pub(crate) fn set_attributes_text<R: std::io::BufRead>(
        &mut self,
        reader: &mut Reader<R>,
        _e: &BytesStart,
    ) {
        let mut buf = Vec::new();
        loop {
            match reader.read_event_into(&mut buf) {
                Ok(Event::Start(ref e)) => match e.name().into_inner() {
                    b"r" => {
                        let mut obj = TextElement::default();
                        obj.set_attributes(reader, e);
                        self.add_rich_text_elements(obj);
                    }
                    _ => (),
                },
                Ok(Event::End(ref e)) => match e.name().into_inner() {
                    b"text" => return,
                    _ => (),
                },
                Ok(Event::Eof) => panic!("Error not find {} end element", "text"),
                Err(e) => panic!("Error at position {}: {:?}", reader.buffer_position(), e),
                _ => (),
            }
            buf.clear();
        }
    }

    pub(crate) fn write_to_none(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
        // none
        self.write_to(writer, "");
    }

    pub(crate) fn write_to_text(&self, writer: &mut Writer<Cursor<Vec<u8>>>) {
        // text
        self.write_to(writer, "text");
    }

    pub(crate) fn write_to(&self, writer: &mut Writer<Cursor<Vec<u8>>>, tag_name: &str) {
        if !tag_name.is_empty() {
            write_start_tag(writer, tag_name, vec![], false);
        }

        // r
        for obj in &self.rich_text_elements {
            obj.write_to(writer);
        }

        if !tag_name.is_empty() {
            write_end_tag(writer, tag_name);
        }
    }
}