sqlitedatabase.h

Go to the documentation of this file.

 
#ifndef SQLDATABASE_H
#define SQLDATABASE_H
 
#ifndef SQLITE_HAS_CODEC
#define SQLITE_HAS_CODEC
#endif
 
#include <filesystem>
#include <functional>
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
#include "sqlite.h"
#include "sqlitefunctioncontext.h"
#include "sqlitestatement.h"
 
namespace Nickvision::Database
{
    using SqliteCustomFunction = std::function<void(SqliteFunctionContext&)>;

    class SqliteDatabase
    {
    public:
        SqliteDatabase(const std::filesystem::path& path, int flags = SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE);
        SqliteDatabase(const SqliteDatabase&) = delete;
        SqliteDatabase(SqliteDatabase&& other) noexcept;
        ~SqliteDatabase() noexcept;
        const std::filesystem::path& getPath() const noexcept;
        bool isEncrypted() const noexcept;
        bool isUnlocked() const noexcept;
        bool unlock(const std::string& password) noexcept;
        bool setPassword(const std::string& password);
        bool registerFunction(const std::string& name, const SqliteCustomFunction& func, int expectedArgs = -1) noexcept;
        SqliteStatement createStatement(const std::string& command);
        bool execute(const std::string& command) noexcept;
        SqliteDatabase& operator=(const SqliteDatabase&) = delete;
        SqliteDatabase& operator=(SqliteDatabase&& other) noexcept;
 
    private:
        mutable std::mutex m_mutex;
        std::filesystem::path m_path;
        int m_flags;
        bool m_isEncrypted;
        bool m_isUnlocked;
        sqlite3* m_database;
        std::unordered_map<std::string, SqliteCustomFunction> m_customFunctions;
    };
}
 
#endif //SQLDATABASE_H