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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
use crate::YaSerialize;
use std::io::{Cursor, Write};
use std::str;
use xml::writer::XmlEvent;
use xml::{EmitterConfig, EventWriter};
pub fn to_string<T: YaSerialize>(model: &T) -> Result<String, String> {
let buf = Cursor::new(Vec::new());
let cursor = serialize_with_writer(model, buf, &Config::default())?;
let data = str::from_utf8(cursor.get_ref()).expect("Found invalid UTF-8");
Ok(String::from(data))
}
pub fn to_string_with_config<T: YaSerialize>(model: &T, config: &Config) -> Result<String, String> {
let buf = Cursor::new(Vec::new());
let cursor = serialize_with_writer(model, buf, config)?;
let data = str::from_utf8(cursor.get_ref()).expect("Found invalid UTF-8");
Ok(String::from(data))
}
pub fn serialize_with_writer<W: Write, T: YaSerialize>(
model: &T,
writer: W,
_config: &Config,
) -> Result<W, String> {
let mut serializer = Serializer::new_from_writer(writer);
match model.serialize(&mut serializer) {
Ok(()) => Ok(serializer.into_inner()),
Err(msg) => Err(msg),
}
}
pub fn to_string_content<T: YaSerialize>(model: &T) -> Result<String, String> {
let buf = Cursor::new(Vec::new());
let cursor = serialize_with_writer_content(model, buf)?;
let data = str::from_utf8(cursor.get_ref()).expect("Found invalid UTF-8");
Ok(String::from(data))
}
pub fn serialize_with_writer_content<W: Write, T: YaSerialize>(
model: &T,
writer: W,
) -> Result<W, String> {
let mut serializer = Serializer::new_for_inner(writer);
serializer.set_skip_start_end(true);
match model.serialize(&mut serializer) {
Ok(()) => Ok(serializer.into_inner()),
Err(msg) => Err(msg),
}
}
pub struct Serializer<W: Write> {
writer: EventWriter<W>,
skip_start_end: bool,
start_event_name: Option<String>,
}
impl<'de, W: Write> Serializer<W> {
pub fn new(writer: EventWriter<W>) -> Self {
Serializer {
writer,
skip_start_end: false,
start_event_name: None,
}
}
pub fn new_from_writer(writer: W) -> Self {
let config = EmitterConfig::new().cdata_to_characters(true);
Self::new(EventWriter::new_with_config(writer, config))
}
pub fn new_for_inner(writer: W) -> Self {
let config = EmitterConfig::new().write_document_declaration(false);
Self::new(EventWriter::new_with_config(writer, config))
}
pub fn into_inner(self) -> W {
self.writer.into_inner()
}
pub fn skip_start_end(&self) -> bool {
self.skip_start_end
}
pub fn set_skip_start_end(&mut self, state: bool) {
self.skip_start_end = state;
}
pub fn get_start_event_name(&self) -> Option<String> {
self.start_event_name.clone()
}
pub fn set_start_event_name(&mut self, name: Option<String>) {
self.start_event_name = name;
}
pub fn write<'a, E>(&mut self, event: E) -> xml::writer::Result<()>
where
E: Into<XmlEvent<'a>>,
{
self.writer.write(event)
}
}
pub struct Config {
pub perform_indent: bool,
pub write_document_declaration: bool,
pub indent_string: Option<String>,
}
impl Default for Config {
fn default() -> Self {
Config {
perform_indent: false,
write_document_declaration: true,
indent_string: None,
}
}
}