Compare commits
1 commit
master
...
make_tests
Author | SHA1 | Date | |
---|---|---|---|
f2d4c88768 |
19 changed files with 128 additions and 296 deletions
2
.gitignore
vendored
2
.gitignore
vendored
|
@ -12,5 +12,3 @@ report.html
|
||||||
*.save*
|
*.save*
|
||||||
.envrc
|
.envrc
|
||||||
.direnv/
|
.direnv/
|
||||||
.hie/
|
|
||||||
stan.html
|
|
||||||
|
|
|
@ -20,7 +20,6 @@ data ServerConfig = ServerConfig
|
||||||
-- , configMaxConnectionsPerClient :: Word
|
-- , configMaxConnectionsPerClient :: Word
|
||||||
-- , configBlockRegistration :: Bool
|
-- , configBlockRegistration :: Bool
|
||||||
, configSendmailPath :: FilePath
|
, configSendmailPath :: FilePath
|
||||||
, configJWTSecret :: T.Text
|
|
||||||
}
|
}
|
||||||
deriving (Show)
|
deriving (Show)
|
||||||
|
|
||||||
|
@ -38,7 +37,6 @@ instance FromJSON ServerConfig where
|
||||||
-- <*> m .:? "max_connections_per_client" .!= 10
|
-- <*> m .:? "max_connections_per_client" .!= 10
|
||||||
-- <*> m .: "block_registration"
|
-- <*> m .: "block_registration"
|
||||||
<*> m .: "sendmail_path"
|
<*> m .: "sendmail_path"
|
||||||
<*> m .: "jwt_secret"
|
|
||||||
parseJSON _ = error "Can not parse configuration"
|
parseJSON _ = error "Can not parse configuration"
|
||||||
|
|
||||||
data Options = Options
|
data Options = Options
|
||||||
|
|
70
app/Main.hs
70
app/Main.hs
|
@ -10,47 +10,40 @@ module Main where
|
||||||
|
|
||||||
import Prelude as P
|
import Prelude as P
|
||||||
|
|
||||||
import Crypto.JWT hiding (Context, header)
|
|
||||||
|
|
||||||
import Control.Concurrent.STM.TVar
|
|
||||||
import Control.Concurrent.STM (newTQueueIO)
|
import Control.Concurrent.STM (newTQueueIO)
|
||||||
import Control.Concurrent (forkIO)
|
import Control.Concurrent (forkIO)
|
||||||
|
|
||||||
import Control.Lens hiding (Context)
|
|
||||||
|
|
||||||
import Control.Monad (void, unless)
|
|
||||||
|
|
||||||
import Control.Monad.Reader
|
|
||||||
|
|
||||||
import Servant
|
import Servant
|
||||||
import Servant.Server.Experimental.Auth
|
import Servant.Server.Experimental.Auth
|
||||||
import qualified Servant.OpenApi as OA
|
import qualified Servant.OpenApi as OA
|
||||||
import Servant.Swagger.UI
|
import Servant.Swagger.UI
|
||||||
import Servant.RawM.Server
|
import Servant.RawM
|
||||||
|
|
||||||
import Data.Set as S (empty)
|
import Data.Set as S (empty)
|
||||||
|
|
||||||
import qualified Data.ByteString.Lazy as B
|
|
||||||
import Data.ByteString.Char8 as B8 hiding (putStrLn)
|
import Data.ByteString.Char8 as B8 hiding (putStrLn)
|
||||||
|
|
||||||
import qualified Data.Text as T
|
import qualified Data.Text as T
|
||||||
import qualified Data.Text.Encoding as TE
|
|
||||||
|
|
||||||
import Data.String
|
import Data.String
|
||||||
|
|
||||||
import Data.Yaml
|
import Data.Yaml
|
||||||
|
|
||||||
import Data.Version (showVersion)
|
import Data.Version (showVersion)
|
||||||
|
import Data.IP
|
||||||
import qualified Data.OpenApi as OA hiding (Server)
|
import qualified Data.OpenApi as OA hiding (Server)
|
||||||
|
|
||||||
import Database.PostgreSQL.Simple
|
import Database.PostgreSQL.Simple
|
||||||
import Database.PostgreSQL.Simple.Migration
|
import Database.PostgreSQL.Simple.Migration
|
||||||
import Database.PostgreSQL.Simple.Util
|
import Database.PostgreSQL.Simple.Util
|
||||||
|
|
||||||
|
import Network.Socket (defaultPort)
|
||||||
|
|
||||||
import Network.Wai
|
import Network.Wai
|
||||||
import Network.Wai.Logger
|
import Network.Wai.Logger
|
||||||
import Network.Wai.Handler.Warp
|
import Network.Wai.Handler.Warp
|
||||||
|
import Network.Wai.Middleware.Throttle
|
||||||
|
|
||||||
|
import Control.Monad.Reader
|
||||||
|
|
||||||
|
import Control.Concurrent.STM.TVar
|
||||||
|
|
||||||
|
import Control.Lens hiding (Context)
|
||||||
|
|
||||||
import Options.Applicative
|
import Options.Applicative
|
||||||
|
|
||||||
|
@ -94,7 +87,6 @@ main = do
|
||||||
-- max_conn_per_client
|
-- max_conn_per_client
|
||||||
-- block_registration
|
-- block_registration
|
||||||
sendmail_path
|
sendmail_path
|
||||||
jwt_secret
|
|
||||||
) -> do
|
) -> do
|
||||||
conn <- connectPostgreSQL (
|
conn <- connectPostgreSQL (
|
||||||
"host='" <> fromString (T.unpack db_host) <> "' " <>
|
"host='" <> fromString (T.unpack db_host) <> "' " <>
|
||||||
|
@ -150,10 +142,22 @@ main = do
|
||||||
, rsSoftwareVersion = T.pack (showVersion version)
|
, rsSoftwareVersion = T.pack (showVersion version)
|
||||||
, rsSendmailPath = sendmail_path
|
, rsSendmailPath = sendmail_path
|
||||||
, rsMailQueue = mailQueue
|
, rsMailQueue = mailQueue
|
||||||
, rsJWTSecret =
|
|
||||||
fromOctets . B.fromStrict $ TE.encodeUtf8 jwt_secret
|
|
||||||
}
|
}
|
||||||
runSettings settings (app initState)
|
expirationSpec = TimeSpec 5 0 -- five seconds
|
||||||
|
throt = (defaultThrottleSettings expirationSpec)
|
||||||
|
{ throttleSettingsRate = 10
|
||||||
|
, throttleSettingsPeriod = 1000
|
||||||
|
}
|
||||||
|
th <- initCustomThrottler throt
|
||||||
|
(\req ->
|
||||||
|
let headers = requestHeaders req
|
||||||
|
in case lookup "x-forwarded-for" headers of
|
||||||
|
Just addrs ->
|
||||||
|
let xaddr = fst (B8.break (== ',') addrs)
|
||||||
|
in Right $ Address $ toSockAddr (read (B8.unpack xaddr), defaultPort)
|
||||||
|
Nothing -> Right $ Address $ remoteHost req
|
||||||
|
)
|
||||||
|
runSettings settings (throttle th (app initState))
|
||||||
where
|
where
|
||||||
opts = info (options <**> helper)
|
opts = info (options <**> helper)
|
||||||
( fullDesc
|
( fullDesc
|
||||||
|
@ -164,8 +168,7 @@ main = do
|
||||||
app :: ReadState -> Application
|
app :: ReadState -> Application
|
||||||
-- app conn = serveWithContext userApi genAuthServerContext (users conn)
|
-- app conn = serveWithContext userApi genAuthServerContext (users conn)
|
||||||
app initState =
|
app initState =
|
||||||
serveWithContext combinedAPI
|
serveWithContext combinedAPI (genAuthServerContext (rsConnection initState)) server
|
||||||
(genAuthServerContext (rsJWTSecret initState) (rsConnection initState)) server
|
|
||||||
where
|
where
|
||||||
server :: Server CombinedAPI
|
server :: Server CombinedAPI
|
||||||
server = appToServer initState mateAPI thisApi :<|>
|
server = appToServer initState mateAPI thisApi :<|>
|
||||||
|
@ -179,6 +182,7 @@ appToServer initState myApi =
|
||||||
|
|
||||||
thisApi :: ServerT MateAPI MateHandler
|
thisApi :: ServerT MateAPI MateHandler
|
||||||
thisApi =
|
thisApi =
|
||||||
|
authGet :<|>
|
||||||
authSend :<|>
|
authSend :<|>
|
||||||
authLogout :<|>
|
authLogout :<|>
|
||||||
|
|
||||||
|
@ -242,22 +246,21 @@ authProxy :: Proxy '[ AuthHandler Request (Maybe (Int, AuthMethod)) ]
|
||||||
authProxy = Proxy
|
authProxy = Proxy
|
||||||
|
|
||||||
genAuthServerContext
|
genAuthServerContext
|
||||||
:: JWK
|
:: Connection
|
||||||
-> Connection
|
|
||||||
-> Context '[ AuthHandler Request (Maybe (Int, AuthMethod)) ]
|
-> Context '[ AuthHandler Request (Maybe (Int, AuthMethod)) ]
|
||||||
genAuthServerContext key conn = authHandler key conn Servant.:. EmptyContext
|
genAuthServerContext conn = authHandler conn Servant.:. EmptyContext
|
||||||
|
|
||||||
type instance AuthServerData (AuthProtect "header-auth") = Maybe (Int, AuthMethod)
|
type instance AuthServerData (AuthProtect "header-auth") = Maybe (Int, AuthMethod)
|
||||||
|
|
||||||
authHandler :: JWK -> Connection -> AuthHandler Request (Maybe (Int, AuthMethod))
|
authHandler :: Connection -> AuthHandler Request (Maybe (Int, AuthMethod))
|
||||||
authHandler key conn = mkAuthHandler handler
|
authHandler conn = mkAuthHandler handler
|
||||||
where
|
where
|
||||||
handler :: Request -> Handler (Maybe (Int, AuthMethod))
|
handler :: Request -> Handler (Maybe (Int, AuthMethod))
|
||||||
handler req = do
|
handler req = do
|
||||||
let headers = requestHeaders req
|
let headers = requestHeaders req
|
||||||
case lookup "Authorization" headers of
|
case lookup "Authentication" headers of
|
||||||
Just hh -> do
|
Just hh ->
|
||||||
validateToken (B8.drop 7 hh) key conn
|
validateToken hh conn
|
||||||
_ ->
|
_ ->
|
||||||
return Nothing
|
return Nothing
|
||||||
|
|
||||||
|
@ -266,6 +269,3 @@ instance OA.HasOpenApi sub => OA.HasOpenApi (AuthProtect "header-auth" :> sub) w
|
||||||
|
|
||||||
instance OA.HasOpenApi (RawM' Application) where
|
instance OA.HasOpenApi (RawM' Application) where
|
||||||
toOpenApi _ = OA.toOpenApi (Proxy :: Proxy (Get '[JSON] NoContent))
|
toOpenApi _ = OA.toOpenApi (Proxy :: Proxy (Get '[JSON] NoContent))
|
||||||
|
|
||||||
instance OA.HasOpenApi (JWS Identity () JWSHeader) where
|
|
||||||
toOpenApi _ = OA.toOpenApi (Proxy :: Proxy (Get '[JSON] NoContent))
|
|
||||||
|
|
4
cabal.project
Normal file
4
cabal.project
Normal file
|
@ -0,0 +1,4 @@
|
||||||
|
-- index-state: 2021-02-05T00:00:00Z
|
||||||
|
packages:
|
||||||
|
./
|
||||||
|
tests: true
|
|
@ -10,5 +10,3 @@ currency: "meow"
|
||||||
currency_fraction: 2
|
currency_fraction: 2
|
||||||
#block_registration: false
|
#block_registration: false
|
||||||
sendmail_path: "/run/wrappers/bin/sendmail"
|
sendmail_path: "/run/wrappers/bin/sendmail"
|
||||||
# Change me !!!
|
|
||||||
jwt_secret: "MySuperDuperSecretJWTKeyChangeMe"
|
|
||||||
|
|
60
flake.lock
60
flake.lock
|
@ -1,60 +0,0 @@
|
||||||
{
|
|
||||||
"nodes": {
|
|
||||||
"flake-utils": {
|
|
||||||
"inputs": {
|
|
||||||
"systems": "systems"
|
|
||||||
},
|
|
||||||
"locked": {
|
|
||||||
"lastModified": 1710146030,
|
|
||||||
"narHash": "sha256-SZ5L6eA7HJ/nmkzGG7/ISclqe6oZdOZTNoesiInkXPQ=",
|
|
||||||
"owner": "numtide",
|
|
||||||
"repo": "flake-utils",
|
|
||||||
"rev": "b1d9ab70662946ef0850d488da1c9019f3a9752a",
|
|
||||||
"type": "github"
|
|
||||||
},
|
|
||||||
"original": {
|
|
||||||
"owner": "numtide",
|
|
||||||
"repo": "flake-utils",
|
|
||||||
"type": "github"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"nixpkgs": {
|
|
||||||
"locked": {
|
|
||||||
"lastModified": 1725059563,
|
|
||||||
"narHash": "sha256-laJvLHrSU5M9zWlejH7H67HdpLhcUI6uPDa4rX7eUuE=",
|
|
||||||
"owner": "NixOS",
|
|
||||||
"repo": "nixpkgs",
|
|
||||||
"rev": "0abfc619bcb605299a0f3f01c1887bb65db61a6b",
|
|
||||||
"type": "github"
|
|
||||||
},
|
|
||||||
"original": {
|
|
||||||
"owner": "NixOS",
|
|
||||||
"repo": "nixpkgs",
|
|
||||||
"type": "github"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"root": {
|
|
||||||
"inputs": {
|
|
||||||
"flake-utils": "flake-utils",
|
|
||||||
"nixpkgs": "nixpkgs"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"systems": {
|
|
||||||
"locked": {
|
|
||||||
"lastModified": 1681028828,
|
|
||||||
"narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
|
|
||||||
"owner": "nix-systems",
|
|
||||||
"repo": "default",
|
|
||||||
"rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
|
|
||||||
"type": "github"
|
|
||||||
},
|
|
||||||
"original": {
|
|
||||||
"owner": "nix-systems",
|
|
||||||
"repo": "default",
|
|
||||||
"type": "github"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"root": "root",
|
|
||||||
"version": 7
|
|
||||||
}
|
|
44
flake.nix
44
flake.nix
|
@ -1,44 +0,0 @@
|
||||||
{
|
|
||||||
description = "A game stub written in Haskell";
|
|
||||||
|
|
||||||
inputs = {
|
|
||||||
nixpkgs.url = "github:NixOS/nixpkgs";
|
|
||||||
flake-utils.url = "github:numtide/flake-utils";
|
|
||||||
};
|
|
||||||
|
|
||||||
outputs = { self, nixpkgs, flake-utils }:
|
|
||||||
flake-utils.lib.eachDefaultSystem (system:
|
|
||||||
let
|
|
||||||
pkgs = nixpkgs.legacyPackages.${system};
|
|
||||||
|
|
||||||
haskellPackages = pkgs.haskellPackages.override {
|
|
||||||
overrides = final: prev: {
|
|
||||||
openapi3 = jailbreakUnbreak (pkgs.haskell.lib.dontCheck prev.openapi3);
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
jailbreakUnbreak = pkg:
|
|
||||||
pkgs.haskell.lib.doJailbreak (pkg.overrideAttrs (_: { meta = { }; }));
|
|
||||||
|
|
||||||
packageName = "mateamt";
|
|
||||||
in rec {
|
|
||||||
packages.${packageName} = # (ref:haskell-package-def)
|
|
||||||
haskellPackages.callCabal2nix packageName self rec {
|
|
||||||
# Dependency overrides go here
|
|
||||||
postgresql-simple-migration = jailbreakUnbreak haskellPackages.postgresql-simple-migration;
|
|
||||||
};
|
|
||||||
|
|
||||||
defaultPackage = self.packages.${system}.${packageName};
|
|
||||||
|
|
||||||
devShell = haskellPackages.shellFor {
|
|
||||||
packages = p: [ defaultPackage ];
|
|
||||||
withHoogle = true;
|
|
||||||
buildInputs = with haskellPackages; [
|
|
||||||
haskell-language-server
|
|
||||||
ghcid
|
|
||||||
cabal-install
|
|
||||||
];
|
|
||||||
};
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
|
@ -72,22 +72,10 @@ library
|
||||||
|
|
||||||
-- Modules included in this library but not exported.
|
-- Modules included in this library but not exported.
|
||||||
-- other-modules:
|
-- other-modules:
|
||||||
default-extensions:
|
|
||||||
StrictData
|
|
||||||
other-extensions:
|
other-extensions:
|
||||||
DataKinds
|
DataKinds TypeOperators FlexibleInstances MultiParamTypeClasses
|
||||||
TypeOperators
|
RankNTypes ScopedTypeVariables FlexibleContexts OverloadedStrings
|
||||||
FlexibleInstances
|
Arrows CPP LambdaCase DeriveGeneric TypeFamilies
|
||||||
MultiParamTypeClasses
|
|
||||||
RankNTypes
|
|
||||||
ScopedTypeVariables
|
|
||||||
FlexibleContexts
|
|
||||||
OverloadedStrings
|
|
||||||
Arrows
|
|
||||||
CPP
|
|
||||||
LambdaCase
|
|
||||||
DeriveGeneric
|
|
||||||
TypeFamilies
|
|
||||||
TypeSynonymInstances
|
TypeSynonymInstances
|
||||||
|
|
||||||
build-depends:
|
build-depends:
|
||||||
|
@ -105,6 +93,7 @@ library
|
||||||
, servant
|
, servant
|
||||||
, servant-server
|
, servant-server
|
||||||
, servant-openapi3
|
, servant-openapi3
|
||||||
|
, servant-rawm >= 0.3.0.0
|
||||||
, servant-rawm-server
|
, servant-rawm-server
|
||||||
, opaleye
|
, opaleye
|
||||||
, aeson
|
, aeson
|
||||||
|
@ -122,14 +111,10 @@ library
|
||||||
, haskell-gettext
|
, haskell-gettext
|
||||||
, mime-mail
|
, mime-mail
|
||||||
, directory
|
, directory
|
||||||
, jose >= 0.10
|
|
||||||
, monad-time
|
|
||||||
|
|
||||||
hs-source-dirs: src
|
hs-source-dirs: src
|
||||||
default-language: Haskell2010
|
default-language: Haskell2010
|
||||||
ghc-options: -Wall
|
ghc-options: -Wall
|
||||||
-fwrite-ide-info
|
|
||||||
-hiedir=.hie
|
|
||||||
|
|
||||||
executable mateamt
|
executable mateamt
|
||||||
main-is: Main.hs
|
main-is: Main.hs
|
||||||
|
@ -140,8 +125,6 @@ executable mateamt
|
||||||
AppTypes.Configuration
|
AppTypes.Configuration
|
||||||
Janitor
|
Janitor
|
||||||
Paths_mateamt
|
Paths_mateamt
|
||||||
default-extensions:
|
|
||||||
StrictData
|
|
||||||
other-extensions:
|
other-extensions:
|
||||||
DataKinds TypeOperators FlexibleInstances MultiParamTypeClasses
|
DataKinds TypeOperators FlexibleInstances MultiParamTypeClasses
|
||||||
RankNTypes ScopedTypeVariables FlexibleContexts OverloadedStrings
|
RankNTypes ScopedTypeVariables FlexibleContexts OverloadedStrings
|
||||||
|
@ -166,27 +149,24 @@ executable mateamt
|
||||||
, network
|
, network
|
||||||
, servant
|
, servant
|
||||||
, servant-server
|
, servant-server
|
||||||
, servant-rawm-server
|
, servant-rawm
|
||||||
, servant-openapi3
|
, servant-openapi3
|
||||||
, servant-swagger-ui
|
, servant-swagger-ui
|
||||||
, servant-swagger-ui-core
|
, servant-swagger-ui-core
|
||||||
, warp
|
, warp
|
||||||
, wai
|
, wai
|
||||||
, wai-logger
|
, wai-logger
|
||||||
|
, wai-middleware-throttle
|
||||||
, yaml
|
, yaml
|
||||||
, optparse-applicative
|
, optparse-applicative
|
||||||
, case-insensitive
|
, case-insensitive
|
||||||
, iproute
|
, iproute
|
||||||
, clock
|
, clock
|
||||||
, tagged
|
, tagged
|
||||||
, jose >= 0.10
|
|
||||||
, aeson
|
|
||||||
|
|
||||||
hs-source-dirs: app
|
hs-source-dirs: app
|
||||||
default-language: Haskell2010
|
default-language: Haskell2010
|
||||||
ghc-options: -Wall
|
ghc-options: -Wall
|
||||||
-fwrite-ide-info
|
|
||||||
-hiedir=.hie
|
|
||||||
|
|
||||||
-- test-suite mateamt-test
|
-- test-suite mateamt-test
|
||||||
-- default-language: Haskell2010
|
-- default-language: Haskell2010
|
||||||
|
|
18
src/API.hs
18
src/API.hs
|
@ -13,14 +13,16 @@ import Servant.Server
|
||||||
|
|
||||||
import Data.Proxy
|
import Data.Proxy
|
||||||
|
|
||||||
import Servant.RawM.Server
|
import Servant.RawM
|
||||||
|
import Servant.RawM.Server ()
|
||||||
|
|
||||||
-- internal imports
|
-- internal imports
|
||||||
|
|
||||||
import Types
|
import Types
|
||||||
|
|
||||||
type MateAPI = "v1" :> (
|
type MateAPI = "v1" :> (
|
||||||
"auth" :> ReqBody '[JSON] AuthRequest :> Post '[JSON] String
|
"auth" :> "get" :> ReqBody '[JSON] TicketRequest :> Post '[JSON] AuthInfo
|
||||||
|
:<|> "auth" :> ReqBody '[JSON] AuthRequest :> Post '[JSON] AuthResult
|
||||||
:<|> "auth" :> AuthProtect "header-auth" :> Delete '[JSON] NoContent
|
:<|> "auth" :> AuthProtect "header-auth" :> Delete '[JSON] NoContent
|
||||||
|
|
||||||
:<|> "auth" :> "manage" :> AuthProtect "header-auth"
|
:<|> "auth" :> "manage" :> AuthProtect "header-auth"
|
||||||
|
@ -96,6 +98,7 @@ type MateAPI = "v1" :> (
|
||||||
:<|> "meta" :> Get '[JSON] MetaInformation
|
:<|> "meta" :> Get '[JSON] MetaInformation
|
||||||
)
|
)
|
||||||
|
|
||||||
|
authGetLink :: Link
|
||||||
authSendLink :: Link
|
authSendLink :: Link
|
||||||
authLogoutLink :: Link
|
authLogoutLink :: Link
|
||||||
|
|
||||||
|
@ -123,7 +126,7 @@ journalShowLink :: Maybe Int -> Maybe Int -> Link
|
||||||
journalPostCheck :: Link
|
journalPostCheck :: Link
|
||||||
|
|
||||||
avatarGetLink :: Int -> Link
|
avatarGetLink :: Int -> Link
|
||||||
avatarInsertLink :: Link
|
avaterInsertLink :: Link
|
||||||
avatarUpdateLink :: Int -> Link
|
avatarUpdateLink :: Int -> Link
|
||||||
avatarListLink :: Link
|
avatarListLink :: Link
|
||||||
|
|
||||||
|
@ -136,11 +139,12 @@ roleAssociationSubmitLink :: Link
|
||||||
roleAssociationDeleteLink :: Link
|
roleAssociationDeleteLink :: Link
|
||||||
|
|
||||||
settingsGetLink :: Link
|
settingsGetLink :: Link
|
||||||
settingsUpdateLink :: Link
|
settingsUpdateLnk :: Link
|
||||||
|
|
||||||
metaGetLink :: Link
|
metaGetLink :: Link
|
||||||
|
|
||||||
( authSendLink :<|>
|
( authGetLink :<|>
|
||||||
|
authSendLink :<|>
|
||||||
authLogoutLink :<|>
|
authLogoutLink :<|>
|
||||||
|
|
||||||
authManageListLink :<|>
|
authManageListLink :<|>
|
||||||
|
@ -167,7 +171,7 @@ metaGetLink :: Link
|
||||||
journalPostCheck :<|>
|
journalPostCheck :<|>
|
||||||
|
|
||||||
avatarGetLink :<|>
|
avatarGetLink :<|>
|
||||||
avatarInsertLink :<|>
|
avaterInsertLink :<|>
|
||||||
avatarUpdateLink :<|>
|
avatarUpdateLink :<|>
|
||||||
avatarListLink :<|>
|
avatarListLink :<|>
|
||||||
|
|
||||||
|
@ -180,7 +184,7 @@ metaGetLink :: Link
|
||||||
roleAssociationDeleteLink :<|>
|
roleAssociationDeleteLink :<|>
|
||||||
|
|
||||||
settingsGetLink :<|>
|
settingsGetLink :<|>
|
||||||
settingsUpdateLink :<|>
|
settingsUpdateLnk :<|>
|
||||||
|
|
||||||
metaGetLink
|
metaGetLink
|
||||||
) = allLinks (Proxy :: Proxy MateAPI)
|
) = allLinks (Proxy :: Proxy MateAPI)
|
||||||
|
|
|
@ -3,8 +3,6 @@ module Control.Auth where
|
||||||
|
|
||||||
import Servant
|
import Servant
|
||||||
|
|
||||||
import Control.Lens (re, view, review)
|
|
||||||
|
|
||||||
import Control.Monad (void)
|
import Control.Monad (void)
|
||||||
|
|
||||||
import Control.Monad.Reader (asks)
|
import Control.Monad.Reader (asks)
|
||||||
|
@ -12,14 +10,8 @@ import Control.Monad.IO.Class (liftIO)
|
||||||
|
|
||||||
import Control.Concurrent.STM (readTVarIO)
|
import Control.Concurrent.STM (readTVarIO)
|
||||||
|
|
||||||
import Crypto.Error
|
|
||||||
import Crypto.JWT (SignedJWT, encodeCompact, base64url)
|
|
||||||
import Crypto.KDF.Argon2
|
import Crypto.KDF.Argon2
|
||||||
|
import Crypto.Error
|
||||||
import Data.Aeson (encode)
|
|
||||||
|
|
||||||
import qualified Data.ByteString.Char8 as B8
|
|
||||||
import qualified Data.ByteString.Lazy as BL
|
|
||||||
|
|
||||||
import Data.String (fromString)
|
import Data.String (fromString)
|
||||||
|
|
||||||
|
@ -40,8 +32,11 @@ authGet (TicketRequest uid method) =
|
||||||
|
|
||||||
authSend
|
authSend
|
||||||
:: AuthRequest
|
:: AuthRequest
|
||||||
-> MateHandler String
|
-> MateHandler AuthResult
|
||||||
authSend req = B8.unpack . BL.toStrict . encodeCompact <$> (processAuthRequest req =<< asks rsConnection)
|
authSend req = uncurry (processAuthRequest req) =<< ((,) <$>
|
||||||
|
(liftIO . readTVarIO =<< asks rsTicketStore) <*>
|
||||||
|
asks rsConnection
|
||||||
|
)
|
||||||
|
|
||||||
authLogout
|
authLogout
|
||||||
:: Maybe (Int, AuthMethod)
|
:: Maybe (Int, AuthMethod)
|
||||||
|
|
|
@ -11,10 +11,8 @@ metaGet :: MateHandler MetaInformation
|
||||||
metaGet = do
|
metaGet = do
|
||||||
symbol <- asks rsCurrencySymbol
|
symbol <- asks rsCurrencySymbol
|
||||||
version <- asks rsSoftwareVersion
|
version <- asks rsSoftwareVersion
|
||||||
decimals <- asks rsCurrencyFraction
|
|
||||||
return (MetaInformation
|
return (MetaInformation
|
||||||
{ metaInfoVersion = version
|
{ metaInfoVersion = version
|
||||||
, metaInfoCurrency = symbol
|
, metaInfoCurrency = symbol
|
||||||
, metaInfoDecimals = fromIntegral decimals
|
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
|
@ -1,8 +1,6 @@
|
||||||
{-# LANGUAGE OverloadedStrings #-}
|
{-# LANGUAGE OverloadedStrings #-}
|
||||||
module Control.Settings where
|
module Control.Settings where
|
||||||
|
|
||||||
import Control.Monad (void, unless)
|
|
||||||
|
|
||||||
import Control.Monad.Reader
|
import Control.Monad.Reader
|
||||||
|
|
||||||
import Servant
|
import Servant
|
||||||
|
|
|
@ -4,7 +4,6 @@
|
||||||
{-# LANGUAGE ScopedTypeVariables #-}
|
{-# LANGUAGE ScopedTypeVariables #-}
|
||||||
{-# LANGUAGE RecordWildCards #-}
|
{-# LANGUAGE RecordWildCards #-}
|
||||||
{-# LANGUAGE FlexibleContexts #-}
|
{-# LANGUAGE FlexibleContexts #-}
|
||||||
{-# LANGUAGE FlexibleInstances #-}
|
|
||||||
module Model.Auth where
|
module Model.Auth where
|
||||||
|
|
||||||
import Servant
|
import Servant
|
||||||
|
@ -18,14 +17,7 @@ import Control.Monad.Reader (asks)
|
||||||
import Control.Concurrent (threadDelay)
|
import Control.Concurrent (threadDelay)
|
||||||
import Control.Concurrent.STM
|
import Control.Concurrent.STM
|
||||||
|
|
||||||
import Control.Lens
|
|
||||||
|
|
||||||
import Crypto.Error
|
import Crypto.Error
|
||||||
import Crypto.JWT
|
|
||||||
|
|
||||||
import qualified Data.Aeson as A
|
|
||||||
|
|
||||||
import Data.ByteString as B (ByteString, fromStrict)
|
|
||||||
|
|
||||||
import Data.Profunctor.Product (p4, p6)
|
import Data.Profunctor.Product (p4, p6)
|
||||||
|
|
||||||
|
@ -42,6 +34,9 @@ import qualified Data.Set as S
|
||||||
|
|
||||||
import Data.Time.Clock
|
import Data.Time.Clock
|
||||||
|
|
||||||
|
import Data.ByteString as B (ByteString)
|
||||||
|
import Data.ByteString.Base64 (encode)
|
||||||
|
|
||||||
import Opaleye hiding (null)
|
import Opaleye hiding (null)
|
||||||
|
|
||||||
-- internal imports
|
-- internal imports
|
||||||
|
@ -54,7 +49,7 @@ import Util.Crypto
|
||||||
initToken :: PGS.Query
|
initToken :: PGS.Query
|
||||||
initToken = mconcat
|
initToken = mconcat
|
||||||
[ "CREATE TABLE IF NOT EXISTS \"token\" ("
|
[ "CREATE TABLE IF NOT EXISTS \"token\" ("
|
||||||
, "token_id SERIAL PRIMARY KEY,"
|
, "token_string BYTEA NOT NULL PRIMARY KEY,"
|
||||||
, "token_user INTEGER REFERENCES \"user\"(user_id) NOT NULL,"
|
, "token_user INTEGER REFERENCES \"user\"(user_id) NOT NULL,"
|
||||||
, "token_expiry TIMESTAMPTZ NOT NULL,"
|
, "token_expiry TIMESTAMPTZ NOT NULL,"
|
||||||
, "token_method INT NOT NULL"
|
, "token_method INT NOT NULL"
|
||||||
|
@ -62,19 +57,19 @@ initToken = mconcat
|
||||||
]
|
]
|
||||||
|
|
||||||
tokenTable :: Table
|
tokenTable :: Table
|
||||||
( Maybe (Field SqlInt4)
|
( Field SqlBytea
|
||||||
, Field SqlInt4
|
, Field SqlInt4
|
||||||
, Field SqlTimestamptz
|
, Field SqlTimestamptz
|
||||||
, Field SqlInt4
|
, Field SqlInt4
|
||||||
)
|
)
|
||||||
( Field SqlInt4
|
( Field SqlBytea
|
||||||
, Field SqlInt4
|
, Field SqlInt4
|
||||||
, Field SqlTimestamptz
|
, Field SqlTimestamptz
|
||||||
, Field SqlInt4
|
, Field SqlInt4
|
||||||
)
|
)
|
||||||
tokenTable = table "token" (
|
tokenTable = table "token" (
|
||||||
p4
|
p4
|
||||||
( tableField "token_id"
|
( tableField "token_string"
|
||||||
, tableField "token_user"
|
, tableField "token_user"
|
||||||
, tableField "token_expiry"
|
, tableField "token_expiry"
|
||||||
, tableField "token_method"
|
, tableField "token_method"
|
||||||
|
@ -219,31 +214,22 @@ deleteAuthDataById adid conn = liftIO $ runDelete_ conn $ Delete
|
||||||
|
|
||||||
validateToken
|
validateToken
|
||||||
:: ByteString
|
:: ByteString
|
||||||
-> JWK
|
|
||||||
-> PGS.Connection
|
-> PGS.Connection
|
||||||
-> Handler (Maybe (Int, AuthMethod))
|
-> Handler (Maybe (Int, AuthMethod))
|
||||||
validateToken authHeader key conn = do
|
validateToken header conn = do
|
||||||
token <- either (error . show) id <$> liftIO (runJOSE $ do
|
|
||||||
jwt <- (decodeCompact (fromStrict authHeader) :: JOSE JWTError IO SignedJWT)
|
|
||||||
liftIO $ print jwt
|
|
||||||
let chk = defaultJWTValidationSettings (const True)
|
|
||||||
verifyJWT chk key jwt :: JOSE JWTError IO AuthResult
|
|
||||||
)
|
|
||||||
tokens <- liftIO $ map fromDatabase <$> runSelect conn (
|
tokens <- liftIO $ map fromDatabase <$> runSelect conn (
|
||||||
proc () -> do
|
proc () -> do
|
||||||
stuff@(_, tUser, _, tMethod) <- (selectTable tokenTable) -< ()
|
stuff@(tstr, _, _, _) <- (selectTable tokenTable) -< ()
|
||||||
restrict -<
|
restrict -< toFields header .== tstr
|
||||||
toFields (authUser token) .== tUser .&&
|
|
||||||
toFields (fromEnum $ authMethod token) .== tMethod
|
|
||||||
returnA -< stuff
|
returnA -< stuff
|
||||||
)
|
)
|
||||||
case tokens of
|
case tokens of
|
||||||
[Token tid uid stamp method] -> do
|
[Token _ uid stamp method] -> do
|
||||||
now_ <- liftIO getCurrentTime
|
now_ <- liftIO getCurrentTime
|
||||||
if diffUTCTime stamp now_ > 0
|
if diffUTCTime stamp now_ > 0
|
||||||
then return $ Just (uid, method)
|
then return $ Just (uid, method)
|
||||||
else do
|
else do
|
||||||
void $ deleteToken tid conn
|
void $ deleteToken header conn
|
||||||
liftIO $ threadDelay delayTime
|
liftIO $ threadDelay delayTime
|
||||||
throwError $ err401
|
throwError $ err401
|
||||||
{ errBody = "Your token expired!"
|
{ errBody = "Your token expired!"
|
||||||
|
@ -256,12 +242,11 @@ validateToken authHeader key conn = do
|
||||||
|
|
||||||
|
|
||||||
generateToken
|
generateToken
|
||||||
:: Int
|
:: Ticket
|
||||||
-> AuthMethod
|
|
||||||
-> AuthResponse
|
-> AuthResponse
|
||||||
-> PGS.Connection
|
-> PGS.Connection
|
||||||
-> MateHandler SignedJWT
|
-> MateHandler AuthResult
|
||||||
generateToken tuid method (AuthResponse response) conn = do
|
generateToken (Ticket _ tuid _ (method, _)) (AuthResponse response) conn = do
|
||||||
authData <- liftIO $ map fromDatabase <$> runSelect conn (
|
authData <- liftIO $ map fromDatabase <$> runSelect conn (
|
||||||
proc () -> do
|
proc () -> do
|
||||||
stuff@(_, auid, amethod, _, _, _) <- selectTable authDataTable -< ()
|
stuff@(_, auid, amethod, _, _, _) <- selectTable authDataTable -< ()
|
||||||
|
@ -278,22 +263,15 @@ generateToken tuid method (AuthResponse response) conn = do
|
||||||
-- liftIO $ print (response : userPayloads)
|
-- liftIO $ print (response : userPayloads)
|
||||||
if authResult
|
if authResult
|
||||||
then do
|
then do
|
||||||
key <- asks rsJWTSecret
|
token <- liftIO $ Token
|
||||||
issuedAt <- liftIO getCurrentTime
|
<$> (decodeUtf8 <$> randomString)
|
||||||
let preToken =
|
<*> pure tuid
|
||||||
( tuid
|
<*> (addUTCTime (23*60) <$> getCurrentTime)
|
||||||
, (addUTCTime (23*60) issuedAt)
|
<*> pure method
|
||||||
, method
|
void $ insertToken token conn
|
||||||
)
|
return $ Granted (AuthToken $ tokenString token)
|
||||||
liftIO $ print $ A.encode key
|
|
||||||
void $ insertToken preToken conn
|
|
||||||
let result = AuthResult issuedAt tuid method -- (AuthToken $ tokenString token)
|
|
||||||
signedJWT <- liftIO $ runJOSE (do
|
|
||||||
algo <- (bestJWSAlg key :: JOSE JWTError IO Alg)
|
|
||||||
signJWT key (newJWSHeader ((), algo)) result)
|
|
||||||
return $ either (error "Signing JWT failed") id signedJWT
|
|
||||||
else
|
else
|
||||||
throwError err401
|
return Denied
|
||||||
where
|
where
|
||||||
validatePass resp =
|
validatePass resp =
|
||||||
foldM
|
foldM
|
||||||
|
@ -309,39 +287,37 @@ generateToken tuid method (AuthResponse response) conn = do
|
||||||
)
|
)
|
||||||
False
|
False
|
||||||
validateChallengeResponse _ _ =
|
validateChallengeResponse _ _ =
|
||||||
throwError err501
|
error "Validation of challenge response authentication not yet implemented"
|
||||||
{ errBody = "Validation of challenge response authentication not yet implemented"
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
insertToken
|
insertToken
|
||||||
:: (Int, UTCTime, AuthMethod)
|
:: Token
|
||||||
-> PGS.Connection
|
-> PGS.Connection
|
||||||
-> MateHandler Int
|
-> MateHandler ByteString
|
||||||
insertToken (tUser, tExpiry, tMethod) conn =
|
insertToken (Token tString tUser tExpiry tMethod) conn =
|
||||||
fmap head $ liftIO $ runInsert_ conn $ Insert
|
fmap head $ liftIO $ runInsert_ conn $ Insert
|
||||||
{ iTable = tokenTable
|
{ iTable = tokenTable
|
||||||
, iRows =
|
, iRows =
|
||||||
[
|
[
|
||||||
( toFields (Nothing :: Maybe Int)
|
( toFields (encodeUtf8 tString)
|
||||||
, toFields tUser
|
, toFields tUser
|
||||||
, toFields tExpiry
|
, toFields tExpiry
|
||||||
, toFields (fromEnum tMethod)
|
, toFields (fromEnum tMethod)
|
||||||
)
|
)
|
||||||
]
|
]
|
||||||
, iReturning = rReturning (\(tid, _, _, _) -> tid)
|
, iReturning = rReturning (\(ident, _, _, _) -> ident)
|
||||||
, iOnConflict = Nothing
|
, iOnConflict = Nothing
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
deleteToken
|
deleteToken
|
||||||
:: Int
|
:: ByteString
|
||||||
-> PGS.Connection
|
-> PGS.Connection
|
||||||
-> Handler Int64
|
-> Handler Int64
|
||||||
deleteToken dtid conn =
|
deleteToken tstr conn =
|
||||||
liftIO $ runDelete_ conn $ Delete
|
liftIO $ runDelete_ conn $ Delete
|
||||||
{ dTable = tokenTable
|
{ dTable = tokenTable
|
||||||
, dWhere = \(tid, _, _, _) -> tid .== toFields dtid
|
, dWhere = \(rtstr, _, _, _) -> rtstr .== toFields tstr
|
||||||
, dReturning = rCount
|
, dReturning = rCount
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -388,14 +364,25 @@ newTicket ident method = do
|
||||||
|
|
||||||
processAuthRequest
|
processAuthRequest
|
||||||
:: AuthRequest
|
:: AuthRequest
|
||||||
|
-> S.Set Ticket
|
||||||
-> PGS.Connection
|
-> PGS.Connection
|
||||||
-> MateHandler SignedJWT
|
-> MateHandler AuthResult
|
||||||
processAuthRequest (AuthRequest user method pass) conn = do
|
processAuthRequest (AuthRequest aticket pass) store conn = do
|
||||||
-- liftIO $ putStrLn "there is a ticket..."
|
let mticket = S.filter (\st -> ticketId st == aticket) store
|
||||||
now_ <- liftIO getCurrentTime
|
case S.toList mticket of
|
||||||
liftIO $ threadDelay delayTime
|
[ticket] -> do
|
||||||
-- liftIO $ putStrLn "...and it is valid"
|
-- liftIO $ putStrLn "there is a ticket..."
|
||||||
generateToken user method pass conn
|
now_ <- liftIO getCurrentTime
|
||||||
|
liftIO $ threadDelay delayTime
|
||||||
|
if now_ > ticketExpiry ticket
|
||||||
|
then
|
||||||
|
return Denied
|
||||||
|
else
|
||||||
|
-- liftIO $ putStrLn "...and it is valid"
|
||||||
|
generateToken ticket pass conn
|
||||||
|
_ -> do
|
||||||
|
liftIO $ threadDelay delayTime
|
||||||
|
return Denied
|
||||||
|
|
||||||
processLogout
|
processLogout
|
||||||
:: Int
|
:: Int
|
||||||
|
|
|
@ -25,7 +25,7 @@ initSettings :: PGS.Query
|
||||||
initSettings = mconcat
|
initSettings = mconcat
|
||||||
[ "CREATE TABLE IF NOT EXISTS \"settings\" ("
|
[ "CREATE TABLE IF NOT EXISTS \"settings\" ("
|
||||||
, "settings_signup_blocked BOOLEAN NOT NULL DEFAULT false,"
|
, "settings_signup_blocked BOOLEAN NOT NULL DEFAULT false,"
|
||||||
, "settings_imprint TEXT DEFAULT \'\',"
|
, "settings_imprint TEXT DEFAULT \'\'"
|
||||||
, "settings_idle_time INTEGER NOT NULL DEFAULT 30"
|
, "settings_idle_time INTEGER NOT NULL DEFAULT 30"
|
||||||
, ");"
|
, ");"
|
||||||
, "INSERT INTO \"settings\" DEFAULT VALUES"
|
, "INSERT INTO \"settings\" DEFAULT VALUES"
|
||||||
|
|
|
@ -1,14 +1,10 @@
|
||||||
{-# LANGUAGE DeriveGeneric #-}
|
{-# LANGUAGE DeriveGeneric #-}
|
||||||
|
{-# LANGUAGE TypeSynonymInstances #-}
|
||||||
{-# LANGUAGE TypeFamilies #-}
|
{-# LANGUAGE TypeFamilies #-}
|
||||||
{-# LANGUAGE FlexibleInstances #-}
|
|
||||||
module Types.Auth where
|
module Types.Auth where
|
||||||
|
|
||||||
import GHC.Generics
|
import GHC.Generics
|
||||||
|
|
||||||
import Control.Lens.Lens
|
|
||||||
|
|
||||||
import Crypto.JWT
|
|
||||||
|
|
||||||
import Data.Aeson
|
import Data.Aeson
|
||||||
|
|
||||||
import qualified Data.Set as S
|
import qualified Data.Set as S
|
||||||
|
@ -113,8 +109,7 @@ instance FromJSON AuthResponse
|
||||||
instance ToSchema AuthResponse
|
instance ToSchema AuthResponse
|
||||||
|
|
||||||
data AuthRequest = AuthRequest
|
data AuthRequest = AuthRequest
|
||||||
{ authRequestUser :: Int
|
{ authRequestTicket :: AuthTicket
|
||||||
, authRequestMethod :: AuthMethod
|
|
||||||
, authRequestPassword :: AuthResponse
|
, authRequestPassword :: AuthResponse
|
||||||
}
|
}
|
||||||
deriving (Show, Generic)
|
deriving (Show, Generic)
|
||||||
|
@ -125,24 +120,18 @@ instance ToJSON AuthRequest where
|
||||||
instance FromJSON AuthRequest
|
instance FromJSON AuthRequest
|
||||||
instance ToSchema AuthRequest
|
instance ToSchema AuthRequest
|
||||||
|
|
||||||
data AuthResult = AuthResult
|
data AuthResult
|
||||||
{ authTime :: UTCTime
|
= Granted
|
||||||
, authUser :: Int
|
{ authToken :: AuthToken
|
||||||
, authMethod :: AuthMethod
|
}
|
||||||
-- , authToken :: AuthToken
|
| Denied
|
||||||
}
|
deriving (Show, Generic)
|
||||||
deriving (Show, Generic)
|
|
||||||
|
|
||||||
instance ToJSON AuthResult where
|
instance ToJSON AuthResult where
|
||||||
toEncoding = genericToEncoding defaultOptions
|
toEncoding = genericToEncoding defaultOptions
|
||||||
|
|
||||||
instance FromJSON AuthResult
|
instance FromJSON AuthResult
|
||||||
instance ToSchema AuthResult
|
instance ToSchema AuthResult
|
||||||
instance HasClaimsSet AuthResult where
|
|
||||||
claimsSet = lens (const emptyClaimsSet) const
|
|
||||||
|
|
||||||
instance FromJSON Token
|
|
||||||
instance ToSchema Token
|
|
||||||
|
|
||||||
newtype AuthToken = AuthToken T.Text deriving (Show, Generic)
|
newtype AuthToken = AuthToken T.Text deriving (Show, Generic)
|
||||||
|
|
||||||
|
@ -153,7 +142,7 @@ instance FromJSON AuthToken
|
||||||
instance ToSchema AuthToken
|
instance ToSchema AuthToken
|
||||||
|
|
||||||
data Token = Token
|
data Token = Token
|
||||||
{ tokenId :: Int
|
{ tokenString :: T.Text
|
||||||
, tokenUser :: Int
|
, tokenUser :: Int
|
||||||
, tokenExpiry :: UTCTime
|
, tokenExpiry :: UTCTime
|
||||||
, tokenMethod :: AuthMethod
|
, tokenMethod :: AuthMethod
|
||||||
|
@ -169,12 +158,12 @@ data Token = Token
|
||||||
|
|
||||||
instance DatabaseRepresentation Token where
|
instance DatabaseRepresentation Token where
|
||||||
|
|
||||||
type Representation Token = (Int, Int, UTCTime, Int)
|
type Representation Token = (ByteString, Int, UTCTime, Int)
|
||||||
|
|
||||||
instance FromDatabase Token where
|
instance FromDatabase Token where
|
||||||
|
|
||||||
fromDatabase (tId, usr, expiry, method) =
|
fromDatabase (string, usr, expiry, method) =
|
||||||
Token tId usr expiry (toEnum method)
|
Token (decodeUtf8 string) usr expiry (toEnum method)
|
||||||
|
|
||||||
|
|
||||||
type TicketStore = TVar (S.Set Ticket)
|
type TicketStore = TVar (S.Set Ticket)
|
||||||
|
|
|
@ -1,5 +1,3 @@
|
||||||
{-# LANGUAGE TypeSynonymInstances #-}
|
|
||||||
{-# LANGUAGE FlexibleInstances #-}
|
|
||||||
module Types.Reader where
|
module Types.Reader where
|
||||||
|
|
||||||
import qualified Data.Text as T
|
import qualified Data.Text as T
|
||||||
|
@ -7,11 +5,7 @@ import qualified Data.Text as T
|
||||||
import Servant (Handler)
|
import Servant (Handler)
|
||||||
|
|
||||||
import Control.Concurrent.STM (TQueue)
|
import Control.Concurrent.STM (TQueue)
|
||||||
import Control.Monad.IO.Class (liftIO)
|
|
||||||
import Control.Monad.Reader (ReaderT)
|
import Control.Monad.Reader (ReaderT)
|
||||||
import Control.Monad.Time
|
|
||||||
|
|
||||||
import Crypto.JOSE.JWK
|
|
||||||
|
|
||||||
import Database.PostgreSQL.Simple (Connection)
|
import Database.PostgreSQL.Simple (Connection)
|
||||||
|
|
||||||
|
@ -29,10 +23,6 @@ data ReadState = ReadState
|
||||||
, rsSoftwareVersion :: T.Text
|
, rsSoftwareVersion :: T.Text
|
||||||
, rsSendmailPath :: FilePath
|
, rsSendmailPath :: FilePath
|
||||||
, rsMailQueue :: TQueue Mail
|
, rsMailQueue :: TQueue Mail
|
||||||
, rsJWTSecret :: JWK
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type MateHandler = ReaderT ReadState Handler
|
type MateHandler = ReaderT ReadState Handler
|
||||||
|
|
||||||
instance MonadTime MateHandler where
|
|
||||||
currentTime = liftIO currentTime
|
|
||||||
|
|
|
@ -72,7 +72,6 @@ initDB conn = do
|
||||||
void $ execute_ conn initJournal
|
void $ execute_ conn initJournal
|
||||||
void $ execute_ conn initRole
|
void $ execute_ conn initRole
|
||||||
void $ execute_ conn initUserToRole
|
void $ execute_ conn initUserToRole
|
||||||
void $ execute_ conn initSettings
|
|
||||||
void $ runInsertInitialRoles conn
|
void $ runInsertInitialRoles conn
|
||||||
|
|
||||||
-- This is only a dummy function.
|
-- This is only a dummy function.
|
||||||
|
|
|
@ -1,5 +1,4 @@
|
||||||
{-# LANGUAGE OverloadedStrings #-}
|
{-# LANGUAGE OverloadedStrings #-}
|
||||||
{-# LANGUAGE PackageImports #-}
|
|
||||||
module AppMainSpec where
|
module AppMainSpec where
|
||||||
|
|
||||||
import Database.PostgreSQL.Transact
|
import Database.PostgreSQL.Transact
|
||||||
|
@ -10,11 +9,11 @@ import Control.Monad.IO.Class (liftIO)
|
||||||
|
|
||||||
-- internal imports
|
-- internal imports
|
||||||
|
|
||||||
import qualified "mateamt" Util as LibUtil
|
import Util
|
||||||
import TestUtil
|
import TestUtil
|
||||||
|
|
||||||
spec :: Spec
|
spec :: Spec
|
||||||
spec = describeDB (const $ return ()) "Initialize DB" $
|
spec = describeDB (const $ return ()) "Initialize DB" $
|
||||||
itDB "call initialization script" $ do
|
itDB "calls initialization script" $ do
|
||||||
conn <- getConnection
|
conn <- getConnection
|
||||||
liftIO $ shouldReturn (LibUtil.initDB conn) ()
|
liftIO $ shouldReturn (initDB conn) ()
|
||||||
|
|
|
@ -1,4 +1,3 @@
|
||||||
{-# LANGUAGE PackageImports #-}
|
|
||||||
module Main where
|
module Main where
|
||||||
|
|
||||||
import Test.Hspec
|
import Test.Hspec
|
||||||
|
|
Loading…
Reference in a new issue