#include "profileselect.h"
#include <QDebug>
#include <QStringListModel>
#include <QStandardItemModel>
#include <QStandardItem>
#include <QDialog>
#include <QLineEdit>
#include <QVBoxLayout>
#include <QCryptographicHash>
#include <QMessageBox>
#include <QTimer>
#include "ui_profileselect.h"
#include "SleepLib/profiles.h"
#include "newprofile.h"

ProfileSelect::ProfileSelect(QWidget *parent) :
    QDialog(parent),
    ui(new Ui::ProfileSelect)
{
    ui->setupUi(this);
    //ui->listView->setViewMode(QListView::IconMode);
    QStringList str;
    QStandardItemModel *model=new QStandardItemModel (0,0);
    //QList<QStandardItem *> items;


    int i=0;
    int sel=-1;
    QString name;
    for (QHash<QString,Profile *>::iterator p=Profiles::profiles.begin();p!=Profiles::profiles.end();p++) {
        //str.append(p.key());
        Profile &profile=**p;
        name=p.key();
       // if (!profile["FirstName"].toString().isEmpty())
       //     name+=" ("+profile["FirstName"].toString()+" "+profile["LastName"].toString()+")";
        QStandardItem *item=new QStandardItem(*new QIcon(":/icons/moon.png"),name);
        if (pref.Exists("Profile") && (name==pref["Profile"].toString())) {
            sel=i;
        }
        item->setData(p.key());
        item->setEditable(false);
        item->setFont(QFont("Sans Serif",18,QFont::Bold,false));
        model->appendRow(item);
        i++;
    }
    ui->listView->setModel(model);
    ui->listView->setSelectionBehavior(QAbstractItemView::SelectRows);
    ui->listView->setSelectionMode(QAbstractItemView::SingleSelection);
    if (sel>=0) ui->listView->setCurrentIndex(model->item(sel)->index());
    m_tries=0;

    pref["SkipLogin"]=false;
    if ((i==1) && pref["SkipLogin"].toBool()) {
        if (!Profiles::profiles.contains(name))
            pref["Profile"]=name;
        QTimer::singleShot(0,this,SLOT(earlyExit()));
        hide();
    }
}

ProfileSelect::~ProfileSelect()
{
    delete ui;
}
void ProfileSelect::earlyExit()
{
    accept();
}
void ProfileSelect::on_selectButton_clicked()
{
    on_listView_activated(ui->listView->currentIndex());
}

void ProfileSelect::on_newProfileButton_clicked()
{
    NewProfile newprof(this);
    newprof.skipWelcomeScreen();
    newprof.exec();
    accept();
}

void ProfileSelect::on_listView_activated(const QModelIndex &index)
{
    QString name=index.data().toString();
    Profile *profile=Profiles::profiles[name];
    if (!profile) return;
    if (!profile->Exists("Password")) {
        m_selectedProfile=name;
        pref["Profile"]=name;
    } else {
        QDialog dialog(this,Qt::Dialog);
        QLineEdit *e=new QLineEdit(&dialog);
        e->setEchoMode(QLineEdit::Password);
        dialog.connect(e,SIGNAL(returnPressed()),&dialog,SLOT(accept()));
        dialog.setWindowTitle("Enter Password");
        QVBoxLayout *lay=new QVBoxLayout();
        dialog.setLayout(lay);
        lay->addWidget(e);
        dialog.exec();
        QByteArray ba=e->text().toUtf8();
        if (QCryptographicHash::hash(ba,QCryptographicHash::Sha1).toHex()==(*profile)["Password"]) {
            m_selectedProfile=name;
            pref["Profile"]=name;
        } else {
            m_tries++;
            if (m_tries>2) {
                QMessageBox::warning(this,"Error","You entered an Incorrect Password too many times. Exiting!",QMessageBox::Ok);
                this->reject();
            } else {
                QMessageBox::warning(this,"Error","Incorrect Password",QMessageBox::Ok);
            }
            return;
        }
    }
    accept();
}