@@ -10,8 +10,8 @@ def initialize
1010 super
1111 end
1212
13- def from_activity ( h )
14- merge ( h )
13+ def from_activity ( from )
14+ merge ( from )
1515 end
1616
1717 def enriched?
@@ -31,7 +31,7 @@ class Enrich
3131 attr_reader :fields
3232
3333 def initialize ( fields = nil )
34- @fields = fields || [ : actor, : object, : target]
34+ @fields = fields || %i[ actor object target ]
3535 end
3636
3737 def add_fields ( new_fields )
@@ -47,9 +47,9 @@ def model_field?(field_value)
4747 begin
4848 model . classify . constantize
4949 rescue NameError
50- return false
50+ false
5151 else
52- return true
52+ true
5353 end
5454 end
5555
@@ -77,6 +77,7 @@ def collect_references(activities)
7777 activities . each do |activity |
7878 activity . select { |k , _v | @fields . include? k . to_sym } . each do |_field , value |
7979 next unless model_field? ( value )
80+
8081 model , _ , id = value . rpartition ( ':' )
8182 model_refs [ model ] [ id ] = 0
8283 end
@@ -85,13 +86,18 @@ def collect_references(activities)
8586 end
8687
8788 def retrieve_objects ( references )
88- Hash [ references . map { |model , ids | [ model , Hash [ model . classify . constantize . where ( model . classify . constantize . primary_key => ids . keys ) . map { |i | [ i . id . to_s , i ] } ] ] } ]
89+ references . map do |model , ids |
90+ [ model , model . classify . constantize . where ( model . classify . constantize . primary_key => ids . keys ) . map do |i |
91+ [ i . id . to_s , i ]
92+ end . to_h ]
93+ end . to_h
8994 end
9095
9196 def inject_objects ( activities , objects )
9297 create_activity_results ( activities ) . each do |activity |
9398 activity . select { |k , _v | @fields . include? k . to_sym } . each do |field , value |
9499 next unless model_field? ( value )
100+
95101 model , _ , id = value . rpartition ( ':' )
96102 activity [ field ] = objects [ model ] [ id ] || value
97103 activity . track_not_enriched_field ( field , value ) if objects [ model ] [ id ] . nil?
@@ -102,7 +108,7 @@ def inject_objects(activities, objects)
102108 private
103109
104110 def create_activity_results ( activities )
105- return activities . map { |a | ActivityResult . new . from_activity ( a ) }
111+ activities . map { |a | ActivityResult . new . from_activity ( a ) }
106112 end
107113 end
108114end
0 commit comments