/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
 *   Copyright 2020-2021 Couchbase, Inc.
 *
 *   Licensed 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 "diagnostics.hxx"

#include <fmt/core.h>

template<>
struct fmt::formatter<couchbase::core::diag::cluster_state> {
    template<typename ParseContext>
    constexpr auto parse(ParseContext& ctx)
    {
        return ctx.begin();
    }

    template<typename FormatContext>
    auto format(couchbase::core::diag::cluster_state state, FormatContext& ctx) const
    {
        string_view name = "unknown";
        switch (state) {
            case couchbase::core::diag::cluster_state::online:
                name = "online";
                break;

            case couchbase::core::diag::cluster_state::degraded:
                name = "degraded";
                break;

            case couchbase::core::diag::cluster_state::offline:
                name = "offline";
                break;
        }
        return format_to(ctx.out(), "{}", name);
    }
};

template<>
struct fmt::formatter<couchbase::core::diag::endpoint_state> {
    template<typename ParseContext>
    constexpr auto parse(ParseContext& ctx)
    {
        return ctx.begin();
    }

    template<typename FormatContext>
    auto format(couchbase::core::diag::endpoint_state state, FormatContext& ctx) const
    {
        string_view name = "unknown";
        switch (state) {
            case couchbase::core::diag::endpoint_state::disconnected:
                name = "disconnected";
                break;

            case couchbase::core::diag::endpoint_state::connecting:
                name = "connecting";
                break;

            case couchbase::core::diag::endpoint_state::connected:
                name = "connected";
                break;

            case couchbase::core::diag::endpoint_state::disconnecting:
                name = "disconnecting";
                break;
        }
        return format_to(ctx.out(), "{}", name);
    }
};

template<>
struct fmt::formatter<couchbase::core::diag::ping_state> {
    template<typename ParseContext>
    constexpr auto parse(ParseContext& ctx)
    {
        return ctx.begin();
    }

    template<typename FormatContext>
    auto format(couchbase::core::diag::ping_state state, FormatContext& ctx) const
    {
        string_view name = "unknown";
        switch (state) {
            case couchbase::core::diag::ping_state::ok:
                name = "ok";
                break;

            case couchbase::core::diag::ping_state::timeout:
                name = "timeout";
                break;

            case couchbase::core::diag::ping_state::error:
                name = "error";
                break;
        }
        return format_to(ctx.out(), "{}", name);
    }
};