Inspector: wrapped inspector agent instances into factory-like wrappers

Source/core/inspector/InspectorBaseAgent.h

 /*
  * Copyright (C) 2011 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 14 matching lines @@
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #ifndef InspectorBaseAgent_h
 #define InspectorBaseAgent_h
 
 #include "InspectorBackendDispatcher.h"
-#include <wtf/Forward.h>
-#include <wtf/Vector.h>
-#include <wtf/text/WTFString.h>
+#include "wtf/Forward.h"
+#include "wtf/PassOwnPtr.h"
+#include "wtf/RefCounted.h"
+#include "wtf/Vector.h"
+#include "wtf/text/WTFString.h"
 
 namespace WebCore {
 
 class InspectorFrontend;
 class InspectorCompositeState;
 class InspectorState;
 class InstrumentingAgents;
 
-class InspectorBaseAgentInterface {
+class InspectorBaseAgent : public RefCounted<InspectorBaseAgent> {
 public:
-    InspectorBaseAgentInterface(const String&, InstrumentingAgents*, InspectorCompositeState*);
-    virtual ~InspectorBaseAgentInterface();
+    InspectorBaseAgent(InstrumentingAgents*, InspectorState*);
+    virtual ~InspectorBaseAgent();
 
+    // FIXME(kaznacheev) Remove the below 4 methods once all agent instances are created on enable (see crbug.com/248092)
     virtual void setFrontend(InspectorFrontend*) { }
     virtual void clearFrontend() { }
     virtual void restore() { }
-    virtual void registerInDispatcher(InspectorBackendDispatcher*) = 0;
     virtual void discardAgent() { }
 
-    String name() { return m_name; }
-
     virtual void reportMemoryUsage(MemoryObjectInfo*) const;
 
+protected:
+    InstrumentingAgents* m_instrumentingAgents;
+    InspectorState* m_state;
+};
+
+class InspectorBaseFactoryInterface {
+public:
+
+    InspectorBaseFactoryInterface(const String& name, InstrumentingAgents*, InspectorCompositeState*);
+    virtual ~InspectorBaseFactoryInterface();
+
+    void setAgent(PassRefPtr<InspectorBaseAgent>);
+
+    virtual void setFrontend(InspectorFrontend*);
+    virtual void clearFrontend();
+    virtual void restore();
+    virtual void registerInDispatcher(InspectorBackendDispatcher*) = 0;
+    virtual void discardDependencies();
+
+    virtual void reportMemoryUsage(MemoryObjectInfo*) const;
+
 protected:
     InstrumentingAgents* m_instrumentingAgents;
     InspectorState* m_state;
 
+    RefPtr<InspectorBaseAgent> m_agent;
+
 private:
-    String m_name;
+    InspectorFrontend* m_frontend;
 };
 
-class InspectorAgentRegistry {
+class InspectorFactoryRegistry {
 public:
-    void append(PassOwnPtr<InspectorBaseAgentInterface>);
+    void append(PassOwnPtr<InspectorBaseFactoryInterface>);
 
     void setFrontend(InspectorFrontend*);
     void clearFrontend();
     void restore();
     void registerInDispatcher(InspectorBackendDispatcher*);
-    void discardAgents();
+    void discardDependencies();
 
     virtual void reportMemoryUsage(MemoryObjectInfo*) const;
 
 private:
-    Vector<OwnPtr<InspectorBaseAgentInterface> > m_agents;
+    Vector<OwnPtr<InspectorBaseFactoryInterface> > m_controllers;
 };
 
-template<typename T>
-class InspectorBaseAgent : public InspectorBaseAgentInterface {
+template<typename T, typename Agent>
+class InspectorBaseFactory : public InspectorBaseFactoryInterface, public Agent::Factory {
 public:
-    virtual ~InspectorBaseAgent() { }
-
+    // InspectorBaseFactoryInterface implementation
     virtual void registerInDispatcher(InspectorBackendDispatcher* dispatcher)
     {
         dispatcher->registerAgent(static_cast<T*>(this));
     }
 
+    Agent* agent() { return static_cast<Agent*>(m_agent.get()); }
+
+    // Agent::Factory implementation
+    virtual Agent* commandHandler() { return agent(); }
+
 protected:
-    InspectorBaseAgent(const String& name, InstrumentingAgents* instrumentingAgents, InspectorCompositeState* inspectorState)
-        : InspectorBaseAgentInterface(name, instrumentingAgents, inspectorState)
+    InspectorBaseFactory(const String& name, InstrumentingAgents* instrumentingAgents, InspectorCompositeState* compositeState)
+        : InspectorBaseFactoryInterface(name, instrumentingAgents, compositeState)
     {
     }
 };
 
 } // namespace WebCore
 
 #endif // !defined(InspectorBaseAgent_h)