#ifndef STORAGE_H
#define STORAGE_H

#include <QString>
#include <QMap>
#include "task.h"
#include "SQLiteCpp/Database.h"

class Storage
{
public:
    Storage();
    ~Storage();

    QString path();
    void setPath(const QString& path);
    QString key();
    void setKey(const QString& key);

    SQLite::Database& database();

    // New database. Returns true or false depending on success of call.
    bool create();

    // Open database, preload task tree. Returns true or false depending on success of operation.
    bool open();

    // Close database
    void close();

    // Ensure database is latest version
    bool upgrade();

    PTask createTask(int index);
    PTask createTask(const PTask& parent, int index);
    // PTask loadTask(Task::Id id, PTask parent);

    // Remove task from database
    enum DeleteOption
    {
        DeleteOption_Total,
        DeleteOption_FromParentAndHash,
        DeleteOption_FromParent
    };

    void deleteTask(const PTask& task, DeleteOption option);

    // Undelete task
    void undeleteTask(const PTask& task);

    // Remove task from internal dictionaries - it makes task unadressable by its id/model id
    void removeTask(const PTask& task);

    void loadTaskContent(Task& task);
    void loadTaskMetadata(Task& task);

    // Save changes in single task
    enum SaveOptions
    {
        Save_Automatic,
        Save_Forced
    };
    void saveTask(const Task& task, SaveOptions options = Save_Automatic);

    // Save Task hierarchy
    enum Depth
    {
        depthSingleTask,
        depthRecursive
    };
    void saveTask(PTask task, Depth depth);

    int findAttachmentCountOnTask(const Task& t);

    // Move task
    bool moveTask(PTask task, PTask newParent, int indexToInsert);

    bool isOpened();
    void save();

    TaskArray& topOfTaskTree();
    PTask findTaskByModelId(Task::ModelId id);
    PTask findTaskById(Task::Id id);
    int findTaskIndexInParent(PTask task);

    // Attachments
    void loadAttachments(PTask task, AttachmentArray& output);
    void deleteAttachment(PAttachment att);
    void undeleteAttachment(PAttachment att);
    Id   saveMetadata(const Attachment& attachment);
    void saveContent(const Attachment& attachment, const QByteArray& content);
    QByteArray loadContent(const Attachment& attachment);

    // Timeline
    Id saveTimeRecord(const TimeRecord& r);
    void deleteTimeRecord(const TimeRecord& r);

    void loadTimeLine(TimeLine& l);
    void saveTimeLime(const TimeLine& l);

    static Storage& instance();

protected:
    QString mPath, mKey;
    QSharedPointer<SQLite::Database> mDatabase;
    TaskArray mTopTasks;
    typedef QMap<Task::ModelId, PTask> TaskModelIdMap;
    TaskModelIdMap mTaskModelIdMap;
    typedef QMap<Task::Id, PTask> TaskIdMap;
    TaskIdMap mTaskIdMap;

    TaskArray mTaskToUpgrade;

    bool hasTable(const QString& tablename);
    void saveSingleTask(PTask task);
    void loadTaskRecord(Task& t, SQLite::Statement& q);
    void loadTaskTree();
    void loadTaskChildren(PTask task);
    bool upgradeFromVersion0();
    bool encryptTask(PTask task);
    bool encryptTaskContent(PTask task);
    bool encryptTaskAttachment(PAttachment attachment);
};

#endif // STORAGE_H