Prepare to use CSS selectors in content extensions.
[WebKit-https.git] / Source / WebCore / page / UserContentController.h
1 /*
2  * Copyright (C) 2014 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #ifndef UserContentController_h
27 #define UserContentController_h
28
29 #include "UserScriptTypes.h"
30 #include "UserStyleSheetTypes.h"
31 #include <wtf/HashSet.h>
32 #include <wtf/RefCounted.h>
33 #include <wtf/RefPtr.h>
34
35 #if ENABLE(USER_MESSAGE_HANDLERS)
36 #include "UserMessageHandlerDescriptorTypes.h"
37 #endif
38
39 namespace WebCore {
40
41 class DOMWrapperWorld;
42 class Page;
43 class URL;
44 class UserScript;
45 class UserStyleSheet;
46 class UserMessageHandlerDescriptor;
47
48 namespace ContentExtensions {
49 class ContentExtensionsBackend;
50 struct Action;
51 }
52
53 class UserContentController : public RefCounted<UserContentController> {
54 public:
55     WEBCORE_EXPORT static RefPtr<UserContentController> create();
56     WEBCORE_EXPORT ~UserContentController();
57
58     void addPage(Page&);
59     void removePage(Page&);
60
61     const UserScriptMap* userScripts() const { return m_userScripts.get(); }
62
63     WEBCORE_EXPORT void addUserScript(DOMWrapperWorld&, std::unique_ptr<UserScript>);
64     WEBCORE_EXPORT void removeUserScript(DOMWrapperWorld&, const URL&);
65     WEBCORE_EXPORT void removeUserScripts(DOMWrapperWorld&);
66
67     const UserStyleSheetMap* userStyleSheets() const { return m_userStyleSheets.get(); }
68
69     WEBCORE_EXPORT void addUserStyleSheet(DOMWrapperWorld&, std::unique_ptr<UserStyleSheet>, UserStyleInjectionTime);
70     WEBCORE_EXPORT void removeUserStyleSheet(DOMWrapperWorld&, const URL&);
71     WEBCORE_EXPORT void removeUserStyleSheets(DOMWrapperWorld&);
72
73     WEBCORE_EXPORT void removeAllUserContent();
74
75 #if ENABLE(USER_MESSAGE_HANDLERS)
76     const UserMessageHandlerDescriptorMap* userMessageHandlerDescriptors() const { return m_userMessageHandlerDescriptors.get(); }
77
78     WEBCORE_EXPORT void addUserMessageHandlerDescriptor(UserMessageHandlerDescriptor&);
79     WEBCORE_EXPORT void removeUserMessageHandlerDescriptor(UserMessageHandlerDescriptor&);
80 #endif
81
82 #if ENABLE(CONTENT_EXTENSIONS)
83     // FIXME: This should really take a pointer to a compiled UserContentFilter.
84     WEBCORE_EXPORT void addUserContentFilter(const String& name, const String& ruleList);
85     WEBCORE_EXPORT void removeUserContentFilter(const String& name);
86     WEBCORE_EXPORT void removeAllUserContentFilters();
87     
88     // FIXME: Consider putting this (and other future content filter predicates) in its own class.
89     Vector<ContentExtensions::Action> actionsForURL(const URL&);
90 #endif
91
92 private:
93     UserContentController();
94
95     void invalidateInjectedStyleSheetCacheInAllFrames();
96
97     HashSet<Page*> m_pages;
98
99     std::unique_ptr<UserScriptMap> m_userScripts;
100     std::unique_ptr<UserStyleSheetMap> m_userStyleSheets;
101 #if ENABLE(USER_MESSAGE_HANDLERS)
102     std::unique_ptr<UserMessageHandlerDescriptorMap> m_userMessageHandlerDescriptors;
103 #endif
104 #if ENABLE(CONTENT_EXTENSIONS)
105     std::unique_ptr<ContentExtensions::ContentExtensionsBackend> m_contentExtensionBackend;
106 #endif
107 };
108
109 } // namespace WebCore
110
111 #endif // UserContentController_h