-
Notifications
You must be signed in to change notification settings - Fork 0
/
Controller.hs
52 lines (44 loc) · 1.66 KB
/
Controller.hs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
{-# LANGUAGE TemplateHaskell #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE OverloadedStrings #-}
{-# OPTIONS_GHC -fno-warn-orphans #-}
module Controller
( withYesod2
, withDevelApp
) where
import Yesod2
import Settings
import Yesod.Helpers.Static
import Yesod.Helpers.Auth
import Yesod.Helpers.Crud
import Database.Persist.GenericSql
import Data.ByteString (ByteString)
import Data.Dynamic (Dynamic, toDyn)
-- Import all relevant handler modules here.
import Handler.Root
import Handler.Products
productCrud :: Yesod2 -> Crud Yesod2 Product
productCrud = defaultCrud
-- This line actually creates our YesodSite instance. It is the second half
-- of the call to mkYesodData which occurs in Yesod2.hs. Please see
-- the comments there for more details.
mkYesodDispatch "Yesod2" resourcesYesod2
-- 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.
withYesod2 :: (Application -> IO a) -> IO a
withYesod2 f = Settings.withConnectionPool $ \p -> do
runConnectionPool (runMigration migrateAll) p
let h = Yesod2 s p
toWaiApp h >>= f
where
s = static Settings.staticdir
withDevelApp :: Dynamic
withDevelApp = toDyn (withYesod2 :: (Application -> IO ()) -> IO ())