Initial implementation DbExecutable: data base functionality for
[emufront] / src / db / databasemanager.cpp
1 // EmuFront
2 // Copyright 2010 Mikko Keinänen
3 //
4 // This file is part of EmuFront.
5 //
6 //
7 // EmuFront is free software: you can redistribute it and/or modify
8 // it under the terms of the GNU General Public License as published by
9 // the Free Software Foundation, either version 3 of the License, or
10 // (at your option) any later version.
11 //
12 // EmuFront is distributed in the hope that it will be useful,
13 // but WITHOUT ANY WARRANTY; without even the implied warranty of
14 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 // GNU General Public License for more details.
16 //
17 // You should have received a copy of the GNU General Public License
18 // along with EmuFront.  If not, see <http://www.gnu.org/licenses/>.
19
20 #include "databasemanager.h"
21 #include <QObject>
22 #include <QSqlDatabase>
23 #include <QSqlTableModel>
24 #include <QSqlError>
25 #include <QSqlQuery>
26 #include <QSqlRecord>
27 #include <QFile>
28 #include <QDir>
29 #include <QVariant>
30 #include <QDebug>
31 #include <QDateTime>
32
33 const QString DatabaseManager::DB_FILENAME = QString("my.db.sqlite");
34 const QString DatabaseManager::DATABASE = QString("QSQLITE");
35 const QString DatabaseManager::DB_TABLE_NAME_MEDIATYPE = QString("mediatype");
36 const QString DatabaseManager::DB_TABLE_NAME_PLATFORM = QString("platform");
37 const QString DatabaseManager::DB_TABLE_NAME_FILE= QString("file");
38 const QString DatabaseManager::DB_TABLE_NAME_FILEPATH = QString("filepath");
39 const QString DatabaseManager::DB_TABLE_NAME_SETUP = QString("setup");
40 const QString DatabaseManager::DB_TABLE_MEDIAIMAGECONTAINER = QString("mediaimagecontainer");
41 const QString DatabaseManager::DB_TABLE_MEDIAIMAGECONTAINER_MEDIAIMAGE = QString("mediaimagecontainer_mediaimage");
42 const QString DatabaseManager::DB_TABLE_EXECUTABLE = QString("executable");
43
44 DatabaseManager::DatabaseManager(QObject *parent)
45         : QObject(parent)
46 {
47     sqlTableModel = 0;
48 }
49
50 DatabaseManager::~DatabaseManager()
51 {
52     // no need to explicitily destroy sqlTableModel
53     // because it is parented QObject and will
54     // be destroyed when parent is destroyed
55 }
56
57 QSqlQueryModel* DatabaseManager::getDataModel(bool update)
58 {
59     if (!sqlTableModel || (sqlTableModel && !update)) sqlTableModel = getData();
60     return sqlTableModel;
61 }
62
63 bool DatabaseManager::openDB()
64 {
65     QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE");
66     db.setDatabaseName(DatabaseManager::getDbPath());
67     return db.open();
68 }
69
70 QString DatabaseManager::getDbPath()
71 {
72         QString path;
73 #ifdef Q_OS_LINUX
74         path.append(QDir::home().path());
75         path.append(QDir::separator()).append(DB_FILENAME);
76 #else
77         path.append(DB_FILENAME);       
78 #endif
79         return path;
80 }
81
82 void DatabaseManager::resetModel()
83 {
84     if (!sqlTableModel) return;
85     clearFilters();
86 }
87
88 // sql must return a count(*) value
89 int DatabaseManager::countRows(QString tableName, QString columnName, int id) const
90 {
91     QString sql = QString("SELECT COUNT(*) FROM %1 WHERE %2 = %3")
92         .arg(tableName).arg(columnName).arg(id);
93     int numEntries = 0;
94     QSqlQuery query(sql);
95     if (query.next())
96         numEntries = query.value(0).toInt();
97     return numEntries;
98 }
99
100 EmuFrontObject* DatabaseManager::getDataObject(int id)
101 {
102     filterById(id);
103     return getFilteredDataObject();
104     /*EmuFrontObject *plf = 0;
105     if (sqlTableModel->rowCount() == 1)
106     {
107         QSqlRecord record = sqlTableModel->record(0);
108         if (record.isEmpty())
109         {
110             throw new EmuFrontException(tr("No data available for id %1").arg(id));
111         }
112         else plf = recordToDataObject(&record);
113     }
114     return plf;*/
115 }
116
117 EmuFrontObject* DatabaseManager::getDataObject(QString filter)
118 {
119     qDebug() << "Filtering data object" << filter;
120     QList<QString> filters;
121     filters.append(filter);
122     filterDataObjects(filters);
123     qDebug() << "...done filtering.";
124     return getFilteredDataObject();
125 }
126
127 EmuFrontObject* DatabaseManager::getFilteredDataObject()
128 {
129     EmuFrontObject *plf = 0;
130     // TODO: if record has more than one the first instance is returned
131     // ... check if this is ok in all cases!
132     if (sqlTableModel->rowCount() >= 1)
133     {
134         QSqlRecord record = sqlTableModel->record(0);
135         if (record.isEmpty())
136         {
137             throw new EmuFrontException(tr("No filtered data available"));
138         }
139         else plf = recordToDataObject(&record);
140     }
141      return plf;
142 }
143
144 EmuFrontObject* DatabaseManager::getDataObjectFromModel(QModelIndex *index)
145 {
146     if (!sqlTableModel) sqlTableModel = getDataModel();
147     QSqlRecord record = sqlTableModel->record(index->row());
148     return recordToDataObject(&record);
149 }
150
151 int DatabaseManager::getCurrentTimeStamp() {
152     return QDateTime::currentDateTime().toTime_t();
153 }