kreuzberg 4.0.0.rc1 → 4.0.0.rc2

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 (342) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +14 -8
  3. data/.rspec +3 -3
  4. data/.rubocop.yaml +1 -534
  5. data/.rubocop.yml +538 -0
  6. data/Gemfile +8 -9
  7. data/Gemfile.lock +9 -109
  8. data/README.md +426 -421
  9. data/Rakefile +25 -25
  10. data/Steepfile +47 -47
  11. data/examples/async_patterns.rb +341 -340
  12. data/ext/kreuzberg_rb/extconf.rb +45 -35
  13. data/ext/kreuzberg_rb/native/Cargo.lock +6535 -0
  14. data/ext/kreuzberg_rb/native/Cargo.toml +44 -36
  15. data/ext/kreuzberg_rb/native/README.md +425 -425
  16. data/ext/kreuzberg_rb/native/build.rs +15 -17
  17. data/ext/kreuzberg_rb/native/include/ieeefp.h +11 -11
  18. data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +14 -14
  19. data/ext/kreuzberg_rb/native/include/strings.h +20 -20
  20. data/ext/kreuzberg_rb/native/include/unistd.h +47 -47
  21. data/ext/kreuzberg_rb/native/src/lib.rs +2998 -2939
  22. data/extconf.rb +28 -28
  23. data/kreuzberg.gemspec +148 -105
  24. data/lib/kreuzberg/api_proxy.rb +142 -142
  25. data/lib/kreuzberg/cache_api.rb +46 -45
  26. data/lib/kreuzberg/cli.rb +55 -55
  27. data/lib/kreuzberg/cli_proxy.rb +127 -127
  28. data/lib/kreuzberg/config.rb +691 -684
  29. data/lib/kreuzberg/error_context.rb +32 -0
  30. data/lib/kreuzberg/errors.rb +118 -50
  31. data/lib/kreuzberg/extraction_api.rb +85 -84
  32. data/lib/kreuzberg/mcp_proxy.rb +186 -186
  33. data/lib/kreuzberg/ocr_backend_protocol.rb +113 -113
  34. data/lib/kreuzberg/post_processor_protocol.rb +86 -86
  35. data/lib/kreuzberg/result.rb +216 -216
  36. data/lib/kreuzberg/setup_lib_path.rb +80 -79
  37. data/lib/kreuzberg/validator_protocol.rb +89 -89
  38. data/lib/kreuzberg/version.rb +5 -5
  39. data/lib/kreuzberg.rb +103 -82
  40. data/sig/kreuzberg/internal.rbs +184 -184
  41. data/sig/kreuzberg.rbs +520 -468
  42. data/spec/binding/cache_spec.rb +227 -227
  43. data/spec/binding/cli_proxy_spec.rb +85 -87
  44. data/spec/binding/cli_spec.rb +55 -54
  45. data/spec/binding/config_spec.rb +345 -345
  46. data/spec/binding/config_validation_spec.rb +283 -283
  47. data/spec/binding/error_handling_spec.rb +213 -213
  48. data/spec/binding/errors_spec.rb +66 -66
  49. data/spec/binding/plugins/ocr_backend_spec.rb +307 -307
  50. data/spec/binding/plugins/postprocessor_spec.rb +269 -269
  51. data/spec/binding/plugins/validator_spec.rb +274 -274
  52. data/spec/fixtures/config.toml +39 -39
  53. data/spec/fixtures/config.yaml +41 -42
  54. data/spec/fixtures/invalid_config.toml +4 -4
  55. data/spec/smoke/package_spec.rb +178 -178
  56. data/spec/spec_helper.rb +42 -42
  57. data/vendor/kreuzberg/Cargo.toml +204 -134
  58. data/vendor/kreuzberg/README.md +175 -175
  59. data/vendor/kreuzberg/benches/otel_overhead.rs +48 -0
  60. data/vendor/kreuzberg/build.rs +474 -460
  61. data/vendor/kreuzberg/src/api/error.rs +81 -81
  62. data/vendor/kreuzberg/src/api/handlers.rs +199 -199
  63. data/vendor/kreuzberg/src/api/mod.rs +79 -79
  64. data/vendor/kreuzberg/src/api/server.rs +353 -353
  65. data/vendor/kreuzberg/src/api/types.rs +170 -170
  66. data/vendor/kreuzberg/src/cache/mod.rs +1167 -1143
  67. data/vendor/kreuzberg/src/chunking/mod.rs +677 -677
  68. data/vendor/kreuzberg/src/core/batch_mode.rs +95 -35
  69. data/vendor/kreuzberg/src/core/config.rs +1032 -1032
  70. data/vendor/kreuzberg/src/core/extractor.rs +1024 -903
  71. data/vendor/kreuzberg/src/core/io.rs +329 -327
  72. data/vendor/kreuzberg/src/core/mime.rs +605 -615
  73. data/vendor/kreuzberg/src/core/mod.rs +45 -42
  74. data/vendor/kreuzberg/src/core/pipeline.rs +984 -906
  75. data/vendor/kreuzberg/src/embeddings.rs +432 -323
  76. data/vendor/kreuzberg/src/error.rs +431 -431
  77. data/vendor/kreuzberg/src/extraction/archive.rs +954 -954
  78. data/vendor/kreuzberg/src/extraction/docx.rs +40 -40
  79. data/vendor/kreuzberg/src/extraction/email.rs +854 -854
  80. data/vendor/kreuzberg/src/extraction/excel.rs +688 -688
  81. data/vendor/kreuzberg/src/extraction/html.rs +553 -553
  82. data/vendor/kreuzberg/src/extraction/image.rs +368 -368
  83. data/vendor/kreuzberg/src/extraction/libreoffice.rs +563 -564
  84. data/vendor/kreuzberg/src/extraction/markdown.rs +213 -0
  85. data/vendor/kreuzberg/src/extraction/mod.rs +81 -77
  86. data/vendor/kreuzberg/src/extraction/office_metadata/app_properties.rs +398 -398
  87. data/vendor/kreuzberg/src/extraction/office_metadata/core_properties.rs +247 -247
  88. data/vendor/kreuzberg/src/extraction/office_metadata/custom_properties.rs +240 -240
  89. data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +130 -128
  90. data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +287 -0
  91. data/vendor/kreuzberg/src/extraction/pptx.rs +3000 -3000
  92. data/vendor/kreuzberg/src/extraction/structured.rs +490 -490
  93. data/vendor/kreuzberg/src/extraction/table.rs +328 -328
  94. data/vendor/kreuzberg/src/extraction/text.rs +269 -269
  95. data/vendor/kreuzberg/src/extraction/xml.rs +333 -333
  96. data/vendor/kreuzberg/src/extractors/archive.rs +446 -425
  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 -479
  100. data/vendor/kreuzberg/src/extractors/email.rs +143 -129
  101. data/vendor/kreuzberg/src/extractors/epub.rs +707 -0
  102. data/vendor/kreuzberg/src/extractors/excel.rs +343 -344
  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 -410
  106. data/vendor/kreuzberg/src/extractors/image.rs +198 -195
  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 -268
  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 -496
  116. data/vendor/kreuzberg/src/extractors/pptx.rs +248 -234
  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 -126
  122. data/vendor/kreuzberg/src/extractors/text.rs +260 -242
  123. data/vendor/kreuzberg/src/extractors/typst.rs +650 -0
  124. data/vendor/kreuzberg/src/extractors/xml.rs +135 -128
  125. data/vendor/kreuzberg/src/image/dpi.rs +164 -164
  126. data/vendor/kreuzberg/src/image/mod.rs +6 -6
  127. data/vendor/kreuzberg/src/image/preprocessing.rs +417 -417
  128. data/vendor/kreuzberg/src/image/resize.rs +89 -89
  129. data/vendor/kreuzberg/src/keywords/config.rs +154 -154
  130. data/vendor/kreuzberg/src/keywords/mod.rs +237 -237
  131. data/vendor/kreuzberg/src/keywords/processor.rs +267 -267
  132. data/vendor/kreuzberg/src/keywords/rake.rs +293 -294
  133. data/vendor/kreuzberg/src/keywords/types.rs +68 -68
  134. data/vendor/kreuzberg/src/keywords/yake.rs +163 -163
  135. data/vendor/kreuzberg/src/language_detection/mod.rs +942 -942
  136. data/vendor/kreuzberg/src/lib.rs +105 -102
  137. data/vendor/kreuzberg/src/mcp/mod.rs +32 -32
  138. data/vendor/kreuzberg/src/mcp/server.rs +1968 -1966
  139. data/vendor/kreuzberg/src/ocr/cache.rs +469 -469
  140. data/vendor/kreuzberg/src/ocr/error.rs +37 -37
  141. data/vendor/kreuzberg/src/ocr/hocr.rs +216 -216
  142. data/vendor/kreuzberg/src/ocr/mod.rs +58 -58
  143. data/vendor/kreuzberg/src/ocr/processor.rs +863 -847
  144. data/vendor/kreuzberg/src/ocr/table/mod.rs +4 -4
  145. data/vendor/kreuzberg/src/ocr/table/tsv_parser.rs +144 -144
  146. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +450 -450
  147. data/vendor/kreuzberg/src/ocr/types.rs +393 -393
  148. data/vendor/kreuzberg/src/ocr/utils.rs +47 -47
  149. data/vendor/kreuzberg/src/ocr/validation.rs +206 -206
  150. data/vendor/kreuzberg/src/panic_context.rs +154 -0
  151. data/vendor/kreuzberg/src/pdf/error.rs +122 -122
  152. data/vendor/kreuzberg/src/pdf/images.rs +139 -139
  153. data/vendor/kreuzberg/src/pdf/metadata.rs +346 -346
  154. data/vendor/kreuzberg/src/pdf/mod.rs +50 -50
  155. data/vendor/kreuzberg/src/pdf/rendering.rs +369 -369
  156. data/vendor/kreuzberg/src/pdf/table.rs +393 -420
  157. data/vendor/kreuzberg/src/pdf/text.rs +158 -161
  158. data/vendor/kreuzberg/src/plugins/extractor.rs +1013 -1010
  159. data/vendor/kreuzberg/src/plugins/mod.rs +209 -209
  160. data/vendor/kreuzberg/src/plugins/ocr.rs +620 -629
  161. data/vendor/kreuzberg/src/plugins/processor.rs +642 -641
  162. data/vendor/kreuzberg/src/plugins/registry.rs +1337 -1324
  163. data/vendor/kreuzberg/src/plugins/traits.rs +258 -258
  164. data/vendor/kreuzberg/src/plugins/validator.rs +956 -955
  165. data/vendor/kreuzberg/src/stopwords/mod.rs +1470 -1470
  166. data/vendor/kreuzberg/src/text/mod.rs +19 -19
  167. data/vendor/kreuzberg/src/text/quality.rs +697 -697
  168. data/vendor/kreuzberg/src/text/string_utils.rs +217 -217
  169. data/vendor/kreuzberg/src/text/token_reduction/cjk_utils.rs +164 -164
  170. data/vendor/kreuzberg/src/text/token_reduction/config.rs +100 -100
  171. data/vendor/kreuzberg/src/text/token_reduction/core.rs +796 -796
  172. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +902 -902
  173. data/vendor/kreuzberg/src/text/token_reduction/mod.rs +160 -160
  174. data/vendor/kreuzberg/src/text/token_reduction/semantic.rs +619 -619
  175. data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +147 -147
  176. data/vendor/kreuzberg/src/types.rs +903 -873
  177. data/vendor/kreuzberg/src/utils/mod.rs +17 -17
  178. data/vendor/kreuzberg/src/utils/quality.rs +959 -959
  179. data/vendor/kreuzberg/src/utils/string_utils.rs +381 -381
  180. data/vendor/kreuzberg/stopwords/af_stopwords.json +53 -53
  181. data/vendor/kreuzberg/stopwords/ar_stopwords.json +482 -482
  182. data/vendor/kreuzberg/stopwords/bg_stopwords.json +261 -261
  183. data/vendor/kreuzberg/stopwords/bn_stopwords.json +400 -400
  184. data/vendor/kreuzberg/stopwords/br_stopwords.json +1205 -1205
  185. data/vendor/kreuzberg/stopwords/ca_stopwords.json +280 -280
  186. data/vendor/kreuzberg/stopwords/cs_stopwords.json +425 -425
  187. data/vendor/kreuzberg/stopwords/da_stopwords.json +172 -172
  188. data/vendor/kreuzberg/stopwords/de_stopwords.json +622 -622
  189. data/vendor/kreuzberg/stopwords/el_stopwords.json +849 -849
  190. data/vendor/kreuzberg/stopwords/en_stopwords.json +1300 -1300
  191. data/vendor/kreuzberg/stopwords/eo_stopwords.json +175 -175
  192. data/vendor/kreuzberg/stopwords/es_stopwords.json +734 -734
  193. data/vendor/kreuzberg/stopwords/et_stopwords.json +37 -37
  194. data/vendor/kreuzberg/stopwords/eu_stopwords.json +100 -100
  195. data/vendor/kreuzberg/stopwords/fa_stopwords.json +801 -801
  196. data/vendor/kreuzberg/stopwords/fi_stopwords.json +849 -849
  197. data/vendor/kreuzberg/stopwords/fr_stopwords.json +693 -693
  198. data/vendor/kreuzberg/stopwords/ga_stopwords.json +111 -111
  199. data/vendor/kreuzberg/stopwords/gl_stopwords.json +162 -162
  200. data/vendor/kreuzberg/stopwords/gu_stopwords.json +226 -226
  201. data/vendor/kreuzberg/stopwords/ha_stopwords.json +41 -41
  202. data/vendor/kreuzberg/stopwords/he_stopwords.json +196 -196
  203. data/vendor/kreuzberg/stopwords/hi_stopwords.json +227 -227
  204. data/vendor/kreuzberg/stopwords/hr_stopwords.json +181 -181
  205. data/vendor/kreuzberg/stopwords/hu_stopwords.json +791 -791
  206. data/vendor/kreuzberg/stopwords/hy_stopwords.json +47 -47
  207. data/vendor/kreuzberg/stopwords/id_stopwords.json +760 -760
  208. data/vendor/kreuzberg/stopwords/it_stopwords.json +634 -634
  209. data/vendor/kreuzberg/stopwords/ja_stopwords.json +136 -136
  210. data/vendor/kreuzberg/stopwords/kn_stopwords.json +84 -84
  211. data/vendor/kreuzberg/stopwords/ko_stopwords.json +681 -681
  212. data/vendor/kreuzberg/stopwords/ku_stopwords.json +64 -64
  213. data/vendor/kreuzberg/stopwords/la_stopwords.json +51 -51
  214. data/vendor/kreuzberg/stopwords/lt_stopwords.json +476 -476
  215. data/vendor/kreuzberg/stopwords/lv_stopwords.json +163 -163
  216. data/vendor/kreuzberg/stopwords/ml_stopwords.json +1 -1
  217. data/vendor/kreuzberg/stopwords/mr_stopwords.json +101 -101
  218. data/vendor/kreuzberg/stopwords/ms_stopwords.json +477 -477
  219. data/vendor/kreuzberg/stopwords/ne_stopwords.json +490 -490
  220. data/vendor/kreuzberg/stopwords/nl_stopwords.json +415 -415
  221. data/vendor/kreuzberg/stopwords/no_stopwords.json +223 -223
  222. data/vendor/kreuzberg/stopwords/pl_stopwords.json +331 -331
  223. data/vendor/kreuzberg/stopwords/pt_stopwords.json +562 -562
  224. data/vendor/kreuzberg/stopwords/ro_stopwords.json +436 -436
  225. data/vendor/kreuzberg/stopwords/ru_stopwords.json +561 -561
  226. data/vendor/kreuzberg/stopwords/si_stopwords.json +193 -193
  227. data/vendor/kreuzberg/stopwords/sk_stopwords.json +420 -420
  228. data/vendor/kreuzberg/stopwords/sl_stopwords.json +448 -448
  229. data/vendor/kreuzberg/stopwords/so_stopwords.json +32 -32
  230. data/vendor/kreuzberg/stopwords/st_stopwords.json +33 -33
  231. data/vendor/kreuzberg/stopwords/sv_stopwords.json +420 -420
  232. data/vendor/kreuzberg/stopwords/sw_stopwords.json +76 -76
  233. data/vendor/kreuzberg/stopwords/ta_stopwords.json +129 -129
  234. data/vendor/kreuzberg/stopwords/te_stopwords.json +54 -54
  235. data/vendor/kreuzberg/stopwords/th_stopwords.json +118 -118
  236. data/vendor/kreuzberg/stopwords/tl_stopwords.json +149 -149
  237. data/vendor/kreuzberg/stopwords/tr_stopwords.json +506 -506
  238. data/vendor/kreuzberg/stopwords/uk_stopwords.json +75 -75
  239. data/vendor/kreuzberg/stopwords/ur_stopwords.json +519 -519
  240. data/vendor/kreuzberg/stopwords/vi_stopwords.json +647 -647
  241. data/vendor/kreuzberg/stopwords/yo_stopwords.json +62 -62
  242. data/vendor/kreuzberg/stopwords/zh_stopwords.json +796 -796
  243. data/vendor/kreuzberg/stopwords/zu_stopwords.json +31 -31
  244. data/vendor/kreuzberg/tests/api_extract_multipart.rs +52 -0
  245. data/vendor/kreuzberg/tests/api_tests.rs +966 -966
  246. data/vendor/kreuzberg/tests/archive_integration.rs +543 -543
  247. data/vendor/kreuzberg/tests/batch_orchestration.rs +556 -542
  248. data/vendor/kreuzberg/tests/batch_processing.rs +316 -304
  249. data/vendor/kreuzberg/tests/bibtex_parity_test.rs +421 -0
  250. data/vendor/kreuzberg/tests/concurrency_stress.rs +525 -509
  251. data/vendor/kreuzberg/tests/config_features.rs +598 -580
  252. data/vendor/kreuzberg/tests/config_loading_tests.rs +415 -439
  253. data/vendor/kreuzberg/tests/core_integration.rs +510 -493
  254. data/vendor/kreuzberg/tests/csv_integration.rs +414 -424
  255. data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +498 -0
  256. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +122 -124
  257. data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +370 -0
  258. data/vendor/kreuzberg/tests/email_integration.rs +325 -325
  259. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +275 -0
  260. data/vendor/kreuzberg/tests/error_handling.rs +393 -393
  261. data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +228 -0
  262. data/vendor/kreuzberg/tests/format_integration.rs +159 -159
  263. data/vendor/kreuzberg/tests/helpers/mod.rs +142 -142
  264. data/vendor/kreuzberg/tests/html_table_test.rs +551 -0
  265. data/vendor/kreuzberg/tests/image_integration.rs +253 -253
  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 -479
  270. data/vendor/kreuzberg/tests/keywords_quality.rs +509 -509
  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 -428
  274. data/vendor/kreuzberg/tests/ocr_configuration.rs +510 -510
  275. data/vendor/kreuzberg/tests/ocr_errors.rs +676 -676
  276. data/vendor/kreuzberg/tests/ocr_quality.rs +627 -627
  277. data/vendor/kreuzberg/tests/ocr_stress.rs +469 -469
  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 -43
  282. data/vendor/kreuzberg/tests/pipeline_integration.rs +1411 -1412
  283. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +771 -771
  284. data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +560 -561
  285. data/vendor/kreuzberg/tests/plugin_system.rs +921 -921
  286. data/vendor/kreuzberg/tests/plugin_validator_test.rs +783 -783
  287. data/vendor/kreuzberg/tests/registry_integration_tests.rs +586 -607
  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 -404
  291. data/vendor/kreuzberg/tests/stopwords_integration_test.rs +888 -888
  292. data/vendor/kreuzberg/tests/test_fastembed.rs +609 -609
  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 -87
  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 +90 -95
  331. data/pkg/kreuzberg-4.0.0.rc1.gem +0 -0
  332. data/spec/examples.txt +0 -104
  333. data/vendor/kreuzberg/src/bin/profile_extract.rs +0 -455
  334. data/vendor/kreuzberg/src/extraction/pandoc/batch.rs +0 -275
  335. data/vendor/kreuzberg/src/extraction/pandoc/mime_types.rs +0 -178
  336. data/vendor/kreuzberg/src/extraction/pandoc/mod.rs +0 -491
  337. data/vendor/kreuzberg/src/extraction/pandoc/server.rs +0 -496
  338. data/vendor/kreuzberg/src/extraction/pandoc/subprocess.rs +0 -1188
  339. data/vendor/kreuzberg/src/extraction/pandoc/version.rs +0 -162
  340. data/vendor/kreuzberg/src/extractors/pandoc.rs +0 -201
  341. data/vendor/kreuzberg/tests/chunking_offset_demo.rs +0 -92
  342. data/vendor/kreuzberg/tests/pandoc_integration.rs +0 -503
