src/Gargantext/Database/Utils.hs

   1 {-|
   2 Module      : Gargantext.Database.Util
   3 Description :
   4 Copyright   : (c) CNRS, 2017-Present
   5 License     : AGPL + CECILL v3
   6 Maintainer  : team@gargantext.org
   7 Stability   : experimental
   8 Portability : POSIX
   9
  10 Here is a longer description of this module, containing some
  11 commentary with @some markup@.
  12 -}
  13
  14 {-# LANGUAGE FlexibleContexts  #-}
  15 {-# LANGUAGE NoImplicitPrelude #-}
  16 {-# LANGUAGE OverloadedStrings #-}
  17
  18 module Gargantext.Database.Utils where
  19
  20 import qualified Database.PostgreSQL.Simple as PGS
  21
  22 import Data.Monoid ((<>))
  23 import Data.Either.Extra (Either(Left, Right))
  24 import Gargantext.Prelude
  25 import Data.Text (unpack, pack)
  26 import Text.Read (read)
  27 import Data.Ini (readIniFile, lookupValue)
  28 import Data.Word (Word16)
  29 import System.IO (FilePath)
  30 import Database.PostgreSQL.Simple (Connection, connect)
  31
  32 -- Utilities
  33 import Opaleye (Query, Unpackspec, showSqlForPostgres)
  34 import Data.Profunctor.Product.Default (Default)
  35 import Data.Maybe (maybe)
  36 -- TODO add a reader Monad here
  37 -- read this in the init file
  38
  39 databaseParameters :: FilePath -> IO PGS.ConnectInfo
  40 databaseParameters fp = do
  41   ini         <- readIniFile fp
  42   let ini'' = case ini of
  43         Left e     -> panic (pack $ "No ini file error" <> show e)
  44         Right ini' -> ini'
  45
  46   let val x = case (lookupValue (pack "django") (pack x) ini'') of
  47         Left _ -> panic (pack $ "no" <> x)
  48         Right p' -> unpack p'
  49
  50   pure $ PGS.ConnectInfo { PGS.connectHost     = val       "DB_HOST"
  51                          , PGS.connectPort     = read (val "DB_PORT") :: Word16
  52                          , PGS.connectUser     = val       "DB_USER"
  53                          , PGS.connectPassword = val       "DB_PASS"
  54                          , PGS.connectDatabase = val       "DB_NAME"
  55                          }
  56
  57 connectGargandb :: FilePath -> IO Connection
  58 connectGargandb fp = databaseParameters fp >>= \params -> connect params
  59
  60 printSql :: Default Unpackspec a a => Query a -> IO ()
  61 printSql = putStrLn . maybe "Empty query" identity . showSqlForPostgres
  62