mastodon/app/models/stream_entry.rb

63 lines
1.5 KiB
Ruby
Raw Normal View History

2016-08-17 16:56:23 +01:00
class StreamEntry < ApplicationRecord
include Paginable
2016-02-22 15:00:20 +00:00
belongs_to :account, inverse_of: :stream_entries
belongs_to :activity, polymorphic: true
belongs_to :status, foreign_type: 'Status', foreign_key: 'activity_id'
belongs_to :follow, foreign_type: 'Follow', foreign_key: 'activity_id'
belongs_to :favourite, foreign_type: 'Favourite', foreign_key: 'activity_id'
2016-02-22 17:10:30 +00:00
validates :account, :activity, presence: true
2016-09-29 20:28:21 +01:00
STATUS_INCLUDES = [:account, :stream_entry, :media_attachments, mentions: :account, reblog: [:stream_entry, :account, mentions: :account], thread: [:stream_entry, :account]].freeze
scope :with_includes, -> { includes(:account, status: STATUS_INCLUDES, favourite: [:account, :stream_entry, status: STATUS_INCLUDES], follow: [:target_account, :stream_entry]) }
2016-03-21 09:31:20 +00:00
2016-02-22 15:00:20 +00:00
def object_type
2016-09-29 20:28:21 +01:00
orphaned? ? :activity : (targeted? ? :activity : activity.object_type)
2016-02-22 15:00:20 +00:00
end
def verb
2016-09-29 20:28:21 +01:00
orphaned? ? :delete : activity.verb
2016-02-22 17:10:30 +00:00
end
def targeted?
[:follow, :share, :favorite].include? verb
2016-02-22 15:00:20 +00:00
end
def target
2016-09-29 20:28:21 +01:00
orphaned? ? nil : activity.target
2016-02-22 17:10:30 +00:00
end
def title
2016-09-29 20:28:21 +01:00
orphaned? ? nil : activity.title
2016-02-22 15:00:20 +00:00
end
def content
2016-09-29 20:28:21 +01:00
orphaned? ? nil : activity.content
2016-02-22 15:00:20 +00:00
end
def threaded?
(verb == :favorite || object_type == :comment) && !thread.nil?
end
def thread
2016-09-29 20:28:21 +01:00
orphaned? ? nil : activity.thread
end
def mentions
activity.respond_to?(:mentions) ? activity.mentions.map(&:account) : []
end
def activity
2016-09-29 20:28:21 +01:00
send(activity_type.downcase.to_sym)
end
private
def orphaned?
2016-09-29 20:28:21 +01:00
activity.nil?
end
2016-02-22 15:00:20 +00:00
end