you can now follow / unfollow a post from the stream; fixed cukes.

1 files changed, 17 insertions, 0 deletions

diff --git a/lib/stream/base.rb b/lib/stream/base.rb
index b3af708ca..27281581a 100644
--- a/lib/stream/base.rb
+++ b/lib/stream/base.rb
@@ -40,6 +40,7 @@ class Stream::Base
   def stream_posts
     self.posts.for_a_stream(max_time, order, self.user).tap do |posts|
       like_posts_for_stream!(posts) #some sql person could probably do this with joins.
+      participation_posts_for_stream!(posts)
     end
   end
 
@@ -112,6 +113,22 @@ class Stream::Base
     end
   end
 
+  # @return [void]
+  def participation_posts_for_stream!(posts)
+    return posts unless @user
+
+    participations = Participation.where(:author_id => @user.person.id, :target_id => posts.map(&:id), :target_type => "Post")
+
+    participation_hash = participations.inject({}) do |hash, participation|
+      hash[participation.target_id] = participation
+      hash
+    end
+
+    posts.each do |post|
+      post.user_participation = participation_hash[post.id]
+    end
+  end
+
   # @return [Hash]
   def publisher_opts
     {}