kreuzberg 4.0.0.rc2 → 4.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/.gitignore +14 -14
- data/.rspec +3 -3
- data/.rubocop.yaml +1 -1
- data/.rubocop.yml +543 -538
- data/Gemfile +8 -8
- data/Gemfile.lock +194 -6
- data/README.md +391 -426
- data/Rakefile +34 -25
- data/Steepfile +51 -47
- data/examples/async_patterns.rb +283 -341
- data/ext/kreuzberg_rb/extconf.rb +65 -45
- data/ext/kreuzberg_rb/native/.cargo/config.toml +23 -0
- data/ext/kreuzberg_rb/native/Cargo.lock +7619 -6535
- data/ext/kreuzberg_rb/native/Cargo.toml +75 -44
- data/ext/kreuzberg_rb/native/README.md +425 -425
- data/ext/kreuzberg_rb/native/build.rs +15 -15
- data/ext/kreuzberg_rb/native/include/ieeefp.h +11 -11
- data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +14 -14
- data/ext/kreuzberg_rb/native/include/strings.h +20 -20
- data/ext/kreuzberg_rb/native/include/unistd.h +47 -47
- data/ext/kreuzberg_rb/native/src/lib.rs +3802 -2998
- data/extconf.rb +60 -28
- data/kreuzberg.gemspec +199 -148
- data/lib/kreuzberg/api_proxy.rb +126 -142
- data/lib/kreuzberg/cache_api.rb +67 -46
- data/lib/kreuzberg/cli.rb +47 -55
- data/lib/kreuzberg/cli_proxy.rb +117 -127
- data/lib/kreuzberg/config.rb +936 -691
- data/lib/kreuzberg/error_context.rb +136 -32
- data/lib/kreuzberg/errors.rb +116 -118
- data/lib/kreuzberg/extraction_api.rb +313 -85
- data/lib/kreuzberg/mcp_proxy.rb +177 -186
- data/lib/kreuzberg/ocr_backend_protocol.rb +40 -113
- data/lib/kreuzberg/post_processor_protocol.rb +15 -86
- data/lib/kreuzberg/result.rb +334 -216
- data/lib/kreuzberg/setup_lib_path.rb +99 -80
- data/lib/kreuzberg/types.rb +170 -0
- data/lib/kreuzberg/validator_protocol.rb +16 -89
- data/lib/kreuzberg/version.rb +5 -5
- data/lib/kreuzberg.rb +96 -103
- data/lib/libpdfium.so +0 -0
- data/sig/kreuzberg/internal.rbs +184 -184
- data/sig/kreuzberg.rbs +561 -520
- data/spec/binding/async_operations_spec.rb +473 -0
- data/spec/binding/batch_operations_spec.rb +595 -0
- data/spec/binding/batch_spec.rb +359 -0
- data/spec/binding/cache_spec.rb +227 -227
- data/spec/binding/cli_proxy_spec.rb +85 -85
- data/spec/binding/cli_spec.rb +55 -55
- data/spec/binding/config_result_spec.rb +377 -0
- data/spec/binding/config_spec.rb +419 -345
- data/spec/binding/config_validation_spec.rb +377 -283
- data/spec/binding/embeddings_spec.rb +816 -0
- data/spec/binding/error_handling_spec.rb +399 -213
- data/spec/binding/error_recovery_spec.rb +488 -0
- data/spec/binding/errors_spec.rb +66 -66
- data/spec/binding/font_config_spec.rb +220 -0
- data/spec/binding/images_spec.rb +738 -0
- data/spec/binding/keywords_extraction_spec.rb +600 -0
- data/spec/binding/metadata_types_spec.rb +1228 -0
- data/spec/binding/pages_extraction_spec.rb +471 -0
- data/spec/binding/plugins/ocr_backend_spec.rb +307 -307
- data/spec/binding/plugins/postprocessor_spec.rb +269 -269
- data/spec/binding/plugins/validator_spec.rb +273 -274
- data/spec/binding/tables_spec.rb +641 -0
- data/spec/fixtures/config.toml +38 -39
- data/spec/fixtures/config.yaml +41 -41
- data/spec/fixtures/invalid_config.toml +3 -4
- data/spec/smoke/package_spec.rb +177 -178
- data/spec/spec_helper.rb +40 -42
- data/spec/unit/config/chunking_config_spec.rb +213 -0
- data/spec/unit/config/embedding_config_spec.rb +343 -0
- data/spec/unit/config/extraction_config_spec.rb +438 -0
- data/spec/unit/config/font_config_spec.rb +285 -0
- data/spec/unit/config/hierarchy_config_spec.rb +314 -0
- data/spec/unit/config/image_extraction_config_spec.rb +209 -0
- data/spec/unit/config/image_preprocessing_config_spec.rb +249 -0
- data/spec/unit/config/keyword_config_spec.rb +229 -0
- data/spec/unit/config/language_detection_config_spec.rb +258 -0
- data/spec/unit/config/ocr_config_spec.rb +171 -0
- data/spec/unit/config/page_config_spec.rb +221 -0
- data/spec/unit/config/pdf_config_spec.rb +267 -0
- data/spec/unit/config/postprocessor_config_spec.rb +290 -0
- data/spec/unit/config/tesseract_config_spec.rb +181 -0
- data/spec/unit/config/token_reduction_config_spec.rb +251 -0
- data/test/metadata_types_test.rb +959 -0
- data/vendor/Cargo.toml +61 -0
- data/vendor/kreuzberg/Cargo.toml +259 -204
- data/vendor/kreuzberg/README.md +263 -175
- data/vendor/kreuzberg/build.rs +782 -474
- data/vendor/kreuzberg/examples/bench_fixes.rs +71 -0
- data/vendor/kreuzberg/examples/test_pdfium_fork.rs +62 -0
- data/vendor/kreuzberg/src/api/error.rs +81 -81
- data/vendor/kreuzberg/src/api/handlers.rs +320 -199
- data/vendor/kreuzberg/src/api/mod.rs +94 -79
- data/vendor/kreuzberg/src/api/server.rs +518 -353
- data/vendor/kreuzberg/src/api/types.rs +206 -170
- data/vendor/kreuzberg/src/cache/mod.rs +1167 -1167
- data/vendor/kreuzberg/src/chunking/mod.rs +2303 -677
- data/vendor/kreuzberg/src/chunking/processor.rs +219 -0
- data/vendor/kreuzberg/src/core/batch_mode.rs +95 -95
- data/vendor/kreuzberg/src/core/batch_optimizations.rs +385 -0
- data/vendor/kreuzberg/src/core/config.rs +1914 -1032
- data/vendor/kreuzberg/src/core/config_validation.rs +949 -0
- data/vendor/kreuzberg/src/core/extractor.rs +1200 -1024
- data/vendor/kreuzberg/src/core/formats.rs +235 -0
- data/vendor/kreuzberg/src/core/io.rs +329 -329
- data/vendor/kreuzberg/src/core/mime.rs +605 -605
- data/vendor/kreuzberg/src/core/mod.rs +61 -45
- data/vendor/kreuzberg/src/core/pipeline.rs +1223 -984
- data/vendor/kreuzberg/src/core/server_config.rs +1220 -0
- data/vendor/kreuzberg/src/embeddings.rs +471 -432
- data/vendor/kreuzberg/src/error.rs +431 -431
- data/vendor/kreuzberg/src/extraction/archive.rs +959 -954
- data/vendor/kreuzberg/src/extraction/capacity.rs +263 -0
- data/vendor/kreuzberg/src/extraction/docx.rs +404 -40
- data/vendor/kreuzberg/src/extraction/email.rs +855 -854
- data/vendor/kreuzberg/src/extraction/excel.rs +697 -688
- data/vendor/kreuzberg/src/extraction/html.rs +1830 -553
- data/vendor/kreuzberg/src/extraction/image.rs +492 -368
- data/vendor/kreuzberg/src/extraction/libreoffice.rs +574 -563
- data/vendor/kreuzberg/src/extraction/markdown.rs +216 -213
- data/vendor/kreuzberg/src/extraction/mod.rs +93 -81
- data/vendor/kreuzberg/src/extraction/office_metadata/app_properties.rs +398 -398
- data/vendor/kreuzberg/src/extraction/office_metadata/core_properties.rs +247 -247
- data/vendor/kreuzberg/src/extraction/office_metadata/custom_properties.rs +240 -240
- data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +130 -130
- data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +284 -287
- data/vendor/kreuzberg/src/extraction/pptx.rs +3102 -3000
- data/vendor/kreuzberg/src/extraction/structured.rs +491 -490
- data/vendor/kreuzberg/src/extraction/table.rs +329 -328
- data/vendor/kreuzberg/src/extraction/text.rs +277 -269
- data/vendor/kreuzberg/src/extraction/xml.rs +333 -333
- data/vendor/kreuzberg/src/extractors/archive.rs +447 -446
- data/vendor/kreuzberg/src/extractors/bibtex.rs +470 -469
- data/vendor/kreuzberg/src/extractors/docbook.rs +504 -502
- data/vendor/kreuzberg/src/extractors/docx.rs +400 -367
- data/vendor/kreuzberg/src/extractors/email.rs +157 -143
- data/vendor/kreuzberg/src/extractors/epub.rs +696 -707
- data/vendor/kreuzberg/src/extractors/excel.rs +385 -343
- data/vendor/kreuzberg/src/extractors/fictionbook.rs +492 -491
- data/vendor/kreuzberg/src/extractors/html.rs +419 -393
- data/vendor/kreuzberg/src/extractors/image.rs +219 -198
- data/vendor/kreuzberg/src/extractors/jats.rs +1054 -1051
- data/vendor/kreuzberg/src/extractors/jupyter.rs +368 -367
- data/vendor/kreuzberg/src/extractors/latex.rs +653 -652
- data/vendor/kreuzberg/src/extractors/markdown.rs +701 -700
- data/vendor/kreuzberg/src/extractors/mod.rs +429 -365
- data/vendor/kreuzberg/src/extractors/odt.rs +628 -628
- data/vendor/kreuzberg/src/extractors/opml.rs +635 -634
- data/vendor/kreuzberg/src/extractors/orgmode.rs +529 -528
- data/vendor/kreuzberg/src/extractors/pdf.rs +761 -493
- data/vendor/kreuzberg/src/extractors/pptx.rs +279 -248
- data/vendor/kreuzberg/src/extractors/rst.rs +577 -576
- data/vendor/kreuzberg/src/extractors/rtf.rs +809 -810
- data/vendor/kreuzberg/src/extractors/security.rs +484 -484
- data/vendor/kreuzberg/src/extractors/security_tests.rs +367 -367
- data/vendor/kreuzberg/src/extractors/structured.rs +142 -140
- data/vendor/kreuzberg/src/extractors/text.rs +265 -260
- data/vendor/kreuzberg/src/extractors/typst.rs +651 -650
- data/vendor/kreuzberg/src/extractors/xml.rs +147 -135
- data/vendor/kreuzberg/src/image/dpi.rs +164 -164
- data/vendor/kreuzberg/src/image/mod.rs +6 -6
- data/vendor/kreuzberg/src/image/preprocessing.rs +417 -417
- data/vendor/kreuzberg/src/image/resize.rs +89 -89
- data/vendor/kreuzberg/src/keywords/config.rs +154 -154
- data/vendor/kreuzberg/src/keywords/mod.rs +237 -237
- data/vendor/kreuzberg/src/keywords/processor.rs +275 -267
- data/vendor/kreuzberg/src/keywords/rake.rs +293 -293
- data/vendor/kreuzberg/src/keywords/types.rs +68 -68
- data/vendor/kreuzberg/src/keywords/yake.rs +163 -163
- data/vendor/kreuzberg/src/language_detection/mod.rs +985 -942
- data/vendor/kreuzberg/src/language_detection/processor.rs +218 -0
- data/vendor/kreuzberg/src/lib.rs +114 -105
- data/vendor/kreuzberg/src/mcp/mod.rs +35 -32
- data/vendor/kreuzberg/src/mcp/server.rs +2090 -1968
- data/vendor/kreuzberg/src/ocr/cache.rs +469 -469
- data/vendor/kreuzberg/src/ocr/error.rs +37 -37
- data/vendor/kreuzberg/src/ocr/hocr.rs +216 -216
- data/vendor/kreuzberg/src/ocr/language_registry.rs +520 -0
- data/vendor/kreuzberg/src/ocr/mod.rs +60 -58
- data/vendor/kreuzberg/src/ocr/processor.rs +858 -863
- data/vendor/kreuzberg/src/ocr/table/mod.rs +4 -4
- data/vendor/kreuzberg/src/ocr/table/tsv_parser.rs +144 -144
- data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +456 -450
- data/vendor/kreuzberg/src/ocr/types.rs +393 -393
- data/vendor/kreuzberg/src/ocr/utils.rs +47 -47
- data/vendor/kreuzberg/src/ocr/validation.rs +206 -206
- data/vendor/kreuzberg/src/panic_context.rs +154 -154
- data/vendor/kreuzberg/src/pdf/bindings.rs +306 -0
- data/vendor/kreuzberg/src/pdf/bundled.rs +408 -0
- data/vendor/kreuzberg/src/pdf/error.rs +214 -122
- data/vendor/kreuzberg/src/pdf/fonts.rs +358 -0
- data/vendor/kreuzberg/src/pdf/hierarchy.rs +903 -0
- data/vendor/kreuzberg/src/pdf/images.rs +139 -139
- data/vendor/kreuzberg/src/pdf/metadata.rs +509 -346
- data/vendor/kreuzberg/src/pdf/mod.rs +81 -50
- data/vendor/kreuzberg/src/pdf/rendering.rs +369 -369
- data/vendor/kreuzberg/src/pdf/table.rs +417 -393
- data/vendor/kreuzberg/src/pdf/text.rs +553 -158
- data/vendor/kreuzberg/src/plugins/extractor.rs +1042 -1013
- data/vendor/kreuzberg/src/plugins/mod.rs +212 -209
- data/vendor/kreuzberg/src/plugins/ocr.rs +637 -620
- data/vendor/kreuzberg/src/plugins/processor.rs +650 -642
- data/vendor/kreuzberg/src/plugins/registry.rs +1339 -1337
- data/vendor/kreuzberg/src/plugins/traits.rs +258 -258
- data/vendor/kreuzberg/src/plugins/validator.rs +967 -956
- data/vendor/kreuzberg/src/stopwords/mod.rs +1470 -1470
- data/vendor/kreuzberg/src/text/mod.rs +27 -19
- data/vendor/kreuzberg/src/text/quality.rs +710 -697
- data/vendor/kreuzberg/src/text/quality_processor.rs +231 -0
- data/vendor/kreuzberg/src/text/string_utils.rs +229 -217
- data/vendor/kreuzberg/src/text/token_reduction/cjk_utils.rs +164 -164
- data/vendor/kreuzberg/src/text/token_reduction/config.rs +100 -100
- data/vendor/kreuzberg/src/text/token_reduction/core.rs +832 -796
- data/vendor/kreuzberg/src/text/token_reduction/filters.rs +923 -902
- data/vendor/kreuzberg/src/text/token_reduction/mod.rs +160 -160
- data/vendor/kreuzberg/src/text/token_reduction/semantic.rs +619 -619
- data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +148 -147
- data/vendor/kreuzberg/src/text/utf8_validation.rs +193 -0
- data/vendor/kreuzberg/src/types.rs +1713 -903
- data/vendor/kreuzberg/src/utils/mod.rs +31 -17
- data/vendor/kreuzberg/src/utils/pool.rs +503 -0
- data/vendor/kreuzberg/src/utils/pool_sizing.rs +364 -0
- data/vendor/kreuzberg/src/utils/quality.rs +968 -959
- data/vendor/kreuzberg/src/utils/string_pool.rs +761 -0
- data/vendor/kreuzberg/src/utils/string_utils.rs +381 -381
- data/vendor/kreuzberg/stopwords/af_stopwords.json +53 -53
- data/vendor/kreuzberg/stopwords/ar_stopwords.json +482 -482
- data/vendor/kreuzberg/stopwords/bg_stopwords.json +261 -261
- data/vendor/kreuzberg/stopwords/bn_stopwords.json +400 -400
- data/vendor/kreuzberg/stopwords/br_stopwords.json +1205 -1205
- data/vendor/kreuzberg/stopwords/ca_stopwords.json +280 -280
- data/vendor/kreuzberg/stopwords/cs_stopwords.json +425 -425
- data/vendor/kreuzberg/stopwords/da_stopwords.json +172 -172
- data/vendor/kreuzberg/stopwords/de_stopwords.json +622 -622
- data/vendor/kreuzberg/stopwords/el_stopwords.json +849 -849
- data/vendor/kreuzberg/stopwords/en_stopwords.json +1300 -1300
- data/vendor/kreuzberg/stopwords/eo_stopwords.json +175 -175
- data/vendor/kreuzberg/stopwords/es_stopwords.json +734 -734
- data/vendor/kreuzberg/stopwords/et_stopwords.json +37 -37
- data/vendor/kreuzberg/stopwords/eu_stopwords.json +100 -100
- data/vendor/kreuzberg/stopwords/fa_stopwords.json +801 -801
- data/vendor/kreuzberg/stopwords/fi_stopwords.json +849 -849
- data/vendor/kreuzberg/stopwords/fr_stopwords.json +693 -693
- data/vendor/kreuzberg/stopwords/ga_stopwords.json +111 -111
- data/vendor/kreuzberg/stopwords/gl_stopwords.json +162 -162
- data/vendor/kreuzberg/stopwords/gu_stopwords.json +226 -226
- data/vendor/kreuzberg/stopwords/ha_stopwords.json +41 -41
- data/vendor/kreuzberg/stopwords/he_stopwords.json +196 -196
- data/vendor/kreuzberg/stopwords/hi_stopwords.json +227 -227
- data/vendor/kreuzberg/stopwords/hr_stopwords.json +181 -181
- data/vendor/kreuzberg/stopwords/hu_stopwords.json +791 -791
- data/vendor/kreuzberg/stopwords/hy_stopwords.json +47 -47
- data/vendor/kreuzberg/stopwords/id_stopwords.json +760 -760
- data/vendor/kreuzberg/stopwords/it_stopwords.json +634 -634
- data/vendor/kreuzberg/stopwords/ja_stopwords.json +136 -136
- data/vendor/kreuzberg/stopwords/kn_stopwords.json +84 -84
- data/vendor/kreuzberg/stopwords/ko_stopwords.json +681 -681
- data/vendor/kreuzberg/stopwords/ku_stopwords.json +64 -64
- data/vendor/kreuzberg/stopwords/la_stopwords.json +51 -51
- data/vendor/kreuzberg/stopwords/lt_stopwords.json +476 -476
- data/vendor/kreuzberg/stopwords/lv_stopwords.json +163 -163
- data/vendor/kreuzberg/stopwords/ml_stopwords.json +1 -1
- data/vendor/kreuzberg/stopwords/mr_stopwords.json +101 -101
- data/vendor/kreuzberg/stopwords/ms_stopwords.json +477 -477
- data/vendor/kreuzberg/stopwords/ne_stopwords.json +490 -490
- data/vendor/kreuzberg/stopwords/nl_stopwords.json +415 -415
- data/vendor/kreuzberg/stopwords/no_stopwords.json +223 -223
- data/vendor/kreuzberg/stopwords/pl_stopwords.json +331 -331
- data/vendor/kreuzberg/stopwords/pt_stopwords.json +562 -562
- data/vendor/kreuzberg/stopwords/ro_stopwords.json +436 -436
- data/vendor/kreuzberg/stopwords/ru_stopwords.json +561 -561
- data/vendor/kreuzberg/stopwords/si_stopwords.json +193 -193
- data/vendor/kreuzberg/stopwords/sk_stopwords.json +420 -420
- data/vendor/kreuzberg/stopwords/sl_stopwords.json +448 -448
- data/vendor/kreuzberg/stopwords/so_stopwords.json +32 -32
- data/vendor/kreuzberg/stopwords/st_stopwords.json +33 -33
- data/vendor/kreuzberg/stopwords/sv_stopwords.json +420 -420
- data/vendor/kreuzberg/stopwords/sw_stopwords.json +76 -76
- data/vendor/kreuzberg/stopwords/ta_stopwords.json +129 -129
- data/vendor/kreuzberg/stopwords/te_stopwords.json +54 -54
- data/vendor/kreuzberg/stopwords/th_stopwords.json +118 -118
- data/vendor/kreuzberg/stopwords/tl_stopwords.json +149 -149
- data/vendor/kreuzberg/stopwords/tr_stopwords.json +506 -506
- data/vendor/kreuzberg/stopwords/uk_stopwords.json +75 -75
- data/vendor/kreuzberg/stopwords/ur_stopwords.json +519 -519
- data/vendor/kreuzberg/stopwords/vi_stopwords.json +647 -647
- data/vendor/kreuzberg/stopwords/yo_stopwords.json +62 -62
- data/vendor/kreuzberg/stopwords/zh_stopwords.json +796 -796
- data/vendor/kreuzberg/stopwords/zu_stopwords.json +31 -31
- data/vendor/kreuzberg/tests/api_embed.rs +360 -0
- data/vendor/kreuzberg/tests/api_extract_multipart.rs +52 -52
- data/vendor/kreuzberg/tests/api_large_pdf_extraction.rs +471 -0
- data/vendor/kreuzberg/tests/api_large_pdf_extraction_diagnostics.rs +289 -0
- data/vendor/kreuzberg/tests/api_tests.rs +1472 -966
- data/vendor/kreuzberg/tests/archive_integration.rs +545 -543
- data/vendor/kreuzberg/tests/batch_orchestration.rs +587 -556
- data/vendor/kreuzberg/tests/batch_pooling_benchmark.rs +154 -0
- data/vendor/kreuzberg/tests/batch_processing.rs +328 -316
- data/vendor/kreuzberg/tests/bibtex_parity_test.rs +421 -421
- data/vendor/kreuzberg/tests/concurrency_stress.rs +541 -525
- data/vendor/kreuzberg/tests/config_features.rs +612 -598
- data/vendor/kreuzberg/tests/config_integration_test.rs +753 -0
- data/vendor/kreuzberg/tests/config_loading_tests.rs +416 -415
- data/vendor/kreuzberg/tests/core_integration.rs +519 -510
- data/vendor/kreuzberg/tests/csv_integration.rs +414 -414
- data/vendor/kreuzberg/tests/data/hierarchy_ground_truth.json +294 -0
- data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +500 -498
- data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +122 -122
- data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +370 -370
- data/vendor/kreuzberg/tests/email_integration.rs +327 -325
- data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +275 -275
- data/vendor/kreuzberg/tests/error_handling.rs +402 -393
- data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +228 -228
- data/vendor/kreuzberg/tests/format_integration.rs +165 -159
- data/vendor/kreuzberg/tests/helpers/mod.rs +202 -142
- data/vendor/kreuzberg/tests/html_table_test.rs +551 -551
- data/vendor/kreuzberg/tests/image_integration.rs +255 -253
- data/vendor/kreuzberg/tests/instrumentation_test.rs +139 -139
- data/vendor/kreuzberg/tests/jats_extractor_tests.rs +639 -639
- data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +704 -704
- data/vendor/kreuzberg/tests/keywords_integration.rs +479 -479
- data/vendor/kreuzberg/tests/keywords_quality.rs +509 -509
- data/vendor/kreuzberg/tests/latex_extractor_tests.rs +496 -496
- data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +490 -490
- data/vendor/kreuzberg/tests/mime_detection.rs +429 -428
- data/vendor/kreuzberg/tests/ocr_configuration.rs +514 -510
- data/vendor/kreuzberg/tests/ocr_errors.rs +698 -676
- data/vendor/kreuzberg/tests/ocr_language_registry.rs +191 -0
- data/vendor/kreuzberg/tests/ocr_quality.rs +629 -627
- data/vendor/kreuzberg/tests/ocr_stress.rs +469 -469
- data/vendor/kreuzberg/tests/odt_extractor_tests.rs +674 -695
- data/vendor/kreuzberg/tests/opml_extractor_tests.rs +616 -616
- data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +822 -822
- data/vendor/kreuzberg/tests/page_markers.rs +297 -0
- data/vendor/kreuzberg/tests/pdf_hierarchy_detection.rs +301 -0
- data/vendor/kreuzberg/tests/pdf_hierarchy_quality.rs +589 -0
- data/vendor/kreuzberg/tests/pdf_integration.rs +45 -43
- data/vendor/kreuzberg/tests/pdf_ocr_triggering.rs +301 -0
- data/vendor/kreuzberg/tests/pdf_text_merging.rs +475 -0
- data/vendor/kreuzberg/tests/pdfium_linking.rs +340 -0
- data/vendor/kreuzberg/tests/pipeline_integration.rs +1446 -1411
- data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +776 -771
- data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +577 -560
- data/vendor/kreuzberg/tests/plugin_system.rs +927 -921
- data/vendor/kreuzberg/tests/plugin_validator_test.rs +783 -783
- data/vendor/kreuzberg/tests/registry_integration_tests.rs +587 -586
- data/vendor/kreuzberg/tests/rst_extractor_tests.rs +694 -692
- data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +775 -776
- data/vendor/kreuzberg/tests/security_validation.rs +416 -415
- data/vendor/kreuzberg/tests/stopwords_integration_test.rs +888 -888
- data/vendor/kreuzberg/tests/test_fastembed.rs +631 -609
- data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +1260 -1259
- data/vendor/kreuzberg/tests/typst_extractor_tests.rs +648 -647
- data/vendor/kreuzberg/tests/xlsx_metadata_extraction_test.rs +87 -87
- data/vendor/kreuzberg-ffi/Cargo.toml +67 -0
- data/vendor/kreuzberg-ffi/README.md +851 -0
- data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +227 -0
- data/vendor/kreuzberg-ffi/build.rs +168 -0
- data/vendor/kreuzberg-ffi/cbindgen.toml +37 -0
- data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +12 -0
- data/vendor/kreuzberg-ffi/kreuzberg.h +3012 -0
- data/vendor/kreuzberg-ffi/src/batch_streaming.rs +588 -0
- data/vendor/kreuzberg-ffi/src/config.rs +1341 -0
- data/vendor/kreuzberg-ffi/src/error.rs +901 -0
- data/vendor/kreuzberg-ffi/src/extraction.rs +555 -0
- data/vendor/kreuzberg-ffi/src/helpers.rs +879 -0
- data/vendor/kreuzberg-ffi/src/lib.rs +977 -0
- data/vendor/kreuzberg-ffi/src/memory.rs +493 -0
- data/vendor/kreuzberg-ffi/src/mime.rs +329 -0
- data/vendor/kreuzberg-ffi/src/panic_shield.rs +265 -0
- data/vendor/kreuzberg-ffi/src/plugins/document_extractor.rs +442 -0
- data/vendor/kreuzberg-ffi/src/plugins/mod.rs +14 -0
- data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +628 -0
- data/vendor/kreuzberg-ffi/src/plugins/post_processor.rs +438 -0
- data/vendor/kreuzberg-ffi/src/plugins/validator.rs +329 -0
- data/vendor/kreuzberg-ffi/src/result.rs +510 -0
- data/vendor/kreuzberg-ffi/src/result_pool.rs +639 -0
- data/vendor/kreuzberg-ffi/src/result_view.rs +773 -0
- data/vendor/kreuzberg-ffi/src/string_intern.rs +568 -0
- data/vendor/kreuzberg-ffi/src/types.rs +363 -0
- data/vendor/kreuzberg-ffi/src/util.rs +210 -0
- data/vendor/kreuzberg-ffi/src/validation.rs +848 -0
- data/vendor/kreuzberg-ffi/tests.disabled/README.md +48 -0
- data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +299 -0
- data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +346 -0
- data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +232 -0
- data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +470 -0
- data/vendor/kreuzberg-tesseract/.commitlintrc.json +13 -0
- data/vendor/kreuzberg-tesseract/.crate-ignore +2 -0
- data/vendor/kreuzberg-tesseract/Cargo.lock +2933 -0
- data/vendor/kreuzberg-tesseract/Cargo.toml +57 -0
- data/vendor/{rb-sys/LICENSE-MIT → kreuzberg-tesseract/LICENSE} +22 -21
- data/vendor/kreuzberg-tesseract/README.md +399 -0
- data/vendor/kreuzberg-tesseract/build.rs +1127 -0
- data/vendor/kreuzberg-tesseract/patches/README.md +71 -0
- data/vendor/kreuzberg-tesseract/patches/tesseract.diff +199 -0
- data/vendor/kreuzberg-tesseract/src/api.rs +1371 -0
- data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +77 -0
- data/vendor/kreuzberg-tesseract/src/enums.rs +297 -0
- data/vendor/kreuzberg-tesseract/src/error.rs +81 -0
- data/vendor/kreuzberg-tesseract/src/lib.rs +145 -0
- data/vendor/kreuzberg-tesseract/src/monitor.rs +57 -0
- data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +197 -0
- data/vendor/kreuzberg-tesseract/src/page_iterator.rs +253 -0
- data/vendor/kreuzberg-tesseract/src/result_iterator.rs +286 -0
- data/vendor/kreuzberg-tesseract/src/result_renderer.rs +183 -0
- data/vendor/kreuzberg-tesseract/tests/integration_test.rs +211 -0
- metadata +196 -45
- data/vendor/kreuzberg/benches/otel_overhead.rs +0 -48
- data/vendor/kreuzberg/src/extractors/fictionbook.rs.backup2 +0 -738
- data/vendor/rb-sys/.cargo-ok +0 -1
- data/vendor/rb-sys/.cargo_vcs_info.json +0 -6
- data/vendor/rb-sys/Cargo.lock +0 -393
- data/vendor/rb-sys/Cargo.toml +0 -70
- data/vendor/rb-sys/Cargo.toml.orig +0 -57
- data/vendor/rb-sys/LICENSE-APACHE +0 -190
- data/vendor/rb-sys/bin/release.sh +0 -21
- data/vendor/rb-sys/build/features.rs +0 -108
- data/vendor/rb-sys/build/main.rs +0 -246
- data/vendor/rb-sys/build/stable_api_config.rs +0 -153
- data/vendor/rb-sys/build/version.rs +0 -48
- data/vendor/rb-sys/readme.md +0 -36
- data/vendor/rb-sys/src/bindings.rs +0 -21
- data/vendor/rb-sys/src/hidden.rs +0 -11
- data/vendor/rb-sys/src/lib.rs +0 -34
- data/vendor/rb-sys/src/macros.rs +0 -371
- data/vendor/rb-sys/src/memory.rs +0 -53
- data/vendor/rb-sys/src/ruby_abi_version.rs +0 -38
- data/vendor/rb-sys/src/special_consts.rs +0 -31
- data/vendor/rb-sys/src/stable_api/compiled.c +0 -179
- data/vendor/rb-sys/src/stable_api/compiled.rs +0 -257
- data/vendor/rb-sys/src/stable_api/ruby_2_6.rs +0 -316
- data/vendor/rb-sys/src/stable_api/ruby_2_7.rs +0 -316
- data/vendor/rb-sys/src/stable_api/ruby_3_0.rs +0 -324
- data/vendor/rb-sys/src/stable_api/ruby_3_1.rs +0 -317
- data/vendor/rb-sys/src/stable_api/ruby_3_2.rs +0 -315
- data/vendor/rb-sys/src/stable_api/ruby_3_3.rs +0 -326
- data/vendor/rb-sys/src/stable_api/ruby_3_4.rs +0 -327
- data/vendor/rb-sys/src/stable_api.rs +0 -261
- data/vendor/rb-sys/src/symbol.rs +0 -31
- data/vendor/rb-sys/src/tracking_allocator.rs +0 -332
- data/vendor/rb-sys/src/utils.rs +0 -89
- data/vendor/rb-sys/src/value_type.rs +0 -7
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
# Disabled Integration Tests
|
|
2
|
+
|
|
3
|
+
## Why These Tests Were Disabled
|
|
4
|
+
|
|
5
|
+
These integration tests were attempting to test the C FFI interface from Rust using `extern "C"` declarations. However, this approach has fundamental issues:
|
|
6
|
+
|
|
7
|
+
1. **Linking Problem**: Integration tests in `tests/` directory run as separate binaries. When the library is compiled as `cdylib` or `staticlib`, these crate types don't automatically link to Rust integration tests.
|
|
8
|
+
|
|
9
|
+
2. **Architecture Mismatch**: These tests were trying to use `unsafe extern "C"` blocks to declare functions that are already defined in the same crate. This creates a circular dependency where the test binary expects to dynamically link against symbols that should be statically linked.
|
|
10
|
+
|
|
11
|
+
3. **Wrong Testing Layer**: C FFI functions should be tested from the actual language bindings (Java, Go, etc.) that consume them, not from Rust integration tests.
|
|
12
|
+
|
|
13
|
+
## Testing Strategy
|
|
14
|
+
|
|
15
|
+
The FFI layer is thoroughly tested through:
|
|
16
|
+
|
|
17
|
+
1. **Unit tests** in `src/lib.rs` that directly call the public Rust FFI functions
|
|
18
|
+
2. **Language binding tests**:
|
|
19
|
+
- Java: `packages/java/src/test/java/`
|
|
20
|
+
- Go: `packages/go/v4/*_test.go`
|
|
21
|
+
- Python: `packages/python/tests/`
|
|
22
|
+
- TypeScript: `packages/typescript/**/*.spec.ts`
|
|
23
|
+
- Ruby: `packages/ruby/spec/`
|
|
24
|
+
|
|
25
|
+
3. **E2E tests** in `e2e/` directory that test the complete integration
|
|
26
|
+
|
|
27
|
+
## Error That Led to Disabling
|
|
28
|
+
|
|
29
|
+
```
|
|
30
|
+
error: linking with `cc` failed: exit status: 1
|
|
31
|
+
rust-lld: error: undefined symbol: kreuzberg_last_error
|
|
32
|
+
rust-lld: error: undefined symbol: kreuzberg_load_extraction_config_from_file
|
|
33
|
+
rust-lld: error: undefined symbol: kreuzberg_free_string
|
|
34
|
+
```
|
|
35
|
+
|
|
36
|
+
This occurred because the integration test binaries couldn't link to the C symbols defined in the FFI crate.
|
|
37
|
+
|
|
38
|
+
## Future Improvements
|
|
39
|
+
|
|
40
|
+
If C FFI integration tests are needed from Rust:
|
|
41
|
+
|
|
42
|
+
1. Create a separate C test harness that links against the compiled `.so`/`.dylib`
|
|
43
|
+
2. Move critical tests into unit tests within `src/lib.rs`
|
|
44
|
+
3. Ensure comprehensive coverage through language binding tests instead
|
|
45
|
+
|
|
46
|
+
## CI Impact
|
|
47
|
+
|
|
48
|
+
This change resolves CI failures in run 19607600358 on both macOS-14 and Ubuntu runners.
|
|
@@ -0,0 +1,299 @@
|
|
|
1
|
+
//! FFI config loading integration tests.
|
|
2
|
+
//!
|
|
3
|
+
//! Tests the FFI layer for configuration loading from files and discovery.
|
|
4
|
+
|
|
5
|
+
use std::ffi::{CStr, CString};
|
|
6
|
+
use std::fs;
|
|
7
|
+
use std::os::raw::c_char;
|
|
8
|
+
use tempfile::TempDir;
|
|
9
|
+
|
|
10
|
+
unsafe extern "C" {
|
|
11
|
+
fn kreuzberg_load_extraction_config_from_file(file_path: *const c_char) -> *mut c_char;
|
|
12
|
+
fn kreuzberg_free_string(s: *mut c_char);
|
|
13
|
+
fn kreuzberg_last_error() -> *const c_char;
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
/// Helper to convert *const c_char to String
|
|
17
|
+
unsafe fn c_str_to_string(ptr: *const c_char) -> Option<String> {
|
|
18
|
+
if ptr.is_null() {
|
|
19
|
+
None
|
|
20
|
+
} else {
|
|
21
|
+
unsafe { Some(CStr::from_ptr(ptr).to_string_lossy().into_owned()) }
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
|
|
25
|
+
/// Helper to get last error message
|
|
26
|
+
unsafe fn get_last_error() -> Option<String> {
|
|
27
|
+
let error_ptr = unsafe { kreuzberg_last_error() };
|
|
28
|
+
unsafe { c_str_to_string(error_ptr) }
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
/// Test successful config loading from TOML file.
|
|
32
|
+
#[test]
|
|
33
|
+
fn test_load_config_from_toml_file_succeeds() {
|
|
34
|
+
let temp_dir = TempDir::new().unwrap();
|
|
35
|
+
let config_path = temp_dir.path().join("config.toml");
|
|
36
|
+
|
|
37
|
+
let toml_content = r#"
|
|
38
|
+
[ocr]
|
|
39
|
+
enabled = true
|
|
40
|
+
backend = "tesseract"
|
|
41
|
+
|
|
42
|
+
[chunking]
|
|
43
|
+
max_chars = 1000
|
|
44
|
+
max_overlap = 100
|
|
45
|
+
"#;
|
|
46
|
+
|
|
47
|
+
fs::write(&config_path, toml_content).unwrap();
|
|
48
|
+
|
|
49
|
+
unsafe {
|
|
50
|
+
let path_cstr = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
51
|
+
let result = kreuzberg_load_extraction_config_from_file(path_cstr.as_ptr());
|
|
52
|
+
|
|
53
|
+
assert!(!result.is_null(), "Result should not be null");
|
|
54
|
+
|
|
55
|
+
let json_str = c_str_to_string(result).expect("Should have valid JSON");
|
|
56
|
+
kreuzberg_free_string(result);
|
|
57
|
+
|
|
58
|
+
let json_value: serde_json::Value = serde_json::from_str(&json_str).unwrap();
|
|
59
|
+
assert!(json_value.get("ocr").is_some(), "Should have OCR config");
|
|
60
|
+
assert!(json_value.get("chunking").is_some(), "Should have chunking config");
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
|
|
64
|
+
/// Test successful config loading from YAML file.
|
|
65
|
+
#[test]
|
|
66
|
+
fn test_load_config_from_yaml_file_succeeds() {
|
|
67
|
+
let temp_dir = TempDir::new().unwrap();
|
|
68
|
+
let config_path = temp_dir.path().join("config.yaml");
|
|
69
|
+
|
|
70
|
+
let yaml_content = r#"
|
|
71
|
+
ocr:
|
|
72
|
+
enabled: true
|
|
73
|
+
backend: tesseract
|
|
74
|
+
chunking:
|
|
75
|
+
max_chars: 1000
|
|
76
|
+
max_overlap: 100
|
|
77
|
+
"#;
|
|
78
|
+
|
|
79
|
+
fs::write(&config_path, yaml_content).unwrap();
|
|
80
|
+
|
|
81
|
+
unsafe {
|
|
82
|
+
let path_cstr = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
83
|
+
let result = kreuzberg_load_extraction_config_from_file(path_cstr.as_ptr());
|
|
84
|
+
|
|
85
|
+
assert!(!result.is_null(), "Result should not be null");
|
|
86
|
+
|
|
87
|
+
let json_str = c_str_to_string(result).expect("Should have valid JSON");
|
|
88
|
+
kreuzberg_free_string(result);
|
|
89
|
+
|
|
90
|
+
let json_value: serde_json::Value = serde_json::from_str(&json_str).unwrap();
|
|
91
|
+
assert!(json_value.get("ocr").is_some(), "Should have OCR config");
|
|
92
|
+
assert!(json_value.get("chunking").is_some(), "Should have chunking config");
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
|
|
96
|
+
/// Test successful config loading from JSON file.
|
|
97
|
+
#[test]
|
|
98
|
+
fn test_load_config_from_json_file_succeeds() {
|
|
99
|
+
let temp_dir = TempDir::new().unwrap();
|
|
100
|
+
let config_path = temp_dir.path().join("config.json");
|
|
101
|
+
|
|
102
|
+
let json_content = r#"
|
|
103
|
+
{
|
|
104
|
+
"ocr": {
|
|
105
|
+
"enabled": true,
|
|
106
|
+
"backend": "tesseract"
|
|
107
|
+
},
|
|
108
|
+
"chunking": {
|
|
109
|
+
"max_chars": 1000,
|
|
110
|
+
"max_overlap": 100
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
"#;
|
|
114
|
+
|
|
115
|
+
fs::write(&config_path, json_content).unwrap();
|
|
116
|
+
|
|
117
|
+
unsafe {
|
|
118
|
+
let path_cstr = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
119
|
+
let result = kreuzberg_load_extraction_config_from_file(path_cstr.as_ptr());
|
|
120
|
+
|
|
121
|
+
assert!(!result.is_null(), "Result should not be null");
|
|
122
|
+
|
|
123
|
+
let json_str = c_str_to_string(result).expect("Should have valid JSON");
|
|
124
|
+
kreuzberg_free_string(result);
|
|
125
|
+
|
|
126
|
+
let json_value: serde_json::Value = serde_json::from_str(&json_str).unwrap();
|
|
127
|
+
assert!(json_value.get("ocr").is_some(), "Should have OCR config");
|
|
128
|
+
assert!(json_value.get("chunking").is_some(), "Should have chunking config");
|
|
129
|
+
}
|
|
130
|
+
}
|
|
131
|
+
|
|
132
|
+
/// Test config loading fails gracefully with invalid file path.
|
|
133
|
+
#[test]
|
|
134
|
+
fn test_load_config_from_invalid_path_fails_gracefully() {
|
|
135
|
+
unsafe {
|
|
136
|
+
let invalid_path = CString::new("/nonexistent/path/config.toml").unwrap();
|
|
137
|
+
let result = kreuzberg_load_extraction_config_from_file(invalid_path.as_ptr());
|
|
138
|
+
|
|
139
|
+
assert!(result.is_null(), "Result should be null for invalid path");
|
|
140
|
+
|
|
141
|
+
let error = get_last_error();
|
|
142
|
+
assert!(error.is_some(), "Should have error message");
|
|
143
|
+
let error_msg = error.unwrap();
|
|
144
|
+
assert!(
|
|
145
|
+
error_msg.contains("No such file") || error_msg.contains("not found"),
|
|
146
|
+
"Error should mention file not found: {}",
|
|
147
|
+
error_msg
|
|
148
|
+
);
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
|
|
152
|
+
/// Test config loading fails gracefully with null pointer.
|
|
153
|
+
#[test]
|
|
154
|
+
fn test_load_config_from_null_pointer_fails_gracefully() {
|
|
155
|
+
unsafe {
|
|
156
|
+
let result = kreuzberg_load_extraction_config_from_file(std::ptr::null());
|
|
157
|
+
|
|
158
|
+
assert!(result.is_null(), "Result should be null for null pointer");
|
|
159
|
+
|
|
160
|
+
let error = get_last_error();
|
|
161
|
+
assert!(error.is_some(), "Should have error message");
|
|
162
|
+
let error_msg = error.unwrap();
|
|
163
|
+
assert!(
|
|
164
|
+
error_msg.contains("null") || error_msg.contains("invalid"),
|
|
165
|
+
"Error should mention null/invalid: {}",
|
|
166
|
+
error_msg
|
|
167
|
+
);
|
|
168
|
+
}
|
|
169
|
+
}
|
|
170
|
+
|
|
171
|
+
/// Test config loading fails gracefully with malformed TOML.
|
|
172
|
+
#[test]
|
|
173
|
+
fn test_load_config_from_malformed_toml_fails_gracefully() {
|
|
174
|
+
let temp_dir = TempDir::new().unwrap();
|
|
175
|
+
let config_path = temp_dir.path().join("config.toml");
|
|
176
|
+
|
|
177
|
+
let malformed_toml = r#"
|
|
178
|
+
[ocr
|
|
179
|
+
enabled = true
|
|
180
|
+
"#;
|
|
181
|
+
|
|
182
|
+
fs::write(&config_path, malformed_toml).unwrap();
|
|
183
|
+
|
|
184
|
+
unsafe {
|
|
185
|
+
let path_cstr = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
186
|
+
let result = kreuzberg_load_extraction_config_from_file(path_cstr.as_ptr());
|
|
187
|
+
|
|
188
|
+
assert!(result.is_null(), "Result should be null for malformed TOML");
|
|
189
|
+
|
|
190
|
+
let error = get_last_error();
|
|
191
|
+
assert!(error.is_some(), "Should have error message");
|
|
192
|
+
let error_msg = error.unwrap();
|
|
193
|
+
assert!(
|
|
194
|
+
error_msg.contains("parse") || error_msg.contains("invalid") || error_msg.contains("TOML"),
|
|
195
|
+
"Error should mention parsing issue: {}",
|
|
196
|
+
error_msg
|
|
197
|
+
);
|
|
198
|
+
}
|
|
199
|
+
}
|
|
200
|
+
|
|
201
|
+
/// Test config loading fails gracefully with malformed JSON.
|
|
202
|
+
#[test]
|
|
203
|
+
fn test_load_config_from_malformed_json_fails_gracefully() {
|
|
204
|
+
let temp_dir = TempDir::new().unwrap();
|
|
205
|
+
let config_path = temp_dir.path().join("config.json");
|
|
206
|
+
|
|
207
|
+
let malformed_json = r#"
|
|
208
|
+
{
|
|
209
|
+
"ocr": {
|
|
210
|
+
"enabled": true
|
|
211
|
+
}
|
|
212
|
+
"chunking": {}
|
|
213
|
+
}
|
|
214
|
+
"#;
|
|
215
|
+
|
|
216
|
+
fs::write(&config_path, malformed_json).unwrap();
|
|
217
|
+
|
|
218
|
+
unsafe {
|
|
219
|
+
let path_cstr = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
220
|
+
let result = kreuzberg_load_extraction_config_from_file(path_cstr.as_ptr());
|
|
221
|
+
|
|
222
|
+
assert!(result.is_null(), "Result should be null for malformed JSON");
|
|
223
|
+
|
|
224
|
+
let error = get_last_error();
|
|
225
|
+
assert!(error.is_some(), "Should have error message");
|
|
226
|
+
let error_msg = error.unwrap();
|
|
227
|
+
assert!(
|
|
228
|
+
error_msg.contains("parse") || error_msg.contains("invalid") || error_msg.contains("JSON"),
|
|
229
|
+
"Error should mention parsing issue: {}",
|
|
230
|
+
error_msg
|
|
231
|
+
);
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
|
|
235
|
+
/// Test config loading with invalid UTF-8 in path.
|
|
236
|
+
#[test]
|
|
237
|
+
fn test_load_config_with_invalid_utf8_fails_gracefully() {
|
|
238
|
+
unsafe {
|
|
239
|
+
let invalid_bytes = vec![0xFF, 0xFE, 0xFD];
|
|
240
|
+
let invalid_cstr = CString::new(invalid_bytes).unwrap_or_else(|_| CString::new("").unwrap());
|
|
241
|
+
|
|
242
|
+
let result = kreuzberg_load_extraction_config_from_file(invalid_cstr.as_ptr());
|
|
243
|
+
|
|
244
|
+
assert!(result.is_null(), "Result should be null for invalid UTF-8");
|
|
245
|
+
|
|
246
|
+
let error = get_last_error();
|
|
247
|
+
assert!(error.is_some(), "Should have error message");
|
|
248
|
+
}
|
|
249
|
+
}
|
|
250
|
+
|
|
251
|
+
/// Test config loading with empty file.
|
|
252
|
+
#[test]
|
|
253
|
+
fn test_load_config_from_empty_file_uses_defaults() {
|
|
254
|
+
let temp_dir = TempDir::new().unwrap();
|
|
255
|
+
let config_path = temp_dir.path().join("config.toml");
|
|
256
|
+
|
|
257
|
+
fs::write(&config_path, "").unwrap();
|
|
258
|
+
|
|
259
|
+
unsafe {
|
|
260
|
+
let path_cstr = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
261
|
+
let result = kreuzberg_load_extraction_config_from_file(path_cstr.as_ptr());
|
|
262
|
+
|
|
263
|
+
assert!(!result.is_null(), "Result should not be null for empty file");
|
|
264
|
+
|
|
265
|
+
let json_str = c_str_to_string(result).expect("Should have valid JSON");
|
|
266
|
+
kreuzberg_free_string(result);
|
|
267
|
+
|
|
268
|
+
let json_value: serde_json::Value = serde_json::from_str(&json_str).unwrap();
|
|
269
|
+
assert!(json_value.is_object(), "Should be a JSON object");
|
|
270
|
+
}
|
|
271
|
+
}
|
|
272
|
+
|
|
273
|
+
/// Test format detection from file extension.
|
|
274
|
+
#[test]
|
|
275
|
+
fn test_config_format_detection_from_extension() {
|
|
276
|
+
let temp_dir = TempDir::new().unwrap();
|
|
277
|
+
|
|
278
|
+
let yml_path = temp_dir.path().join("config.yml");
|
|
279
|
+
let yaml_content = "ocr:\n enabled: true";
|
|
280
|
+
fs::write(&yml_path, yaml_content).unwrap();
|
|
281
|
+
|
|
282
|
+
unsafe {
|
|
283
|
+
let path_cstr = CString::new(yml_path.to_str().unwrap()).unwrap();
|
|
284
|
+
let result = kreuzberg_load_extraction_config_from_file(path_cstr.as_ptr());
|
|
285
|
+
assert!(!result.is_null(), ".yml extension should be recognized");
|
|
286
|
+
kreuzberg_free_string(result);
|
|
287
|
+
}
|
|
288
|
+
|
|
289
|
+
let json_path = temp_dir.path().join("config.json");
|
|
290
|
+
let json_content = r#"{"ocr": {"enabled": true}}"#;
|
|
291
|
+
fs::write(&json_path, json_content).unwrap();
|
|
292
|
+
|
|
293
|
+
unsafe {
|
|
294
|
+
let path_cstr = CString::new(json_path.to_str().unwrap()).unwrap();
|
|
295
|
+
let result = kreuzberg_load_extraction_config_from_file(path_cstr.as_ptr());
|
|
296
|
+
assert!(!result.is_null(), ".json extension should be recognized");
|
|
297
|
+
kreuzberg_free_string(result);
|
|
298
|
+
}
|
|
299
|
+
}
|
|
@@ -0,0 +1,346 @@
|
|
|
1
|
+
use std::ffi::{CStr, CString};
|
|
2
|
+
use std::fs;
|
|
3
|
+
use std::os::raw::c_char;
|
|
4
|
+
use std::ptr;
|
|
5
|
+
use tempfile::TempDir;
|
|
6
|
+
|
|
7
|
+
unsafe extern "C" {
|
|
8
|
+
fn kreuzberg_config_from_file(path: *const c_char) -> *mut std::ffi::c_void;
|
|
9
|
+
|
|
10
|
+
fn kreuzberg_config_discover() -> *mut std::ffi::c_void;
|
|
11
|
+
|
|
12
|
+
fn kreuzberg_last_error() -> *const c_char;
|
|
13
|
+
|
|
14
|
+
fn kreuzberg_free_config(config: *mut std::ffi::c_void);
|
|
15
|
+
}
|
|
16
|
+
|
|
17
|
+
#[test]
|
|
18
|
+
fn test_config_from_file_toml() {
|
|
19
|
+
unsafe {
|
|
20
|
+
let temp_dir = TempDir::new().unwrap();
|
|
21
|
+
let config_path = temp_dir.path().join("kreuzberg.toml");
|
|
22
|
+
|
|
23
|
+
let config_content = r#"
|
|
24
|
+
[ocr]
|
|
25
|
+
enabled = true
|
|
26
|
+
backend = "tesseract"
|
|
27
|
+
|
|
28
|
+
[chunking]
|
|
29
|
+
enabled = false
|
|
30
|
+
"#;
|
|
31
|
+
|
|
32
|
+
fs::write(&config_path, config_content).unwrap();
|
|
33
|
+
|
|
34
|
+
let path_str = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
35
|
+
let config_ptr = kreuzberg_config_from_file(path_str.as_ptr());
|
|
36
|
+
|
|
37
|
+
assert!(!config_ptr.is_null(), "Config should be loaded successfully");
|
|
38
|
+
|
|
39
|
+
kreuzberg_free_config(config_ptr);
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
#[test]
|
|
44
|
+
fn test_config_from_file_yaml() {
|
|
45
|
+
unsafe {
|
|
46
|
+
let temp_dir = TempDir::new().unwrap();
|
|
47
|
+
let config_path = temp_dir.path().join("kreuzberg.yaml");
|
|
48
|
+
|
|
49
|
+
let config_content = r#"
|
|
50
|
+
ocr:
|
|
51
|
+
enabled: true
|
|
52
|
+
backend: tesseract
|
|
53
|
+
|
|
54
|
+
chunking:
|
|
55
|
+
enabled: false
|
|
56
|
+
"#;
|
|
57
|
+
|
|
58
|
+
fs::write(&config_path, config_content).unwrap();
|
|
59
|
+
|
|
60
|
+
let path_str = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
61
|
+
let config_ptr = kreuzberg_config_from_file(path_str.as_ptr());
|
|
62
|
+
|
|
63
|
+
assert!(!config_ptr.is_null(), "Config should be loaded successfully");
|
|
64
|
+
|
|
65
|
+
kreuzberg_free_config(config_ptr);
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
|
|
69
|
+
#[test]
|
|
70
|
+
fn test_config_from_file_json() {
|
|
71
|
+
unsafe {
|
|
72
|
+
let temp_dir = TempDir::new().unwrap();
|
|
73
|
+
let config_path = temp_dir.path().join("kreuzberg.json");
|
|
74
|
+
|
|
75
|
+
let config_content = r#"
|
|
76
|
+
{
|
|
77
|
+
"ocr": {
|
|
78
|
+
"enabled": true,
|
|
79
|
+
"backend": "tesseract"
|
|
80
|
+
},
|
|
81
|
+
"chunking": {
|
|
82
|
+
"enabled": false
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
"#;
|
|
86
|
+
|
|
87
|
+
fs::write(&config_path, config_content).unwrap();
|
|
88
|
+
|
|
89
|
+
let path_str = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
90
|
+
let config_ptr = kreuzberg_config_from_file(path_str.as_ptr());
|
|
91
|
+
|
|
92
|
+
assert!(!config_ptr.is_null(), "Config should be loaded successfully");
|
|
93
|
+
|
|
94
|
+
kreuzberg_free_config(config_ptr);
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
#[test]
|
|
99
|
+
fn test_config_from_file_null_path() {
|
|
100
|
+
unsafe {
|
|
101
|
+
let config_ptr = kreuzberg_config_from_file(ptr::null());
|
|
102
|
+
|
|
103
|
+
assert!(config_ptr.is_null(), "Should return NULL for NULL path");
|
|
104
|
+
|
|
105
|
+
let error = kreuzberg_last_error();
|
|
106
|
+
assert!(!error.is_null());
|
|
107
|
+
let error_str = CStr::from_ptr(error).to_str().unwrap();
|
|
108
|
+
assert!(error_str.contains("NULL"), "Error should mention NULL: {}", error_str);
|
|
109
|
+
}
|
|
110
|
+
}
|
|
111
|
+
|
|
112
|
+
#[test]
|
|
113
|
+
fn test_config_from_file_nonexistent() {
|
|
114
|
+
unsafe {
|
|
115
|
+
let path = CString::new("/nonexistent/path/kreuzberg.toml").unwrap();
|
|
116
|
+
let config_ptr = kreuzberg_config_from_file(path.as_ptr());
|
|
117
|
+
|
|
118
|
+
assert!(config_ptr.is_null(), "Should return NULL for nonexistent file");
|
|
119
|
+
|
|
120
|
+
let error = kreuzberg_last_error();
|
|
121
|
+
assert!(!error.is_null());
|
|
122
|
+
let error_str = CStr::from_ptr(error).to_str().unwrap();
|
|
123
|
+
assert!(
|
|
124
|
+
error_str.contains("IO") || error_str.contains("not found") || error_str.contains("No such"),
|
|
125
|
+
"Error should indicate file not found: {}",
|
|
126
|
+
error_str
|
|
127
|
+
);
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
|
|
131
|
+
#[test]
|
|
132
|
+
fn test_config_from_file_invalid_toml() {
|
|
133
|
+
unsafe {
|
|
134
|
+
let temp_dir = TempDir::new().unwrap();
|
|
135
|
+
let config_path = temp_dir.path().join("invalid.toml");
|
|
136
|
+
|
|
137
|
+
let config_content = r#"
|
|
138
|
+
[ocr
|
|
139
|
+
enabled = true # Missing closing bracket
|
|
140
|
+
"#;
|
|
141
|
+
|
|
142
|
+
fs::write(&config_path, config_content).unwrap();
|
|
143
|
+
|
|
144
|
+
let path_str = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
145
|
+
let config_ptr = kreuzberg_config_from_file(path_str.as_ptr());
|
|
146
|
+
|
|
147
|
+
assert!(config_ptr.is_null(), "Should return NULL for invalid TOML");
|
|
148
|
+
|
|
149
|
+
let error = kreuzberg_last_error();
|
|
150
|
+
assert!(!error.is_null());
|
|
151
|
+
}
|
|
152
|
+
}
|
|
153
|
+
|
|
154
|
+
#[test]
|
|
155
|
+
fn test_config_from_file_invalid_json() {
|
|
156
|
+
unsafe {
|
|
157
|
+
let temp_dir = TempDir::new().unwrap();
|
|
158
|
+
let config_path = temp_dir.path().join("invalid.json");
|
|
159
|
+
|
|
160
|
+
let config_content = r#"
|
|
161
|
+
{
|
|
162
|
+
"ocr": {
|
|
163
|
+
"enabled": true,
|
|
164
|
+
} // Trailing comma is invalid in strict JSON
|
|
165
|
+
}
|
|
166
|
+
"#;
|
|
167
|
+
|
|
168
|
+
fs::write(&config_path, config_content).unwrap();
|
|
169
|
+
|
|
170
|
+
let path_str = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
171
|
+
let config_ptr = kreuzberg_config_from_file(path_str.as_ptr());
|
|
172
|
+
|
|
173
|
+
assert!(config_ptr.is_null(), "Should return NULL for invalid JSON");
|
|
174
|
+
|
|
175
|
+
let error = kreuzberg_last_error();
|
|
176
|
+
assert!(!error.is_null());
|
|
177
|
+
}
|
|
178
|
+
}
|
|
179
|
+
|
|
180
|
+
#[test]
|
|
181
|
+
fn test_config_from_file_no_extension() {
|
|
182
|
+
unsafe {
|
|
183
|
+
let temp_dir = TempDir::new().unwrap();
|
|
184
|
+
let config_path = temp_dir.path().join("kreuzberg");
|
|
185
|
+
|
|
186
|
+
fs::write(&config_path, "some content").unwrap();
|
|
187
|
+
|
|
188
|
+
let path_str = CString::new(config_path.to_str().unwrap()).unwrap();
|
|
189
|
+
let config_ptr = kreuzberg_config_from_file(path_str.as_ptr());
|
|
190
|
+
|
|
191
|
+
assert!(config_ptr.is_null(), "Should return NULL for file without extension");
|
|
192
|
+
|
|
193
|
+
let error = kreuzberg_last_error();
|
|
194
|
+
assert!(!error.is_null());
|
|
195
|
+
let error_str = CStr::from_ptr(error).to_str().unwrap();
|
|
196
|
+
assert!(
|
|
197
|
+
error_str.contains("extension") || error_str.contains("format"),
|
|
198
|
+
"Error should mention extension: {}",
|
|
199
|
+
error_str
|
|
200
|
+
);
|
|
201
|
+
}
|
|
202
|
+
}
|
|
203
|
+
|
|
204
|
+
#[test]
|
|
205
|
+
fn test_config_from_file_invalid_utf8_path() {
|
|
206
|
+
unsafe {
|
|
207
|
+
let invalid_path = b"/tmp/test\xFF\xFEinvalid.toml\0";
|
|
208
|
+
|
|
209
|
+
let config_ptr = kreuzberg_config_from_file(invalid_path.as_ptr() as *const c_char);
|
|
210
|
+
|
|
211
|
+
assert!(config_ptr.is_null(), "Should return NULL for invalid UTF-8 path");
|
|
212
|
+
|
|
213
|
+
let error = kreuzberg_last_error();
|
|
214
|
+
assert!(!error.is_null());
|
|
215
|
+
let error_str = CStr::from_ptr(error).to_str().unwrap();
|
|
216
|
+
assert!(error_str.contains("UTF-8"));
|
|
217
|
+
}
|
|
218
|
+
}
|
|
219
|
+
|
|
220
|
+
#[test]
|
|
221
|
+
fn test_config_discover_not_found() {
|
|
222
|
+
unsafe {
|
|
223
|
+
let temp_dir = TempDir::new().unwrap();
|
|
224
|
+
let original_dir = std::env::current_dir().unwrap();
|
|
225
|
+
|
|
226
|
+
std::env::set_current_dir(&temp_dir).unwrap();
|
|
227
|
+
|
|
228
|
+
let config_ptr = kreuzberg_config_discover();
|
|
229
|
+
|
|
230
|
+
assert!(config_ptr.is_null(), "Should return NULL when no config found");
|
|
231
|
+
|
|
232
|
+
let error = kreuzberg_last_error();
|
|
233
|
+
if !error.is_null() {
|
|
234
|
+
let error_str = CStr::from_ptr(error).to_str().unwrap();
|
|
235
|
+
assert!(
|
|
236
|
+
error_str.is_empty() || error_str.contains("not found") || error_str.contains("IO"),
|
|
237
|
+
"Error should be empty or indicate not found: {}",
|
|
238
|
+
error_str
|
|
239
|
+
);
|
|
240
|
+
}
|
|
241
|
+
|
|
242
|
+
std::env::set_current_dir(original_dir).unwrap();
|
|
243
|
+
}
|
|
244
|
+
}
|
|
245
|
+
|
|
246
|
+
#[test]
|
|
247
|
+
fn test_config_discover_toml() {
|
|
248
|
+
unsafe {
|
|
249
|
+
let temp_dir = TempDir::new().unwrap();
|
|
250
|
+
let config_path = temp_dir.path().join("kreuzberg.toml");
|
|
251
|
+
|
|
252
|
+
let config_content = r#"
|
|
253
|
+
[ocr]
|
|
254
|
+
enabled = true
|
|
255
|
+
"#;
|
|
256
|
+
|
|
257
|
+
fs::write(&config_path, config_content).unwrap();
|
|
258
|
+
|
|
259
|
+
let original_dir = std::env::current_dir().unwrap();
|
|
260
|
+
std::env::set_current_dir(&temp_dir).unwrap();
|
|
261
|
+
|
|
262
|
+
let config_ptr = kreuzberg_config_discover();
|
|
263
|
+
|
|
264
|
+
assert!(!config_ptr.is_null(), "Should discover config in current directory");
|
|
265
|
+
|
|
266
|
+
kreuzberg_free_config(config_ptr);
|
|
267
|
+
|
|
268
|
+
std::env::set_current_dir(original_dir).unwrap();
|
|
269
|
+
}
|
|
270
|
+
}
|
|
271
|
+
|
|
272
|
+
#[test]
|
|
273
|
+
fn test_config_discover_yaml() {
|
|
274
|
+
unsafe {
|
|
275
|
+
let temp_dir = TempDir::new().unwrap();
|
|
276
|
+
let config_path = temp_dir.path().join("kreuzberg.yaml");
|
|
277
|
+
|
|
278
|
+
let config_content = r#"
|
|
279
|
+
ocr:
|
|
280
|
+
enabled: true
|
|
281
|
+
"#;
|
|
282
|
+
|
|
283
|
+
fs::write(&config_path, config_content).unwrap();
|
|
284
|
+
|
|
285
|
+
let original_dir = std::env::current_dir().unwrap();
|
|
286
|
+
std::env::set_current_dir(&temp_dir).unwrap();
|
|
287
|
+
|
|
288
|
+
let config_ptr = kreuzberg_config_discover();
|
|
289
|
+
|
|
290
|
+
assert!(!config_ptr.is_null(), "Should discover YAML config");
|
|
291
|
+
|
|
292
|
+
kreuzberg_free_config(config_ptr);
|
|
293
|
+
|
|
294
|
+
std::env::set_current_dir(original_dir).unwrap();
|
|
295
|
+
}
|
|
296
|
+
}
|
|
297
|
+
|
|
298
|
+
#[test]
|
|
299
|
+
fn test_config_discover_parent_directory() {
|
|
300
|
+
unsafe {
|
|
301
|
+
let temp_dir = TempDir::new().unwrap();
|
|
302
|
+
let config_path = temp_dir.path().join("kreuzberg.toml");
|
|
303
|
+
|
|
304
|
+
let config_content = r#"
|
|
305
|
+
[ocr]
|
|
306
|
+
enabled = true
|
|
307
|
+
"#;
|
|
308
|
+
|
|
309
|
+
fs::write(&config_path, config_content).unwrap();
|
|
310
|
+
|
|
311
|
+
let subdir = temp_dir.path().join("subdir");
|
|
312
|
+
fs::create_dir(&subdir).unwrap();
|
|
313
|
+
|
|
314
|
+
let original_dir = std::env::current_dir().unwrap();
|
|
315
|
+
std::env::set_current_dir(&subdir).unwrap();
|
|
316
|
+
|
|
317
|
+
let config_ptr = kreuzberg_config_discover();
|
|
318
|
+
|
|
319
|
+
assert!(!config_ptr.is_null(), "Should discover config in parent directory");
|
|
320
|
+
|
|
321
|
+
kreuzberg_free_config(config_ptr);
|
|
322
|
+
|
|
323
|
+
std::env::set_current_dir(original_dir).unwrap();
|
|
324
|
+
}
|
|
325
|
+
}
|
|
326
|
+
|
|
327
|
+
#[test]
|
|
328
|
+
fn test_config_discover_preference_order() {
|
|
329
|
+
unsafe {
|
|
330
|
+
let temp_dir = TempDir::new().unwrap();
|
|
331
|
+
|
|
332
|
+
fs::write(temp_dir.path().join("kreuzberg.toml"), "[ocr]\nenabled = true").unwrap();
|
|
333
|
+
fs::write(temp_dir.path().join("kreuzberg.yaml"), "ocr:\n enabled: false").unwrap();
|
|
334
|
+
|
|
335
|
+
let original_dir = std::env::current_dir().unwrap();
|
|
336
|
+
std::env::set_current_dir(&temp_dir).unwrap();
|
|
337
|
+
|
|
338
|
+
let config_ptr = kreuzberg_config_discover();
|
|
339
|
+
|
|
340
|
+
assert!(!config_ptr.is_null(), "Should discover a config file");
|
|
341
|
+
|
|
342
|
+
kreuzberg_free_config(config_ptr);
|
|
343
|
+
|
|
344
|
+
std::env::set_current_dir(original_dir).unwrap();
|
|
345
|
+
}
|
|
346
|
+
}
|