Add a WebDocument::watchCssSelectors(selectors)

Source/core/dom/Document.cpp

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
  *           (C) 2001 Dirk Mueller (mueller@kde.org)
  *           (C) 2006 Alexey Proskuryakov (ap@webkit.org)
  * Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009, 2011, 2012 Apple Inc. All rights reserved.
  * Copyright (C) 2008, 2009 Torch Mobile Inc. All rights reserved. (http://www.torchmobile.com/)
  * Copyright (C) 2008, 2009, 2011, 2012 Google Inc. All rights reserved.
  * Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies)
  * Copyright (C) Research In Motion Limited 2010-2011. All rights reserved.
@@ skipping 22 matching lines @@
 #include "RuntimeEnabledFeatures.h"
 #include "SVGElementFactory.h"
 #include "SVGNames.h"
 #include "XMLNSNames.h"
 #include "XMLNames.h"
 #include "bindings/v8/CustomElementConstructorBuilder.h"
 #include "bindings/v8/Dictionary.h"
 #include "bindings/v8/ScriptController.h"
 #include "core/accessibility/AXObjectCache.h"
 #include "core/animation/DocumentTimeline.h"
+#include "core/css/CSSParser.h"
 #include "core/css/CSSStyleDeclaration.h"
 #include "core/css/CSSStyleSheet.h"
 #include "core/css/FontLoader.h"
 #include "core/css/MediaQueryMatcher.h"
 #include "core/css/StylePropertySet.h"
 #include "core/css/StyleSheetContents.h"
 #include "core/css/StyleSheetList.h"
 #include "core/css/resolver/StyleResolver.h"
 #include "core/dom/Attr.h"
 #include "core/dom/CDATASection.h"
@@ skipping 61 matching lines @@
 #include "core/html/PluginDocument.h"
 #include "core/html/parser/HTMLDocumentParser.h"
 #include "core/html/parser/HTMLParserIdioms.h"
 #include "core/html/parser/NestingLevelIncrementer.h"
 #include "core/inspector/InspectorCounters.h"
 #include "core/inspector/InspectorInstrumentation.h"
 #include "core/inspector/ScriptCallStack.h"
 #include "core/loader/CookieJar.h"
 #include "core/loader/DocumentLoader.h"
 #include "core/loader/FrameLoader.h"
+#include "core/loader/FrameLoaderClient.h"
 #include "core/loader/ImageLoader.h"
 #include "core/loader/Prerenderer.h"
 #include "core/loader/TextResourceDecoder.h"
 #include "core/loader/cache/CachedResourceLoader.h"
 #include "core/page/Chrome.h"
 #include "core/page/ChromeClient.h"
 #include "core/page/ContentSecurityPolicy.h"
 #include "core/page/DOMSecurityPolicy.h"
 #include "core/page/DOMWindow.h"
 #include "core/page/EventHandler.h"
@@ skipping 305 matching lines @@
     , m_directionSetOnDocumentElement(false)
     , m_writingModeSetOnDocumentElement(false)
     , m_writeRecursionIsTooDeep(false)
     , m_writeRecursionDepth(0)
     , m_wheelEventHandlerCount(0)
     , m_lastHandledUserGestureTimestamp(0)
     , m_prerenderer(Prerenderer::create(this))
     , m_textAutosizer(TextAutosizer::create(this))
     , m_pendingTasksTimer(this, &Document::pendingTasksTimerFired)
     , m_scheduledTasksAreSuspended(false)
+    , m_callbackSelectorChangeTimer(this, &Document::callbackSelectorChangeTimerFired)
     , m_sharedObjectPoolClearTimer(this, &Document::sharedObjectPoolClearTimerFired)
 #ifndef NDEBUG
     , m_didDispatchViewportPropertiesChanged(false)
 #endif
     , m_timeline(DocumentTimeline::create(this))
     , m_templateDocumentHost(0)
     , m_fontloader(0)
     , m_didAssociateFormControlsTimer(this, &Document::didAssociateFormControlsTimerFired)
 {
     ScriptWrappable::init(this);
@@ skipping 1384 matching lines @@
         matchAuthorAndUserStyles = docSettings->authorAndUserStylesEnabled();
     m_styleResolver = adoptPtr(new StyleResolver(this, matchAuthorAndUserStyles));
     m_styleSheetCollection->combineCSSFeatureFlags();
 }
 
 void Document::clearStyleResolver()
 {
     m_styleResolver.clear();
 }
 
