Make all of the WebContext sub objects MessageReceivers
[WebKit-https.git] / Source / WebKit2 / UIProcess / WebIconDatabase.h
1 /*
2  * Copyright (C) 2011 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 WebIconDatabase_h
27 #define WebIconDatabase_h
28
29 #include "APIObject.h"
30
31 #include "Connection.h"
32 #include "WebIconDatabaseClient.h"
33 #include <WebCore/IconDatabaseClient.h>
34 #include <WebCore/ImageSource.h>
35 #include <WebCore/IntSize.h>
36 #include <wtf/Forward.h>
37 #include <wtf/PassRefPtr.h>
38 #include <wtf/RefPtr.h>
39 #include <wtf/Vector.h>
40 #include <wtf/text/StringHash.h>
41
42 namespace CoreIPC {
43 class ArgumentDecoder;
44 class DataReference;
45 class MessageID;
46 }
47
48 namespace WebCore {
49 class IconDatabase;
50 class Image;
51 }
52
53 namespace WebKit {
54
55 class WebContext;
56
57 class WebIconDatabase : public APIObject, public WebCore::IconDatabaseClient, private CoreIPC::MessageReceiver {
58 public:
59     static const Type APIType = TypeIconDatabase;
60
61     static PassRefPtr<WebIconDatabase> create(WebContext*);
62     virtual ~WebIconDatabase();
63
64     void invalidate();
65     void clearContext() { m_webContext = 0; }
66     void setDatabasePath(const String&);
67     void enableDatabaseCleanup();
68
69     void retainIconForPageURL(const String&);
70     void releaseIconForPageURL(const String&);
71     void setIconURLForPageURL(const String&, const String&);
72     void setIconDataForIconURL(const CoreIPC::DataReference&, const String&);
73     
74     void synchronousIconDataForPageURL(const String&, CoreIPC::DataReference&);
75     void synchronousIconURLForPageURL(const String&, String&);
76     void synchronousIconDataKnownForIconURL(const String&, bool&) const;
77     void synchronousLoadDecisionForIconURL(const String&, int&) const;
78     
79     void getLoadDecisionForIconURL(const String&, uint64_t callbackID);
80     void didReceiveIconForPageURL(const String&);
81
82     WebCore::Image* imageForPageURL(const String&, const WebCore::IntSize& iconSize = WebCore::IntSize(32, 32));
83     WebCore::NativeImagePtr nativeImageForPageURL(const String&, const WebCore::IntSize& iconSize = WebCore::IntSize(32, 32));
84     bool isOpen();
85     bool isUrlImportCompleted();
86
87     void removeAllIcons();
88     void checkIntegrityBeforeOpening();
89     void close();
90
91     void initializeIconDatabaseClient(const WKIconDatabaseClient*);
92
93 private:
94     WebIconDatabase(WebContext*);
95
96     virtual Type type() const { return APIType; }
97
98     // WebCore::IconDatabaseClient
99     virtual void didImportIconURLForPageURL(const String&);
100     virtual void didImportIconDataForPageURL(const String&);
101     virtual void didChangeIconForPageURL(const String&);
102     virtual void didRemoveAllIcons();
103     virtual void didFinishURLImport();
104
105     // CoreIPC::MessageReceiver
106     virtual void didReceiveMessage(CoreIPC::Connection*, CoreIPC::MessageID, CoreIPC::ArgumentDecoder*) OVERRIDE;
107     virtual void didReceiveSyncMessage(CoreIPC::Connection*, CoreIPC::MessageID, CoreIPC::ArgumentDecoder*, OwnPtr<CoreIPC::ArgumentEncoder>&) OVERRIDE;
108
109     void didReceiveWebIconDatabaseMessage(CoreIPC::Connection*, CoreIPC::MessageID, CoreIPC::ArgumentDecoder*);
110     void didReceiveSyncWebIconDatabaseMessage(CoreIPC::Connection*, CoreIPC::MessageID, CoreIPC::ArgumentDecoder*, OwnPtr<CoreIPC::ArgumentEncoder>&);
111
112     void notifyIconDataReadyForPageURL(const String&);
113
114     WebContext* m_webContext;
115     
116     OwnPtr<WebCore::IconDatabase> m_iconDatabaseImpl;
117     bool m_urlImportCompleted;
118     bool m_databaseCleanupDisabled;
119     HashMap<uint64_t, String> m_pendingLoadDecisionURLMap;
120
121     WebIconDatabaseClient m_iconDatabaseClient;
122 };
123
124 } // namespace WebKit
125
126 #endif // WebIconDatabase_h