lib/looksist/hashed.rb in looksist-0.1.3 vs lib/looksist/hashed.rb in looksist-0.1.4

@@ -2,10 +2,11 @@
 require 'json'
 
 module Looksist
   module Hashed
     extend ActiveSupport::Concern
+    include Looksist::Common
 
     module ClassMethods
       def inject(opts)
         raise 'Incorrect usage' unless [:after, :using, :populate].all? { |e| opts.keys.include? e }
 
@@ -36,12 +37,9 @@
     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)
       return nil unless hash_offset
       keys = hash_offset[opts[:using]]
       entity_name = entity(opts[:using])