Changes in the data model: this is not tested yet at all! (compiles
[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 // Foobar 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 Foobar.  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
32 const QString DatabaseManager::DB_FILENAME = QString("my.db.sqlite");
33 const QString DatabaseManager::DATABASE = QString("QSQLITE");
34 const QString DatabaseManager::DB_TABLE_NAME_MEDIATYPE = QString("mediatype");
35 const QString DatabaseManager::DB_TABLE_NAME_PLATFORM = QString("platform");
36 const QString DatabaseManager::DB_TABLE_NAME_FILE= QString("file");
37 const QString DatabaseManager::DB_TABLE_NAME_FILEPATH = QString("filepath");
38 const QString DatabaseManager::DB_TABLE_NAME_SETUP = QString("setup");
39
40 DatabaseManager::DatabaseManager(QObject *parent)
41         : QObject(parent)
42 {
43     sqlTableModel = 0;
44 }
45
46 DatabaseManager::~DatabaseManager()
47 {
48     // no need to explicitily destroy sqlTableModel
49     // because it is parented QObject and will
50     // be destroyed when parent is destroyed
51 }
52
53 QSqlQueryModel* DatabaseManager::getDataModel()
54 {
55     if (!sqlTableModel) sqlTableModel = getData();
56     return sqlTableModel;
57 }
58
59 bool DatabaseManager::openDB()
60 {
61     QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE");
62     db.setDatabaseName(DatabaseManager::getDbPath());
63     return db.open();
64 }
65
66 QString DatabaseManager::getDbPath()
67 {
68         QString path;
69 #ifdef Q_OS_LINUX
70         path.append(QDir::home().path());
71         path.append(QDir::separator()).append(DB_FILENAME);
72 #else
73         path.append(DB_FILENAME);       
74 #endif
75         return path;
76 }
77
78 void DatabaseManager::resetModel()
79 {
80     if (!sqlTableModel) return;
81     clearFilters();
82 }
83
84 // sql must return a count(*) value
85 int DatabaseManager::countRows(QString tableName, QString columnName, int id) const
86 {
87     QString sql = QString("SELECT COUNT(*) FROM %1 WHERE %2 = %3")
88         .arg(tableName).arg(columnName).arg(id);
89     int numEntries = 0;
90     QSqlQuery query(sql);
91     if (query.next())
92         numEntries = query.value(0).toInt();
93     return numEntries;
94 }
95
96 EmuFrontObject* DatabaseManager::getDataObject(int id)
97 {
98     filterById(id);
99     EmuFrontObject *plf = 0;
100     if (sqlTableModel->rowCount() == 1)
101     {
102         QSqlRecord record = sqlTableModel->record(0);
103         if (record.isEmpty())
104         {
105             throw new EmuFrontException(tr("No data available for id %1").arg(id));
106         }
107         else plf = recordToDataObject(&record);
108     }
109     return plf;
110 }
111
112 EmuFrontObject* DatabaseManager::getDataObjectFromModel(QModelIndex *index)
113 {
114     if (!sqlTableModel) sqlTableModel = getDataModel();
115     QSqlRecord record = sqlTableModel->record(index->row());
116     return recordToDataObject(&record);
117 }