// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "remoting/host/host_extension_session_manager.h"

#include "base/check.h"
#include "remoting/base/capabilities.h"
#include "remoting/host/client_session_details.h"
#include "remoting/host/host_extension.h"
#include "remoting/host/host_extension_session.h"

namespace remoting {

HostExtensionSessionManager::HostExtensionSessionManager(
    const HostExtensions& extensions,
    ClientSessionDetails* client_session_details)
    : client_session_details_(client_session_details),
      client_stub_(nullptr),
      extensions_(extensions) {}

HostExtensionSessionManager::~HostExtensionSessionManager() = default;

std::string HostExtensionSessionManager::GetCapabilities() const {
  std::string capabilities;
  for (auto* extension : extensions_) {
    const std::string& capability = extension->capability();
    if (capability.empty()) {
      continue;
    }
    if (!capabilities.empty()) {
      capabilities.append(" ");
    }
    capabilities.append(capability);
  }
  return capabilities;
}

HostExtensionSession* HostExtensionSessionManager::FindExtensionSession(
    const std::string& capability) {
  auto iter = extension_sessions_.find(capability);
  if (iter == extension_sessions_.end()) {
    return nullptr;
  }
  return iter->second.get();
}

void HostExtensionSessionManager::OnNegotiatedCapabilities(
    protocol::ClientStub* client_stub,
    const std::string& capabilities) {
  DCHECK(client_stub);
  DCHECK(!client_stub_);

  client_stub_ = client_stub;

  for (auto* extension : extensions_) {
    // If the extension requires a capability that was not negotiated then do
    // not instantiate it.
    if (!extension->capability().empty() &&
        !HasCapability(capabilities, extension->capability())) {
      continue;
    }

    std::unique_ptr<HostExtensionSession> extension_session =
        extension->CreateExtensionSession(client_session_details_,
                                          client_stub_);
    DCHECK(extension_session);

    extension_sessions_.emplace(extension->capability(),
                                std::move(extension_session));
  }
}

bool HostExtensionSessionManager::OnExtensionMessage(
    const protocol::ExtensionMessage& message) {
  for (const auto& [capability, session] : extension_sessions_) {
    if (session->OnExtensionMessage(client_session_details_, client_stub_,
                                    message)) {
      return true;
    }
  }
  return false;
}

}  // namespace remoting