Nut/src/query.h

/**************************************************************************
**
** This file is part of Nut project.
** https://github.com/HamedMasafi/Nut
**
** Nut is free software: you can redistribute it and/or modify
** it under the terms of the GNU Lesser General Public License as published by
** the Free Software Foundation, either version 3 of the License, or
** (at your option) any later version.
**
** Nut is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU Lesser General Public License for more details.
**
** You should have received a copy of the GNU Lesser General Public License
** along with Nut.  If not, see <http://www.gnu.org/licenses/>.
**
**************************************************************************/

#ifndef QUERY_H
#define QUERY_H

#include <QtCore/QVariant>
#include <QtCore/QDebug>
#include <QtCore/QScopedPointer>
#include <QtCore/QRegularExpression>
#include <QtCore/QMetaObject>

#include "query_p.h"
#include "database.h"
#include "databasemodel.h"
#include "tablesetbase_p.h"
#include "sqlgeneratorbase_p.h"
#include "querybase_p.h"
#include "wherephrase.h"
#include "tablemodel.h"

NUT_BEGIN_NAMESPACE

template <class T>
class NUT_EXPORT Query : public QueryBase
{
    QueryPrivate *d_ptr;
    Q_DECLARE_PRIVATE(Query)

    bool m_autoDelete;

public:
    Query(Database *database, TableSetBase *tableSet, bool autoDelete);

    ~Query();

    Query<T> *join(const QString &tableName);
    Query<T> *setWhere(WherePhrase where);

    Query<T> *join(Table *c)
    {
        join(c->metaObject()->className());
        return this;
    }

//    Query<T> *orderBy(QString fieldName, QString type);
    Query<T> *skip(int &n);
    Query<T> *take(int &n);
    Query<T> *orderBy(WherePhrase phrase);
    Query<T> *include(TableSetBase *t);
    Query<T> *include(Table *t);

    int count();
    QVariant max(FieldPhrase<int> &f);
    QVariant min(FieldPhrase<int> &f);
    QVariant average(FieldPhrase<int> &f);
    T *first();
    QList<T *> toList(int count = -1);
    template <typename F>
    QList<F> select(const FieldPhrase<F> f);

    int update(WherePhrase phrase);
    int remove();

    QString sqlCommand() const;
};

template <typename T>
inline Query<T> *createQuery(TableSet<T> *tableSet)
{
    return tableSet->query();
}

template <class T>
Q_OUTOFLINE_TEMPLATE Query<T>::Query(Database *database, TableSetBase *tableSet,
                                     bool autoDelete)
    : QueryBase(database), d_ptr(new QueryPrivate(this)),
      m_autoDelete(autoDelete)
{
    Q_D(Query);

    d->database = database;
    d->tableSet = tableSet;
    d->tableName
        = // TableModel::findByClassName(T::staticMetaObject.className())->name();
        d->database->model()
            .tableByClassName(T::staticMetaObject.className())
            ->name();
}

template <class T>
Q_OUTOFLINE_TEMPLATE Query<T>::~Query()
{
    Q_D(Query);

    delete d;
}

