Mandoline DevTools service: remote debugging protocol over HTTP/WebSocket.

components/devtools_service/devtools_registry_impl.cc

 // Copyright 2015 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 "components/devtools_service/devtools_registry_impl.h"
 
 #include "base/logging.h"
+#include "components/devtools_service/devtools_agent_host.h"
 
 namespace devtools_service {
 
+DevToolsRegistryImpl::Iterator::Iterator(DevToolsRegistryImpl* registry)
+    : registry_(registry), iter_(registry->agents_.begin()) {
+}
+
+DevToolsRegistryImpl::Iterator::~Iterator() {
+}
+
 DevToolsRegistryImpl::DevToolsRegistryImpl(DevToolsService* service)
     : service_(service) {
 }
 
 DevToolsRegistryImpl::~DevToolsRegistryImpl() {
 }
 
 void DevToolsRegistryImpl::BindToRegistryRequest(
     mojo::InterfaceRequest<DevToolsRegistry> request) {
   bindings_.AddBinding(this, request.Pass());
 }
 
+DevToolsAgentHost* DevToolsRegistryImpl::GetAgentById(const std::string& id) {
+  auto iter = agents_.find(id);
+  if (iter == agents_.end())
+    return nullptr;
+
+  return iter->second.get();
+}
+
 void DevToolsRegistryImpl::RegisterAgent(DevToolsAgentPtr agent) {
-  // TODO(yzshen): Implement it.
-  NOTIMPLEMENTED();
+  linked_ptr<DevToolsAgentHost> agent_host(new DevToolsAgentHost(agent.Pass()));
+  std::string id = agent_host->id();
+  agent_host->set_agent_connection_error_handler(
+      [this, id]() { OnAgentConnectionError(id); });
+
+  agents_[agent_host->id()] = agent_host;
+}
+
+void DevToolsRegistryImpl::OnAgentConnectionError(const std::string& id) {
+  DCHECK(agents_.find(id) != agents_.end());
+  agents_.erase(id);
 }
 
 }  // namespace devtools_service