bytemapper 1.0.13 → 1.0.19

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b7915e2b8be3bcd3f677ed08c5a82e8fd135ab941ff9ab8b04c2148cfc7b8441
-  data.tar.gz: c44bf0499a0d15dc77728770edf55a995a12b00987e76ea651eb249c09f0d2e7
+  metadata.gz: 9651d363a3ac3e436372c5527f7f8bed34015ea0687eaa93b3b89b31bd0bce81
+  data.tar.gz: 7ed18fc3bfd35527aea3eef93e038adc785019ae4857d00008639512d18bf974
 SHA512:
-  metadata.gz: ea2aa731fde9ad3ef8f22a08d1b0abfa4de8803e7d7849af1ad1c0aebc9c865640e9e62f1ebc91293025c1b76b21cb5bd1b469d510005dead3a6723a6b4ef314
-  data.tar.gz: 9cbb50a69eea518d210ac6fb7bcaeeb9a74ff8b0e6ad3a9ade5f68988d6839c7d2464e9c276781d45fac856704de011352463c60a3b03d563ff90fec16ccf488
+  metadata.gz: 6d1b379ef5e3322c9b010377702cc7d4c5d1043ebbb9038a56e723df6537c558e6c4ca9b8f0b812015de4dc1199cb7043ada03f00d3850292ef60e55949ab20f
+  data.tar.gz: 93db768253293483ff66349868b577704e75cce5f18169cd7b0eb3fc7e52906ebb1c0d0a28b944d260918bbc21987a59d84dfff1f3de08af7d30ff37be0b84a0

data/lib/bytemapper.rb

@@ -1,4 +1,4 @@
-# Bytemapper - Model arbitrary bytestrings as Ruby objects.  
+# Bytemapper - Model arbitrary bytestrings as Ruby objects.
 # Copyright (C) 2020 Jefferson Hudson
 #
 # This program is free software: you can redistribute it and/or modify it under
@@ -16,54 +16,87 @@
 
 module Bytemapper
   require 'bytemapper/registry'
+  require 'bytemapper/shape'
+  require 'bytemapper/type'
   require 'bytemapper/nameable'
-  require 'bytemapper/flattenable'
   require 'bytemapper/chunk'
 
   @@registry = Registry.new
 
-  def self.wrap(obj, name = nil, wrapper = {})
-    if (obj.is_a?(Array) || obj.is_a?(String) || obj.is_a?(Symbol))
-      obj = registry.get(obj, name)
-      raise ArgumentError.new "Failed to resolve symbol #{name}" if obj.nil?
-    elsif obj.is_a?(Hash)
-      obj.each do |k, v|
-        wrapper[k] = wrap(v, k)
-        wrapper.define_singleton_method(k) { self.send(:fetch, k) }
+  class << self
+
+    def register(obj, name, fqname = [])
+      return if obj.nil?
+      name = name.downcase.to_sym unless name.nil?
+      fqname << name unless name.nil?
+      if is_a_type?(obj)
+        name = fqname.size > 1 ? fqname.join('.') : fqname.first
+        obj = Type.new(obj)
+        put(obj, name)
+      elsif is_a_name?(obj)
+        register(get(obj), nil, fqname)
+      elsif is_a_shape?(obj)
+        if registered?(obj)
+          obj = get(obj)
+          put(obj, name)
+        else
+          shape = Shape.new
+          obj.each do |k,v|
+            shape[k] = register(v, k, [].concat(fqname))
+          end
+          put(shape, name)
+        end
+      else
+        put(obj, name)
       end
-      wrapper.extend(Flattenable)
-      obj = registry.put(wrapper, name)
-    else
-      raise ArgumentError.new "Invalid object"
     end
-    obj
-  end
+    alias :wrap :register
 
-  def self.map(bytes, shape, name = nil)
-    bytes.force_encoding(Encoding::ASCII_8BIT)
-    bytes = StringIO.new(bytes)
-    wrapper = self.wrap(shape, name)
-    Chunk.new(bytes, wrapper, name)
-  end
+    def is_a_type?(obj)
+      obj.is_a?(Type) ||
+      obj.is_a?(Array) &&
+      obj.size == 2 &&
+      obj.first.is_a?(Integer) &&
+      obj.last.is_a?(String)
+    end
 
-  def self.registry
-    @@registry
-  end
+    def is_a_shape?(obj)
+      obj.is_a?(Hash)
+    end
+
+    def is_a_name?(obj)
+      obj.is_a?(String) || obj.is_a?(Symbol)
+    end
+
+    def map(bytes, shape, name = nil)
+      bytes.force_encoding(Encoding::ASCII_8BIT)
+      bytes = StringIO.new(bytes)
+      shape = wrap(shape, name)
+      Chunk.new(bytes, shape, name)
+    end
+
+    def registered?(obj)
+      registry.registered?(obj)
+    end
+
+    def get(obj)
+      registry.get(obj)
+    end
+
+    def put(obj, name)
+      registry.put(obj, name)
+    end
+
+    def print
+      registry.print
+    end
+
+    def registry
+      @@registry
+    end
 
