// Licensed to the LF AI & Data foundation under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#pragma once

#include <cstdint>
#include <sstream>
#include <string>
#include <unordered_map>

#include "common/EasyAssert.h"
#include "common/Types.h"
#include "fmt/core.h"
#include "pb/index_coord.pb.h"
#include "pb/schema.pb.h"

namespace milvus::storage {

using DataType = milvus::DataType;
using Timestamp = milvus::Timestamp;

const int32_t MAGIC_NUM = 0xfffabc;

enum StorageType {
    None = 0,
    Memory = 1,
    LocalDisk = 2,
    Remote = 3,
};

enum class FileType : int32_t {
    InsertBinlog = 0,     // InsertBinlog FileType for insert data
    DeleteBinlog = 1,     // DeleteBinlog FileType for delete data
    DDLBinlog = 2,        // DDLBinlog FileType for DDL
    IndexFileBinlog = 3,  // IndexFileBinlog FileType for index

    KWInsertBinlog =
        100,  // InsertBinlog FileType for insert data prepared for knowhere
    KWIndexFileBinlog =
        101,  // IndexFileBinlog FileType for index generated by knowhere
};

enum class EventType : int8_t {
    DescriptorEvent = 0,
    InsertEvent = 1,
    DeleteEvent = 2,
    CreateCollectionEvent = 3,
    DropCollectionEvent = 4,
    CreatePartitionEvent = 5,
    DropPartitionEvent = 6,
    IndexFileEvent = 7,
    EventTypeEnd = 8,
};

// segment/field meta information corresponding to binlog file data
struct FieldDataMeta {
    int64_t collection_id;
    int64_t partition_id;
    int64_t segment_id;
    int64_t field_id;
    proto::schema::FieldSchema field_schema;
};

// Column name mapping used when reading StorageV3 manifests. The index config
// must not carry raw external_spec because Knowhere also consumes that config.
struct StorageColumnMapping {
    std::string schema_column_name;
    std::string storage_column_name;
    bool is_external_column = false;
};

using StorageColumnMappings = std::unordered_map<int64_t, StorageColumnMapping>;

enum CodecType {
    InvalidCodecType = 0,
    InsertDataType = 1,
    IndexDataType = 2,
};

// index meta information corresponding to index file data
struct IndexMeta {
    int64_t segment_id;
    int64_t field_id;
    int64_t build_id;
    int64_t index_version;
    std::string key;
    std::string field_name;
    DataType field_type;
    int64_t dim;
    bool index_non_encoding;
    // Path format version used by the worker/loader to assemble the object-storage prefix.
    // See milvus.proto.index.IndexStorePathVersion.
    milvus::proto::index::IndexStorePathVersion index_store_path_version{
        milvus::proto::index::IndexStorePathVersion::
            INDEX_STORE_PATH_VERSION_BUILD_ROOTED};
};

struct StorageConfig {
    std::string address = "localhost:9000";
    std::string bucket_name = "a-bucket";
    std::string access_key_id = "minioadmin";
    std::string access_key_value = "minioadmin";
    std::string root_path = "files";
    std::string storage_type = "minio";
    std::string cloud_provider = "aws";
    std::string iam_endpoint = "";
    std::string log_level = "warn";
    std::string region = "";
    bool useSSL = false;
    std::string sslCACert = "";
    bool useIAM = false;
    bool useVirtualHost = false;
    int64_t requestTimeoutMs = 3000;
    bool gcp_native_without_auth = false;
    std::string gcp_credential_json = "";
    uint32_t max_connections = 100;
    std::string tls_min_version = "";
    bool use_crc32c_checksum = false;

    std::string
    ToString() const {
        std::stringstream ss;
        ss << "[address=" << address << ", bucket_name=" << bucket_name
           << ", root_path=" << root_path << ", storage_type=" << storage_type
           << ", cloud_provider=" << cloud_provider
           << ", iam_endpoint=" << iam_endpoint << ", log_level=" << log_level
           << ", region=" << region << ", useSSL=" << std::boolalpha << useSSL
           << ", sslCACert=" << sslCACert.size()  // only print cert length
           << ", useIAM=" << std::boolalpha << useIAM
           << ", useVirtualHost=" << std::boolalpha << useVirtualHost
           << ", requestTimeoutMs=" << requestTimeoutMs
           << ", maxConnections=" << max_connections
           << ", gcp_native_without_auth=" << std::boolalpha
           << gcp_native_without_auth << ", tls_min_version=" << tls_min_version
           << ", use_crc32c_checksum=" << std::boolalpha << use_crc32c_checksum
           << "]";

        return ss.str();
    }
};

struct MmapConfig {
    std::string cache_read_ahead_policy;
    std::string mmap_path;
    uint64_t disk_limit;
    uint64_t fix_file_size;
    bool growing_enable_mmap;
    bool scalar_index_enable_mmap;
    bool scalar_field_enable_mmap;
    bool vector_index_enable_mmap;
    bool vector_field_enable_mmap;
    bool mmap_populate;
    bool json_stats_enable_mmap;
    std::string json_stats_mmap_path;
    bool
    GetEnableGrowingMmap() const {
        return growing_enable_mmap;
    }
    void
    SetEnableGrowingMmap(bool flag) {
        this->growing_enable_mmap = flag;
    }
    bool
    GetScalarIndexEnableMmap() const {
        return scalar_index_enable_mmap;
    }
    void
    SetScalarIndexEnableMmap(bool flag) {
        this->scalar_index_enable_mmap = flag;
    }
    bool
    GetScalarFieldEnableMmap() const {
        return scalar_field_enable_mmap;
    }
    void
    SetScalarFieldEnableMmap(bool flag) {
        this->scalar_field_enable_mmap = flag;
    }
    bool
    GetVectorIndexEnableMmap() const {
        return vector_index_enable_mmap;
    }
    void
    SetVectorIndexEnableMmap(bool flag) {
        this->vector_index_enable_mmap = flag;
    }
    bool
    GetVectorFieldEnableMmap() const {
        return vector_field_enable_mmap;
    }
    void
    SetVectorFieldEnableMmap(bool flag) {
        this->vector_field_enable_mmap = flag;
    }
    [[nodiscard]] bool
    GetMmapPopulate() const {
        return mmap_populate;
    }

