kreuzberg 4.0.0.pre.rc.6

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.
Files changed (330) hide show
  1. checksums.yaml +7 -0
  2. data/.gitignore +14 -0
  3. data/.rspec +3 -0
  4. data/.rubocop.yaml +1 -0
  5. data/.rubocop.yml +538 -0
  6. data/Gemfile +8 -0
  7. data/Gemfile.lock +157 -0
  8. data/README.md +426 -0
  9. data/Rakefile +25 -0
  10. data/Steepfile +47 -0
  11. data/examples/async_patterns.rb +341 -0
  12. data/ext/kreuzberg_rb/extconf.rb +45 -0
  13. data/ext/kreuzberg_rb/native/Cargo.lock +6535 -0
  14. data/ext/kreuzberg_rb/native/Cargo.toml +44 -0
  15. data/ext/kreuzberg_rb/native/README.md +425 -0
  16. data/ext/kreuzberg_rb/native/build.rs +15 -0
  17. data/ext/kreuzberg_rb/native/include/ieeefp.h +11 -0
  18. data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +14 -0
  19. data/ext/kreuzberg_rb/native/include/strings.h +20 -0
  20. data/ext/kreuzberg_rb/native/include/unistd.h +47 -0
  21. data/ext/kreuzberg_rb/native/src/lib.rs +2998 -0
  22. data/extconf.rb +28 -0
  23. data/kreuzberg.gemspec +148 -0
  24. data/lib/kreuzberg/api_proxy.rb +142 -0
  25. data/lib/kreuzberg/cache_api.rb +46 -0
  26. data/lib/kreuzberg/cli.rb +55 -0
  27. data/lib/kreuzberg/cli_proxy.rb +127 -0
  28. data/lib/kreuzberg/config.rb +691 -0
  29. data/lib/kreuzberg/error_context.rb +32 -0
  30. data/lib/kreuzberg/errors.rb +118 -0
  31. data/lib/kreuzberg/extraction_api.rb +85 -0
  32. data/lib/kreuzberg/mcp_proxy.rb +186 -0
  33. data/lib/kreuzberg/ocr_backend_protocol.rb +113 -0
  34. data/lib/kreuzberg/post_processor_protocol.rb +86 -0
  35. data/lib/kreuzberg/result.rb +216 -0
  36. data/lib/kreuzberg/setup_lib_path.rb +80 -0
  37. data/lib/kreuzberg/validator_protocol.rb +89 -0
  38. data/lib/kreuzberg/version.rb +5 -0
  39. data/lib/kreuzberg.rb +103 -0
  40. data/sig/kreuzberg/internal.rbs +184 -0
  41. data/sig/kreuzberg.rbs +520 -0
  42. data/spec/binding/cache_spec.rb +227 -0
  43. data/spec/binding/cli_proxy_spec.rb +85 -0
  44. data/spec/binding/cli_spec.rb +55 -0
  45. data/spec/binding/config_spec.rb +345 -0
  46. data/spec/binding/config_validation_spec.rb +283 -0
  47. data/spec/binding/error_handling_spec.rb +213 -0
  48. data/spec/binding/errors_spec.rb +66 -0
  49. data/spec/binding/plugins/ocr_backend_spec.rb +307 -0
  50. data/spec/binding/plugins/postprocessor_spec.rb +269 -0
  51. data/spec/binding/plugins/validator_spec.rb +274 -0
  52. data/spec/fixtures/config.toml +39 -0
  53. data/spec/fixtures/config.yaml +41 -0
  54. data/spec/fixtures/invalid_config.toml +4 -0
  55. data/spec/smoke/package_spec.rb +178 -0
  56. data/spec/spec_helper.rb +42 -0
  57. data/vendor/kreuzberg/Cargo.toml +204 -0
  58. data/vendor/kreuzberg/README.md +175 -0
  59. data/vendor/kreuzberg/benches/otel_overhead.rs +48 -0
  60. data/vendor/kreuzberg/build.rs +474 -0
  61. data/vendor/kreuzberg/src/api/error.rs +81 -0
  62. data/vendor/kreuzberg/src/api/handlers.rs +199 -0
  63. data/vendor/kreuzberg/src/api/mod.rs +79 -0
  64. data/vendor/kreuzberg/src/api/server.rs +353 -0
  65. data/vendor/kreuzberg/src/api/types.rs +170 -0
  66. data/vendor/kreuzberg/src/cache/mod.rs +1167 -0
  67. data/vendor/kreuzberg/src/chunking/mod.rs +677 -0
  68. data/vendor/kreuzberg/src/core/batch_mode.rs +95 -0
  69. data/vendor/kreuzberg/src/core/config.rs +1032 -0
  70. data/vendor/kreuzberg/src/core/extractor.rs +1024 -0
  71. data/vendor/kreuzberg/src/core/io.rs +329 -0
  72. data/vendor/kreuzberg/src/core/mime.rs +605 -0
  73. data/vendor/kreuzberg/src/core/mod.rs +45 -0
  74. data/vendor/kreuzberg/src/core/pipeline.rs +984 -0
  75. data/vendor/kreuzberg/src/embeddings.rs +432 -0
  76. data/vendor/kreuzberg/src/error.rs +431 -0
  77. data/vendor/kreuzberg/src/extraction/archive.rs +954 -0
  78. data/vendor/kreuzberg/src/extraction/docx.rs +40 -0
  79. data/vendor/kreuzberg/src/extraction/email.rs +854 -0
  80. data/vendor/kreuzberg/src/extraction/excel.rs +688 -0
  81. data/vendor/kreuzberg/src/extraction/html.rs +553 -0
  82. data/vendor/kreuzberg/src/extraction/image.rs +368 -0
  83. data/vendor/kreuzberg/src/extraction/libreoffice.rs +563 -0
  84. data/vendor/kreuzberg/src/extraction/markdown.rs +213 -0
  85. data/vendor/kreuzberg/src/extraction/mod.rs +81 -0
  86. data/vendor/kreuzberg/src/extraction/office_metadata/app_properties.rs +398 -0
  87. data/vendor/kreuzberg/src/extraction/office_metadata/core_properties.rs +247 -0
  88. data/vendor/kreuzberg/src/extraction/office_metadata/custom_properties.rs +240 -0
  89. data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +130 -0
  90. data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +287 -0
  91. data/vendor/kreuzberg/src/extraction/pptx.rs +3000 -0
  92. data/vendor/kreuzberg/src/extraction/structured.rs +490 -0
  93. data/vendor/kreuzberg/src/extraction/table.rs +328 -0
  94. data/vendor/kreuzberg/src/extraction/text.rs +269 -0
  95. data/vendor/kreuzberg/src/extraction/xml.rs +333 -0
  96. data/vendor/kreuzberg/src/extractors/archive.rs +446 -0
  97. data/vendor/kreuzberg/src/extractors/bibtex.rs +469 -0
  98. data/vendor/kreuzberg/src/extractors/docbook.rs +502 -0
  99. data/vendor/kreuzberg/src/extractors/docx.rs +367 -0
  100. data/vendor/kreuzberg/src/extractors/email.rs +143 -0
  101. data/vendor/kreuzberg/src/extractors/epub.rs +707 -0
  102. data/vendor/kreuzberg/src/extractors/excel.rs +343 -0
  103. data/vendor/kreuzberg/src/extractors/fictionbook.rs +491 -0
  104. data/vendor/kreuzberg/src/extractors/fictionbook.rs.backup2 +738 -0
  105. data/vendor/kreuzberg/src/extractors/html.rs +393 -0
  106. data/vendor/kreuzberg/src/extractors/image.rs +198 -0
  107. data/vendor/kreuzberg/src/extractors/jats.rs +1051 -0
  108. data/vendor/kreuzberg/src/extractors/jupyter.rs +367 -0
  109. data/vendor/kreuzberg/src/extractors/latex.rs +652 -0
  110. data/vendor/kreuzberg/src/extractors/markdown.rs +700 -0
  111. data/vendor/kreuzberg/src/extractors/mod.rs +365 -0
  112. data/vendor/kreuzberg/src/extractors/odt.rs +628 -0
  113. data/vendor/kreuzberg/src/extractors/opml.rs +634 -0
  114. data/vendor/kreuzberg/src/extractors/orgmode.rs +528 -0
  115. data/vendor/kreuzberg/src/extractors/pdf.rs +493 -0
  116. data/vendor/kreuzberg/src/extractors/pptx.rs +248 -0
  117. data/vendor/kreuzberg/src/extractors/rst.rs +576 -0
  118. data/vendor/kreuzberg/src/extractors/rtf.rs +810 -0
  119. data/vendor/kreuzberg/src/extractors/security.rs +484 -0
  120. data/vendor/kreuzberg/src/extractors/security_tests.rs +367 -0
  121. data/vendor/kreuzberg/src/extractors/structured.rs +140 -0
  122. data/vendor/kreuzberg/src/extractors/text.rs +260 -0
  123. data/vendor/kreuzberg/src/extractors/typst.rs +650 -0
  124. data/vendor/kreuzberg/src/extractors/xml.rs +135 -0
  125. data/vendor/kreuzberg/src/image/dpi.rs +164 -0
  126. data/vendor/kreuzberg/src/image/mod.rs +6 -0
  127. data/vendor/kreuzberg/src/image/preprocessing.rs +417 -0
  128. data/vendor/kreuzberg/src/image/resize.rs +89 -0
  129. data/vendor/kreuzberg/src/keywords/config.rs +154 -0
  130. data/vendor/kreuzberg/src/keywords/mod.rs +237 -0
  131. data/vendor/kreuzberg/src/keywords/processor.rs +267 -0
  132. data/vendor/kreuzberg/src/keywords/rake.rs +293 -0
  133. data/vendor/kreuzberg/src/keywords/types.rs +68 -0
  134. data/vendor/kreuzberg/src/keywords/yake.rs +163 -0
  135. data/vendor/kreuzberg/src/language_detection/mod.rs +942 -0
  136. data/vendor/kreuzberg/src/lib.rs +105 -0
  137. data/vendor/kreuzberg/src/mcp/mod.rs +32 -0
  138. data/vendor/kreuzberg/src/mcp/server.rs +1968 -0
  139. data/vendor/kreuzberg/src/ocr/cache.rs +469 -0
  140. data/vendor/kreuzberg/src/ocr/error.rs +37 -0
  141. data/vendor/kreuzberg/src/ocr/hocr.rs +216 -0
  142. data/vendor/kreuzberg/src/ocr/mod.rs +58 -0
  143. data/vendor/kreuzberg/src/ocr/processor.rs +863 -0
  144. data/vendor/kreuzberg/src/ocr/table/mod.rs +4 -0
  145. data/vendor/kreuzberg/src/ocr/table/tsv_parser.rs +144 -0
  146. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +450 -0
  147. data/vendor/kreuzberg/src/ocr/types.rs +393 -0
  148. data/vendor/kreuzberg/src/ocr/utils.rs +47 -0
  149. data/vendor/kreuzberg/src/ocr/validation.rs +206 -0
  150. data/vendor/kreuzberg/src/panic_context.rs +154 -0
  151. data/vendor/kreuzberg/src/pdf/error.rs +122 -0
  152. data/vendor/kreuzberg/src/pdf/images.rs +139 -0
  153. data/vendor/kreuzberg/src/pdf/metadata.rs +346 -0
  154. data/vendor/kreuzberg/src/pdf/mod.rs +50 -0
  155. data/vendor/kreuzberg/src/pdf/rendering.rs +369 -0
  156. data/vendor/kreuzberg/src/pdf/table.rs +393 -0
  157. data/vendor/kreuzberg/src/pdf/text.rs +158 -0
  158. data/vendor/kreuzberg/src/plugins/extractor.rs +1013 -0
  159. data/vendor/kreuzberg/src/plugins/mod.rs +209 -0
  160. data/vendor/kreuzberg/src/plugins/ocr.rs +620 -0
  161. data/vendor/kreuzberg/src/plugins/processor.rs +642 -0
  162. data/vendor/kreuzberg/src/plugins/registry.rs +1337 -0
  163. data/vendor/kreuzberg/src/plugins/traits.rs +258 -0
  164. data/vendor/kreuzberg/src/plugins/validator.rs +956 -0
  165. data/vendor/kreuzberg/src/stopwords/mod.rs +1470 -0
  166. data/vendor/kreuzberg/src/text/mod.rs +19 -0
  167. data/vendor/kreuzberg/src/text/quality.rs +697 -0
  168. data/vendor/kreuzberg/src/text/string_utils.rs +217 -0
  169. data/vendor/kreuzberg/src/text/token_reduction/cjk_utils.rs +164 -0
  170. data/vendor/kreuzberg/src/text/token_reduction/config.rs +100 -0
  171. data/vendor/kreuzberg/src/text/token_reduction/core.rs +796 -0
  172. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +902 -0
  173. data/vendor/kreuzberg/src/text/token_reduction/mod.rs +160 -0
  174. data/vendor/kreuzberg/src/text/token_reduction/semantic.rs +619 -0
  175. data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +147 -0
  176. data/vendor/kreuzberg/src/types.rs +903 -0
  177. data/vendor/kreuzberg/src/utils/mod.rs +17 -0
  178. data/vendor/kreuzberg/src/utils/quality.rs +959 -0
  179. data/vendor/kreuzberg/src/utils/string_utils.rs +381 -0
  180. data/vendor/kreuzberg/stopwords/af_stopwords.json +53 -0
  181. data/vendor/kreuzberg/stopwords/ar_stopwords.json +482 -0
  182. data/vendor/kreuzberg/stopwords/bg_stopwords.json +261 -0
  183. data/vendor/kreuzberg/stopwords/bn_stopwords.json +400 -0
  184. data/vendor/kreuzberg/stopwords/br_stopwords.json +1205 -0
  185. data/vendor/kreuzberg/stopwords/ca_stopwords.json +280 -0
  186. data/vendor/kreuzberg/stopwords/cs_stopwords.json +425 -0
  187. data/vendor/kreuzberg/stopwords/da_stopwords.json +172 -0
  188. data/vendor/kreuzberg/stopwords/de_stopwords.json +622 -0
  189. data/vendor/kreuzberg/stopwords/el_stopwords.json +849 -0
  190. data/vendor/kreuzberg/stopwords/en_stopwords.json +1300 -0
  191. data/vendor/kreuzberg/stopwords/eo_stopwords.json +175 -0
  192. data/vendor/kreuzberg/stopwords/es_stopwords.json +734 -0
  193. data/vendor/kreuzberg/stopwords/et_stopwords.json +37 -0
  194. data/vendor/kreuzberg/stopwords/eu_stopwords.json +100 -0
  195. data/vendor/kreuzberg/stopwords/fa_stopwords.json +801 -0
  196. data/vendor/kreuzberg/stopwords/fi_stopwords.json +849 -0
  197. data/vendor/kreuzberg/stopwords/fr_stopwords.json +693 -0
  198. data/vendor/kreuzberg/stopwords/ga_stopwords.json +111 -0
  199. data/vendor/kreuzberg/stopwords/gl_stopwords.json +162 -0
  200. data/vendor/kreuzberg/stopwords/gu_stopwords.json +226 -0
  201. data/vendor/kreuzberg/stopwords/ha_stopwords.json +41 -0
  202. data/vendor/kreuzberg/stopwords/he_stopwords.json +196 -0
  203. data/vendor/kreuzberg/stopwords/hi_stopwords.json +227 -0
  204. data/vendor/kreuzberg/stopwords/hr_stopwords.json +181 -0
  205. data/vendor/kreuzberg/stopwords/hu_stopwords.json +791 -0
  206. data/vendor/kreuzberg/stopwords/hy_stopwords.json +47 -0
  207. data/vendor/kreuzberg/stopwords/id_stopwords.json +760 -0
  208. data/vendor/kreuzberg/stopwords/it_stopwords.json +634 -0
  209. data/vendor/kreuzberg/stopwords/ja_stopwords.json +136 -0
  210. data/vendor/kreuzberg/stopwords/kn_stopwords.json +84 -0
  211. data/vendor/kreuzberg/stopwords/ko_stopwords.json +681 -0
  212. data/vendor/kreuzberg/stopwords/ku_stopwords.json +64 -0
  213. data/vendor/kreuzberg/stopwords/la_stopwords.json +51 -0
  214. data/vendor/kreuzberg/stopwords/lt_stopwords.json +476 -0
  215. data/vendor/kreuzberg/stopwords/lv_stopwords.json +163 -0
  216. data/vendor/kreuzberg/stopwords/ml_stopwords.json +1 -0
  217. data/vendor/kreuzberg/stopwords/mr_stopwords.json +101 -0
  218. data/vendor/kreuzberg/stopwords/ms_stopwords.json +477 -0
  219. data/vendor/kreuzberg/stopwords/ne_stopwords.json +490 -0
  220. data/vendor/kreuzberg/stopwords/nl_stopwords.json +415 -0
  221. data/vendor/kreuzberg/stopwords/no_stopwords.json +223 -0
  222. data/vendor/kreuzberg/stopwords/pl_stopwords.json +331 -0
  223. data/vendor/kreuzberg/stopwords/pt_stopwords.json +562 -0
  224. data/vendor/kreuzberg/stopwords/ro_stopwords.json +436 -0
  225. data/vendor/kreuzberg/stopwords/ru_stopwords.json +561 -0
  226. data/vendor/kreuzberg/stopwords/si_stopwords.json +193 -0
  227. data/vendor/kreuzberg/stopwords/sk_stopwords.json +420 -0
  228. data/vendor/kreuzberg/stopwords/sl_stopwords.json +448 -0
  229. data/vendor/kreuzberg/stopwords/so_stopwords.json +32 -0
  230. data/vendor/kreuzberg/stopwords/st_stopwords.json +33 -0
  231. data/vendor/kreuzberg/stopwords/sv_stopwords.json +420 -0
  232. data/vendor/kreuzberg/stopwords/sw_stopwords.json +76 -0
  233. data/vendor/kreuzberg/stopwords/ta_stopwords.json +129 -0
  234. data/vendor/kreuzberg/stopwords/te_stopwords.json +54 -0
  235. data/vendor/kreuzberg/stopwords/th_stopwords.json +118 -0
  236. data/vendor/kreuzberg/stopwords/tl_stopwords.json +149 -0
  237. data/vendor/kreuzberg/stopwords/tr_stopwords.json +506 -0
  238. data/vendor/kreuzberg/stopwords/uk_stopwords.json +75 -0
  239. data/vendor/kreuzberg/stopwords/ur_stopwords.json +519 -0
  240. data/vendor/kreuzberg/stopwords/vi_stopwords.json +647 -0
  241. data/vendor/kreuzberg/stopwords/yo_stopwords.json +62 -0
  242. data/vendor/kreuzberg/stopwords/zh_stopwords.json +796 -0
  243. data/vendor/kreuzberg/stopwords/zu_stopwords.json +31 -0
  244. data/vendor/kreuzberg/tests/api_extract_multipart.rs +52 -0
  245. data/vendor/kreuzberg/tests/api_tests.rs +966 -0
  246. data/vendor/kreuzberg/tests/archive_integration.rs +543 -0
  247. data/vendor/kreuzberg/tests/batch_orchestration.rs +556 -0
  248. data/vendor/kreuzberg/tests/batch_processing.rs +316 -0
  249. data/vendor/kreuzberg/tests/bibtex_parity_test.rs +421 -0
  250. data/vendor/kreuzberg/tests/concurrency_stress.rs +525 -0
  251. data/vendor/kreuzberg/tests/config_features.rs +598 -0
  252. data/vendor/kreuzberg/tests/config_loading_tests.rs +415 -0
  253. data/vendor/kreuzberg/tests/core_integration.rs +510 -0
  254. data/vendor/kreuzberg/tests/csv_integration.rs +414 -0
  255. data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +498 -0
  256. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +122 -0
  257. data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +370 -0
  258. data/vendor/kreuzberg/tests/email_integration.rs +325 -0
  259. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +275 -0
  260. data/vendor/kreuzberg/tests/error_handling.rs +393 -0
  261. data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +228 -0
  262. data/vendor/kreuzberg/tests/format_integration.rs +159 -0
  263. data/vendor/kreuzberg/tests/helpers/mod.rs +142 -0
  264. data/vendor/kreuzberg/tests/html_table_test.rs +551 -0
  265. data/vendor/kreuzberg/tests/image_integration.rs +253 -0
  266. data/vendor/kreuzberg/tests/instrumentation_test.rs +139 -0
  267. data/vendor/kreuzberg/tests/jats_extractor_tests.rs +639 -0
  268. data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +704 -0
  269. data/vendor/kreuzberg/tests/keywords_integration.rs +479 -0
  270. data/vendor/kreuzberg/tests/keywords_quality.rs +509 -0
  271. data/vendor/kreuzberg/tests/latex_extractor_tests.rs +496 -0
  272. data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +490 -0
  273. data/vendor/kreuzberg/tests/mime_detection.rs +428 -0
  274. data/vendor/kreuzberg/tests/ocr_configuration.rs +510 -0
  275. data/vendor/kreuzberg/tests/ocr_errors.rs +676 -0
  276. data/vendor/kreuzberg/tests/ocr_quality.rs +627 -0
  277. data/vendor/kreuzberg/tests/ocr_stress.rs +469 -0
  278. data/vendor/kreuzberg/tests/odt_extractor_tests.rs +695 -0
  279. data/vendor/kreuzberg/tests/opml_extractor_tests.rs +616 -0
  280. data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +822 -0
  281. data/vendor/kreuzberg/tests/pdf_integration.rs +43 -0
  282. data/vendor/kreuzberg/tests/pipeline_integration.rs +1411 -0
  283. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +771 -0
  284. data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +560 -0
  285. data/vendor/kreuzberg/tests/plugin_system.rs +921 -0
  286. data/vendor/kreuzberg/tests/plugin_validator_test.rs +783 -0
  287. data/vendor/kreuzberg/tests/registry_integration_tests.rs +586 -0
  288. data/vendor/kreuzberg/tests/rst_extractor_tests.rs +692 -0
  289. data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +776 -0
  290. data/vendor/kreuzberg/tests/security_validation.rs +415 -0
  291. data/vendor/kreuzberg/tests/stopwords_integration_test.rs +888 -0
  292. data/vendor/kreuzberg/tests/test_fastembed.rs +609 -0
  293. data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +1259 -0
  294. data/vendor/kreuzberg/tests/typst_extractor_tests.rs +647 -0
  295. data/vendor/kreuzberg/tests/xlsx_metadata_extraction_test.rs +87 -0
  296. data/vendor/rb-sys/.cargo-ok +1 -0
  297. data/vendor/rb-sys/.cargo_vcs_info.json +6 -0
  298. data/vendor/rb-sys/Cargo.lock +393 -0
  299. data/vendor/rb-sys/Cargo.toml +70 -0
  300. data/vendor/rb-sys/Cargo.toml.orig +57 -0
  301. data/vendor/rb-sys/LICENSE-APACHE +190 -0
  302. data/vendor/rb-sys/LICENSE-MIT +21 -0
  303. data/vendor/rb-sys/bin/release.sh +21 -0
  304. data/vendor/rb-sys/build/features.rs +108 -0
  305. data/vendor/rb-sys/build/main.rs +246 -0
  306. data/vendor/rb-sys/build/stable_api_config.rs +153 -0
  307. data/vendor/rb-sys/build/version.rs +48 -0
  308. data/vendor/rb-sys/readme.md +36 -0
  309. data/vendor/rb-sys/src/bindings.rs +21 -0
  310. data/vendor/rb-sys/src/hidden.rs +11 -0
  311. data/vendor/rb-sys/src/lib.rs +34 -0
  312. data/vendor/rb-sys/src/macros.rs +371 -0
  313. data/vendor/rb-sys/src/memory.rs +53 -0
  314. data/vendor/rb-sys/src/ruby_abi_version.rs +38 -0
  315. data/vendor/rb-sys/src/special_consts.rs +31 -0
  316. data/vendor/rb-sys/src/stable_api/compiled.c +179 -0
  317. data/vendor/rb-sys/src/stable_api/compiled.rs +257 -0
  318. data/vendor/rb-sys/src/stable_api/ruby_2_6.rs +316 -0
  319. data/vendor/rb-sys/src/stable_api/ruby_2_7.rs +316 -0
  320. data/vendor/rb-sys/src/stable_api/ruby_3_0.rs +324 -0
  321. data/vendor/rb-sys/src/stable_api/ruby_3_1.rs +317 -0
  322. data/vendor/rb-sys/src/stable_api/ruby_3_2.rs +315 -0
  323. data/vendor/rb-sys/src/stable_api/ruby_3_3.rs +326 -0
  324. data/vendor/rb-sys/src/stable_api/ruby_3_4.rs +327 -0
  325. data/vendor/rb-sys/src/stable_api.rs +261 -0
  326. data/vendor/rb-sys/src/symbol.rs +31 -0
  327. data/vendor/rb-sys/src/tracking_allocator.rs +332 -0
  328. data/vendor/rb-sys/src/utils.rs +89 -0
  329. data/vendor/rb-sys/src/value_type.rs +7 -0
  330. metadata +536 -0
@@ -0,0 +1,810 @@
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
+ }