#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;
    mView->setModel(mViewModel);
    mSearchModel = new QStandardItemModel;

    //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::populateByArtist(QStandardItem *parent, const QString &filter){
    QSqlDatabase db = QSqlDatabase::database("beetplayerdb");
    QStandardItem *root = parent;
    QSqlQuery artistsQ(db);
    if(!filter.isEmpty()){
        artistsQ.prepare("SELECT iartists_id, tartists_name FROM artists WHERE tartists_name ~ :f ORDER BY tartists_name ASC");
        artistsQ.bindValue(":f", filter);
    }else{
        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");

    //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::populateBySong(QStandardItem *parent, const QString &filter){
    QSqlDatabase db = QSqlDatabase::database("beetplayerdb");
    QStandardItem *root = parent;
    QSqlQuery songQ = QSqlQuery(db);
    songQ.prepare("SELECT sipos, ttitle, tfullpath, igenres_id, artists.tartists_name FROM songs, artists WHERE ttitle ~ :f AND songs.iartists_id = artists.iartists_id ORDER BY ttitle ASC");
    songQ.bindValue(":f", filter);
    songQ.exec();
    while(songQ.next()){
        QStandardItem *curSong = new QStandardItem;
        curSong->setFont(QFont("courier"));
        QString songText = QString(tr("%1 %2 (%3)")).arg(QString("🅢")).arg(songQ.value(1).toString()).arg(songQ.value(4).toString());
        curSong->setText(songText);
        curSong->setData(songQ.value(0), IdRole);
        curSong->setData(songQ.value(2), FullPathRole);
        curSong->setData(songQ.value(3), GenreRole);
        root->appendRow(curSong);
    }
}

void PlayerWidget::populateByGenre(QStandardItem *parent, const QString &filter){
    QSqlDatabase db = QSqlDatabase::database("beetplayerdb");
    QStandardItem *root = parent;
    QSqlQuery genreQ(db);
    genreQ.prepare("SELECT igenres_id, tgenres_name FROM genres WHERE tgenres_name ~ :f");
    genreQ.bindValue(":f", filter);
    genreQ.exec();
    while(genreQ.next()){
        QStandardItem *curGenre = new QStandardItem;
        curGenre->setFont(QFont("courier"));
        QString genreText = QString(tr("%1 %2")).arg(QString("🅖")).arg(genreQ.value(1).toString());
        curGenre->setText(genreText);
        curGenre->setData(Genre, TypeRole);
        curGenre->setData(genreQ.value(0), IdRole);
        root->appendRow(curGenre);
        QSqlQuery songQ = QSqlQuery(db);
        songQ.prepare("SELECT sipos, ttitle, tfullpath, igenres_id, artists.tartists_name FROM songs, artists WHERE igenres_id = :id AND songs.iartists_id = artists.iartists_id ORDER BY ttitle ASC");
        songQ.bindValue(":id", genreQ.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(1).toString()).arg(songQ.value(4).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);
            curGenre->appendRow(curSong);
        }
    }
}

void PlayerWidget::populate(){
    mView->setModel(mViewModel);
    mViewModel->clear();
    mViewModel->setHorizontalHeaderLabels(QStringList() << tr("Name"));
    QStandardItem *root = mViewModel->invisibleRootItem();
    populateByArtist(root, QString());
}

void PlayerWidget::doFilter(){
    QString filter = mFilter->text();
    if(filter.isEmpty()){
        mView->setModel(mViewModel);
    }
    mSearchModel->clear();
    mSearchModel->setHorizontalHeaderLabels(QStringList() << tr("Name"));
    mView->setModel(mSearchModel);
    QStandardItem *root = mSearchModel->invisibleRootItem();
    populateByArtist(root, filter);
    populateByGenre(root, filter);
    populateBySong(root, filter);
}

void PlayerWidget::clearFilter(){
    mFilter->clear();
    mView->setModel(mViewModel);
}