/**************************************************************************
**
** 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/>.
**
**************************************************************************/

#include <QMetaMethod>
#include <QVariant>
#include "table.h"
#include "table_p.h"
#include "database.h"
#include "databasemodel.h"
#include "generators/sqlgeneratorbase_p.h"

NUT_BEGIN_NAMESPACE

/*
 * FIXME:
 *  Qt can not access metaObject inside of constructor
 *  so, if we can't initalize myModel inside of ctor. in
 *  other side myModel inited in propertyChanged signal, so
 *  any method that uses myModel (like: primaryKey, ...) can't
 *  be accessed before any property set. So ugly, but there are
 *  no other way for now.
 *
 *  This should be fixed to v1.2
 */

Table::Table(QObject *parent) : QObject(parent),
    d_ptr(new TablePrivate(this))
{
    Q_D(Table);
    d->status = NewCreated;
    d->model = TableModel::findByClassName(metaObject()->className());
}

Table::~Table()
{
    Q_D(Table);

    if (d->parentTableSet)
        d->parentTableSet->remove(this);
}

void Table::add(TableSetBase *t)
{
    Q_D(Table);
    d->childTableSets.insert(t);
}

QString Table::primaryKey() const
{
    Q_D(const Table);
    return d->model->primaryKey();
}

bool Table::isPrimaryKeyAutoIncrement() const
{
    Q_D(const Table);
    FieldModel *pk = d->model->field(d->model->primaryKey());
    if (!pk)
        return false;
    return pk->isAutoIncrement;
}


QVariant Table::primaryValue() const
{
    return property(primaryKey().toLatin1().data());
}

void Table::propertyChanged(const QString &propName)
{
    Q_D(Table);
    if (!d->model)
         d->model = TableModel::findByClassName(metaObject()->className());

    if (!d->model)
        qFatal ("model for class '%s' not found", qPrintable(metaObject()->className()));

    foreach (FieldModel *f, d->model->fields())
        if(f->isPrimaryKey && propName == f->name && f->isAutoIncrement)
            return;

    d->changedProperties.insert(propName);
    if (d->status == FeatchedFromDB)
        d->status = Modified;

    if (d->status == NewCreated)
        d->status = Added;
}

void Table::clear()
{
    Q_D(Table);
    d->changedProperties.clear();
}

QSet<QString> Table::changedProperties() const
{
    Q_D(const Table);
    return d->changedProperties;
}

bool Table::setParentTable(Table *master)
{
    Q_D(Table);

    QString masterClassName = master->metaObject()->className();

    foreach (RelationModel *r, d->model->foregionKeys())
        if(r->masterClassName == masterClassName)
        {
            setProperty(QString(r->localColumn).toLatin1().data(),
                        master->primaryValue());
            d->changedProperties.insert(r->localColumn);
            return true;
        }

    return false;
}

TableSetBase *Table::parentTableSet() const
{
    Q_D(const Table);
    return d->parentTableSet;
}

void Table::setParentTableSet(TableSetBase *parent)
{
    Q_D(Table);
    d->parentTableSet = parent;

    if (parent)
        d->parentTableSet->add(this);
}

TableSetBase *Table::childTableSet(const QString &name) const
{
    Q_D(const Table);
    foreach (TableSetBase *t, d->childTableSets)
        if (t->childClassName() == name)
            return t;
    return Q_NULLPTR;
}

int Table::save(Database *db)
{
    Q_D(Table);

    QSqlQuery q = db->exec(db->sqlGenertor()->saveRecord(this, db->tableName(metaObject()->className())));

    if(status() == Added && isPrimaryKeyAutoIncrement())
        setProperty(primaryKey().toLatin1().data(), q.lastInsertId());

    foreach(TableSetBase *ts, d->childTableSets)
        ts->save(db);
    setStatus(FeatchedFromDB);

    return q.numRowsAffected();
}

Table::Status Table::status() const
{
    Q_D(const Table);
    return static_cast<Status>(d->status);
}

void Table::setStatus(const Status &status)
{
    Q_D(Table);
    d->status = status;
}



TablePrivate::TablePrivate(Table *parent) : q_ptr(parent),
    status(Table::NewCreated), parentTableSet(nullptr)
{

}

NUT_END_NAMESPACE