// Copyright 2023 Huawei Cloud Computing Technology Co., Ltd.
//
// 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.

#ifndef INCLUDED_SRC_BUILDTOOL_SERVE_API_REMOTE_SERVE_API_HPP
#define INCLUDED_SRC_BUILDTOOL_SERVE_API_REMOTE_SERVE_API_HPP

#include <memory>
#include <optional>
#include <string>

#include "src/buildtool/common/artifact.hpp"
#include "src/buildtool/common/artifact_digest.hpp"
#include "src/buildtool/common/remote/port.hpp"
#include "src/buildtool/file_system/symlinks_map/pragma_special.hpp"
#include "src/buildtool/serve_api/remote/config.hpp"
#include "src/buildtool/serve_api/remote/configuration_client.hpp"
#include "src/buildtool/serve_api/remote/source_tree_client.hpp"
#include "src/buildtool/serve_api/remote/target_client.hpp"

class ServeApi final {
  public:
    ServeApi(ServeApi const&) = delete;
    ~ServeApi() = default;

    auto operator=(ServeApi const&) -> ServeApi& = delete;
    auto operator=(ServeApi&&) -> ServeApi& = delete;

    [[nodiscard]] static auto Instance() noexcept -> ServeApi& {
        static ServeApi instance = ServeApi::init();
        return instance;
    }

    [[nodiscard]] static auto RetrieveTreeFromCommit(
        std::string const& commit,
        std::string const& subdir = ".",
        bool sync_tree = false) noexcept -> std::variant<bool, std::string> {
        return Instance().stc_->ServeCommitTree(commit, subdir, sync_tree);
    }

    [[nodiscard]] static auto RetrieveTreeFromArchive(
        std::string const& content,
        std::string const& archive_type = "archive",
        std::string const& subdir = ".",
        std::optional<PragmaSpecial> const& resolve_symlinks = std::nullopt,
        bool sync_tree = false) noexcept -> std::variant<bool, std::string> {
        return Instance().stc_->ServeArchiveTree(
            content, archive_type, subdir, resolve_symlinks, sync_tree);
    }

    [[nodiscard]] static auto RetrieveTreeFromDistdir(
        std::shared_ptr<std::unordered_map<std::string, std::string>> const&
            distfiles,
        bool sync_tree = false) noexcept -> std::variant<bool, std::string> {
        return Instance().stc_->ServeDistdirTree(distfiles, sync_tree);
    }

    [[nodiscard]] static auto ContentInRemoteCAS(
        std::string const& content) noexcept -> bool {
        return Instance().stc_->ServeContent(content);
    }

    [[nodiscard]] static auto TreeInRemoteCAS(
        std::string const& tree_id) noexcept -> bool {
        return Instance().stc_->ServeTree(tree_id);
    }

    [[nodiscard]] static auto CheckRootTree(std::string const& tree_id) noexcept
        -> std::optional<bool> {
        return Instance().stc_->CheckRootTree(tree_id);
    }

    [[nodiscard]] static auto GetTreeFromRemote(
        std::string const& tree_id) noexcept -> bool {
        return Instance().stc_->GetRemoteTree(tree_id);
    }

    [[nodiscard]] static auto ServeTargetVariables(
        std::string const& target_root_id,
        std::string const& target_file,
        std::string const& target) noexcept
        -> std::optional<std::vector<std::string>> {
        return Instance().tc_->ServeTargetVariables(
            target_root_id, target_file, target);
    }

    [[nodiscard]] static auto ServeTargetDescription(
        std::string const& target_root_id,
        std::string const& target_file,
        std::string const& target) noexcept -> std::optional<ArtifactDigest> {
        return Instance().tc_->ServeTargetDescription(
            target_root_id, target_file, target);
    }

    [[nodiscard]] static auto ServeTarget(const TargetCacheKey& key,
                                          const std::string& repo_key) noexcept
        -> std::optional<std::pair<TargetCacheEntry, Artifact::ObjectInfo>> {
        return Instance().tc_->ServeTarget(key, repo_key);
    }

    [[nodiscard]] static auto CheckServeRemoteExecution() noexcept -> bool {
        return Instance().cc_->CheckServeRemoteExecution();
    }

    [[nodiscard]] static auto IsCompatible() noexcept -> std::optional<bool> {
        return Instance().cc_->IsCompatible();
    }

  private:
    ServeApi(std::string const& host, Port port) noexcept
        : stc_{std::make_unique<SourceTreeClient>(host, port)},
          tc_{std::make_unique<TargetClient>(host, port)},
          cc_{std::make_unique<ConfigurationClient>(host, port)} {}

    ServeApi(ServeApi&& other) noexcept = default;

    [[nodiscard]] static auto init() noexcept -> ServeApi {
        auto sadd = RemoteServeConfig::RemoteAddress();
        return ServeApi{sadd->host, sadd->port};
    }

    // source tree service client
    std::unique_ptr<SourceTreeClient> stc_;
    // target service client
    std::unique_ptr<TargetClient> tc_;
    // configuration service client
    std::unique_ptr<ConfigurationClient> cc_;
};

#endif  // INCLUDED_SRC_BUILDTOOL_SERVE_API_REMOTE_SERVE_API_HPP