kworship/KwDatabaseSetup.cpp

   1 /***************************************************************************
   2  *   This file is part of KWorship.                                        *
   3  *   Copyright 2008 James Hogan <james@albanarts.com>                      *
   4  *                                                                         *
   5  *   KWorship is free software: you can redistribute it and/or modify      *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation, either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   KWorship is distributed in the hope that it will be useful,           *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with KWorship.  If not, write to the Free Software Foundation,  *
  17  *   Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.   *
  18  ***************************************************************************/
  19
  20 /**
  21  * @file KwDatabaseSetup.cpp
  22  * @brief Database setup manager.
  23  * @author James Hogan <james@albanarts.com>
  24  */
  25
  26 #include "KwDatabaseSetup.h"
  27 #include "settings.h"
  28
  29 #include <kstandarddirs.h>
  30
  31 /*
  32  * Constructors + destructor
  33  */
  34
  35 /// Default constructor.
  36 KwDatabaseSetup::KwDatabaseSetup(bool testing)
  37 : m_testing(testing)
  38 , m_database()
  39 {
  40 }
  41
  42 /// Destructor.
  43 KwDatabaseSetup::~KwDatabaseSetup()
  44 {
  45 }
  46
  47 /*
  48  * Initialisation
  49  */
  50
  51 /// Initialise from configuration.
  52 bool KwDatabaseSetup::initialiseFromConfig()
  53 {
  54   int type = Settings::databaseType();
  55   switch (type)
  56   {
  57     case Settings::EnumDatabaseType::MySQL:
  58     case Settings::EnumDatabaseType::PostgreSQL:
  59       {
  60         QString strType;
  61         if (type == Settings::EnumDatabaseType::MySQL)
  62         {
  63           strType = "QMYSQL";
  64         }
  65         else
  66         {
  67           strType = "QPSQL";
  68         }
  69         return initialiseConnection(strType,
  70                                     Settings::databaseHost(),
  71                                     Settings::databaseName(),
  72                                     Settings::databaseUsername(),
  73                                     Settings::databasePassword());
  74       }
  75       break;
  76     case Settings::EnumDatabaseType::SQLite:
  77       {
  78         QString strType = "QSQLITE";
  79         QUrl url;
  80         if (Settings::databaseLocationCustom())
  81         {
  82           url = Settings::databaseLocation();
  83         }
  84         else
  85         {
  86           url = KStandardDirs::locateLocal("appdata", "kworship.db");
  87         }
  88         return initialiseFile(strType,
  89                               url);
  90       }
  91       break;
  92     default:
  93       break;
  94   }
  95   return false;
  96 }
  97
  98 /// Initialise connection to a remote database.
  99 bool KwDatabaseSetup::initialiseConnection(QString type, QString host, QString dbname, QString username, QString password)
 100 {
 101   if (type != "QMYSQL" && type != "QPSQL")
 102   {
 103     return false;
 104   }
 105
 106   // Setup connection
 107   if (m_testing)
 108   {
 109     m_database = QSqlDatabase::addDatabase(type, "test");
 110   }
 111   else
 112   {
 113     m_database = QSqlDatabase::addDatabase(type);
 114   }
 115   m_database.setHostName(host);
 116   m_database.setDatabaseName(dbname);
 117   m_database.setUserName(username);
 118   if (!password.isNull())
 119   {
 120     m_database.setPassword(password);
 121   }
 122
 123   // Connect
 124   bool connected = m_database.open();
 125
 126   if (connected && !m_testing)
 127   {
 128     // Ensure the database is setup ok
 129     /// @todo implement
 130   }
 131
 132   return connected;
 133 }
 134
 135 /// Initialise local file-based database.
 136 bool KwDatabaseSetup::initialiseFile(QString type, QUrl url)
 137 {
 138   if (type != "QSQLITE")
 139   {
 140     return false;
 141   }
 142
 143   m_database = QSqlDatabase::addDatabase(type);
 144   m_database.setDatabaseName(url.toLocalFile());
 145   bool connected = m_database.open();
 146
 147   if (connected)
 148   {
 149     // Ensure database is setup ok
 150     /// @todo implement
 151   }
 152
 153   return connected;
 154 }
 155
 156 /*
 157  * Accessors
 158  */
 159
 160 /// Get the database object.
 161 QSqlDatabase& KwDatabaseSetup::database()
 162 {
 163   return m_database;
 164 }
 165