protobuf: Translate unsigned integers to corresponding Word types (#246)

Author: akshaymankar

adapter/protobuf/src/Mu/Adapter/ProtoBuf.hs

@@ -44,6 +44,7 @@ import           Data.Int
 import           Data.SOP                 (All)
 import qualified Data.Text                as T
 import qualified Data.Text.Lazy           as LT
+import           Data.Word                (Word64, Word32)
 import           GHC.TypeLits
 import           Proto3.Wire
 import qualified Proto3.Wire.Decode       as PBDec
@@ -420,6 +421,22 @@ instance ProtoBridgeOneFieldValue sch ('TPrimitive Int64) where
     = PBEnc.packedVarints fid $ map (\(FPrimitive i) -> fromIntegral i) vs
   protoToPackedFieldValue = map FPrimitive <$> PBDec.packedVarints
 
+instance ProtoBridgeOneFieldValue sch ('TPrimitive Word32) where
+  defaultOneFieldValue = Just $ FPrimitive 0
+  oneFieldValueToProto fid (FPrimitive n) = PBEnc.uint32 fid n
+  protoToOneFieldValue = FPrimitive <$> PBDec.uint32
+  supportsPacking _ = True
+  packedFieldValueToProto fid vs = PBEnc.packedVarints fid $ map (\(FPrimitive i) -> fromIntegral i) vs
+  protoToPackedFieldValue = map FPrimitive <$> PBDec.packedVarints
+
+instance ProtoBridgeOneFieldValue sch ('TPrimitive Word64) where
+  defaultOneFieldValue = Just $ FPrimitive 0
+  oneFieldValueToProto fid (FPrimitive n) = PBEnc.uint64 fid n
+  protoToOneFieldValue = FPrimitive <$> PBDec.uint64
+  supportsPacking _ = True
+  packedFieldValueToProto fid vs = PBEnc.packedVarints fid $ map (\(FPrimitive i) -> i) vs
+  protoToPackedFieldValue = map FPrimitive <$> PBDec.packedVarints
+
 -- WARNING! These instances may go out of bounds
 instance ProtoBridgeOneFieldValue sch ('TPrimitive Integer) where
   defaultOneFieldValue = Just $ FPrimitive 0

adapter/protobuf/src/Mu/Quasi/ProtoBuf.hs

@@ -22,6 +22,7 @@ import           Control.Monad                   (when)
 import           Control.Monad.IO.Class
 import qualified Data.ByteString                 as B
 import           Data.Int
+import           Data.Word
 import qualified Data.Text                       as T
 import           Language.Haskell.TH
 import           Language.ProtocolBuffers.Parser
@@ -112,10 +113,10 @@ pbTypeDeclToType (P.DMessage name _ _ fields _) = do
 
     pbFieldTypeToType :: P.FieldType -> Q Type
     pbFieldTypeToType P.TInt32     = [t|'TPrimitive Int32|]
-    pbFieldTypeToType P.TUInt32    = fail "unsigned integers are not currently supported"
+    pbFieldTypeToType P.TUInt32    = [t|'TPrimitive Word32|]
     pbFieldTypeToType P.TSInt32    = [t|'TPrimitive Int32|]
     pbFieldTypeToType P.TInt64     = [t|'TPrimitive Int64|]
-    pbFieldTypeToType P.TUInt64    = fail "unsigned integers are not currently supported"
+    pbFieldTypeToType P.TUInt64    = [t|'TPrimitive Word64|]
     pbFieldTypeToType P.TSInt64    = [t|'TPrimitive Int64|]
     pbFieldTypeToType P.TFixed32   = fail "fixed integers are not currently supported"
     pbFieldTypeToType P.TFixed64   = fail "fixed integers are not currently supported"