|
| 1 | +{-# LANGUAGE GeneralizedNewtypeDeriving #-} |
| 2 | +{-# LANGUAGE OverloadedStrings #-} |
| 3 | +{-# LANGUAGE TemplateHaskell #-} |
| 4 | + |
| 5 | +module Demo where |
| 6 | + |
| 7 | +import Control.Exception.Safe as E |
| 8 | +import Control.Exception.Safe ( MonadCatch |
| 9 | + , MonadMask |
| 10 | + , MonadThrow |
| 11 | + ) |
| 12 | +import Control.Monad.IO.Class ( MonadIO |
| 13 | + , liftIO |
| 14 | + ) |
| 15 | +import Control.Monad.Logger as L |
| 16 | +import Control.Monad.Logger ( LoggingT |
| 17 | + , MonadLogger |
| 18 | + ) |
| 19 | +import Control.Monad.Reader ( MonadReader |
| 20 | + , ReaderT |
| 21 | + , ask |
| 22 | + , runReaderT |
| 23 | + ) |
| 24 | +import Control.Monad.Trans.Class ( MonadTrans |
| 25 | + , lift |
| 26 | + ) |
| 27 | +import Data.Default.Class ( Default |
| 28 | + , def |
| 29 | + ) |
| 30 | +import Data.IP ( IP ) |
| 31 | +import Data.String ( fromString ) |
| 32 | +import Network.HTTP.Types as HTTP |
| 33 | +import Network.HTTP.Types.Header as HTTPHeaders |
| 34 | +import Network.Socket ( PortNumber ) |
| 35 | +import Network.Wai ( Application ) |
| 36 | +import Network.Wai as Wai |
| 37 | +import Network.Wai.Handler.Warp as Warp |
| 38 | +import System.Environment ( lookupEnv ) |
| 39 | + |
| 40 | + |
| 41 | +data DemoEnv = DemoEnv { demoEnvHost :: IP, |
| 42 | + demoEnvPort :: PortNumber |
| 43 | + } deriving (Show) |
| 44 | + |
| 45 | +-- this is the default demoenv |
| 46 | +instance Default DemoEnv where |
| 47 | + def = DemoEnv "127.0.0.1" 55555 |
| 48 | + |
| 49 | +newtype DemoT m a = DemoT |
| 50 | + { runDemoT :: ReaderT DemoEnv (LoggingT m) a } |
| 51 | + deriving ( |
| 52 | + Functor, |
| 53 | + Applicative, |
| 54 | + Monad, |
| 55 | + MonadIO, |
| 56 | + MonadThrow, |
| 57 | + MonadCatch, |
| 58 | + MonadLogger, |
| 59 | + MonadReader DemoEnv |
| 60 | + ) |
| 61 | + |
| 62 | +type Demo = DemoT IO |
| 63 | + |
| 64 | +runDemo :: DemoEnv -> Demo a -> IO a |
| 65 | +runDemo env demo = do |
| 66 | + runStderrLoggingT $ runReaderT (runDemoT demo) env |
| 67 | + |
| 68 | +warpApp :: Application |
| 69 | +warpApp req respond = E.bracket_ |
| 70 | + (runStderrLoggingT ($(L.logInfo) "Try IO Block")) |
| 71 | + (runStderrLoggingT ($(L.logInfo) "Clean IO Block")) |
| 72 | + (respond $ Wai.responseLBS HTTP.status200 |
| 73 | + [(HTTPHeaders.hContentType, "text/plain")] |
| 74 | + "Hello from Demo!\n" |
| 75 | + ) |
| 76 | + |
| 77 | +demoApp :: Demo () |
| 78 | +demoApp = do |
| 79 | + $(L.logInfo) "Starting Demo Server" |
| 80 | + DemoEnv ip port <- ask |
| 81 | + let settings = |
| 82 | + Warp.setHost (fromString $ show ip) |
| 83 | + $ Warp.setPort (fromIntegral port) |
| 84 | + $ Warp.defaultSettings |
| 85 | + $(L.logInfo) $ fromString $ "Running on " ++ show ip ++ ":" ++ show port |
| 86 | + liftIO $ Warp.runSettings settings warpApp |
| 87 | + $(L.logInfo) "Terminated Demo Server" |
| 88 | + |
| 89 | +runDemoApp :: IO () |
| 90 | +runDemoApp = do |
| 91 | + demoHost <- lookupEnv "DEMO_HOST" |
| 92 | + demoPort <- lookupEnv "DEMO_PORT" |
| 93 | + let defaultEnv = def :: DemoEnv |
| 94 | + defaultEnv <- return $ maybe |
| 95 | + defaultEnv |
| 96 | + (\host -> defaultEnv { demoEnvHost = read host }) |
| 97 | + demoHost |
| 98 | + defaultEnv <- return $ maybe |
| 99 | + defaultEnv |
| 100 | + (\port -> defaultEnv { demoEnvPort = read port }) |
| 101 | + demoPort |
| 102 | + runDemo defaultEnv demoApp |
0 commit comments