-  def reset(with_basic_types = true)
-    [
-      [:uint8_t, [8,'C']],
-      [:bool, [8,'C']],
-      [:uint16_t, [16,'S']],
-      [:uint32_t, [32,'L']],
-      [:uint64_t, [64,'Q']],
-      [:int8_t, [8,'c']],
-      [:int16_t, [16,'s']],
-      [:int32_t, [32,'l']],
-      [:int64_t, [64,'q']]
-    ].each do |name, type|
-      @@registry.put(type, name)
+    def reset(with_basic_types = true)
+      @@registry = Registry.new(with_basic_types)
     end
-    @@registry
   end
 end

data/lib/bytemapper/chunk.rb

@@ -15,17 +15,25 @@
 # along with this program. If not, see <https://www.gnu.org/licenses/>.
 
 module Bytemapper
-  class Chunk
+  require 'bytemapper/flattenable'
+  class Chunk < Hash
+    include Flattenable
     attr_reader :bytes, :shape, :name
 
     def initialize(bytes, shape, name)
-      @bytes = bytes 
-      @shape = shape
       @name = name
-
-      shape.flatten.each do |k,v|
+      @shape = shape
+      @bytes = bytes.is_a?(StringIO) ? bytes : StringIO.new(bytes)
+      @bytes.truncate(shape.size)
+      replace(shape)
+      each_pair do |k,v|
+        self[k] = if v.is_a?(Hash)
+                    Chunk.new(@bytes.read(v.size), v, k)
+                  else
+                    unpack(v)
+                  end
         singleton_class.instance_eval { attr_reader k }
-        instance_variable_set("@#{k.to_s}", unpack(v))
+        instance_variable_set("@#{k.to_s}", self[k])
       end
     end
 
@@ -42,7 +50,10 @@ module Bytemapper
     end
 
     def size
-      @bytes.size
+      bytes.size
+    end
+
+    def print
     end
 
     def unpack(value, endian = nil)

data/lib/bytemapper/registry.rb

@@ -1,18 +1,19 @@
-# Bytemapper - Model arbitrary bytestrings as Ruby objects.  
+# Bytemapper - Model arbitrary bytestrings as Ruby objects.
 # Copyright (C) 2020 Jefferson Hudson
 #
 # This program is free software: you can redistribute it and/or modify it under
 # the terms of the GNU Affero General Public License as published by the Free
 # Software Foundation, either version 3 of the License, or (at your option) any
 # later version.
-
+#
 # This program is distributed in the hope that it will be useful, but WITHOUT
 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 # FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
 # details.
-
+#
 # You should have received a copy of the GNU Affero General Public License
 # along with this program. If not, see <https://www.gnu.org/licenses/>.
+
 module Bytemapper
   class Registry
     require 'bytemapper/nameable'
@@ -36,56 +37,62 @@ module Bytemapper
     end
 
     def registered?(obj)
-      names.key?(obj) || objects.key?(obj.hash)
+      v = registered_name?(obj) if obj.respond_to?(:to_sym)
+      v || registered_obj?(obj)
     end
 
-    def get(obj, name = nil)
-      if (obj.is_a?(String) || obj.is_a?(Symbol))
-        name = obj
-        key = @names[obj]
-      elsif (obj.is_a?(Array) || obj.is_a?(Hash))
-        put(obj, name)
-        key = obj.hash
+    def get(obj)
+      if registered_name?(obj)
+        obj = obj.to_sym.downcase
+        @objects.fetch(@names[obj]) unless obj.nil?
+      elsif registered_obj?(obj)
+        @objects.fetch(obj.hash)
       else
-        raise ArgumentError "Invalid obj"
+        nil
       end
-      @objects[key]
     end
 
     def put(obj, name = nil)
       obj = register_obj(obj)
-      register_name(obj, name)
+      register_name(obj, name) unless name.nil?
       obj
     end
 
     def registered_name?(name)
-      @names.key?(name) 
+      if name.respond_to?(:to_sym)
+        name = name.to_sym.downcase
+        @names.key?(name)
+      else
+        false
+      end
     end
 
     def registered_obj?(obj)
-      @objects.key?(obj.hash) 
+      @objects.key?(obj.hash)
     end
 
     def register_obj(obj)
       unless registered_obj?(obj)
-        obj.extend(Nameable)
-        obj.names = Set.new
+        begin
+          obj.extend(Nameable)
+          obj.names = Set.new
+        rescue TypeError
+        end
       end
       @objects[obj.hash] ||= obj
     end
 
     def register_name(obj, name)
-      unless name.nil?
-        if registered_name?(name) && get(name) != obj
-          raise ArgumentError.new 'Name is already registered' 
-        else
-          name = name.to_sym
-          @names[name] ||= obj.hash
-          obj.names << name
-        end
+      if registered_name?(name) && get(name) != obj
+        raise ArgumentError.new 'Name is already registered'
+      else
+        name = name.to_sym.downcase
+        @names[name] ||= obj.hash
+        obj.names << name if obj.respond_to?(:names)
       end
       obj
     end
