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