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 | #pragma once |
27 | |
28 | #include "UserContentProvider.h" |
29 | #include "UserScriptTypes.h" |
30 | #include "UserStyleSheetTypes.h" |
31 | |
32 | namespace WebCore { |
33 | |
34 | class UserContentController final : public UserContentProvider { |
35 | public: |
36 | WEBCORE_EXPORT static Ref<UserContentController> create(); |
37 | WEBCORE_EXPORT ~UserContentController(); |
38 | |
39 | WEBCORE_EXPORT void addUserScript(DOMWrapperWorld&, std::unique_ptr<UserScript>); |
40 | WEBCORE_EXPORT void removeUserScript(DOMWrapperWorld&, const URL&); |
41 | WEBCORE_EXPORT void removeUserScripts(DOMWrapperWorld&); |
42 | |
43 | WEBCORE_EXPORT void addUserStyleSheet(DOMWrapperWorld&, std::unique_ptr<UserStyleSheet>, UserStyleInjectionTime); |
44 | WEBCORE_EXPORT void removeUserStyleSheet(DOMWrapperWorld&, const URL&); |
45 | WEBCORE_EXPORT void removeUserStyleSheets(DOMWrapperWorld&); |
46 | |
47 | WEBCORE_EXPORT void removeAllUserContent(); |
48 | |
49 | private: |
50 | UserContentController(); |
51 | |
52 | // UserContentProvider |
53 | void forEachUserScript(Function<void(DOMWrapperWorld&, const UserScript&)>&&) const final; |
54 | void forEachUserStyleSheet(Function<void(const UserStyleSheet&)>&&) const final; |
55 | #if ENABLE(USER_MESSAGE_HANDLERS) |
56 | void forEachUserMessageHandler(Function<void(const UserMessageHandlerDescriptor&)>&&) const final; |
57 | #endif |
58 | #if ENABLE(CONTENT_EXTENSIONS) |
59 | ContentExtensions::ContentExtensionsBackend& userContentExtensionBackend() override { return m_contentExtensionBackend; } |
60 | #endif |
61 | |
62 | UserScriptMap m_userScripts; |
63 | UserStyleSheetMap m_userStyleSheets; |
64 | #if ENABLE(CONTENT_EXTENSIONS) |
65 | ContentExtensions::ContentExtensionsBackend m_contentExtensionBackend; |
66 | #endif |
67 | }; |
68 | |
69 | } // namespace WebCore |
70 | |