recursive_struct 0.0.5 → 0.0.6

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    N2UxYTExZjRiNjU2YjY2ZmE4ZTA0OTFhMDcyYjIzMGRiMjFhZTRhNQ==
+    ZDZjODIwNzIwODg3YjQ1ZmM2OGM4Zjg4M2E5ZDZmZGVjZjdhMjFkMg==
   data.tar.gz: !binary |-
-    NjA2M2I0ZDljNjAwM2UwM2MwNWE1NjY5MDIyZmRkZjQ4NDM4OWZjMg==
+    OWRmNWUwMWZlMTkyNWZiM2ZiMGI3ODZiMjFhNjQzNjkxNzJmNjMyOQ==
 SHA512:
   metadata.gz: !binary |-
-    MTBlNzY4MmVhNDJjMWJiOTQyYjI3ZThjYmUzNjI2ZTBhOTczZTk3M2JlNjkz
-    MzYzYmUyMjk0ZjhhMmJjY2ZjZjA1MzMyYjczNWY4NmZmYzM1NjM3MjIxYTk0
-    MDU3NmM3ZDBmMmJiN2E0NWQ0OTI4YjFkZDNhNWNmMTRkODY1ZTc=
+    ZGI0NWFkMzc2NWM4NDRmMTkzYjcyZmYzMzRjODZhMjA4MDY1YzEzNzYyYWVk
+    YWJmY2IwMmY0MjBhZmI0ODY1N2QyMDFjNDVjMzJiZWZkMTI2MDIxMTI0YmNm
+    ZmM1OWQzNTgxNDM1NmExYzFjNzBjODIzM2UwMTc2ZWU2YmExZDY=
   data.tar.gz: !binary |-
-    YjY5ZjczNGRkZmQwYjg1ZTlhMDQwYzMxNzI5MzFjYTNmMWY4NDQ1ODQ4MTY2
-    YTYyNmFjNjAxYTUwMDA1NzcxNzc5ZTNkZTFmYTg2ZmRjMTYzMTMyZDhmNDJj
-    MTBiMzhhZTRjMTFkYWVhMDlkZDM0NTU0OWZiNTI4Njk2NzQ0NzM=
+    MzY5NmE3ZWJhZTFjZDljNDM1MTIwMTQyN2U5NDdiYWIwODJmNmQ2NmQ5ZjVi
+    MzU4OWM1MjFiNzQ4MGJlODk5YzRkOTgyYjJmYWEzY2EyYzhlZWVkMDRjMDI0
+    NWVjMzc2MWVhOWVjYTA2NTFmNzU5ZGFhYTE5ZDU5N2ZiMjM0MDU=

data/lib/recursive_struct.rb

@@ -1,8 +1,12 @@
+require 'recursive_struct/data'
+
 class RecursiveStruct
+  include RecursiveStruct::Data
+
   def initialize(hash = nil)
     if hash
       hash.each do |key, value|
-        add_data(key, process(value))
+        set_data(key, value)
       end
     end
   end
@@ -10,8 +14,10 @@ class RecursiveStruct
   def method_missing(name, *args)
     key = name.to_s
 
-    if key.chomp!('=') && args.length == 1
-      add_data(key, process(args.first))
+    if key.include?('=')
+      add_setter(key.chomp('='), *args)
+    elsif data.respond_to?(key)
+      send_data(key, *args)
     elsif args.length == 0
       get_data(key)
     else
@@ -25,25 +31,11 @@ class RecursiveStruct
     value.is_a?(Hash) ? self.class.new(value) : value
   end
 
-  def data
-    @data ||= {}
-  end
-
-  def add_data(key, value)
-    key = key.to_sym
-
-    data[key] = value
-    define_methods(key)
-  end
-
-  def get_data(key)
-    data[key.to_sym]
-  end
-
-  def define_methods(key)
-    unless respond_to?(key)
-      define_singleton_method(key) { data[key] }
-      define_singleton_method("#{key}=") { |value| data[key] = value }
+  def add_setter(key, *args)
+    if key == '[]' && args.length == 2
+      set_data(args.first, args.last)
+    elsif args.length == 1
+      set_data(key, args.first)
     end
   end
 end

data/lib/recursive_struct/data.rb

@@ -0,0 +1,35 @@
+class RecursiveStruct
+  module Data
+    def data
+      @data ||= {}
+    end
+
+    def set_data(key, value)
+      key = key.to_sym
+
+      data[key] = process(value)
+      define_methods(key)
+    end
+
+    def get_data(key)
+      data[key.to_sym]
+    end
+
+    def send_data(key, *args)
+      data.send(key, *args)
+    end
+
+    private
+
+    def process(value)
+      value.is_a?(Hash) ? RecursiveStruct.new(value) : value
+    end
+
+    def define_methods(key)
+      unless respond_to?(key)
+        define_singleton_method(key) { data[key] }
+        define_singleton_method("#{key}=") { |value| data[key] = value }
+      end
+    end
+  end
+end

data/lib/recursive_struct/version.rb

@@ -1,3 +1,3 @@
 class RecursiveStruct
-  VERSION = '0.0.5'
+  VERSION = '0.0.6'
 end

data/spec/recursive_struct_spec.rb

@@ -54,6 +54,12 @@ describe RecursiveStruct do
         it { expect(subject.one).to be true }
       end
 
+      describe 'hash setter' do
+        before { subject[:one] = true }
+        it { expect(subject[:one]).to be true }
+        it { expect(subject.one).to be true }
+      end
+
       describe 'array value' do
         before { subject.one = [1, 2] }
         it { expect(subject.one).not_to be_empty }
@@ -72,5 +78,9 @@ describe RecursiveStruct do
     describe 'invalid method' do
       it { expect { subject.one(1) }.to raise_error NoMethodError }
     end
+
+    describe 'responds to hash methods' do
+      it { expect { subject.keys }.not_to raise_error }
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: recursive_struct
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Phillip Boksz
@@ -81,6 +81,7 @@ files:
 - README.md
 - Rakefile
 - lib/recursive_struct.rb
+- lib/recursive_struct/data.rb
 - lib/recursive_struct/version.rb
 - recursive_struct.gemspec
 - spec/recursive_struct_spec.rb