query.hpp

#pragma once
 
#include <cstddef>
#include <optional>
#include <utility>
 
#include "model.hpp"
#include "query/QueryData.hpp"
 
namespace orm
{
class Database;
template <typename T>
class Query
{
public:
    Query() : data{.modelInfo = Model<T>().getModelInfo()} {}

    auto where(const query::Predicate& predicate) -> Query<T>&
    {
        data.predicate = predicate;
 
        return *this;
    }

    auto andWhere(const query::Predicate& predicate) -> Query<T>&
    {
        data.predicate = data.predicate.has_value() ? data.predicate.value() && predicate : predicate;
 
        return *this;
    }

    auto orWhere(const query::Predicate& predicate) -> Query<T>&
    {
        data.predicate = data.predicate.has_value() ? data.predicate.value() || predicate : predicate;
 
        return *this;
    }

    template <typename... Orders>
    auto orderBy(Orders... orders) -> Query<T>&
    {
        data.orderBy = {std::move(orders)...};
 
        return *this;
    }

    inline auto distinct() -> Query<T>&
    {
        data.isDistinct = true;
 
        return *this;
    }

    inline auto offset(std::size_t offset) -> Query<T>&
    {
        data.offset = offset;
 
        return *this;
    }

    inline auto limit(std::size_t limit) -> Query<T>&
    {
        data.limit = limit;
 
        return *this;
    }

    inline auto disableJoining() -> Query<T>&
    {
        data.shouldJoin = false;
 
        return *this;
    }
 
private:
    friend class orm::Database;

    [[nodiscard]] inline auto getData() const -> const query::QueryData&
    {
        return data;
    }
 
    query::QueryData data; 
};
} // namespace orm