#ifndef CONNECTION_H
#define CONNECTION_H

#include <QMessageBox>
#include <QSqlDatabase>

inline bool createConnection()
{
    QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE");
    db.setDatabaseName("../cdcollection/cdcollection.dat");
    if (!db.open()) {
        QMessageBox::warning(0, QObject::tr("Database Error"),
                             db.lastError().text());
        return false;
    }
    return true;
}

inline void createTables()
{
    QSqlQuery query;
    query.exec("CREATE TABLE artist ("
               "id INTEGER PRIMARY KEY, "
               "name VARCHAR(40) NOT NULL, "
               "country VARCHAR(40))");
    query.exec("CREATE TABLE cd ("
               "id INTEGER PRIMARY KEY, "
               "title VARCHAR(40) NOT NULL, "
               "artistid INTEGER NOT NULL, "
               "year INTEGER NOT NULL, "
               "FOREIGN KEY (artistid) REFERENCES artist)");
    query.exec("CREATE TABLE track ("
               "id INTEGER PRIMARY KEY, "
               "title VARCHAR(40) NOT NULL, "
               "duration INTEGER NOT NULL, "
               "cdid INTEGER NOT NULL)");
}

#endif