about summary refs log tree commit diff
path: root/app/serializers/activitypub
diff options
context:
space:
mode:
authornullkal <nullkal@nil.nu>2017-09-19 23:37:06 +0900
committerEugen Rochko <eugen@zeonfederated.com>2017-09-19 16:37:06 +0200
commit7d16bb379d1463471faf264bb89e24d5b8a505ca (patch)
treeda43f72d3d7772af68f477e8c28767b12beffa68 /app/serializers/activitypub
parent0401a24558294b6941c30c922af3f2063dfd305e (diff)
Use OrderedCollectionPage to return followers/following list (#4949)
Diffstat (limited to 'app/serializers/activitypub')
-rw-r--r--app/serializers/activitypub/collection_serializer.rb25
1 files changed, 20 insertions, 5 deletions
diff --git a/app/serializers/activitypub/collection_serializer.rb b/app/serializers/activitypub/collection_serializer.rb
index d01dead28..9832133fc 100644
--- a/app/serializers/activitypub/collection_serializer.rb
+++ b/app/serializers/activitypub/collection_serializer.rb
@@ -3,23 +3,38 @@
 class ActivityPub::CollectionSerializer < ActiveModel::Serializer
   def self.serializer_for(model, options)
     return ActivityPub::ActivitySerializer if model.class.name == 'Status'
+    return ActivityPub::CollectionSerializer if model.class.name == 'ActivityPub::CollectionPresenter'
     super
   end
 
   attributes :id, :type, :total_items
+  attribute :next, if: -> { object.next.present? }
+  attribute :prev, if: -> { object.prev.present? }
+  attribute :part_of, if: -> { object.part_of.present? }
 
-  has_many :items, key: :ordered_items
+  has_one :first, if: -> { object.first.present? }
+  has_many :items, key: :items, if: -> { (object.items.present? || page?) && !ordered? }
+  has_many :items, key: :ordered_items, if: -> { (object.items.present? || page?) && ordered? }
 
   def type
-    case object.type
-    when :ordered
-      'OrderedCollection'
+    if page?
+      ordered? ? 'OrderedCollectionPage' : 'CollectionPage'
     else
-      'Collection'
+      ordered? ? 'OrderedCollection' : 'Collection'
     end
   end
 
   def total_items
     object.size
   end
+
+  private
+
+  def ordered?
+    object.type == :ordered
+  end
+
+  def page?
+    object.part_of.present?
+  end
 end