#pragma once
#include "DatabaseDetails.h"
#include "ExceptionOr.h"
#include "SQLiteDatabase.h"
#include "SecurityOriginData.h"
#include "SecurityOriginHash.h"
#include <wtf/HashCountedSet.h>
#include <wtf/HashMap.h>
#include <wtf/HashSet.h>
#include <wtf/text/StringHash.h>
namespace WebCore {
class Database;
class DatabaseContext;
class DatabaseManagerClient;
class OriginLock;
class SecurityOrigin;
struct SecurityOriginData;
enum class CurrentQueryBehavior { Interrupt, RunToCompletion };
class DatabaseTracker {
WTF_MAKE_NONCOPYABLE(DatabaseTracker); WTF_MAKE_FAST_ALLOCATED;
public:
WEBCORE_EXPORT static std::unique_ptr<DatabaseTracker> trackerWithDatabasePath(const String& databasePath);
static void initializeTracker(const String& databasePath);
WEBCORE_EXPORT static DatabaseTracker& singleton();
ExceptionOr<void> canEstablishDatabase(DatabaseContext&, const String& name, unsigned estimatedSize);
ExceptionOr<void> retryCanEstablishDatabase(DatabaseContext&, const String& name, unsigned estimatedSize);
void setDatabaseDetails(const SecurityOriginData&, const String& name, const String& displayName, unsigned estimatedSize);
String fullPathForDatabase(const SecurityOriginData&, const String& name, bool createIfDoesNotExist);
void addOpenDatabase(Database&);
void removeOpenDatabase(Database&);
unsigned long long maximumSize(Database&);
WEBCORE_EXPORT void closeAllDatabases(CurrentQueryBehavior = CurrentQueryBehavior::RunToCompletion);
WEBCORE_EXPORT Vector<SecurityOriginData> origins();
WEBCORE_EXPORT Vector<String> databaseNames(const SecurityOriginData&);
DatabaseDetails detailsForNameAndOrigin(const String&, const SecurityOriginData&);
WEBCORE_EXPORT unsigned long long usage(const SecurityOriginData&);
WEBCORE_EXPORT unsigned long long quota(const SecurityOriginData&);
WEBCORE_EXPORT void setQuota(const SecurityOriginData&, unsigned long long);
RefPtr<OriginLock> originLockFor(const SecurityOriginData&);
WEBCORE_EXPORT void deleteAllDatabasesImmediately();
WEBCORE_EXPORT void deleteDatabasesModifiedSince(std::chrono::system_clock::time_point);
WEBCORE_EXPORT bool deleteOrigin(const SecurityOriginData&);
WEBCORE_EXPORT bool deleteDatabase(const SecurityOriginData&, const String& name);
#if PLATFORM(IOS)
WEBCORE_EXPORT void removeDeletedOpenedDatabases();
WEBCORE_EXPORT static bool deleteDatabaseFileIfEmpty(const String&);
static Lock& openDatabaseMutex();
WEBCORE_EXPORT static void emptyDatabaseFilesRemovalTaskWillBeScheduled();
WEBCORE_EXPORT static void emptyDatabaseFilesRemovalTaskDidFinish();
#endif
void setClient(DatabaseManagerClient*);
void scheduleNotifyDatabaseChanged(const SecurityOriginData&, const String& name);
void doneCreatingDatabase(Database&);
private:
explicit DatabaseTracker(const String& databasePath);
ExceptionOr<void> hasAdequateQuotaForOrigin(const SecurityOriginData&, unsigned estimatedSize);
bool hasEntryForOriginNoLock(const SecurityOriginData&);
String fullPathForDatabaseNoLock(const SecurityOriginData&, const String& name, bool createIfDoesNotExist);
Vector<String> databaseNamesNoLock(const SecurityOriginData&);
unsigned long long quotaNoLock(const SecurityOriginData&);
String trackerDatabasePath() const;
enum TrackerCreationAction {
DontCreateIfDoesNotExist,
CreateIfDoesNotExist
};
void openTrackerDatabase(TrackerCreationAction);
String originPath(const SecurityOriginData&) const;
bool hasEntryForDatabase(const SecurityOriginData&, const String& databaseIdentifier);
bool addDatabase(const SecurityOriginData&, const String& name, const String& path);
enum class DeletionMode {
Immediate,
#if PLATFORM(IOS)
Deferred,
Default = Deferred
#else
Default = Immediate
#endif
};
bool deleteOrigin(const SecurityOriginData&, DeletionMode);
bool deleteDatabaseFile(const SecurityOriginData&, const String& name, DeletionMode);
void deleteOriginLockFor(const SecurityOriginData&);
using DatabaseSet = HashSet<Database*>;
using DatabaseNameMap = HashMap<String, DatabaseSet*>;
using DatabaseOriginMap = HashMap<SecurityOriginData, DatabaseNameMap*>;
Lock m_openDatabaseMapGuard;
mutable std::unique_ptr<DatabaseOriginMap> m_openDatabaseMap;
Lock m_databaseGuard;
SQLiteDatabase m_database;
using OriginLockMap = HashMap<String, RefPtr<OriginLock>>;
OriginLockMap m_originLockMap;
String m_databaseDirectoryPath;
DatabaseManagerClient* m_client { nullptr };
HashMap<SecurityOriginData, std::unique_ptr<HashCountedSet<String>>> m_beingCreated;
HashMap<SecurityOriginData, std::unique_ptr<HashSet<String>>> m_beingDeleted;
HashSet<SecurityOriginData> m_originsBeingDeleted;
bool isDeletingDatabaseOrOriginFor(const SecurityOriginData&, const String& name);
void recordCreatingDatabase(const SecurityOriginData&, const String& name);
void doneCreatingDatabase(const SecurityOriginData&, const String& name);
bool creatingDatabase(const SecurityOriginData&, const String& name);
bool canDeleteDatabase(const SecurityOriginData&, const String& name);
void recordDeletingDatabase(const SecurityOriginData&, const String& name);
void doneDeletingDatabase(const SecurityOriginData&, const String& name);
bool isDeletingDatabase(const SecurityOriginData&, const String& name);
bool canDeleteOrigin(const SecurityOriginData&);
bool isDeletingOrigin(const SecurityOriginData&);
void recordDeletingOrigin(const SecurityOriginData&);
void doneDeletingOrigin(const SecurityOriginData&);
static void scheduleForNotification();
static void notifyDatabasesChanged();
};
}