Merge DevTools Refactor CL to Blink36

Source/core/inspector/InjectedScriptManager.cpp

 /*
  * Copyright (C) 2007, 2008 Apple Inc. All rights reserved.
  * Copyright (C) 2008 Matt Lilek <webkit@mattlilek.com>
  * Copyright (C) 2012 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:
  *
  * 1.  Redistributions of source code must retain the above copyright
@@ skipping 14 matching lines @@
  * LOSS OF USE, 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.
  */
 
 #include "config.h"
 #include "core/inspector/InjectedScriptManager.h"
 
 #include "InjectedScriptSource.h"
+#include "bindings/dart/DartInjectedScript.h"
+#include "bindings/dart/DartScriptState.h"
 #include "bindings/v8/ScriptObject.h"
 #include "core/inspector/InjectedScript.h"
 #include "core/inspector/InjectedScriptHost.h"
 #include "core/inspector/JSONParser.h"
 #include "platform/JSONValues.h"
 #include "wtf/PassOwnPtr.h"
 
 namespace WebCore {
 
 PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForPage()
 {
     return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWindow));
 }
 
 PassOwnPtr<InjectedScriptManager> InjectedScriptManager::createForWorker()
 {
     return adoptPtr(new InjectedScriptManager(&InjectedScriptManager::canAccessInspectedWorkerGlobalScope));
 }
 
 InjectedScriptManager::InjectedScriptManager(InspectedStateAccessCheck accessCheck)
     : m_nextInjectedScriptId(1)
     , m_injectedScriptHost(InjectedScriptHost::create())
     , m_inspectedStateAccessCheck(accessCheck)
+    , m_placeholderInjectedScript(0)
 {
 }
 
 InjectedScriptManager::~InjectedScriptManager()
 {
+    delete m_placeholderInjectedScript;
 }
 
 void InjectedScriptManager::disconnect()
 {
     m_injectedScriptHost->disconnect();
     m_injectedScriptHost.clear();
 }
 
+InjectedScript& InjectedScriptManager::placeholderInjectedScript()
+{
+    if (!m_placeholderInjectedScript)
+        m_placeholderInjectedScript = new V8InjectedScript();
+    return *m_placeholderInjectedScript;
+}
+
 InjectedScriptHost* InjectedScriptManager::injectedScriptHost()
 {
     return m_injectedScriptHost.get();
 }
 
-InjectedScript InjectedScriptManager::injectedScriptForId(int id)
+InjectedScript& InjectedScriptManager::injectedScriptForId(int id)
 {
     IdToInjectedScriptMap::iterator it = m_idToInjectedScript.find(id);
     if (it != m_idToInjectedScript.end())
-        return it->value;
+        return *(it->value);
     for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
         if (it->value == id)
             return injectedScriptFor(it->key.get());
     }
-    return InjectedScript();
+    return placeholderInjectedScript();
 }
 
 int InjectedScriptManager::injectedScriptIdFor(ScriptState* scriptState)
 {
     ScriptStateToId::iterator it = m_scriptStateToId.find(scriptState);
     if (it != m_scriptStateToId.end())
         return it->value;
     int id = m_nextInjectedScriptId++;
     m_scriptStateToId.set(scriptState, id);
     return id;
 }
 
