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
125
126
127
128
129
130
131
132
133
134
135
136
use std::io::{Cursor, Read};
use quick_xml::events::Event;
use zip::ZipArchive;
use crate::{error::ParsingError, parse_text};
use super::sheets;
pub fn parse_pptx(content: impl AsRef<[u8]>) -> Result<String, ParsingError> {
let mut archive = ZipArchive::new(Cursor::new(content))
.map_err(|e| ParsingError::ArchiveError(e.to_string()))?;
let mut markdown = String::new();
let mut slide_num = 1;
for i in 0..archive.len() {
let mut file = archive
.by_index(i)
.map_err(|e| ParsingError::ArchiveError(e.to_string()))?;
let file_name = file.name().to_string();
if file_name.starts_with("ppt/slides/") && file_name.ends_with(".xml") {
markdown.push_str(&format!("\n\n## Slide number {}\n", slide_num));
slide_num += 1;
let mut content = String::new();
file.read_to_string(&mut content)?;
let mut reader = quick_xml::Reader::from_str(&content);
let mut buf = Vec::new();
let mut table_rows: Vec<Vec<String>> = Vec::new();
let mut current_row: Vec<String> = Vec::new();
let mut cell_text = String::new();
let mut in_text_body = false;
let mut in_title = false;
let mut in_table = false;
let mut in_row = false;
let mut in_cell = false;
loop {
match reader.read_event_into(&mut buf) {
Ok(Event::Start(ref e)) => match e.name().as_ref() {
b"p:txBody" => {
in_text_body = true;
}
b"p:title" => {
in_title = true;
}
b"a:tbl" => {
in_table = true;
table_rows.clear();
}
b"a:tr" => {
if in_table {
in_row = true;
current_row = Vec::new();
}
}
b"a:br" => {
if in_text_body {
markdown.push_str(" \n");
}
}
b"a:tc" => {
if in_row {
in_cell = true;
cell_text.clear();
}
}
_ => {}
},
Ok(Event::Text(e)) => {
let text = parse_text(&*e)?;
if in_text_body && !text.trim().is_empty() {
if in_title {
markdown.push_str(&format!("### {}", text.trim()));
} else {
markdown.push_str(&format!("{} ", text.trim()));
}
}
if in_cell {
cell_text.push_str(&text);
}
}
Ok(Event::End(ref e)) => match e.name().as_ref() {
b"p:txBody" => {
in_text_body = false;
markdown.push_str(" \n");
}
b"p:title" => {
in_title = false;
markdown.push_str(" \n");
}
b"a:tbl" => {
in_table = false;
if !table_rows.is_empty() {
let headers = table_rows[0].clone();
let data_rows = if table_rows.len() > 1 {
table_rows[1..].to_vec()
} else {
Vec::new()
};
markdown.push_str(&sheets::to_markdown_table(&headers, &data_rows));
markdown.push('\n');
}
}
b"a:tr" => {
in_row = false;
if !current_row.is_empty() {
table_rows.push(current_row.clone());
}
}
b"a:tc" => {
in_cell = false;
if in_row {
current_row.push(cell_text.trim().to_string());
}
}
_ => {}
},
Ok(Event::Eof) => break,
Err(e) => {
return Err(ParsingError::ParsingError(format!(
"Error at position {}: {:?}",
reader.buffer_position(),
e
)));
}
_ => {}
}
buf.clear();
}
}
}
Ok(markdown.trim().to_string())
}