DbFilePath (insert and update functionality). getDataObjectFromModel and
[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
31 const QString DatabaseManager::DB_FILENAME = QString("my.db.sqlite");
32 const QString DatabaseManager::DATABASE = QString("QSQLITE");
33 const QString DatabaseManager::DB_TABLE_NAME_MEDIATYPE = QString("mediatype");
34 const QString DatabaseManager::DB_TABLE_NAME_PLATFORM = QString("platform");
35 const QString DatabaseManager::DB_TABLE_NAME_FILEPATH = QString("filepath");
36
37 DatabaseManager::DatabaseManager(QObject *parent)
38         : QObject(parent)
39 {}
40
41 DatabaseManager::~DatabaseManager()
42 {
43     // no need to explicitily destroy sqlTableModel
44     // because it is parented QObject and will
45     // be destroyed when parent is destroyed
46 }
47
48 bool DatabaseManager::openDB()
49 {
50     QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE");
51     db.setDatabaseName(DatabaseManager::getDbPath());
52     return db.open();
53 }
54
55 QString DatabaseManager::getDbPath()
56 {
57         QString path;
58 #ifdef Q_OS_LINUX
59         path.append(QDir::home().path());
60         path.append(QDir::separator()).append(DB_FILENAME);
61 #else
62         path.append(DB_FILENAME);       
63 #endif
64         return path;
65 }
66
67 void DatabaseManager::resetModel() const
68 {
69     if (!sqlTableModel) return;
70     sqlTableModel->setFilter("");
71     sqlTableModel->select();
72 }
73
74 // sql must return a count(*) value
75 int DatabaseManager::countRows(QString tableName, QString columnName, int id) const
76 {
77     QString sql = QString("SELECT COUNT(*) FROM %1 WHERE %2 = %3")
78         .arg(tableName).arg(columnName).arg(id);
79     int numEntries = 0;
80     QSqlQuery query(sql);
81     if (query.next())
82         numEntries = query.value(0).toInt();
83     return numEntries;
84 }
85
86 EmuFrontObject* DatabaseManager::getDataObject(int id) const
87 {
88     sqlTableModel->setFilter(QString("id = %1").arg(id));
89     sqlTableModel->select();
90     EmuFrontObject *plf = 0;
91     if (sqlTableModel->rowCount() == 1)
92     {
93         QSqlRecord record = sqlTableModel->record(0);
94         plf = recordToDataObject(&record);
95     }
96     return plf;
97 }
98
99 EmuFrontObject* DatabaseManager::getDataObjectFromModel(QModelIndex *index)
100 {
101     QSqlRecord record = sqlTableModel->record(index->row());
102     return recordToDataObject(&record);
103 }