57 lines
1.8 KiB
Plaintext
57 lines
1.8 KiB
Plaintext
{-# LANGUAGE TemplateHaskell #-}
|
|
{-# LANGUAGE MultiParamTypeClasses #-}
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
{-# OPTIONS_GHC -fno-warn-orphans #-}
|
|
module Controller
|
|
( with~sitearg~
|
|
, withDevelApp
|
|
) where
|
|
|
|
import ~sitearg~
|
|
import Settings
|
|
import Yesod.Helpers.Static
|
|
import Yesod.Helpers.Auth
|
|
import Database.Persist.GenericSql
|
|
import Data.ByteString (ByteString)
|
|
import Data.Dynamic (Dynamic, toDyn)
|
|
|
|
-- Import all relevant handler modules here.
|
|
import Handler.Root
|
|
|
|
-- This line actually creates our YesodSite instance. It is the second half
|
|
-- of the call to mkYesodData which occurs in ~sitearg~.hs. Please see
|
|
-- the comments there for more details.
|
|
mkYesodDispatch "~sitearg~" resources~sitearg~
|
|
|
|
-- Some default handlers that ship with the Yesod site template. You will
|
|
-- very rarely need to modify this.
|
|
getFaviconR :: Handler ()
|
|
getFaviconR = sendFile "image/x-icon" "config/favicon.ico"
|
|
|
|
getRobotsR :: Handler RepPlain
|
|
getRobotsR = return $ RepPlain $ toContent ("User-agent: *" :: ByteString)
|
|
|
|
-- This function allocates resources (such as a database connection pool),
|
|
-- performs initialization and creates a WAI application. This is also the
|
|
-- place to put your migrate statements to have automatic database
|
|
-- migrations handled by Yesod.
|
|
with~sitearg~ :: AppConfig -> (Application -> IO a) -> IO a
|
|
with~sitearg~ conf f = do
|
|
Settings.withConnectionPool conf $ \p -> do
|
|
runConnectionPool (runMigration migrateAll) p
|
|
let h = ~sitearg~ conf s p
|
|
toWaiApp h >>= f
|
|
where
|
|
s = static Settings.staticDir
|
|
|
|
with~sitearg~LoadConfig :: Settings.AppEnvironment -> (Application -> IO a) -> IO a
|
|
with~sitearg~LoadConfig env f = do
|
|
conf <- Settings.loadConfig env
|
|
withFoobar conf f
|
|
|
|
-- for yesod devel
|
|
withDevelApp :: Dynamic
|
|
withDevelApp = do
|
|
toDyn ((with~sitearg~LoadConfig Settings.Development):: (Application -> IO ()) -> IO ())
|
|
|