026bbe2fd5ae3e0cb54ca03ffd7b59d97dce760e
[WebKit-https.git] / Source / WebKit2 / UIProcess / UserContent / WebUserContentControllerProxy.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 WebUserContentControllerProxy_h
27 #define WebUserContentControllerProxy_h
28
29 #include "MessageReceiver.h"
30 #include <wtf/Forward.h>
31 #include <wtf/HashMap.h>
32 #include <wtf/HashSet.h>
33 #include <wtf/PassRefPtr.h>
34 #include <wtf/RefCounted.h>
35 #include <wtf/Vector.h>
36
37 namespace IPC {
38 class DataReference;
39 }
40
41 namespace WebCore {
42 class UserScript;
43 class UserStyleSheet;
44 }
45
46 namespace WebKit {
47
48 class WebProcessProxy;
49 class WebScriptMessageHandler;
50
51 class WebUserContentControllerProxy : public RefCounted<WebUserContentControllerProxy>, private IPC::MessageReceiver {
52 public:
53     static PassRefPtr<WebUserContentControllerProxy> create();
54     ~WebUserContentControllerProxy();
55
56     uint64_t identifier() const { return m_identifier; }
57
58     void addProcess(WebProcessProxy&);
59     void removeProcess(WebProcessProxy&);
60
61     void addUserScript(WebCore::UserScript);
62     void removeAllUserScripts();
63
64     void addUserStyleSheet(WebCore::UserStyleSheet);
65     void removeAllUserStyleSheets();
66
67     // Returns false if there was a name conflict.
68     bool addUserScriptMessageHandler(WebScriptMessageHandler*);
69     void removeUserMessageHandlerForName(const String&);
70
71 private:
72     explicit WebUserContentControllerProxy();
73
74     // IPC::MessageReceiver.
75     virtual void didReceiveMessage(IPC::Connection*, IPC::MessageDecoder&) override;
76
77     void didPostMessage(IPC::Connection&, uint64_t pageID, uint64_t frameID, uint64_t messageHandlerID, const IPC::DataReference&);
78
79     uint64_t m_identifier;
80     HashSet<WebProcessProxy*> m_processes;
81
82     Vector<WebCore::UserScript> m_userScripts;
83     Vector<WebCore::UserStyleSheet> m_userStyleSheets;
84     HashMap<uint64_t, RefPtr<WebScriptMessageHandler>> m_scriptMessageHandlers;
85 };
86
87 } // namespace WebKit
88
89 #endif // WebUserContentControllerProxy_h