#ifndef DatabaseTracker_h
#define DatabaseTracker_h
#include "DatabaseDetails.h"
#include "DatabaseError.h"
#include "SQLiteDatabase.h"
#include "SecurityOriginHash.h"
#include <wtf/HashMap.h>
#include <wtf/HashSet.h>
#include <wtf/text/StringHash.h>
#include <wtf/text/WTFString.h>
namespace WebCore {
class Database;
class DatabaseContext;
class DatabaseManagerClient;
class OriginLock;
class SecurityOrigin;
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& tracker();
bool canEstablishDatabase(DatabaseContext*, const String& name, unsigned long estimatedSize, DatabaseError&);
bool retryCanEstablishDatabase(DatabaseContext*, const String& name, unsigned long estimatedSize, DatabaseError&);
void setDatabaseDetails(SecurityOrigin*, const String& name, const String& displayName, unsigned long estimatedSize);
String fullPathForDatabase(SecurityOrigin*, const String& name, bool createIfDoesNotExist = true);
void addOpenDatabase(Database*);
void removeOpenDatabase(Database*);
void getOpenDatabases(SecurityOrigin*, const String& name, HashSet<RefPtr<Database>>* databases);
unsigned long long getMaxSizeForDatabase(const Database*);
WEBCORE_EXPORT void closeAllDatabases();
void interruptAllDatabasesForContext(const DatabaseContext*);
private:
explicit DatabaseTracker(const String& databasePath);
bool hasAdequateQuotaForOrigin(SecurityOrigin*, unsigned long estimatedSize, DatabaseError&);
public:
void setDatabaseDirectoryPath(const String&);
String databaseDirectoryPath() const;
WEBCORE_EXPORT void origins(Vector<RefPtr<SecurityOrigin>>& result);
bool databaseNamesForOrigin(SecurityOrigin*, Vector<String>& result);
DatabaseDetails detailsForNameAndOrigin(const String&, SecurityOrigin*);
unsigned long long usageForOrigin(SecurityOrigin*);
unsigned long long quotaForOrigin(SecurityOrigin*);
void setQuota(SecurityOrigin*, unsigned long long);
RefPtr<OriginLock> originLockFor(SecurityOrigin*);
void deleteAllDatabases();
WEBCORE_EXPORT void deleteDatabasesModifiedSince(std::chrono::system_clock::time_point);
WEBCORE_EXPORT bool deleteOrigin(SecurityOrigin*);
bool deleteDatabase(SecurityOrigin*, const String& name);
#if PLATFORM(IOS)
WEBCORE_EXPORT void removeDeletedOpenedDatabases();
WEBCORE_EXPORT static bool deleteDatabaseFileIfEmpty(const String&);
static Mutex& openDatabaseMutex();
WEBCORE_EXPORT static void emptyDatabaseFilesRemovalTaskWillBeScheduled();
WEBCORE_EXPORT static void emptyDatabaseFilesRemovalTaskDidFinish();
WEBCORE_EXPORT void setDatabasesPaused(bool);
#endif
void setClient(DatabaseManagerClient*);
void scheduleNotifyDatabaseChanged(SecurityOrigin*, const String& name);
bool hasEntryForOrigin(SecurityOrigin*);
void doneCreatingDatabase(Database*);
private:
bool hasEntryForOriginNoLock(SecurityOrigin* origin);
String fullPathForDatabaseNoLock(SecurityOrigin*, const String& name, bool createIfDoesNotExist);
bool databaseNamesForOriginNoLock(SecurityOrigin* origin, Vector<String>& resultVector);
unsigned long long quotaForOriginNoLock(SecurityOrigin* origin);
String trackerDatabasePath() const;
enum TrackerCreationAction {
DontCreateIfDoesNotExist,
CreateIfDoesNotExist
};
void openTrackerDatabase(TrackerCreationAction);
String originPath(SecurityOrigin*) const;
bool hasEntryForDatabase(SecurityOrigin*, const String& databaseIdentifier);
bool addDatabase(SecurityOrigin*, const String& name, const String& path);
bool deleteDatabaseFile(SecurityOrigin*, const String& name);
void deleteOriginLockFor(SecurityOrigin*);
typedef HashSet<Database*> DatabaseSet;
typedef HashMap<String, DatabaseSet*> DatabaseNameMap;
typedef HashMap<RefPtr<SecurityOrigin>, DatabaseNameMap*> DatabaseOriginMap;
Mutex m_openDatabaseMapGuard;
mutable std::unique_ptr<DatabaseOriginMap> m_openDatabaseMap;
Mutex m_databaseGuard;
SQLiteDatabase m_database;
typedef HashMap<String, RefPtr<OriginLock>> OriginLockMap;
OriginLockMap m_originLockMap;
String m_databaseDirectoryPath;
DatabaseManagerClient* m_client;
typedef HashMap<String, long> NameCountMap;
typedef HashMap<RefPtr<SecurityOrigin>, NameCountMap*, SecurityOriginHash> CreateSet;
CreateSet m_beingCreated;
typedef HashSet<String> NameSet;
HashMap<RefPtr<SecurityOrigin>, NameSet*> m_beingDeleted;
HashSet<RefPtr<SecurityOrigin>> m_originsBeingDeleted;
bool isDeletingDatabaseOrOriginFor(SecurityOrigin*, const String& name);
void recordCreatingDatabase(SecurityOrigin*, const String& name);
void doneCreatingDatabase(SecurityOrigin*, const String& name);
bool creatingDatabase(SecurityOrigin*, const String& name);
bool canDeleteDatabase(SecurityOrigin*, const String& name);
void recordDeletingDatabase(SecurityOrigin*, const String& name);
void doneDeletingDatabase(SecurityOrigin*, const String& name);
bool isDeletingDatabase(SecurityOrigin*, const String& name);
bool canDeleteOrigin(SecurityOrigin*);
bool isDeletingOrigin(SecurityOrigin*);
void recordDeletingOrigin(SecurityOrigin*);
void doneDeletingOrigin(SecurityOrigin*);
static void scheduleForNotification();
static void notifyDatabasesChanged(void*);
};
}
#endif // DatabaseTracker_h