template <class T>
Q_OUTOFLINE_TEMPLATE QList<T *> Query<T>::toList(int count)
{
    Q_D(Query);
    QList<T *> result;
    d->select = "*";

    //    QSqlQuery q =
    //    d->database->exec(d->database->sqlGenertor()->selectCommand(d->wheres,
    //    d->orders, d->tableName, d->joinClassName));
    d->sql = d->database->sqlGenertor()->selectCommand(
        SqlGeneratorBase::SelectAll, "", d->wheres, d->orderPhrases,
        d->tableName, d->joinClassName, d->skip, d->take);
    QSqlQuery q = d->database->exec(d->sql);

    //    QString pk = TableModel::findByName(d->tableName)->primaryKey();
    QString pk = d->database->model().tableByName(d->tableName)->primaryKey();
    QVariant lastPkValue = QVariant();
    int childTypeId = 0;
    T *lastRow = 0;
    TableSetBase *childTableSet = Q_NULLPTR;

    // FIXME: getting table error
    //    QStringList masterFields =
    //    TableModel::findByName(d->tableName)->fieldsNames();
    QStringList masterFields
        = d->database->model().tableByName(d->tableName)->fieldsNames();
    QStringList childFields;
    if (!d->joinClassName.isNull()) {
        TableModel *joinTableModel
            = TableModel::findByClassName(d->joinClassName);
        if (joinTableModel) {
            //            childFields =
            //            d->database->model().modelByClass(d->joinClassName)->fieldsNames();
            childFields
                = TableModel::findByClassName(d->joinClassName)->fieldsNames();
            QString joinTableName = d->database->tableName(d->joinClassName);
            childTypeId = d->database->model().tableByName(joinTableName)->typeId();
            //            childTypeId =
            //            TableModel::findByName(joinTableName)->typeId();
        }
    }

    while (q.next()) {
        if (lastPkValue != q.value(pk)) {
            T *t = new T();
            foreach (QString field, masterFields)
                t->setProperty(field.toLatin1().data(), q.value(field));
            //            for (int i = 0; i < t->metaObject()->propertyCount();
            //            i++) {
            //                const QMetaProperty p =
            //                t->metaObject()->property(i);

            //                p.write(t,
            //                d->database->sqlGenertor()->readValue(p.type(),
            //                q.value(p.name())));
            //            }

            t->setTableSet(d->tableSet);
            t->setStatus(Table::FeatchedFromDB);
            t->setParent(this);
            t->clear();

            result.append(t);
            lastRow = t;

            if (childTypeId) {
                QSet<TableSetBase *> tableSets = t->tableSets;
                foreach (TableSetBase *ts, tableSets)
                    if (ts->childClassName() == d->joinClassName)
                        childTableSet = ts;
            }
        }

        if (childTypeId) {
            const QMetaObject *childMetaObject
                = QMetaType::metaObjectForType(childTypeId);
            Table *childTable
                = qobject_cast<Table *>(childMetaObject->newInstance());

            foreach (QString field, childFields)
                childTable->setProperty(field.toLatin1().data(),
                                        q.value(field));
            // TODO: set database for table
            childTable->setParent(this);
            childTable->setParentTable(lastRow);
            childTable->setStatus(Table::FeatchedFromDB);
            childTable->setTableSet(childTableSet);
            childTable->clear();
            childTableSet->add(childTable);
        }
        lastPkValue = q.value(pk);

        if (!--count)
            break;
    }

    if (m_autoDelete)
        deleteLater();
    return result;
}

template <typename T>
template <typename F>
Q_OUTOFLINE_TEMPLATE QList<F> Query<T>::select(const FieldPhrase<F> f)
{
    Q_D(Query);
    QList<F> ret;
    d->sql = d->database->sqlGenertor()->selectCommand(
        SqlGeneratorBase::SignleField, f.data()->text, d->wheres,
        d->orderPhrases, d->tableName, d->joinClassName, d->skip, d->take);

    QSqlQuery q = d->database->exec(d->sql);

    while (q.next()) {
        QVariant v = q.value(0);
        ret.append(v.value<F>());
    }

    if (m_autoDelete)
        deleteLater();
    return ret;
}

template <class T>
Q_OUTOFLINE_TEMPLATE T *Query<T>::first()
{
    QList<T *> list = toList(1);

    if (list.count())
        return list.first();
    else
        return 0;
}

template <class T>
Q_OUTOFLINE_TEMPLATE int Query<T>::count()
{
    Q_D(Query);

    d->select = "COUNT(*)";
    d->sql = d->database->sqlGenertor()->selectCommand(SqlGeneratorBase::Count,
        QStringLiteral("*"), d->wheres, d->orderPhrases, d->tableName, d->joinClassName);
    QSqlQuery q = d->database->exec(d->sql);

    if (q.next())
        return q.value(0).toInt();
    return 0;
}

