#include <QtGui/QApplication>
#include "mainwindow.h"
#include "accountswindow.h"
#include "globals.h"
#include "contacts.h"
#include "skynet.h"
#include "msguirouter.h"
#include "picaactioncenter.h"
#include "picasystray.h"
#include <QDir>
#include <QMessageBox>
#include <QString>
#include <QIcon>
#include "openssltool.h"
#include "../PICA_client.h"

//globals
QString config_dir;
QString config_dbname;
SkyNet *skynet;
MainWindow *mainwindow;
class AskPassword *askpassword;
class MsgUIRouter *msguirouter;

QIcon picapica_ico_sit;
QIcon picapica_ico_fly;

class PicaActionCenter *action_center;
class PicaSysTray *systray;

// /usr/share/pica-client/pica-pica_CA.pem
//QString config_defaultCA("/home/root_jr/files/projects/picapica_wc/localhost/project_picapica/trunk/tests/trusted_CA.pem");
#ifndef WIN32
QString config_defaultCA(PICA_INSTALLPREFIX"/share/pica-client/CA.pem");
#else
QString config_defaultCA("share\\CA.pem");
#endif

static bool create_database()
{
    QMessageBox msgBox;

    QSqlDatabase dbconn=QSqlDatabase::addDatabase("QSQLITE");
    dbconn.setDatabaseName(config_dbname);

    if (!dbconn.open())
    {

        msgBox.setText(dbconn.lastError().text());
        msgBox.exec();
        return false;
    }

    QSqlQuery query;

    query.exec("PRAGMA foreign_keys=ON;");

    if (query.lastError().isValid())
            goto showerror;

    query.exec("create table accounts \
               (\
                   id blob primary key, \
                   name varchar(64), \
                   cert_file varchar(255) not null, \
                   pkey_file varchar(255) not null, \
                   ca_file varchar(255) default null \
                   );");

    if (query.lastError().isValid())
                    goto showerror;

    query.exec("create table contacts \
               (\
                   id blob,\
                   name varchar(64), \
                   cert_pem text(2048), \
                   account_id blob not null, \
                   primary key(id, account_id),\
                   foreign key(account_id) references accounts(id) on delete cascade\
                );");

    if (query.lastError().isValid())
            goto showerror;

    query.exec(
                "create table nodes \
                (\
                    address varchar(255) not null,\
                    port int not null,\
                    last_active int not null,\
                    inactive_count int not null,\
                    constraint pk primary key (address,port) on conflict replace \
                 );");
     query.exec("insert into nodes values(\"picapica.im\", 2233, 0, 0);");
     query.exec("insert into nodes values(\"picapica.im\", 2299, 0, 0);");
     query.exec("insert into nodes values(\"picapica.ge\", 2299, 0, 0);");

     if (query.lastError().isValid())
             goto showerror;

     query.exec("create table history \
                (\
                    id integer primary key autoincrement, \
                    contact_id blob not null, \
                    account_id blob not null, \
                    timestamp int not null, \
                    is_me int not null, \
                    is_delivered int not null, \
                    message text(65536) not null, \
                    foreign key(account_id) references accounts(id) on delete cascade\
                );"
                 );

     if (query.lastError().isValid())
             goto showerror;

     query.exec("create table schema_version \
                  ( \
                    ver integer primary key, \
                    timestamp int not null \
                  );"
                );

     if (query.lastError().isValid())
             goto showerror;

     query.exec("insert into schema_version values (2, strftime('%s','now'));"); //current schema version. update when needed

    if (query.lastError().isValid())
    showerror:
    {
        msgBox.setText(query.lastError().text());
        msgBox.exec();
        return false;
    }

    return true;
}