+#if ENABLE(CSS_CALLBACKS)
+void Document::callbackSelectorChangeTimerFired(Timer<Document>*)
+{
+    if (m_addedSelectors.isEmpty() && m_removedSelectors.isEmpty())
+        return;
+    if (frame()) {
+        Vector<String> addedSelectors, removedSelectors;
+        copyToVector(m_addedSelectors, addedSelectors);
+        copyToVector(m_removedSelectors, removedSelectors);
+        frame()->loader()->client()->selectorMatchChanged(addedSelectors, removedSelectors);
+    }
+    m_addedSelectors.clear();
+    m_removedSelectors.clear();
+}
+
+void Document::addSelectorMatch(const String& selector, StyleRareNonInheritedData* data)
+{
+    // Set m_document here to avoid exposing the weakFactory outside of Document.
+    data->m_document = m_weakFactory.createWeakPtr();
+
+    HashMap<String, int>::iterator count = m_matchingCallbackSelectors.find(selector);
+    if (count != m_matchingCallbackSelectors.end()) {
+        ++count->value;
+        return;
+    }
+    if (!m_callbackSelectorChangeTimer.isActive())
+        m_callbackSelectorChangeTimer.startOneShot(0);
+
+    m_matchingCallbackSelectors.set(selector, 1);
+    if (m_removedSelectors.contains(selector))
+        m_removedSelectors.remove(selector);
+    else
+        m_addedSelectors.add(selector);
+}
+
+void Document::removeSelectorMatch(const String& selector)
+{
+    HashMap<String, int>::iterator count = m_matchingCallbackSelectors.find(selector);
+    if (count == m_matchingCallbackSelectors.end())
+        return;
+    --count->value;
+    if (!count->value) {
+        if (!m_callbackSelectorChangeTimer.isActive())
+            m_callbackSelectorChangeTimer.startOneShot(0);
+
+        m_matchingCallbackSelectors.remove(count);
+        if (m_addedSelectors.contains(selector))
+            m_addedSelectors.remove(selector);
+        else
+            m_removedSelectors.add(selector);
+    }
+}
+#endif // ENABLE(CSS_CALLBACKS)
+
 void Document::attach(const AttachContext& context)
 {
     ASSERT(!attached());
     ASSERT(!m_axObjectCache || this != topDocument());
 
     if (!m_renderArena)
         m_renderArena = RenderArena::create();
 
     // Create the rendering tree
     setRenderer(new (m_renderArena.get()) RenderView(this));
@@ skipping 1134 matching lines @@
     // Seamless child frames are expected to notify their seamless children recursively, so we only do direct children.
     for (Frame* child = frame()->tree()->firstChild(); child; child = child->tree()->nextSibling()) {
         Document* childDocument = child->document();
         if (childDocument->shouldDisplaySeamlesslyWithParent()) {
             ASSERT(childDocument->seamlessParentIFrame()->document() == this);
             childDocument->seamlessParentUpdatedStylesheets();
         }
     }
 }
 
+void Document::watchCssSelectors(const Vector<String>& selectors)
+{
+    m_watchedCallbackSelectors.clear();
+    CSSParserContext context(UASheetMode);
+    CSSParser parser(context);
+
+    const CSSProperty callbackProperty(CSSPropertyInternalCallback, CSSPrimitiveValue::createIdentifier(CSSValueInternalPresence));
+    const RefPtr<StylePropertySet> callbackPropertySet = ImmutableStylePropertySet::create(&callbackProperty, 1, UASheetMode);
+
+    CSSSelectorList selectorList;
+    for (unsigned i = 0; i < selectors.size(); ++i) {
+        RefPtr<StyleRule> rule = StyleRule::create(i);
+
+        parser.parseSelector(selectors[i], selectorList);
+        rule->wrapperAdoptSelectorList(selectorList);
+
+        rule->setProperties(callbackPropertySet);
+
+        m_watchedCallbackSelectors.append(rule.release());
+    }
+    styleResolverChanged(RecalcStyleImmediately, FullStyleUpdate);
+}
+
 void Document::setHoverNode(PassRefPtr<Node> newHoverNode)
 {
     m_hoverNode = newHoverNode;
 }
 
 void Document::setActiveElement(PassRefPtr<Element> newActiveElement)
 {
     if (!newActiveElement) {
         m_activeElement.clear();
         return;
@@ skipping 2120 matching lines @@
 {
     return DocumentLifecycleNotifier::create(this);
 }
 
 DocumentLifecycleNotifier* Document::lifecycleNotifier()
 {
     return static_cast<DocumentLifecycleNotifier*>(ScriptExecutionContext::lifecycleNotifier());
 }
 
 } // namespace WebCore