@@ -1,479 +1,367 @@
1
- //! DOCX extractor using docx-lite for high-performance text extraction.
2
- //!
3
- //! Supports: Microsoft Word (.docx)
4
-
5
- use crate::Result;
6
- use crate::core::config::ExtractionConfig;
7
- use crate::extraction::office_metadata;
8
- use crate::plugins::{DocumentExtractor, Plugin};
9
- use crate::types::{ExtractionResult, Metadata, Table};
10
- use async_trait::async_trait;
11
- use std::io::Cursor;
12
-
13
- /// High-performance DOCX extractor using docx-lite.
14
- ///
15
- /// This extractor provides:
16
- /// - Fast text extraction via streaming XML parsing (~160 MB/s average)
17
- /// - Comprehensive metadata extraction (core.xml, app.xml, custom.xml)
18
- /// - ~400x faster than Pandoc subprocess approach
19
- pub struct DocxExtractor;
20
-
21
- impl DocxExtractor {
22
- /// Create a new DOCX extractor.
23
- pub fn new() -> Self {
24
- Self
25
- }
26
- }
27
-
28
- impl Default for DocxExtractor {
29
- fn default() -> Self {
30
- Self::new()
31
- }
32
- }
33
-
34
- impl Plugin for DocxExtractor {
35
- fn name(&self) -> &str {
36
- "docx-extractor"
37
- }
38
-
39
- fn version(&self) -> String {
40
- env!("CARGO_PKG_VERSION").to_string()
41
- }
42
-
43
- fn initialize(&self) -> Result<()> {
44
- Ok(())
45
- }
46
-
47
- fn shutdown(&self) -> Result<()> {
48
- Ok(())
49
- }
50
-
51
- fn description(&self) -> &str {
52
- "High-performance DOCX text extraction using docx-lite with metadata support"
53
- }
54
-
55
- fn author(&self) -> &str {
56
- "Kreuzberg Team"
57
- }
58
- }
59
-
60
- /// Convert docx-lite table to Kreuzberg Table struct with markdown representation.
61
- ///
62
- /// # Arguments
63
- /// * `docx_table` - The table from docx-lite library
64
- /// * `table_index` - Index of the table in the document (used as page_number)
65
- ///
66
- /// # Returns
67
- /// * `Table` - Converted table with cells and markdown representation
68
- fn convert_docx_table_to_table(docx_table: &docx_lite::Table, table_index: usize) -> Table {
69
- // Extract cells as 2D vector
70
- let cells: Vec<Vec<String>> = docx_table
71
- .rows
72
- .iter()
73
- .map(|row| {
74
- row.cells
75
- .iter()
76
- .map(|cell| {
77
- // Extract text from all paragraphs in the cell
78
- cell.paragraphs
79
- .iter()
80
- .map(|para| para.to_text())
81
- .collect::<Vec<_>>()
82
- .join(" ")
83
- .trim()
84
- .to_string()
85
- })
86
- .collect()
87
- })
88
- .collect();
89
-
90
- // Generate markdown representation
91
- let markdown = cells_to_markdown(&cells);
92
-
93
- Table {
94
- cells,
95
- markdown,
96
- page_number: table_index + 1, // 1-indexed
97
- }
98
- }
99
-
100
- /// Convert 2D cell data to markdown table format.
101
- ///
102
- /// # Arguments
103
- /// * `cells` - 2D vector of cell strings (rows × columns)
104
- ///
105
- /// # Returns
106
- /// * `String` - Markdown formatted table
107
- fn cells_to_markdown(cells: &[Vec<String>]) -> String {
108
- if cells.is_empty() {
109
- return String::new();
110
- }
111
-
112
- let mut markdown = String::new();
113
-
114
- // Determine number of columns from first row
115
- let num_cols = cells.first().map(|r| r.len()).unwrap_or(0);
116
- if num_cols == 0 {
117
- return String::new();
118
- }
119
-
120
- // Header row (first row)
121
- if let Some(header) = cells.first() {
122
- markdown.push_str("| ");
123
- for cell in header {
124
- // Escape pipe characters in cell content
125
- let escaped = cell.replace('|', "\\|");
126
- markdown.push_str(&escaped);
127
- markdown.push_str(" | ");
128
- }
129
- markdown.push('\n');
130
-
131
- // Separator row
132
- markdown.push('|');
133
- for _ in 0..num_cols {
134
- markdown.push_str("------|");
135
- }
136
- markdown.push('\n');
137
- }
138
-
139
- // Data rows (skip first row as it's the header)
140
- for row in cells.iter().skip(1) {
141
- markdown.push_str("| ");
142
- for (idx, cell) in row.iter().enumerate() {
143
- if idx >= num_cols {
144
- break; // Handle irregular tables
145
- }
146
- // Escape pipe characters in cell content
147
- let escaped = cell.replace('|', "\\|");
148
- markdown.push_str(&escaped);
149
- markdown.push_str(" | ");
150
- }
151
- // Pad with empty cells if row is shorter than expected
152
- for _ in row.len()..num_cols {
153
- markdown.push_str(" | ");
154
- }
155
- markdown.push('\n');
156
- }
157
-
158
- markdown
159
- }
160
-
161
- #[async_trait]
162
- impl DocumentExtractor for DocxExtractor {
163
- async fn extract_bytes(
164
- &self,
165
- content: &[u8],
166
- mime_type: &str,
167
- _config: &ExtractionConfig,
168
- ) -> Result<ExtractionResult> {
169
- // Parse the DOCX document to extract both text and tables
170
- let (text, tables) = if crate::core::batch_mode::is_batch_mode() {
171
- // Batch mode: Use spawn_blocking for parallelism
172
- let content_owned = content.to_vec();
173
- tokio::task::spawn_blocking(move || -> crate::error::Result<(String, Vec<Table>)> {
174
- // Parse document structure
175
- let cursor = Cursor::new(&content_owned);
176
- let doc = docx_lite::parse_document(cursor)
177
- .map_err(|e| crate::error::KreuzbergError::parsing(format!("DOCX parsing failed: {}", e)))?;
178
-
179
- // Extract text
180
- let text = doc.extract_text();
181
-
182
- // Extract tables
183
- let tables: Vec<Table> = doc
184
- .tables
185
- .iter()
186
- .enumerate()
187
- .map(|(idx, table)| convert_docx_table_to_table(table, idx))
188
- .collect();
189
-
190
- Ok((text, tables))
191
- })
192
- .await
193
- .map_err(|e| crate::error::KreuzbergError::parsing(format!("DOCX extraction task failed: {}", e)))??
194
- } else {
195
- // Single-file mode: Direct extraction (no spawn overhead)
196
- let cursor = Cursor::new(content);
197
- let doc = docx_lite::parse_document(cursor)
198
- .map_err(|e| crate::error::KreuzbergError::parsing(format!("DOCX parsing failed: {}", e)))?;
199
-
200
- // Extract text
201
- let text = doc.extract_text();
202
-
203
- // Extract tables
204
- let tables: Vec<Table> = doc
205
- .tables
206
- .iter()
207
- .enumerate()
208
- .map(|(idx, table)| convert_docx_table_to_table(table, idx))
209
- .collect();
210
-
211
- (text, tables)
212
- };
213
-
214
- // Extract metadata using existing office_metadata module
215
- let mut archive = if crate::core::batch_mode::is_batch_mode() {
216
- // Batch mode: Use spawn_blocking for parallelism
217
- let content_owned = content.to_vec();
218
- tokio::task::spawn_blocking(move || -> crate::error::Result<_> {
219
- let cursor = Cursor::new(content_owned);
220
- zip::ZipArchive::new(cursor)
221
- .map_err(|e| crate::error::KreuzbergError::parsing(format!("Failed to open ZIP archive: {}", e)))
222
- })
223
- .await
224
- .map_err(|e| crate::error::KreuzbergError::parsing(format!("Task join error: {}", e)))??
225
- } else {
226
- // Single-file mode: Direct extraction (no spawn overhead)
227
- // Note: We still need to clone for ZipArchive type consistency with batch mode
228
- let content_owned = content.to_vec();
229
- let cursor = Cursor::new(content_owned);
230
- zip::ZipArchive::new(cursor)
231
- .map_err(|e| crate::error::KreuzbergError::parsing(format!("Failed to open ZIP archive: {}", e)))?
232
- };
233
-
234
- let mut metadata_map = std::collections::HashMap::new();
235
-
236
- // Extract core properties (title, creator, dates, keywords, etc.)
237
- if let Ok(core) = office_metadata::extract_core_properties(&mut archive) {
238
- if let Some(title) = core.title {
239
- metadata_map.insert("title".to_string(), serde_json::Value::String(title));
240
- }
241
- if let Some(creator) = core.creator {
242
- metadata_map.insert(
243
- "authors".to_string(),
244
- serde_json::Value::Array(vec![serde_json::Value::String(creator.clone())]),
245
- );
246
- metadata_map.insert("created_by".to_string(), serde_json::Value::String(creator));
247
- }
248
- if let Some(subject) = core.subject {
249
- metadata_map.insert("subject".to_string(), serde_json::Value::String(subject));
250
- }
251
- if let Some(keywords) = core.keywords {
252
- metadata_map.insert("keywords".to_string(), serde_json::Value::String(keywords));
253
- }
254
- if let Some(description) = core.description {
255
- metadata_map.insert("description".to_string(), serde_json::Value::String(description));
256
- }
257
- if let Some(modified_by) = core.last_modified_by {
258
- metadata_map.insert("modified_by".to_string(), serde_json::Value::String(modified_by));
259
- }
260
- if let Some(created) = core.created {
261
- metadata_map.insert("created_at".to_string(), serde_json::Value::String(created));
262
- }
263
- if let Some(modified) = core.modified {
264
- metadata_map.insert("modified_at".to_string(), serde_json::Value::String(modified));
265
- }
266
- if let Some(revision) = core.revision {
267
- metadata_map.insert("revision".to_string(), serde_json::Value::String(revision));
268
- }
269
- if let Some(category) = core.category {
270
- metadata_map.insert("category".to_string(), serde_json::Value::String(category));
271
- }
272
- if let Some(content_status) = core.content_status {
273
- metadata_map.insert("content_status".to_string(), serde_json::Value::String(content_status));
274
- }
275
- if let Some(language) = core.language {
276
- metadata_map.insert("language".to_string(), serde_json::Value::String(language));
277
- }
278
- }
279
-
280
- // Extract app properties (page count, word count, etc.)
281
- if let Ok(app) = office_metadata::extract_docx_app_properties(&mut archive) {
282
- if let Some(pages) = app.pages {
283
- metadata_map.insert("page_count".to_string(), serde_json::Value::Number(pages.into()));
284
- }
285
- if let Some(words) = app.words {
286
- metadata_map.insert("word_count".to_string(), serde_json::Value::Number(words.into()));
287
- }
288
- if let Some(chars) = app.characters {
289
- metadata_map.insert("character_count".to_string(), serde_json::Value::Number(chars.into()));
290
- }
291
- if let Some(lines) = app.lines {
292
- metadata_map.insert("line_count".to_string(), serde_json::Value::Number(lines.into()));
293
- }
294
- if let Some(paragraphs) = app.paragraphs {
295
- metadata_map.insert(
296
- "paragraph_count".to_string(),
297
- serde_json::Value::Number(paragraphs.into()),
298
- );
299
- }
300
- if let Some(template) = app.template {
301
- metadata_map.insert("template".to_string(), serde_json::Value::String(template));
302
- }
303
- if let Some(company) = app.company {
304
- metadata_map.insert("organization".to_string(), serde_json::Value::String(company));
305
- }
306
- if let Some(time) = app.total_time {
307
- metadata_map.insert(
308
- "total_editing_time_minutes".to_string(),
309
- serde_json::Value::Number(time.into()),
310
- );
311
- }
312
- if let Some(application) = app.application {
313
- metadata_map.insert("application".to_string(), serde_json::Value::String(application));
314
- }
315
- }
316
-
317
- // Extract custom properties
318
- if let Ok(custom) = office_metadata::extract_custom_properties(&mut archive) {
319
- for (key, value) in custom {
320
- metadata_map.insert(format!("custom_{}", key), value);
321
- }
322
- }
323
-
324
- Ok(ExtractionResult {
325
- content: text,
326
- mime_type: mime_type.to_string(),
327
- metadata: Metadata {
328
- additional: metadata_map,
329
- ..Default::default()
330
- },
331
- tables,
332
- detected_languages: None,
333
- chunks: None,
334
- images: None,
335
- })
336
- }
337
-
338
- fn supported_mime_types(&self) -> &[&str] {
339
- &["application/vnd.openxmlformats-officedocument.wordprocessingml.document"]
340
- }
341
-
342
- fn priority(&self) -> i32 {
343
- 50 // Higher priority than Pandoc (40) to take precedence
344
- }
345
- }
346
-
347
- #[cfg(test)]
348
- mod tests {
349
- use super::*;
350
-
351
- #[tokio::test]
352
- async fn test_docx_extractor_plugin_interface() {
353
- let extractor = DocxExtractor::new();
354
- assert_eq!(extractor.name(), "docx-extractor");
355
- assert_eq!(extractor.version(), env!("CARGO_PKG_VERSION"));
356
- assert_eq!(extractor.priority(), 50);
357
- assert_eq!(extractor.supported_mime_types().len(), 1);
358
- }
359
-
360
- #[tokio::test]
361
- async fn test_docx_extractor_supports_docx() {
362
- let extractor = DocxExtractor::new();
363
- assert!(
364
- extractor
365
- .supported_mime_types()
366
- .contains(&"application/vnd.openxmlformats-officedocument.wordprocessingml.document")
367
- );
368
- }
369
-
370
- #[tokio::test]
371
- async fn test_docx_extractor_default() {
372
- let extractor = DocxExtractor;
373
- assert_eq!(extractor.name(), "docx-extractor");
374
- }
375
-
376
- #[tokio::test]
377
- async fn test_docx_extractor_initialize_shutdown() {
378
- let extractor = DocxExtractor::new();
379
- assert!(extractor.initialize().is_ok());
380
- assert!(extractor.shutdown().is_ok());
381
- }
382
-
383
- #[test]
384
- fn test_cells_to_markdown_basic_table() {
385
- let cells = vec![
386
- vec!["Header1".to_string(), "Header2".to_string()],
387
- vec!["Row1Col1".to_string(), "Row1Col2".to_string()],
388
- vec!["Row2Col1".to_string(), "Row2Col2".to_string()],
389
- ];
390
-
391
- let markdown = cells_to_markdown(&cells);
392
-
393
- assert!(markdown.contains("| Header1 | Header2 |"));
394
- assert!(markdown.contains("|------|------|"));
395
- assert!(markdown.contains("| Row1Col1 | Row1Col2 |"));
396
- assert!(markdown.contains("| Row2Col1 | Row2Col2 |"));
397
- }
398
-
399
- #[test]
400
- fn test_cells_to_markdown_empty() {
401
- let cells: Vec<Vec<String>> = vec![];
402
- let markdown = cells_to_markdown(&cells);
403
- assert_eq!(markdown, "");
404
- }
405
-
406
- #[test]
407
- fn test_cells_to_markdown_escape_pipes() {
408
- let cells = vec![vec!["Header".to_string()], vec!["Cell with | pipe".to_string()]];
409
-
410
- let markdown = cells_to_markdown(&cells);
411
- assert!(markdown.contains("Cell with \\| pipe"));
412
- }
413
-
414
- #[test]
415
- fn test_cells_to_markdown_irregular_rows() {
416
- let cells = vec![
417
- vec!["H1".to_string(), "H2".to_string(), "H3".to_string()],
418
- vec!["R1C1".to_string(), "R1C2".to_string()], // Missing third column
419
- vec!["R2C1".to_string(), "R2C2".to_string(), "R2C3".to_string()],
420
- ];
421
-
422
- let markdown = cells_to_markdown(&cells);
423
-
424
- // Should have 3 columns in header
425
- assert!(markdown.contains("| H1 | H2 | H3 |"));
426
- // Should pad short rows
427
- assert!(markdown.contains("| R1C1 | R1C2 | |"));
428
- }
429
-
430
- #[test]
431
- fn test_convert_docx_table_to_table() {
432
- use docx_lite::{Paragraph, Run, Table as DocxTable, TableCell, TableRow};
433
-
434
- // Create a simple docx-lite table
435
- let mut table = DocxTable::new();
436
-
437
- // Header row
438
- let mut header_row = TableRow::default();
439
- let mut cell1 = TableCell::default();
440
- let mut para1 = Paragraph::new();
441
- para1.add_run(Run::new("Name".to_string()));
442
- cell1.paragraphs.push(para1);
443
- header_row.cells.push(cell1);
444
-
445
- let mut cell2 = TableCell::default();
446
- let mut para2 = Paragraph::new();
447
- para2.add_run(Run::new("Age".to_string()));
448
- cell2.paragraphs.push(para2);
449
- header_row.cells.push(cell2);
450
-
451
- table.rows.push(header_row);
452
-
453
- // Data row
454
- let mut data_row = TableRow::default();
455
- let mut cell3 = TableCell::default();
456
- let mut para3 = Paragraph::new();
457
- para3.add_run(Run::new("Alice".to_string()));
458
- cell3.paragraphs.push(para3);
459
- data_row.cells.push(cell3);
460
-
461
- let mut cell4 = TableCell::default();
462
- let mut para4 = Paragraph::new();
463
- para4.add_run(Run::new("30".to_string()));
464
- cell4.paragraphs.push(para4);
465
- data_row.cells.push(cell4);
466
-
467
- table.rows.push(data_row);
468
-
469
- // Convert to Kreuzberg Table
470
- let result = convert_docx_table_to_table(&table, 0);
471
-
472
- assert_eq!(result.page_number, 1); // 0 + 1 = 1 (1-indexed)
473
- assert_eq!(result.cells.len(), 2); // 2 rows
474
- assert_eq!(result.cells[0], vec!["Name", "Age"]);
475
- assert_eq!(result.cells[1], vec!["Alice", "30"]);
476
- assert!(result.markdown.contains("| Name | Age |"));
477
- assert!(result.markdown.contains("| Alice | 30 |"));
478
- }
479
- }
1
+ //! DOCX extractor using docx-lite for high-performance text extraction.
2
+ //!
3
+ //! Supports: Microsoft Word (.docx)
4
+
5
+ use crate::Result;
6
+ use crate::core::config::ExtractionConfig;
7
+ use crate::extraction::{cells_to_markdown, office_metadata};
8
+ use crate::plugins::{DocumentExtractor, Plugin};
9
+ use crate::types::{ExtractionResult, Metadata, Table};
10
+ use async_trait::async_trait;
11
+ use std::io::Cursor;
12
+
13
+ /// High-performance DOCX extractor using docx-lite.
14
+ ///
15
+ /// This extractor provides:
16
+ /// - Fast text extraction via streaming XML parsing (~160 MB/s average)
17
+ /// - Comprehensive metadata extraction (core.xml, app.xml, custom.xml)
18
+ pub struct DocxExtractor;
19
+
20
+ impl DocxExtractor {
21
+ /// Create a new DOCX extractor.
22
+ pub fn new() -> Self {
23
+ Self
24
+ }
25
+ }
26
+
27
+ impl Default for DocxExtractor {
28
+ fn default() -> Self {
29
+ Self::new()
30
+ }
31
+ }
32
+
33
+ impl Plugin for DocxExtractor {
34
+ fn name(&self) -> &str {
35
+ "docx-extractor"
36
+ }
37
+
38
+ fn version(&self) -> String {
39
+ env!("CARGO_PKG_VERSION").to_string()
40
+ }
41
+
42
+ fn initialize(&self) -> Result<()> {
43
+ Ok(())
44
+ }
45
+
46
+ fn shutdown(&self) -> Result<()> {
47
+ Ok(())
48
+ }
49
+
50
+ fn description(&self) -> &str {
51
+ "High-performance DOCX text extraction using docx-lite with metadata support"
52
+ }
53
+
54
+ fn author(&self) -> &str {
55
+ "Kreuzberg Team"
56
+ }
57
+ }
58
+
59
+ /// Convert docx-lite table to Kreuzberg Table struct with markdown representation.
60
+ ///
61
+ /// # Arguments
62
+ /// * `docx_table` - The table from docx-lite library
63
+ /// * `table_index` - Index of the table in the document (used as page_number)
64
+ ///
65
+ /// # Returns
66
+ /// * `Table` - Converted table with cells and markdown representation
67
+ fn convert_docx_table_to_table(docx_table: &docx_lite::Table, table_index: usize) -> Table {
68
+ let cells: Vec<Vec<String>> = docx_table
69
+ .rows
70
+ .iter()
71
+ .map(|row| {
72
+ row.cells
73
+ .iter()
74
+ .map(|cell| {
75
+ cell.paragraphs
76
+ .iter()
77
+ .map(|para| para.to_text())
78
+ .collect::<Vec<_>>()
79
+ .join(" ")
80
+ .trim()
81
+ .to_string()
82
+ })
83
+ .collect()
84
+ })
85
+ .collect();
86
+
87
+ let markdown = cells_to_markdown(&cells);
88
+
89
+ Table {
90
+ cells,
91
+ markdown,
92
+ page_number: table_index + 1,
93
+ }
94
+ }
95
+
96
+ /// Convert 2D cell data to markdown table format.
97
+ ///
98
+ /// # Arguments
99
+ /// * `cells` - 2D vector of cell strings (rows × columns)
100
+ ///
101
+ /// # Returns
102
+ /// * `String` - Markdown formatted table
103
+
104
+ #[async_trait]
105
+ impl DocumentExtractor for DocxExtractor {
106
+ #[cfg_attr(feature = "otel", tracing::instrument(
107
+ skip(self, content, _config),
108
+ fields(
109
+ extractor.name = self.name(),
110
+ content.size_bytes = content.len(),
111
+ )
112
+ ))]
113
+ async fn extract_bytes(
114
+ &self,
115
+ content: &[u8],
116
+ mime_type: &str,
117
+ _config: &ExtractionConfig,
118
+ ) -> Result<ExtractionResult> {
119
+ let (text, tables) = if crate::core::batch_mode::is_batch_mode() {
120
+ let content_owned = content.to_vec();
121
+ let span = tracing::Span::current();
122
+ tokio::task::spawn_blocking(move || -> crate::error::Result<(String, Vec<Table>)> {
123
+ let _guard = span.entered();
124
+ let cursor = Cursor::new(&content_owned);
125
+ let doc = docx_lite::parse_document(cursor)
126
+ .map_err(|e| crate::error::KreuzbergError::parsing(format!("DOCX parsing failed: {}", e)))?;
127
+
128
+ let text = doc.extract_text();
129
+
130
+ let tables: Vec<Table> = doc
131
+ .tables
132
+ .iter()
133
+ .enumerate()
134
+ .map(|(idx, table)| convert_docx_table_to_table(table, idx))
135
+ .collect();
136
+
137
+ Ok((text, tables))
138
+ })
139
+ .await
140
+ .map_err(|e| crate::error::KreuzbergError::parsing(format!("DOCX extraction task failed: {}", e)))??
141
+ } else {
142
+ let cursor = Cursor::new(content);
143
+ let doc = docx_lite::parse_document(cursor)
144
+ .map_err(|e| crate::error::KreuzbergError::parsing(format!("DOCX parsing failed: {}", e)))?;
145
+
146
+ let text = doc.extract_text();
147
+
148
+ let tables: Vec<Table> = doc
149
+ .tables
150
+ .iter()
151
+ .enumerate()
152
+ .map(|(idx, table)| convert_docx_table_to_table(table, idx))
153
+ .collect();
154
+
155
+ (text, tables)
156
+ };
157
+
158
+ let mut archive = if crate::core::batch_mode::is_batch_mode() {
159
+ let content_owned = content.to_vec();
160
+ let span = tracing::Span::current();
161
+ tokio::task::spawn_blocking(move || -> crate::error::Result<_> {
162
+ let _guard = span.entered();
163
+ let cursor = Cursor::new(content_owned);
164
+ zip::ZipArchive::new(cursor)
165
+ .map_err(|e| crate::error::KreuzbergError::parsing(format!("Failed to open ZIP archive: {}", e)))
166
+ })
167
+ .await
168
+ .map_err(|e| crate::error::KreuzbergError::parsing(format!("Task join error: {}", e)))??
169
+ } else {
170
+ let content_owned = content.to_vec();
171
+ let cursor = Cursor::new(content_owned);
172
+ zip::ZipArchive::new(cursor)
173
+ .map_err(|e| crate::error::KreuzbergError::parsing(format!("Failed to open ZIP archive: {}", e)))?
174
+ };
175
+
176
+ let mut metadata_map = std::collections::HashMap::new();
177
+
178
+ if let Ok(core) = office_metadata::extract_core_properties(&mut archive) {
179
+ if let Some(title) = core.title {
180
+ metadata_map.insert("title".to_string(), serde_json::Value::String(title));
181
+ }
182
+ if let Some(creator) = core.creator {
183
+ metadata_map.insert(
184
+ "authors".to_string(),
185
+ serde_json::Value::Array(vec![serde_json::Value::String(creator.clone())]),
186
+ );
187
+ metadata_map.insert("created_by".to_string(), serde_json::Value::String(creator));
188
+ }
189
+ if let Some(subject) = core.subject {
190
+ metadata_map.insert("subject".to_string(), serde_json::Value::String(subject));
191
+ }
192
+ if let Some(keywords) = core.keywords {
193
+ metadata_map.insert("keywords".to_string(), serde_json::Value::String(keywords));
194
+ }
195
+ if let Some(description) = core.description {
196
+ metadata_map.insert("description".to_string(), serde_json::Value::String(description));
197
+ }
198
+ if let Some(modified_by) = core.last_modified_by {
199
+ metadata_map.insert("modified_by".to_string(), serde_json::Value::String(modified_by));
200
+ }
201
+ if let Some(created) = core.created {
202
+ metadata_map.insert("created_at".to_string(), serde_json::Value::String(created));
203
+ }
204
+ if let Some(modified) = core.modified {
205
+ metadata_map.insert("modified_at".to_string(), serde_json::Value::String(modified));
206
+ }
207
+ if let Some(revision) = core.revision {
208
+ metadata_map.insert("revision".to_string(), serde_json::Value::String(revision));
209
+ }
210
+ if let Some(category) = core.category {
211
+ metadata_map.insert("category".to_string(), serde_json::Value::String(category));
212
+ }
213
+ if let Some(content_status) = core.content_status {
214
+ metadata_map.insert("content_status".to_string(), serde_json::Value::String(content_status));
215
+ }
216
+ if let Some(language) = core.language {
217
+ metadata_map.insert("language".to_string(), serde_json::Value::String(language));
218
+ }
219
+ }
220
+
221
+ if let Ok(app) = office_metadata::extract_docx_app_properties(&mut archive) {
222
+ if let Some(pages) = app.pages {
223
+ metadata_map.insert("page_count".to_string(), serde_json::Value::Number(pages.into()));
224
+ }
225
+ if let Some(words) = app.words {
226
+ metadata_map.insert("word_count".to_string(), serde_json::Value::Number(words.into()));
227
+ }
228
+ if let Some(chars) = app.characters {
229
+ metadata_map.insert("character_count".to_string(), serde_json::Value::Number(chars.into()));
230
+ }
231
+ if let Some(lines) = app.lines {
232
+ metadata_map.insert("line_count".to_string(), serde_json::Value::Number(lines.into()));
233
+ }
234
+ if let Some(paragraphs) = app.paragraphs {
235
+ metadata_map.insert(
236
+ "paragraph_count".to_string(),
237
+ serde_json::Value::Number(paragraphs.into()),
238
+ );
239
+ }
240
+ if let Some(template) = app.template {
241
+ metadata_map.insert("template".to_string(), serde_json::Value::String(template));
242
+ }
243
+ if let Some(company) = app.company {
244
+ metadata_map.insert("organization".to_string(), serde_json::Value::String(company));
245
+ }
246
+ if let Some(time) = app.total_time {
247
+ metadata_map.insert(
248
+ "total_editing_time_minutes".to_string(),
249
+ serde_json::Value::Number(time.into()),
250
+ );
251
+ }
252
+ if let Some(application) = app.application {
253
+ metadata_map.insert("application".to_string(), serde_json::Value::String(application));
254
+ }
255
+ }
256
+
257
+ if let Ok(custom) = office_metadata::extract_custom_properties(&mut archive) {
258
+ for (key, value) in custom {
259
+ metadata_map.insert(format!("custom_{}", key), value);
260
+ }
261
+ }
262
+
263
+ Ok(ExtractionResult {
264
+ content: text,
265
+ mime_type: mime_type.to_string(),
266
+ metadata: Metadata {
267
+ additional: metadata_map,
268
+ ..Default::default()
269
+ },
270
+ tables,
271
+ detected_languages: None,
272
+ chunks: None,
273
+ images: None,
274
+ })
275
+ }
276
+
277
+ fn supported_mime_types(&self) -> &[&str] {
278
+ &["application/vnd.openxmlformats-officedocument.wordprocessingml.document"]
279
+ }
280
+
281
+ fn priority(&self) -> i32 {
282
+ 50
283
+ }
284
+ }
285
+
286
+ #[cfg(test)]
287
+ mod tests {
288
+ use super::*;
289
+
290
+ #[tokio::test]
291
+ async fn test_docx_extractor_plugin_interface() {
292
+ let extractor = DocxExtractor::new();
293
+ assert_eq!(extractor.name(), "docx-extractor");
294
+ assert_eq!(extractor.version(), env!("CARGO_PKG_VERSION"));
295
+ assert_eq!(extractor.priority(), 50);
296
+ assert_eq!(extractor.supported_mime_types().len(), 1);
297
+ }
298
+
299
+ #[tokio::test]
300
+ async fn test_docx_extractor_supports_docx() {
301
+ let extractor = DocxExtractor::new();
302
+ assert!(
303
+ extractor
304
+ .supported_mime_types()
305
+ .contains(&"application/vnd.openxmlformats-officedocument.wordprocessingml.document")
306
+ );
307
+ }
308
+
309
+ #[tokio::test]
310
+ async fn test_docx_extractor_default() {
311
+ let extractor = DocxExtractor;
312
+ assert_eq!(extractor.name(), "docx-extractor");
313
+ }
314
+
315
+ #[tokio::test]
316
+ async fn test_docx_extractor_initialize_shutdown() {
317
+ let extractor = DocxExtractor::new();
318
+ assert!(extractor.initialize().is_ok());
319
+ assert!(extractor.shutdown().is_ok());
320
+ }
321
+
322
+ #[test]
323
+ fn test_convert_docx_table_to_table() {
324
+ use docx_lite::{Paragraph, Run, Table as DocxTable, TableCell, TableRow};
325
+
326
+ let mut table = DocxTable::new();
327
+
328
+ let mut header_row = TableRow::default();
329
+ let mut cell1 = TableCell::default();
330
+ let mut para1 = Paragraph::new();
331
+ para1.add_run(Run::new("Name".to_string()));
332
+ cell1.paragraphs.push(para1);
333
+ header_row.cells.push(cell1);
334
+
335
+ let mut cell2 = TableCell::default();
336
+ let mut para2 = Paragraph::new();
337
+ para2.add_run(Run::new("Age".to_string()));
338
+ cell2.paragraphs.push(para2);
339
+ header_row.cells.push(cell2);
340
+
341
+ table.rows.push(header_row);
342
+
343
+ let mut data_row = TableRow::default();
344
+ let mut cell3 = TableCell::default();
345
+ let mut para3 = Paragraph::new();
346
+ para3.add_run(Run::new("Alice".to_string()));
347
+ cell3.paragraphs.push(para3);
348
+ data_row.cells.push(cell3);
349
+
350
+ let mut cell4 = TableCell::default();
351
+ let mut para4 = Paragraph::new();
352
+ para4.add_run(Run::new("30".to_string()));
353
+ cell4.paragraphs.push(para4);
354
+ data_row.cells.push(cell4);
355
+
356
+ table.rows.push(data_row);
357
+
358
+ let result = convert_docx_table_to_table(&table, 0);
359
+
360
+ assert_eq!(result.page_number, 1);
361
+ assert_eq!(result.cells.len(), 2);
362
+ assert_eq!(result.cells[0], vec!["Name", "Age"]);
363
+ assert_eq!(result.cells[1], vec!["Alice", "30"]);
364
+ assert!(result.markdown.contains("| Name | Age |"));
365
+ assert!(result.markdown.contains("| Alice | 30 |"));
366
+ }
367
+ }