e5d1dad79fa4f9acf2f4613e5efc46889f061cb1
[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 version 2 as published by
9 // the Free Software Foundation and appearing in the file gpl.txt included in the
10 // packaging of this file.
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 }
105
106 EmuFrontObject* DatabaseManager::getDataObject(QString filter)
107 {
108     qDebug() << "Filtering data object" << filter;
109     QList<QString> filters;
110     filters.append(filter);
111     filterDataObjects(filters);
112     qDebug() << "...done filtering.";
113     return getFilteredDataObject();
114 }
115
116 EmuFrontObject* DatabaseManager::getFilteredDataObject()
117 {
118     EmuFrontObject *plf = 0;
119     // TODO: if record has more than one the first instance is returned
120     // ... check if this is ok in all cases!
121     if (sqlTableModel->rowCount() >= 1)
122     {
123         QSqlRecord record = sqlTableModel->record(0);
124         if (record.isEmpty())
125         {
126             throw new EmuFrontException(tr("No filtered data available"));
127         }
128         else plf = recordToDataObject(&record);
129     }
130      return plf;
131 }
132
133 EmuFrontObject* DatabaseManager::getDataObjectFromModel(QModelIndex *index)
134 {
135     if (!sqlTableModel) sqlTableModel = getDataModel();
136     QSqlRecord record = sqlTableModel->record(index->row());
137     return recordToDataObject(&record);
138 }
139
140 int DatabaseManager::getCurrentTimeStamp() {
141     return QDateTime::currentDateTime().toTime_t();
142 }
143
144 int DatabaseManager::countDataObjectRefs(int id) const
145 {
146     int ret = 0;
147     QSqlQuery q;
148     q.prepare(getCountRefsSelect(id));
149     q.exec();
150     QSqlRecord rec;
151     if (q.next()) {
152         rec = q.record();
153         ret = rec.value(0).toInt();
154     }
155     qDebug() << "Found " << ret << " references.";
156     return ret;
157 }