kreuzberg 4.0.0.pre.rc.13 → 4.0.0.pre.rc.15
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 +538 -538
- data/Gemfile +8 -8
- data/Gemfile.lock +104 -2
- data/README.md +454 -454
- data/Rakefile +33 -25
- data/Steepfile +47 -47
- data/examples/async_patterns.rb +341 -341
- data/ext/kreuzberg_rb/extconf.rb +45 -45
- data/ext/kreuzberg_rb/native/.cargo/config.toml +2 -2
- data/ext/kreuzberg_rb/native/Cargo.lock +6750 -6941
- data/ext/kreuzberg_rb/native/Cargo.toml +53 -54
- data/ext/kreuzberg_rb/native/README.md +425 -425
- data/ext/kreuzberg_rb/native/build.rs +52 -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 +3158 -3158
- data/extconf.rb +28 -28
- data/kreuzberg.gemspec +214 -214
- data/lib/kreuzberg/api_proxy.rb +142 -142
- data/lib/kreuzberg/cache_api.rb +81 -81
- data/lib/kreuzberg/cli.rb +55 -55
- data/lib/kreuzberg/cli_proxy.rb +127 -127
- data/lib/kreuzberg/config.rb +724 -724
- data/lib/kreuzberg/error_context.rb +80 -80
- data/lib/kreuzberg/errors.rb +118 -118
- data/lib/kreuzberg/extraction_api.rb +340 -340
- data/lib/kreuzberg/mcp_proxy.rb +186 -186
- data/lib/kreuzberg/ocr_backend_protocol.rb +113 -113
- data/lib/kreuzberg/post_processor_protocol.rb +86 -86
- data/lib/kreuzberg/result.rb +279 -279
- data/lib/kreuzberg/setup_lib_path.rb +80 -80
- data/lib/kreuzberg/validator_protocol.rb +89 -89
- data/lib/kreuzberg/version.rb +5 -5
- data/lib/kreuzberg.rb +109 -109
- data/lib/{pdfium.dll → libpdfium.so} +0 -0
- data/sig/kreuzberg/internal.rbs +184 -184
- data/sig/kreuzberg.rbs +546 -546
- 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_spec.rb +345 -345
- data/spec/binding/config_validation_spec.rb +283 -283
- data/spec/binding/error_handling_spec.rb +213 -213
- data/spec/binding/errors_spec.rb +66 -66
- 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 +274 -274
- data/spec/fixtures/config.toml +39 -39
- data/spec/fixtures/config.yaml +41 -41
- data/spec/fixtures/invalid_config.toml +4 -4
- data/spec/smoke/package_spec.rb +178 -178
- data/spec/spec_helper.rb +42 -42
- data/vendor/Cargo.toml +2 -2
- data/vendor/kreuzberg/Cargo.toml +5 -5
- data/vendor/kreuzberg/README.md +230 -230
- data/vendor/kreuzberg/benches/otel_overhead.rs +48 -48
- data/vendor/kreuzberg/build.rs +887 -843
- data/vendor/kreuzberg/src/api/error.rs +81 -81
- data/vendor/kreuzberg/src/api/handlers.rs +199 -199
- data/vendor/kreuzberg/src/api/mod.rs +87 -79
- data/vendor/kreuzberg/src/api/server.rs +353 -353
- data/vendor/kreuzberg/src/api/types.rs +170 -170
- data/vendor/kreuzberg/src/cache/mod.rs +1167 -1167
- data/vendor/kreuzberg/src/chunking/mod.rs +1877 -1877
- data/vendor/kreuzberg/src/chunking/processor.rs +220 -220
- data/vendor/kreuzberg/src/core/batch_mode.rs +95 -95
- data/vendor/kreuzberg/src/core/config.rs +1080 -1080
- data/vendor/kreuzberg/src/core/extractor.rs +1156 -1156
- 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 +47 -47
- data/vendor/kreuzberg/src/core/pipeline.rs +1184 -1184
- data/vendor/kreuzberg/src/embeddings.rs +500 -500
- data/vendor/kreuzberg/src/error.rs +431 -431
- data/vendor/kreuzberg/src/extraction/archive.rs +954 -954
- data/vendor/kreuzberg/src/extraction/docx.rs +398 -398
- data/vendor/kreuzberg/src/extraction/email.rs +854 -854
- data/vendor/kreuzberg/src/extraction/excel.rs +688 -688
- data/vendor/kreuzberg/src/extraction/html.rs +634 -601
- data/vendor/kreuzberg/src/extraction/image.rs +491 -491
- data/vendor/kreuzberg/src/extraction/libreoffice.rs +574 -574
- data/vendor/kreuzberg/src/extraction/markdown.rs +213 -213
- data/vendor/kreuzberg/src/extraction/mod.rs +81 -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 -284
- data/vendor/kreuzberg/src/extraction/pptx.rs +3100 -3100
- data/vendor/kreuzberg/src/extraction/structured.rs +490 -490
- data/vendor/kreuzberg/src/extraction/table.rs +328 -328
- data/vendor/kreuzberg/src/extraction/text.rs +269 -269
- data/vendor/kreuzberg/src/extraction/xml.rs +333 -333
- data/vendor/kreuzberg/src/extractors/archive.rs +447 -447
- data/vendor/kreuzberg/src/extractors/bibtex.rs +470 -470
- data/vendor/kreuzberg/src/extractors/docbook.rs +504 -504
- data/vendor/kreuzberg/src/extractors/docx.rs +400 -400
- data/vendor/kreuzberg/src/extractors/email.rs +157 -157
- data/vendor/kreuzberg/src/extractors/epub.rs +708 -708
- data/vendor/kreuzberg/src/extractors/excel.rs +345 -345
- data/vendor/kreuzberg/src/extractors/fictionbook.rs +492 -492
- data/vendor/kreuzberg/src/extractors/html.rs +407 -407
- data/vendor/kreuzberg/src/extractors/image.rs +219 -219
- data/vendor/kreuzberg/src/extractors/jats.rs +1054 -1054
- data/vendor/kreuzberg/src/extractors/jupyter.rs +368 -368
- data/vendor/kreuzberg/src/extractors/latex.rs +653 -653
- data/vendor/kreuzberg/src/extractors/markdown.rs +701 -701
- data/vendor/kreuzberg/src/extractors/mod.rs +429 -429
- data/vendor/kreuzberg/src/extractors/odt.rs +628 -628
- data/vendor/kreuzberg/src/extractors/opml.rs +635 -635
- data/vendor/kreuzberg/src/extractors/orgmode.rs +529 -529
- data/vendor/kreuzberg/src/extractors/pdf.rs +749 -749
- data/vendor/kreuzberg/src/extractors/pptx.rs +267 -267
- data/vendor/kreuzberg/src/extractors/rst.rs +577 -577
- data/vendor/kreuzberg/src/extractors/rtf.rs +809 -809
- 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 -142
- data/vendor/kreuzberg/src/extractors/text.rs +265 -265
- data/vendor/kreuzberg/src/extractors/typst.rs +651 -651
- data/vendor/kreuzberg/src/extractors/xml.rs +147 -147
- 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 -275
- 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 -985
- data/vendor/kreuzberg/src/language_detection/processor.rs +219 -219
- data/vendor/kreuzberg/src/lib.rs +113 -113
- data/vendor/kreuzberg/src/mcp/mod.rs +35 -35
- data/vendor/kreuzberg/src/mcp/server.rs +2076 -2076
- 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/mod.rs +58 -58
- data/vendor/kreuzberg/src/ocr/processor.rs +863 -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 +452 -452
- 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 +44 -44
- data/vendor/kreuzberg/src/pdf/bundled.rs +452 -346
- data/vendor/kreuzberg/src/pdf/error.rs +130 -130
- data/vendor/kreuzberg/src/pdf/images.rs +139 -139
- data/vendor/kreuzberg/src/pdf/metadata.rs +489 -489
- data/vendor/kreuzberg/src/pdf/mod.rs +68 -68
- data/vendor/kreuzberg/src/pdf/rendering.rs +368 -368
- data/vendor/kreuzberg/src/pdf/table.rs +420 -420
- data/vendor/kreuzberg/src/pdf/text.rs +240 -240
- data/vendor/kreuzberg/src/plugins/extractor.rs +1044 -1044
- data/vendor/kreuzberg/src/plugins/mod.rs +212 -212
- data/vendor/kreuzberg/src/plugins/ocr.rs +639 -639
- data/vendor/kreuzberg/src/plugins/processor.rs +650 -650
- data/vendor/kreuzberg/src/plugins/registry.rs +1339 -1339
- data/vendor/kreuzberg/src/plugins/traits.rs +258 -258
- data/vendor/kreuzberg/src/plugins/validator.rs +967 -967
- data/vendor/kreuzberg/src/stopwords/mod.rs +1470 -1470
- data/vendor/kreuzberg/src/text/mod.rs +25 -25
- data/vendor/kreuzberg/src/text/quality.rs +697 -697
- data/vendor/kreuzberg/src/text/quality_processor.rs +219 -219
- data/vendor/kreuzberg/src/text/string_utils.rs +217 -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 +796 -796
- data/vendor/kreuzberg/src/text/token_reduction/filters.rs +902 -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 +147 -147
- data/vendor/kreuzberg/src/types.rs +1055 -1055
- data/vendor/kreuzberg/src/utils/mod.rs +17 -17
- data/vendor/kreuzberg/src/utils/quality.rs +959 -959
- 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_extract_multipart.rs +52 -52
- data/vendor/kreuzberg/tests/api_tests.rs +966 -966
- data/vendor/kreuzberg/tests/archive_integration.rs +545 -545
- data/vendor/kreuzberg/tests/batch_orchestration.rs +556 -556
- data/vendor/kreuzberg/tests/batch_processing.rs +318 -318
- data/vendor/kreuzberg/tests/bibtex_parity_test.rs +421 -421
- data/vendor/kreuzberg/tests/concurrency_stress.rs +533 -533
- data/vendor/kreuzberg/tests/config_features.rs +612 -612
- data/vendor/kreuzberg/tests/config_loading_tests.rs +416 -416
- data/vendor/kreuzberg/tests/core_integration.rs +510 -510
- data/vendor/kreuzberg/tests/csv_integration.rs +414 -414
- data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +500 -500
- 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 -327
- data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +275 -275
- data/vendor/kreuzberg/tests/error_handling.rs +402 -402
- data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +228 -228
- data/vendor/kreuzberg/tests/format_integration.rs +165 -164
- data/vendor/kreuzberg/tests/helpers/mod.rs +142 -142
- data/vendor/kreuzberg/tests/html_table_test.rs +551 -551
- data/vendor/kreuzberg/tests/image_integration.rs +255 -255
- 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 -429
- data/vendor/kreuzberg/tests/ocr_configuration.rs +514 -514
- data/vendor/kreuzberg/tests/ocr_errors.rs +698 -698
- data/vendor/kreuzberg/tests/ocr_quality.rs +629 -629
- data/vendor/kreuzberg/tests/ocr_stress.rs +469 -469
- data/vendor/kreuzberg/tests/odt_extractor_tests.rs +674 -674
- data/vendor/kreuzberg/tests/opml_extractor_tests.rs +616 -616
- data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +822 -822
- data/vendor/kreuzberg/tests/pdf_integration.rs +45 -45
- data/vendor/kreuzberg/tests/pdfium_linking.rs +374 -374
- data/vendor/kreuzberg/tests/pipeline_integration.rs +1436 -1436
- data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +776 -776
- data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +560 -560
- data/vendor/kreuzberg/tests/plugin_system.rs +927 -927
- data/vendor/kreuzberg/tests/plugin_validator_test.rs +783 -783
- data/vendor/kreuzberg/tests/registry_integration_tests.rs +587 -587
- data/vendor/kreuzberg/tests/rst_extractor_tests.rs +694 -694
- data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +775 -775
- data/vendor/kreuzberg/tests/security_validation.rs +416 -416
- data/vendor/kreuzberg/tests/stopwords_integration_test.rs +888 -888
- data/vendor/kreuzberg/tests/test_fastembed.rs +631 -631
- data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +1260 -1260
- data/vendor/kreuzberg/tests/typst_extractor_tests.rs +648 -648
- data/vendor/kreuzberg/tests/xlsx_metadata_extraction_test.rs +87 -87
- data/vendor/kreuzberg-tesseract/.commitlintrc.json +13 -13
- data/vendor/kreuzberg-tesseract/.crate-ignore +2 -2
- data/vendor/kreuzberg-tesseract/Cargo.lock +2933 -2933
- data/vendor/kreuzberg-tesseract/Cargo.toml +2 -2
- data/vendor/kreuzberg-tesseract/LICENSE +22 -22
- data/vendor/kreuzberg-tesseract/README.md +399 -399
- data/vendor/kreuzberg-tesseract/build.rs +1354 -1354
- data/vendor/kreuzberg-tesseract/patches/README.md +71 -71
- data/vendor/kreuzberg-tesseract/patches/tesseract.diff +199 -199
- data/vendor/kreuzberg-tesseract/src/api.rs +1371 -1371
- data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +77 -77
- data/vendor/kreuzberg-tesseract/src/enums.rs +297 -297
- data/vendor/kreuzberg-tesseract/src/error.rs +81 -81
- data/vendor/kreuzberg-tesseract/src/lib.rs +145 -145
- data/vendor/kreuzberg-tesseract/src/monitor.rs +57 -57
- data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +197 -197
- data/vendor/kreuzberg-tesseract/src/page_iterator.rs +253 -253
- data/vendor/kreuzberg-tesseract/src/result_iterator.rs +286 -286
- data/vendor/kreuzberg-tesseract/src/result_renderer.rs +183 -183
- data/vendor/kreuzberg-tesseract/tests/integration_test.rs +211 -211
- data/vendor/rb-sys/.cargo_vcs_info.json +5 -5
- data/vendor/rb-sys/Cargo.lock +393 -393
- data/vendor/rb-sys/Cargo.toml +70 -70
- data/vendor/rb-sys/Cargo.toml.orig +57 -57
- data/vendor/rb-sys/LICENSE-APACHE +190 -190
- data/vendor/rb-sys/LICENSE-MIT +21 -21
- data/vendor/rb-sys/build/features.rs +111 -111
- data/vendor/rb-sys/build/main.rs +286 -286
- data/vendor/rb-sys/build/stable_api_config.rs +155 -155
- data/vendor/rb-sys/build/version.rs +50 -50
- data/vendor/rb-sys/readme.md +36 -36
- data/vendor/rb-sys/src/bindings.rs +21 -21
- data/vendor/rb-sys/src/hidden.rs +11 -11
- data/vendor/rb-sys/src/lib.rs +35 -35
- data/vendor/rb-sys/src/macros.rs +371 -371
- data/vendor/rb-sys/src/memory.rs +53 -53
- data/vendor/rb-sys/src/ruby_abi_version.rs +38 -38
- data/vendor/rb-sys/src/special_consts.rs +31 -31
- data/vendor/rb-sys/src/stable_api/compiled.c +179 -179
- data/vendor/rb-sys/src/stable_api/compiled.rs +257 -257
- data/vendor/rb-sys/src/stable_api/ruby_2_7.rs +324 -324
- data/vendor/rb-sys/src/stable_api/ruby_3_0.rs +332 -332
- data/vendor/rb-sys/src/stable_api/ruby_3_1.rs +325 -325
- data/vendor/rb-sys/src/stable_api/ruby_3_2.rs +323 -323
- data/vendor/rb-sys/src/stable_api/ruby_3_3.rs +339 -339
- data/vendor/rb-sys/src/stable_api/ruby_3_4.rs +339 -339
- data/vendor/rb-sys/src/stable_api.rs +260 -260
- data/vendor/rb-sys/src/symbol.rs +31 -31
- data/vendor/rb-sys/src/tracking_allocator.rs +330 -330
- data/vendor/rb-sys/src/utils.rs +89 -89
- data/vendor/rb-sys/src/value_type.rs +7 -7
- metadata +81 -22
- data/vendor/kreuzberg-ffi/Cargo.toml +0 -63
- data/vendor/kreuzberg-ffi/README.md +0 -851
- data/vendor/kreuzberg-ffi/build.rs +0 -176
- data/vendor/kreuzberg-ffi/cbindgen.toml +0 -27
- data/vendor/kreuzberg-ffi/kreuzberg-ffi-install.pc +0 -12
- data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +0 -12
- data/vendor/kreuzberg-ffi/kreuzberg.h +0 -1087
- data/vendor/kreuzberg-ffi/src/lib.rs +0 -3616
- data/vendor/kreuzberg-ffi/src/panic_shield.rs +0 -247
- data/vendor/kreuzberg-ffi/tests.disabled/README.md +0 -48
- data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +0 -299
- data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +0 -346
- data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +0 -232
- data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +0 -470
|
@@ -1,697 +1,697 @@
|
|
|
1
|
-
use once_cell::sync::Lazy;
|
|
2
|
-
use regex::Regex;
|
|
3
|
-
use std::borrow::Cow;
|
|
4
|
-
use std::collections::HashMap;
|
|
5
|
-
|
|
6
|
-
use crate::utils::quality::{collapse_scattered_ascii, normalize_whitespace_ascii};
|
|
7
|
-
|
|
8
|
-
// ============================================================================
|
|
9
|
-
// ============================================================================
|
|
10
|
-
|
|
11
|
-
const OCR_PENALTY_WEIGHT: f64 = 0.3;
|
|
12
|
-
const SCRIPT_PENALTY_WEIGHT: f64 = 0.2;
|
|
13
|
-
const NAV_PENALTY_WEIGHT: f64 = 0.1;
|
|
14
|
-
const STRUCTURE_BONUS_WEIGHT: f64 = 0.2;
|
|
15
|
-
const METADATA_BONUS_WEIGHT: f64 = 0.1;
|
|
16
|
-
|
|
17
|
-
const MIN_TEXT_LENGTH: usize = 10;
|
|
18
|
-
const LARGE_TEXT_LENGTH: usize = 1000;
|
|
19
|
-
const MIN_SENTENCE_WORDS: f64 = 10.0;
|
|
20
|
-
const MAX_SENTENCE_WORDS: f64 = 30.0;
|
|
21
|
-
const MIN_PARAGRAPH_WORDS: f64 = 50.0;
|
|
22
|
-
const MAX_PARAGRAPH_WORDS: f64 = 300.0;
|
|
23
|
-
|
|
24
|
-
static SCATTERED_CHARS_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
25
|
-
Regex::new(r"\b[a-zA-Z]\s{2,}[a-zA-Z]\s{2,}[a-zA-Z]\b")
|
|
26
|
-
.expect("Scattered chars regex pattern is valid and should compile")
|
|
27
|
-
});
|
|
28
|
-
static REPEATED_PUNCT_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
29
|
-
Regex::new(r"[.]{3,}|[_]{3,}").expect("Repeated punctuation regex pattern is valid and should compile")
|
|
30
|
-
});
|
|
31
|
-
static DASH_PATTERN: Lazy<Regex> =
|
|
32
|
-
Lazy::new(|| Regex::new(r"[-]{3,}").expect("Dash pattern regex is valid and should compile"));
|
|
33
|
-
static ISOLATED_PUNCT_PATTERN: Lazy<Regex> =
|
|
34
|
-
Lazy::new(|| Regex::new(r"\s[.,;:!?]\s").expect("Isolated punctuation regex pattern is valid and should compile"));
|
|
35
|
-
static MALFORMED_WORDS_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
36
|
-
Regex::new(r"\b[a-zA-Z]+[0-9]+[a-zA-Z]+[a-zA-Z0-9]*\b")
|
|
37
|
-
.expect("Malformed words regex pattern is valid and should compile")
|
|
38
|
-
});
|
|
39
|
-
static EXCESSIVE_WHITESPACE_PATTERN: Lazy<Regex> =
|
|
40
|
-
Lazy::new(|| Regex::new(r"\s{3,}").expect("Excessive whitespace regex pattern is valid and should compile"));
|
|
41
|
-
|
|
42
|
-
static JS_FUNCTION_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
43
|
-
Regex::new(r"(?i)function\s+\w+\s*\([^)]*\)\s*\{[^}]*\}")
|
|
44
|
-
.expect("JavaScript function regex pattern is valid and should compile")
|
|
45
|
-
});
|
|
46
|
-
static CSS_RULES_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
47
|
-
Regex::new(r"(?i)\.[a-zA-Z][\w-]*\s*\{[^}]*\}").expect("CSS rules regex pattern is valid and should compile")
|
|
48
|
-
});
|
|
49
|
-
static SCRIPT_TAG_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
50
|
-
Regex::new(r"(?is)<script[^>]*>.*?</script>").expect("Script tag regex pattern is valid and should compile")
|
|
51
|
-
});
|
|
52
|
-
static STYLE_TAG_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
53
|
-
Regex::new(r"(?is)<style[^>]*>.*?</style>").expect("Style tag regex pattern is valid and should compile")
|
|
54
|
-
});
|
|
55
|
-
|
|
56
|
-
static NAV_WORDS_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
57
|
-
Regex::new(r"(?i)\b(?:Skip to main content|Back to top|Main navigation|Site navigation)\b")
|
|
58
|
-
.expect("Navigation words regex pattern is valid and should compile")
|
|
59
|
-
});
|
|
60
|
-
static BREADCRUMB_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
61
|
-
Regex::new(r"(?:Home\s*[>»]\s*|[>»]\s*){2,}").expect("Breadcrumb regex pattern is valid and should compile")
|
|
62
|
-
});
|
|
63
|
-
static PAGINATION_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
64
|
-
Regex::new(r"(?i)\b(?:Page \d+ of \d+|First page|Last page|Previous page|Next page|^\d+ of \d+$)\b")
|
|
65
|
-
.expect("Pagination regex pattern is valid and should compile")
|
|
66
|
-
});
|
|
67
|
-
|
|
68
|
-
static SENTENCE_DETECT: Lazy<Regex> =
|
|
69
|
-
Lazy::new(|| Regex::new(r"[.!?]\s+[A-Z]").expect("Sentence detection regex pattern is valid and should compile"));
|
|
70
|
-
static PUNCTUATION_DETECT: Lazy<Regex> =
|
|
71
|
-
Lazy::new(|| Regex::new(r"[.!?]").expect("Punctuation detection regex pattern is valid and should compile"));
|
|
72
|
-
|
|
73
|
-
static WHITESPACE_NORMALIZE: Lazy<Regex> = Lazy::new(|| {
|
|
74
|
-
Regex::new(r"[ \t\f\v\r\xa0\u{2000}-\u{200b}\u{2028}\u{2029}\u{3000}]+")
|
|
75
|
-
.expect("Whitespace normalization regex pattern is valid and should compile")
|
|
76
|
-
});
|
|
77
|
-
static NEWLINE_NORMALIZE: Lazy<Regex> = Lazy::new(|| {
|
|
78
|
-
Regex::new(r"\n\s*\n\s*\n+").expect("Newline normalization regex pattern is valid and should compile")
|
|
79
|
-
});
|
|
80
|
-
static NEWLINE_CLEANUP: Lazy<Regex> =
|
|
81
|
-
Lazy::new(|| Regex::new(r"\n+").expect("Newline cleanup regex pattern is valid and should compile"));
|
|
82
|
-
|
|
83
|
-
#[inline]
|
|
84
|
-
fn sum_match_lengths(text: &str, pattern: &Regex) -> usize {
|
|
85
|
-
pattern.find_iter(text).map(|m| m.len()).sum()
|
|
86
|
-
}
|
|
87
|
-
|
|
88
|
-
fn chain_replacements<'a>(mut text: Cow<'a, str>, replacements: &[(&Regex, &str)]) -> Cow<'a, str> {
|
|
89
|
-
for (pattern, replacement) in replacements {
|
|
90
|
-
if pattern.is_match(&text) {
|
|
91
|
-
text = Cow::Owned(pattern.replace_all(&text, *replacement).into_owned());
|
|
92
|
-
}
|
|
93
|
-
}
|
|
94
|
-
text
|
|
95
|
-
}
|
|
96
|
-
|
|
97
|
-
#[inline]
|
|
98
|
-
fn replace_with_if_matches<'a, F>(text: &'a str, pattern: &Regex, replacer: F) -> Cow<'a, str>
|
|
99
|
-
where
|
|
100
|
-
F: FnMut(®ex::Captures) -> String,
|
|
101
|
-
{
|
|
102
|
-
if pattern.is_match(text) {
|
|
103
|
-
Cow::Owned(pattern.replace_all(text, replacer).into_owned())
|
|
104
|
-
} else {
|
|
105
|
-
Cow::Borrowed(text)
|
|
106
|
-
}
|
|
107
|
-
}
|
|
108
|
-
|
|
109
|
-
pub fn calculate_quality_score(text: &str, metadata: Option<&HashMap<String, String>>) -> f64 {
|
|
110
|
-
if text.is_empty() || text.trim().is_empty() {
|
|
111
|
-
return 0.0;
|
|
112
|
-
}
|
|
113
|
-
|
|
114
|
-
let total_chars = text.len() as f64;
|
|
115
|
-
|
|
116
|
-
if text.len() < MIN_TEXT_LENGTH {
|
|
117
|
-
return 0.1;
|
|
118
|
-
}
|
|
119
|
-
|
|
120
|
-
let mut score = 1.0;
|
|
121
|
-
|
|
122
|
-
if text.len() > LARGE_TEXT_LENGTH {
|
|
123
|
-
let ocr_penalty = calculate_ocr_penalty(text, total_chars);
|
|
124
|
-
let script_penalty = calculate_script_penalty(text, total_chars);
|
|
125
|
-
let nav_penalty = calculate_navigation_penalty(text, total_chars);
|
|
126
|
-
let structure_bonus = calculate_structure_bonus(text);
|
|
127
|
-
|
|
128
|
-
score -= ocr_penalty * OCR_PENALTY_WEIGHT;
|
|
129
|
-
score -= script_penalty * SCRIPT_PENALTY_WEIGHT;
|
|
130
|
-
score -= nav_penalty * NAV_PENALTY_WEIGHT;
|
|
131
|
-
score += structure_bonus * STRUCTURE_BONUS_WEIGHT;
|
|
132
|
-
} else {
|
|
133
|
-
score -= calculate_ocr_penalty(text, total_chars) * OCR_PENALTY_WEIGHT;
|
|
134
|
-
score += calculate_structure_bonus(text) * STRUCTURE_BONUS_WEIGHT;
|
|
135
|
-
}
|
|
136
|
-
|
|
137
|
-
if let Some(metadata) = metadata {
|
|
138
|
-
score += calculate_metadata_bonus(metadata) * METADATA_BONUS_WEIGHT;
|
|
139
|
-
}
|
|
140
|
-
|
|
141
|
-
score.clamp(0.0, 1.0)
|
|
142
|
-
}
|
|
143
|
-
|
|
144
|
-
#[inline]
|
|
145
|
-
fn calculate_ocr_penalty(text: &str, total_chars: f64) -> f64 {
|
|
146
|
-
if total_chars == 0.0 {
|
|
147
|
-
return 0.0;
|
|
148
|
-
}
|
|
149
|
-
|
|
150
|
-
if !text.contains(" ") && !text.contains("...") {
|
|
151
|
-
return 0.0;
|
|
152
|
-
}
|
|
153
|
-
|
|
154
|
-
let artifact_chars = sum_match_lengths(text, &SCATTERED_CHARS_PATTERN)
|
|
155
|
-
+ sum_match_lengths(text, &REPEATED_PUNCT_PATTERN)
|
|
156
|
-
+ count_non_table_dash_artifacts(text)
|
|
157
|
-
+ sum_match_lengths(text, &ISOLATED_PUNCT_PATTERN)
|
|
158
|
-
+ sum_match_lengths(text, &MALFORMED_WORDS_PATTERN)
|
|
159
|
-
+ sum_match_lengths(text, &EXCESSIVE_WHITESPACE_PATTERN);
|
|
160
|
-
|
|
161
|
-
(artifact_chars as f64 / total_chars).min(1.0)
|
|
162
|
-
}
|
|
163
|
-
|
|
164
|
-
#[inline]
|
|
165
|
-
fn count_non_table_dash_artifacts(text: &str) -> usize {
|
|
166
|
-
let mut artifact_count = 0;
|
|
167
|
-
|
|
168
|
-
for line in text.lines() {
|
|
169
|
-
let trimmed = line.trim();
|
|
170
|
-
let is_table_separator = trimmed.starts_with('|')
|
|
171
|
-
&& trimmed.ends_with('|')
|
|
172
|
-
&& trimmed
|
|
173
|
-
.chars()
|
|
174
|
-
.all(|c| c == '|' || c == '-' || c.is_whitespace() || c == ':');
|
|
175
|
-
|
|
176
|
-
if !is_table_separator {
|
|
177
|
-
for m in DASH_PATTERN.find_iter(line) {
|
|
178
|
-
artifact_count += m.len();
|
|
179
|
-
}
|
|
180
|
-
}
|
|
181
|
-
}
|
|
182
|
-
|
|
183
|
-
artifact_count
|
|
184
|
-
}
|
|
185
|
-
|
|
186
|
-
#[inline]
|
|
187
|
-
fn calculate_script_penalty(text: &str, total_chars: f64) -> f64 {
|
|
188
|
-
if total_chars == 0.0 {
|
|
189
|
-
return 0.0;
|
|
190
|
-
}
|
|
191
|
-
|
|
192
|
-
if !text.contains("function") && !text.contains("<script") && !text.contains("<style") {
|
|
193
|
-
return 0.0;
|
|
194
|
-
}
|
|
195
|
-
|
|
196
|
-
let script_chars = sum_match_lengths(text, &JS_FUNCTION_PATTERN)
|
|
197
|
-
+ sum_match_lengths(text, &CSS_RULES_PATTERN)
|
|
198
|
-
+ sum_match_lengths(text, &SCRIPT_TAG_PATTERN)
|
|
199
|
-
+ sum_match_lengths(text, &STYLE_TAG_PATTERN);
|
|
200
|
-
|
|
201
|
-
(script_chars as f64 / total_chars).min(1.0)
|
|
202
|
-
}
|
|
203
|
-
|
|
204
|
-
#[inline]
|
|
205
|
-
fn calculate_navigation_penalty(text: &str, total_chars: f64) -> f64 {
|
|
206
|
-
if total_chars == 0.0 {
|
|
207
|
-
return 0.0;
|
|
208
|
-
}
|
|
209
|
-
|
|
210
|
-
let nav_chars = sum_match_lengths(text, &NAV_WORDS_PATTERN)
|
|
211
|
-
+ sum_match_lengths(text, &BREADCRUMB_PATTERN)
|
|
212
|
-
+ sum_match_lengths(text, &PAGINATION_PATTERN);
|
|
213
|
-
|
|
214
|
-
(nav_chars as f64 / total_chars).min(1.0)
|
|
215
|
-
}
|
|
216
|
-
|
|
217
|
-
#[inline]
|
|
218
|
-
fn calculate_structure_bonus(text: &str) -> f64 {
|
|
219
|
-
if text.is_empty() {
|
|
220
|
-
return 0.0;
|
|
221
|
-
}
|
|
222
|
-
|
|
223
|
-
let sentence_count = SENTENCE_DETECT.find_iter(text).count() as f64;
|
|
224
|
-
let paragraph_count = text.matches("\n\n").count() as f64 + 1.0;
|
|
225
|
-
let words = text.split_whitespace().count() as f64;
|
|
226
|
-
|
|
227
|
-
if words == 0.0 {
|
|
228
|
-
return 0.0;
|
|
229
|
-
}
|
|
230
|
-
|
|
231
|
-
let avg_words_per_sentence = words / sentence_count.max(1.0);
|
|
232
|
-
let avg_words_per_paragraph = words / paragraph_count;
|
|
233
|
-
|
|
234
|
-
let mut structure_score: f64 = 0.0;
|
|
235
|
-
|
|
236
|
-
if (MIN_SENTENCE_WORDS..=MAX_SENTENCE_WORDS).contains(&avg_words_per_sentence) {
|
|
237
|
-
structure_score += 0.3;
|
|
238
|
-
}
|
|
239
|
-
|
|
240
|
-
if (MIN_PARAGRAPH_WORDS..=MAX_PARAGRAPH_WORDS).contains(&avg_words_per_paragraph) {
|
|
241
|
-
structure_score += 0.3;
|
|
242
|
-
}
|
|
243
|
-
|
|
244
|
-
if paragraph_count > 1.0 {
|
|
245
|
-
structure_score += 0.2;
|
|
246
|
-
}
|
|
247
|
-
|
|
248
|
-
if PUNCTUATION_DETECT.is_match(text) {
|
|
249
|
-
structure_score += 0.2;
|
|
250
|
-
}
|
|
251
|
-
|
|
252
|
-
structure_score.min(1.0)
|
|
253
|
-
}
|
|
254
|
-
|
|
255
|
-
#[inline]
|
|
256
|
-
fn calculate_metadata_bonus(metadata: &HashMap<String, String>) -> f64 {
|
|
257
|
-
const IMPORTANT_FIELDS: &[&str] = &["title", "author", "subject", "description", "keywords"];
|
|
258
|
-
|
|
259
|
-
let present_fields = IMPORTANT_FIELDS
|
|
260
|
-
.iter()
|
|
261
|
-
.filter(|&&field| metadata.contains_key(field))
|
|
262
|
-
.count();
|
|
263
|
-
|
|
264
|
-
present_fields as f64 / IMPORTANT_FIELDS.len() as f64
|
|
265
|
-
}
|
|
266
|
-
|
|
267
|
-
pub fn clean_extracted_text(text: &str) -> String {
|
|
268
|
-
if text.is_empty() {
|
|
269
|
-
return String::new();
|
|
270
|
-
}
|
|
271
|
-
|
|
272
|
-
let mut working_text = Cow::Borrowed(text);
|
|
273
|
-
|
|
274
|
-
working_text = clean_scripts(working_text);
|
|
275
|
-
|
|
276
|
-
working_text = clean_ocr_artifacts_cow(working_text);
|
|
277
|
-
|
|
278
|
-
working_text = clean_navigation_elements_cow(working_text);
|
|
279
|
-
|
|
280
|
-
working_text = normalize_whitespace_cow(working_text);
|
|
281
|
-
|
|
282
|
-
working_text.trim().to_string()
|
|
283
|
-
}
|
|
284
|
-
|
|
285
|
-
#[inline]
|
|
286
|
-
fn clean_scripts<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
287
|
-
let script_replacements = [
|
|
288
|
-
(&*SCRIPT_TAG_PATTERN, " "),
|
|
289
|
-
(&*STYLE_TAG_PATTERN, " "),
|
|
290
|
-
(&*JS_FUNCTION_PATTERN, " "),
|
|
291
|
-
(&*CSS_RULES_PATTERN, " "),
|
|
292
|
-
];
|
|
293
|
-
chain_replacements(text, &script_replacements)
|
|
294
|
-
}
|
|
295
|
-
|
|
296
|
-
#[inline]
|
|
297
|
-
fn normalize_whitespace_cow<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
298
|
-
if let Some(fast) = normalize_whitespace_ascii(text.as_ref()) {
|
|
299
|
-
return Cow::Owned(fast);
|
|
300
|
-
}
|
|
301
|
-
|
|
302
|
-
let mut result = text;
|
|
303
|
-
|
|
304
|
-
if WHITESPACE_NORMALIZE.is_match(&result) {
|
|
305
|
-
result = Cow::Owned(WHITESPACE_NORMALIZE.replace_all(&result, " ").into_owned());
|
|
306
|
-
}
|
|
307
|
-
|
|
308
|
-
if NEWLINE_NORMALIZE.is_match(&result) {
|
|
309
|
-
result = Cow::Owned(NEWLINE_NORMALIZE.replace_all(&result, "\n\n").into_owned());
|
|
310
|
-
}
|
|
311
|
-
|
|
312
|
-
result
|
|
313
|
-
}
|
|
314
|
-
|
|
315
|
-
#[inline]
|
|
316
|
-
fn clean_ocr_artifacts_cow<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
317
|
-
let result = if let Some(fixed) = collapse_scattered_ascii(&text) {
|
|
318
|
-
Cow::Owned(fixed)
|
|
319
|
-
} else if SCATTERED_CHARS_PATTERN.is_match(&text) {
|
|
320
|
-
Cow::Owned(
|
|
321
|
-
replace_with_if_matches(&text, &SCATTERED_CHARS_PATTERN, |caps: ®ex::Captures| {
|
|
322
|
-
caps[0].chars().filter(|c| !c.is_whitespace()).collect::<String>()
|
|
323
|
-
})
|
|
324
|
-
.into_owned(),
|
|
325
|
-
)
|
|
326
|
-
} else {
|
|
327
|
-
text
|
|
328
|
-
};
|
|
329
|
-
|
|
330
|
-
let result = clean_dashes_preserve_tables(result);
|
|
331
|
-
|
|
332
|
-
let ocr_replacements = [
|
|
333
|
-
(&*REPEATED_PUNCT_PATTERN, "..."),
|
|
334
|
-
(&*ISOLATED_PUNCT_PATTERN, " "),
|
|
335
|
-
(&*MALFORMED_WORDS_PATTERN, " "),
|
|
336
|
-
(&*EXCESSIVE_WHITESPACE_PATTERN, " "),
|
|
337
|
-
];
|
|
338
|
-
|
|
339
|
-
chain_replacements(result, &ocr_replacements)
|
|
340
|
-
}
|
|
341
|
-
|
|
342
|
-
#[inline]
|
|
343
|
-
fn clean_dashes_preserve_tables<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
344
|
-
if !DASH_PATTERN.is_match(&text) {
|
|
345
|
-
return text;
|
|
346
|
-
}
|
|
347
|
-
|
|
348
|
-
let mut result = String::with_capacity(text.len());
|
|
349
|
-
let lines: Vec<&str> = text.lines().collect();
|
|
350
|
-
|
|
351
|
-
for (i, line) in lines.iter().enumerate() {
|
|
352
|
-
if i > 0 {
|
|
353
|
-
result.push('\n');
|
|
354
|
-
}
|
|
355
|
-
|
|
356
|
-
let trimmed = line.trim();
|
|
357
|
-
let is_table_separator = trimmed.starts_with('|')
|
|
358
|
-
&& trimmed.ends_with('|')
|
|
359
|
-
&& trimmed
|
|
360
|
-
.chars()
|
|
361
|
-
.all(|c| c == '|' || c == '-' || c.is_whitespace() || c == ':');
|
|
362
|
-
|
|
363
|
-
if is_table_separator {
|
|
364
|
-
result.push_str(line);
|
|
365
|
-
} else {
|
|
366
|
-
let cleaned_line = DASH_PATTERN.replace_all(line, "...");
|
|
367
|
-
result.push_str(&cleaned_line);
|
|
368
|
-
}
|
|
369
|
-
}
|
|
370
|
-
|
|
371
|
-
Cow::Owned(result)
|
|
372
|
-
}
|
|
373
|
-
|
|
374
|
-
#[inline]
|
|
375
|
-
fn clean_navigation_elements_cow<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
376
|
-
let nav_replacements = [
|
|
377
|
-
(&*NAV_WORDS_PATTERN, " "),
|
|
378
|
-
(&*BREADCRUMB_PATTERN, " "),
|
|
379
|
-
(&*PAGINATION_PATTERN, " "),
|
|
380
|
-
];
|
|
381
|
-
|
|
382
|
-
chain_replacements(text, &nav_replacements)
|
|
383
|
-
}
|
|
384
|
-
|
|
385
|
-
pub fn normalize_spaces(text: &str) -> String {
|
|
386
|
-
if text.is_empty() || text.trim().is_empty() {
|
|
387
|
-
return String::new();
|
|
388
|
-
}
|
|
389
|
-
|
|
390
|
-
let mut result = String::with_capacity(text.len());
|
|
391
|
-
|
|
392
|
-
let mut first = true;
|
|
393
|
-
for paragraph in text.split("\n\n") {
|
|
394
|
-
let trimmed = paragraph.trim();
|
|
395
|
-
if trimmed.is_empty() {
|
|
396
|
-
continue;
|
|
397
|
-
}
|
|
398
|
-
|
|
399
|
-
if !first {
|
|
400
|
-
result.push_str("\n\n");
|
|
401
|
-
}
|
|
402
|
-
first = false;
|
|
403
|
-
|
|
404
|
-
let cleaned = WHITESPACE_NORMALIZE.replace_all(paragraph, " ");
|
|
405
|
-
let cleaned = NEWLINE_CLEANUP.replace_all(&cleaned, "\n");
|
|
406
|
-
|
|
407
|
-
let mut first_line = true;
|
|
408
|
-
for line in cleaned.split('\n') {
|
|
409
|
-
let line = line.trim();
|
|
410
|
-
if !line.is_empty() {
|
|
411
|
-
if !first_line {
|
|
412
|
-
result.push('\n');
|
|
413
|
-
}
|
|
414
|
-
result.push_str(line);
|
|
415
|
-
first_line = false;
|
|
416
|
-
}
|
|
417
|
-
}
|
|
418
|
-
}
|
|
419
|
-
|
|
420
|
-
result
|
|
421
|
-
}
|
|
422
|
-
|
|
423
|
-
#[cfg(test)]
|
|
424
|
-
mod tests {
|
|
425
|
-
use super::*;
|
|
426
|
-
|
|
427
|
-
#[test]
|
|
428
|
-
fn test_calculate_quality_score_empty_text() {
|
|
429
|
-
assert_eq!(calculate_quality_score("", None), 0.0);
|
|
430
|
-
assert_eq!(calculate_quality_score(" ", None), 0.0);
|
|
431
|
-
assert_eq!(calculate_quality_score("\n\n\n", None), 0.0);
|
|
432
|
-
}
|
|
433
|
-
|
|
434
|
-
#[test]
|
|
435
|
-
fn test_calculate_quality_score_short_text() {
|
|
436
|
-
let text = "Hello";
|
|
437
|
-
let score = calculate_quality_score(text, None);
|
|
438
|
-
assert_eq!(score, 0.1);
|
|
439
|
-
}
|
|
440
|
-
|
|
441
|
-
#[test]
|
|
442
|
-
fn test_calculate_quality_score_normal_text() {
|
|
443
|
-
let text =
|
|
444
|
-
"This is a normal sentence with proper punctuation. It has multiple sentences. And proper structure.";
|
|
445
|
-
let score = calculate_quality_score(text, None);
|
|
446
|
-
assert!(score > 0.5);
|
|
447
|
-
assert!(score <= 1.0);
|
|
448
|
-
}
|
|
449
|
-
|
|
450
|
-
#[test]
|
|
451
|
-
fn test_clean_extracted_text_empty() {
|
|
452
|
-
assert_eq!(clean_extracted_text(""), "");
|
|
453
|
-
assert_eq!(clean_extracted_text(" "), "");
|
|
454
|
-
}
|
|
455
|
-
|
|
456
|
-
#[test]
|
|
457
|
-
fn test_clean_extracted_text_removes_scripts() {
|
|
458
|
-
let text = "Before <script>alert('test');</script> After";
|
|
459
|
-
let cleaned = clean_extracted_text(text);
|
|
460
|
-
assert!(!cleaned.contains("<script"));
|
|
461
|
-
assert!(cleaned.contains("Before"));
|
|
462
|
-
assert!(cleaned.contains("After"));
|
|
463
|
-
}
|
|
464
|
-
|
|
465
|
-
#[test]
|
|
466
|
-
fn test_normalize_spaces_empty() {
|
|
467
|
-
assert_eq!(normalize_spaces(""), "");
|
|
468
|
-
assert_eq!(normalize_spaces(" "), "");
|
|
469
|
-
}
|
|
470
|
-
|
|
471
|
-
#[test]
|
|
472
|
-
fn test_normalize_spaces_single_paragraph() {
|
|
473
|
-
let text = "This is a test";
|
|
474
|
-
let normalized = normalize_spaces(text);
|
|
475
|
-
assert_eq!(normalized, "This is a test");
|
|
476
|
-
}
|
|
477
|
-
|
|
478
|
-
#[test]
|
|
479
|
-
fn test_calculate_quality_score_with_metadata() {
|
|
480
|
-
let text = "This is a normal text with proper structure.";
|
|
481
|
-
let mut metadata = HashMap::new();
|
|
482
|
-
metadata.insert("title".to_string(), "Test Title".to_string());
|
|
483
|
-
metadata.insert("author".to_string(), "Test Author".to_string());
|
|
484
|
-
|
|
485
|
-
let score = calculate_quality_score(text, Some(&metadata));
|
|
486
|
-
assert!(score > 0.0);
|
|
487
|
-
assert!(score <= 1.0);
|
|
488
|
-
}
|
|
489
|
-
|
|
490
|
-
#[test]
|
|
491
|
-
fn test_calculate_ocr_penalty_clean_text() {
|
|
492
|
-
let text = "This is clean text without artifacts";
|
|
493
|
-
let penalty = calculate_ocr_penalty(text, text.len() as f64);
|
|
494
|
-
assert_eq!(penalty, 0.0);
|
|
495
|
-
}
|
|
496
|
-
|
|
497
|
-
#[test]
|
|
498
|
-
fn test_calculate_ocr_penalty_with_artifacts() {
|
|
499
|
-
let text = "Text with excessive spaces and ....... dots";
|
|
500
|
-
let penalty = calculate_ocr_penalty(text, text.len() as f64);
|
|
501
|
-
assert!(penalty > 0.0);
|
|
502
|
-
assert!(penalty <= 1.0);
|
|
503
|
-
}
|
|
504
|
-
|
|
505
|
-
#[test]
|
|
506
|
-
fn test_calculate_script_penalty_clean_text() {
|
|
507
|
-
let text = "This is clean text without scripts";
|
|
508
|
-
let penalty = calculate_script_penalty(text, text.len() as f64);
|
|
509
|
-
assert_eq!(penalty, 0.0);
|
|
510
|
-
}
|
|
511
|
-
|
|
512
|
-
#[test]
|
|
513
|
-
fn test_calculate_script_penalty_with_js() {
|
|
514
|
-
let text = "function test() { return 42; }";
|
|
515
|
-
let penalty = calculate_script_penalty(text, text.len() as f64);
|
|
516
|
-
assert!(penalty > 0.0);
|
|
517
|
-
}
|
|
518
|
-
|
|
519
|
-
#[test]
|
|
520
|
-
fn test_calculate_navigation_penalty_clean_text() {
|
|
521
|
-
let text = "This is clean text without navigation";
|
|
522
|
-
let penalty = calculate_navigation_penalty(text, text.len() as f64);
|
|
523
|
-
assert_eq!(penalty, 0.0);
|
|
524
|
-
}
|
|
525
|
-
|
|
526
|
-
#[test]
|
|
527
|
-
fn test_calculate_navigation_penalty_with_nav() {
|
|
528
|
-
let text = "Skip to main content and Back to top links everywhere";
|
|
529
|
-
let penalty = calculate_navigation_penalty(text, text.len() as f64);
|
|
530
|
-
assert!(penalty > 0.0);
|
|
531
|
-
}
|
|
532
|
-
|
|
533
|
-
#[test]
|
|
534
|
-
fn test_calculate_structure_bonus_empty() {
|
|
535
|
-
assert_eq!(calculate_structure_bonus(""), 0.0);
|
|
536
|
-
}
|
|
537
|
-
|
|
538
|
-
#[test]
|
|
539
|
-
fn test_calculate_structure_bonus_well_structured() {
|
|
540
|
-
let text = "This is a sentence. This is another sentence.\n\nNew paragraph here. More content.";
|
|
541
|
-
let bonus = calculate_structure_bonus(text);
|
|
542
|
-
assert!(bonus > 0.0);
|
|
543
|
-
assert!(bonus <= 1.0);
|
|
544
|
-
}
|
|
545
|
-
|
|
546
|
-
#[test]
|
|
547
|
-
fn test_calculate_metadata_bonus_empty() {
|
|
548
|
-
let metadata = HashMap::new();
|
|
549
|
-
let bonus = calculate_metadata_bonus(&metadata);
|
|
550
|
-
assert_eq!(bonus, 0.0);
|
|
551
|
-
}
|
|
552
|
-
|
|
553
|
-
#[test]
|
|
554
|
-
fn test_calculate_metadata_bonus_full() {
|
|
555
|
-
let mut metadata = HashMap::new();
|
|
556
|
-
metadata.insert("title".to_string(), "Title".to_string());
|
|
557
|
-
metadata.insert("author".to_string(), "Author".to_string());
|
|
558
|
-
metadata.insert("subject".to_string(), "Subject".to_string());
|
|
559
|
-
metadata.insert("description".to_string(), "Description".to_string());
|
|
560
|
-
metadata.insert("keywords".to_string(), "Keywords".to_string());
|
|
561
|
-
|
|
562
|
-
let bonus = calculate_metadata_bonus(&metadata);
|
|
563
|
-
assert_eq!(bonus, 1.0);
|
|
564
|
-
}
|
|
565
|
-
|
|
566
|
-
#[test]
|
|
567
|
-
fn test_clean_extracted_text_removes_styles() {
|
|
568
|
-
let text = "Before <style>.class { color: red; }</style> After";
|
|
569
|
-
let cleaned = clean_extracted_text(text);
|
|
570
|
-
assert!(!cleaned.contains("<style"));
|
|
571
|
-
assert!(cleaned.contains("Before"));
|
|
572
|
-
assert!(cleaned.contains("After"));
|
|
573
|
-
}
|
|
574
|
-
|
|
575
|
-
#[test]
|
|
576
|
-
fn test_clean_extracted_text_ocr_artifacts() {
|
|
577
|
-
let text = "Text with excessive spaces";
|
|
578
|
-
let cleaned = clean_extracted_text(text);
|
|
579
|
-
assert!(!cleaned.contains(" "));
|
|
580
|
-
}
|
|
581
|
-
|
|
582
|
-
#[test]
|
|
583
|
-
fn test_clean_extracted_text_navigation() {
|
|
584
|
-
let text = "Content Skip to main content more content";
|
|
585
|
-
let cleaned = clean_extracted_text(text);
|
|
586
|
-
assert!(cleaned.contains("Content"));
|
|
587
|
-
assert!(cleaned.contains("more content"));
|
|
588
|
-
}
|
|
589
|
-
|
|
590
|
-
#[test]
|
|
591
|
-
fn test_normalize_spaces_multiple_paragraphs() {
|
|
592
|
-
let text = "First paragraph.\n\nSecond paragraph.";
|
|
593
|
-
let normalized = normalize_spaces(text);
|
|
594
|
-
assert!(normalized.contains("\n\n"));
|
|
595
|
-
}
|
|
596
|
-
|
|
597
|
-
#[test]
|
|
598
|
-
fn test_normalize_spaces_preserves_paragraphs() {
|
|
599
|
-
let text = "Para 1\n\n\n\nPara 2";
|
|
600
|
-
let normalized = normalize_spaces(text);
|
|
601
|
-
assert_eq!(normalized, "Para 1\n\nPara 2");
|
|
602
|
-
}
|
|
603
|
-
|
|
604
|
-
#[test]
|
|
605
|
-
fn test_count_non_table_dash_artifacts() {
|
|
606
|
-
let text = "Some text --- with dashes";
|
|
607
|
-
let count = count_non_table_dash_artifacts(text);
|
|
608
|
-
assert!(count > 0);
|
|
609
|
-
}
|
|
610
|
-
|
|
611
|
-
#[test]
|
|
612
|
-
fn test_count_non_table_dash_artifacts_preserves_tables() {
|
|
613
|
-
let text = "| Header |\n|--------|\n| Data |";
|
|
614
|
-
let count = count_non_table_dash_artifacts(text);
|
|
615
|
-
assert_eq!(count, 0);
|
|
616
|
-
}
|
|
617
|
-
|
|
618
|
-
#[test]
|
|
619
|
-
fn test_clean_dashes_preserve_tables_simple() {
|
|
620
|
-
let text = Cow::Borrowed("| Col1 |\n|------|\n| Data |");
|
|
621
|
-
let result = clean_dashes_preserve_tables(text);
|
|
622
|
-
assert!(result.contains("|------"));
|
|
623
|
-
}
|
|
624
|
-
|
|
625
|
-
#[test]
|
|
626
|
-
fn test_clean_dashes_preserve_tables_replaces_non_table() {
|
|
627
|
-
let text = Cow::Borrowed("Text with --- dashes");
|
|
628
|
-
let result = clean_dashes_preserve_tables(text);
|
|
629
|
-
assert!(result.contains("..."));
|
|
630
|
-
assert!(!result.contains("---"));
|
|
631
|
-
}
|
|
632
|
-
|
|
633
|
-
#[test]
|
|
634
|
-
fn test_sum_match_lengths() {
|
|
635
|
-
let text = "test ... test ... test";
|
|
636
|
-
let count = sum_match_lengths(text, &REPEATED_PUNCT_PATTERN);
|
|
637
|
-
assert!(count > 0);
|
|
638
|
-
}
|
|
639
|
-
|
|
640
|
-
#[test]
|
|
641
|
-
fn test_quality_score_large_text_with_ocr_issues() {
|
|
642
|
-
let text = "a".repeat(2000) + " " + &"b".repeat(2000);
|
|
643
|
-
let score = calculate_quality_score(&text, None);
|
|
644
|
-
assert!(score >= 0.0);
|
|
645
|
-
assert!(score <= 1.0);
|
|
646
|
-
}
|
|
647
|
-
|
|
648
|
-
#[test]
|
|
649
|
-
fn test_quality_score_clamped_to_range() {
|
|
650
|
-
let perfect_text = "This is perfect text. ".repeat(100);
|
|
651
|
-
let score = calculate_quality_score(&perfect_text, None);
|
|
652
|
-
assert!(score >= 0.0);
|
|
653
|
-
assert!(score <= 1.0);
|
|
654
|
-
}
|
|
655
|
-
|
|
656
|
-
#[test]
|
|
657
|
-
fn test_clean_extracted_text_scattered_chars() {
|
|
658
|
-
let text = "a b c scattered";
|
|
659
|
-
let cleaned = clean_extracted_text(text);
|
|
660
|
-
assert!(!cleaned.is_empty());
|
|
661
|
-
}
|
|
662
|
-
|
|
663
|
-
#[test]
|
|
664
|
-
fn test_normalize_whitespace_cow_no_changes() {
|
|
665
|
-
let text = Cow::Borrowed("normaltext");
|
|
666
|
-
let result = normalize_whitespace_cow(text);
|
|
667
|
-
assert_eq!(result.as_ref(), "normaltext");
|
|
668
|
-
}
|
|
669
|
-
|
|
670
|
-
#[test]
|
|
671
|
-
fn test_normalize_whitespace_cow_with_changes() {
|
|
672
|
-
let text = Cow::Borrowed("text with spaces");
|
|
673
|
-
let result = normalize_whitespace_cow(text);
|
|
674
|
-
assert!(matches!(result, Cow::Owned(_)));
|
|
675
|
-
}
|
|
676
|
-
|
|
677
|
-
#[test]
|
|
678
|
-
fn test_clean_scripts_no_scripts() {
|
|
679
|
-
let text = Cow::Borrowed("clean text");
|
|
680
|
-
let result = clean_scripts(text);
|
|
681
|
-
assert!(matches!(result, Cow::Borrowed(_)));
|
|
682
|
-
}
|
|
683
|
-
|
|
684
|
-
#[test]
|
|
685
|
-
fn test_clean_scripts_with_script_tag() {
|
|
686
|
-
let text = Cow::Borrowed("<script>code</script>");
|
|
687
|
-
let result = clean_scripts(text);
|
|
688
|
-
assert!(!result.contains("<script"));
|
|
689
|
-
}
|
|
690
|
-
|
|
691
|
-
#[test]
|
|
692
|
-
fn test_quality_constants() {
|
|
693
|
-
assert_eq!(MIN_TEXT_LENGTH, 10);
|
|
694
|
-
assert_eq!(LARGE_TEXT_LENGTH, 1000);
|
|
695
|
-
assert_eq!(OCR_PENALTY_WEIGHT, 0.3);
|
|
696
|
-
}
|
|
697
|
-
}
|
|
1
|
+
use once_cell::sync::Lazy;
|
|
2
|
+
use regex::Regex;
|
|
3
|
+
use std::borrow::Cow;
|
|
4
|
+
use std::collections::HashMap;
|
|
5
|
+
|
|
6
|
+
use crate::utils::quality::{collapse_scattered_ascii, normalize_whitespace_ascii};
|
|
7
|
+
|
|
8
|
+
// ============================================================================
|
|
9
|
+
// ============================================================================
|
|
10
|
+
|
|
11
|
+
const OCR_PENALTY_WEIGHT: f64 = 0.3;
|
|
12
|
+
const SCRIPT_PENALTY_WEIGHT: f64 = 0.2;
|
|
13
|
+
const NAV_PENALTY_WEIGHT: f64 = 0.1;
|
|
14
|
+
const STRUCTURE_BONUS_WEIGHT: f64 = 0.2;
|
|
15
|
+
const METADATA_BONUS_WEIGHT: f64 = 0.1;
|
|
16
|
+
|
|
17
|
+
const MIN_TEXT_LENGTH: usize = 10;
|
|
18
|
+
const LARGE_TEXT_LENGTH: usize = 1000;
|
|
19
|
+
const MIN_SENTENCE_WORDS: f64 = 10.0;
|
|
20
|
+
const MAX_SENTENCE_WORDS: f64 = 30.0;
|
|
21
|
+
const MIN_PARAGRAPH_WORDS: f64 = 50.0;
|
|
22
|
+
const MAX_PARAGRAPH_WORDS: f64 = 300.0;
|
|
23
|
+
|
|
24
|
+
static SCATTERED_CHARS_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
25
|
+
Regex::new(r"\b[a-zA-Z]\s{2,}[a-zA-Z]\s{2,}[a-zA-Z]\b")
|
|
26
|
+
.expect("Scattered chars regex pattern is valid and should compile")
|
|
27
|
+
});
|
|
28
|
+
static REPEATED_PUNCT_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
29
|
+
Regex::new(r"[.]{3,}|[_]{3,}").expect("Repeated punctuation regex pattern is valid and should compile")
|
|
30
|
+
});
|
|
31
|
+
static DASH_PATTERN: Lazy<Regex> =
|
|
32
|
+
Lazy::new(|| Regex::new(r"[-]{3,}").expect("Dash pattern regex is valid and should compile"));
|
|
33
|
+
static ISOLATED_PUNCT_PATTERN: Lazy<Regex> =
|
|
34
|
+
Lazy::new(|| Regex::new(r"\s[.,;:!?]\s").expect("Isolated punctuation regex pattern is valid and should compile"));
|
|
35
|
+
static MALFORMED_WORDS_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
36
|
+
Regex::new(r"\b[a-zA-Z]+[0-9]+[a-zA-Z]+[a-zA-Z0-9]*\b")
|
|
37
|
+
.expect("Malformed words regex pattern is valid and should compile")
|
|
38
|
+
});
|
|
39
|
+
static EXCESSIVE_WHITESPACE_PATTERN: Lazy<Regex> =
|
|
40
|
+
Lazy::new(|| Regex::new(r"\s{3,}").expect("Excessive whitespace regex pattern is valid and should compile"));
|
|
41
|
+
|
|
42
|
+
static JS_FUNCTION_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
43
|
+
Regex::new(r"(?i)function\s+\w+\s*\([^)]*\)\s*\{[^}]*\}")
|
|
44
|
+
.expect("JavaScript function regex pattern is valid and should compile")
|
|
45
|
+
});
|
|
46
|
+
static CSS_RULES_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
47
|
+
Regex::new(r"(?i)\.[a-zA-Z][\w-]*\s*\{[^}]*\}").expect("CSS rules regex pattern is valid and should compile")
|
|
48
|
+
});
|
|
49
|
+
static SCRIPT_TAG_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
50
|
+
Regex::new(r"(?is)<script[^>]*>.*?</script>").expect("Script tag regex pattern is valid and should compile")
|
|
51
|
+
});
|
|
52
|
+
static STYLE_TAG_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
53
|
+
Regex::new(r"(?is)<style[^>]*>.*?</style>").expect("Style tag regex pattern is valid and should compile")
|
|
54
|
+
});
|
|
55
|
+
|
|
56
|
+
static NAV_WORDS_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
57
|
+
Regex::new(r"(?i)\b(?:Skip to main content|Back to top|Main navigation|Site navigation)\b")
|
|
58
|
+
.expect("Navigation words regex pattern is valid and should compile")
|
|
59
|
+
});
|
|
60
|
+
static BREADCRUMB_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
61
|
+
Regex::new(r"(?:Home\s*[>»]\s*|[>»]\s*){2,}").expect("Breadcrumb regex pattern is valid and should compile")
|
|
62
|
+
});
|
|
63
|
+
static PAGINATION_PATTERN: Lazy<Regex> = Lazy::new(|| {
|
|
64
|
+
Regex::new(r"(?i)\b(?:Page \d+ of \d+|First page|Last page|Previous page|Next page|^\d+ of \d+$)\b")
|
|
65
|
+
.expect("Pagination regex pattern is valid and should compile")
|
|
66
|
+
});
|
|
67
|
+
|
|
68
|
+
static SENTENCE_DETECT: Lazy<Regex> =
|
|
69
|
+
Lazy::new(|| Regex::new(r"[.!?]\s+[A-Z]").expect("Sentence detection regex pattern is valid and should compile"));
|
|
70
|
+
static PUNCTUATION_DETECT: Lazy<Regex> =
|
|
71
|
+
Lazy::new(|| Regex::new(r"[.!?]").expect("Punctuation detection regex pattern is valid and should compile"));
|
|
72
|
+
|
|
73
|
+
static WHITESPACE_NORMALIZE: Lazy<Regex> = Lazy::new(|| {
|
|
74
|
+
Regex::new(r"[ \t\f\v\r\xa0\u{2000}-\u{200b}\u{2028}\u{2029}\u{3000}]+")
|
|
75
|
+
.expect("Whitespace normalization regex pattern is valid and should compile")
|
|
76
|
+
});
|
|
77
|
+
static NEWLINE_NORMALIZE: Lazy<Regex> = Lazy::new(|| {
|
|
78
|
+
Regex::new(r"\n\s*\n\s*\n+").expect("Newline normalization regex pattern is valid and should compile")
|
|
79
|
+
});
|
|
80
|
+
static NEWLINE_CLEANUP: Lazy<Regex> =
|
|
81
|
+
Lazy::new(|| Regex::new(r"\n+").expect("Newline cleanup regex pattern is valid and should compile"));
|
|
82
|
+
|
|
83
|
+
#[inline]
|
|
84
|
+
fn sum_match_lengths(text: &str, pattern: &Regex) -> usize {
|
|
85
|
+
pattern.find_iter(text).map(|m| m.len()).sum()
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
fn chain_replacements<'a>(mut text: Cow<'a, str>, replacements: &[(&Regex, &str)]) -> Cow<'a, str> {
|
|
89
|
+
for (pattern, replacement) in replacements {
|
|
90
|
+
if pattern.is_match(&text) {
|
|
91
|
+
text = Cow::Owned(pattern.replace_all(&text, *replacement).into_owned());
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
text
|
|
95
|
+
}
|
|
96
|
+
|
|
97
|
+
#[inline]
|
|
98
|
+
fn replace_with_if_matches<'a, F>(text: &'a str, pattern: &Regex, replacer: F) -> Cow<'a, str>
|
|
99
|
+
where
|
|
100
|
+
F: FnMut(®ex::Captures) -> String,
|
|
101
|
+
{
|
|
102
|
+
if pattern.is_match(text) {
|
|
103
|
+
Cow::Owned(pattern.replace_all(text, replacer).into_owned())
|
|
104
|
+
} else {
|
|
105
|
+
Cow::Borrowed(text)
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
|
|
109
|
+
pub fn calculate_quality_score(text: &str, metadata: Option<&HashMap<String, String>>) -> f64 {
|
|
110
|
+
if text.is_empty() || text.trim().is_empty() {
|
|
111
|
+
return 0.0;
|
|
112
|
+
}
|
|
113
|
+
|
|
114
|
+
let total_chars = text.len() as f64;
|
|
115
|
+
|
|
116
|
+
if text.len() < MIN_TEXT_LENGTH {
|
|
117
|
+
return 0.1;
|
|
118
|
+
}
|
|
119
|
+
|
|
120
|
+
let mut score = 1.0;
|
|
121
|
+
|
|
122
|
+
if text.len() > LARGE_TEXT_LENGTH {
|
|
123
|
+
let ocr_penalty = calculate_ocr_penalty(text, total_chars);
|
|
124
|
+
let script_penalty = calculate_script_penalty(text, total_chars);
|
|
125
|
+
let nav_penalty = calculate_navigation_penalty(text, total_chars);
|
|
126
|
+
let structure_bonus = calculate_structure_bonus(text);
|
|
127
|
+
|
|
128
|
+
score -= ocr_penalty * OCR_PENALTY_WEIGHT;
|
|
129
|
+
score -= script_penalty * SCRIPT_PENALTY_WEIGHT;
|
|
130
|
+
score -= nav_penalty * NAV_PENALTY_WEIGHT;
|
|
131
|
+
score += structure_bonus * STRUCTURE_BONUS_WEIGHT;
|
|
132
|
+
} else {
|
|
133
|
+
score -= calculate_ocr_penalty(text, total_chars) * OCR_PENALTY_WEIGHT;
|
|
134
|
+
score += calculate_structure_bonus(text) * STRUCTURE_BONUS_WEIGHT;
|
|
135
|
+
}
|
|
136
|
+
|
|
137
|
+
if let Some(metadata) = metadata {
|
|
138
|
+
score += calculate_metadata_bonus(metadata) * METADATA_BONUS_WEIGHT;
|
|
139
|
+
}
|
|
140
|
+
|
|
141
|
+
score.clamp(0.0, 1.0)
|
|
142
|
+
}
|
|
143
|
+
|
|
144
|
+
#[inline]
|
|
145
|
+
fn calculate_ocr_penalty(text: &str, total_chars: f64) -> f64 {
|
|
146
|
+
if total_chars == 0.0 {
|
|
147
|
+
return 0.0;
|
|
148
|
+
}
|
|
149
|
+
|
|
150
|
+
if !text.contains(" ") && !text.contains("...") {
|
|
151
|
+
return 0.0;
|
|
152
|
+
}
|
|
153
|
+
|
|
154
|
+
let artifact_chars = sum_match_lengths(text, &SCATTERED_CHARS_PATTERN)
|
|
155
|
+
+ sum_match_lengths(text, &REPEATED_PUNCT_PATTERN)
|
|
156
|
+
+ count_non_table_dash_artifacts(text)
|
|
157
|
+
+ sum_match_lengths(text, &ISOLATED_PUNCT_PATTERN)
|
|
158
|
+
+ sum_match_lengths(text, &MALFORMED_WORDS_PATTERN)
|
|
159
|
+
+ sum_match_lengths(text, &EXCESSIVE_WHITESPACE_PATTERN);
|
|
160
|
+
|
|
161
|
+
(artifact_chars as f64 / total_chars).min(1.0)
|
|
162
|
+
}
|
|
163
|
+
|
|
164
|
+
#[inline]
|
|
165
|
+
fn count_non_table_dash_artifacts(text: &str) -> usize {
|
|
166
|
+
let mut artifact_count = 0;
|
|
167
|
+
|
|
168
|
+
for line in text.lines() {
|
|
169
|
+
let trimmed = line.trim();
|
|
170
|
+
let is_table_separator = trimmed.starts_with('|')
|
|
171
|
+
&& trimmed.ends_with('|')
|
|
172
|
+
&& trimmed
|
|
173
|
+
.chars()
|
|
174
|
+
.all(|c| c == '|' || c == '-' || c.is_whitespace() || c == ':');
|
|
175
|
+
|
|
176
|
+
if !is_table_separator {
|
|
177
|
+
for m in DASH_PATTERN.find_iter(line) {
|
|
178
|
+
artifact_count += m.len();
|
|
179
|
+
}
|
|
180
|
+
}
|
|
181
|
+
}
|
|
182
|
+
|
|
183
|
+
artifact_count
|
|
184
|
+
}
|
|
185
|
+
|
|
186
|
+
#[inline]
|
|
187
|
+
fn calculate_script_penalty(text: &str, total_chars: f64) -> f64 {
|
|
188
|
+
if total_chars == 0.0 {
|
|
189
|
+
return 0.0;
|
|
190
|
+
}
|
|
191
|
+
|
|
192
|
+
if !text.contains("function") && !text.contains("<script") && !text.contains("<style") {
|
|
193
|
+
return 0.0;
|
|
194
|
+
}
|
|
195
|
+
|
|
196
|
+
let script_chars = sum_match_lengths(text, &JS_FUNCTION_PATTERN)
|
|
197
|
+
+ sum_match_lengths(text, &CSS_RULES_PATTERN)
|
|
198
|
+
+ sum_match_lengths(text, &SCRIPT_TAG_PATTERN)
|
|
199
|
+
+ sum_match_lengths(text, &STYLE_TAG_PATTERN);
|
|
200
|
+
|
|
201
|
+
(script_chars as f64 / total_chars).min(1.0)
|
|
202
|
+
}
|
|
203
|
+
|
|
204
|
+
#[inline]
|
|
205
|
+
fn calculate_navigation_penalty(text: &str, total_chars: f64) -> f64 {
|
|
206
|
+
if total_chars == 0.0 {
|
|
207
|
+
return 0.0;
|
|
208
|
+
}
|
|
209
|
+
|
|
210
|
+
let nav_chars = sum_match_lengths(text, &NAV_WORDS_PATTERN)
|
|
211
|
+
+ sum_match_lengths(text, &BREADCRUMB_PATTERN)
|
|
212
|
+
+ sum_match_lengths(text, &PAGINATION_PATTERN);
|
|
213
|
+
|
|
214
|
+
(nav_chars as f64 / total_chars).min(1.0)
|
|
215
|
+
}
|
|
216
|
+
|
|
217
|
+
#[inline]
|
|
218
|
+
fn calculate_structure_bonus(text: &str) -> f64 {
|
|
219
|
+
if text.is_empty() {
|
|
220
|
+
return 0.0;
|
|
221
|
+
}
|
|
222
|
+
|
|
223
|
+
let sentence_count = SENTENCE_DETECT.find_iter(text).count() as f64;
|
|
224
|
+
let paragraph_count = text.matches("\n\n").count() as f64 + 1.0;
|
|
225
|
+
let words = text.split_whitespace().count() as f64;
|
|
226
|
+
|
|
227
|
+
if words == 0.0 {
|
|
228
|
+
return 0.0;
|
|
229
|
+
}
|
|
230
|
+
|
|
231
|
+
let avg_words_per_sentence = words / sentence_count.max(1.0);
|
|
232
|
+
let avg_words_per_paragraph = words / paragraph_count;
|
|
233
|
+
|
|
234
|
+
let mut structure_score: f64 = 0.0;
|
|
235
|
+
|
|
236
|
+
if (MIN_SENTENCE_WORDS..=MAX_SENTENCE_WORDS).contains(&avg_words_per_sentence) {
|
|
237
|
+
structure_score += 0.3;
|
|
238
|
+
}
|
|
239
|
+
|
|
240
|
+
if (MIN_PARAGRAPH_WORDS..=MAX_PARAGRAPH_WORDS).contains(&avg_words_per_paragraph) {
|
|
241
|
+
structure_score += 0.3;
|
|
242
|
+
}
|
|
243
|
+
|
|
244
|
+
if paragraph_count > 1.0 {
|
|
245
|
+
structure_score += 0.2;
|
|
246
|
+
}
|
|
247
|
+
|
|
248
|
+
if PUNCTUATION_DETECT.is_match(text) {
|
|
249
|
+
structure_score += 0.2;
|
|
250
|
+
}
|
|
251
|
+
|
|
252
|
+
structure_score.min(1.0)
|
|
253
|
+
}
|
|
254
|
+
|
|
255
|
+
#[inline]
|
|
256
|
+
fn calculate_metadata_bonus(metadata: &HashMap<String, String>) -> f64 {
|
|
257
|
+
const IMPORTANT_FIELDS: &[&str] = &["title", "author", "subject", "description", "keywords"];
|
|
258
|
+
|
|
259
|
+
let present_fields = IMPORTANT_FIELDS
|
|
260
|
+
.iter()
|
|
261
|
+
.filter(|&&field| metadata.contains_key(field))
|
|
262
|
+
.count();
|
|
263
|
+
|
|
264
|
+
present_fields as f64 / IMPORTANT_FIELDS.len() as f64
|
|
265
|
+
}
|
|
266
|
+
|
|
267
|
+
pub fn clean_extracted_text(text: &str) -> String {
|
|
268
|
+
if text.is_empty() {
|
|
269
|
+
return String::new();
|
|
270
|
+
}
|
|
271
|
+
|
|
272
|
+
let mut working_text = Cow::Borrowed(text);
|
|
273
|
+
|
|
274
|
+
working_text = clean_scripts(working_text);
|
|
275
|
+
|
|
276
|
+
working_text = clean_ocr_artifacts_cow(working_text);
|
|
277
|
+
|
|
278
|
+
working_text = clean_navigation_elements_cow(working_text);
|
|
279
|
+
|
|
280
|
+
working_text = normalize_whitespace_cow(working_text);
|
|
281
|
+
|
|
282
|
+
working_text.trim().to_string()
|
|
283
|
+
}
|
|
284
|
+
|
|
285
|
+
#[inline]
|
|
286
|
+
fn clean_scripts<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
287
|
+
let script_replacements = [
|
|
288
|
+
(&*SCRIPT_TAG_PATTERN, " "),
|
|
289
|
+
(&*STYLE_TAG_PATTERN, " "),
|
|
290
|
+
(&*JS_FUNCTION_PATTERN, " "),
|
|
291
|
+
(&*CSS_RULES_PATTERN, " "),
|
|
292
|
+
];
|
|
293
|
+
chain_replacements(text, &script_replacements)
|
|
294
|
+
}
|
|
295
|
+
|
|
296
|
+
#[inline]
|
|
297
|
+
fn normalize_whitespace_cow<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
298
|
+
if let Some(fast) = normalize_whitespace_ascii(text.as_ref()) {
|
|
299
|
+
return Cow::Owned(fast);
|
|
300
|
+
}
|
|
301
|
+
|
|
302
|
+
let mut result = text;
|
|
303
|
+
|
|
304
|
+
if WHITESPACE_NORMALIZE.is_match(&result) {
|
|
305
|
+
result = Cow::Owned(WHITESPACE_NORMALIZE.replace_all(&result, " ").into_owned());
|
|
306
|
+
}
|
|
307
|
+
|
|
308
|
+
if NEWLINE_NORMALIZE.is_match(&result) {
|
|
309
|
+
result = Cow::Owned(NEWLINE_NORMALIZE.replace_all(&result, "\n\n").into_owned());
|
|
310
|
+
}
|
|
311
|
+
|
|
312
|
+
result
|
|
313
|
+
}
|
|
314
|
+
|
|
315
|
+
#[inline]
|
|
316
|
+
fn clean_ocr_artifacts_cow<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
317
|
+
let result = if let Some(fixed) = collapse_scattered_ascii(&text) {
|
|
318
|
+
Cow::Owned(fixed)
|
|
319
|
+
} else if SCATTERED_CHARS_PATTERN.is_match(&text) {
|
|
320
|
+
Cow::Owned(
|
|
321
|
+
replace_with_if_matches(&text, &SCATTERED_CHARS_PATTERN, |caps: ®ex::Captures| {
|
|
322
|
+
caps[0].chars().filter(|c| !c.is_whitespace()).collect::<String>()
|
|
323
|
+
})
|
|
324
|
+
.into_owned(),
|
|
325
|
+
)
|
|
326
|
+
} else {
|
|
327
|
+
text
|
|
328
|
+
};
|
|
329
|
+
|
|
330
|
+
let result = clean_dashes_preserve_tables(result);
|
|
331
|
+
|
|
332
|
+
let ocr_replacements = [
|
|
333
|
+
(&*REPEATED_PUNCT_PATTERN, "..."),
|
|
334
|
+
(&*ISOLATED_PUNCT_PATTERN, " "),
|
|
335
|
+
(&*MALFORMED_WORDS_PATTERN, " "),
|
|
336
|
+
(&*EXCESSIVE_WHITESPACE_PATTERN, " "),
|
|
337
|
+
];
|
|
338
|
+
|
|
339
|
+
chain_replacements(result, &ocr_replacements)
|
|
340
|
+
}
|
|
341
|
+
|
|
342
|
+
#[inline]
|
|
343
|
+
fn clean_dashes_preserve_tables<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
344
|
+
if !DASH_PATTERN.is_match(&text) {
|
|
345
|
+
return text;
|
|
346
|
+
}
|
|
347
|
+
|
|
348
|
+
let mut result = String::with_capacity(text.len());
|
|
349
|
+
let lines: Vec<&str> = text.lines().collect();
|
|
350
|
+
|
|
351
|
+
for (i, line) in lines.iter().enumerate() {
|
|
352
|
+
if i > 0 {
|
|
353
|
+
result.push('\n');
|
|
354
|
+
}
|
|
355
|
+
|
|
356
|
+
let trimmed = line.trim();
|
|
357
|
+
let is_table_separator = trimmed.starts_with('|')
|
|
358
|
+
&& trimmed.ends_with('|')
|
|
359
|
+
&& trimmed
|
|
360
|
+
.chars()
|
|
361
|
+
.all(|c| c == '|' || c == '-' || c.is_whitespace() || c == ':');
|
|
362
|
+
|
|
363
|
+
if is_table_separator {
|
|
364
|
+
result.push_str(line);
|
|
365
|
+
} else {
|
|
366
|
+
let cleaned_line = DASH_PATTERN.replace_all(line, "...");
|
|
367
|
+
result.push_str(&cleaned_line);
|
|
368
|
+
}
|
|
369
|
+
}
|
|
370
|
+
|
|
371
|
+
Cow::Owned(result)
|
|
372
|
+
}
|
|
373
|
+
|
|
374
|
+
#[inline]
|
|
375
|
+
fn clean_navigation_elements_cow<'a>(text: Cow<'a, str>) -> Cow<'a, str> {
|
|
376
|
+
let nav_replacements = [
|
|
377
|
+
(&*NAV_WORDS_PATTERN, " "),
|
|
378
|
+
(&*BREADCRUMB_PATTERN, " "),
|
|
379
|
+
(&*PAGINATION_PATTERN, " "),
|
|
380
|
+
];
|
|
381
|
+
|
|
382
|
+
chain_replacements(text, &nav_replacements)
|
|
383
|
+
}
|
|
384
|
+
|
|
385
|
+
pub fn normalize_spaces(text: &str) -> String {
|
|
386
|
+
if text.is_empty() || text.trim().is_empty() {
|
|
387
|
+
return String::new();
|
|
388
|
+
}
|
|
389
|
+
|
|
390
|
+
let mut result = String::with_capacity(text.len());
|
|
391
|
+
|
|
392
|
+
let mut first = true;
|
|
393
|
+
for paragraph in text.split("\n\n") {
|
|
394
|
+
let trimmed = paragraph.trim();
|
|
395
|
+
if trimmed.is_empty() {
|
|
396
|
+
continue;
|
|
397
|
+
}
|
|
398
|
+
|
|
399
|
+
if !first {
|
|
400
|
+
result.push_str("\n\n");
|
|
401
|
+
}
|
|
402
|
+
first = false;
|
|
403
|
+
|
|
404
|
+
let cleaned = WHITESPACE_NORMALIZE.replace_all(paragraph, " ");
|
|
405
|
+
let cleaned = NEWLINE_CLEANUP.replace_all(&cleaned, "\n");
|
|
406
|
+
|
|
407
|
+
let mut first_line = true;
|
|
408
|
+
for line in cleaned.split('\n') {
|
|
409
|
+
let line = line.trim();
|
|
410
|
+
if !line.is_empty() {
|
|
411
|
+
if !first_line {
|
|
412
|
+
result.push('\n');
|
|
413
|
+
}
|
|
414
|
+
result.push_str(line);
|
|
415
|
+
first_line = false;
|
|
416
|
+
}
|
|
417
|
+
}
|
|
418
|
+
}
|
|
419
|
+
|
|
420
|
+
result
|
|
421
|
+
}
|
|
422
|
+
|
|
423
|
+
#[cfg(test)]
|
|
424
|
+
mod tests {
|
|
425
|
+
use super::*;
|
|
426
|
+
|
|
427
|
+
#[test]
|
|
428
|
+
fn test_calculate_quality_score_empty_text() {
|
|
429
|
+
assert_eq!(calculate_quality_score("", None), 0.0);
|
|
430
|
+
assert_eq!(calculate_quality_score(" ", None), 0.0);
|
|
431
|
+
assert_eq!(calculate_quality_score("\n\n\n", None), 0.0);
|
|
432
|
+
}
|
|
433
|
+
|
|
434
|
+
#[test]
|
|
435
|
+
fn test_calculate_quality_score_short_text() {
|
|
436
|
+
let text = "Hello";
|
|
437
|
+
let score = calculate_quality_score(text, None);
|
|
438
|
+
assert_eq!(score, 0.1);
|
|
439
|
+
}
|
|
440
|
+
|
|
441
|
+
#[test]
|
|
442
|
+
fn test_calculate_quality_score_normal_text() {
|
|
443
|
+
let text =
|
|
444
|
+
"This is a normal sentence with proper punctuation. It has multiple sentences. And proper structure.";
|
|
445
|
+
let score = calculate_quality_score(text, None);
|
|
446
|
+
assert!(score > 0.5);
|
|
447
|
+
assert!(score <= 1.0);
|
|
448
|
+
}
|
|
449
|
+
|
|
450
|
+
#[test]
|
|
451
|
+
fn test_clean_extracted_text_empty() {
|
|
452
|
+
assert_eq!(clean_extracted_text(""), "");
|
|
453
|
+
assert_eq!(clean_extracted_text(" "), "");
|
|
454
|
+
}
|
|
455
|
+
|
|
456
|
+
#[test]
|
|
457
|
+
fn test_clean_extracted_text_removes_scripts() {
|
|
458
|
+
let text = "Before <script>alert('test');</script> After";
|
|
459
|
+
let cleaned = clean_extracted_text(text);
|
|
460
|
+
assert!(!cleaned.contains("<script"));
|
|
461
|
+
assert!(cleaned.contains("Before"));
|
|
462
|
+
assert!(cleaned.contains("After"));
|
|
463
|
+
}
|
|
464
|
+
|
|
465
|
+
#[test]
|
|
466
|
+
fn test_normalize_spaces_empty() {
|
|
467
|
+
assert_eq!(normalize_spaces(""), "");
|
|
468
|
+
assert_eq!(normalize_spaces(" "), "");
|
|
469
|
+
}
|
|
470
|
+
|
|
471
|
+
#[test]
|
|
472
|
+
fn test_normalize_spaces_single_paragraph() {
|
|
473
|
+
let text = "This is a test";
|
|
474
|
+
let normalized = normalize_spaces(text);
|
|
475
|
+
assert_eq!(normalized, "This is a test");
|
|
476
|
+
}
|
|
477
|
+
|
|
478
|
+
#[test]
|
|
479
|
+
fn test_calculate_quality_score_with_metadata() {
|
|
480
|
+
let text = "This is a normal text with proper structure.";
|
|
481
|
+
let mut metadata = HashMap::new();
|
|
482
|
+
metadata.insert("title".to_string(), "Test Title".to_string());
|
|
483
|
+
metadata.insert("author".to_string(), "Test Author".to_string());
|
|
484
|
+
|
|
485
|
+
let score = calculate_quality_score(text, Some(&metadata));
|
|
486
|
+
assert!(score > 0.0);
|
|
487
|
+
assert!(score <= 1.0);
|
|
488
|
+
}
|
|
489
|
+
|
|
490
|
+
#[test]
|
|
491
|
+
fn test_calculate_ocr_penalty_clean_text() {
|
|
492
|
+
let text = "This is clean text without artifacts";
|
|
493
|
+
let penalty = calculate_ocr_penalty(text, text.len() as f64);
|
|
494
|
+
assert_eq!(penalty, 0.0);
|
|
495
|
+
}
|
|
496
|
+
|
|
497
|
+
#[test]
|
|
498
|
+
fn test_calculate_ocr_penalty_with_artifacts() {
|
|
499
|
+
let text = "Text with excessive spaces and ....... dots";
|
|
500
|
+
let penalty = calculate_ocr_penalty(text, text.len() as f64);
|
|
501
|
+
assert!(penalty > 0.0);
|
|
502
|
+
assert!(penalty <= 1.0);
|
|
503
|
+
}
|
|
504
|
+
|
|
505
|
+
#[test]
|
|
506
|
+
fn test_calculate_script_penalty_clean_text() {
|
|
507
|
+
let text = "This is clean text without scripts";
|
|
508
|
+
let penalty = calculate_script_penalty(text, text.len() as f64);
|
|
509
|
+
assert_eq!(penalty, 0.0);
|
|
510
|
+
}
|
|
511
|
+
|
|
512
|
+
#[test]
|
|
513
|
+
fn test_calculate_script_penalty_with_js() {
|
|
514
|
+
let text = "function test() { return 42; }";
|
|
515
|
+
let penalty = calculate_script_penalty(text, text.len() as f64);
|
|
516
|
+
assert!(penalty > 0.0);
|
|
517
|
+
}
|
|
518
|
+
|
|
519
|
+
#[test]
|
|
520
|
+
fn test_calculate_navigation_penalty_clean_text() {
|
|
521
|
+
let text = "This is clean text without navigation";
|
|
522
|
+
let penalty = calculate_navigation_penalty(text, text.len() as f64);
|
|
523
|
+
assert_eq!(penalty, 0.0);
|
|
524
|
+
}
|
|
525
|
+
|
|
526
|
+
#[test]
|
|
527
|
+
fn test_calculate_navigation_penalty_with_nav() {
|
|
528
|
+
let text = "Skip to main content and Back to top links everywhere";
|
|
529
|
+
let penalty = calculate_navigation_penalty(text, text.len() as f64);
|
|
530
|
+
assert!(penalty > 0.0);
|
|
531
|
+
}
|
|
532
|
+
|
|
533
|
+
#[test]
|
|
534
|
+
fn test_calculate_structure_bonus_empty() {
|
|
535
|
+
assert_eq!(calculate_structure_bonus(""), 0.0);
|
|
536
|
+
}
|
|
537
|
+
|
|
538
|
+
#[test]
|
|
539
|
+
fn test_calculate_structure_bonus_well_structured() {
|
|
540
|
+
let text = "This is a sentence. This is another sentence.\n\nNew paragraph here. More content.";
|
|
541
|
+
let bonus = calculate_structure_bonus(text);
|
|
542
|
+
assert!(bonus > 0.0);
|
|
543
|
+
assert!(bonus <= 1.0);
|
|
544
|
+
}
|
|
545
|
+
|
|
546
|
+
#[test]
|
|
547
|
+
fn test_calculate_metadata_bonus_empty() {
|
|
548
|
+
let metadata = HashMap::new();
|
|
549
|
+
let bonus = calculate_metadata_bonus(&metadata);
|
|
550
|
+
assert_eq!(bonus, 0.0);
|
|
551
|
+
}
|
|
552
|
+
|
|
553
|
+
#[test]
|
|
554
|
+
fn test_calculate_metadata_bonus_full() {
|
|
555
|
+
let mut metadata = HashMap::new();
|
|
556
|
+
metadata.insert("title".to_string(), "Title".to_string());
|
|
557
|
+
metadata.insert("author".to_string(), "Author".to_string());
|
|
558
|
+
metadata.insert("subject".to_string(), "Subject".to_string());
|
|
559
|
+
metadata.insert("description".to_string(), "Description".to_string());
|
|
560
|
+
metadata.insert("keywords".to_string(), "Keywords".to_string());
|
|
561
|
+
|
|
562
|
+
let bonus = calculate_metadata_bonus(&metadata);
|
|
563
|
+
assert_eq!(bonus, 1.0);
|
|
564
|
+
}
|
|
565
|
+
|
|
566
|
+
#[test]
|
|
567
|
+
fn test_clean_extracted_text_removes_styles() {
|
|
568
|
+
let text = "Before <style>.class { color: red; }</style> After";
|
|
569
|
+
let cleaned = clean_extracted_text(text);
|
|
570
|
+
assert!(!cleaned.contains("<style"));
|
|
571
|
+
assert!(cleaned.contains("Before"));
|
|
572
|
+
assert!(cleaned.contains("After"));
|
|
573
|
+
}
|
|
574
|
+
|
|
575
|
+
#[test]
|
|
576
|
+
fn test_clean_extracted_text_ocr_artifacts() {
|
|
577
|
+
let text = "Text with excessive spaces";
|
|
578
|
+
let cleaned = clean_extracted_text(text);
|
|
579
|
+
assert!(!cleaned.contains(" "));
|
|
580
|
+
}
|
|
581
|
+
|
|
582
|
+
#[test]
|
|
583
|
+
fn test_clean_extracted_text_navigation() {
|
|
584
|
+
let text = "Content Skip to main content more content";
|
|
585
|
+
let cleaned = clean_extracted_text(text);
|
|
586
|
+
assert!(cleaned.contains("Content"));
|
|
587
|
+
assert!(cleaned.contains("more content"));
|
|
588
|
+
}
|
|
589
|
+
|
|
590
|
+
#[test]
|
|
591
|
+
fn test_normalize_spaces_multiple_paragraphs() {
|
|
592
|
+
let text = "First paragraph.\n\nSecond paragraph.";
|
|
593
|
+
let normalized = normalize_spaces(text);
|
|
594
|
+
assert!(normalized.contains("\n\n"));
|
|
595
|
+
}
|
|
596
|
+
|
|
597
|
+
#[test]
|
|
598
|
+
fn test_normalize_spaces_preserves_paragraphs() {
|
|
599
|
+
let text = "Para 1\n\n\n\nPara 2";
|
|
600
|
+
let normalized = normalize_spaces(text);
|
|
601
|
+
assert_eq!(normalized, "Para 1\n\nPara 2");
|
|
602
|
+
}
|
|
603
|
+
|
|
604
|
+
#[test]
|
|
605
|
+
fn test_count_non_table_dash_artifacts() {
|
|
606
|
+
let text = "Some text --- with dashes";
|
|
607
|
+
let count = count_non_table_dash_artifacts(text);
|
|
608
|
+
assert!(count > 0);
|
|
609
|
+
}
|
|
610
|
+
|
|
611
|
+
#[test]
|
|
612
|
+
fn test_count_non_table_dash_artifacts_preserves_tables() {
|
|
613
|
+
let text = "| Header |\n|--------|\n| Data |";
|
|
614
|
+
let count = count_non_table_dash_artifacts(text);
|
|
615
|
+
assert_eq!(count, 0);
|
|
616
|
+
}
|
|
617
|
+
|
|
618
|
+
#[test]
|
|
619
|
+
fn test_clean_dashes_preserve_tables_simple() {
|
|
620
|
+
let text = Cow::Borrowed("| Col1 |\n|------|\n| Data |");
|
|
621
|
+
let result = clean_dashes_preserve_tables(text);
|
|
622
|
+
assert!(result.contains("|------"));
|
|
623
|
+
}
|
|
624
|
+
|
|
625
|
+
#[test]
|
|
626
|
+
fn test_clean_dashes_preserve_tables_replaces_non_table() {
|
|
627
|
+
let text = Cow::Borrowed("Text with --- dashes");
|
|
628
|
+
let result = clean_dashes_preserve_tables(text);
|
|
629
|
+
assert!(result.contains("..."));
|
|
630
|
+
assert!(!result.contains("---"));
|
|
631
|
+
}
|
|
632
|
+
|
|
633
|
+
#[test]
|
|
634
|
+
fn test_sum_match_lengths() {
|
|
635
|
+
let text = "test ... test ... test";
|
|
636
|
+
let count = sum_match_lengths(text, &REPEATED_PUNCT_PATTERN);
|
|
637
|
+
assert!(count > 0);
|
|
638
|
+
}
|
|
639
|
+
|
|
640
|
+
#[test]
|
|
641
|
+
fn test_quality_score_large_text_with_ocr_issues() {
|
|
642
|
+
let text = "a".repeat(2000) + " " + &"b".repeat(2000);
|
|
643
|
+
let score = calculate_quality_score(&text, None);
|
|
644
|
+
assert!(score >= 0.0);
|
|
645
|
+
assert!(score <= 1.0);
|
|
646
|
+
}
|
|
647
|
+
|
|
648
|
+
#[test]
|
|
649
|
+
fn test_quality_score_clamped_to_range() {
|
|
650
|
+
let perfect_text = "This is perfect text. ".repeat(100);
|
|
651
|
+
let score = calculate_quality_score(&perfect_text, None);
|
|
652
|
+
assert!(score >= 0.0);
|
|
653
|
+
assert!(score <= 1.0);
|
|
654
|
+
}
|
|
655
|
+
|
|
656
|
+
#[test]
|
|
657
|
+
fn test_clean_extracted_text_scattered_chars() {
|
|
658
|
+
let text = "a b c scattered";
|
|
659
|
+
let cleaned = clean_extracted_text(text);
|
|
660
|
+
assert!(!cleaned.is_empty());
|
|
661
|
+
}
|
|
662
|
+
|
|
663
|
+
#[test]
|
|
664
|
+
fn test_normalize_whitespace_cow_no_changes() {
|
|
665
|
+
let text = Cow::Borrowed("normaltext");
|
|
666
|
+
let result = normalize_whitespace_cow(text);
|
|
667
|
+
assert_eq!(result.as_ref(), "normaltext");
|
|
668
|
+
}
|
|
669
|
+
|
|
670
|
+
#[test]
|
|
671
|
+
fn test_normalize_whitespace_cow_with_changes() {
|
|
672
|
+
let text = Cow::Borrowed("text with spaces");
|
|
673
|
+
let result = normalize_whitespace_cow(text);
|
|
674
|
+
assert!(matches!(result, Cow::Owned(_)));
|
|
675
|
+
}
|
|
676
|
+
|
|
677
|
+
#[test]
|
|
678
|
+
fn test_clean_scripts_no_scripts() {
|
|
679
|
+
let text = Cow::Borrowed("clean text");
|
|
680
|
+
let result = clean_scripts(text);
|
|
681
|
+
assert!(matches!(result, Cow::Borrowed(_)));
|
|
682
|
+
}
|
|
683
|
+
|
|
684
|
+
#[test]
|
|
685
|
+
fn test_clean_scripts_with_script_tag() {
|
|
686
|
+
let text = Cow::Borrowed("<script>code</script>");
|
|
687
|
+
let result = clean_scripts(text);
|
|
688
|
+
assert!(!result.contains("<script"));
|
|
689
|
+
}
|
|
690
|
+
|
|
691
|
+
#[test]
|
|
692
|
+
fn test_quality_constants() {
|
|
693
|
+
assert_eq!(MIN_TEXT_LENGTH, 10);
|
|
694
|
+
assert_eq!(LARGE_TEXT_LENGTH, 1000);
|
|
695
|
+
assert_eq!(OCR_PENALTY_WEIGHT, 0.3);
|
|
696
|
+
}
|
|
697
|
+
}
|