f594d7cb921c0d86cd3fafb29d08f83b578ef701
[emufront] / src / db / dbsetup.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 <QDebug>
21 #include <QStringList>
22 #include <QSqlRecord>
23 #include <QSqlQuery>
24 #include <QSqlError>
25 #include <QSqlRelationalTableModel>
26 #include "dbsetup.h"
27
28 const QString DbSetup::FILE_TYPE_EXTENSION_SEPARATOR = QString("|");
29
30 DbSetup::DbSetup(QObject *parent) : DbQueryModelManager(parent)
31 {
32     tableName = DbSetup::DB_TABLE_NAME_SETUP;
33     dbPlatform = new DbPlatform(this);
34     dbMediaType = new DbMediaType(this);
35 }
36
37 /* Throws EmuFrontException */
38 EmuFrontObject* DbSetup::recordToDataObject(const QSqlRecord *rec)
39 {
40     Setup *s = 0;
41     if (!rec) return s;
42     int id = rec->value(Setup_Id).toInt();
43     QString extensions = rec->value(Setup_FileTypeExtensions).toString().trimmed();
44     QStringList list;
45     if (!extensions.isEmpty())
46         list = extensions.split(FILE_TYPE_EXTENSION_SEPARATOR);
47     int plfId = rec->value(Setup_PlatformId).toInt();
48     int mtId = rec->value(Setup_MediaTypeId).toInt();
49     Platform *plf = dynamic_cast<Platform*>(dbPlatform->getDataObject(plfId)); /* Throws EmuFrontException */
50     MediaType *mt = dynamic_cast<MediaType*>(dbMediaType->getDataObject(mtId)); /* Throws EmuFrontException */
51     s = new Setup(id, plf, mt, list);
52     return s;
53 }
54
55 bool DbSetup::updateDataObjectToModel(const EmuFrontObject *ob)
56 {
57     const Setup *fpo = dynamic_cast<const Setup*>(ob);
58     bool ret = false;
59     QSqlQuery query;
60     query.prepare(QString("UPDATE setup SET "
61         "platformid=:platformid, "
62         "mediatypeid=:mediatypeid, "
63         "filetypeextensions=:filetypeextensions "
64         "WHERE id = :id"));
65     if (fpo->getPlatform())
66         query.bindValue(":platformid", fpo->getPlatform()->getId());
67     if (fpo->getMediaType())
68         query.bindValue(":mediatypeid", fpo->getMediaType()->getId());
69     query.bindValue(":filetypeextensions", supportedExtensionsToDb(fpo->getSupportedFileTypeExtensions()));
70     query.bindValue(":id", fpo->getId());
71     ret = query.exec();
72     if (ret) resetModel();
73     if (!ret) qDebug() << query.lastError().text() << query.executedQuery();
74     return ret;
75 }
76
77 QString DbSetup::supportedExtensionsToDb(QStringList list)
78 {
79     return list.isEmpty() ? "" : list.join(FILE_TYPE_EXTENSION_SEPARATOR);
80 }
81
82 /* Returns id of inserted data item after succesful insert, -1 if insert failed */
83 int DbSetup::insertDataObjectToModel(const EmuFrontObject *ob)
84 {
85     qDebug() << "Inserting setup to database...";
86     const Setup *fpo = dynamic_cast<const Setup*>(ob);
87     QSqlQuery query;
88     query.prepare("INSERT INTO setup (id, platformid, mediatypeid, filetypeextensions)"
89         "VALUES (NULL, :platformid, :mediatypeid, :fileextensions)");
90     int plfId = fpo->getPlatform() ? fpo->getPlatform()->getId() : -1;
91     int mtId = fpo->getMediaType() ? fpo->getMediaType()->getId() : -1;
92     query.bindValue(":platformid", plfId);
93     query.bindValue(":mediatypeid", mtId);
94     query.bindValue(":filetypeextensions", supportedExtensionsToDb(fpo->getSupportedFileTypeExtensions()));
95     int id = -1;
96     if (query.exec())
97         id = query.lastInsertId().toInt();
98     else
99         qDebug() << query.lastError().text() << query.executedQuery();
100     return id;
101 }
102
103 QString DbSetup::constructSelect(QString where) const
104 {
105     return QString(
106         "SELECT setup.id AS SetupId, "
107         "setup.platformid AS PlatformId, "
108         "setup.mediatypeid AS MediaTypeId, "
109         "setup.filetypeextensions AS SupportedFileTypeExtensions, "
110         "platform.name || ' ' || mediatype.name AS SetupName "
111         "FROM setup "
112         "INNER JOIN platform ON setup.platformid=platform.id "
113         "INNER JOIN mediatype ON setup.mediatypeid=mediatype.id %1 "
114         "ORDER BY SetupName"
115         ).arg(where);
116 }
117
118 QString DbSetup::constructFilterById(int id) const
119 {
120      return QString("setup.id = %1").arg(id);
121 }
122
123
124 QString DbSetup::constructSelectById(int id) const
125 {
126     return constructSelect(
127         QString("WHERE %1").arg(constructFilterById(id)));
128 }
129
130 // WARNING: this will delete also all the databindings to selected media image path
131 bool DbSetup::deleteDataObjectFromModel(QModelIndex */*index*/)
132 {
133     qDebug() << "This is not currently supported";
134     return false;
135 }
136
137 QSqlQueryModel* DbSetup::getData()
138 {
139     QSqlQueryModel *model = new QSqlQueryModel(this);
140     QString select = constructSelect();
141     qDebug() << select;
142     model->setQuery(select);
143     model->setHeaderData(Setup_Id, Qt::Horizontal, tr("Id"));
144     model->setHeaderData(Setup_PlatformId, Qt::Horizontal, tr("Platform id"));
145     model->setHeaderData(Setup_MediaTypeId, Qt::Horizontal, tr("Media type id"));
146     model->setHeaderData(Setup_FileTypeExtensions, Qt::Horizontal, tr("File types"));
147     model->setHeaderData(Setup_Name, Qt::Horizontal, tr("Name"));
148     return model;
149 }
150
151 QString DbSetup::getCountRefsSelect(int id) const
152 {
153     /* setups are referenced by executable and filepath */
154     return QString("SELECT count(*) FROM "
155             "(SELECT setup.id FROM setup "
156               "INNER JOIN executable ON setup.id=executable.setupid "
157               "WHERE setup.id=%1 "
158               "UNION ALL "
159               "SELECT setup.id FROM setup "
160               "INNER JOIN filepath ON setup.id=filepath.setupid "
161               "WHERE setup.id=%1)").arg(id);
162 }