+    alias :register_alias :register_name
 
     def print
       puts to_s
@@ -97,7 +104,7 @@ module Bytemapper
       # Buffer to build up output.
       buf = StringIO.new
 
-      # Calculate the width of each column.
+      # Calculate the true max width of each column.
       widths = [
         @names.keys.size.zero? ? 0 : @names.keys.map(&:size).max + 1, # add space for the `:`
         7, # length of ID to print
@@ -105,6 +112,10 @@ module Bytemapper
         @objects.values.map { |v| v.to_s.size }.max
       ]
 
+      # Truncate more than `max_width` num chars
+      max_width = 60
+      widths = widths.map { |w| w > max_width ? max_width : w }
+
       # Add an extra space at the beginning and end of each column.
       widths = widths.map { |p| p += 2 }
 
@@ -125,21 +136,29 @@ module Bytemapper
 
           # Fixup the id string so it pads nicely
           idstr = id.positive? ? id.to_s[..5] : id.to_s[..6]
-          idstr = id.positive? ? "  #{idstr}" : " #{idstr}"
+          idstr = id.positive? ? " #{idstr}" : "#{idstr}"
 
-          # Wrap each column value with whitespace.
+          # Generate the column values
           values = [
-          name.empty? ? name : " :#{name} ",
-          idstr,
-          " #{obj.class.to_s} ",
-          " #{obj.to_s} "
+            name.empty? ? name : ":#{name}",
+            idstr,
+            "#{obj.class.to_s}",
+            "#{obj.to_s}"
           ]
-          
-          # Calculate padding for each column.
-          pads = widths.zip(values).map { |a,b| a - b.size }
+
+          # Pad the values to fit in their respective columns, truncating as
+          # needed to stay within `max_width`
+          values = widths.zip(values).map do |w,v|
+            q = w - v.size
+            if q < 2
+              " #{v[..w-7]} ... "
+            else
+              " #{v}#{" "*(q-1)}"
+            end
+          end
 
           values.size.times do |i|
-            buf << "#{values[i]}#{' '*pads[i]}|"
+            buf << "#{values[i]}|"
           end
           buf << "\n"
         end
@@ -163,15 +182,20 @@ module Bytemapper
     def register_basic_types
       [
         [:uint8_t, [8,'C']],
+        [:uchar, [8,'C']],
         [:bool, [8,'C']],
         [:uint16_t, [16,'S']],
         [:uint32_t, [32,'L']],
         [:uint64_t, [64,'Q']],
         [:int8_t, [8,'c']],
+        [:char, [8,'c']],
         [:int16_t, [16,'s']],
         [:int32_t, [32,'l']],
         [:int64_t, [64,'q']]
-      ].each { |name, type| put(type, name) }
+      ].each do |name, type|
+        type = Type.new(type)
+        put(type, name)
+      end
     end
   end
 end

data/lib/bytemapper/shape.rb

@@ -0,0 +1,30 @@
+# Bytemapper - Model arbitrary bytestrings as Ruby objects.
+# Copyright (C) 2020 Jefferson Hudson
+#
+# This program is free software: you can redistribute it and/or modify it under
+# the terms of the GNU Affero General Public License as published by the Free
+# Software Foundation, either version 3 of the License, or (at your option) any
+# later version.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
+# details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+module Bytemapper
+  require 'bytemapper/flattenable'
+  require 'bytemapper/nameable'
+  class Shape < Hash
+    include Flattenable
+    include Nameable
+
+    def []=(k,v)
+      super
+      singleton_class.instance_eval { attr_reader k }
+      instance_variable_set("@#{k.to_s}", self[k])
+    end
+  end
+end

data/lib/bytemapper/type.rb

@@ -0,0 +1,26 @@
+# Bytemapper - Model arbitrary bytestrings as Ruby objects.
+# Copyright (C) 2020 Jefferson Hudson
+#
+# This program is free software: you can redistribute it and/or modify it under
+# the terms of the GNU Affero General Public License as published by the Free
+# Software Foundation, either version 3 of the License, or (at your option) any
+# later version.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
+# details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+module Bytemapper
+  require 'bytemapper/nameable'
+  class Type < Array
+    include Nameable
+
+    def size
+      first / 8
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bytemapper
 version: !ruby/object:Gem::Version
-  version: 1.0.13
+  version: 1.0.19
 platform: ruby
 authors:
 - Jefferson Hudson
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-05-26 00:00:00.000000000 Z
+date: 2020-06-01 00:00:00.000000000 Z
 dependencies: []
 description: Model and interact with bytestrings using Ruby objects.
 email: jefferson.hudson@gmail.com
@@ -21,6 +21,8 @@ files:
 - lib/bytemapper/flattenable.rb
 - lib/bytemapper/nameable.rb
 - lib/bytemapper/registry.rb
+- lib/bytemapper/shape.rb
+- lib/bytemapper/type.rb
 homepage: https://github.com/l4cr0ss/bytemapper
 licenses:
 - AGPL-3.0-or-later