socket.hs

{-# LANGUAGE OverloadedStrings #-}

import           Control.Concurrent        (forkIO)
import           Control.Monad             (forever, unless)

import qualified Data.ByteString.Lazy      as BL

import           Network.Connection
import qualified Network.WebSockets        as WS
import qualified Network.WebSockets.Stream as WS

import           Data.Text                 (Text)
import qualified Data.Text                 as T
import qualified Data.Text.IO              as T

import           Data.Maybe                (maybe)

app :: WS.ClientApp ()
app conn = do
    putStrLn "Connected!"

    _ <- forkIO $ forever $ do
          msg <- WS.receiveData conn
          T.putStrLn msg

    let loop = do
          line <- T.getLine
          unless (T.null line) $ WS.sendTextData conn line >> loop

    loop
    WS.sendClose conn ("Bye!" :: Text)

main :: IO ()
main = do
  ctx <- initConnectionContext
  con <- connectTo ctx $ ConnectionParams
                          { connectionHostname  = hostname
                          , connectionPort      = port
                          , connectionUseSecure =
                            Just $ TLSSettingsSimple
                              { settingDisableCertificateValidation = True
                              , settingDisableSession = False
                              , settingUseServerName  = False
                              }
                          , connectionUseSocks  = Nothing
                          }
  stream <- WS.makeStream (fmap Just $ connectionGetChunk con)
                          (maybe (return ()) (connectionPut con . BL.toStrict))
  WS.runClientWithStream stream hostname path WS.defaultConnectionOptions [] app
 where
   hostname = "echo.websocket.org"
   port = 443
   path = "/"