    [[nodiscard]] bool
    GetJsonStatsEnableMmap() const {
        return json_stats_enable_mmap;
    }

    [[nodiscard]] const std::string&
    GetJsonStatsMmapPath() const {
        return json_stats_mmap_path;
    }

    std::string
    GetMmapPath() {
        return mmap_path;
    }
    std::string
    ToString() const {
        std::stringstream ss;
        ss << "[cache_read_ahead_policy=" << cache_read_ahead_policy
           << ", mmap_path=" << mmap_path
           << ", disk_limit=" << disk_limit / (1024 * 1024) << "MB"
           << ", fix_file_size=" << fix_file_size / (1024 * 1024) << "MB"
           << ", growing_enable_mmap=" << std::boolalpha << growing_enable_mmap
           << ", scalar_index_enable_mmap=" << std::boolalpha
           << scalar_index_enable_mmap
           << ", scalar_field_enable_mmap=" << std::boolalpha
           << scalar_field_enable_mmap
           << ", vector_index_enable_mmap=" << std::boolalpha
           << vector_index_enable_mmap
           << ", vector_field_enable_mmap=" << std::boolalpha
           << vector_field_enable_mmap
           << ", json_stats_enable_mmap=" << std::boolalpha
           << json_stats_enable_mmap
           << ", json_stats_mmap_path=" << json_stats_mmap_path << "]";
        return ss.str();
    }
};

}  // namespace milvus::storage

template <>
struct fmt::formatter<milvus::storage::EventType> : formatter<string_view> {
    auto
    format(milvus::storage::EventType c, format_context& ctx) const {
        string_view name = "unknown";
        switch (c) {
            case milvus::storage::EventType::DescriptorEvent:
                name = "DescriptorEvent";
                break;
            case milvus::storage::EventType::InsertEvent:
                name = "InsertEvent";
                break;
            case milvus::storage::EventType::DeleteEvent:
                name = "DeleteEvent";
                break;
            case milvus::storage::EventType::CreateCollectionEvent:
                name = "CreateCollectionEvent";
                break;
            case milvus::storage::EventType::DropCollectionEvent:
                name = "DropCollectionEvent";
                break;
            case milvus::storage::EventType::CreatePartitionEvent:
                name = "CreatePartitionEvent";
                break;
            case milvus::storage::EventType::DropPartitionEvent:
                name = "DropPartitionEvent";
                break;
            case milvus::storage::EventType::IndexFileEvent:
                name = "IndexFileEvent";
                break;
            case milvus::storage::EventType::EventTypeEnd:
                name = "EventTypeEnd";
                break;
        }
        return formatter<string_view>::format(name, ctx);
    }
};

template <>
struct fmt::formatter<milvus::storage::StorageType> : formatter<string_view> {
    auto
    format(milvus::storage::StorageType c, format_context& ctx) const {
        switch (c) {
            case milvus::storage::StorageType::None:
                return formatter<string_view>::format("None", ctx);
            case milvus::storage::StorageType::Memory:
                return formatter<string_view>::format("Memory", ctx);
            case milvus::storage::StorageType::LocalDisk:
                return formatter<string_view>::format("LocalDisk", ctx);
            case milvus::storage::StorageType::Remote:
                return formatter<string_view>::format("Remote", ctx);
        }
        return formatter<string_view>::format("unknown", ctx);
    }
};

struct BytesBuf {
 public:
    explicit BytesBuf(const uint8_t* data, int64_t length)
        : data_(data), size_(length) {
        AssertInfo(data != nullptr, "Data pointer for slice cannot be null.");
    }

    int64_t
    Size() const {
        return size_;
    }

    const uint8_t*
    Data() const {
        return data_;
    }

 private:
    const uint8_t* data_;  // Only used when not owning the data
    int64_t size_;
};
