1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
/*
* Copyright (C) 2019 Igalia S.L.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public License
* along with this library; see the file COPYING.LIB. If not, write to
* the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#pragma once
#include <WebCore/NativeImage.h>
#include <WebCore/SQLiteDatabase.h>
#include <WebCore/SQLiteStatement.h>
#include <wtf/CompletionHandler.h>
#include <wtf/HashMap.h>
#include <wtf/Lock.h>
#include <wtf/ThreadSafeRefCounted.h>
#include <wtf/WorkQueue.h>
namespace WebKit {
class IconDatabase : public ThreadSafeRefCounted<IconDatabase> {
public:
enum class AllowDatabaseWrite : bool { No, Yes };
static Ref<IconDatabase> create(const String& path, AllowDatabaseWrite allowDatabaseWrite)
{
return adoptRef(*new IconDatabase(path, allowDatabaseWrite));
}
~IconDatabase();
void invalidate();
void checkIconURLAndSetPageURLIfNeeded(const String& iconURL, const String& pageURL, AllowDatabaseWrite, CompletionHandler<void(bool, bool)>&&);
void loadIconForPageURL(const String&, AllowDatabaseWrite, CompletionHandler<void(WebCore::PlatformImagePtr&&)>&&);
String iconURLForPageURL(const String&);
void setIconForPageURL(const String& iconURL, std::span<const uint8_t>, const String& pageURL, AllowDatabaseWrite, CompletionHandler<void(bool)>&&);
void clear(CompletionHandler<void()>&&);
private:
IconDatabase(const String&, AllowDatabaseWrite);
bool createTablesIfNeeded();
void populatePageURLToIconURLMap();
void pruneTimerFired();
void startPruneTimer();
void clearStatements();
void clearLoadedIconsTimerFired();
void startClearLoadedIconsTimer();
std::optional<int64_t> iconIDForIconURL(const String&, bool& expired);
bool setIconIDForPageURL(int64_t, const String&);
Vector<uint8_t> iconData(int64_t);
std::optional<int64_t> addIcon(const String&, const Vector<uint8_t>&);
void updateIconTimestamp(int64_t iconID, int64_t timestamp);
void deleteIcon(int64_t);
Ref<WorkQueue> m_workQueue;
AllowDatabaseWrite m_allowDatabaseWrite { AllowDatabaseWrite::Yes };
WebCore::SQLiteDatabase m_db;
HashMap<String, String> m_pageURLToIconURLMap;
Lock m_pageURLToIconURLMapLock;
HashMap<String, std::pair<WebCore::PlatformImagePtr, MonotonicTime>> m_loadedIcons WTF_GUARDED_BY_LOCK(m_loadedIconsLock);
Lock m_loadedIconsLock;
std::unique_ptr<WebCore::SQLiteStatement> m_iconIDForIconURLStatement;
std::unique_ptr<WebCore::SQLiteStatement> m_setIconIDForPageURLStatement;
std::unique_ptr<WebCore::SQLiteStatement> m_iconDataStatement;
std::unique_ptr<WebCore::SQLiteStatement> m_addIconStatement;
std::unique_ptr<WebCore::SQLiteStatement> m_addIconDataStatement;
std::unique_ptr<WebCore::SQLiteStatement> m_updateIconTimestampStatement;
std::unique_ptr<WebCore::SQLiteStatement> m_deletePageURLsForIconStatement;
std::unique_ptr<WebCore::SQLiteStatement> m_deleteIconDataStatement;
std::unique_ptr<WebCore::SQLiteStatement> m_deleteIconStatement;
std::unique_ptr<WebCore::SQLiteStatement> m_pruneIconsStatement;
std::unique_ptr<RunLoop::Timer> m_pruneTimer;
RunLoop::Timer m_clearLoadedIconsTimer;
};
} // namespace WebKit
|
