[import] support document.stylesheets in imported documents.

Source/core/dom/DocumentStyleSheetCollector.h

 /*
  * Copyright (C) 2014 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.
  *     * Neither the name of Google Inc. nor the names of its
  * contributors may be used to endorse or promote products derived from
  * this software without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 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.
  */
 
 #ifndef DocumentStyleSheetCollector_h
 #define DocumentStyleSheetCollector_h
 
-#include "core/dom/StyleSheetCollection.h"
-#include "wtf/HashSet.h"
+#include "wtf/RefPtr.h"
+#include "wtf/Vector.h"
 
 namespace WebCore {
 
-class DocumentStyleSheetCollector FINAL {
+class CSSStyleSheet;
+class StyleSheetCollectionBase;
+class StyleSheet;
+
+class DocumentStyleSheetCollector {
 public:
-    DocumentStyleSheetCollector(TreeScope& root)
-        : m_root(root)
-    { }
+    friend class ImportedDocumentStyleSheetCollector;
 
-    bool isCollectingForList(TreeScope&) const;
-    void setCollectionTo(StyleSheetCollectionBase&);
+    DocumentStyleSheetCollector(Vector<RefPtr<StyleSheet> >& sheetsForList, Vector<RefPtr<CSSStyleSheet> >& activeList);
+    ~DocumentStyleSheetCollector();
 
-    void appendActiveStyleSheets(const Vector<RefPtr<CSSStyleSheet> >& sheets)  { m_collection.appendActiveStyleSheets(sheets); }
-    void appendActiveStyleSheet(CSSStyleSheet* sheet) { m_collection.appendActiveStyleSheet(sheet); }
-    void appendSheetForList(StyleSheet* sheet) { m_collection.appendSheetForList(sheet); }
-    StyleSheetCollectionBase& collection() { return m_collection; }
-    void markVisited(Document* document) { m_documentsVisited.add(document); }
-    bool hasVisited(Document* document) const { return m_documentsVisited.contains(document); }
+    void appendActiveStyleSheets(const Vector<RefPtr<CSSStyleSheet> >&);
+    void appendActiveStyleSheet(CSSStyleSheet*);
+    void appendSheetForList(StyleSheet*);
 
 private:
-    TreeScope& m_root;
-    StyleSheetCollectionBase m_collection;
-    HashSet<Document*> m_documentsVisited;
+    Vector<RefPtr<StyleSheet> >& m_styleSheetsForStyleSheetList;
+    Vector<RefPtr<CSSStyleSheet> >& m_activeAuthorStyleSheets;
+};
+
+class ActiveDocumentStyleSheetCollector FINAL : public DocumentStyleSheetCollector {
+public:
+    ActiveDocumentStyleSheetCollector(StyleSheetCollectionBase&);
+};
+
+class ImportedDocumentStyleSheetCollector FINAL : public DocumentStyleSheetCollector {
+public:
+    ImportedDocumentStyleSheetCollector(DocumentStyleSheetCollector&, Vector<RefPtr<StyleSheet> >&);
 };
 
 } // namespace WebCore
 
 #endif // DocumentStyleSheetCollector_h