lib/looksist/hashed.rb in looksist-0.0.4 vs lib/looksist/hashed.rb in looksist-0.0.5

@@ -1,5 +1,8 @@
+require 'jsonpath'
+require 'json'
+
 module Looksist
   module Hashed
     extend ActiveSupport::Concern
 
     class << self;
@@ -16,26 +19,37 @@
         return if @rules[opts[:after]].length > 1
 
         define_method("#{opts[:after]}_with_inject") do |*args|
           hash = send("#{opts[:after]}_without_inject".to_sym, *args)
           self.class.instance_variable_get(:@rules)[opts[:after]].each do |opts|
-            keys = hash[opts[:at]][opts[:using]]
-            entity_name = entity(opts[:using])
-            values = Hashed.redis_service.send("#{entity_name}_for", keys)
-            hash[opts[:at]][opts[:populate]] = values
+            if opts[:at].is_a? String
+              hash = JsonPath.for(hash.with_indifferent_access).gsub(opts[:at]) do |i|
+                inject_attributes_at(i, opts)
+              end.to_hash.deep_symbolize_keys
+            else
+              inject_attributes_at(hash[opts[:at]], opts)
+            end
           end
           hash
         end
-
         alias_method_chain opts[:after], :inject
       end
     end
 
     included do |base|
       base.class_attribute :rules
     end
 
+    private
     def entity(entity_id)
       entity_id.to_s.gsub('_id', '')
+    end
+
+    def inject_attributes_at(hash_offset, opts)
+      keys = hash_offset[opts[:using]]
+      entity_name = entity(opts[:using])
+      values = Hashed.redis_service.send("#{entity_name}_for", keys)
+      hash_offset[opts[:populate]] = values
+      hash_offset
     end
   end
 end