account.rb 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201
  1. # frozen_string_literal: true
  2. class Account < ApplicationRecord
  3. include Targetable
  4. include PgSearch
  5. MENTION_RE = /(?:^|[^\/\w])@([a-z0-9_]+(?:@[a-z0-9\.\-]+)?)/i
  6. IMAGE_MIME_TYPES = ['image/jpeg', 'image/png', 'image/gif'].freeze
  7. # Local users
  8. has_one :user, inverse_of: :account
  9. validates :username, presence: true, format: { with: /\A[a-z0-9_]+\z/i, message: 'only letters, numbers and underscores' }, uniqueness: { scope: :domain, case_sensitive: false }, length: { maximum: 30 }, if: 'local?'
  10. validates :username, presence: true, uniqueness: { scope: :domain, case_sensitive: true }, unless: 'local?'
  11. # Avatar upload
  12. has_attached_file :avatar, styles: { original: '120x120#' }, convert_options: { all: '-quality 80 -strip' }
  13. validates_attachment_content_type :avatar, content_type: IMAGE_MIME_TYPES
  14. validates_attachment_size :avatar, less_than: 2.megabytes
  15. # Header upload
  16. has_attached_file :header, styles: { original: '700x335#' }, convert_options: { all: '-quality 80 -strip' }
  17. validates_attachment_content_type :header, content_type: IMAGE_MIME_TYPES
  18. validates_attachment_size :header, less_than: 2.megabytes
  19. # Local user profile validations
  20. validates :display_name, length: { maximum: 30 }, if: 'local?'
  21. validates :note, length: { maximum: 160 }, if: 'local?'
  22. # Timelines
  23. has_many :stream_entries, inverse_of: :account, dependent: :destroy
  24. has_many :statuses, inverse_of: :account, dependent: :destroy
  25. has_many :favourites, inverse_of: :account, dependent: :destroy
  26. has_many :mentions, inverse_of: :account, dependent: :destroy
  27. has_many :notifications, inverse_of: :account, dependent: :destroy
  28. # Follow relations
  29. has_many :follow_requests, dependent: :destroy
  30. has_many :active_relationships, class_name: 'Follow', foreign_key: 'account_id', dependent: :destroy
  31. has_many :passive_relationships, class_name: 'Follow', foreign_key: 'target_account_id', dependent: :destroy
  32. has_many :following, -> { order('follows.id desc') }, through: :active_relationships, source: :target_account
  33. has_many :followers, -> { order('follows.id desc') }, through: :passive_relationships, source: :account
  34. # Block relationships
  35. has_many :block_relationships, class_name: 'Block', foreign_key: 'account_id', dependent: :destroy
  36. has_many :blocking, -> { order('blocks.id desc') }, through: :block_relationships, source: :target_account
  37. # Media
  38. has_many :media_attachments, dependent: :destroy
  39. # PuSH subscriptions
  40. has_many :subscriptions, dependent: :destroy
  41. pg_search_scope :search_for, against: { username: 'A', domain: 'B' },
  42. using: { tsearch: { prefix: true } }
  43. scope :remote, -> { where.not(domain: nil) }
  44. scope :local, -> { where(domain: nil) }
  45. scope :without_followers, -> { where('(select count(f.id) from follows as f where f.target_account_id = accounts.id) = 0') }
  46. scope :with_followers, -> { where('(select count(f.id) from follows as f where f.target_account_id = accounts.id) > 0') }
  47. scope :expiring, ->(time) { where(subscription_expires_at: nil).or(where('subscription_expires_at < ?', time)).remote.with_followers }
  48. scope :silenced, -> { where(silenced: true) }
  49. scope :suspended, -> { where(suspended: true) }
  50. scope :recent, -> { reorder('id desc') }
  51. scope :alphabetic, -> { order('domain ASC, username ASC') }
  52. def follow!(other_account)
  53. active_relationships.where(target_account: other_account).first_or_create!(target_account: other_account)
  54. end
  55. def block!(other_account)
  56. block_relationships.where(target_account: other_account).first_or_create!(target_account: other_account)
  57. end
  58. def unfollow!(other_account)
  59. follow = active_relationships.find_by(target_account: other_account)
  60. follow&.destroy
  61. end
  62. def unblock!(other_account)
  63. block = block_relationships.find_by(target_account: other_account)
  64. block&.destroy
  65. end
  66. def following?(other_account)
  67. following.include?(other_account)
  68. end
  69. def blocking?(other_account)
  70. blocking.include?(other_account)
  71. end
  72. def requested?(other_account)
  73. follow_requests.where(target_account: other_account).exists?
  74. end
  75. def local?
  76. domain.nil?
  77. end
  78. def acct
  79. local? ? username : "#{username}@#{domain}"
  80. end
  81. def subscribed?
  82. !subscription_expires_at.nil?
  83. end
  84. def favourited?(status)
  85. (status.reblog? ? status.reblog : status).favourites.where(account: self).count.positive?
  86. end
  87. def reblogged?(status)
  88. (status.reblog? ? status.reblog : status).reblogs.where(account: self).count.positive?
  89. end
  90. def keypair
  91. private_key.nil? ? OpenSSL::PKey::RSA.new(public_key) : OpenSSL::PKey::RSA.new(private_key)
  92. end
  93. def subscription(webhook_url)
  94. OStatus2::Subscription.new(remote_url, secret: secret, lease_seconds: 86_400 * 30, webhook: webhook_url, hub: hub_url)
  95. end
  96. def save_with_optional_avatar!
  97. save!
  98. rescue ActiveRecord::RecordInvalid => invalid
  99. if invalid.record.errors[:avatar_file_size] || invalid[:avatar_content_type]
  100. self.avatar = nil
  101. retry
  102. end
  103. raise invalid
  104. end
  105. def avatar_remote_url=(url)
  106. parsed_url = URI.parse(url)
  107. return if !%w(http https).include?(parsed_url.scheme) || self[:avatar_remote_url] == url
  108. self.avatar = parsed_url
  109. self[:avatar_remote_url] = url
  110. rescue OpenURI::HTTPError => e
  111. Rails.logger.debug "Error fetching remote avatar: #{e}"
  112. end
  113. def object_type
  114. :person
  115. end
  116. def to_param
  117. username
  118. end
  119. class << self
  120. def find_local!(username)
  121. find_remote!(username, nil)
  122. end
  123. def find_remote!(username, domain)
  124. where(arel_table[:username].matches(username.gsub(/[%_]/, '\\\\\0'))).where(domain.nil? ? { domain: nil } : arel_table[:domain].matches(domain.gsub(/[%_]/, '\\\\\0'))).take!
  125. end
  126. def find_local(username)
  127. find_local!(username)
  128. rescue ActiveRecord::RecordNotFound
  129. nil
  130. end
  131. def find_remote(username, domain)
  132. find_remote!(username, domain)
  133. rescue ActiveRecord::RecordNotFound
  134. nil
  135. end
  136. def following_map(target_account_ids, account_id)
  137. Follow.where(target_account_id: target_account_ids).where(account_id: account_id).map { |f| [f.target_account_id, true] }.to_h
  138. end
  139. def followed_by_map(target_account_ids, account_id)
  140. Follow.where(account_id: target_account_ids).where(target_account_id: account_id).map { |f| [f.account_id, true] }.to_h
  141. end
  142. def blocking_map(target_account_ids, account_id)
  143. Block.where(target_account_id: target_account_ids).where(account_id: account_id).map { |b| [b.target_account_id, true] }.to_h
  144. end
  145. def requested_map(target_account_ids, account_id)
  146. FollowRequest.where(target_account_id: target_account_ids).where(account_id: account_id).map { |r| [r.target_account_id, true] }.to_h
  147. end
  148. end
  149. before_create do
  150. if local?
  151. keypair = OpenSSL::PKey::RSA.new(Rails.env.test? ? 1024 : 2048)
  152. self.private_key = keypair.to_pem
  153. self.public_key = keypair.public_key.to_pem
  154. end
  155. end
  156. end