-InjectedScript InjectedScriptManager::injectedScriptForObjectId(const String& objectId)
+InjectedScript& InjectedScriptManager::injectedScriptForObjectId(const String& objectId)
 {
     RefPtr<JSONValue> parsedObjectId = parseJSON(objectId);
     if (parsedObjectId && parsedObjectId->type() == JSONValue::TypeObject) {
         long injectedScriptId = 0;
         bool success = parsedObjectId->asObject()->getNumber("injectedScriptId", &injectedScriptId);
-        if (success)
-            return m_idToInjectedScript.get(injectedScriptId);
+        if (success) {
+            IdToInjectedScriptMap::iterator s = m_idToInjectedScript.find(injectedScriptId);
+            if (s != m_idToInjectedScript.end())
+                return *(s->value);
+        }
     }
-    return InjectedScript();
+    return placeholderInjectedScript();
 }
 
 void InjectedScriptManager::discardInjectedScripts()
 {
     m_idToInjectedScript.clear();
     m_scriptStateToId.clear();
 }
 
 void InjectedScriptManager::discardInjectedScriptsFor(DOMWindow* window)
 {
     if (m_scriptStateToId.isEmpty())
         return;
 
     Vector<long> idsToRemove;
     IdToInjectedScriptMap::iterator end = m_idToInjectedScript.end();
     for (IdToInjectedScriptMap::iterator it = m_idToInjectedScript.begin(); it != end; ++it) {
-        ScriptState* scriptState = it->value.scriptState();
+        ScriptState* scriptState = it->value->scriptState();
         if (window != scriptState->domWindow())
             continue;
         m_scriptStateToId.remove(scriptState);
         idsToRemove.append(it->key);
     }
+    for (size_t i = 0; i < idsToRemove.size(); i++)
+        delete m_idToInjectedScript.get(idsToRemove[i]);
     m_idToInjectedScript.removeAll(idsToRemove);
 
+
     // Now remove script states that have id but no injected script.
     Vector<ScriptState*> scriptStatesToRemove;
     for (ScriptStateToId::iterator it = m_scriptStateToId.begin(); it != m_scriptStateToId.end(); ++it) {
         ScriptState* scriptState = it->key.get();
         if (window == scriptState->domWindow())
             scriptStatesToRemove.append(scriptState);
     }
     m_scriptStateToId.removeAll(scriptStatesToRemove);
 }
 
 bool InjectedScriptManager::canAccessInspectedWorkerGlobalScope(ScriptState*)
 {
     return true;
 }
 
 void InjectedScriptManager::releaseObjectGroup(const String& objectGroup)
 {
     Vector<int> keys;
     keys.appendRange(m_idToInjectedScript.keys().begin(), m_idToInjectedScript.keys().end());
     for (Vector<int>::iterator k = keys.begin(); k != keys.end(); ++k) {
         IdToInjectedScriptMap::iterator s = m_idToInjectedScript.find(*k);
         if (s != m_idToInjectedScript.end())
-            s->value.releaseObjectGroup(objectGroup); // m_idToInjectedScript may change here.
+            s->value->releaseObjectGroup(objectGroup); // m_idToInjectedScript may change here.
     }
 }
 
 String InjectedScriptManager::injectedScriptSource()
 {
     return String(reinterpret_cast<const char*>(InjectedScriptSource_js), sizeof(InjectedScriptSource_js));
 }
 
-InjectedScript InjectedScriptManager::injectedScriptFor(ScriptState* inspectedScriptState)
+InjectedScript& InjectedScriptManager::injectedScriptFor(ScriptState* inspectedScriptState)
 {
     ScriptStateToId::iterator it = m_scriptStateToId.find(inspectedScriptState);
     if (it != m_scriptStateToId.end()) {
         IdToInjectedScriptMap::iterator it1 = m_idToInjectedScript.find(it->value);
         if (it1 != m_idToInjectedScript.end())
-            return it1->value;
+            return *(it1->value);
     }
 
     if (!m_inspectedStateAccessCheck(inspectedScriptState))
-        return InjectedScript();
+        return placeholderInjectedScript();
 
     int id = injectedScriptIdFor(inspectedScriptState);
-    ScriptObject injectedScriptObject = createInjectedScript(injectedScriptSource(), inspectedScriptState, id);
-    InjectedScript result(injectedScriptObject, m_inspectedStateAccessCheck);
+
+    InjectedScript* result;
+    if (inspectedScriptState->isJavaScript()) {
+        ScriptObject injectedScriptObject = createInjectedScript(injectedScriptSource(), inspectedScriptState, id);
+        result = new V8InjectedScript(injectedScriptObject, m_inspectedStateAccessCheck);
+    } else {
+        result = new DartInjectedScript(static_cast<DartScriptState*>(inspectedScriptState), m_inspectedStateAccessCheck, id, injectedScriptHost());
+    }
     m_idToInjectedScript.set(id, result);
-    return result;
+    return *result;
 }
 
 } // namespace WebCore