Merges annotations

lib/merged_comment_tree.rb

@@ -0,0 +1,28 @@
+class MergedCommentTree < CommentTree
+  attr_accessor :commentables, :array_order
+
+  def initialize(commentables, page, order = 'confidence_score')
+    @commentables = commentables
+    @commentable = commentables.first
+    @page = page
+    @order = order
+    @array_order = set_array_order(order)
+
+    @comments = root_comments + root_descendants
+  end
+
+  def root_comments
+    Kaminari.paginate_array(commentables.flatten.map(&:comments).map(&:roots).flatten.sort_by {|a| a[array_order]}.reverse).
+    page(page).per(ROOT_COMMENTS_PER_PAGE)
+  end
+
+  def set_array_order(order)
+    case order
+    when "newest"
+      "created_at"
+    else
+      "confidence_score"
+    end
+  end
+
+end