// Copyright 2023 PingCAP, 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.

#include <Common/Exception.h>
#include <Core/TiFlashDisaggregatedMode.h>
#include <Poco/Util/LayeredConfiguration.h>

namespace DB
{
DisaggOptions DisaggOptions::parseFromConfig(const Poco::Util::LayeredConfiguration & config)
{
    static const std::string config_key = "flash.disaggregated_mode";
    static const std::string autoscaler_config_key = "flash.use_autoscaler";
    static const std::string columnar_config_key = "flash.use_columnar";
    DisaggregatedMode mode = DisaggregatedMode::None;
    if (config.has(config_key))
    {
        std::string mode_str = config.getString(config_key);
        RUNTIME_ASSERT(
            mode_str == DISAGGREGATED_MODE_WRITE || mode_str == DISAGGREGATED_MODE_STORAGE // backward compatibility
                || mode_str == DISAGGREGATED_MODE_COMPUTE,
            "Expect disaggregated_mode is {} or {}, got: {}",
            DISAGGREGATED_MODE_WRITE,
            DISAGGREGATED_MODE_COMPUTE,
            mode_str);
        if (mode_str == DISAGGREGATED_MODE_COMPUTE)
        {
            mode = DisaggregatedMode::Compute;
        }
        else
        {
            mode = DisaggregatedMode::Storage;
        }
    }


    bool use_autoscaler = false;
    if (config.has(autoscaler_config_key))
        use_autoscaler = config.getBool(autoscaler_config_key);

    bool use_columnar = false;
    if (config.has(columnar_config_key))
        use_columnar = config.getBool(columnar_config_key);


    return DisaggOptions{mode, use_autoscaler, use_columnar};
}

bool useColumnar(const Poco::Util::LayeredConfiguration & config)
{
    static const std::string columnar_config_key = "flash.use_columnar";
    bool use_columnar = false;
    if (config.has(columnar_config_key))
        use_columnar = config.getBool(columnar_config_key);
    return use_columnar;
}

std::string getProxyLabelByDisaggregatedMode(DisaggregatedMode mode)
{
    switch (mode)
    {
    case DisaggregatedMode::Compute:
        return DISAGGREGATED_MODE_COMPUTE_PROXY_LABEL;
    case DisaggregatedMode::Storage:
    case DisaggregatedMode::None:
        return DEF_PROXY_LABEL;
    default:
        __builtin_unreachable();
    };
}
} // namespace DB