multihash 0.1.0 → 0.2.0

data/lib/multihash/multihash.rb

@@ -3,6 +3,30 @@
 # several mappings to a single value, without doing anything particularly
 # gross with Array#include? lookups.
 class MultiHash
+  attr_accessor :immutable
+  alias_method :immutable?, :immutable
+
+  HASH_METHOD_WHITELIST = [
+    :[],
+    :assoc,
+    :default,
+    :empty?,
+    :each,
+    :each_key,
+    :each_pair,
+    :each_value,
+    :fetch,
+    :keys,
+    :select,
+    :values_at
+  ]
+
+  HASH_METHOD_WHITELIST.each do |mth|
+    define_method mth do |*args, &blk|
+      @_hash.send(mth, *args, &blk)
+    end
+  end
+
   # Initialize an immutable multihash with the provided hash as its source.
   # For example, we might want to look up the nation which produces different
   # types of cars:
@@ -15,16 +39,70 @@ class MultiHash
   #
   # @param original_hash [Hash<Array|Object, Object>] the hash to convert into
   # a multihash, with arrays or single objects as keys.
-  def initialize(original_hash)
-    @_hash = Hash.new
+  def initialize(original_hash={}, immutable=true)
+    @_hash = Hash.new(original_hash.default)
+    @immutable = immutable
+
+    groups = []
 
     original_hash.each do |orig_key, value|
       key_array = orig_key.is_a?(Array) && orig_key || [orig_key]
-      key_array.each { |k| @_hash[k] = value }
+      groups << key_array
+      key_array.each { |k| map_value(k, value) }
     end
+
+    store_groups(groups) unless immutable?
+  end
+
+  def grouped
+    Hash[@_groups.values.map { |g| [g, self[g.first]] }]
+  end
+
+  def set(key, value)
+    fail 'Cannot modify an immutable MultiHash!' if immutable?
+    split_key_from_group(key)
+    map_value(key, value)
+  end
+  alias_method :[]=, :set
+
+  def set_group(key, value)
+    fail 'Cannot modify an immutable MultiHash!' if immutable?
+    @_groups[key].each { |k| map_value(k, value) }
+  end
+
+  def to_h
+    @_hash
+  end
+
+  def values
+    @_values ||= @_hash.values.uniq
+  end
+
+  def self.[](args)
+    MultiHash.new(Hash[args])
+  end
+
+  private
+
+  def dirty
+    @_grouped = nil
+    @_values = nil
+  end
+
+  def map_value(key, value)
+    dirty
+    @_hash[key] = value
+  end
+
+  def split_key_from_group(key)
+    former_group = @_groups[key]
+    other_groups = @_groups.values - former_group
+    new_group = key.is_a?(Array) ? key : [key]
+    modified_group = former_group - new_group
+    store_groups(other_groups + modified_group + new_group)
   end
 
-  def [](key)
-    @_hash[key]
+  def store_groups(groups)
+    @_groups = MultiHash.new(Hash[groups.map { |g| [g, g] }])
   end
 end

metadata

@@ -1,14 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: multihash
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
+  prerelease: 
 platform: ruby
 authors:
 - Blake Hyde
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-12-17 00:00:00.000000000 Z
+date: 2014-12-18 00:00:00.000000000 Z
 dependencies: []
 description: MultiHash allows multiple keys to the same value, concisely.
 email: syrion@gmail.com
@@ -20,25 +21,26 @@ files:
 homepage: https://github.com/asthasr/multihash
 licenses:
 - MIT
-metadata: {}
 post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
   requirements:
-  - - ">="
+  - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
   requirements:
-  - - ">="
+  - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 1.8.23
 signing_key: 
-specification_version: 4
+specification_version: 3
 summary: A many-to-one, immutable hash.
 test_files: []

checksums.yaml

@@ -1,7 +0,0 @@
----
-SHA1:
-  metadata.gz: dbead59572ba7080786ff23f566971f2c5323d0d
-  data.tar.gz: 21d075eb849f528c9849ec17ecba9d87ca51a5e3
-SHA512:
-  metadata.gz: 8d2aa119dabc2d5980152b50a1a7b53bfb205b51c98db9c9e67b4ebbb035969bf327b5983d053b6b5257b00f840818e0a22d301175590aeeb5d7535d75db2a92
-  data.tar.gz: abb7eff1d113406685f45e47fade2b826a6a2118cd244e94796c04b4479d59990ffcee0ecca4e44742f61715037dc1d617aa56908f0e86ad0e0763a1c98ad99e