status.rb 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480
  1. # frozen_string_literal: true
  2. # == Schema Information
  3. #
  4. # Table name: statuses
  5. #
  6. # id :bigint(8) not null, primary key
  7. # uri :string
  8. # text :text default(""), not null
  9. # created_at :datetime not null
  10. # updated_at :datetime not null
  11. # in_reply_to_id :bigint(8)
  12. # reblog_of_id :bigint(8)
  13. # url :string
  14. # sensitive :boolean default(FALSE), not null
  15. # visibility :integer default("public"), not null
  16. # spoiler_text :text default(""), not null
  17. # reply :boolean default(FALSE), not null
  18. # language :string
  19. # conversation_id :bigint(8)
  20. # local :boolean
  21. # account_id :bigint(8) not null
  22. # application_id :bigint(8)
  23. # in_reply_to_account_id :bigint(8)
  24. # poll_id :bigint(8)
  25. #
  26. class Status < ApplicationRecord
  27. before_destroy :unlink_from_conversations
  28. include Paginable
  29. include Cacheable
  30. include StatusThreadingConcern
  31. # If `override_timestamps` is set at creation time, Snowflake ID creation
  32. # will be based on current time instead of `created_at`
  33. attr_accessor :override_timestamps
  34. update_index('statuses#status', :proper) if Chewy.enabled?
  35. enum visibility: [:public, :unlisted, :private, :direct, :limited], _suffix: :visibility
  36. belongs_to :application, class_name: 'Doorkeeper::Application', optional: true
  37. belongs_to :account, inverse_of: :statuses
  38. belongs_to :in_reply_to_account, foreign_key: 'in_reply_to_account_id', class_name: 'Account', optional: true
  39. belongs_to :conversation, optional: true
  40. belongs_to :preloadable_poll, class_name: 'Poll', foreign_key: 'poll_id', optional: true
  41. belongs_to :thread, foreign_key: 'in_reply_to_id', class_name: 'Status', inverse_of: :replies, optional: true
  42. belongs_to :reblog, foreign_key: 'reblog_of_id', class_name: 'Status', inverse_of: :reblogs, optional: true
  43. has_many :favourites, inverse_of: :status, dependent: :destroy
  44. has_many :reblogs, foreign_key: 'reblog_of_id', class_name: 'Status', inverse_of: :reblog, dependent: :destroy
  45. has_many :replies, foreign_key: 'in_reply_to_id', class_name: 'Status', inverse_of: :thread
  46. has_many :mentions, dependent: :destroy, inverse_of: :status
  47. has_many :active_mentions, -> { active }, class_name: 'Mention', inverse_of: :status
  48. has_many :media_attachments, dependent: :nullify
  49. has_and_belongs_to_many :tags
  50. has_and_belongs_to_many :preview_cards
  51. has_one :notification, as: :activity, dependent: :destroy
  52. has_one :status_stat, inverse_of: :status
  53. has_one :poll, inverse_of: :status, dependent: :destroy
  54. validates :uri, uniqueness: true, presence: true, unless: :local?
  55. validates :text, presence: true, unless: -> { with_media? || reblog? }
  56. validates_with StatusLengthValidator
  57. validates_with DisallowedHashtagsValidator
  58. validates :reblog, uniqueness: { scope: :account }, if: :reblog?
  59. validates :visibility, exclusion: { in: %w(direct limited) }, if: :reblog?
  60. accepts_nested_attributes_for :poll
  61. default_scope { recent }
  62. scope :recent, -> { reorder(id: :desc) }
  63. scope :remote, -> { where(local: false).where.not(uri: nil) }
  64. scope :local, -> { where(local: true).or(where(uri: nil)) }
  65. scope :without_replies, -> { where('statuses.reply = FALSE OR statuses.in_reply_to_account_id = statuses.account_id') }
  66. scope :without_reblogs, -> { where('statuses.reblog_of_id IS NULL') }
  67. scope :with_public_visibility, -> { where(visibility: :public) }
  68. scope :tagged_with, ->(tag) { joins(:statuses_tags).where(statuses_tags: { tag_id: tag }) }
  69. scope :excluding_silenced_accounts, -> { left_outer_joins(:account).where(accounts: { silenced_at: nil }) }
  70. scope :including_silenced_accounts, -> { left_outer_joins(:account).where.not(accounts: { silenced_at: nil }) }
  71. scope :not_excluded_by_account, ->(account) { where.not(account_id: account.excluded_from_timeline_account_ids) }
  72. scope :not_domain_blocked_by_account, ->(account) { account.excluded_from_timeline_domains.blank? ? left_outer_joins(:account) : left_outer_joins(:account).where('accounts.domain IS NULL OR accounts.domain NOT IN (?)', account.excluded_from_timeline_domains) }
  73. scope :tagged_with_all, ->(tags) {
  74. Array(tags).map(&:id).map(&:to_i).reduce(self) do |result, id|
  75. result.joins("INNER JOIN statuses_tags t#{id} ON t#{id}.status_id = statuses.id AND t#{id}.tag_id = #{id}")
  76. end
  77. }
  78. scope :tagged_with_none, ->(tags) {
  79. Array(tags).map(&:id).map(&:to_i).reduce(self) do |result, id|
  80. result.joins("LEFT OUTER JOIN statuses_tags t#{id} ON t#{id}.status_id = statuses.id AND t#{id}.tag_id = #{id}")
  81. .where("t#{id}.tag_id IS NULL")
  82. end
  83. }
  84. cache_associated :application,
  85. :media_attachments,
  86. :conversation,
  87. :status_stat,
  88. :tags,
  89. :preview_cards,
  90. :preloadable_poll,
  91. account: :account_stat,
  92. active_mentions: { account: :account_stat },
  93. reblog: [
  94. :application,
  95. :tags,
  96. :preview_cards,
  97. :media_attachments,
  98. :conversation,
  99. :status_stat,
  100. :preloadable_poll,
  101. account: :account_stat,
  102. active_mentions: { account: :account_stat },
  103. ],
  104. thread: { account: :account_stat }
  105. delegate :domain, to: :account, prefix: true
  106. REAL_TIME_WINDOW = 6.hours
  107. def searchable_by(preloaded = nil)
  108. ids = [account_id]
  109. if preloaded.nil?
  110. ids += mentions.pluck(:account_id)
  111. ids += favourites.pluck(:account_id)
  112. ids += reblogs.pluck(:account_id)
  113. else
  114. ids += preloaded.mentions[id] || []
  115. ids += preloaded.favourites[id] || []
  116. ids += preloaded.reblogs[id] || []
  117. end
  118. ids.uniq
  119. end
  120. def reply?
  121. !in_reply_to_id.nil? || attributes['reply']
  122. end
  123. def local?
  124. attributes['local'] || uri.nil?
  125. end
  126. def reblog?
  127. !reblog_of_id.nil?
  128. end
  129. def within_realtime_window?
  130. created_at >= REAL_TIME_WINDOW.ago
  131. end
  132. def verb
  133. if destroyed?
  134. :delete
  135. else
  136. reblog? ? :share : :post
  137. end
  138. end
  139. def object_type
  140. reply? ? :comment : :note
  141. end
  142. def proper
  143. reblog? ? reblog : self
  144. end
  145. def content
  146. proper.text
  147. end
  148. def target
  149. reblog
  150. end
  151. def preview_card
  152. preview_cards.first
  153. end
  154. def title
  155. if destroyed?
  156. "#{account.acct} deleted status"
  157. else
  158. reblog? ? "#{account.acct} shared a status by #{reblog.account.acct}" : "New status by #{account.acct}"
  159. end
  160. end
  161. def hidden?
  162. !distributable?
  163. end
  164. def distributable?
  165. public_visibility? || unlisted_visibility?
  166. end
  167. alias sign? distributable?
  168. def with_media?
  169. media_attachments.any?
  170. end
  171. def non_sensitive_with_media?
  172. !sensitive? && with_media?
  173. end
  174. def emojis
  175. return @emojis if defined?(@emojis)
  176. fields = [spoiler_text, text]
  177. fields += preloadable_poll.options unless preloadable_poll.nil?
  178. @emojis = CustomEmoji.from_text(fields.join(' '), account.domain)
  179. end
  180. def mark_for_mass_destruction!
  181. @marked_for_mass_destruction = true
  182. end
  183. def marked_for_mass_destruction?
  184. @marked_for_mass_destruction
  185. end
  186. def replies_count
  187. status_stat&.replies_count || 0
  188. end
  189. def reblogs_count
  190. status_stat&.reblogs_count || 0
  191. end
  192. def favourites_count
  193. status_stat&.favourites_count || 0
  194. end
  195. def increment_count!(key)
  196. update_status_stat!(key => public_send(key) + 1)
  197. end
  198. def decrement_count!(key)
  199. update_status_stat!(key => [public_send(key) - 1, 0].max)
  200. end
  201. after_create_commit :increment_counter_caches
  202. after_destroy_commit :decrement_counter_caches
  203. after_create_commit :store_uri, if: :local?
  204. after_create_commit :update_statistics, if: :local?
  205. around_create Mastodon::Snowflake::Callbacks
  206. before_validation :prepare_contents, if: :local?
  207. before_validation :set_reblog
  208. before_validation :set_visibility
  209. before_validation :set_conversation
  210. before_validation :set_local
  211. after_create :set_poll_id
  212. class << self
  213. def selectable_visibilities
  214. visibilities.keys - %w(direct limited)
  215. end
  216. def in_chosen_languages(account)
  217. where(language: nil).or where(language: account.chosen_languages)
  218. end
  219. def as_home_timeline(account)
  220. where(account: [account] + account.following).where(visibility: [:public, :unlisted, :private])
  221. end
  222. def as_public_timeline(account = nil, local_only = false)
  223. query = timeline_scope(local_only).without_replies
  224. apply_timeline_filters(query, account, local_only)
  225. end
  226. def as_tag_timeline(tag, account = nil, local_only = false)
  227. query = timeline_scope(local_only).tagged_with(tag)
  228. apply_timeline_filters(query, account, local_only)
  229. end
  230. def as_outbox_timeline(account)
  231. where(account: account, visibility: :public)
  232. end
  233. def favourites_map(status_ids, account_id)
  234. Favourite.select('status_id').where(status_id: status_ids).where(account_id: account_id).each_with_object({}) { |f, h| h[f.status_id] = true }
  235. end
  236. def reblogs_map(status_ids, account_id)
  237. select('reblog_of_id').where(reblog_of_id: status_ids).where(account_id: account_id).reorder(nil).each_with_object({}) { |s, h| h[s.reblog_of_id] = true }
  238. end
  239. def mutes_map(conversation_ids, account_id)
  240. ConversationMute.select('conversation_id').where(conversation_id: conversation_ids).where(account_id: account_id).each_with_object({}) { |m, h| h[m.conversation_id] = true }
  241. end
  242. def pins_map(status_ids, account_id)
  243. StatusPin.select('status_id').where(status_id: status_ids).where(account_id: account_id).each_with_object({}) { |p, h| h[p.status_id] = true }
  244. end
  245. def reload_stale_associations!(cached_items)
  246. account_ids = []
  247. cached_items.each do |item|
  248. account_ids << item.account_id
  249. account_ids << item.reblog.account_id if item.reblog?
  250. end
  251. account_ids.uniq!
  252. return if account_ids.empty?
  253. accounts = Account.where(id: account_ids).includes(:account_stat).each_with_object({}) { |a, h| h[a.id] = a }
  254. cached_items.each do |item|
  255. item.account = accounts[item.account_id]
  256. item.reblog.account = accounts[item.reblog.account_id] if item.reblog?
  257. end
  258. end
  259. def permitted_for(target_account, account)
  260. visibility = [:public, :unlisted]
  261. if account.nil?
  262. where(visibility: visibility)
  263. elsif target_account.blocking?(account) # get rid of blocked peeps
  264. none
  265. elsif account.id == target_account.id # author can see own stuff
  266. all
  267. else
  268. # followers can see followers-only stuff, but also things they are mentioned in.
  269. # non-followers can see everything that isn't private/direct, but can see stuff they are mentioned in.
  270. visibility.push(:private) if account.following?(target_account)
  271. scope = left_outer_joins(:reblog)
  272. scope.where(visibility: visibility)
  273. .or(scope.where(id: account.mentions.select(:status_id)))
  274. .merge(scope.where(reblog_of_id: nil).or(scope.where.not(reblogs_statuses: { account_id: account.excluded_from_timeline_account_ids })))
  275. end
  276. end
  277. private
  278. def timeline_scope(local_only = false)
  279. starting_scope = local_only ? Status.local : Status
  280. starting_scope
  281. .with_public_visibility
  282. .without_reblogs
  283. end
  284. def apply_timeline_filters(query, account, local_only)
  285. if account.nil?
  286. filter_timeline_default(query)
  287. else
  288. filter_timeline_for_account(query, account, local_only)
  289. end
  290. end
  291. def filter_timeline_for_account(query, account, local_only)
  292. query = query.not_excluded_by_account(account)
  293. query = query.not_domain_blocked_by_account(account) unless local_only
  294. query = query.in_chosen_languages(account) if account.chosen_languages.present?
  295. query.merge(account_silencing_filter(account))
  296. end
  297. def filter_timeline_default(query)
  298. query.excluding_silenced_accounts
  299. end
  300. def account_silencing_filter(account)
  301. if account.silenced?
  302. including_myself = left_outer_joins(:account).where(account_id: account.id).references(:accounts)
  303. excluding_silenced_accounts.or(including_myself)
  304. else
  305. excluding_silenced_accounts
  306. end
  307. end
  308. end
  309. private
  310. def update_status_stat!(attrs)
  311. return if marked_for_destruction? || destroyed?
  312. record = status_stat || build_status_stat
  313. record.update(attrs)
  314. end
  315. def store_uri
  316. update_column(:uri, ActivityPub::TagManager.instance.uri_for(self)) if uri.nil?
  317. end
  318. def prepare_contents
  319. text&.strip!
  320. spoiler_text&.strip!
  321. end
  322. def set_reblog
  323. self.reblog = reblog.reblog if reblog? && reblog.reblog?
  324. end
  325. def set_poll_id
  326. update_column(:poll_id, poll.id) unless poll.nil?
  327. end
  328. def set_visibility
  329. self.visibility = reblog.visibility if reblog? && visibility.nil?
  330. self.visibility = (account.locked? ? :private : :public) if visibility.nil?
  331. self.sensitive = false if sensitive.nil?
  332. end
  333. def set_conversation
  334. self.thread = thread.reblog if thread&.reblog?
  335. self.reply = !(in_reply_to_id.nil? && thread.nil?) unless reply
  336. if reply? && !thread.nil?
  337. self.in_reply_to_account_id = carried_over_reply_to_account_id
  338. self.conversation_id = thread.conversation_id if conversation_id.nil?
  339. elsif conversation_id.nil?
  340. self.conversation = Conversation.new
  341. end
  342. end
  343. def carried_over_reply_to_account_id
  344. if thread.account_id == account_id && thread.reply?
  345. thread.in_reply_to_account_id
  346. else
  347. thread.account_id
  348. end
  349. end
  350. def set_local
  351. self.local = account.local?
  352. end
  353. def update_statistics
  354. return unless distributable?
  355. ActivityTracker.increment('activity:statuses:local')
  356. end
  357. def increment_counter_caches
  358. return if direct_visibility?
  359. account&.increment_count!(:statuses_count)
  360. reblog&.increment_count!(:reblogs_count) if reblog?
  361. thread&.increment_count!(:replies_count) if in_reply_to_id.present? && distributable?
  362. end
  363. def decrement_counter_caches
  364. return if direct_visibility? || marked_for_mass_destruction?
  365. account&.decrement_count!(:statuses_count)
  366. reblog&.decrement_count!(:reblogs_count) if reblog?
  367. thread&.decrement_count!(:replies_count) if in_reply_to_id.present? && distributable?
  368. end
  369. def unlink_from_conversations
  370. return unless direct_visibility?
  371. mentioned_accounts = mentions.includes(:account).map(&:account)
  372. inbox_owners = mentioned_accounts.select(&:local?) + (account.local? ? [account] : [])
  373. inbox_owners.each do |inbox_owner|
  374. AccountConversation.remove_status(inbox_owner, self)
  375. end
  376. end
  377. end