File: mixin.rb

package info (click to toggle)
ruby-packable 1.3.14-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 204 kB
  • sloc: ruby: 609; makefile: 4
file content (61 lines) | stat: -rw-r--r-- 1,767 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# The Packable mixin itself...

require 'stringio'

module Packable
  def self.included(base) #:nodoc:
    base.class_eval do
      class << self
        include PackersClassMethod
        include ClassMethods
      end
    end
  end

  # +options+ can be a Hash, a shortcut (Symbol) or a String (old-style)
  def pack(options = :default)
    return [self].pack(options) if options.is_a? String
    (StringIO.new.packed << [self, options]).string
  end

  module PackersClassMethod

    # Returns or yields the <tt>Packers.for(class)</tt>
    # Normal use is packers.set ...
    # (see docs or Packers::set for usage)
    #
    def packers
      yield packers if block_given?
      Packers.for(self)
    end
  end

  module ClassMethods

    def unpack(s, options = :default)
      return s.unpack(options).first if options.is_a? String
      StringIO.new(s).packed.read(self, options)
    end

    # Default +read_packed+ calls either the instance method <tt>read_packed</tt> or the
    # class method +unpack_string+. Choose:
    # * define a class method +read_packed+ that returns the newly read object
    # * define an instance method +read_packed+ which reads the io into +self+
    # * define a class method +unpack_string+ that reads and returns an object from the string. In this case, options[:bytes] should be specified!
    #
    def read_packed(io, options)
      if method_defined? :read_packed
        mandatory = instance_method(:initialize).arity
        mandatory = -1-mandatory if mandatory < 0
        obj = new(*[nil]*mandatory)
        obj.read_packed(io, options)
        obj
      else
        len = options[:bytes]
        s = len ? io.read_exactly(len) : io.read
        unpack_string(s, options) unless s.nil?
      end
    end

  end
end