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

#include "content/browser/devtools/target_registry.h"

#include "base/strings/stringprintf.h"
#include "content/browser/devtools/devtools_session.h"

namespace content {

TargetRegistry::TargetRegistry(DevToolsSession* root_session)
    : root_session_(root_session) {}

TargetRegistry::~TargetRegistry() {}

void TargetRegistry::AttachSubtargetSession(const std::string& session_id,
                                            DevToolsAgentHostImpl* agent_host,
                                            DevToolsAgentHostClient* client) {
  sessions_[session_id] = std::make_pair(agent_host, client);
  agent_host->AttachSubtargetClient(client, this);
}
void TargetRegistry::DetachSubtargetSession(const std::string& session_id) {
  sessions_.erase(session_id);
}

bool TargetRegistry::DispatchMessageOnAgentHost(
    const std::string& message,
    base::DictionaryValue* parsed_message) {
  std::string session_id;
  if (!parsed_message->GetString("sessionId", &session_id))
    return false;
  auto it = sessions_.find(session_id);
  if (it == sessions_.end()) {
    LOG(ERROR) << "Unknown session " << session_id;
    return true;
  }
  scoped_refptr<DevToolsAgentHostImpl> agent_host = it->second.first;
  DevToolsAgentHostClient* client = it->second.second;
  return agent_host->DispatchProtocolMessage(client, message, parsed_message);
}

void TargetRegistry::SendMessageToClient(const std::string& session_id,
                                         const std::string& message) {
  DCHECK(message[message.length() - 1] == '}');
  std::string suffix =
      base::StringPrintf(", \"sessionId\": \"%s\"}", session_id.c_str());
  std::string patched;
  patched.reserve(message.length() + suffix.length() - 1);
  patched.append(message.data(), message.length() - 1);
  patched.append(suffix);
  root_session_->client()->DispatchProtocolMessage(root_session_->agent_host(),
                                                   patched);
}

}  // namespace content