64 lines
2.2 KiB
Haskell
64 lines
2.2 KiB
Haskell
{-# LANGUAGE CPP #-}
|
|
{-# LANGUAGE TemplateHaskell #-}
|
|
{-# LANGUAGE MultiParamTypeClasses #-}
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
{-# OPTIONS_GHC -fno-warn-orphans #-}
|
|
module Hledger.Web.Application (
|
|
withApp
|
|
,withDevelAppPort
|
|
)
|
|
where
|
|
|
|
import Data.Dynamic (Dynamic, toDyn)
|
|
import Network.Wai (Application)
|
|
import Network.Wai.Middleware.Debug (debugHandle)
|
|
import Yesod.Core hiding (AppConfig,loadConfig,appPort)
|
|
import Yesod.Logger (makeLogger, flushLogger, Logger, logLazyText, logString)
|
|
import Yesod.Static
|
|
|
|
import Hledger.Web.Foundation
|
|
import Hledger.Web.Handlers
|
|
import Hledger.Web.Options
|
|
import Hledger.Web.Settings
|
|
|
|
-- This line actually creates our YesodSite instance. It is the second half
|
|
-- of the call to mkYesodData which occurs in App.hs. Please see
|
|
-- the comments there for more details.
|
|
mkYesodDispatch "App" resourcesApp
|
|
|
|
-- 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.
|
|
withApp :: AppConfig -> Logger -> WebOpts -> (Application -> IO a) -> IO a
|
|
withApp conf logger opts f = do
|
|
#ifdef PRODUCTION
|
|
putStrLn $ "Production mode, using embedded web files"
|
|
let s = $(embed staticDir)
|
|
#else
|
|
putStrLn $ "Not in production mode, using web files from " ++ staticDir ++ "/"
|
|
s <- staticDevel staticDir
|
|
#endif
|
|
let a = App {settings=conf
|
|
,getLogger=logger
|
|
,getStatic=s
|
|
,appOpts=opts
|
|
}
|
|
toWaiApp a >>= f
|
|
|
|
-- for yesod devel
|
|
withDevelAppPort :: Dynamic
|
|
withDevelAppPort =
|
|
toDyn go
|
|
where
|
|
go :: ((Int, Application) -> IO ()) -> IO ()
|
|
go f = do
|
|
conf <- loadConfig Development
|
|
let port = appPort conf
|
|
logger <- makeLogger
|
|
logString logger $ "Devel application launched with default options, listening on port " ++ show port
|
|
withApp conf logger defwebopts $ \app -> f (port, debugHandle (logHandle logger) app)
|
|
flushLogger logger
|
|
where
|
|
logHandle logger msg = logLazyText logger msg >> flushLogger logger
|