72 lines
2.6 KiB
Haskell
72 lines
2.6 KiB
Haskell
|
module Main where
|
||
|
|
|
||
|
|
import Paths_postgrest (version)
|
||
|
|
|
||
|
|
import PostgREST.App
|
||
|
|
import PostgREST.Middleware
|
||
|
|
import PostgREST.Error(errResponse)
|
||
|
|
|
||
|
|
import Control.Monad (unless)
|
||
|
|
import Control.Monad.IO.Class (liftIO)
|
||
|
|
import Data.String.Conversions (cs)
|
||
|
|
import Network.Wai (strictRequestBody)
|
||
|
|
import Network.Wai.Middleware.Cors (cors)
|
||
|
|
import Network.Wai.Handler.Warp hiding (Connection)
|
||
|
|
import Network.Wai.Middleware.Gzip (gzip, def)
|
||
|
|
import Network.Wai.Middleware.Static (staticPolicy, only)
|
||
|
|
import Network.Wai.Middleware.RequestLogger (logStdout)
|
||
|
|
import Data.List (intercalate)
|
||
|
|
import Data.Version (versionBranch)
|
||
|
|
import qualified Hasql as H
|
||
|
|
import qualified Hasql.Postgres as P
|
||
|
|
import Options.Applicative hiding (columns)
|
||
|
|
|
||
|
|
import PostgREST.Config (AppConfig(..), argParser, corsPolicy)
|
||
|
|
|
||
|
|
main :: IO ()
|
||
|
|
main = do
|
||
|
|
let opts = info (helper <*> argParser) $
|
||
|
|
fullDesc
|
||
|
|
<> progDesc (
|
||
|
|
"PostgREST "
|
||
|
|
<> prettyVersion
|
||
|
|
<> " / create a REST API to an existing Postgres database"
|
||
|
|
)
|
||
|
|
parserPrefs = prefs showHelpOnError
|
||
|
|
conf <- customExecParser parserPrefs opts
|
||
|
|
let port = configPort conf
|
||
|
|
|
||
|
|
unless (configSecure conf) $
|
||
|
|
putStrLn "WARNING, running in insecure mode, auth will be in plaintext"
|
||
|
|
unless ("secret" /= configJwtSecret conf) $
|
||
|
|
putStrLn "WARNING, running in insecure mode, JWT secret is the default value"
|
||
|
|
Prelude.putStrLn $ "Listening on port " ++
|
||
|
|
(show $ configPort conf :: String)
|
||
|
|
|
||
|
|
let pgSettings = P.ParamSettings (cs $ configDbHost conf)
|
||
|
|
(fromIntegral $ configDbPort conf)
|
||
|
|
(cs $ configDbUser conf)
|
||
|
|
(cs $ configDbPass conf)
|
||
|
|
(cs $ configDbName conf)
|
||
|
|
appSettings = setPort port
|
||
|
|
. setServerName (cs $ "postgrest/" <> prettyVersion)
|
||
|
|
$ defaultSettings
|
||
|
|
middle = logStdout
|
||
|
|
. (if configSecure conf then redirectInsecure else id)
|
||
|
|
. gzip def . cors corsPolicy
|
||
|
|
. staticPolicy (only [("favicon.ico", "static/favicon.ico")])
|
||
|
|
|
||
|
|
poolSettings <- maybe (fail "Improper session settings") return $
|
||
|
|
H.poolSettings (fromIntegral $ configPool conf) 30
|
||
|
|
pool :: H.Pool P.Postgres
|
||
|
|
<- H.acquirePool pgSettings poolSettings
|
||
|
|
|
||
|
|
runSettings appSettings $ middle $ \req respond -> do
|
||
|
|
body <- strictRequestBody req
|
||
|
|
resOrError <- liftIO $ H.session pool $ H.tx Nothing $
|
||
|
|
authenticated conf (app conf body) req
|
||
|
|
either (respond . errResponse) respond resOrError
|
||
|
|
|
||
|
|
where
|
||
|
|
prettyVersion = intercalate "." $ map show $ versionBranch version
|