connection.hpp

// Copyright (c) 2022 Klemens D. Morgenstern
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
#ifndef BOOST_SQLITE_CONNECTION_HPP
#define BOOST_SQLITE_CONNECTION_HPP
 
#include <boost/sqlite/detail/config.hpp>
#include <boost/sqlite/error.hpp>
#include <boost/sqlite/resultset.hpp>
#include <boost/sqlite/statement.hpp>
#include <memory>
#include <boost/system/system_error.hpp>
 
BOOST_SQLITE_BEGIN_NAMESPACE
 
template<typename T>
struct static_resultset;
 
struct connection
{
    using handle_type = sqlite3*;
    handle_type handle() const { return impl_.get(); }
    handle_type release() &&    { return impl_.release(); }
 
    connection() = default;
    explicit connection(handle_type handle, bool take_ownership = true) : impl_(handle, take_ownership) {}
    connection(connection && ) = default;
    connection& operator=(connection && ) = default;
 
    connection(cstring_ref filename,
               int flags = SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE) { connect(filename, flags); }
 
 
    BOOST_SQLITE_DECL void connect(cstring_ref filename, int flags = SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE);
    BOOST_SQLITE_DECL void connect(cstring_ref filename, int flags, system::error_code & ec);
 
    BOOST_SQLITE_DECL void close();
    BOOST_SQLITE_DECL void close(system::error_code & ec, error_info & ei);
 
    bool valid() const {return impl_ != nullptr;}
 
 
    BOOST_SQLITE_DECL resultset query(
            core::string_view q,
            system::error_code & ec,
            error_info & ei);
 
    BOOST_SQLITE_DECL resultset query(core::string_view q);
 
    template<typename T>
    static_resultset<T> query(
        core::string_view q,
        system::error_code & ec,
        error_info & ei)
    {
        static_resultset<T> tmp = query(q, ec, ei);
        if (ec)
            return {};
        tmp.check_columns_(ec, ei);
        if (ec)
            return {};
 
        return tmp;
    }
 
    template<typename T>
    static_resultset<T> query(core::string_view q)
    {
        system::error_code ec;
        error_info ei;
        auto tmp = query<T>(q, ec, ei);
        if (ec)
            throw_exception(system::system_error(ec, ei.message()));
        return tmp;
    }
 
    BOOST_SQLITE_DECL void execute(
        cstring_ref q,
        system::error_code & ec,
        error_info & ei);
 
    BOOST_SQLITE_DECL void execute(cstring_ref q);
 
    template<typename StringLike,
             typename = decltype(std::declval<const StringLike&>().c_str())>
    void execute(
        const StringLike& q,
        system::error_code & ec,
        error_info & ei)
    {
        execute(q.c_str(), ec, ei);
    }
    template<typename StringLike,
             typename = decltype(std::declval<const StringLike&>().c_str())>
    void execute(const StringLike & q) { execute(q.c_str());}
 
    BOOST_SQLITE_DECL statement prepare(
            core::string_view q,
            system::error_code & ec,
            error_info & ei);
 
    BOOST_SQLITE_DECL statement prepare(core::string_view q);
 
    bool has_table(
        cstring_ref table,
        cstring_ref db_name = "main") const
    {
      return sqlite3_table_column_metadata(impl_.get(), db_name.c_str(), table.c_str(),
                                           nullptr, nullptr, nullptr, nullptr, nullptr, nullptr)
             == SQLITE_OK;
    }
 
    bool has_column(
        cstring_ref table,
        cstring_ref column,
        cstring_ref db_name = "main") const
    {
      return sqlite3_table_column_metadata(impl_.get(), db_name.c_str(), table.c_str(), column.c_str(),
                                           nullptr, nullptr, nullptr, nullptr, nullptr)
             == SQLITE_OK;
    }
 private:
    struct deleter_
    {
        deleter_(bool owned = true) : owned_(owned) {}
        bool owned_ = true;
        void operator()(sqlite3  *impl)
        {
            if (owned_)
              sqlite3_close_v2(impl);
        }
    };
 
    std::unique_ptr<sqlite3, deleter_> impl_{nullptr, deleter_{}};
};
 
BOOST_SQLITE_END_NAMESPACE
 
 
#endif //BOOST_SQLITE_CONNECTION_HPP