Upgrade elements that are created before a custom element definition is registered

Source/bindings/v8/CustomElementHelpers.cpp

 /*
  * 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:
  *
  *     * 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 17 matching lines @@
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "config.h"
 
 #include "bindings/v8/CustomElementHelpers.h"
 
 #include "HTMLNames.h"
 #include "V8CustomElementConstructor.h"
 #include "V8HTMLElementWrapperFactory.h"
+#include "bindings/v8/DOMDataStore.h"
 #include "bindings/v8/DOMWrapperWorld.h"
 #include "bindings/v8/ScriptController.h"
 #include "core/dom/CustomElementRegistry.h"
 #include "core/dom/Node.h"
 #include "core/html/HTMLElement.h"
 #include "core/html/HTMLUnknownElement.h"
 
 #if ENABLE(SVG)
 #include "V8SVGElementWrapperFactory.h"
 #include "SVGNames.h"
@@ skipping 191 matching lines @@
         return 0;
     if (const QualifiedName* htmlName = findHTMLTagNameOfV8Type(type))
         return htmlName;
 #if ENABLE(SVG)
     if (const QualifiedName* svgName = findSVGTagNameOfV8Type(type))
         return svgName;
 #endif
     return 0;
 }
 
+void CustomElementHelpers::upgradeWrappers(ScriptExecutionContext* executionContext, const HashSet<Element*>& elements, const ScriptValue& prototype)
+{
+    if (elements.isEmpty())
+        return;
+
+    v8::HandleScope handleScope;
+    v8::Handle<v8::Context> context = toV8Context(executionContext, mainThreadNormalWorld());
+    v8::Context::Scope scope(context);
+
+    v8::Handle<v8::Value> v8Prototype = prototype.v8Value();
+
+    for (HashSet<Element*>::const_iterator it = elements.begin(); it != elements.end(); ++it) {
+        v8::Handle<v8::Object> wrapper = DOMDataStore::getWrapperForMainWorld(*it);
+        if (wrapper.IsEmpty()) {
+            // The wrapper will be created with the right prototype when
+            // retrieved; we don't need to eagerly create the wrapper.
+            continue;
+        }
+        wrapper->SetPrototype(v8Prototype);
+    }
+}
+
 void CustomElementHelpers::invokeReadyCallbackIfNeeded(Element* element, v8::Handle<v8::Context> context)
 {
     v8::Handle<v8::Value> wrapperValue = toV8(element, context->Global(), context->GetIsolate());
     if (wrapperValue.IsEmpty() || !wrapperValue->IsObject())
         return;
     v8::Handle<v8::Object> wrapper = v8::Handle<v8::Object>::Cast(wrapperValue);
     v8::Handle<v8::Value> prototypeValue = wrapper->GetPrototype();
     if (prototypeValue.IsEmpty() || !prototypeValue->IsObject())
         return;
     v8::Handle<v8::Object> prototype = v8::Handle<v8::Object>::Cast(prototypeValue);
@@ skipping 19 matching lines @@
         return;
     v8::Context::Scope scope(context);
 
     for (size_t i = 0; i < invocations.size(); ++i) {
         ASSERT(executionContext == invocations[i].element()->document());
         invokeReadyCallbackIfNeeded(invocations[i].element(), context);
     }
 }
 
 } // namespace WebCore