summaryrefslogtreecommitdiff
path: root/lib/ostruct.rb
blob: 8bc0628ac46b0fbecd4935bb92f30d46607ac88c (plain)
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
# ostruct.rb - Python Style Object
#  just assign to create field
#
# s = OpenStruct.new
# s.foo = 25
# p s.foo
# s.bar = 2
# p s.bar
# p s

class OpenStruct
  def initialize(hash=nil)
    @table = {}
    if hash
      for k,v in hash
	@table[k] = v.to_sym
      end
    end
  end

  def method_missing(mid, *args)
    mname = mid.id2name
    len = args.length
    if mname =~ /=$/
      if len != 1
	raise ArgumentError, "wrong # of arguments (#{len} for 1)", caller(1)
      end
      mname.chop!
      @table[mname.intern] = args[0]
    elsif args.length == 0
      @table[mid]
    else
      raise NoMethodError, "undefined method `#{mname}' for #{self}", caller(1)
    end
  end

  def delete_field(name)
    @table.delete name.to_sym
  end

  def inspect
    str = "<#{self.class}"
    for k,v in @table
      str += " "
      str += k.to_s
      str += "="
      str += v.inspect
    end
    str += ">"
    str
  end
end