3cc3c5c986
This is largely Aaron Levin's work. But it was done in a merge commit,
so couldn't be cherry-picked in a sensible way. Thanks Aaron!
54 lines
1.2 KiB
Haskell
54 lines
1.2 KiB
Haskell
{-# 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
|
|
:<|> Raw
|
|
|
|
|
|
api :: Proxy API
|
|
api = Proxy
|
|
|
|
|
|
server :: WaiMonad () -> Server API
|
|
server sHandler = socketIOHandler
|
|
:<|> serveDirectory "socket-io-chat/resources"
|
|
|
|
where
|
|
socketIOHandler req respond = toWaiApplication sHandler req respond
|
|
|
|
|
|
app :: WaiMonad () -> Application
|
|
app sHandler = serve api EmptyConfig $ 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
|
|
|
|
|