Upgrade in-document custom elements when an element is defined.

third_party/WebKit/Source/core/dom/custom/CustomElementsRegistry.h

Index: third_party/WebKit/Source/core/dom/custom/CustomElementsRegistry.h
diff --git a/third_party/WebKit/Source/core/dom/custom/CustomElementsRegistry.h b/third_party/WebKit/Source/core/dom/custom/CustomElementsRegistry.h
index 292f86bed06af745541b05496ea41a545bdde531..2221b45f4679ebc65425ec5e163faf672275cd06 100644
--- a/third_party/WebKit/Source/core/dom/custom/CustomElementsRegistry.h
+++ b/third_party/WebKit/Source/core/dom/custom/CustomElementsRegistry.h
@@ -5,6 +5,7 @@
 #ifndef CustomElementsRegistry_h
 #define CustomElementsRegistry_h
 
+#include "base/gtest_prod_util.h"
 #include "bindings/core/v8/ScriptWrappable.h"
 #include "core/CoreExport.h"
 #include "platform/heap/Handle.h"
@@ -16,6 +17,9 @@ namespace blink {
 
 class CustomElementDefinition;
 class CustomElementDefinitionBuilder;
+class CustomElementDescriptor;
+class Document;
+class Element;
 class ElementRegistrationOptions;
 class ExceptionState;
 class ScriptState;
@@ -28,8 +32,7 @@ class CORE_EXPORT CustomElementsRegistry final
     DEFINE_WRAPPERTYPEINFO();
     WTF_MAKE_NONCOPYABLE(CustomElementsRegistry);
 public:
-    static CustomElementsRegistry* create(
-        V0CustomElementRegistrationContext*);
+    static CustomElementsRegistry* create(Document*);

[yosin_UTC9, 2016/06/01 06:15:44]

(^_^)b

[dominicc (has gone to gerrit), 2016/06/01 23:30:57]

Yeah, I feel a bit nervous about adding document, window, etc. here but we do
need to retrieve the document to do upgrade.

I think windows may change documents (document.open), so we may need something
more dynamic here.

 
     void define(
         ScriptState*,
@@ -44,24 +47,39 @@ public:
         const ElementRegistrationOptions&,
         ExceptionState&);
 
-    bool nameIsDefined(const AtomicString& name) const
-    {
-        return m_definitions.contains(name);
-    }
+    bool nameIsDefined(const AtomicString& name) const;
 
     CustomElementDefinition* definitionForName(const AtomicString& name) const;
 
+    // TODO(dominicc): Consider broadening this API when type extensions are
+    // implemented.
+    void addCandidate(Element*);
+
     DECLARE_TRACE();
 
 private:
-    CustomElementsRegistry(const V0CustomElementRegistrationContext*);
-    bool v0NameIsDefined(const AtomicString&) const;
+    friend class CustomElementsRegistryTestBase;
+
+    CustomElementsRegistry(Document*);
+
+    V0CustomElementRegistrationContext* v0();
+    bool v0NameIsDefined(const AtomicString& name);
+
+    void collectCandidates(
+        const CustomElementDescriptor&,
+        HeapVector<Member<Element>>*);
 
     using DefinitionMap =
         HeapHashMap<AtomicString, Member<CustomElementDefinition>>;
     DefinitionMap m_definitions;
 
-    Member<const V0CustomElementRegistrationContext> m_v0;
+    Member<Document> m_document;
+
+    using UpgradeCandidateSet = HeapHashSet<WeakMember<Element>>;
+    using UpgradeCandidateMap = HeapHashMap<
+        AtomicString,
+        Member<UpgradeCandidateSet>>;
+    Member<UpgradeCandidateMap> m_upgradeCandidates;
 };
 
 } // namespace blink