servant/servant-examples/socket-io-chat/socket-io-chat.hs

{-# LANGUAGE CPP               #-}
{-# LANGUAGE DataKinds         #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TypeOperators     #-}


import           Data.Monoid              ((<>))
#if !MIN_VERSION_base(4,8,0)
import           Control.Applicative      ((<$>))
#endif
import           Network.EngineIO.Wai
import           Network.Wai
import           Network.Wai.Handler.Warp (run)
import           Servant


import qualified Control.Concurrent.STM   as STM
import qualified Network.SocketIO         as SocketIO


import           Chat                     (ServerState (..), eioServer)


type API = "socket.io" :> Raw IO Application
    :<|> Raw IO Application


api :: Proxy API
api = Proxy


server :: WaiMonad () -> Server API
server sHandler = socketIOHandler
    :<|> serveDirectory "socket-io-chat/resources"

    where
        socketIOHandler = Raw $ toWaiApplication sHandler


app :: WaiMonad () -> Application
app sHandler = serve api $ server sHandler

port :: Int
port = 3001


main :: IO ()
main = do
    state <- ServerState <$> STM.newTVarIO 0
    sHandler <- SocketIO.initialize waiAPI (eioServer state)
    putStrLn $ "Running on " <> show port
    run port $ app sHandler