static bool update_database()
{
    QMessageBox msgBox;
    int schema_ver;

    QSqlDatabase dbconn=QSqlDatabase::addDatabase("QSQLITE");
    dbconn.setDatabaseName(config_dbname);

    if (!dbconn.open())
    {

        msgBox.setText(dbconn.lastError().text());
        msgBox.exec();
        return false;
    }

    QSqlQuery query;

    //checking if table schema_version exists. If not, then database belongs to pica-client before 0.5.3 version
    query.exec("select count(*) from sqlite_master where name=\"schema_version\"");
    query.next();

    if (query.lastError().isValid())
        goto showerror;

    if (query.value(0).toInt() == 0)
    {
        //--------updating from pre0.5.3 pica-client database
        msgBox.setText("Updating database to newer version");
        msgBox.exec();

        query.exec("create table history \
                   (\
                       id integer primary key autoincrement, \
                       contact_id int not null, \
                       account_id int not null, \
                       timestamp int not null, \
                       is_me int not null, \
                       is_delivered int not null, \
                       message text(65536) not null, \
                       foreign key(account_id) references accounts(id) on delete cascade\
                   );"
                    );

        if (query.lastError().isValid())
                goto showerror;

        query.exec("\
                   create table schema_version \
                     ( \
                       ver integer primary key, \
                       timestamp int not null \
                     );"
                   );

        if (query.lastError().isValid())
                goto showerror;

        query.exec("insert into schema_version values (1, strftime('%s','now'));");
        //--------------------------------------------------
        schema_ver = 1;
    }
    else
    {
        query.exec("select max(ver) from schema_version");
        query.next();

        if (query.lastError().isValid())
                goto showerror;

        schema_ver = query.value(0).toInt();
    }

    if (schema_ver == 1)
    {
        query.exec("PRAGMA foreign_keys=OFF;");

        if (query.lastError().isValid())
            goto showerror;

        query.exec("alter table accounts rename to accounts_old_schemav1;");

        if (query.lastError().isValid())
            goto showerror;

        query.exec("alter table contacts rename to contacts_old_schemav1;");

        if (query.lastError().isValid())
            goto showerror;

        query.exec("alter table history rename to history_old_schemav1;");

        if (query.lastError().isValid())
            goto showerror;

        query.exec("create table accounts \
               (\
                   id blob primary key, \
                   name varchar(64), \
                   cert_file varchar(255) not null, \
                   pkey_file varchar(255) not null, \
                   ca_file varchar(255) default null \
                   );"
                   );

                    if (query.lastError().isValid())
                    goto showerror;

        query.exec("create table contacts \
                     (\
                      id blob,\
                      name varchar(64), \
                      cert_pem text(2048), \
                      account_id blob not null, \
                      primary key(id, account_id),\
                      foreign key(account_id) references accounts(id) on delete cascade\
                      );"
                      );

        if (query.lastError().isValid())
            goto showerror;

        query.exec("create table history \
                (\
                    id integer primary key autoincrement, \
                    contact_id blob not null, \
                    account_id blob not null, \
                    timestamp int not null, \
                    is_me int not null, \
                    is_delivered int not null, \
                    message text(65536) not null, \
                    foreign key(account_id) references accounts(id) on delete cascade\
                );"
                 );
        if (query.lastError().isValid())
            goto showerror;

        query.exec("alter table accounts_old_schemav1 add column new_id blob;");
        query.exec("alter table contacts_old_schemav1 add column new_id blob;");

        if (query.lastError().isValid())
            goto showerror;

        query.exec("select id, cert_file from accounts_old_schemav1");
        if (query.lastError().isValid())
            goto showerror;

        while(query.next())
        {
            quint32 old_id;
            QString cert_file;
            QSqlQuery q;
            QByteArray new_id(PICA_ID_SIZE, 0);

            old_id = query.value(0).toUInt();
            cert_file = query.value(1).toString();

            if (PICA_get_id_from_cert_file(cert_file.toUtf8().constData(), (unsigned char*)new_id.data()))
            {
                QSqlQuery q;

                q.prepare("update accounts_old_schemav1 set new_id = :new_id where id = :id");
                q.bindValue(":new_id", new_id);
                q.bindValue(":id", old_id);
                q.exec();
            }
        }

        query.exec("select id, cert_pem from contacts_old_schemav1");
        if (query.lastError().isValid())
            goto showerror;

        while(query.next())
        {
            quint32 old_id;
            QString cert_pem;
            QSqlQuery q;
            QByteArray new_id(PICA_ID_SIZE, 0);

            old_id = query.value(0).toUInt();
            cert_pem = query.value(1).toString();

            if (PICA_get_id_from_cert_string(cert_pem.toUtf8().constData(), (unsigned char *)new_id.data()))
            {
                QSqlQuery q;

                q.prepare("update contacts_old_schemav1 set new_id = :new_id where id = :id");
                q.bindValue(":new_id", new_id);
                q.bindValue(":id", old_id);
                q.exec();
            }
        }


        query.exec("insert into accounts select new_id, name, cert_file, pkey_file, ca_file from accounts_old_schemav1 where new_id is not null");

        if (query.lastError().isValid())
            goto showerror;

        query.exec("insert into contacts select c.new_id, c.name, c.cert_pem, a.new_id from accounts_old_schemav1 as a, contacts_old_schemav1 as c on c.account_id = a.id where c.new_id is not null and a.new_id is not null");

        if (query.lastError().isValid())
            goto showerror;

        query.exec("insert into history select h.id, c.new_id, a.new_id, h.timestamp, h.is_me, h.is_delivered, h.message \
                    from history_old_schemav1 as h, accounts_old_schemav1 as a, contacts_old_schemav1 as c on h.contact_id = c.id and h.account_id = a.id and a.id = c.account_id order by h.id");

        if (query.lastError().isValid())
            goto showerror;

        query.exec("insert into schema_version values (2, strftime('%s','now'));");
        schema_ver = 2;
    }

if (query.lastError().isValid())
showerror:
    {
        msgBox.setText(query.lastError().text() + " " + query.lastQuery());
        msgBox.exec();
        return false;
    }

    return true;
}

