From 7b485780a5dda526edd91ac43a89f6b01c51ffbb Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sun, 15 Feb 2009 12:43:46 +0000
Subject: * lib/ostruct.rb (OpenStruct#new_ostruct_member): checks if frozen.  
 [ruby-talk:328195], [ruby-core:22142]

git-svn-id: http://svn.ruby-lang.org/repos/ruby/trunk@22332 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 lib/ostruct.rb | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

(limited to 'lib')

diff --git a/lib/ostruct.rb b/lib/ostruct.rb
index 35a14b492..45ebb8083 100644
--- a/lib/ostruct.rb
+++ b/lib/ostruct.rb
@@ -67,29 +67,33 @@ class OpenStruct
     @table.each_key{|key| new_ostruct_member(key)}
   end
 
+  def modifiable
+    if self.frozen?
+      raise TypeError, "can't modify frozen #{self.class}", caller(2)
+    end
+    @table
+  end
+  protected :modifiable
+
   def new_ostruct_member(name)
     name = name.to_sym
     unless self.respond_to?(name)
       class << self; self; end.class_eval do
         define_method(name) { @table[name] }
-        define_method(:"#{name}=") { |x| @table[name] = x }
+        define_method("#{name}=") { |x| modifiable[name] = x }
       end
     end
+    name
   end
 
   def method_missing(mid, *args) # :nodoc:
     mname = mid.id2name
     len = args.length
-    if mname =~ /=$/
+    if mname.chomp!('=')
       if len != 1
         raise ArgumentError, "wrong number of arguments (#{len} for 1)", caller(1)
       end
-      if self.frozen?
-        raise TypeError, "can't modify frozen #{self.class}", caller(1)
-      end
-      mname.chop!
-      self.new_ostruct_member(mname)
-      @table[mname.intern] = args[0]
+      modifiable[new_ostruct_member(mname)] = args[0]
     elsif len == 0
       @table[mid]
     else
-- 
cgit