module Rattletrap.Type.Attribute.Enum where

import qualified Data.Word as Word
import qualified Rattletrap.BitGet as BitGet
import qualified Rattletrap.BitPut as BitPut
import qualified Rattletrap.Schema as Schema
import qualified Rattletrap.Utility.Json as Json
import Prelude hiding (Enum)

newtype Enum = Enum
  { value :: Word.Word16
  }
  deriving (Eq, Show)

instance Json.FromJSON Enum where
  parseJSON = fmap Enum . Json.parseJSON

instance Json.ToJSON Enum where
  toJSON = Json.toJSON . value

schema :: Schema.Schema
schema = Schema.named "attribute-enum" $ Schema.ref Schema.integer

bitPut :: Enum -> BitPut.BitPut
bitPut enumAttribute = BitPut.bits 11 (value enumAttribute)

bitGet :: BitGet.BitGet Enum
bitGet = BitGet.label "Enum" $ do
  value <- BitGet.label "value" $ BitGet.bits 11
  pure Enum {value}