/*
 * Copyright (C) 2025-present ScyllaDB
 *
 */

/*
 * SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
 */

#pragma once

#include <string>
#include <variant>
#include <compare>
#include <fmt/core.h>
#include "utils/s3/creds.hh"

namespace YAML {
    class Node;
}

namespace db {

class object_storage_endpoint_param {
public:
    struct s3_storage {
        std::string endpoint;
        s3::endpoint_config config;

        std::strong_ordering operator<=>(const s3_storage&) const = default;
        std::string to_json_string() const;
        std::string key() const;
    };
    struct gs_storage {
        std::string endpoint; // "default"/empty or an URI
        std::string credentials_file; // optional

        std::strong_ordering operator<=>(const gs_storage&) const = default;
        std::string to_json_string() const;
        std::string key() const;
    };

    object_storage_endpoint_param();
    object_storage_endpoint_param(const object_storage_endpoint_param&);
    object_storage_endpoint_param(s3_storage);
    object_storage_endpoint_param(std::string endpoint, s3::endpoint_config config);
    object_storage_endpoint_param(gs_storage);

    std::strong_ordering operator<=>(const object_storage_endpoint_param&) const;
    bool operator==(const object_storage_endpoint_param&) const;

    std::string to_json_string() const;
    std::string key() const;
    const std::string& type() const;

    bool is_s3_storage() const;
    bool is_gs_storage() const;

    bool is_storage_of_type(std::string_view) const;

    const s3_storage& get_s3_storage() const;
    const gs_storage& get_gs_storage() const;

    static object_storage_endpoint_param decode(const YAML::Node&);
    static const std::string s3_type; // "s3"
    static const std::string gs_type; // "gs"
private:
    friend fmt::formatter<object_storage_endpoint_param>;
    std::variant<s3_storage, gs_storage> _data;
};

std::istream& operator>>(std::istream& is, object_storage_endpoint_param& f);

}

template <>
struct fmt::formatter<db::object_storage_endpoint_param> : fmt::formatter<std::string_view> {
    auto format(const db::object_storage_endpoint_param&, fmt::format_context& ctx) const -> decltype(ctx.out());
};