More pair -> Entity changes

This commit is contained in:
Michael Snoyman 2012-01-10 11:20:17 +02:00
parent f57efbe1b0
commit ca9bb6e449
3 changed files with 6 additions and 5 deletions

View File

@ -90,7 +90,7 @@ import Data.Digest.Pure.SHA (sha1, showDigest)
import Data.Text (Text, pack, unpack, append) import Data.Text (Text, pack, unpack, append)
import Data.Maybe (fromMaybe) import Data.Maybe (fromMaybe)
import System.Random (randomRIO) import System.Random (randomRIO)
import Database.Persist.Store (Entity (..))
-- | Interface for data type which holds user info. It's just a -- | Interface for data type which holds user info. It's just a
-- collection of getters and setters -- collection of getters and setters
@ -153,7 +153,7 @@ validateUser userID passwd = do
return $ hash == saltedHash salt passwd return $ hash == saltedHash salt passwd
-- Get user data -- Get user data
user <- runDB $ getBy userID user <- runDB $ getBy userID
return $ fromMaybe False $ validate . snd =<< user return $ fromMaybe False $ validate . entityVal =<< user
login :: AuthRoute login :: AuthRoute
@ -206,7 +206,7 @@ getAuthIdHashDB authR uniq creds = do
Just u -> runDB (getBy u) Just u -> runDB (getBy u)
case x of case x of
-- user exists -- user exists
Just (uid, _) -> return $ Just uid Just (Entity uid _) -> return $ Just uid
Nothing -> do Nothing -> do
setMessage [QQ(shamlet)| User not found |] setMessage [QQ(shamlet)| User not found |]
redirect $ authR LoginR redirect $ authR LoginR

View File

@ -53,7 +53,7 @@ import Data.Time (Day, TimeOfDay(..))
import qualified Text.Email.Validate as Email import qualified Text.Email.Validate as Email
import Network.URI (parseURI) import Network.URI (parseURI)
import Database.Persist (PersistField) import Database.Persist (PersistField)
import Database.Persist.Query (Entity (..)) import Database.Persist.Store (Entity (..))
import Text.HTML.SanitizeXSS (sanitizeBalance) import Text.HTML.SanitizeXSS (sanitizeBalance)
import Control.Monad (when, unless) import Control.Monad (when, unless)
import Data.List (intersect, nub) import Data.List (intersect, nub)

View File

@ -12,6 +12,7 @@ module Yesod.Persist
import Database.Persist import Database.Persist
import Database.Persist.Query import Database.Persist.Query
import Database.Persist.Store (Entity (..))
import Database.Persist.TH import Database.Persist.TH
import Control.Monad.Trans.Class (MonadTrans) import Control.Monad.Trans.Class (MonadTrans)
@ -45,7 +46,7 @@ getBy404 :: ( PersistUnique b m
, Monad (b m) , Monad (b m)
, MonadTrans b , MonadTrans b
) )
=> Unique val b -> b m (Key b val, val) => Unique val b -> b m (Entity b val)
getBy404 key = do getBy404 key = do
mres <- getBy key mres <- getBy key
case mres of case mres of