template <class T>
Q_OUTOFLINE_TEMPLATE QVariant Query<T>::max(FieldPhrase<int> &f)
{
    Q_D(Query);

    d->sql = d->database->sqlGenertor()->selectCommand(
        SqlGeneratorBase::Max, f.data()->text, d->wheres, d->orderPhrases,
        d->tableName, d->joinClassName);
    QSqlQuery q = d->database->exec(d->sql);

    if (q.next())
        return q.value(0).toInt();
    return 0;
}

template <class T>
Q_OUTOFLINE_TEMPLATE QVariant Query<T>::min(FieldPhrase<int> &f)
{
    Q_D(Query);

    d->sql = d->database->sqlGenertor()->selectCommand(
        SqlGeneratorBase::Min, f.data()->text, d->wheres, d->orderPhrases,
        d->tableName, d->joinClassName);
    QSqlQuery q = d->database->exec(d->sql);

    if (q.next())
        return q.value(0).toInt();
    return 0;
}

template <class T>
Q_OUTOFLINE_TEMPLATE QVariant Query<T>::average(FieldPhrase<int> &f)
{
    Q_D(Query);

    d->sql = d->database->sqlGenertor()->selectCommand(
        SqlGeneratorBase::Average, f.data()->text, d->wheres, d->orderPhrases,
        d->tableName, d->joinClassName);
    QSqlQuery q = d->database->exec(d->sql);

    if (q.next())
        return q.value(0).toInt();
    return 0;
}

template <class T>
Q_OUTOFLINE_TEMPLATE Query<T> *Query<T>::join(const QString &tableName)
{
    Q_D(Query);
    d->joinClassName = tableName;
    return this;
}

template <class T>
Q_OUTOFLINE_TEMPLATE Query<T> *Query<T>::setWhere(WherePhrase where)
{
    Q_D(Query);
    d->wheres.append(where);
    return this;
}

template<class T>
Q_OUTOFLINE_TEMPLATE Query<T> *Query<T>::skip(int &n)
{
    Q_D(Query);
    d->skip = n;
    return this;
}

template<class T>
Q_OUTOFLINE_TEMPLATE Query<T> *Query<T>::take(int &n)
{
    Q_D(Query);
    d->take = n;
    return this;
}

//template <class T>
//Q_OUTOFLINE_TEMPLATE Query<T> *Query<T>::orderBy(QString fieldName,
//                                                 QString type)
//{
//    Q_D(Query);
//    d->orderPhrases.append(fieldName, type);
//    return this;
//}

template <class T>
Q_OUTOFLINE_TEMPLATE Query<T> *Query<T>::orderBy(WherePhrase phrase)
{
    Q_D(Query);
    d->orderPhrases.append(phrase);
    return this;
}

template <class T>
Q_OUTOFLINE_TEMPLATE Query<T> *Query<T>::include(TableSetBase *t)
{
    Q_D(Query);
    d->joinClassName = t->childClassName();
    return this;
}

template <class T>
Q_OUTOFLINE_TEMPLATE Query<T> *Query<T>::include(Table *t)
{
    Q_D(Query);
    d->joinClassName = t->metaObject()->className();
    return this;
}

template <class T>
Q_OUTOFLINE_TEMPLATE int Query<T>::update(WherePhrase phrase)
{
    Q_D(Query);

    d->sql = d->database->sqlGenertor()->updateCommand(phrase, d->wheres,
                                                       d->tableName);
    QSqlQuery q = d->database->exec(d->sql);

    if (m_autoDelete)
        deleteLater();
    return q.numRowsAffected();
}

template <class T>
Q_OUTOFLINE_TEMPLATE int Query<T>::remove()
{
    Q_D(Query);

    d->sql = d->database->sqlGenertor()->deleteCommand(d->wheres, d->tableName);
    QSqlQuery q = d->database->exec(d->sql);

    if (m_autoDelete)
        deleteLater();
    return q.numRowsAffected();
}

template <class T>
Q_OUTOFLINE_TEMPLATE QString Query<T>::sqlCommand() const
{
    Q_D(const Query);
    return d->sql;
}

NUT_END_NAMESPACE

#endif // QUERY_H