kreuzberg 4.0.0.pre.rc.29 → 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.
Files changed (321) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +0 -6
  3. data/.rubocop.yaml +534 -1
  4. data/Gemfile +2 -1
  5. data/Gemfile.lock +28 -116
  6. data/README.md +269 -629
  7. data/Rakefile +0 -9
  8. data/Steepfile +4 -8
  9. data/examples/async_patterns.rb +58 -1
  10. data/ext/kreuzberg_rb/extconf.rb +5 -35
  11. data/ext/kreuzberg_rb/native/Cargo.toml +16 -55
  12. data/ext/kreuzberg_rb/native/build.rs +14 -12
  13. data/ext/kreuzberg_rb/native/include/ieeefp.h +1 -1
  14. data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +1 -1
  15. data/ext/kreuzberg_rb/native/include/strings.h +2 -2
  16. data/ext/kreuzberg_rb/native/include/unistd.h +1 -1
  17. data/ext/kreuzberg_rb/native/src/lib.rs +34 -897
  18. data/extconf.rb +6 -38
  19. data/kreuzberg.gemspec +20 -114
  20. data/lib/kreuzberg/api_proxy.rb +18 -2
  21. data/lib/kreuzberg/cache_api.rb +0 -22
  22. data/lib/kreuzberg/cli.rb +10 -2
  23. data/lib/kreuzberg/cli_proxy.rb +10 -0
  24. data/lib/kreuzberg/config.rb +22 -274
  25. data/lib/kreuzberg/errors.rb +7 -73
  26. data/lib/kreuzberg/extraction_api.rb +8 -237
  27. data/lib/kreuzberg/mcp_proxy.rb +11 -2
  28. data/lib/kreuzberg/ocr_backend_protocol.rb +73 -0
  29. data/lib/kreuzberg/post_processor_protocol.rb +71 -0
  30. data/lib/kreuzberg/result.rb +33 -151
  31. data/lib/kreuzberg/setup_lib_path.rb +2 -22
  32. data/lib/kreuzberg/validator_protocol.rb +73 -0
  33. data/lib/kreuzberg/version.rb +1 -1
  34. data/lib/kreuzberg.rb +13 -27
  35. data/pkg/kreuzberg-4.0.0.rc1.gem +0 -0
  36. data/sig/kreuzberg.rbs +12 -105
  37. data/spec/binding/cache_spec.rb +22 -22
  38. data/spec/binding/cli_proxy_spec.rb +4 -2
  39. data/spec/binding/cli_spec.rb +11 -12
  40. data/spec/binding/config_spec.rb +0 -74
  41. data/spec/binding/config_validation_spec.rb +6 -100
  42. data/spec/binding/error_handling_spec.rb +97 -283
  43. data/spec/binding/plugins/ocr_backend_spec.rb +8 -8
  44. data/spec/binding/plugins/postprocessor_spec.rb +11 -11
  45. data/spec/binding/plugins/validator_spec.rb +13 -12
  46. data/spec/examples.txt +104 -0
  47. data/spec/fixtures/config.toml +1 -0
  48. data/spec/fixtures/config.yaml +1 -0
  49. data/spec/fixtures/invalid_config.toml +1 -0
  50. data/spec/smoke/package_spec.rb +3 -2
  51. data/spec/spec_helper.rb +3 -1
  52. data/vendor/kreuzberg/Cargo.toml +67 -192
  53. data/vendor/kreuzberg/README.md +9 -97
  54. data/vendor/kreuzberg/build.rs +194 -516
  55. data/vendor/kreuzberg/src/api/handlers.rs +9 -130
  56. data/vendor/kreuzberg/src/api/mod.rs +3 -18
  57. data/vendor/kreuzberg/src/api/server.rs +71 -236
  58. data/vendor/kreuzberg/src/api/types.rs +7 -43
  59. data/vendor/kreuzberg/src/bin/profile_extract.rs +455 -0
  60. data/vendor/kreuzberg/src/cache/mod.rs +3 -27
  61. data/vendor/kreuzberg/src/chunking/mod.rs +79 -1705
  62. data/vendor/kreuzberg/src/core/batch_mode.rs +0 -60
  63. data/vendor/kreuzberg/src/core/config.rs +23 -905
  64. data/vendor/kreuzberg/src/core/extractor.rs +106 -403
  65. data/vendor/kreuzberg/src/core/io.rs +2 -4
  66. data/vendor/kreuzberg/src/core/mime.rs +12 -2
  67. data/vendor/kreuzberg/src/core/mod.rs +3 -22
  68. data/vendor/kreuzberg/src/core/pipeline.rs +78 -395
  69. data/vendor/kreuzberg/src/embeddings.rs +21 -169
  70. data/vendor/kreuzberg/src/error.rs +2 -2
  71. data/vendor/kreuzberg/src/extraction/archive.rs +31 -36
  72. data/vendor/kreuzberg/src/extraction/docx.rs +1 -365
  73. data/vendor/kreuzberg/src/extraction/email.rs +11 -12
  74. data/vendor/kreuzberg/src/extraction/excel.rs +129 -138
  75. data/vendor/kreuzberg/src/extraction/html.rs +170 -1447
  76. data/vendor/kreuzberg/src/extraction/image.rs +14 -138
  77. data/vendor/kreuzberg/src/extraction/libreoffice.rs +3 -13
  78. data/vendor/kreuzberg/src/extraction/mod.rs +5 -21
  79. data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +0 -2
  80. data/vendor/kreuzberg/src/extraction/pandoc/batch.rs +275 -0
  81. data/vendor/kreuzberg/src/extraction/pandoc/mime_types.rs +178 -0
  82. data/vendor/kreuzberg/src/extraction/pandoc/mod.rs +491 -0
  83. data/vendor/kreuzberg/src/extraction/pandoc/server.rs +496 -0
  84. data/vendor/kreuzberg/src/extraction/pandoc/subprocess.rs +1188 -0
  85. data/vendor/kreuzberg/src/extraction/pandoc/version.rs +162 -0
  86. data/vendor/kreuzberg/src/extraction/pptx.rs +94 -196
  87. data/vendor/kreuzberg/src/extraction/structured.rs +4 -5
  88. data/vendor/kreuzberg/src/extraction/table.rs +1 -2
  89. data/vendor/kreuzberg/src/extraction/text.rs +10 -18
  90. data/vendor/kreuzberg/src/extractors/archive.rs +0 -22
  91. data/vendor/kreuzberg/src/extractors/docx.rs +148 -69
  92. data/vendor/kreuzberg/src/extractors/email.rs +9 -37
  93. data/vendor/kreuzberg/src/extractors/excel.rs +40 -81
  94. data/vendor/kreuzberg/src/extractors/html.rs +173 -182
  95. data/vendor/kreuzberg/src/extractors/image.rs +8 -32
  96. data/vendor/kreuzberg/src/extractors/mod.rs +10 -171
  97. data/vendor/kreuzberg/src/extractors/pandoc.rs +201 -0
  98. data/vendor/kreuzberg/src/extractors/pdf.rs +64 -329
  99. data/vendor/kreuzberg/src/extractors/pptx.rs +34 -79
  100. data/vendor/kreuzberg/src/extractors/structured.rs +0 -16
  101. data/vendor/kreuzberg/src/extractors/text.rs +7 -30
  102. data/vendor/kreuzberg/src/extractors/xml.rs +8 -27
  103. data/vendor/kreuzberg/src/keywords/processor.rs +1 -9
  104. data/vendor/kreuzberg/src/keywords/rake.rs +1 -0
  105. data/vendor/kreuzberg/src/language_detection/mod.rs +51 -94
  106. data/vendor/kreuzberg/src/lib.rs +5 -17
  107. data/vendor/kreuzberg/src/mcp/mod.rs +1 -4
  108. data/vendor/kreuzberg/src/mcp/server.rs +21 -145
  109. data/vendor/kreuzberg/src/ocr/mod.rs +0 -2
  110. data/vendor/kreuzberg/src/ocr/processor.rs +8 -19
  111. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +0 -2
  112. data/vendor/kreuzberg/src/pdf/error.rs +1 -93
  113. data/vendor/kreuzberg/src/pdf/metadata.rs +100 -263
  114. data/vendor/kreuzberg/src/pdf/mod.rs +2 -33
  115. data/vendor/kreuzberg/src/pdf/rendering.rs +12 -12
  116. data/vendor/kreuzberg/src/pdf/table.rs +64 -61
  117. data/vendor/kreuzberg/src/pdf/text.rs +24 -416
  118. data/vendor/kreuzberg/src/plugins/extractor.rs +8 -40
  119. data/vendor/kreuzberg/src/plugins/mod.rs +0 -3
  120. data/vendor/kreuzberg/src/plugins/ocr.rs +14 -22
  121. data/vendor/kreuzberg/src/plugins/processor.rs +1 -10
  122. data/vendor/kreuzberg/src/plugins/registry.rs +0 -15
  123. data/vendor/kreuzberg/src/plugins/validator.rs +8 -20
  124. data/vendor/kreuzberg/src/stopwords/mod.rs +2 -2
  125. data/vendor/kreuzberg/src/text/mod.rs +0 -8
  126. data/vendor/kreuzberg/src/text/quality.rs +15 -28
  127. data/vendor/kreuzberg/src/text/string_utils.rs +10 -22
  128. data/vendor/kreuzberg/src/text/token_reduction/core.rs +50 -86
  129. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +16 -37
  130. data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +1 -2
  131. data/vendor/kreuzberg/src/types.rs +67 -907
  132. data/vendor/kreuzberg/src/utils/mod.rs +0 -14
  133. data/vendor/kreuzberg/src/utils/quality.rs +3 -12
  134. data/vendor/kreuzberg/tests/api_tests.rs +0 -506
  135. data/vendor/kreuzberg/tests/archive_integration.rs +0 -2
  136. data/vendor/kreuzberg/tests/batch_orchestration.rs +12 -57
  137. data/vendor/kreuzberg/tests/batch_processing.rs +8 -32
  138. data/vendor/kreuzberg/tests/chunking_offset_demo.rs +92 -0
  139. data/vendor/kreuzberg/tests/concurrency_stress.rs +8 -40
  140. data/vendor/kreuzberg/tests/config_features.rs +1 -33
  141. data/vendor/kreuzberg/tests/config_loading_tests.rs +39 -16
  142. data/vendor/kreuzberg/tests/core_integration.rs +9 -35
  143. data/vendor/kreuzberg/tests/csv_integration.rs +81 -71
  144. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +25 -23
  145. data/vendor/kreuzberg/tests/email_integration.rs +1 -3
  146. data/vendor/kreuzberg/tests/error_handling.rs +34 -43
  147. data/vendor/kreuzberg/tests/format_integration.rs +1 -7
  148. data/vendor/kreuzberg/tests/helpers/mod.rs +0 -60
  149. data/vendor/kreuzberg/tests/image_integration.rs +0 -2
  150. data/vendor/kreuzberg/tests/mime_detection.rs +16 -17
  151. data/vendor/kreuzberg/tests/ocr_configuration.rs +0 -4
  152. data/vendor/kreuzberg/tests/ocr_errors.rs +0 -22
  153. data/vendor/kreuzberg/tests/ocr_quality.rs +0 -2
  154. data/vendor/kreuzberg/tests/pandoc_integration.rs +503 -0
  155. data/vendor/kreuzberg/tests/pdf_integration.rs +0 -2
  156. data/vendor/kreuzberg/tests/pipeline_integration.rs +2 -36
  157. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +0 -5
  158. data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +1 -17
  159. data/vendor/kreuzberg/tests/plugin_system.rs +0 -6
  160. data/vendor/kreuzberg/tests/registry_integration_tests.rs +22 -2
  161. data/vendor/kreuzberg/tests/security_validation.rs +1 -13
  162. data/vendor/kreuzberg/tests/test_fastembed.rs +23 -45
  163. metadata +25 -171
  164. data/.rubocop.yml +0 -543
  165. data/ext/kreuzberg_rb/native/.cargo/config.toml +0 -23
  166. data/ext/kreuzberg_rb/native/Cargo.lock +0 -7619
  167. data/lib/kreuzberg/error_context.rb +0 -136
  168. data/lib/kreuzberg/types.rb +0 -170
  169. data/lib/libpdfium.so +0 -0
  170. data/spec/binding/async_operations_spec.rb +0 -473
  171. data/spec/binding/batch_operations_spec.rb +0 -595
  172. data/spec/binding/batch_spec.rb +0 -359
  173. data/spec/binding/config_result_spec.rb +0 -377
  174. data/spec/binding/embeddings_spec.rb +0 -816
  175. data/spec/binding/error_recovery_spec.rb +0 -488
  176. data/spec/binding/font_config_spec.rb +0 -220
  177. data/spec/binding/images_spec.rb +0 -738
  178. data/spec/binding/keywords_extraction_spec.rb +0 -600
  179. data/spec/binding/metadata_types_spec.rb +0 -1228
  180. data/spec/binding/pages_extraction_spec.rb +0 -471
  181. data/spec/binding/tables_spec.rb +0 -641
  182. data/spec/unit/config/chunking_config_spec.rb +0 -213
  183. data/spec/unit/config/embedding_config_spec.rb +0 -343
  184. data/spec/unit/config/extraction_config_spec.rb +0 -438
  185. data/spec/unit/config/font_config_spec.rb +0 -285
  186. data/spec/unit/config/hierarchy_config_spec.rb +0 -314
  187. data/spec/unit/config/image_extraction_config_spec.rb +0 -209
  188. data/spec/unit/config/image_preprocessing_config_spec.rb +0 -249
  189. data/spec/unit/config/keyword_config_spec.rb +0 -229
  190. data/spec/unit/config/language_detection_config_spec.rb +0 -258
  191. data/spec/unit/config/ocr_config_spec.rb +0 -171
  192. data/spec/unit/config/page_config_spec.rb +0 -221
  193. data/spec/unit/config/pdf_config_spec.rb +0 -267
  194. data/spec/unit/config/postprocessor_config_spec.rb +0 -290
  195. data/spec/unit/config/tesseract_config_spec.rb +0 -181
  196. data/spec/unit/config/token_reduction_config_spec.rb +0 -251
  197. data/test/metadata_types_test.rb +0 -959
  198. data/vendor/Cargo.toml +0 -61
  199. data/vendor/kreuzberg/examples/bench_fixes.rs +0 -71
  200. data/vendor/kreuzberg/examples/test_pdfium_fork.rs +0 -62
  201. data/vendor/kreuzberg/src/chunking/processor.rs +0 -219
  202. data/vendor/kreuzberg/src/core/batch_optimizations.rs +0 -385
  203. data/vendor/kreuzberg/src/core/config_validation.rs +0 -949
  204. data/vendor/kreuzberg/src/core/formats.rs +0 -235
  205. data/vendor/kreuzberg/src/core/server_config.rs +0 -1220
  206. data/vendor/kreuzberg/src/extraction/capacity.rs +0 -263
  207. data/vendor/kreuzberg/src/extraction/markdown.rs +0 -216
  208. data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +0 -284
  209. data/vendor/kreuzberg/src/extractors/bibtex.rs +0 -470
  210. data/vendor/kreuzberg/src/extractors/docbook.rs +0 -504
  211. data/vendor/kreuzberg/src/extractors/epub.rs +0 -696
  212. data/vendor/kreuzberg/src/extractors/fictionbook.rs +0 -492
  213. data/vendor/kreuzberg/src/extractors/jats.rs +0 -1054
  214. data/vendor/kreuzberg/src/extractors/jupyter.rs +0 -368
  215. data/vendor/kreuzberg/src/extractors/latex.rs +0 -653
  216. data/vendor/kreuzberg/src/extractors/markdown.rs +0 -701
  217. data/vendor/kreuzberg/src/extractors/odt.rs +0 -628
  218. data/vendor/kreuzberg/src/extractors/opml.rs +0 -635
  219. data/vendor/kreuzberg/src/extractors/orgmode.rs +0 -529
  220. data/vendor/kreuzberg/src/extractors/rst.rs +0 -577
  221. data/vendor/kreuzberg/src/extractors/rtf.rs +0 -809
  222. data/vendor/kreuzberg/src/extractors/security.rs +0 -484
  223. data/vendor/kreuzberg/src/extractors/security_tests.rs +0 -367
  224. data/vendor/kreuzberg/src/extractors/typst.rs +0 -651
  225. data/vendor/kreuzberg/src/language_detection/processor.rs +0 -218
  226. data/vendor/kreuzberg/src/ocr/language_registry.rs +0 -520
  227. data/vendor/kreuzberg/src/panic_context.rs +0 -154
  228. data/vendor/kreuzberg/src/pdf/bindings.rs +0 -306
  229. data/vendor/kreuzberg/src/pdf/bundled.rs +0 -408
  230. data/vendor/kreuzberg/src/pdf/fonts.rs +0 -358
  231. data/vendor/kreuzberg/src/pdf/hierarchy.rs +0 -903
  232. data/vendor/kreuzberg/src/text/quality_processor.rs +0 -231
  233. data/vendor/kreuzberg/src/text/utf8_validation.rs +0 -193
  234. data/vendor/kreuzberg/src/utils/pool.rs +0 -503
  235. data/vendor/kreuzberg/src/utils/pool_sizing.rs +0 -364
  236. data/vendor/kreuzberg/src/utils/string_pool.rs +0 -761
  237. data/vendor/kreuzberg/tests/api_embed.rs +0 -360
  238. data/vendor/kreuzberg/tests/api_extract_multipart.rs +0 -52
  239. data/vendor/kreuzberg/tests/api_large_pdf_extraction.rs +0 -471
  240. data/vendor/kreuzberg/tests/api_large_pdf_extraction_diagnostics.rs +0 -289
  241. data/vendor/kreuzberg/tests/batch_pooling_benchmark.rs +0 -154
  242. data/vendor/kreuzberg/tests/bibtex_parity_test.rs +0 -421
  243. data/vendor/kreuzberg/tests/config_integration_test.rs +0 -753
  244. data/vendor/kreuzberg/tests/data/hierarchy_ground_truth.json +0 -294
  245. data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +0 -500
  246. data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +0 -370
  247. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +0 -275
  248. data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +0 -228
  249. data/vendor/kreuzberg/tests/html_table_test.rs +0 -551
  250. data/vendor/kreuzberg/tests/instrumentation_test.rs +0 -139
  251. data/vendor/kreuzberg/tests/jats_extractor_tests.rs +0 -639
  252. data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +0 -704
  253. data/vendor/kreuzberg/tests/latex_extractor_tests.rs +0 -496
  254. data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +0 -490
  255. data/vendor/kreuzberg/tests/ocr_language_registry.rs +0 -191
  256. data/vendor/kreuzberg/tests/odt_extractor_tests.rs +0 -674
  257. data/vendor/kreuzberg/tests/opml_extractor_tests.rs +0 -616
  258. data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +0 -822
  259. data/vendor/kreuzberg/tests/page_markers.rs +0 -297
  260. data/vendor/kreuzberg/tests/pdf_hierarchy_detection.rs +0 -301
  261. data/vendor/kreuzberg/tests/pdf_hierarchy_quality.rs +0 -589
  262. data/vendor/kreuzberg/tests/pdf_ocr_triggering.rs +0 -301
  263. data/vendor/kreuzberg/tests/pdf_text_merging.rs +0 -475
  264. data/vendor/kreuzberg/tests/pdfium_linking.rs +0 -340
  265. data/vendor/kreuzberg/tests/rst_extractor_tests.rs +0 -694
  266. data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +0 -775
  267. data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +0 -1260
  268. data/vendor/kreuzberg/tests/typst_extractor_tests.rs +0 -648
  269. data/vendor/kreuzberg-ffi/Cargo.toml +0 -67
  270. data/vendor/kreuzberg-ffi/README.md +0 -851
  271. data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +0 -227
  272. data/vendor/kreuzberg-ffi/build.rs +0 -168
  273. data/vendor/kreuzberg-ffi/cbindgen.toml +0 -37
  274. data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +0 -12
  275. data/vendor/kreuzberg-ffi/kreuzberg.h +0 -3012
  276. data/vendor/kreuzberg-ffi/src/batch_streaming.rs +0 -588
  277. data/vendor/kreuzberg-ffi/src/config.rs +0 -1341
  278. data/vendor/kreuzberg-ffi/src/error.rs +0 -901
  279. data/vendor/kreuzberg-ffi/src/extraction.rs +0 -555
  280. data/vendor/kreuzberg-ffi/src/helpers.rs +0 -879
  281. data/vendor/kreuzberg-ffi/src/lib.rs +0 -977
  282. data/vendor/kreuzberg-ffi/src/memory.rs +0 -493
  283. data/vendor/kreuzberg-ffi/src/mime.rs +0 -329
  284. data/vendor/kreuzberg-ffi/src/panic_shield.rs +0 -265
  285. data/vendor/kreuzberg-ffi/src/plugins/document_extractor.rs +0 -442
  286. data/vendor/kreuzberg-ffi/src/plugins/mod.rs +0 -14
  287. data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +0 -628
  288. data/vendor/kreuzberg-ffi/src/plugins/post_processor.rs +0 -438
  289. data/vendor/kreuzberg-ffi/src/plugins/validator.rs +0 -329
  290. data/vendor/kreuzberg-ffi/src/result.rs +0 -510
  291. data/vendor/kreuzberg-ffi/src/result_pool.rs +0 -639
  292. data/vendor/kreuzberg-ffi/src/result_view.rs +0 -773
  293. data/vendor/kreuzberg-ffi/src/string_intern.rs +0 -568
  294. data/vendor/kreuzberg-ffi/src/types.rs +0 -363
  295. data/vendor/kreuzberg-ffi/src/util.rs +0 -210
  296. data/vendor/kreuzberg-ffi/src/validation.rs +0 -848
  297. data/vendor/kreuzberg-ffi/tests.disabled/README.md +0 -48
  298. data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +0 -299
  299. data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +0 -346
  300. data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +0 -232
  301. data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +0 -470
  302. data/vendor/kreuzberg-tesseract/.commitlintrc.json +0 -13
  303. data/vendor/kreuzberg-tesseract/.crate-ignore +0 -2
  304. data/vendor/kreuzberg-tesseract/Cargo.lock +0 -2933
  305. data/vendor/kreuzberg-tesseract/Cargo.toml +0 -57
  306. data/vendor/kreuzberg-tesseract/LICENSE +0 -22
  307. data/vendor/kreuzberg-tesseract/README.md +0 -399
  308. data/vendor/kreuzberg-tesseract/build.rs +0 -1127
  309. data/vendor/kreuzberg-tesseract/patches/README.md +0 -71
  310. data/vendor/kreuzberg-tesseract/patches/tesseract.diff +0 -199
  311. data/vendor/kreuzberg-tesseract/src/api.rs +0 -1371
  312. data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +0 -77
  313. data/vendor/kreuzberg-tesseract/src/enums.rs +0 -297
  314. data/vendor/kreuzberg-tesseract/src/error.rs +0 -81
  315. data/vendor/kreuzberg-tesseract/src/lib.rs +0 -145
  316. data/vendor/kreuzberg-tesseract/src/monitor.rs +0 -57
  317. data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +0 -197
  318. data/vendor/kreuzberg-tesseract/src/page_iterator.rs +0 -253
  319. data/vendor/kreuzberg-tesseract/src/result_iterator.rs +0 -286
  320. data/vendor/kreuzberg-tesseract/src/result_renderer.rs +0 -183
  321. data/vendor/kreuzberg-tesseract/tests/integration_test.rs +0 -211
