summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
authorVladimir Kochnev <hashtable@yandex.ru>2015-11-15 16:57:51 +0300
committerVladimir Kochnev <hashtable@yandex.ru>2019-01-28 21:14:08 +0300
commit250f174f48a7115c832690fd052dcaf63b6debc9 (patch)
tree84c55562cdd1281072bcd84114d11859a7f5e528 /lib
parentcd30488f9e4f4cbdfe0ccada40297e6f6d7e5610 (diff)
downloadhashie-250f174f48a7115c832690fd052dcaf63b6debc9.tar.gz
Add Hashie::Extensions::Mash::DefineAccessors.
This patch adds an extension for Mash that makes it behave like `OpenStruct`. It reduces overhead of `method_missing?` magic which is a good thing! It's inspired by the recent @sferik's work on `OpenStruct` — https://github.com/ruby/ruby/pull/1033. When using it in `Mash` subclasses it makes them *remember* methods so then it's more like `ActiveModel` than `OpenStruct` in this case. To use it like `OpenStruct` one could use this shortcut: ```ruby { foo: 1, bar: 2 }.to_mash.with_accessors! ``` Implementation details: It injects to class an anonymous module that stores accessor method definitions. This is inspired by `ActiveModel` / `ActiveRecord`. It allows to override accessors in subclass and call them via `super` if this is intended.
Diffstat (limited to 'lib')
-rw-r--r--lib/hashie.rb1
-rw-r--r--lib/hashie/extensions/mash/define_accessors.rb90
-rw-r--r--lib/hashie/mash.rb4
3 files changed, 95 insertions, 0 deletions
diff --git a/lib/hashie.rb b/lib/hashie.rb
index b6e1d13..fd93e05 100644
--- a/lib/hashie.rb
+++ b/lib/hashie.rb
@@ -47,6 +47,7 @@ module Hashie
autoload :KeepOriginalKeys, 'hashie/extensions/mash/keep_original_keys'
autoload :SafeAssignment, 'hashie/extensions/mash/safe_assignment'
autoload :SymbolizeKeys, 'hashie/extensions/mash/symbolize_keys'
+ autoload :DefineAccessors, 'hashie/extensions/mash/define_accessors'
end
module Array
diff --git a/lib/hashie/extensions/mash/define_accessors.rb b/lib/hashie/extensions/mash/define_accessors.rb
new file mode 100644
index 0000000..f13ea2e
--- /dev/null
+++ b/lib/hashie/extensions/mash/define_accessors.rb
@@ -0,0 +1,90 @@
+module Hashie
+ module Extensions
+ module Mash
+ module DefineAccessors
+ def self.included(klass)
+ klass.class_eval do
+ mod = Ext.new
+ include mod
+ end
+ end
+
+ def self.extended(obj)
+ included(obj.singleton_class)
+ end
+
+ class Ext < Module
+ def initialize
+ mod = self
+ define_method(:method_missing) do |method_name, *args, &block|
+ key, suffix = method_name_and_suffix(method_name)
+ case suffix
+ when '='.freeze
+ mod.define_writer(key, method_name)
+ when '?'.freeze
+ mod.define_predicate(key, method_name)
+ when '!'.freeze
+ mod.define_initializing_reader(key, method_name)
+ when '_'.freeze
+ mod.define_underbang_reader(key, method_name)
+ else
+ mod.define_reader(key, method_name)
+ end
+ send(method_name, *args, &block)
+ end
+ end
+
+ def define_reader(key, method_name)
+ define_method(method_name) do |&block|
+ if key? method_name
+ self.[](method_name, &block)
+ else
+ self.[](key, &block)
+ end
+ end
+ end
+
+ def define_writer(key, method_name)
+ define_method(method_name) do |value = nil|
+ if key? method_name
+ self.[](method_name, &proc)
+ else
+ assign_property(key, value)
+ end
+ end
+ end
+
+ def define_predicate(key, method_name)
+ define_method(method_name) do
+ if key? method_name
+ self.[](method_name, &proc)
+ else
+ !!self[key]
+ end
+ end
+ end
+
+ def define_initializing_reader(key, method_name)
+ define_method(method_name) do
+ if key? method_name
+ self.[](method_name, &proc)
+ else
+ initializing_reader(key)
+ end
+ end
+ end
+
+ def define_underbang_reader(key, method_name)
+ define_method(method_name) do
+ if key? method_name
+ self.[](key, &proc)
+ else
+ underbang_reader(key)
+ end
+ end
+ end
+ end
+ end
+ end
+ end
+end
diff --git a/lib/hashie/mash.rb b/lib/hashie/mash.rb
index 6e45c09..f41d745 100644
--- a/lib/hashie/mash.rb
+++ b/lib/hashie/mash.rb
@@ -119,6 +119,10 @@ module Hashie
end
end
+ def with_accessors!
+ extend Hashie::Extensions::Mash::DefineAccessors
+ end
+
alias to_s inspect
# If you pass in an existing hash, it will