#pragma once
#include <functional>
#include <sqlite3.h>
#include <wtf/Lock.h>
#include <wtf/Threading.h>
#include <wtf/text/CString.h>
#include <wtf/text/WTFString.h>
#if COMPILER(MSVC)
#pragma warning(disable: 4800)
#endif
struct sqlite3;
namespace WebCore {
class DatabaseAuthorizer;
class SQLiteStatement;
class SQLiteTransaction;
class SQLiteDatabase {
WTF_MAKE_FAST_ALLOCATED;
WTF_MAKE_NONCOPYABLE(SQLiteDatabase);
friend class SQLiteTransaction;
public:
WEBCORE_EXPORT SQLiteDatabase();
WEBCORE_EXPORT ~SQLiteDatabase();
enum class OpenMode { ReadOnly, ReadWrite, ReadWriteCreate };
WEBCORE_EXPORT bool open(const String& filename, OpenMode = OpenMode::ReadWriteCreate);
bool isOpen() const { return m_db; }
WEBCORE_EXPORT void close();
void updateLastChangesCount();
WEBCORE_EXPORT bool executeCommand(const String&);
bool returnsAtLeastOneResult(const String&);
WEBCORE_EXPORT bool tableExists(const String&);
WEBCORE_EXPORT void clearAllTables();
WEBCORE_EXPORT int runVacuumCommand();
int runIncrementalVacuumCommand();
bool transactionInProgress() const { return m_transactionInProgress; }
void interrupt();
int64_t lastInsertRowID();
int lastChanges();
void setBusyTimeout(int ms);
void setBusyHandler(int(*)(void*, int));
void setFullsync(bool);
int64_t maximumSize();
void setMaximumSize(int64_t);
int64_t freeSpaceSize();
int64_t totalSize();
enum SynchronousPragma { SyncOff = 0, SyncNormal = 1, SyncFull = 2 };
void setSynchronous(SynchronousPragma);
WEBCORE_EXPORT int lastError();
WEBCORE_EXPORT const char* lastErrorMsg();
sqlite3* sqlite3Handle() const
{
#if !PLATFORM(IOS_FAMILY)
ASSERT(m_sharable || m_openingThread == &Thread::current() || !m_db);
#endif
return m_db;
}
void setAuthorizer(DatabaseAuthorizer&);
Lock& databaseMutex() { return m_lockingMutex; }
bool isAutoCommitOn() const;
enum AutoVacuumPragma { AutoVacuumNone = 0, AutoVacuumFull = 1, AutoVacuumIncremental = 2 };
WEBCORE_EXPORT bool turnOnIncrementalAutoVacuum();
WEBCORE_EXPORT void setCollationFunction(const String& collationName, WTF::Function<int(int, const void*, int, const void*)>&&);
void removeCollationFunction(const String& collationName);
#ifndef NDEBUG
WEBCORE_EXPORT void disableThreadingChecks();
#else
void disableThreadingChecks() { }
#endif
WEBCORE_EXPORT static void setIsDatabaseOpeningForbidden(bool);
private:
static int authorizerFunction(void*, int, const char*, const char*, const char*, const char*);
void enableAuthorizer(bool enable);
void useWALJournalMode();
int pageSize();
void overrideUnauthorizedFunctions();
sqlite3* m_db { nullptr };
int m_pageSize { -1 };
bool m_transactionInProgress { false };
#ifndef NDEBUG
bool m_sharable { false };
#endif
bool m_useWAL { false };
Lock m_authorizerLock;
RefPtr<DatabaseAuthorizer> m_authorizer;
Lock m_lockingMutex;
RefPtr<Thread> m_openingThread { nullptr };
Lock m_databaseClosingMutex;
int m_openError { SQLITE_ERROR };
CString m_openErrorMessage;
int m_lastChangesCount { 0 };
};
}