static bool create_config_dir()
{
    QMessageBox msgBox;
    config_dir=QDir::homePath()+QDir::separator() +QString(PICA_CLIENT_CONFIGDIR);

    if (!QFile::exists(config_dir))
    {
        QDir dir;

        if (!dir.mkpath(config_dir))
        {
            msgBox.setText(QString(QObject::tr("Unable to create configuration directory:\n %1")).arg(config_dir));
            msgBox.exec();
            return false;
        }
    }

    config_dbname=config_dir+QDir::separator()+QString(PICA_CLIENT_STORAGEDB);

    if (!QFile::exists(config_dbname))
    {
        if (!create_database())
            return false;
    }
    else
        update_database();

    if (!QFile::exists(config_dir + QDir::separator() + PICA_CLIENT_DHPARAMFILE))
    {
        OpenSSLTool osslt;
        msgBox.setText(QString(QObject::tr("Diffie-Hellman parameters will be generated. Please wait. This process can take several minutes...")));
        msgBox.exec();

        if (!osslt.GenDHParam(4096, config_dir + QDir::separator() + PICA_CLIENT_DHPARAMFILE))
        {
            msgBox.setText(QString(QObject::tr("Failed to generate Diffie-Hellman parameters")));
            msgBox.exec();
            return false;
        }
    }

    return true;
}

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);

    if (!create_config_dir())
        return -1;
#ifndef WIN32
    picapica_ico_sit = QIcon(PICA_INSTALLPREFIX"/share/pica-client/picapica-icon-sit.png");
    picapica_ico_fly = QIcon(PICA_INSTALLPREFIX"/share/pica-client/picapica-icon-fly.png");
#else
	picapica_ico_sit = QIcon("share\\picapica-icon-sit.png");
    picapica_ico_fly = QIcon("share\\picapica-icon-fly.png");
#endif

    PicaActionCenter ac;
    action_center = &ac;

    AccountsWindow aw;

    SkyNet s;
    skynet=&s;

    PicaSysTray pst;
    systray = &pst;

    AskPassword askpsw;
    askpassword=&askpsw;

    MsgUIRouter muir;
    msguirouter = &muir;


    aw.show();

    a.setQuitOnLastWindowClosed(false);

    return a.exec();
}