datasaver.cpp

#include "datasaver.h"

#include <domain/project.h>
#include <domain/taskmanagerdata.h>
#include <domain/user.h>
#include <domain/schedule.h>
#include <domain/taskstates.h>
#include <domain/reservation.h>

#include <QDomDocument>
#include <QDebug>
#include <QFile>

namespace control
{

struct DataSaverPrivate
{
    DataSaverPrivate(domain::TaskManagerData* d) : data(d) {}

    void createIds();
    QDomElement writeProjects(QDomDocument doc);
    QDomElement writeResources(QDomDocument doc);
    QDomElement writeUser(domain::User* user, QDomDocument doc);
    QDomElement writeTasks(domain::User* user, QDomDocument doc);
    QDomElement writeReservations(domain::User* user, QDomDocument doc);

    QDomElement createTextElement(const QString& name, const QString& value,
                                  QDomDocument& doc);

    domain::TaskManagerData* data;
    QMap<const domain::StorableData*, QString> idMap;
};

}

using namespace control;

ControllerInterface::ActionType DataSaver::actionType() const
{
    return SaveAction;
}

ControllerInterface::DataType DataSaver::dataType() const
{
    return FileData;
}

QString DataSaver::name() const
{
    return "DataSaver";
}

QString DataSaver::description() const
{
    return "Save data to file";
}

void DataSaver::init(domain::TaskManagerData* data, ui::UiInterface* ui,
                     ControllerFactory* factory)
{
    control::ControllerInterface::init(data, ui, factory);

    d = new DataSaverPrivate(data);
}

bool DataSaver::userIsValid() const
{
    return true;
}

void DataSaver::execute()
{
    QString fileName = ui->getSaveFileName("", "XML files(*.xml)");
    QFile file(fileName);

    if (!file.open(QFile::WriteOnly))
    {
        ui->showError("Failed to open file " + fileName);
        return;
    }

    QDomDocument doc("TaskPlanner");
    QDomElement root = doc.createElement("mop:taskplanner");
    doc.appendChild(root);

    d->createIds();
    root.appendChild(d->writeProjects(doc));
    root.appendChild(d->writeResources(doc));

    Q_FOREACH (domain::User* user, data->users())
        root.appendChild(d->writeUser(user, doc));

    qDebug() << doc.toString(4);

    QTextStream stream(&file);
    doc.save(stream, 4);
    file.close();

    ui->showMessage("Data successfully saved to " + fileName);
}

void DataSaverPrivate::createIds()
{
    idMap.clear();

    //projects
    int count = 0;
    Q_FOREACH (domain::Project* project, data->projects())
        idMap[project] = QString("project%1").arg(count++);

    //resources
    count = 0;
    Q_FOREACH (domain::Resource* resource, data->resources())
        idMap[resource] = QString("resource%1").arg(count++);

    //tasks
    count = 0;
    Q_FOREACH (domain::Task* task, data->tasks())
        idMap[task] = QString("task%1").arg(count++);
}

QDomElement DataSaverPrivate::writeProjects(QDomDocument doc)
{
    QDomElement projects = doc.createElement("mop:projects");

    Q_FOREACH (domain::Project* project, data->projects())
    {
        Q_ASSERT(idMap.contains(project));

        QDomElement projectElement = doc.createElement("mop:project");
        projects.appendChild(projectElement);

        projectElement.setAttribute("id", idMap[project]);
        projectElement.appendChild(createTextElement("mop:description",
                                                     project->description(),
                                                     doc));
    }

    return projects;
}

QDomElement DataSaverPrivate::writeResources(QDomDocument doc)
{
    QDomElement resources = doc.createElement("mop:resources");

    Q_FOREACH (domain::Resource* resource, data->resources())
    {
        QDomElement resourceElement = doc.createElement("mop:resource");
        resources.appendChild(resourceElement);

        resourceElement.setAttribute("id", idMap[resource]);
        resourceElement.appendChild(createTextElement("mop:description",
                                                      resource->description(),
                                                      doc));
        resourceElement.appendChild(createTextElement("mop:type",
                                                      resource->resourceTypeName(),
                                                      doc));
    }

    return resources;
}

QDomElement DataSaverPrivate::writeUser(domain::User* user, QDomDocument doc)
{
    QDomElement userElement = doc.createElement("mop:user");
    userElement.appendChild(createTextElement("mop:name", user->name(), doc));
    userElement.appendChild(writeTasks(user, doc));
    userElement.appendChild(writeReservations(user, doc));
    return userElement;
}

QDomElement DataSaverPrivate::writeTasks(domain::User* user, QDomDocument doc)
{
    QDomElement tasks = doc.createElement("mop:tasks");

    Q_FOREACH (const domain::Task* task, user->tasks())
    {
        QDomElement taskElement = doc.createElement("mop:task");
        tasks.appendChild(taskElement);

        taskElement.setAttribute("id", idMap[task]);

        taskElement.appendChild(createTextElement(
                            "mop:startDate",
                            task->schedule().startDate().toString(Qt::ISODate),
                            doc));
        taskElement.appendChild(createTextElement(
                            "mop:endDate",
                            task->schedule().dueDate().toString(Qt::ISODate),
                            doc));

        taskElement.appendChild(createTextElement(
                "mop:duration",
                QString::number(task->schedule().duration().totalSeconds() / 60),
                doc));

        taskElement.appendChild(createTextElement("mop:status",
                        domain::TaskState::nameForEnum(task->state()), doc));

        if (task->project() != 0)
        {
            taskElement.appendChild(createTextElement("mop:refProject",
                                                      idMap[task->project()],
                                                      doc));
        }

        QDomElement subTasks = doc.createElement("mop:dependsOn");
        taskElement.appendChild(subTasks);

        Q_FOREACH (const domain::Task* subTask, task->subTasks())
        {
            subTasks.appendChild(createTextElement("mop:refTask",
                                                   idMap[subTask],
                                                   doc));
        }
    }

    return tasks;
}

QDomElement DataSaverPrivate::writeReservations(domain::User*, QDomDocument)
{
//     QDomElement reservations = doc.createElement("mop:reservations");
// 
//     Q_FOREACH (const domain::Reservation* reservation, user->reservations())
//     {
//         QDomElement reservationElement = doc.createElement("mop:reservation");
//         reservations.appendChild(reservationElement);
// 
//         reservationElement.appendChild(createTextElement(
//                                     "mop:time",
//                                     reservation->time().toString(Qt::ISODate),
//                                     doc));
// 
//         reservationElement.appendChild(createTextElement(
//                     "mop:duration",
//                     QString::number(reservation->duration().totalSeconds()),
//                     doc));
// 
//         reservationElement.appendChild(createTextElement(
//                                                 "mop:refResource",
//                                                 idMap[reservation->resource()],
//                                                 doc));
//     }
// 
//     return reservations;
// TODO
    return QDomElement();
}

QDomElement DataSaverPrivate::createTextElement(const QString& name,
                                                const QString& value,
                                                QDomDocument& doc)
{
    QDomElement text = doc.createElement(name);
    text.appendChild(doc.createTextNode(value));
    return text;
}

Q_EXPORT_PLUGIN2(controller_datasaver, DataSaver)

---