#include <QGroupBox>
#include <QLineEdit>
#include <QPushButton>
#include <QLabel>
#include <QTextEdit>
#include <QSlider>
#include <QTreeView>
#include <QSplitter>
#include <QStandardItemModel>
#include <QStandardItem>
#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QSqlDatabase>
#include <QSqlQuery>

#include "playerwidget.h"
#include "beetplayerproxy.h"

PlayerWidget::PlayerWidget(QWidget *parent) : QWidget(parent){
    setupGui();
}

void PlayerWidget::setupGui(){
    //THE view
    mView = new QTreeView;
    mViewModel = new QStandardItemModel;
    mViewProxy = new BeetPlayerProxy;
    mViewProxy->setSourceModel(mViewModel);
    mView->setModel(mViewProxy);

    //filter
    QGroupBox *filterGB = new QGroupBox(tr("Filter"));
    mFilter = new QLineEdit;
    connect(mFilter, SIGNAL(returnPressed()), this, SLOT(doFilter()));
    QPushButton *filterB = new QPushButton(tr("Go!"));
    connect(filterB, SIGNAL(clicked()), this, SLOT(doFilter()));
    QPushButton *clearFilterB = new QPushButton(tr("Clear"));
    connect(clearFilterB, SIGNAL(clicked()), this, SLOT(clearFilter()));
    QHBoxLayout *filterLayout = new QHBoxLayout;
    filterLayout->addWidget(mFilter);
    filterLayout->addWidget(filterB);
    filterLayout->addWidget(clearFilterB);
    filterGB->setLayout(filterLayout);

    //left widget
    QWidget *leftWidget = new QWidget;
    QVBoxLayout *leftWidgetL = new QVBoxLayout;
    leftWidgetL->addWidget(filterGB);
    leftWidgetL->addWidget(mView);
    leftWidget->setLayout(leftWidgetL);

    //now playing label
    mNowPlayingL = new QLabel;
    mNowPlayingL->setAlignment(Qt::AlignCenter);
    mNowPlayingL->setFont(QFont("courier new", 20, QFont::Bold));
    mNowPlayingL->setText(tr("(none)"));

    //slider
    mSlider = new QSlider;
    mSlider->setOrientation(Qt::Horizontal);

    //current info
    QGroupBox *currentInfoGB = new QGroupBox(tr("Current"));
    mCurrentTE = new QTextEdit;
    mCurrentTE->setReadOnly(true);
    QVBoxLayout *currentInfoL = new QVBoxLayout;
    currentInfoL->addWidget(mCurrentTE);
    currentInfoGB->setLayout(currentInfoL);

    //center widget
    QWidget *centerWidget = new QWidget;
    QVBoxLayout *centerWidgetL = new QVBoxLayout;
    centerWidgetL->addWidget(mNowPlayingL);
    centerWidgetL->addWidget(mSlider);
    centerWidgetL->addWidget(currentInfoGB);
    centerWidget->setLayout(centerWidgetL);

    //playlist
    mPlayListModel = new QStandardItemModel;
    mPlayListView = new QTreeView;
    mPlayListView->setModel(mPlayListModel);
    QGroupBox *playListGB = new QGroupBox(tr("Playlist"));
    QVBoxLayout *playListL = new QVBoxLayout;
    playListL->addWidget(mPlayListView);
    playListGB->setLayout(playListL);

    //right widget
    QWidget *rightWidget = new QWidget;
    QVBoxLayout *rightWidgetL = new QVBoxLayout;
    rightWidgetL->addWidget(playListGB);
    rightWidget->setLayout(rightWidgetL);

    //put it all together
    QSplitter *splitter = new QSplitter;
    splitter->addWidget(leftWidget);
    splitter->addWidget(centerWidget);
    splitter->addWidget(rightWidget);
    QVBoxLayout *mainLayout = new QVBoxLayout;
    mainLayout->addWidget(splitter);
    setLayout(mainLayout);
    populate();
}

void PlayerWidget::populate(){
    //prepare Queries
    QSqlDatabase db = QSqlDatabase::database("beetplayerdb");
    QSqlQuery artistsQ(db);
    artistsQ.prepare("SELECT iartists_id, tartists_name FROM artists ORDER BY tartists_name ASC");
    QSqlQuery albumQ(db);
    albumQ.prepare("SELECT DISTINCT(songs.ialbums_id), talbum_name, siyear FROM songs, albums WHERE songs.iartists_id = :artistid AND songs.ialbums_id = albums.ialbums_id ORDER BY siyear ASC");
    QSqlQuery songQ(db);
    songQ.prepare("SELECT sipos, ttitle, tfullpath, igenres_id FROM songs WHERE ialbums_id = :alid AND iartists_id = :arid ORDER BY sipos ASC");

    //reset view+model
    mView->setSortingEnabled(false);
    mViewModel->clear();
    mViewModel->setHorizontalHeaderLabels(QStringList() << tr("Name"));
    QStandardItem *root = mViewModel->invisibleRootItem();

    //read data
    artistsQ.exec();
    while(artistsQ.next()){
        QStandardItem *curArtist = new QStandardItem;
        curArtist->setFont(QFont("courier"));
        QString artistText = QString(tr("%1 %2")).arg(QString("🅐")).arg(artistsQ.value(1).toString()); //I guess this will bite me!
        curArtist->setText(artistText);
        curArtist->setData(Artist, TypeRole);
        curArtist->setData(artistsQ.value(0).toInt(), IdRole);
        root->appendRow(curArtist);
        albumQ.bindValue(":artistid", artistsQ.value(0));
        albumQ.exec();
        while(albumQ.next()){
            QStandardItem *curAlbum = new QStandardItem;
            curAlbum->setFont(QFont("courier"));
            QString albumText = QString(tr("%1 %2 - %3")).arg(QChar(0x29BE)).arg(QString::number(albumQ.value(2).toInt())).arg(albumQ.value(1).toString());
            curAlbum->setText(albumText);
            curAlbum->setData(Album, TypeRole);
            curAlbum->setData(albumQ.value(0), IdRole);
            curArtist->appendRow(curAlbum);
            songQ.bindValue(":alid", albumQ.value(0));
            songQ.bindValue(":arid", artistsQ.value(0));
            songQ.exec();
            while(songQ.next()){
                QStandardItem *curSong = new QStandardItem;
                curSong->setFont(QFont("courier"));
                QString songText = QString(tr("%1 %2 - %3")).arg(QChar(0x266C)).arg(songQ.value(0).toInt(), 3, 10, QChar('0')).arg(songQ.value(1).toString());
                curSong->setText(songText);
                curSong->setData(Song, TypeRole);
                curSong->setData(songQ.value(0), IdRole);
                curSong->setData(songQ.value(2), FullPathRole);
                curSong->setData(songQ.value(3), GenreRole);
                curAlbum->appendRow(curSong);
            }
        }
    }
}

void PlayerWidget::doFilter(){
    mViewProxy->setFilterRegExp(QRegExp(mFilter->text()));
}

void PlayerWidget::clearFilter(){
    mFilter->setText(QString());
    doFilter();
}