# -*- mode: python -*-

Import("env")

env = env.Clone()

env.Benchmark(
    target='key_gen_bm',
    source='key_gen_bm.cpp',
    LIBDEPS=[
        '$BUILD_DIR/mongo/base',
        'index_access_method',
    ],
)

env.Library(
    target='expression_params',
    source=[
        'expression_params.cpp',
        's2_common.cpp',
    ],
    LIBDEPS_PRIVATE=[
        '$BUILD_DIR/mongo/base',
        '$BUILD_DIR/mongo/bson/util/bson_extract',
        '$BUILD_DIR/mongo/db/geo/geometry',
        '$BUILD_DIR/mongo/db/geo/geoparser',
        '$BUILD_DIR/mongo/db/mongohasher',
        '$BUILD_DIR/mongo/db/query/collation/collator_interface',
        '$BUILD_DIR/mongo/db/storage/key_string',
        '$BUILD_DIR/third_party/s2/s2',
    ],
)

iamEnv = env.Clone()
iamEnv.InjectThirdParty(libraries=['snappy'])
iamEnv.Library(
    target='index_access_method',
    source=[
        '2d_access_method.cpp',
        'btree_access_method.cpp',
        'btree_key_generator.cpp',
        'column_key_generator.cpp',
        'columns_access_method.cpp',
        'duplicate_key_tracker.cpp',
        'expression_keys_private.cpp',
        'fts_access_method.cpp',
        'hash_access_method.cpp',
        'index_access_method.cpp',
        'index_build_interceptor.cpp',
        'index_build_interceptor.idl',
        'index_descriptor.cpp',
        's2_access_method.cpp',
        's2_bucket_access_method.cpp',
        'skipped_record_tracker.cpp',
        'sort_key_generator.cpp',
        'wildcard_access_method.cpp',
        'wildcard_key_generator.cpp',
        'wildcard_validation.cpp',
    ],
    LIBDEPS_PRIVATE=[
        '$BUILD_DIR/mongo/db/bson/dotted_path_support',
        '$BUILD_DIR/mongo/db/catalog/index_catalog',
        '$BUILD_DIR/mongo/db/catalog/index_catalog_entry',
        '$BUILD_DIR/mongo/db/concurrency/exception_util',
        '$BUILD_DIR/mongo/db/exec/projection_executor',
        '$BUILD_DIR/mongo/db/exec/working_set',
        '$BUILD_DIR/mongo/db/fts/base_fts',
        '$BUILD_DIR/mongo/db/geo/geoparser',
        '$BUILD_DIR/mongo/db/mongohasher',
        '$BUILD_DIR/mongo/db/multi_key_path_tracker',
        '$BUILD_DIR/mongo/db/pipeline/document_path_support',
        '$BUILD_DIR/mongo/db/query/collation/collator_factory_interface',
        '$BUILD_DIR/mongo/db/query/op_metrics',
        '$BUILD_DIR/mongo/db/query/projection_ast',
        '$BUILD_DIR/mongo/db/query/sort_pattern',
        '$BUILD_DIR/mongo/db/query_expressions',
        '$BUILD_DIR/mongo/db/record_id_helpers',
        '$BUILD_DIR/mongo/db/resumable_index_builds_idl',
        '$BUILD_DIR/mongo/db/server_base',
        '$BUILD_DIR/mongo/db/service_context',
        '$BUILD_DIR/mongo/db/shard_role_api_stor_ex',
        '$BUILD_DIR/mongo/db/sorter/sorter_idl',
        '$BUILD_DIR/mongo/db/sorter/sorter_stats',
        '$BUILD_DIR/mongo/db/storage/encryption_hooks',
        '$BUILD_DIR/mongo/db/storage/execution_context',
        '$BUILD_DIR/mongo/db/storage/index_entry_comparison',
        '$BUILD_DIR/mongo/db/storage/key_string',
        '$BUILD_DIR/mongo/db/storage/record_store_base',
        '$BUILD_DIR/mongo/db/storage/storage_options',
        '$BUILD_DIR/mongo/db/timeseries/timeseries_conversion_util',
        '$BUILD_DIR/mongo/db/vector_clock',
        '$BUILD_DIR/mongo/util/progress_meter',
        '$BUILD_DIR/third_party/s2/s2',
        '$BUILD_DIR/third_party/shim_snappy',
        'column_store_index',
        'expression_params',
    ],
)

iamEnv.Library(
    target='column_store_index',
    source=[
        'column_cell.cpp',
        'column_store_sorter.cpp',
    ],
    LIBDEPS_PRIVATE=[
        '$BUILD_DIR/mongo/base',
        '$BUILD_DIR/mongo/db/sorter/sorter_idl',
        '$BUILD_DIR/mongo/db/sorter/sorter_stats',
        '$BUILD_DIR/mongo/db/storage/encryption_hooks',
        '$BUILD_DIR/mongo/db/storage/storage_options',
        '$BUILD_DIR/mongo/s/is_mongos',
        '$BUILD_DIR/third_party/shim_snappy',
    ],
)

indexTestEnv = env.Clone()
indexTestEnv.InjectThirdParty(libraries=['snappy'])
indexTestEnv.CppUnitTest(
    target='db_index_test',
    source=[
        '2d_key_generator_test.cpp',
        'btree_key_generator_test.cpp',
        'column_cell_test.cpp',
        'column_key_generator_test.cpp',
        'column_store_sorter_test.cpp',
        'columns_access_method_test.cpp',
        'hash_key_generator_test.cpp',
        'index_build_interceptor_test.cpp',
        's2_key_generator_test.cpp',
        's2_bucket_key_generator_test.cpp',
        'sort_key_generator_test.cpp',
        'wildcard_key_generator_test.cpp',
        "wildcard_validation_test.cpp",
    ],
    LIBDEPS=[
        '$BUILD_DIR/mongo/db/catalog/catalog_test_fixture',
        '$BUILD_DIR/mongo/db/exec/document_value/document_value_test_util',
        '$BUILD_DIR/mongo/db/exec/working_set',
        '$BUILD_DIR/mongo/db/mongohasher',
        '$BUILD_DIR/mongo/db/query/collation/collator_interface_mock',
        '$BUILD_DIR/mongo/db/query/query_test_service_context',
        '$BUILD_DIR/mongo/db/query/sort_pattern',
        '$BUILD_DIR/mongo/db/query_expressions',
        '$BUILD_DIR/mongo/db/record_id_helpers',
        '$BUILD_DIR/mongo/db/shard_role',
        '$BUILD_DIR/mongo/db/sorter/sorter_stats',
        'column_store_index',
        'expression_params',
        'index_access_method',
    ],
)