/*
  This program 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
  2 of the License, or (at your option) any later version.
*/

#ifndef DBANALYZER_H
#define DBANALYZER_H

#include <QThread>
#include <QVariant>
#include <QList>
#include <QHash>
#include <QDialog>
#include <QSqlDatabase>
#include <QMutex>
#include <QModelIndex>

class QSqlQuery;
class QTabWidget;
class QTreeView;
class QPushButton;
class SmTreeModel;
class DbAnalyzer;
class QLabel;
class QStackedLayout;

class DbAnalyzerDialog : public QDialog {
    Q_OBJECT
    public:
        enum Mode { Actors, Genres };
        explicit DbAnalyzerDialog(QWidget *parent = 0, Qt::WindowFlags f = 0);

     public slots:
        void refresh();

    signals:
        void partClicked(int seriesPartId, int seriesId);
        void delItems(int mode, QList<int> &ids);

    private slots:
        void cancelAnalyzer();
        void analyzerStarted();
        void analyzerFinished();
        void noDataDoubleClicked(const QModelIndex &);
        void deleteItems();
        void tabChanged(int index);
        void setMarks();
        void deleteMarks();

    private:
        void populate(QTreeView *view ,SmTreeModel *model, const QList<QList<QVariant> > &data, const QHash<int, QVariant> &marks = QHash<int, QVariant>());
        const QList<int> currentIds() const;
        QTabWidget *mTab;
        QTreeView *mNoActorsV;
        QTreeView *mNoCoversV;
        QTreeView *mStrayActorsV;
        QTreeView *mStrayGenresV;
        SmTreeModel *mNoActorsM;
        SmTreeModel *mNoCoversM;
        SmTreeModel *mStrayActorsM;
        SmTreeModel *mStrayGenresM;
        QPushButton *mClose;
        QPushButton *mCancel;
        QPushButton *mMark;
        QPushButton *mRefresh;
        QPushButton *mDelete;
        QLabel *mTotal;
        QList<QString> mTotals;
        DbAnalyzer *mAnalyzer;
        int mMarkMode;
        QTreeView *mCurrentView;
        QStackedLayout *mButtonStack;
};

class DbAnalyzer : public QThread {
    Q_OBJECT
    public:
        enum Status { Ok, Fail };
        enum Marks { NOMARKS = 0, MARKS_ACTORS = 1, MARKS_COVERS = 2 };
        explicit DbAnalyzer(QObject *parent = 0);
        ~DbAnalyzer();
        int status() { return mStatus; }
        const QList<QList<QVariant> > noActors() const { return mNoActorR; }
        const QHash<int, QVariant> actorMarks() const { return mActorMarks; }
        const QList<QList<QVariant> > noCovers() const { return mNoCoverR; }
        const QHash<int, QVariant> coverMarks() const { return mCoverMarks; }
        const QList<QList<QVariant> > strayActors() { return mStrayActorR; }
        const QList<QList<QVariant> > strayGenres() { return mStrayGenresR; }

    public slots:
        void setCancel(bool canceled);
        void setMarks(const QList<int> &ids, int reason, int mark_id);
        void deleteMarks(const QList<int> &ids, int reason);

    signals:
        void message(const QString &);

    protected:
        void run();

    private:
        const QList<QList<QVariant> > noDataCheck(QSqlQuery *query);
        const QList<QList<QVariant> > strayCheck(QSqlQuery *query);
        QHash<int, QVariant> marks(int markType);
        void setStatus(int status);
        QSqlDatabase mDb;
        QSqlQuery *mNoActorQuery;
        QSqlQuery *mNoCoverQuery;
        QSqlQuery *mStrayActorsQuery;
        QSqlQuery *mStrayGenresQuery;
        QSqlQuery *mMarksQuery;
        QSqlQuery *mSetMarksQuery;
        QSqlQuery *mDeleteMarksQuery;
        QList<QList<QVariant> > mNoActorR;
        QList<QList<QVariant> > mNoCoverR;
        QList<QList<QVariant> > mStrayActorR;
        QList<QList<QVariant> > mStrayGenresR;
        QHash<int, QVariant> mActorMarks;
        QHash<int, QVariant> mCoverMarks;
        QMutex mCancelMutex;
        QMutex mStatusMutex;
        bool mCanceled;
        int mStatus;
};

#endif // DBANALYZER_H