/**************************************************************************
**
** 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 "table.h"
#include "database.h"
#include "abstracttableset.h"
#include "databasemodel.h"
#include "abstracttablesetdata.h"

NUT_BEGIN_NAMESPACE

AbstractTableSet::AbstractTableSet(Database *parent) : QObject(parent),
    data(new AbstractTableSetData(parent))
{
    parent->add(this);
}

AbstractTableSet::AbstractTableSet(Table *parent) : QObject(parent),
    data(new AbstractTableSetData(parent))
{
    parent->add(this);
}

AbstractTableSet::~AbstractTableSet()
{
    Q_FOREACH (Row<Table> t, data->childs)
        if (t)
            t->setParentTableSet(nullptr);
}

int AbstractTableSet::save(Database *db, bool cleanUp)
{
    int rowsAffected = 0;
    TableModel *masterModel = nullptr;
    if (data->table)
        masterModel = db->model().tableByClassName(QString::fromUtf8(data->table->metaObject()->className()));

    Q_FOREACH (Row<Table> t, data->childs) {
        if (data->table)
            t->setParentTable(data->table,
                              masterModel,
                              db->model().tableByClassName(QString::fromUtf8(t->metaObject()->className())));

        if (t->status() == Table::Added
            || t->status() == Table::Modified
            || t->status() == Table::Deleted) {
            rowsAffected += t->save(db);
            if (cleanUp)
#ifdef NUT_RAW_POINTER
                t->deleteLater();
#else
                remove(t);
#endif
        }
    }

    if (cleanUp)
        data->childs.clear();

    return rowsAffected;
}

void AbstractTableSet::clearChilds()
{
#ifdef NUT_RAW_POINTER
    Q_FOREACH (Table *t, data->childs)
        t->deleteLater();
#endif
    data->childs.clear();
}

void AbstractTableSet::add(Row<Table> t)
{
    data.detach();
    data->childs.append(t);
    t->setParentTableSet(this);
}

void AbstractTableSet::remove(Row<Table> t)
{
    data.detach();
    data->childs.removeAll(t);
}

QString AbstractTableSet::childClassName() const
{
    return data->childClassName;
}

Database *AbstractTableSet::database() const
{
    return data->database;
}

void AbstractTableSet::setDatabase(Database *database)
{
    data.detach();
    data->database = database;
}

int AbstractTableSet::size() const
{
    return data->childs.size();
}

NUT_END_NAMESPACE