/***************************************************************************
 *   This file is part of awesome-widgets                                  *
 *                                                                         *
 *   awesome-widgets is free software: you can redistribute it and/or      *
 *   modify it under the terms of the GNU General Public License as        *
 *   published by the Free Software Foundation, either version 3 of the    *
 *   License, or (at your option) any later version.                       *
 *                                                                         *
 *   awesome-widgets 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 General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with awesome-widgets. If not, see http://www.gnu.org/licenses/  *
 ***************************************************************************/

#ifndef EXTITEMAGGREGATOR_H
#define EXTITEMAGGREGATOR_H

#include <QDebug>
#include <QDir>
#include <QInputDialog>
#include <QLineEdit>
#include <QSettings>
#include <QStandardPaths>

#include <pdebug/pdebug.h>

#include "abstractextitemaggregator.h"


template <class T> class ExtItemAggregator : public AbstractExtItemAggregator
{
public:
    explicit ExtItemAggregator(QWidget *parent, const QString type,
                               const bool debugCmd = false)
        : AbstractExtItemAggregator(parent, debugCmd),
          debug(debugCmd),
          m_type(type)
    {
        initItems();
    };

    ~ExtItemAggregator()
    {
        if (debug) qDebug() << PDEBUG;

        m_items.clear();
    }

    QString getName() const
    {
        if (debug) qDebug() << PDEBUG;

        bool ok;
        QString name = QInputDialog::getText(nullptr, tr("Enter file name"),
                                             tr("File name"), QLineEdit::Normal,
                                             QString(""), &ok);
        if ((!ok) || (name.isEmpty())) return QString("");
        if (!name.endsWith(QString(".desktop"))) name += QString(".desktop");

        return name;
    };

    void editItems()
    {
        if (debug) qDebug() << PDEBUG;

        repaint();
        int ret = dialog->exec();
        if (debug) qDebug() << PDEBUG << ":" << "Dialog returns" << ret;
    };

    void initItems()
    {
        if (debug) qDebug() << PDEBUG;

        m_items.clear();
        m_items = getItems();
    };

    T *itemFromWidget() const
    {
        if (debug) qDebug() << PDEBUG;

        QListWidgetItem *item = widgetDialog->currentItem();
        if (item == nullptr) return nullptr;

        int originalItem = -1;
        for (int i=0; i<m_items.count(); i++) {
            if (m_items[i]->fileName() != item->text()) continue;
            originalItem = i;
            break;
        }

        return originalItem == -1 ? nullptr : m_items[originalItem];
    };

    QList<T *> items() const
    {
        if (debug) qDebug() << PDEBUG;

        return m_items;
    };

    int uniqNumber() const
    {
        if (debug) qDebug() << PDEBUG;

        QList<int> tagList;
        for (int i=0; i<m_items.count(); i++)
            tagList.append(m_items[i]->number());
        int number = 0;
        while (tagList.contains(number))
            number++;

        return number;
    };

private:
    bool debug;
    QList<T *> m_items;
    QString m_type;
    // init method
    QList<T *> getItems()
    {
        if (debug) qDebug() << PDEBUG;

        // create directory at $HOME
        QString localDir = QString("%1/awesomewidgets/%2")
                           .arg(QStandardPaths::writableLocation(QStandardPaths::GenericDataLocation))
                           .arg(m_type);
        QDir localDirectory;
        if (localDirectory.mkpath(localDir))
            if (debug) qDebug() << PDEBUG << ":" << "Created directory" << localDir;

        QStringList dirs = QStandardPaths::locateAll(QStandardPaths::GenericDataLocation,
                           QString("awesomewidgets/%1").arg(m_type),
                           QStandardPaths::LocateDirectory);
        QStringList names;
        QList<T *> items;
        for (int i=0; i<dirs.count(); i++) {
            QStringList files = QDir(dirs[i]).entryList(QDir::Files, QDir::Name);
            for (int j=0; j<files.count(); j++) {
                if (!files[j].endsWith(QString(".desktop"))) continue;
                if (names.contains(files[j])) continue;
                if (debug) qDebug() << PDEBUG << ":" << "Found file" << files[j] << "in" << dirs[i];
                names.append(files[j]);
                items.append(new T(this, files[j], dirs, debug));
            }
        }

        return items;
    };

    void repaint()
    {
        widgetDialog->clear();
        for (int i=0; i<m_items.count(); i++) {
            QListWidgetItem *item = new QListWidgetItem(m_items[i]->fileName(), widgetDialog);
            QStringList tooltip;
            tooltip.append(tr("Name: %1").arg(m_items[i]->name()));
            tooltip.append(tr("Comment: %1").arg(m_items[i]->comment()));
            tooltip.append(tr("Identity: %1").arg(m_items[i]->uniq()));
            item->setToolTip(tooltip.join(QChar('\n')));
            widgetDialog->addItem(item);
        }
    };
    // methods
    void copyItem()
    {
        if (debug) qDebug() << PDEBUG;

        T *source = itemFromWidget();
        QString fileName = getName();
        int number = uniqNumber();
        if ((source == nullptr) || (fileName.isEmpty())) return;

        T *newItem = source->copy(fileName, number);
        if (newItem->showConfiguration() == 1) {
            initItems();
            repaint();
        }
    };

    void createItem()
    {
        if (debug) qDebug() << PDEBUG;

        QString fileName = getName();
        int number = uniqNumber();
        QStringList dirs = QStandardPaths::locateAll(QStandardPaths::GenericDataLocation,
                           QString("awesomewidgets/%1").arg(m_type),
                           QStandardPaths::LocateDirectory);
        if (fileName.isEmpty()) return;

        T *newItem = new T(this, fileName, dirs, debug);
        newItem->setNumber(number);
        if (newItem->showConfiguration() == 1) {
            initItems();
            repaint();
        }
    };

    void deleteItem()
    {
        if (debug) qDebug() << PDEBUG;

        T *source = itemFromWidget();
        if (source == nullptr) return;

        if (source->tryDelete()) {
            initItems();
            repaint();
        }
    };

    void editItem()
    {
        if (debug) qDebug() << PDEBUG;

        T *source = itemFromWidget();
        if (source == nullptr) return;

        if (source->showConfiguration() == 1) {
            initItems();
            repaint();
        }
    };
};


#endif /* EXTITEMAGGREGATOR_H */