Added some exception handling. Marked all the functions throwing
[emufront] / src / db / dbfile.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 <QSqlRecord>
22 #include <QSqlQuery>
23 #include <QSqlRelationalTableModel>
24 #include "dbfile.h"
25
26
27 DbFile::DbFile(QObject *parent) : DbQueryModelManager(parent)
28 {
29     tableName = DbFile::DB_TABLE_NAME_FILE;
30     type = -1;
31 }
32
33 EmuFrontObject* DbFile::recordToDataObject(const QSqlRecord *record)
34 {
35     int id = record->value(File_Id).toInt();
36     QString name = record->value(File_Name).toString();
37     QString checksum = record->value(File_CheckSum).toString();
38     int size = record->value(File_FileSize).toInt();
39     int type = record->value(File_FileType).toInt();
40     return new EmuFrontFile(id, name, checksum, size, type);
41 }
42
43 bool DbFile::updateDataObjectToModel(const EmuFrontObject *ob)
44 {
45     const EmuFrontFile *plf = dynamic_cast<const EmuFrontFile*>(ob);
46     bool ret = false;
47
48     QSqlQuery query;
49     query.prepare(QString("UPDATE file SET"
50                           "name=:name, "
51                           "type=:type, "
52                           "checksum=:checksum, "
53                           "size=:size, "
54                           "updatetime:=updatetime "
55                           "WHERE id=:id"));
56     query.bindValue(":name", plf->getName());
57     query.bindValue(":type", plf->getType());
58     query.bindValue(":checksum", plf->getCheckSum());
59     query.bindValue(":size", plf->getSize());
60     query.bindValue(":updatetime", getCurrentTimeStamp());
61     ret = query.exec();
62     if (ret) resetModel();
63     return ret;
64 }
65
66 int DbFile::insertDataObjectToModel(const EmuFrontObject *ob)
67 {
68     const EmuFrontFile *fi = dynamic_cast<const EmuFrontFile*>(ob);
69     QSqlQuery q;
70     q.prepare("INSERT INTO file "
71               "(id, name, type, checksum, size, updatetime) "
72               "VALUES (NULL, :name, :type, :checksum, :size, :updatetime)");
73     q.bindValue(":name", fi->getName());
74     q.bindValue(":type", fi->getType());
75     q.bindValue(":checksum", fi->getCheckSum());
76     q.bindValue(":size", fi->getSize());
77     q.bindValue(":updatetime", DatabaseManager::getCurrentTimeStamp());
78     int id = -1;
79     if (q.exec())
80         id = q.lastInsertId().toInt();
81     return id;
82 }
83
84 // WARNING: this will delete also all the databindings to selected platform
85 // the delete must be confirmed in the UI
86 bool DbFile::deleteDataObjectFromModel(QModelIndex *index)
87 {
88     return false;
89 }
90
91 bool DbFile::deleteDataObject(int id) const
92 {
93     if (countDataObjectRefs(id) > 0)
94         // TODO
95         return false;
96     QSqlQuery q;
97     q.prepare(QString("DELETE FROM file WHERE id=:id"));
98     q.bindValue(":id", id);
99     return q.exec();
100 }
101
102 QString DbFile::constructSelect(QString where) const
103 {
104     return QString("SELECT file.id AS FileId, "
105                    "file.name AS Name, "
106                    "file.type AS FileType, "
107                    "file.checksum AS Checksum, "
108                    "file.size AS FileSize, "
109                    "file.updatetime AS UpdateTime "
110                    "FROM file "
111                    "%1 "
112                    "ORDER BY Name").arg(where);
113 }
114
115 QString DbFile::constructFilterById(int id) const
116 {
117     return QString("file.id = %1").arg(id);
118 }
119
120 QString DbFile::constructSelectById(int id) const
121 {
122     return constructSelect(QString("WHERE %1").arg(constructFilterById(id)));
123 }
124
125 QSqlQueryModel* DbFile::getData()
126 {
127     QSqlQueryModel *model = new QSqlQueryModel(this);
128     model->setQuery(constructSelect());
129     model->setHeaderData(File_Name, Qt::Horizontal, tr("Name"));
130     model->setHeaderData(File_FileType, Qt::Horizontal, tr("Type"));
131     model->setHeaderData(File_CheckSum, Qt::Horizontal, tr("Checksum"));
132     model->setHeaderData(File_FileSize, Qt::Horizontal, tr("Size"));
133     model->setHeaderData(File_UpdateTime, Qt::Horizontal, tr("Updated"));
134     return model;
135 }
136
137 /* Throws EmuFrontException */
138 EmuFrontObject* DbFile::getFileByChecksum(QString checksum)
139 {
140     return getDataObject(QString("checksum LIKE '%1'").arg(checksum));
141 }
142
143 QString DbFile::getCountRefsSelect(int id) const
144 {
145     /* files are referenced from platform, mediatype, mediaimagecontainer_mediaimage and mediaimagecontainer. */
146     return QString("SELECT count(*) FROM ("
147               "SELECT file.id FROM file "
148               "INNER JOIN platform ON file.id=platform.fileid "
149               "WHERE file.id=%1 "
150               "UNION ALL "
151               "SELECT file.id FROM file "
152               "INNER JOIN mediatype ON file.id=mediatype.fileid "
153               "WHERE file.id=%1 "
154               "UNION ALL "
155               "SELECT file.id FROM file "
156               "INNER JOIN mediaimagecontainer_mediaimage "
157               "ON (mediaimagecontainerid=file.id OR mediaimageid=file.id) "
158               "WHERE file.id=%1 "
159               ")").arg(id);
160 }