@@ -1,879 +0,0 @@
1
- //! Helper utilities for FFI operations.
2
- //!
3
- //! This module contains shared helper functions for error handling, string conversion,
4
- //! and type conversion between Rust and C types.
5
-
6
- use std::cell::RefCell;
7
- use std::ffi::CString;
8
- use std::os::raw::c_char;
9
- use std::ptr;
10
-
11
- use kreuzberg::core::config::ExtractionConfig;
12
- use kreuzberg::types::ExtractionResult;
13
-
14
- use crate::panic_shield::{ErrorCode, StructuredError, clear_structured_error, set_structured_error};
15
- use crate::types::{CExtractionResult, CStringGuard};
16
-
17
- /// Type alias for FFI results that can fail with a String error message
18
- pub type FfiResult<T> = std::result::Result<T, String>;
19
-
20
- // Thread-local storage for the last error message (for backward compatibility)
21
- thread_local! {
22
- pub(crate) static LAST_ERROR_C_STRING: RefCell<Option<CString>> = const { RefCell::new(None) };
23
- }
24
-
25
- /// Set the last error message (convenience wrapper for backward compatibility)
26
- pub fn set_last_error(err: String) {
27
- if let Ok(c_str) = CString::new(err.clone()) {
28
- LAST_ERROR_C_STRING.with(|last| *last.borrow_mut() = Some(c_str));
29
- }
30
-
31
- let structured_err = StructuredError::from_message(err, ErrorCode::GenericError);
32
- set_structured_error(structured_err);
33
- }
34
-
35
- /// Clear the last error message
36
- pub fn clear_last_error() {
37
- LAST_ERROR_C_STRING.with(|last| *last.borrow_mut() = None);
38
- clear_structured_error();
39
- }
40
-
41
- /// Convert a Rust String to a C string pointer
42
- pub fn string_to_c_string(value: String) -> std::result::Result<*mut c_char, String> {
43
- CString::new(value)
44
- .map(CString::into_raw)
45
- .map_err(|e| format!("Failed to create C string: {}", e))
46
- }
47
-
48
- /// Parse extraction configuration from JSON string
49
- pub fn parse_extraction_config_from_json(config_str: &str) -> FfiResult<ExtractionConfig> {
50
- use html_to_markdown_rs::options::{
51
- CodeBlockStyle, ConversionOptions, HeadingStyle, HighlightStyle, ListIndentType, NewlineStyle,
52
- PreprocessingPreset, WhitespaceMode,
53
- };
54
-
55
- fn parse_enum<T, F>(value: Option<&serde_json::Value>, parse_fn: F) -> FfiResult<Option<T>>
56
- where
57
- F: Fn(&str) -> std::result::Result<T, String>,
58
- {
59
- if let Some(raw) = value {
60
- let text = raw
61
- .as_str()
62
- .ok_or_else(|| "Expected string for html_options enum field".to_string())?;
63
- return parse_fn(text).map(Some);
64
- }
65
- Ok(None)
66
- }
67
-
68
- fn parse_heading_style(value: &str) -> FfiResult<HeadingStyle> {
69
- match value.to_lowercase().as_str() {
70
- "atx" => Ok(HeadingStyle::Atx),
71
- "underlined" => Ok(HeadingStyle::Underlined),
72
- "atx_closed" => Ok(HeadingStyle::AtxClosed),
73
- other => Err(format!(
74
- "Invalid heading_style '{}'. Expected one of: atx, underlined, atx_closed",
75
- other
76
- )),
77
- }
78
- }
79
-
80
- fn parse_list_indent_type(value: &str) -> FfiResult<ListIndentType> {
81
- match value.to_lowercase().as_str() {
82
- "spaces" => Ok(ListIndentType::Spaces),
83
- "tabs" => Ok(ListIndentType::Tabs),
84
- other => Err(format!(
85
- "Invalid list_indent_type '{}'. Expected 'spaces' or 'tabs'",
86
- other
87
- )),
88
- }
89
- }
90
-
91
- fn parse_highlight_style(value: &str) -> FfiResult<HighlightStyle> {
92
- match value.to_lowercase().as_str() {
93
- "double_equal" | "==" | "highlight" => Ok(HighlightStyle::DoubleEqual),
94
- "html" => Ok(HighlightStyle::Html),
95
- "bold" => Ok(HighlightStyle::Bold),
96
- "none" => Ok(HighlightStyle::None),
97
- other => Err(format!(
98
- "Invalid highlight_style '{}'. Expected one of: double_equal, html, bold, none",
99
- other
100
- )),
101
- }
102
- }
103
-
104
- fn parse_whitespace_mode(value: &str) -> FfiResult<WhitespaceMode> {
105
- match value.to_lowercase().as_str() {
106
- "normalized" => Ok(WhitespaceMode::Normalized),
107
- "strict" => Ok(WhitespaceMode::Strict),
108
- other => Err(format!(
109
- "Invalid whitespace_mode '{}'. Expected 'normalized' or 'strict'",
110
- other
111
- )),
112
- }
113
- }
114
-
115
- fn parse_newline_style(value: &str) -> FfiResult<NewlineStyle> {
116
- match value.to_lowercase().as_str() {
117
- "spaces" => Ok(NewlineStyle::Spaces),
118
- "backslash" => Ok(NewlineStyle::Backslash),
119
- other => Err(format!(
120
- "Invalid newline_style '{}'. Expected 'spaces' or 'backslash'",
121
- other
122
- )),
123
- }
124
- }
125
-
126
- fn parse_code_block_style(value: &str) -> FfiResult<CodeBlockStyle> {
127
- match value.to_lowercase().as_str() {
128
- "indented" => Ok(CodeBlockStyle::Indented),
129
- "backticks" => Ok(CodeBlockStyle::Backticks),
130
- "tildes" => Ok(CodeBlockStyle::Tildes),
131
- other => Err(format!(
132
- "Invalid code_block_style '{}'. Expected 'indented', 'backticks', or 'tildes'",
133
- other
134
- )),
135
- }
136
- }
137
-
138
- fn parse_preprocessing_preset(value: &str) -> FfiResult<PreprocessingPreset> {
139
- match value.to_lowercase().as_str() {
140
- "minimal" => Ok(PreprocessingPreset::Minimal),
141
- "standard" => Ok(PreprocessingPreset::Standard),
142
- "aggressive" => Ok(PreprocessingPreset::Aggressive),
143
- other => Err(format!(
144
- "Invalid preprocessing.preset '{}'. Expected one of: minimal, standard, aggressive",
145
- other
146
- )),
147
- }
148
- }
149
-
150
- fn parse_html_options(value: &serde_json::Value) -> FfiResult<ConversionOptions> {
151
- let mut opts = ConversionOptions::default();
152
- let obj = value
153
- .as_object()
154
- .ok_or_else(|| "html_options must be an object".to_string())?;
155
-
156
- if let Some(val) = obj.get("heading_style") {
157
- opts.heading_style = parse_enum(Some(val), parse_heading_style)?.unwrap_or(opts.heading_style);
158
- }
159
-
160
- if let Some(val) = obj.get("list_indent_type") {
161
- opts.list_indent_type = parse_enum(Some(val), parse_list_indent_type)?.unwrap_or(opts.list_indent_type);
162
- }
163
-
164
- if let Some(val) = obj.get("list_indent_width") {
165
- opts.list_indent_width = val
166
- .as_u64()
167
- .map(|v| v as usize)
168
- .ok_or_else(|| "list_indent_width must be an integer".to_string())?;
169
- }
170
-
171
- if let Some(val) = obj.get("bullets") {
172
- opts.bullets = val
173
- .as_str()
174
- .map(str::to_string)
175
- .ok_or_else(|| "bullets must be a string".to_string())?;
176
- }
177
-
178
- if let Some(val) = obj.get("strong_em_symbol") {
179
- let symbol = val
180
- .as_str()
181
- .ok_or_else(|| "strong_em_symbol must be a string".to_string())?;
182
- let mut chars = symbol.chars();
183
- opts.strong_em_symbol = chars
184
- .next()
185
- .ok_or_else(|| "strong_em_symbol must not be empty".to_string())?;
186
- }
187
-
188
- if let Some(val) = obj.get("escape_asterisks") {
189
- opts.escape_asterisks = val
190
- .as_bool()
191
- .ok_or_else(|| "escape_asterisks must be a boolean".to_string())?;
192
- }
193
- if let Some(val) = obj.get("escape_underscores") {
194
- opts.escape_underscores = val
195
- .as_bool()
196
- .ok_or_else(|| "escape_underscores must be a boolean".to_string())?;
197
- }
198
- if let Some(val) = obj.get("escape_misc") {
199
- opts.escape_misc = val
200
- .as_bool()
201
- .ok_or_else(|| "escape_misc must be a boolean".to_string())?;
202
- }
203
- if let Some(val) = obj.get("escape_ascii") {
204
- opts.escape_ascii = val
205
- .as_bool()
206
- .ok_or_else(|| "escape_ascii must be a boolean".to_string())?;
207
- }
208
-
209
- if let Some(val) = obj.get("code_language") {
210
- opts.code_language = val
211
- .as_str()
212
- .map(str::to_string)
213
- .ok_or_else(|| "code_language must be a string".to_string())?;
214
- }
215
-
216
- if let Some(val) = obj.get("autolinks") {
217
- opts.autolinks = val.as_bool().ok_or_else(|| "autolinks must be a boolean".to_string())?;
218
- }
219
-
220
- if let Some(val) = obj.get("default_title") {
221
- opts.default_title = val
222
- .as_bool()
223
- .ok_or_else(|| "default_title must be a boolean".to_string())?;
224
- }
225
-
226
- if let Some(val) = obj.get("br_in_tables") {
227
- opts.br_in_tables = val
228
- .as_bool()
229
- .ok_or_else(|| "br_in_tables must be a boolean".to_string())?;
230
- }
231
-
232
- if let Some(val) = obj.get("hocr_spatial_tables") {
233
- opts.hocr_spatial_tables = val
234
- .as_bool()
235
- .ok_or_else(|| "hocr_spatial_tables must be a boolean".to_string())?;
236
- }
237
-
238
- if let Some(val) = obj.get("highlight_style") {
239
- opts.highlight_style = parse_enum(Some(val), parse_highlight_style)?.unwrap_or(opts.highlight_style);
240
- }
241
-
242
- if let Some(val) = obj.get("extract_metadata") {
243
- opts.extract_metadata = val
244
- .as_bool()
245
- .ok_or_else(|| "extract_metadata must be a boolean".to_string())?;
246
- }
247
-
248
- if let Some(val) = obj.get("whitespace_mode") {
249
- opts.whitespace_mode = parse_enum(Some(val), parse_whitespace_mode)?.unwrap_or(opts.whitespace_mode);
250
- }
251
-
252
- if let Some(val) = obj.get("strip_newlines") {
253
- opts.strip_newlines = val
254
- .as_bool()
255
- .ok_or_else(|| "strip_newlines must be a boolean".to_string())?;
256
- }
257
-
258
- if let Some(val) = obj.get("wrap") {
259
- opts.wrap = val.as_bool().ok_or_else(|| "wrap must be a boolean".to_string())?;
260
- }
261
-
262
- if let Some(val) = obj.get("wrap_width") {
263
- opts.wrap_width = val
264
- .as_u64()
265
- .map(|v| v as usize)
266
- .ok_or_else(|| "wrap_width must be an integer".to_string())?;
267
- }
268
-
269
- if let Some(val) = obj.get("convert_as_inline") {
270
- opts.convert_as_inline = val
271
- .as_bool()
272
- .ok_or_else(|| "convert_as_inline must be a boolean".to_string())?;
273
- }
274
-
275
- if let Some(val) = obj.get("sub_symbol") {
276
- opts.sub_symbol = val
277
- .as_str()
278
- .map(str::to_string)
279
- .ok_or_else(|| "sub_symbol must be a string".to_string())?;
280
- }
281
-
282
- if let Some(val) = obj.get("sup_symbol") {
283
- opts.sup_symbol = val
284
- .as_str()
285
- .map(str::to_string)
286
- .ok_or_else(|| "sup_symbol must be a string".to_string())?;
287
- }
288
-
289
- if let Some(val) = obj.get("newline_style") {
290
- opts.newline_style = parse_enum(Some(val), parse_newline_style)?.unwrap_or(opts.newline_style);
291
- }
292
-
293
- if let Some(val) = obj.get("code_block_style") {
294
- opts.code_block_style = parse_enum(Some(val), parse_code_block_style)?.unwrap_or(opts.code_block_style);
295
- }
296
-
297
- if let Some(val) = obj.get("keep_inline_images_in") {
298
- opts.keep_inline_images_in = val
299
- .as_array()
300
- .ok_or_else(|| "keep_inline_images_in must be an array".to_string())?
301
- .iter()
302
- .map(|v| {
303
- v.as_str()
304
- .map(str::to_string)
305
- .ok_or_else(|| "keep_inline_images_in entries must be strings".to_string())
306
- })
307
- .collect::<std::result::Result<Vec<_>, _>>()?;
308
- }
309
-
310
- if let Some(val) = obj.get("encoding") {
311
- opts.encoding = val
312
- .as_str()
313
- .map(str::to_string)
314
- .ok_or_else(|| "encoding must be a string".to_string())?;
315
- }
316
-
317
- if let Some(val) = obj.get("debug") {
318
- opts.debug = val.as_bool().ok_or_else(|| "debug must be a boolean".to_string())?;
319
- }
320
-
321
- if let Some(val) = obj.get("strip_tags") {
322
- opts.strip_tags = val
323
- .as_array()
324
- .ok_or_else(|| "strip_tags must be an array".to_string())?
325
- .iter()
326
- .map(|v| {
327
- v.as_str()
328
- .map(str::to_string)
329
- .ok_or_else(|| "strip_tags entries must be strings".to_string())
330
- })
331
- .collect::<std::result::Result<Vec<_>, _>>()?;
332
- }
333
-
334
- if let Some(val) = obj.get("preserve_tags") {
335
- opts.preserve_tags = val
336
- .as_array()
337
- .ok_or_else(|| "preserve_tags must be an array".to_string())?
338
- .iter()
339
- .map(|v| {
340
- v.as_str()
341
- .map(str::to_string)
342
- .ok_or_else(|| "preserve_tags entries must be strings".to_string())
343
- })
344
- .collect::<std::result::Result<Vec<_>, _>>()?;
345
- }
346
-
347
- if let Some(val) = obj.get("preprocessing") {
348
- let pre = val
349
- .as_object()
350
- .ok_or_else(|| "preprocessing must be an object".to_string())?;
351
- let mut preprocessing = opts.preprocessing.clone();
352
-
353
- if let Some(v) = pre.get("enabled") {
354
- preprocessing.enabled = v
355
- .as_bool()
356
- .ok_or_else(|| "preprocessing.enabled must be a boolean".to_string())?;
357
- }
358
-
359
- if let Some(v) = pre.get("preset") {
360
- let preset = v
361
- .as_str()
362
- .ok_or_else(|| "preprocessing.preset must be a string".to_string())?;
363
- preprocessing.preset = parse_preprocessing_preset(preset)?;
364
- }
365
-
366
- if let Some(v) = pre.get("remove_navigation") {
367
- preprocessing.remove_navigation = v
368
- .as_bool()
369
- .ok_or_else(|| "preprocessing.remove_navigation must be a boolean".to_string())?;
370
- }
371
-
372
- if let Some(v) = pre.get("remove_forms") {
373
- preprocessing.remove_forms = v
374
- .as_bool()
375
- .ok_or_else(|| "preprocessing.remove_forms must be a boolean".to_string())?;
376
- }
377
-
378
- opts.preprocessing = preprocessing;
379
- }
380
-
381
- Ok(opts)
382
- }
383
-
384
- let value: serde_json::Value =
385
- serde_json::from_str(config_str).map_err(|e| format!("Failed to parse config JSON: {}", e))?;
386
-
387
- let html_options = value.get("html_options").map(parse_html_options).transpose()?;
388
-
389
- let mut config: ExtractionConfig =
390
- serde_json::from_value(value).map_err(|e| format!("Failed to parse config JSON: {}", e))?;
391
-
392
- if let Some(options) = html_options {
393
- config.html_options = Some(options);
394
- }
395
-
396
- Ok(config)
397
- }
398
-
399
- /// Convert a Rust ExtractionResult to a C-compatible CExtractionResult
400
- pub fn to_c_extraction_result(result: ExtractionResult) -> std::result::Result<*mut CExtractionResult, String> {
401
- let ExtractionResult {
402
- content,
403
- mime_type,
404
- metadata,
405
- tables,
406
- detected_languages,
407
- chunks,
408
- images,
409
- pages,
410
- } = result;
411
-
412
- let sanitized_content = if content.contains('\0') {
413
- content.replace('\0', "\u{FFFD}")
414
- } else {
415
- content
416
- };
417
-
418
- let content_guard = CStringGuard::new(
419
- CString::new(sanitized_content).map_err(|e| format!("Failed to convert content to C string: {}", e))?,
420
- );
421
-
422
- let mime_type_guard = CStringGuard::new(
423
- CString::new(mime_type).map_err(|e| format!("Failed to convert MIME type to C string: {}", e))?,
424
- );
425
-
426
- let language_guard = match &metadata.language {
427
- Some(lang) => Some(CStringGuard::new(
428
- CString::new(lang.as_str()).map_err(|e| format!("Failed to convert language to C string: {}", e))?,
429
- )),
430
- None => None,
431
- };
432
-
433
- let date_guard = match &metadata.created_at {
434
- Some(d) => {
435
- Some(CStringGuard::new(CString::new(d.as_str()).map_err(|e| {
436
- format!("Failed to convert created_at to C string: {}", e)
437
- })?))
438
- }
439
- None => None,
440
- };
441
-
442
- let subject_guard = match &metadata.subject {
443
- Some(subj) => Some(CStringGuard::new(
444
- CString::new(subj.as_str()).map_err(|e| format!("Failed to convert subject to C string: {}", e))?,
445
- )),
446
- None => None,
447
- };
448
-
449
- let tables_json_guard = if !tables.is_empty() {
450
- let json = serde_json::to_string(&tables).map_err(|e| format!("Failed to serialize tables to JSON: {}", e))?;
451
- Some(CStringGuard::new(CString::new(json).map_err(|e| {
452
- format!("Failed to convert tables JSON to C string: {}", e)
453
- })?))
454
- } else {
455
- None
456
- };
457
-
458
- let detected_languages_json_guard = match detected_languages {
459
- Some(langs) if !langs.is_empty() => {
460
- let json = serde_json::to_string(&langs)
461
- .map_err(|e| format!("Failed to serialize detected languages to JSON: {}", e))?;
462
- Some(CStringGuard::new(CString::new(json).map_err(|e| {
463
- format!("Failed to convert detected languages JSON to C string: {}", e)
464
- })?))
465
- }
466
- _ => None,
467
- };
468
-
469
- let metadata_json_guard = {
470
- let json =
471
- serde_json::to_string(&metadata).map_err(|e| format!("Failed to serialize metadata to JSON: {}", e))?;
472
- Some(CStringGuard::new(CString::new(json).map_err(|e| {
473
- format!("Failed to convert metadata JSON to C string: {}", e)
474
- })?))
475
- };
476
-
477
- let chunks_json_guard = match chunks {
478
- Some(chunks) if !chunks.is_empty() => {
479
- let json =
480
- serde_json::to_string(&chunks).map_err(|e| format!("Failed to serialize chunks to JSON: {}", e))?;
481
- Some(CStringGuard::new(CString::new(json).map_err(|e| {
482
- format!("Failed to convert chunks JSON to C string: {}", e)
483
- })?))
484
- }
485
- _ => None,
486
- };
487
-
488
- let images_json_guard = match images {
489
- Some(images) if !images.is_empty() => {
490
- let json =
491
- serde_json::to_string(&images).map_err(|e| format!("Failed to serialize images to JSON: {}", e))?;
492
- Some(CStringGuard::new(CString::new(json).map_err(|e| {
493
- format!("Failed to convert images JSON to C string: {}", e)
494
- })?))
495
- }
496
- _ => None,
497
- };
498
-
499
- let page_structure_json_guard = match &metadata.pages {
500
- Some(page_structure) => {
501
- let json = serde_json::to_string(&page_structure)
502
- .map_err(|e| format!("Failed to serialize page structure to JSON: {}", e))?;
503
- Some(CStringGuard::new(CString::new(json).map_err(|e| {
504
- format!("Failed to convert page structure JSON to C string: {}", e)
505
- })?))
506
- }
507
- _ => None,
508
- };
509
-
510
- let pages_json_guard = match pages {
511
- Some(pages) if !pages.is_empty() => {
512
- let json =
513
- serde_json::to_string(&pages).map_err(|e| format!("Failed to serialize pages to JSON: {}", e))?;
514
- Some(CStringGuard::new(CString::new(json).map_err(|e| {
515
- format!("Failed to convert pages JSON to C string: {}", e)
516
- })?))
517
- }
518
- _ => None,
519
- };
520
-
521
- Ok(Box::into_raw(Box::new(CExtractionResult {
522
- content: content_guard.into_raw(),
523
- mime_type: mime_type_guard.into_raw(),
524
- language: language_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
525
- date: date_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
526
- subject: subject_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
527
- tables_json: tables_json_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
528
- detected_languages_json: detected_languages_json_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
529
- metadata_json: metadata_json_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
530
- chunks_json: chunks_json_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
531
- images_json: images_json_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
532
- page_structure_json: page_structure_json_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
533
- pages_json: pages_json_guard.map_or(ptr::null_mut(), |g| g.into_raw()),
534
- success: true,
535
- _padding1: [0u8; 7],
536
- })))
537
- }
538
-
539
- #[cfg(test)]
540
- mod tests {
541
- use super::*;
542
- use kreuzberg::types::{Chunk, ChunkMetadata, ExtractionResult, Metadata, Table};
543
- use std::ffi::CStr;
544
-
545
- #[test]
546
- fn test_set_and_clear_error() {
547
- // Test that error functions don't panic
548
- set_last_error("test error".to_string());
549
- clear_last_error();
550
- }
551
-
552
- #[test]
553
- fn test_string_to_c_string_success() {
554
- let result = string_to_c_string("hello world".to_string());
555
- assert!(result.is_ok());
556
-
557
- // Clean up the allocated string
558
- if let Ok(ptr) = result {
559
- unsafe {
560
- let _ = CString::from_raw(ptr);
561
- }
562
- }
563
- }
564
-
565
- #[test]
566
- fn test_string_to_c_string_with_null_byte() {
567
- let result = string_to_c_string("hello\0world".to_string());
568
- assert!(result.is_err());
569
- assert!(result.unwrap_err().contains("Failed to create C string"));
570
- }
571
-
572
- #[test]
573
- fn test_parse_extraction_config_empty_json() {
574
- let result = parse_extraction_config_from_json("{}");
575
- assert!(result.is_ok());
576
- let config = result.unwrap();
577
- // Default config should be created
578
- assert!(config.html_options.is_none());
579
- }
580
-
581
- #[test]
582
- fn test_parse_extraction_config_with_basic_options() {
583
- let json = r#"{
584
- "use_cache": true,
585
- "enable_quality_processing": false
586
- }"#;
587
-
588
- let result = parse_extraction_config_from_json(json);
589
- assert!(result.is_ok());
590
- let config = result.unwrap();
591
- assert!(config.use_cache);
592
- assert!(!config.enable_quality_processing);
593
- }
594
-
595
- #[test]
596
- fn test_parse_extraction_config_with_html_options() {
597
- let json = r#"{
598
- "html_options": {
599
- "heading_style": "atx",
600
- "escape_asterisks": true,
601
- "autolinks": false
602
- }
603
- }"#;
604
-
605
- let result = parse_extraction_config_from_json(json);
606
- assert!(result.is_ok());
607
- let config = result.unwrap();
608
- assert!(config.html_options.is_some());
609
-
610
- let html_opts = config.html_options.unwrap();
611
- assert!(html_opts.escape_asterisks);
612
- assert!(!html_opts.autolinks);
613
- }
614
-
615
- #[test]
616
- fn test_parse_extraction_config_invalid_json() {
617
- let result = parse_extraction_config_from_json("not valid json");
618
- assert!(result.is_err());
619
- assert!(result.unwrap_err().contains("Failed to parse config JSON"));
620
- }
621
-
622
- #[test]
623
- fn test_parse_extraction_config_invalid_heading_style() {
624
- let json = r#"{
625
- "html_options": {
626
- "heading_style": "invalid_style"
627
- }
628
- }"#;
629
-
630
- let result = parse_extraction_config_from_json(json);
631
- assert!(result.is_err());
632
- assert!(result.unwrap_err().contains("Invalid heading_style"));
633
- }
634
-
635
- #[test]
636
- fn test_parse_extraction_config_all_heading_styles() {
637
- let styles = vec![
638
- ("atx", "Atx"),
639
- ("underlined", "Underlined"),
640
- ("atx_closed", "AtxClosed"),
641
- ];
642
-
643
- for (input, _expected) in styles {
644
- let json = format!(r#"{{"html_options": {{"heading_style": "{}"}}}}"#, input);
645
- let result = parse_extraction_config_from_json(&json);
646
- assert!(result.is_ok(), "Failed to parse heading_style: {}", input);
647
- }
648
- }
649
-
650
- #[test]
651
- fn test_parse_extraction_config_preprocessing() {
652
- let json = r#"{
653
- "html_options": {
654
- "preprocessing": {
655
- "enabled": true,
656
- "preset": "aggressive",
657
- "remove_navigation": true,
658
- "remove_forms": false
659
- }
660
- }
661
- }"#;
662
-
663
- let result = parse_extraction_config_from_json(json);
664
- assert!(result.is_ok());
665
- let config = result.unwrap();
666
- assert!(config.html_options.is_some());
667
-
668
- let preprocessing = &config.html_options.unwrap().preprocessing;
669
- assert!(preprocessing.enabled);
670
- assert!(preprocessing.remove_navigation);
671
- assert!(!preprocessing.remove_forms);
672
- }
673
-
674
- #[test]
675
- fn test_to_c_extraction_result_basic() {
676
- let result = ExtractionResult {
677
- content: "Test content".to_string(),
678
- mime_type: "text/plain".to_string(),
679
- metadata: Metadata::default(),
680
- tables: vec![],
681
- detected_languages: None,
682
- chunks: None,
683
- images: None,
684
- pages: None,
685
- };
686
-
687
- let c_result = to_c_extraction_result(result);
688
- assert!(c_result.is_ok());
689
-
690
- // Clean up the allocated result
691
- if let Ok(ptr) = c_result {
692
- unsafe {
693
- let boxed = Box::from_raw(ptr);
694
- // Verify success flag
695
- assert!(boxed.success);
696
-
697
- // Clean up strings
698
- if !boxed.content.is_null() {
699
- let _ = CString::from_raw(boxed.content);
700
- }
701
- if !boxed.mime_type.is_null() {
702
- let _ = CString::from_raw(boxed.mime_type);
703
- }
704
- if !boxed.metadata_json.is_null() {
705
- let _ = CString::from_raw(boxed.metadata_json);
706
- }
707
- }
708
- }
709
- }
710
-
711
- #[test]
712
- fn test_to_c_extraction_result_with_null_bytes() {
713
- let result = ExtractionResult {
714
- content: "Test\0content with null".to_string(),
715
- mime_type: "text/plain".to_string(),
716
- metadata: Metadata::default(),
717
- tables: vec![],
718
- detected_languages: None,
719
- chunks: None,
720
- images: None,
721
- pages: None,
722
- };
723
-
724
- let c_result = to_c_extraction_result(result);
725
- assert!(c_result.is_ok());
726
-
727
- // Clean up
728
- if let Ok(ptr) = c_result {
729
- unsafe {
730
- let boxed = Box::from_raw(ptr);
731
-
732
- // Verify null bytes were replaced with replacement character
733
- let content_str = CStr::from_ptr(boxed.content).to_str().unwrap();
734
- assert!(!content_str.contains('\0'));
735
- assert!(content_str.contains('\u{FFFD}'));
736
-
737
- // Clean up strings
738
- if !boxed.content.is_null() {
739
- let _ = CString::from_raw(boxed.content);
740
- }
741
- if !boxed.mime_type.is_null() {
742
- let _ = CString::from_raw(boxed.mime_type);
743
- }
744
- if !boxed.metadata_json.is_null() {
745
- let _ = CString::from_raw(boxed.metadata_json);
746
- }
747
- }
748
- }
749
- }
750
-
751
- #[test]
752
- fn test_to_c_extraction_result_with_metadata() {
753
- let metadata = Metadata {
754
- language: Some("en".to_string()),
755
- created_at: Some("2024-01-01".to_string()),
756
- subject: Some("Test Subject".to_string()),
757
- ..Default::default()
758
- };
759
-
760
- let result = ExtractionResult {
761
- content: "Test content".to_string(),
762
- mime_type: "text/plain".to_string(),
763
- metadata,
764
- tables: vec![],
765
- detected_languages: Some(vec!["en".to_string(), "de".to_string()]),
766
- chunks: None,
767
- images: None,
768
- pages: None,
769
- };
770
-
771
- let c_result = to_c_extraction_result(result);
772
- assert!(c_result.is_ok());
773
-
774
- // Clean up
775
- if let Ok(ptr) = c_result {
776
- unsafe {
777
- let boxed = Box::from_raw(ptr);
778
-
779
- // Verify metadata fields are not null
780
- assert!(!boxed.language.is_null());
781
- assert!(!boxed.date.is_null());
782
- assert!(!boxed.subject.is_null());
783
- assert!(!boxed.detected_languages_json.is_null());
784
-
785
- // Clean up all allocated strings
786
- if !boxed.content.is_null() {
787
- let _ = CString::from_raw(boxed.content);
788
- }
789
- if !boxed.mime_type.is_null() {
790
- let _ = CString::from_raw(boxed.mime_type);
791
- }
792
- if !boxed.language.is_null() {
793
- let _ = CString::from_raw(boxed.language);
794
- }
795
- if !boxed.date.is_null() {
796
- let _ = CString::from_raw(boxed.date);
797
- }
798
- if !boxed.subject.is_null() {
799
- let _ = CString::from_raw(boxed.subject);
800
- }
801
- if !boxed.metadata_json.is_null() {
802
- let _ = CString::from_raw(boxed.metadata_json);
803
- }
804
- if !boxed.detected_languages_json.is_null() {
805
- let _ = CString::from_raw(boxed.detected_languages_json);
806
- }
807
- }
808
- }
809
- }
810
-
811
- #[test]
812
- fn test_to_c_extraction_result_with_tables_and_chunks() {
813
- let table = Table {
814
- cells: vec![
815
- vec!["Col1".to_string(), "Col2".to_string()],
816
- vec!["A1".to_string(), "A2".to_string()],
817
- vec!["B1".to_string(), "B2".to_string()],
818
- ],
819
- markdown: "| Col1 | Col2 |\n|------|------|\n| A1 | A2 |\n| B1 | B2 |".to_string(),
820
- page_number: 1,
821
- };
822
-
823
- let chunk = Chunk {
824
- content: "Chunk content".to_string(),
825
- embedding: None,
826
- metadata: ChunkMetadata {
827
- byte_start: 0,
828
- byte_end: 13,
829
- token_count: None,
830
- chunk_index: 0,
831
- total_chunks: 1,
832
- first_page: Some(1),
833
- last_page: Some(1),
834
- },
835
- };
836
-
837
- let result = ExtractionResult {
838
- content: "Test content".to_string(),
839
- mime_type: "text/plain".to_string(),
840
- metadata: Metadata::default(),
841
- tables: vec![table],
842
- detected_languages: None,
843
- chunks: Some(vec![chunk]),
844
- images: None,
845
- pages: None,
846
- };
847
-
848
- let c_result = to_c_extraction_result(result);
849
- assert!(c_result.is_ok());
850
-
851
- // Clean up
852
- if let Ok(ptr) = c_result {
853
- unsafe {
854
- let boxed = Box::from_raw(ptr);
855
-
856
- // Verify JSON fields are not null
857
- assert!(!boxed.tables_json.is_null());
858
- assert!(!boxed.chunks_json.is_null());
859
-
860
- // Clean up all allocated strings
861
- if !boxed.content.is_null() {
862
- let _ = CString::from_raw(boxed.content);
863
- }
864
- if !boxed.mime_type.is_null() {
865
- let _ = CString::from_raw(boxed.mime_type);
866
- }
867
- if !boxed.metadata_json.is_null() {
868
- let _ = CString::from_raw(boxed.metadata_json);
869
- }
870
- if !boxed.tables_json.is_null() {
871
- let _ = CString::from_raw(boxed.tables_json);
872
- }
873
- if !boxed.chunks_json.is_null() {
874
- let _ = CString::from_raw(boxed.chunks_json);
875
- }
876
- }
877
- }
878
- }
879
- }