refactor: Internal helper #_primary_type

Author: DanSnow

src/binary_parser/macros/_primary_type.cr

@@ -0,0 +1,34 @@
+class BinaryParser
+  # :nodoc:
+  macro _primary_type(name, suffix)
+    # Internal helper for define primary type like `UInt32`
+    #
+    # ## Argument:
+    # - name: field name
+    # - suffix: suffix for declare, like `u32`
+
+    _primary_type({{name}}, {{suffix}}, {{suffix.starts_with?("u") ? "UInt#{suffix[1..-1].id}" : "Int#{suffix[1..-1].id}"}})
+  end
+
+  # :nodoc:
+  macro _primary_type(name, suffix, type)
+    # ## Argument:
+    # - name: field name
+    # - suffix: suffix for declare, like `u32`
+    # - type: like `UInt32`
+    property! :{{name.id}}
+    @{{name.id}} = 0{{suffix.id}}
+
+    def _read_{{name.id}}(io : IO)
+      @{{name.id}} = io.not_nil!.read_bytes({{type.id}}).as({{type.id}})
+    end
+
+    def _write_{{name.id}}(io : IO)
+      io.not_nil!.write_bytes(@{{name.id}}.not_nil!)
+    end
+
+    def _size_static_{{name.id}}
+      sizeof({{type.id}})
+    end
+  end
+end

src/binary_parser/macros/int16.cr

@@ -1,3 +1,5 @@
+require "./_primary_type"
+
 class BinaryParser
   #
   # Declare a int16 field
@@ -6,19 +8,6 @@ class BinaryParser
   # int16 :value # name of field
   # ```
   macro int16(name)
-    property! :{{name.id}}
-    @{{name.id}} = 0i16
-
-    def _read_{{name.id}}(io : IO)
-      @{{name.id}} = io.not_nil!.read_bytes(Int16).as(Int16)
-    end
-
-    def _write_{{name.id}}(io : IO)
-      io.not_nil!.write_bytes(@{{name.id}}.not_nil!)
-    end
-
-    def _size_static_{{name.id}}
-      sizeof(Int16)
-    end
+    _primary_type({{name}}, "i16")
   end
 end

src/binary_parser/macros/int32.cr

@@ -1,23 +1,12 @@
+require "./_primary_type"
+
 class BinaryParser
   # Declare a int32 field
   #
   # ```crystal
   # int32 :value # name of field
   # ```
   macro int32(name)
-    property! :{{name.id}}
-    @{{name.id}} = 0i32
-
-    def _read_{{name.id}}(io : IO)
-      @{{name.id}} = io.not_nil!.read_bytes(Int32).as(Int32)
-    end
-
-    def _write_{{name.id}}(io : IO)
-      io.not_nil!.write_bytes(@{{name.id}}.not_nil!)
-    end
-
-    def _size_static_{{name.id}}
-      sizeof(Int32)
-    end
+    _primary_type({{name}}, "i32")
   end
 end

src/binary_parser/macros/int64.cr

@@ -1,23 +1,12 @@
+require "./_primary_type"
+
 class BinaryParser
   # Declare a int64 field
   #
   # ```crystal
   # int64 :value # name of field
   # ```
   macro int64(name)
-    property! :{{name.id}}
-    @{{name.id}} = 0i64
-
-    def _read_{{name.id}}(io : IO)
-      @{{name.id}} = io.not_nil!.read_bytes(Int64).as(Int64)
-    end
-
-    def _write_{{name.id}}(io : IO)
-      io.not_nil!.write_bytes(@{{name.id}}.not_nil!)
-    end
-
-    def _size_static_{{name.id}}
-      sizeof(Int64)
-    end
+    _primary_type({{name}}, "i64")
   end
 end

src/binary_parser/macros/int8.cr

@@ -1,23 +1,12 @@
+require "./_primary_type"
+
 class BinaryParser
   # Declare a int8 field
   #
   # ```crystal
   # int8 :value # name of field
   # ```
   macro int8(name)
-    property! :{{name.id}}
-    @{{name.id}} = 0i8
-
-    def _read_{{name.id}}(io : IO)
-      @{{name.id}} = io.not_nil!.read_bytes(Int8).as(Int8)
-    end
-
-    def _write_{{name.id}}(io : IO)
-      io.not_nil!.write_bytes(@{{name.id}}.not_nil!)
-    end
-
-    def _size_static_{{name.id}}
-      sizeof(Int8)
-    end
+    _primary_type({{name}}, "i8")
   end
 end

src/binary_parser/macros/uint16.cr

@@ -1,23 +1,12 @@
+require "./_primary_type"
+
 class BinaryParser
   # Declare a uint16 field
   #
   # ```crystal
   # uint16 :value # name of field
   # ```
   macro uint16(name)
-    property! :{{name.id}}
-    @{{name.id}} = 0u16
-
-    def _read_{{name.id}}(io : IO)
-      @{{name.id}} = io.not_nil!.read_bytes(UInt16).as(UInt16)
-    end
-
-    def _write_{{name.id}}(io : IO)
-      io.not_nil!.write_bytes(@{{name.id}}.not_nil!)
-    end
-
-    def _size_static_{{name.id}}
-      sizeof(UInt16)
-    end
+    _primary_type({{name}}, "u16")
   end
 end

src/binary_parser/macros/uint32.cr

@@ -1,23 +1,12 @@
+require "./_primary_type"
+
 class BinaryParser
   # Declare a uint32 field
   #
   # ```crystal
   # uint32 :value # name of field
   # ```
   macro uint32(name)
-    property! :{{name.id}}
-    @{{name.id}} = 0u32
-
-    def _read_{{name.id}}(io : IO)
-      @{{name.id}} = io.not_nil!.read_bytes(UInt32).as(UInt32)
-    end
-
-    def _write_{{name.id}}(io : IO)
-      io.not_nil!.write_bytes(@{{name.id}}.not_nil!)
-    end
-
-    def _size_static_{{name.id}}
-      sizeof(UInt32)
-    end
+    _primary_type({{name}}, "u32")
   end
 end

src/binary_parser/macros/uint64.cr

@@ -1,23 +1,12 @@
+require "./_primary_type"
+
 class BinaryParser
   # Declare a uint64 field
   #
   # ```crystal
   # uint64 :value # name of field
   # ```
   macro uint64(name)
-    property! :{{name.id}}
-    @{{name.id}} = 0u64
-
-    def _read_{{name.id}}(io : IO)
-      @{{name.id}} = io.not_nil!.read_bytes(UInt64).as(UInt64)
-    end
-
-    def _write_{{name.id}}(io : IO)
-      io.not_nil!.write_bytes(@{{name.id}}.not_nil!)
-    end
-
-    def _size_static_{{name.id}}
-      sizeof(UInt64)
-    end
+    _primary_type({{name}}, "u64")
   end
 end

src/binary_parser/macros/uint8.cr

@@ -1,23 +1,12 @@
+require "./_primary_type"
+
 class BinaryParser
   # Declare a uint8 field
   #
   # ```crystal
   # uint8 :value # name of field
   # ```
   macro uint8(name)
-    property! :{{name.id}}
-    @{{name.id}} = 0u8
-
-    def _read_{{name.id}}(io : IO)
-      @{{name.id}} = io.not_nil!.read_bytes(UInt8).as(UInt8)
-    end
-
-    def _write_{{name.id}}(io : IO)
-      io.not_nil!.write_bytes(@{{name.id}}.not_nil!)
-    end
-
-    def _size_static_{{name.id}}
-      sizeof(UInt8)
-    end
+    _primary_type({{name}}, "u8")
   end
 end