wizard-wipeout/src-server/Server/Communication/Receive.hs

{-# LANGUAGE OverloadedStrings #-}
module Server.Communication.Receive where

import Control.Concurrent (threadDelay)

import qualified Control.Concurrent.STM as STM

import Control.Exception

import Control.Monad.IO.Class

import qualified Data.Aeson as A

import qualified Data.ByteString as B
import qualified Data.ByteString.Lazy.Char8 as B8

import Foreign.Marshal hiding (void)

import Network.Socket

import System.Random

-- internal imports

import Library.Types

import Server.Communication.Send
import Server.Types

-- | receive a 'ClientMessage'
receiveMessage
  :: STM.TMVar Socket
  -> STM.TQueue ClientMessage
  -> IO ()
receiveMessage sockContainer queue = do
  -- randSleep <- randomRIO (1, 1000)
  -- threadDelay randSleep
  sock <- STM.atomically $ STM.readTMVar sockContainer
  let maxBufferLength = 4096
  putStrLn "read socket container for receiving"
  ptr <- mallocArray maxBufferLength
  putStrLn "receiving data"
  eBufferLength <-
    try $ recvBuf sock ptr maxBufferLength
  putStrLn $ "received raw buffer length of: " <> show eBufferLength
  bufferLength <- case eBufferLength of
    Left (e :: IOException) -> do
      -- putStrLn ("Socket vanished, cleaning up after " <> show e)
      -- dropClient clientList sock
      pure 0
    Right len -> pure len
  putStrLn $ "received buffer of length: " <> show bufferLength
  rawMsg <- B.pack <$> peekArray bufferLength ptr
  free ptr
  putStrLn $ "received data: " <> show rawMsg
  let msgs =
        if B.length rawMsg < 1
        then [] ::  [B8.ByteString]
        else map B8.tail $ init $ B8.split '>' $ B8.fromStrict rawMsg
  putStrLn $ "received messages: " <> show msgs
  print msgs
  mapM_
    (\msg -> do
      let mJsonMsg = A.decode' msg
      maybe
        (putStrLn $ "received garbled data: " <> B8.unpack (B8.fromStrict rawMsg))
        (\jsonMsg -> do
          STM.atomically $ STM.writeTQueue queue jsonMsg
        )
        mJsonMsg
    )
    msgs