RemoteId.hs

module Rattletrap.RemoteId where

import Rattletrap.Primitive
import Rattletrap.Utility

import qualified Data.Binary.Bits.Get as BinaryBit
import qualified Data.Binary.Bits.Put as BinaryBit
import qualified Data.ByteString.Lazy as ByteString
import qualified Data.Text as Text
import qualified Data.Text.Encoding as Encoding
import qualified Data.Word as Word

data RemoteId
  = PlayStationId Text.Text
                  [Word.Word8]
  | SplitscreenId Word.Word32
  | SteamId Word64
  | XboxId Word64
  deriving (Eq, Ord, Show)

getRemoteId :: Word8 -> BinaryBit.BitGet RemoteId
getRemoteId systemId =
  case word8Value systemId of
    0 -> do
      word24 <- BinaryBit.getWord32be 24
      pure (SplitscreenId word24)
    1 -> do
      word64 <- getWord64Bits
      pure (SteamId word64)
    2 -> do
      rawName <- BinaryBit.getLazyByteString 16
      let name =
            Text.dropWhileEnd
              (== '\x00')
              (Encoding.decodeLatin1
                 (ByteString.toStrict (reverseBytes rawName)))
      bytes <- BinaryBit.getLazyByteString 16
      pure (PlayStationId name (ByteString.unpack bytes))
    4 -> do
      word64 <- getWord64Bits
      pure (XboxId word64)
    _ -> fail ("unknown system id " ++ show systemId)

putRemoteId :: RemoteId -> BinaryBit.BitPut ()
putRemoteId remoteId =
  case remoteId of
    PlayStationId name bytes -> do
      let rawName =
            ByteString.toStrict
              (reverseBytes (padBytes (16 :: Int) (encodeLatin1 name)))
      BinaryBit.putByteString rawName
      BinaryBit.putByteString (ByteString.toStrict (ByteString.pack bytes))
    SplitscreenId word24 -> BinaryBit.putWord32be 24 word24
    SteamId word64 -> putWord64Bits word64
    XboxId word64 -> putWord64Bits word64