kreuzberg 4.0.0.pre.rc.6 → 4.0.0.rc1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/.gitignore +0 -6
- data/.rubocop.yaml +534 -1
- data/Gemfile +2 -1
- data/Gemfile.lock +11 -11
- data/README.md +5 -10
- data/examples/async_patterns.rb +0 -1
- data/ext/kreuzberg_rb/extconf.rb +0 -10
- data/ext/kreuzberg_rb/native/Cargo.toml +15 -23
- data/ext/kreuzberg_rb/native/build.rs +2 -0
- data/ext/kreuzberg_rb/native/include/ieeefp.h +1 -1
- data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +1 -1
- data/ext/kreuzberg_rb/native/include/strings.h +2 -2
- data/ext/kreuzberg_rb/native/include/unistd.h +1 -1
- data/ext/kreuzberg_rb/native/src/lib.rs +16 -75
- data/kreuzberg.gemspec +14 -57
- data/lib/kreuzberg/cache_api.rb +0 -1
- data/lib/kreuzberg/cli.rb +2 -2
- data/lib/kreuzberg/config.rb +2 -9
- data/lib/kreuzberg/errors.rb +7 -75
- data/lib/kreuzberg/extraction_api.rb +0 -1
- data/lib/kreuzberg/setup_lib_path.rb +0 -1
- data/lib/kreuzberg/version.rb +1 -1
- data/lib/kreuzberg.rb +0 -21
- data/pkg/kreuzberg-4.0.0.rc1.gem +0 -0
- data/sig/kreuzberg.rbs +3 -55
- data/spec/binding/cli_proxy_spec.rb +4 -2
- data/spec/binding/cli_spec.rb +11 -12
- data/spec/examples.txt +104 -0
- data/spec/fixtures/config.yaml +1 -0
- data/spec/spec_helper.rb +1 -1
- data/vendor/kreuzberg/Cargo.toml +42 -112
- data/vendor/kreuzberg/README.md +2 -2
- data/vendor/kreuzberg/build.rs +4 -18
- data/vendor/kreuzberg/src/bin/profile_extract.rs +455 -0
- data/vendor/kreuzberg/src/cache/mod.rs +3 -27
- data/vendor/kreuzberg/src/core/batch_mode.rs +0 -60
- data/vendor/kreuzberg/src/core/extractor.rs +81 -202
- data/vendor/kreuzberg/src/core/io.rs +2 -4
- data/vendor/kreuzberg/src/core/mime.rs +12 -2
- data/vendor/kreuzberg/src/core/mod.rs +1 -4
- data/vendor/kreuzberg/src/core/pipeline.rs +33 -111
- data/vendor/kreuzberg/src/embeddings.rs +16 -125
- data/vendor/kreuzberg/src/error.rs +1 -1
- data/vendor/kreuzberg/src/extraction/docx.rs +1 -1
- data/vendor/kreuzberg/src/extraction/image.rs +13 -13
- data/vendor/kreuzberg/src/extraction/libreoffice.rs +1 -0
- data/vendor/kreuzberg/src/extraction/mod.rs +5 -9
- data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +0 -2
- data/vendor/kreuzberg/src/extraction/pandoc/batch.rs +275 -0
- data/vendor/kreuzberg/src/extraction/pandoc/mime_types.rs +178 -0
- data/vendor/kreuzberg/src/extraction/pandoc/mod.rs +491 -0
- data/vendor/kreuzberg/src/extraction/pandoc/server.rs +496 -0
- data/vendor/kreuzberg/src/extraction/pandoc/subprocess.rs +1188 -0
- data/vendor/kreuzberg/src/extraction/pandoc/version.rs +162 -0
- data/vendor/kreuzberg/src/extractors/archive.rs +0 -21
- data/vendor/kreuzberg/src/extractors/docx.rs +128 -16
- data/vendor/kreuzberg/src/extractors/email.rs +0 -14
- data/vendor/kreuzberg/src/extractors/excel.rs +20 -19
- data/vendor/kreuzberg/src/extractors/html.rs +154 -137
- data/vendor/kreuzberg/src/extractors/image.rs +4 -7
- data/vendor/kreuzberg/src/extractors/mod.rs +9 -106
- data/vendor/kreuzberg/src/extractors/pandoc.rs +201 -0
- data/vendor/kreuzberg/src/extractors/pdf.rs +15 -12
- data/vendor/kreuzberg/src/extractors/pptx.rs +3 -17
- data/vendor/kreuzberg/src/extractors/structured.rs +0 -14
- data/vendor/kreuzberg/src/extractors/text.rs +5 -23
- data/vendor/kreuzberg/src/extractors/xml.rs +0 -7
- data/vendor/kreuzberg/src/keywords/rake.rs +1 -0
- data/vendor/kreuzberg/src/lib.rs +1 -4
- data/vendor/kreuzberg/src/mcp/mod.rs +1 -1
- data/vendor/kreuzberg/src/mcp/server.rs +3 -5
- data/vendor/kreuzberg/src/ocr/processor.rs +2 -18
- data/vendor/kreuzberg/src/pdf/error.rs +1 -1
- data/vendor/kreuzberg/src/pdf/table.rs +44 -17
- data/vendor/kreuzberg/src/pdf/text.rs +3 -0
- data/vendor/kreuzberg/src/plugins/extractor.rs +5 -8
- data/vendor/kreuzberg/src/plugins/ocr.rs +11 -2
- data/vendor/kreuzberg/src/plugins/processor.rs +1 -2
- data/vendor/kreuzberg/src/plugins/registry.rs +0 -13
- data/vendor/kreuzberg/src/plugins/validator.rs +8 -9
- data/vendor/kreuzberg/src/stopwords/mod.rs +2 -2
- data/vendor/kreuzberg/src/types.rs +12 -42
- data/vendor/kreuzberg/tests/batch_orchestration.rs +5 -19
- data/vendor/kreuzberg/tests/batch_processing.rs +3 -15
- data/vendor/kreuzberg/tests/chunking_offset_demo.rs +92 -0
- data/vendor/kreuzberg/tests/concurrency_stress.rs +1 -17
- data/vendor/kreuzberg/tests/config_features.rs +0 -18
- data/vendor/kreuzberg/tests/config_loading_tests.rs +39 -15
- data/vendor/kreuzberg/tests/core_integration.rs +7 -24
- data/vendor/kreuzberg/tests/csv_integration.rs +81 -71
- data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +25 -23
- data/vendor/kreuzberg/tests/pandoc_integration.rs +503 -0
- data/vendor/kreuzberg/tests/pipeline_integration.rs +1 -0
- data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +1 -0
- data/vendor/kreuzberg/tests/registry_integration_tests.rs +22 -1
- data/vendor/kreuzberg/tests/security_validation.rs +1 -12
- metadata +25 -90
- data/.rubocop.yml +0 -538
- data/ext/kreuzberg_rb/native/Cargo.lock +0 -6535
- data/lib/kreuzberg/error_context.rb +0 -32
- data/vendor/kreuzberg/benches/otel_overhead.rs +0 -48
- data/vendor/kreuzberg/src/extraction/markdown.rs +0 -213
- data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +0 -287
- data/vendor/kreuzberg/src/extractors/bibtex.rs +0 -469
- data/vendor/kreuzberg/src/extractors/docbook.rs +0 -502
- data/vendor/kreuzberg/src/extractors/epub.rs +0 -707
- data/vendor/kreuzberg/src/extractors/fictionbook.rs +0 -491
- data/vendor/kreuzberg/src/extractors/fictionbook.rs.backup2 +0 -738
- data/vendor/kreuzberg/src/extractors/jats.rs +0 -1051
- data/vendor/kreuzberg/src/extractors/jupyter.rs +0 -367
- data/vendor/kreuzberg/src/extractors/latex.rs +0 -652
- data/vendor/kreuzberg/src/extractors/markdown.rs +0 -700
- data/vendor/kreuzberg/src/extractors/odt.rs +0 -628
- data/vendor/kreuzberg/src/extractors/opml.rs +0 -634
- data/vendor/kreuzberg/src/extractors/orgmode.rs +0 -528
- data/vendor/kreuzberg/src/extractors/rst.rs +0 -576
- data/vendor/kreuzberg/src/extractors/rtf.rs +0 -810
- data/vendor/kreuzberg/src/extractors/security.rs +0 -484
- data/vendor/kreuzberg/src/extractors/security_tests.rs +0 -367
- data/vendor/kreuzberg/src/extractors/typst.rs +0 -650
- data/vendor/kreuzberg/src/panic_context.rs +0 -154
- data/vendor/kreuzberg/tests/api_extract_multipart.rs +0 -52
- data/vendor/kreuzberg/tests/bibtex_parity_test.rs +0 -421
- data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +0 -498
- data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +0 -370
- data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +0 -275
- data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +0 -228
- data/vendor/kreuzberg/tests/html_table_test.rs +0 -551
- data/vendor/kreuzberg/tests/instrumentation_test.rs +0 -139
- data/vendor/kreuzberg/tests/jats_extractor_tests.rs +0 -639
- data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +0 -704
- data/vendor/kreuzberg/tests/latex_extractor_tests.rs +0 -496
- data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +0 -490
- data/vendor/kreuzberg/tests/odt_extractor_tests.rs +0 -695
- data/vendor/kreuzberg/tests/opml_extractor_tests.rs +0 -616
- data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +0 -822
- data/vendor/kreuzberg/tests/rst_extractor_tests.rs +0 -692
- data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +0 -776
- data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +0 -1259
- data/vendor/kreuzberg/tests/typst_extractor_tests.rs +0 -647
- data/vendor/rb-sys/.cargo-ok +0 -1
- data/vendor/rb-sys/.cargo_vcs_info.json +0 -6
- data/vendor/rb-sys/Cargo.lock +0 -393
- data/vendor/rb-sys/Cargo.toml +0 -70
- data/vendor/rb-sys/Cargo.toml.orig +0 -57
- data/vendor/rb-sys/LICENSE-APACHE +0 -190
- data/vendor/rb-sys/LICENSE-MIT +0 -21
- data/vendor/rb-sys/bin/release.sh +0 -21
- data/vendor/rb-sys/build/features.rs +0 -108
- data/vendor/rb-sys/build/main.rs +0 -246
- data/vendor/rb-sys/build/stable_api_config.rs +0 -153
- data/vendor/rb-sys/build/version.rs +0 -48
- data/vendor/rb-sys/readme.md +0 -36
- data/vendor/rb-sys/src/bindings.rs +0 -21
- data/vendor/rb-sys/src/hidden.rs +0 -11
- data/vendor/rb-sys/src/lib.rs +0 -34
- data/vendor/rb-sys/src/macros.rs +0 -371
- data/vendor/rb-sys/src/memory.rs +0 -53
- data/vendor/rb-sys/src/ruby_abi_version.rs +0 -38
- data/vendor/rb-sys/src/special_consts.rs +0 -31
- data/vendor/rb-sys/src/stable_api/compiled.c +0 -179
- data/vendor/rb-sys/src/stable_api/compiled.rs +0 -257
- data/vendor/rb-sys/src/stable_api/ruby_2_6.rs +0 -316
- data/vendor/rb-sys/src/stable_api/ruby_2_7.rs +0 -316
- data/vendor/rb-sys/src/stable_api/ruby_3_0.rs +0 -324
- data/vendor/rb-sys/src/stable_api/ruby_3_1.rs +0 -317
- data/vendor/rb-sys/src/stable_api/ruby_3_2.rs +0 -315
- data/vendor/rb-sys/src/stable_api/ruby_3_3.rs +0 -326
- data/vendor/rb-sys/src/stable_api/ruby_3_4.rs +0 -327
- data/vendor/rb-sys/src/stable_api.rs +0 -261
- data/vendor/rb-sys/src/symbol.rs +0 -31
- data/vendor/rb-sys/src/tracking_allocator.rs +0 -332
- data/vendor/rb-sys/src/utils.rs +0 -89
- data/vendor/rb-sys/src/value_type.rs +0 -7
|
@@ -1,810 +0,0 @@
|
|
|
1
|
-
//! RTF (Rich Text Format) extractor.
|
|
2
|
-
//!
|
|
3
|
-
//! Supports: Rich Text Format (.rtf)
|
|
4
|
-
//!
|
|
5
|
-
//! This native Rust extractor provides text extraction from RTF documents with:
|
|
6
|
-
//! - Character encoding support (Windows-1252 for 0x80-0x9F range)
|
|
7
|
-
//! - Common RTF control words (paragraph breaks, tabs, bullets, quotes, dashes)
|
|
8
|
-
//! - Unicode escape sequences
|
|
9
|
-
//! - Image metadata extraction
|
|
10
|
-
//! - Whitespace normalization
|
|
11
|
-
|
|
12
|
-
use crate::Result;
|
|
13
|
-
use crate::core::config::ExtractionConfig;
|
|
14
|
-
use crate::extraction::cells_to_markdown;
|
|
15
|
-
use crate::plugins::{DocumentExtractor, Plugin};
|
|
16
|
-
use crate::types::{ExtractionResult, Metadata, Table};
|
|
17
|
-
use async_trait::async_trait;
|
|
18
|
-
use serde_json::Value;
|
|
19
|
-
use std::collections::HashMap;
|
|
20
|
-
|
|
21
|
-
/// Native Rust RTF extractor.
|
|
22
|
-
///
|
|
23
|
-
/// Extracts text content, metadata, and structure from RTF documents
|
|
24
|
-
pub struct RtfExtractor;
|
|
25
|
-
|
|
26
|
-
impl RtfExtractor {
|
|
27
|
-
/// Create a new RTF extractor.
|
|
28
|
-
pub fn new() -> Self {
|
|
29
|
-
Self
|
|
30
|
-
}
|
|
31
|
-
}
|
|
32
|
-
|
|
33
|
-
impl Default for RtfExtractor {
|
|
34
|
-
fn default() -> Self {
|
|
35
|
-
Self::new()
|
|
36
|
-
}
|
|
37
|
-
}
|
|
38
|
-
|
|
39
|
-
impl Plugin for RtfExtractor {
|
|
40
|
-
fn name(&self) -> &str {
|
|
41
|
-
"rtf-extractor"
|
|
42
|
-
}
|
|
43
|
-
|
|
44
|
-
fn version(&self) -> String {
|
|
45
|
-
env!("CARGO_PKG_VERSION").to_string()
|
|
46
|
-
}
|
|
47
|
-
|
|
48
|
-
fn initialize(&self) -> Result<()> {
|
|
49
|
-
Ok(())
|
|
50
|
-
}
|
|
51
|
-
|
|
52
|
-
fn shutdown(&self) -> Result<()> {
|
|
53
|
-
Ok(())
|
|
54
|
-
}
|
|
55
|
-
|
|
56
|
-
fn description(&self) -> &str {
|
|
57
|
-
"Extracts content from RTF (Rich Text Format) files with native Rust parsing"
|
|
58
|
-
}
|
|
59
|
-
|
|
60
|
-
fn author(&self) -> &str {
|
|
61
|
-
"Kreuzberg Team"
|
|
62
|
-
}
|
|
63
|
-
}
|
|
64
|
-
|
|
65
|
-
/// Convert a hex digit character to its numeric value.
|
|
66
|
-
///
|
|
67
|
-
/// Returns None if the character is not a valid hex digit.
|
|
68
|
-
#[inline]
|
|
69
|
-
fn hex_digit_to_u8(c: char) -> Option<u8> {
|
|
70
|
-
match c {
|
|
71
|
-
'0'..='9' => Some((c as u8) - b'0'),
|
|
72
|
-
'a'..='f' => Some((c as u8) - b'a' + 10),
|
|
73
|
-
'A'..='F' => Some((c as u8) - b'A' + 10),
|
|
74
|
-
_ => None,
|
|
75
|
-
}
|
|
76
|
-
}
|
|
77
|
-
|
|
78
|
-
/// Parse a hex-encoded byte from two characters.
|
|
79
|
-
///
|
|
80
|
-
/// Returns the decoded byte if both characters are valid hex digits.
|
|
81
|
-
#[inline]
|
|
82
|
-
fn parse_hex_byte(h1: char, h2: char) -> Option<u8> {
|
|
83
|
-
let high = hex_digit_to_u8(h1)?;
|
|
84
|
-
let low = hex_digit_to_u8(h2)?;
|
|
85
|
-
Some((high << 4) | low)
|
|
86
|
-
}
|
|
87
|
-
|
|
88
|
-
/// Parse an RTF control word and extract its value.
|
|
89
|
-
///
|
|
90
|
-
/// Returns a tuple of (control_word, optional_numeric_value)
|
|
91
|
-
fn parse_rtf_control_word(chars: &mut std::iter::Peekable<std::str::Chars>) -> (String, Option<i32>) {
|
|
92
|
-
let mut word = String::new();
|
|
93
|
-
let mut num_str = String::new();
|
|
94
|
-
let mut is_negative = false;
|
|
95
|
-
|
|
96
|
-
while let Some(&c) = chars.peek() {
|
|
97
|
-
if c.is_alphabetic() {
|
|
98
|
-
word.push(c);
|
|
99
|
-
chars.next();
|
|
100
|
-
} else {
|
|
101
|
-
break;
|
|
102
|
-
}
|
|
103
|
-
}
|
|
104
|
-
|
|
105
|
-
if let Some(&c) = chars.peek()
|
|
106
|
-
&& c == '-'
|
|
107
|
-
{
|
|
108
|
-
is_negative = true;
|
|
109
|
-
chars.next();
|
|
110
|
-
}
|
|
111
|
-
|
|
112
|
-
while let Some(&c) = chars.peek() {
|
|
113
|
-
if c.is_ascii_digit() {
|
|
114
|
-
num_str.push(c);
|
|
115
|
-
chars.next();
|
|
116
|
-
} else {
|
|
117
|
-
break;
|
|
118
|
-
}
|
|
119
|
-
}
|
|
120
|
-
|
|
121
|
-
let num_value = if !num_str.is_empty() {
|
|
122
|
-
let val = num_str.parse::<i32>().unwrap_or(0);
|
|
123
|
-
Some(if is_negative { -val } else { val })
|
|
124
|
-
} else {
|
|
125
|
-
None
|
|
126
|
-
};
|
|
127
|
-
|
|
128
|
-
(word, num_value)
|
|
129
|
-
}
|
|
130
|
-
|
|
131
|
-
/// Extract text and image metadata from RTF document.
|
|
132
|
-
///
|
|
133
|
-
/// This function extracts plain text from an RTF document by:
|
|
134
|
-
/// 1. Tokenizing control sequences and text
|
|
135
|
-
/// 2. Converting encoded characters to Unicode
|
|
136
|
-
/// 3. Extracting text while skipping formatting groups
|
|
137
|
-
/// 4. Detecting and extracting image metadata (\pict sections)
|
|
138
|
-
/// 5. Normalizing whitespace
|
|
139
|
-
fn extract_text_from_rtf(content: &str) -> (String, Vec<Table>) {
|
|
140
|
-
struct TableState {
|
|
141
|
-
rows: Vec<Vec<String>>,
|
|
142
|
-
current_row: Vec<String>,
|
|
143
|
-
current_cell: String,
|
|
144
|
-
in_row: bool,
|
|
145
|
-
}
|
|
146
|
-
|
|
147
|
-
fn push_cell(state: &mut TableState) {
|
|
148
|
-
let cell = state.current_cell.trim().to_string();
|
|
149
|
-
state.current_row.push(cell);
|
|
150
|
-
state.current_cell.clear();
|
|
151
|
-
}
|
|
152
|
-
|
|
153
|
-
fn push_row(state: &mut TableState) {
|
|
154
|
-
if state.in_row || !state.current_cell.is_empty() {
|
|
155
|
-
push_cell(state);
|
|
156
|
-
state.in_row = false;
|
|
157
|
-
}
|
|
158
|
-
if !state.current_row.is_empty() {
|
|
159
|
-
state.rows.push(state.current_row.clone());
|
|
160
|
-
state.current_row.clear();
|
|
161
|
-
}
|
|
162
|
-
}
|
|
163
|
-
|
|
164
|
-
fn finalize_table(state_opt: &mut Option<TableState>, tables: &mut Vec<Table>) {
|
|
165
|
-
if let Some(mut state) = state_opt.take() {
|
|
166
|
-
if state.in_row || !state.current_cell.is_empty() || !state.current_row.is_empty() {
|
|
167
|
-
push_row(&mut state);
|
|
168
|
-
}
|
|
169
|
-
if !state.rows.is_empty() {
|
|
170
|
-
let markdown = cells_to_markdown(&state.rows);
|
|
171
|
-
tables.push(Table {
|
|
172
|
-
cells: state.rows,
|
|
173
|
-
markdown,
|
|
174
|
-
page_number: 1,
|
|
175
|
-
});
|
|
176
|
-
}
|
|
177
|
-
}
|
|
178
|
-
}
|
|
179
|
-
|
|
180
|
-
let mut result = String::new();
|
|
181
|
-
let mut chars = content.chars().peekable();
|
|
182
|
-
let mut tables: Vec<Table> = Vec::new();
|
|
183
|
-
let mut table_state: Option<TableState> = None;
|
|
184
|
-
|
|
185
|
-
let ensure_table = |table_state: &mut Option<TableState>| {
|
|
186
|
-
if table_state.is_none() {
|
|
187
|
-
*table_state = Some(TableState {
|
|
188
|
-
rows: Vec::new(),
|
|
189
|
-
current_row: Vec::new(),
|
|
190
|
-
current_cell: String::new(),
|
|
191
|
-
in_row: false,
|
|
192
|
-
});
|
|
193
|
-
}
|
|
194
|
-
};
|
|
195
|
-
|
|
196
|
-
while let Some(ch) = chars.next() {
|
|
197
|
-
match ch {
|
|
198
|
-
'\\' => {
|
|
199
|
-
if let Some(&next_ch) = chars.peek() {
|
|
200
|
-
match next_ch {
|
|
201
|
-
'\\' | '{' | '}' => {
|
|
202
|
-
chars.next();
|
|
203
|
-
result.push(next_ch);
|
|
204
|
-
}
|
|
205
|
-
'\'' => {
|
|
206
|
-
chars.next();
|
|
207
|
-
let hex1 = chars.next();
|
|
208
|
-
let hex2 = chars.next();
|
|
209
|
-
if let (Some(h1), Some(h2)) = (hex1, hex2)
|
|
210
|
-
&& let Some(byte) = parse_hex_byte(h1, h2)
|
|
211
|
-
{
|
|
212
|
-
let decoded = match byte {
|
|
213
|
-
0x80 => '\u{20AC}',
|
|
214
|
-
0x81 => '?',
|
|
215
|
-
0x82 => '\u{201A}',
|
|
216
|
-
0x83 => '\u{0192}',
|
|
217
|
-
0x84 => '\u{201E}',
|
|
218
|
-
0x85 => '\u{2026}',
|
|
219
|
-
0x86 => '\u{2020}',
|
|
220
|
-
0x87 => '\u{2021}',
|
|
221
|
-
0x88 => '\u{02C6}',
|
|
222
|
-
0x89 => '\u{2030}',
|
|
223
|
-
0x8A => '\u{0160}',
|
|
224
|
-
0x8B => '\u{2039}',
|
|
225
|
-
0x8C => '\u{0152}',
|
|
226
|
-
0x8D => '?',
|
|
227
|
-
0x8E => '\u{017D}',
|
|
228
|
-
0x8F => '?',
|
|
229
|
-
0x90 => '?',
|
|
230
|
-
0x91 => '\u{2018}',
|
|
231
|
-
0x92 => '\u{2019}',
|
|
232
|
-
0x93 => '\u{201C}',
|
|
233
|
-
0x94 => '\u{201D}',
|
|
234
|
-
0x95 => '\u{2022}',
|
|
235
|
-
0x96 => '\u{2013}',
|
|
236
|
-
0x97 => '\u{2014}',
|
|
237
|
-
0x98 => '\u{02DC}',
|
|
238
|
-
0x99 => '\u{2122}',
|
|
239
|
-
0x9A => '\u{0161}',
|
|
240
|
-
0x9B => '\u{203A}',
|
|
241
|
-
0x9C => '\u{0153}',
|
|
242
|
-
0x9D => '?',
|
|
243
|
-
0x9E => '\u{017E}',
|
|
244
|
-
0x9F => '\u{0178}',
|
|
245
|
-
_ => byte as char,
|
|
246
|
-
};
|
|
247
|
-
result.push(decoded);
|
|
248
|
-
if let Some(state) = table_state.as_mut()
|
|
249
|
-
&& state.in_row
|
|
250
|
-
{
|
|
251
|
-
state.current_cell.push(decoded);
|
|
252
|
-
}
|
|
253
|
-
}
|
|
254
|
-
}
|
|
255
|
-
'u' => {
|
|
256
|
-
chars.next();
|
|
257
|
-
let mut num_str = String::new();
|
|
258
|
-
while let Some(&c) = chars.peek() {
|
|
259
|
-
if c.is_ascii_digit() || c == '-' {
|
|
260
|
-
num_str.push(c);
|
|
261
|
-
chars.next();
|
|
262
|
-
} else {
|
|
263
|
-
break;
|
|
264
|
-
}
|
|
265
|
-
}
|
|
266
|
-
if let Ok(code_num) = num_str.parse::<i32>() {
|
|
267
|
-
let code_u = if code_num < 0 {
|
|
268
|
-
(code_num + 65536) as u32
|
|
269
|
-
} else {
|
|
270
|
-
code_num as u32
|
|
271
|
-
};
|
|
272
|
-
if let Some(c) = char::from_u32(code_u) {
|
|
273
|
-
result.push(c);
|
|
274
|
-
if let Some(state) = table_state.as_mut()
|
|
275
|
-
&& state.in_row
|
|
276
|
-
{
|
|
277
|
-
state.current_cell.push(c);
|
|
278
|
-
}
|
|
279
|
-
}
|
|
280
|
-
}
|
|
281
|
-
}
|
|
282
|
-
_ => {
|
|
283
|
-
let (control_word, _) = parse_rtf_control_word(&mut chars);
|
|
284
|
-
|
|
285
|
-
match control_word.as_str() {
|
|
286
|
-
"pict" => {
|
|
287
|
-
let image_metadata = extract_image_metadata(&mut chars);
|
|
288
|
-
if !image_metadata.is_empty() {
|
|
289
|
-
result.push('!');
|
|
290
|
-
result.push('[');
|
|
291
|
-
result.push_str("image");
|
|
292
|
-
result.push(']');
|
|
293
|
-
result.push('(');
|
|
294
|
-
result.push_str(&image_metadata);
|
|
295
|
-
result.push(')');
|
|
296
|
-
result.push(' ');
|
|
297
|
-
if let Some(state) = table_state.as_mut()
|
|
298
|
-
&& state.in_row
|
|
299
|
-
{
|
|
300
|
-
state.current_cell.push('!');
|
|
301
|
-
state.current_cell.push('[');
|
|
302
|
-
state.current_cell.push_str("image");
|
|
303
|
-
state.current_cell.push(']');
|
|
304
|
-
state.current_cell.push('(');
|
|
305
|
-
state.current_cell.push_str(&image_metadata);
|
|
306
|
-
state.current_cell.push(')');
|
|
307
|
-
state.current_cell.push(' ');
|
|
308
|
-
}
|
|
309
|
-
}
|
|
310
|
-
}
|
|
311
|
-
"par" => {
|
|
312
|
-
if table_state.is_some() {
|
|
313
|
-
finalize_table(&mut table_state, &mut tables);
|
|
314
|
-
}
|
|
315
|
-
if !result.is_empty() && !result.ends_with('\n') {
|
|
316
|
-
result.push('\n');
|
|
317
|
-
result.push('\n');
|
|
318
|
-
}
|
|
319
|
-
}
|
|
320
|
-
"tab" => {
|
|
321
|
-
result.push('\t');
|
|
322
|
-
if let Some(state) = table_state.as_mut()
|
|
323
|
-
&& state.in_row
|
|
324
|
-
{
|
|
325
|
-
state.current_cell.push('\t');
|
|
326
|
-
}
|
|
327
|
-
}
|
|
328
|
-
"bullet" => {
|
|
329
|
-
result.push('•');
|
|
330
|
-
}
|
|
331
|
-
"lquote" => {
|
|
332
|
-
result.push('\u{2018}');
|
|
333
|
-
}
|
|
334
|
-
"rquote" => {
|
|
335
|
-
result.push('\u{2019}');
|
|
336
|
-
}
|
|
337
|
-
"ldblquote" => {
|
|
338
|
-
result.push('\u{201C}');
|
|
339
|
-
}
|
|
340
|
-
"rdblquote" => {
|
|
341
|
-
result.push('\u{201D}');
|
|
342
|
-
}
|
|
343
|
-
"endash" => {
|
|
344
|
-
result.push('\u{2013}');
|
|
345
|
-
}
|
|
346
|
-
"emdash" => {
|
|
347
|
-
result.push('\u{2014}');
|
|
348
|
-
}
|
|
349
|
-
"trowd" => {
|
|
350
|
-
ensure_table(&mut table_state);
|
|
351
|
-
if let Some(state) = table_state.as_mut() {
|
|
352
|
-
if state.in_row {
|
|
353
|
-
push_row(state);
|
|
354
|
-
}
|
|
355
|
-
state.in_row = true;
|
|
356
|
-
state.current_cell.clear();
|
|
357
|
-
state.current_row.clear();
|
|
358
|
-
}
|
|
359
|
-
if !result.is_empty() && !result.ends_with('\n') {
|
|
360
|
-
result.push('\n');
|
|
361
|
-
}
|
|
362
|
-
if !result.ends_with('|') {
|
|
363
|
-
result.push('|');
|
|
364
|
-
result.push(' ');
|
|
365
|
-
}
|
|
366
|
-
}
|
|
367
|
-
"cell" => {
|
|
368
|
-
if !result.ends_with('|') {
|
|
369
|
-
if !result.ends_with(' ') && !result.is_empty() {
|
|
370
|
-
result.push(' ');
|
|
371
|
-
}
|
|
372
|
-
result.push('|');
|
|
373
|
-
}
|
|
374
|
-
if !result.ends_with(' ') {
|
|
375
|
-
result.push(' ');
|
|
376
|
-
}
|
|
377
|
-
}
|
|
378
|
-
"row" => {
|
|
379
|
-
ensure_table(&mut table_state);
|
|
380
|
-
if let Some(state) = table_state.as_mut()
|
|
381
|
-
&& (state.in_row || !state.current_cell.is_empty())
|
|
382
|
-
{
|
|
383
|
-
push_row(state);
|
|
384
|
-
}
|
|
385
|
-
if !result.ends_with('|') {
|
|
386
|
-
result.push('|');
|
|
387
|
-
}
|
|
388
|
-
if !result.ends_with('\n') {
|
|
389
|
-
result.push('\n');
|
|
390
|
-
}
|
|
391
|
-
if let Some(state) = table_state.as_ref()
|
|
392
|
-
&& !state.in_row
|
|
393
|
-
&& !state.rows.is_empty()
|
|
394
|
-
{
|
|
395
|
-
// We'll finalize once we see content outside the table
|
|
396
|
-
}
|
|
397
|
-
}
|
|
398
|
-
_ => {}
|
|
399
|
-
}
|
|
400
|
-
}
|
|
401
|
-
}
|
|
402
|
-
}
|
|
403
|
-
}
|
|
404
|
-
'{' | '}' => {
|
|
405
|
-
if !result.is_empty() && !result.ends_with(' ') {
|
|
406
|
-
result.push(' ');
|
|
407
|
-
}
|
|
408
|
-
}
|
|
409
|
-
' ' | '\t' | '\n' | '\r' => {
|
|
410
|
-
if !result.is_empty() && !result.ends_with(' ') {
|
|
411
|
-
result.push(' ');
|
|
412
|
-
}
|
|
413
|
-
if let Some(state) = table_state.as_mut()
|
|
414
|
-
&& state.in_row
|
|
415
|
-
&& !state.current_cell.ends_with(' ')
|
|
416
|
-
{
|
|
417
|
-
state.current_cell.push(' ');
|
|
418
|
-
}
|
|
419
|
-
}
|
|
420
|
-
_ => {
|
|
421
|
-
if let Some(state) = table_state.as_ref()
|
|
422
|
-
&& !state.in_row
|
|
423
|
-
&& !state.rows.is_empty()
|
|
424
|
-
{
|
|
425
|
-
finalize_table(&mut table_state, &mut tables);
|
|
426
|
-
}
|
|
427
|
-
result.push(ch);
|
|
428
|
-
if let Some(state) = table_state.as_mut()
|
|
429
|
-
&& state.in_row
|
|
430
|
-
{
|
|
431
|
-
state.current_cell.push(ch);
|
|
432
|
-
}
|
|
433
|
-
}
|
|
434
|
-
}
|
|
435
|
-
}
|
|
436
|
-
|
|
437
|
-
if table_state.is_some() {
|
|
438
|
-
finalize_table(&mut table_state, &mut tables);
|
|
439
|
-
}
|
|
440
|
-
|
|
441
|
-
(normalize_whitespace(&result), tables)
|
|
442
|
-
}
|
|
443
|
-
|
|
444
|
-
/// Normalize whitespace in a string using a single-pass algorithm.
|
|
445
|
-
///
|
|
446
|
-
/// Collapses multiple consecutive whitespace characters into single spaces
|
|
447
|
-
/// and trims leading/trailing whitespace.
|
|
448
|
-
fn normalize_whitespace(s: &str) -> String {
|
|
449
|
-
let mut result = String::with_capacity(s.len());
|
|
450
|
-
let mut last_was_space = false;
|
|
451
|
-
|
|
452
|
-
for ch in s.chars() {
|
|
453
|
-
if ch.is_whitespace() {
|
|
454
|
-
if !last_was_space {
|
|
455
|
-
result.push(' ');
|
|
456
|
-
last_was_space = true;
|
|
457
|
-
}
|
|
458
|
-
} else {
|
|
459
|
-
result.push(ch);
|
|
460
|
-
last_was_space = false;
|
|
461
|
-
}
|
|
462
|
-
}
|
|
463
|
-
|
|
464
|
-
result.trim().to_string()
|
|
465
|
-
}
|
|
466
|
-
|
|
467
|
-
/// Parse a `{\\creatim ...}` or `{\\revtim ...}` RTF info block into ISO 8601 format.
|
|
468
|
-
fn parse_rtf_datetime(segment: &str) -> Option<String> {
|
|
469
|
-
let mut year: Option<i32> = None;
|
|
470
|
-
let mut month: Option<i32> = None;
|
|
471
|
-
let mut day: Option<i32> = None;
|
|
472
|
-
let mut hour: Option<i32> = None;
|
|
473
|
-
let mut minute: Option<i32> = None;
|
|
474
|
-
|
|
475
|
-
let mut chars = segment.chars().peekable();
|
|
476
|
-
while let Some(&ch) = chars.peek() {
|
|
477
|
-
if ch != '\\' {
|
|
478
|
-
chars.next();
|
|
479
|
-
continue;
|
|
480
|
-
}
|
|
481
|
-
chars.next();
|
|
482
|
-
let (word, value) = parse_rtf_control_word(&mut chars);
|
|
483
|
-
if let Some(v) = value {
|
|
484
|
-
match word.as_str() {
|
|
485
|
-
"yr" => year = Some(v),
|
|
486
|
-
"mo" => month = Some(v),
|
|
487
|
-
"dy" => day = Some(v),
|
|
488
|
-
"hr" => hour = Some(v),
|
|
489
|
-
"min" => minute = Some(v),
|
|
490
|
-
_ => {}
|
|
491
|
-
}
|
|
492
|
-
}
|
|
493
|
-
}
|
|
494
|
-
|
|
495
|
-
let year = year?;
|
|
496
|
-
let month = month.unwrap_or(1).max(1) as u32;
|
|
497
|
-
let day = day.unwrap_or(1).max(1) as u32;
|
|
498
|
-
let hour = hour.unwrap_or(0).max(0) as u32;
|
|
499
|
-
let minute = minute.unwrap_or(0).max(0) as u32;
|
|
500
|
-
|
|
501
|
-
Some(format!(
|
|
502
|
-
"{:04}-{:02}-{:02}T{:02}:{:02}:00Z",
|
|
503
|
-
year, month, day, hour, minute
|
|
504
|
-
))
|
|
505
|
-
}
|
|
506
|
-
|
|
507
|
-
/// Extract metadata from the RTF `\\info` block and augment with computed statistics.
|
|
508
|
-
fn extract_rtf_metadata(rtf_content: &str, extracted_text: &str) -> HashMap<String, Value> {
|
|
509
|
-
let mut metadata: HashMap<String, Value> = HashMap::new();
|
|
510
|
-
|
|
511
|
-
if let Some(start) = rtf_content.find("{\\info") {
|
|
512
|
-
let slice = &rtf_content[start..];
|
|
513
|
-
let mut depth = 0usize;
|
|
514
|
-
let mut end_offset: Option<usize> = None;
|
|
515
|
-
|
|
516
|
-
for (idx, ch) in slice.char_indices() {
|
|
517
|
-
match ch {
|
|
518
|
-
'{' => depth += 1,
|
|
519
|
-
'}' => {
|
|
520
|
-
if depth == 0 {
|
|
521
|
-
break;
|
|
522
|
-
}
|
|
523
|
-
depth -= 1;
|
|
524
|
-
if depth == 0 {
|
|
525
|
-
end_offset = Some(idx + 1);
|
|
526
|
-
break;
|
|
527
|
-
}
|
|
528
|
-
}
|
|
529
|
-
_ => {}
|
|
530
|
-
}
|
|
531
|
-
}
|
|
532
|
-
|
|
533
|
-
let info_block = end_offset.map(|end| &slice[..end]).unwrap_or(slice);
|
|
534
|
-
|
|
535
|
-
let mut segments: Vec<String> = Vec::new();
|
|
536
|
-
let mut seg_depth = 0usize;
|
|
537
|
-
let mut current = String::new();
|
|
538
|
-
let mut in_segment = false;
|
|
539
|
-
|
|
540
|
-
for ch in info_block.chars() {
|
|
541
|
-
if ch == '{' {
|
|
542
|
-
seg_depth += 1;
|
|
543
|
-
if seg_depth == 2 {
|
|
544
|
-
in_segment = true;
|
|
545
|
-
current.clear();
|
|
546
|
-
continue;
|
|
547
|
-
}
|
|
548
|
-
} else if ch == '}' {
|
|
549
|
-
if seg_depth == 2 && in_segment {
|
|
550
|
-
segments.push(current.clone());
|
|
551
|
-
in_segment = false;
|
|
552
|
-
}
|
|
553
|
-
seg_depth = seg_depth.saturating_sub(1);
|
|
554
|
-
continue;
|
|
555
|
-
}
|
|
556
|
-
|
|
557
|
-
if in_segment {
|
|
558
|
-
current.push(ch);
|
|
559
|
-
}
|
|
560
|
-
}
|
|
561
|
-
|
|
562
|
-
for segment in segments {
|
|
563
|
-
if !segment.starts_with('\\') {
|
|
564
|
-
continue;
|
|
565
|
-
}
|
|
566
|
-
|
|
567
|
-
let cleaned_segment = if segment.starts_with("\\*\\") {
|
|
568
|
-
segment.replacen("\\*\\", "\\", 1)
|
|
569
|
-
} else {
|
|
570
|
-
segment.clone()
|
|
571
|
-
};
|
|
572
|
-
|
|
573
|
-
let mut chars = cleaned_segment.chars().peekable();
|
|
574
|
-
chars.next(); // consume the leading backslash
|
|
575
|
-
let (keyword, numeric) = parse_rtf_control_word(&mut chars);
|
|
576
|
-
let remaining: String = chars.collect();
|
|
577
|
-
let trimmed = remaining.trim();
|
|
578
|
-
|
|
579
|
-
match keyword.as_str() {
|
|
580
|
-
"author" => {
|
|
581
|
-
if !trimmed.is_empty() {
|
|
582
|
-
let author = trimmed.to_string();
|
|
583
|
-
metadata.insert("created_by".to_string(), Value::String(author.clone()));
|
|
584
|
-
metadata.insert("authors".to_string(), Value::Array(vec![Value::String(author)]));
|
|
585
|
-
}
|
|
586
|
-
}
|
|
587
|
-
"operator" => {
|
|
588
|
-
if !trimmed.is_empty() {
|
|
589
|
-
metadata.insert("modified_by".to_string(), Value::String(trimmed.to_string()));
|
|
590
|
-
}
|
|
591
|
-
}
|
|
592
|
-
"title" => {
|
|
593
|
-
if !trimmed.is_empty() {
|
|
594
|
-
metadata.insert("title".to_string(), Value::String(trimmed.to_string()));
|
|
595
|
-
}
|
|
596
|
-
}
|
|
597
|
-
"subject" => {
|
|
598
|
-
if !trimmed.is_empty() {
|
|
599
|
-
metadata.insert("subject".to_string(), Value::String(trimmed.to_string()));
|
|
600
|
-
}
|
|
601
|
-
}
|
|
602
|
-
"generator" => {
|
|
603
|
-
if !trimmed.is_empty() {
|
|
604
|
-
metadata.insert("generator".to_string(), Value::String(trimmed.to_string()));
|
|
605
|
-
}
|
|
606
|
-
}
|
|
607
|
-
"creatim" => {
|
|
608
|
-
if let Some(dt) = parse_rtf_datetime(trimmed) {
|
|
609
|
-
metadata.insert("created_at".to_string(), Value::String(dt));
|
|
610
|
-
}
|
|
611
|
-
}
|
|
612
|
-
"revtim" => {
|
|
613
|
-
if let Some(dt) = parse_rtf_datetime(trimmed) {
|
|
614
|
-
metadata.insert("modified_at".to_string(), Value::String(dt));
|
|
615
|
-
}
|
|
616
|
-
}
|
|
617
|
-
"version" => {
|
|
618
|
-
if let Some(val) = numeric.or_else(|| trimmed.parse::<i32>().ok()) {
|
|
619
|
-
metadata.insert("revision".to_string(), Value::String(val.to_string()));
|
|
620
|
-
}
|
|
621
|
-
}
|
|
622
|
-
"nofpages" => {
|
|
623
|
-
if let Some(val) = numeric.or_else(|| trimmed.parse::<i32>().ok()) {
|
|
624
|
-
metadata.insert("page_count".to_string(), Value::Number(val.into()));
|
|
625
|
-
}
|
|
626
|
-
}
|
|
627
|
-
"nofwords" => {
|
|
628
|
-
if let Some(val) = numeric.or_else(|| trimmed.parse::<i32>().ok()) {
|
|
629
|
-
metadata.insert("word_count".to_string(), Value::Number(val.into()));
|
|
630
|
-
}
|
|
631
|
-
}
|
|
632
|
-
"nofchars" => {
|
|
633
|
-
if let Some(val) = numeric.or_else(|| trimmed.parse::<i32>().ok()) {
|
|
634
|
-
metadata.insert("character_count".to_string(), Value::Number(val.into()));
|
|
635
|
-
}
|
|
636
|
-
}
|
|
637
|
-
"lines" => {
|
|
638
|
-
if let Some(val) = numeric.or_else(|| trimmed.parse::<i32>().ok()) {
|
|
639
|
-
metadata.insert("line_count".to_string(), Value::Number(val.into()));
|
|
640
|
-
}
|
|
641
|
-
}
|
|
642
|
-
"paragraphs" => {
|
|
643
|
-
if let Some(val) = numeric.or_else(|| trimmed.parse::<i32>().ok()) {
|
|
644
|
-
metadata.insert("paragraph_count".to_string(), Value::Number(val.into()));
|
|
645
|
-
}
|
|
646
|
-
}
|
|
647
|
-
_ => {}
|
|
648
|
-
}
|
|
649
|
-
}
|
|
650
|
-
}
|
|
651
|
-
|
|
652
|
-
let cleaned_text = extracted_text.trim();
|
|
653
|
-
if !cleaned_text.is_empty() {
|
|
654
|
-
let word_count = cleaned_text.split_whitespace().count() as i64;
|
|
655
|
-
metadata
|
|
656
|
-
.entry("word_count".to_string())
|
|
657
|
-
.or_insert(Value::Number(word_count.into()));
|
|
658
|
-
|
|
659
|
-
let character_count = cleaned_text.chars().count() as i64;
|
|
660
|
-
metadata
|
|
661
|
-
.entry("character_count".to_string())
|
|
662
|
-
.or_insert(Value::Number(character_count.into()));
|
|
663
|
-
|
|
664
|
-
let line_count = cleaned_text.lines().count() as i64;
|
|
665
|
-
metadata
|
|
666
|
-
.entry("line_count".to_string())
|
|
667
|
-
.or_insert(Value::Number(line_count.into()));
|
|
668
|
-
|
|
669
|
-
let paragraph_count = cleaned_text.split("\n\n").filter(|p| !p.trim().is_empty()).count() as i64;
|
|
670
|
-
metadata
|
|
671
|
-
.entry("paragraph_count".to_string())
|
|
672
|
-
.or_insert(Value::Number(paragraph_count.into()));
|
|
673
|
-
}
|
|
674
|
-
|
|
675
|
-
metadata
|
|
676
|
-
}
|
|
677
|
-
|
|
678
|
-
/// Extract image metadata from within a \pict group.
|
|
679
|
-
///
|
|
680
|
-
/// Looks for image type (jpegblip, pngblip, etc.) and dimensions.
|
|
681
|
-
fn extract_image_metadata(chars: &mut std::iter::Peekable<std::str::Chars>) -> String {
|
|
682
|
-
let mut metadata = String::new();
|
|
683
|
-
let mut image_type: Option<&str> = None;
|
|
684
|
-
let mut width_goal: Option<i32> = None;
|
|
685
|
-
let mut height_goal: Option<i32> = None;
|
|
686
|
-
let mut depth = 0;
|
|
687
|
-
|
|
688
|
-
while let Some(&ch) = chars.peek() {
|
|
689
|
-
match ch {
|
|
690
|
-
'{' => {
|
|
691
|
-
depth += 1;
|
|
692
|
-
chars.next();
|
|
693
|
-
}
|
|
694
|
-
'}' => {
|
|
695
|
-
if depth == 0 {
|
|
696
|
-
break;
|
|
697
|
-
}
|
|
698
|
-
depth -= 1;
|
|
699
|
-
chars.next();
|
|
700
|
-
}
|
|
701
|
-
'\\' => {
|
|
702
|
-
chars.next();
|
|
703
|
-
let (control_word, value) = parse_rtf_control_word(chars);
|
|
704
|
-
|
|
705
|
-
match control_word.as_str() {
|
|
706
|
-
"jpegblip" => image_type = Some("jpg"),
|
|
707
|
-
"pngblip" => image_type = Some("png"),
|
|
708
|
-
"wmetafile" => image_type = Some("wmf"),
|
|
709
|
-
"dibitmap" => image_type = Some("bmp"),
|
|
710
|
-
"picwgoal" => width_goal = value,
|
|
711
|
-
"pichgoal" => height_goal = value,
|
|
712
|
-
"bin" => break,
|
|
713
|
-
_ => {}
|
|
714
|
-
}
|
|
715
|
-
}
|
|
716
|
-
' ' => {
|
|
717
|
-
chars.next();
|
|
718
|
-
}
|
|
719
|
-
_ => {
|
|
720
|
-
chars.next();
|
|
721
|
-
}
|
|
722
|
-
}
|
|
723
|
-
}
|
|
724
|
-
|
|
725
|
-
if let Some(itype) = image_type {
|
|
726
|
-
metadata.push_str("image.");
|
|
727
|
-
metadata.push_str(itype);
|
|
728
|
-
}
|
|
729
|
-
|
|
730
|
-
if let Some(width) = width_goal {
|
|
731
|
-
let width_inches = f64::from(width) / 1440.0;
|
|
732
|
-
metadata.push_str(&format!(" width=\"{:.1}in\"", width_inches));
|
|
733
|
-
}
|
|
734
|
-
|
|
735
|
-
if let Some(height) = height_goal {
|
|
736
|
-
let height_inches = f64::from(height) / 1440.0;
|
|
737
|
-
metadata.push_str(&format!(" height=\"{:.1}in\"", height_inches));
|
|
738
|
-
}
|
|
739
|
-
|
|
740
|
-
if metadata.is_empty() {
|
|
741
|
-
metadata.push_str("image.jpg");
|
|
742
|
-
}
|
|
743
|
-
|
|
744
|
-
metadata
|
|
745
|
-
}
|
|
746
|
-
|
|
747
|
-
#[async_trait]
|
|
748
|
-
impl DocumentExtractor for RtfExtractor {
|
|
749
|
-
#[cfg_attr(feature = "otel", tracing::instrument(
|
|
750
|
-
skip(self, content, _config),
|
|
751
|
-
fields(
|
|
752
|
-
extractor.name = self.name(),
|
|
753
|
-
content.size_bytes = content.len(),
|
|
754
|
-
)
|
|
755
|
-
))]
|
|
756
|
-
async fn extract_bytes(
|
|
757
|
-
&self,
|
|
758
|
-
content: &[u8],
|
|
759
|
-
mime_type: &str,
|
|
760
|
-
_config: &ExtractionConfig,
|
|
761
|
-
) -> Result<ExtractionResult> {
|
|
762
|
-
let rtf_content = String::from_utf8_lossy(content);
|
|
763
|
-
|
|
764
|
-
let (extracted_text, tables) = extract_text_from_rtf(&rtf_content);
|
|
765
|
-
let metadata_map = extract_rtf_metadata(&rtf_content, &extracted_text);
|
|
766
|
-
|
|
767
|
-
Ok(ExtractionResult {
|
|
768
|
-
content: extracted_text,
|
|
769
|
-
mime_type: mime_type.to_string(),
|
|
770
|
-
metadata: Metadata {
|
|
771
|
-
additional: metadata_map,
|
|
772
|
-
..Default::default()
|
|
773
|
-
},
|
|
774
|
-
tables,
|
|
775
|
-
detected_languages: None,
|
|
776
|
-
chunks: None,
|
|
777
|
-
images: None,
|
|
778
|
-
})
|
|
779
|
-
}
|
|
780
|
-
|
|
781
|
-
fn supported_mime_types(&self) -> &[&str] {
|
|
782
|
-
&["application/rtf", "text/rtf"]
|
|
783
|
-
}
|
|
784
|
-
|
|
785
|
-
fn priority(&self) -> i32 {
|
|
786
|
-
50
|
|
787
|
-
}
|
|
788
|
-
}
|
|
789
|
-
|
|
790
|
-
#[cfg(test)]
|
|
791
|
-
mod tests {
|
|
792
|
-
use super::*;
|
|
793
|
-
|
|
794
|
-
#[tokio::test]
|
|
795
|
-
async fn test_rtf_extractor_plugin_interface() {
|
|
796
|
-
let extractor = RtfExtractor::new();
|
|
797
|
-
assert_eq!(extractor.name(), "rtf-extractor");
|
|
798
|
-
assert_eq!(extractor.version(), env!("CARGO_PKG_VERSION"));
|
|
799
|
-
assert!(extractor.supported_mime_types().contains(&"application/rtf"));
|
|
800
|
-
assert_eq!(extractor.priority(), 50);
|
|
801
|
-
}
|
|
802
|
-
|
|
803
|
-
#[test]
|
|
804
|
-
fn test_simple_rtf_extraction() {
|
|
805
|
-
let _extractor = RtfExtractor;
|
|
806
|
-
let rtf_content = r#"{\rtf1 Hello World}"#;
|
|
807
|
-
let (extracted, _) = extract_text_from_rtf(rtf_content);
|
|
808
|
-
assert!(extracted.contains("Hello") || extracted.contains("World"));
|
|
809
|
-
}
|
|
810
|
-
}
|