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

Source/core/dom/CustomElementRegistry.h

 /*
  * 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
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
@@ skipping 16 matching lines @@
  * (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 CustomElementRegistry_h
 #define CustomElementRegistry_h
 
 #include "bindings/v8/ScriptState.h"
 #include "core/dom/ContextDestructionObserver.h"
 #include "core/dom/CustomElementConstructor.h"
+#include "core/dom/CustomElementUpgradeCandidateMap.h"
 #include "core/dom/ExceptionCode.h"
 #include "core/dom/QualifiedName.h"
-#include <wtf/HashSet.h>
-#include <wtf/ListHashSet.h>
-#include <wtf/PassRefPtr.h>
-#include <wtf/RefCounted.h>
-#include <wtf/RefPtr.h>
-#include <wtf/Vector.h>
-#include <wtf/text/AtomicString.h>
-#include <wtf/text/AtomicStringHash.h>
+#include "wtf/HashSet.h"
+#include "wtf/ListHashSet.h"
+#include "wtf/PassRefPtr.h"
+#include "wtf/RefCounted.h"
+#include "wtf/RefPtr.h"
+#include "wtf/Vector.h"
+#include "wtf/text/AtomicString.h"
+#include "wtf/text/AtomicStringHash.h"
 
 namespace WebCore {
 
 class CustomElementDefinition;
 class Dictionary;
 class Document;
 class Element;
 
 class CustomElementInvocation {
 public:
@@ skipping 23 matching lines @@
     PassRefPtr<CustomElementConstructor> registerElement(WebCore::ScriptState*, const AtomicString& name, const Dictionary& options, ExceptionCode&);
 
     bool isUnresolved(Element*) const;
     PassRefPtr<CustomElementDefinition> findFor(Element*) const;
     PassRefPtr<CustomElementDefinition> findAndCheckNamespace(const AtomicString& type, const AtomicString& namespaceURI) const;
 
     PassRefPtr<Element> createCustomTagElement(const QualifiedName& localName);
 
     Document* document() const;
 
-    void didGiveTypeExtension(Element*);
+    void didGiveTypeExtension(Element*, const AtomicString&);
     void customElementWasDestroyed(Element*);
 
     static bool isCustomTagName(const AtomicString& name) { return isValidName(name); }
 
     static void deliverAllLifecycleCallbacks();
     static void deliverAllLifecycleCallbacksIfNeeded();
 
 private:
     typedef HashMap<AtomicString, RefPtr<CustomElementDefinition> > DefinitionMap;
-    typedef HashSet<AtomicString> NameSet;
     typedef ListHashSet<CustomElementRegistry*> InstanceSet;
-    typedef HashSet<Element*> ElementSet;
 
     static bool isValidName(const AtomicString&);
 
     static InstanceSet& activeCustomElementRegistries();
     void activate(const CustomElementInvocation&);
     void deactivate();
     void deliverLifecycleCallbacks();
 
     void didCreateCustomTagElement(Element*);
-    void didCreateUnresolvedElement(Element*);
+    void didCreateUnresolvedElement(CustomElementDefinition::CustomElementKind, const AtomicString& type, Element*);
 
     DefinitionMap m_definitions;
+    CustomElementUpgradeCandidateMap m_candidates;
+
     Vector<CustomElementInvocation> m_invocations;
-    ElementSet m_unresolvedElements;
 };
 
 inline void CustomElementRegistry::deliverAllLifecycleCallbacksIfNeeded()
 {
     if (!activeCustomElementRegistries().isEmpty())
         deliverAllLifecycleCallbacks();
     ASSERT(activeCustomElementRegistries().isEmpty());
 }
 
 inline CustomElementRegistry::InstanceSet& CustomElementRegistry::activeCustomElementRegistries()
 {
     DEFINE_STATIC_LOCAL(InstanceSet, activeInstances, ());
     return activeInstances;
 }
 
 
 } // namespace WebCore
 
 #endif