blob: 48b4483b24d0c59f3ee8fccd1838e37a96b09770 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
module WebSocket (
WebSocketAddress(..),
startWebsocketServer,
) where
import Control.Concurrent
import Control.Exception
import Control.Monad
import Data.ByteString.Lazy qualified as BL
import Data.Unique
import Erebos.Network
import Network.WebSockets qualified as WS
data WebSocketAddress = WebSocketAddress Unique WS.Connection
instance Eq WebSocketAddress where
WebSocketAddress u _ == WebSocketAddress u' _ = u == u'
instance Ord WebSocketAddress where
compare (WebSocketAddress u _) (WebSocketAddress u' _) = compare u u'
instance Show WebSocketAddress where
show (WebSocketAddress _ _) = "websocket"
instance PeerAddressType WebSocketAddress where
sendBytesToAddress (WebSocketAddress _ conn) msg = do
WS.sendDataMessage conn $ WS.Binary $ BL.fromStrict msg
connectionToAddressClosed (WebSocketAddress _ conn) = do
WS.sendClose conn BL.empty `catch` \e -> if
| Just WS.ConnectionClosed <- fromException e -> return ()
| otherwise -> throwIO e
startWebsocketServer :: Server -> String -> Int -> (String -> IO ()) -> IO ()
startWebsocketServer server addr port logd = do
void $ forkIO $ do
WS.runServer addr port $ \pending -> do
conn <- WS.acceptRequest pending
u <- newUnique
let paddr = WebSocketAddress u conn
void $ serverPeerCustom server paddr
let handler e
| Just WS.CloseRequest {} <- fromException e = do
dropPeerAddress server $ CustomPeerAddress paddr
| Just WS.ConnectionClosed <- fromException e = do
dropPeerAddress server $ CustomPeerAddress paddr
| otherwise = do
logd $ "WebSocket thread exception: " ++ show e
handle handler $ do
WS.withPingThread conn 30 (return ()) $ do
forever $ do
WS.receiveDataMessage conn >>= \case
WS.Binary msg -> receivedFromCustomAddress server paddr $ BL.toStrict msg
WS.Text {} -> logd $ "unexpected websocket text message"
|