2020-05-10 06:35:32 +00:00
|
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
{-# LANGUAGE FlexibleContexts #-}
|
|
|
|
{-# LANGUAGE Arrows #-}
|
|
|
|
{-# LANGUAGE ScopedTypeVariables #-}
|
|
|
|
module Model.Role where
|
2020-05-16 08:02:26 +00:00
|
|
|
|
|
|
|
import qualified Database.PostgreSQL.Simple as PGS
|
|
|
|
|
|
|
|
import Data.Profunctor.Product (p2, p13)
|
|
|
|
|
|
|
|
import Opaleye as O hiding (null, not)
|
|
|
|
import Opaleye.Constant as C
|
|
|
|
|
2020-07-19 04:24:22 +00:00
|
|
|
import Control.Monad
|
2020-05-16 14:12:49 +00:00
|
|
|
import Control.Monad.IO.Class (liftIO)
|
|
|
|
|
|
|
|
-- internal imports
|
|
|
|
|
|
|
|
import Types
|
|
|
|
|
2020-05-16 08:02:26 +00:00
|
|
|
initRole :: PGS.Query
|
|
|
|
initRole = mconcat
|
|
|
|
[ "CREATE TABLE IF NOT EXISTS \"role\" ("
|
|
|
|
, "role_id SERIAL NOT NULL,"
|
|
|
|
, "role_name TEXT NOT NULL,"
|
|
|
|
, "role_can_refill_stock BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_add_product BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_view_journal BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_pay_invoice BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_pay_out BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_manage_products BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_manage_journal BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_manage_users BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_manage_roles BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_manage_suppliers BOOLEAN NOT NULL,"
|
|
|
|
, "role_can_manage_settings BOOLEAN NOT NULL,"
|
|
|
|
, "PRIMARY KEY (role_id)"
|
2020-07-19 04:31:35 +00:00
|
|
|
, "UNIQUE (role_name)"
|
2020-05-16 08:02:26 +00:00
|
|
|
, ")"
|
|
|
|
]
|
|
|
|
|
|
|
|
initUserToRole :: PGS.Query
|
|
|
|
initUserToRole = mconcat
|
|
|
|
[ "CREATE TABLE IF NOT EXISTS \"user_to_role\" ("
|
|
|
|
, "user_id INTEGER NOT NULL,"
|
|
|
|
, "role_id INTEGER NOT NULL,"
|
|
|
|
, "PRIMARY KEY (user_id, role_id)"
|
|
|
|
, ")"
|
|
|
|
]
|
|
|
|
|
|
|
|
roleTable :: Table
|
|
|
|
( Maybe (Field SqlInt4)
|
|
|
|
, Field SqlText
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
)
|
|
|
|
( Field SqlInt4
|
|
|
|
, Field SqlText
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
, Field SqlBool
|
|
|
|
)
|
|
|
|
roleTable = table "role" (
|
|
|
|
p13
|
|
|
|
( tableField "role_id"
|
|
|
|
, tableField "role_name"
|
|
|
|
, tableField "role_can_refill_stock"
|
|
|
|
, tableField "role_can_add_product"
|
|
|
|
, tableField "role_can_view_journal"
|
|
|
|
, tableField "role_can_pay_invoice"
|
|
|
|
, tableField "role_can_pay_out"
|
|
|
|
, tableField "role_can_manage_products"
|
|
|
|
, tableField "role_can_manage_journal"
|
|
|
|
, tableField "role_can_manage_users"
|
|
|
|
, tableField "role_can_manage_roles"
|
|
|
|
, tableField "role_can_manage_suppliers"
|
|
|
|
, tableField "role_can_manage_settings"
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
userToRoleTable :: Table
|
|
|
|
( Field SqlInt4
|
|
|
|
, Field SqlInt4
|
|
|
|
)
|
|
|
|
( Field SqlInt4
|
|
|
|
, Field SqlInt4
|
|
|
|
)
|
|
|
|
userToRoleTable = table "user_to_role" (
|
|
|
|
p2
|
|
|
|
( tableField "user_id"
|
|
|
|
, tableField "role_id"
|
|
|
|
)
|
|
|
|
)
|
2020-05-16 13:33:27 +00:00
|
|
|
|
2020-07-19 04:24:22 +00:00
|
|
|
insertInitialRoles :: PGS.Connection -> MateHandler ()
|
|
|
|
insertInitialRoles conn =
|
|
|
|
void $ liftIO $ runInsertInitialRoles conn
|
2020-05-16 13:33:27 +00:00
|
|
|
|
2020-07-19 04:24:22 +00:00
|
|
|
runInsertInitialRoles :: PGS.Connection -> IO [Int]
|
|
|
|
runInsertInitialRoles conn = do
|
|
|
|
a <- runInsert_ conn $ Insert
|
2020-05-16 13:33:27 +00:00
|
|
|
{ iTable = roleTable
|
|
|
|
, iRows =
|
|
|
|
[
|
|
|
|
( C.constant (Nothing :: Maybe Int)
|
2020-05-16 14:12:49 +00:00
|
|
|
, C.constant ("Administrator" :: String)
|
2020-05-16 13:33:27 +00:00
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
, C.constant True
|
|
|
|
)
|
|
|
|
]
|
2020-07-19 04:24:22 +00:00
|
|
|
, iReturning = rReturning (\(id_, _, _, _, _, _, _, _, _, _, _, _, _) -> id_ )
|
2020-05-16 13:33:27 +00:00
|
|
|
, iOnConflict = Nothing
|
|
|
|
}
|
2020-07-19 04:24:22 +00:00
|
|
|
b <- runInsert_ conn $ Insert
|
|
|
|
{ iTable = roleTable
|
|
|
|
, iRows =
|
|
|
|
[
|
|
|
|
( C.constant (Nothing :: Maybe Int)
|
|
|
|
, C.constant ("User" :: String)
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
, C.constant False
|
|
|
|
)
|
|
|
|
]
|
|
|
|
, iReturning = rReturning (\(id_, _, _, _, _, _, _, _, _, _, _, _, _) -> id_ )
|
|
|
|
, iOnConflict = Nothing
|
|
|
|
}
|
|
|
|
return $ a ++ b
|
2020-07-19 05:01:53 +00:00
|
|
|
|
|
|
|
queryRoleIdByName
|
|
|
|
:: T.Text
|
|
|
|
-> PGS.Connection
|
|
|
|
-> MateHandler Int
|
|
|
|
queryRoleIdByName name con = do
|
|
|
|
roles <- liftIO $ runSelect conn (
|
|
|
|
keepWhen (\(_, rname, _, _, _, _, _, _, _, _, _, _, _) ->
|
|
|
|
C.constant name == rname) <<< queryTable roleTable
|
|
|
|
) :: MateHandler
|
|
|
|
( Int
|
|
|
|
, T.Text
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
)
|
|
|
|
return $ (\(rid, _, _, _, _, _, _, _, _, _, _, _, _) -> rid) (head roles)
|
|
|
|
|
|
|
|
queryRoleIdByCapabilities
|
|
|
|
:: (Bool, Bool, Bool, Bool, Bool, Bool, Bool, Bool, Bool, Bool, Bool)
|
|
|
|
-> PGS.Connection
|
|
|
|
-> MateHandler Int
|
|
|
|
queryRoleIdByName caps con = do
|
|
|
|
roles <- liftIO $ runSelect conn (
|
|
|
|
keepWhen (\(_, _, c1, c2, c3, c4, c5, c6, c7, c8, c9, c10, c11) ->
|
|
|
|
C.constant caps == (c1, c2, c3, vc4, c5, c6, c7, c8, c9, c10, c11))
|
|
|
|
<<< queryTable roleTable
|
|
|
|
) :: MateHandler
|
|
|
|
( Int
|
|
|
|
, T.Text
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
, Bool
|
|
|
|
)
|
|
|
|
return $ (\(rid, _, _, _, _, _, _, _, _, _, _, _, _) -> rid) (head roles)
|
|
|
|
|
|
|
|
associateUserToRole
|
|
|
|
:: Int -- ^ User id
|
|
|
|
:: Int -- ^ Role id
|
|
|
|
:: PGS.Connection
|
|
|
|
:: Matehandler Int -- ^ Resulting UserToRole id
|
|
|
|
associateUserToRole uid rid conn =
|
|
|
|
a <- runInsert_ conn $ Insert
|
|
|
|
{ iTable = userToRoleTable
|
|
|
|
, iRows =
|
|
|
|
[
|
|
|
|
( C.constant (Nothing :: Maybe Int)
|
|
|
|
, C.constant uid
|
|
|
|
, C.constant rid
|
|
|
|
)
|
|
|
|
]
|
|
|
|
, iReturning = rReturning (\(id_, _, _) -> id_ )
|
|
|
|
, iOnConflict = Nothing
|
|
|
|
}
|