src/Tools/SQL.hs

   1 {-
   2     Tools/SQL.hs
   3
   4     Copyright 2013 Louis-Guillaume Gagnon <louis.guillaume.gagnon@gmail.com>
   5
   6     This program is free software: you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation, either version 3 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 -}
  19
  20 module Tools.SQL (
  21     createdb,
  22     opendb
  23 ) where
  24
  25 import Database.HDBC
  26 import Database.HDBC.Sqlite3
  27
  28 import Tools.Constants
  29
  30 createdb :: Connection -> IO Connection
  31 createdb conn = do run conn ("CREATE TABLE " ++ tableName ++ "(id       INTEGER PRIMARY KEY,\
  32                                                         \ Path     VARCHAR(1000),\
  33                                                         \ Title    VARCHAR(1000),\
  34                                                         \ Authors  VARCHAR(1000),\
  35                                                         \ Keywords VARCHAR(1000),\
  36                                                         \ Journal  VARCHAR(1000),\
  37                                                         \ Volume   VARCHAR(1000),\
  38                                                         \ Year     VARCHAR(1000),\
  39                                                         \ Pages    VARCHAR(1000),\
  40                                                         \ Bookmarked VARCHAR(1000));") []
  41                    commit conn
  42                    return conn
  43
  44 opendb :: IO Connection
  45 opendb = do
  46     conn <- connectSqlite3 dbName
  47     tables <- getTables conn
  48     if not (tableName `elem` tables)
  49         then do createdb conn
  50         else do return conn
  51
  52
  53