//===-- ProtocolServer.cpp ------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "lldb/Core/ProtocolServer.h"
#include "lldb/Core/PluginManager.h"
#include "llvm/Support/Error.h"

using namespace lldb_private;
using namespace lldb;

static std::pair<llvm::StringMap<ProtocolServerUP> &, std::mutex &> Servers() {
  static llvm::StringMap<ProtocolServerUP> g_protocol_server_instances;
  static std::mutex g_mutex;
  return {g_protocol_server_instances, g_mutex};
}

ProtocolServer *ProtocolServer::GetOrCreate(llvm::StringRef name) {
  auto [protocol_server_instances, mutex] = Servers();

  std::lock_guard<std::mutex> guard(mutex);

  auto it = protocol_server_instances.find(name);
  if (it != protocol_server_instances.end())
    return it->second.get();

  if (ProtocolServerCreateInstance create_callback =
          PluginManager::GetProtocolCreateCallbackForPluginName(name)) {
    auto pair = protocol_server_instances.try_emplace(name, create_callback());
    return pair.first->second.get();
  }

  return nullptr;
}

std::vector<llvm::StringRef> ProtocolServer::GetSupportedProtocols() {
  std::vector<llvm::StringRef> supported_protocols;
  size_t i = 0;

  for (llvm::StringRef protocol_name =
           PluginManager::GetProtocolServerPluginNameAtIndex(i++);
       !protocol_name.empty();
       protocol_name = PluginManager::GetProtocolServerPluginNameAtIndex(i++)) {
    supported_protocols.push_back(protocol_name);
  }

  return supported_protocols;
}

llvm::Error ProtocolServer::Terminate() {
  llvm::Error error = llvm::Error::success();

  auto [protocol_server_instances, mutex] = Servers();
  std::lock_guard<std::mutex> guard(mutex);
  for (auto &instance : protocol_server_instances) {
    if (llvm::Error instance_error = instance.second->Stop())
      error = llvm::joinErrors(std::move(error), std::move(instance_error));
  }

  protocol_server_instances.clear();

  return error;
}