[DevTools] Support multiple clients in DevToolsAgentHost

content/browser/devtools/devtools_agent_host_impl.h

 // Copyright (c) 2012 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.
 
 #ifndef CONTENT_BROWSER_DEVTOOLS_DEVTOOLS_AGENT_HOST_IMPL_H_
 #define CONTENT_BROWSER_DEVTOOLS_DEVTOOLS_AGENT_HOST_IMPL_H_
 
 #include <stdint.h>
 
 #include <string>
 
 #include "base/compiler_specific.h"
 #include "base/containers/flat_set.h"
 #include "content/browser/devtools/devtools_io_context.h"
 #include "content/common/content_export.h"
 #include "content/common/devtools_messages.h"
 #include "content/public/browser/devtools_agent_host.h"
 
 namespace content {
 
 class BrowserContext;
 class DevToolsSession;
 
 // Describes interface for managing devtools agents from the browser process.
 class CONTENT_EXPORT DevToolsAgentHostImpl : public DevToolsAgentHost {
  public:
   // DevToolsAgentHost implementation.
   bool AttachClient(DevToolsAgentHostClient* client) override;
   void ForceAttachClient(DevToolsAgentHostClient* client) override;
+  bool AttachMultiClient(DevToolsAgentHostClient* client) override;
   bool DetachClient(DevToolsAgentHostClient* client) override;
   bool DispatchProtocolMessage(DevToolsAgentHostClient* client,
                                const std::string& message) override;
   bool IsAttached() override;
   void InspectElement(DevToolsAgentHostClient* client, int x, int y) override;
   std::string GetId() override;
   std::string GetParentId() override;
   std::string GetDescription() override;
   GURL GetFaviconURL() override;
   std::string GetFrontendURL() override;
@@ skipping 12 matching lines @@
   static bool ShouldForceCreation();
 
   virtual void AttachSession(DevToolsSession* session) = 0;
   virtual void DetachSession(int session_id) = 0;
   virtual bool DispatchProtocolMessage(
       DevToolsSession* session,
       const std::string& message) = 0;
   virtual void InspectElement(DevToolsSession* session, int x, int y);
 
   void NotifyCreated();
-  void ForceDetach(bool replaced);
+  void ForceDetachAllClients(bool replaced);
   DevToolsIOContext* GetIOContext() { return &io_context_; }
-
-  // TODO(dgozman): remove this accessor.
-  DevToolsSession* session() { return session_; }
+  const base::flat_map<int, DevToolsSession*>& sessions() {
+    return session_by_id_;
+  }
 
  private:
   friend class DevToolsAgentHost; // for static methods
   friend class DevToolsSession;
-  bool InnerAttachClient(DevToolsAgentHostClient* client, bool force);
-  void InnerDetachClient();
+  void InnerAttachClient(DevToolsAgentHostClient* client);
+  void InnerDetachClient(DevToolsAgentHostClient* client);
   void NotifyAttached();
   void NotifyDetached();
   void NotifyDestroyed();
 
   const std::string id_;
   int last_session_id_;
-  // TODO(dgozman): remove this together with accessor above.
-  DevToolsSession* session_;
-  base::flat_set<std::unique_ptr<DevToolsSession>> sessions_;
+  base::flat_map<DevToolsAgentHostClient*, std::unique_ptr<DevToolsSession>>
+      session_by_client_;
+  base::flat_map<int, DevToolsSession*> session_by_id_;
   DevToolsIOContext io_context_;
   static int s_attached_count_;
   static int s_force_creation_count_;
 };
 
 class DevToolsMessageChunkProcessor {
  public:
   using SendMessageCallback = base::Callback<void(int, const std::string&)>;
   explicit DevToolsMessageChunkProcessor(const SendMessageCallback& callback);
   ~DevToolsMessageChunkProcessor();
 
   std::string state_cookie() const { return state_cookie_; }
   void set_state_cookie(const std::string& cookie) { state_cookie_ = cookie; }
   int last_call_id() const { return last_call_id_; }
   bool ProcessChunkedMessageFromAgent(const DevToolsMessageChunk& chunk);
 
  private:
   SendMessageCallback callback_;
   std::string message_buffer_;
   uint32_t message_buffer_size_;
   std::string state_cookie_;
   int last_call_id_;
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_DEVTOOLS_DEVTOOLS_AGENT_HOST_IMPL_H_