This repository has been archived on 2019-05-14. You can view files and clone it, but cannot push or open issues or pull requests.
mastodon/app/models/account.rb

146 lines
4.7 KiB
Ruby
Raw Normal View History

2016-08-17 15:56:23 +00:00
class Account < ApplicationRecord
include Targetable
MENTION_RE = /(?:^|\s|\.|>)@([a-z0-9_]+(?:@[a-z0-9\.\-]+)?)/i
IMAGE_MIME_TYPES = ['image/jpeg', 'image/png', 'image/gif']
2016-02-22 15:00:20 +00:00
# Local users
has_one :user, inverse_of: :account
validates :username, presence: true, format: { with: /\A[a-z0-9_]+\z/i, message: 'only letters, numbers and underscores' }, uniqueness: { scope: :domain, case_sensitive: false }, if: 'local?'
validates :username, presence: true, uniqueness: { scope: :domain, case_sensitive: true }, unless: 'local?'
2016-02-22 15:00:20 +00:00
# Avatar upload
2016-09-08 16:23:59 +00:00
has_attached_file :avatar, styles: { large: '300x300#', medium: '96x96#', small: '48x48#' }
validates_attachment_content_type :avatar, content_type: IMAGE_MIME_TYPES
validates_attachment_size :avatar, less_than: 2.megabytes
# Header upload
has_attached_file :header, styles: { medium: '700x335#' }
validates_attachment_content_type :header, content_type: IMAGE_MIME_TYPES
validates_attachment_size :header, less_than: 2.megabytes
2016-03-16 10:18:09 +00:00
# Local user profile validations
validates :display_name, length: { maximum: 30 }, if: 'local?'
validates :note, length: { maximum: 124 }, if: 'local?'
2016-02-22 15:00:20 +00:00
# Timelines
has_many :stream_entries, inverse_of: :account
2016-02-20 21:53:20 +00:00
has_many :statuses, inverse_of: :account
has_many :favourites, inverse_of: :account
has_many :mentions, inverse_of: :account
2016-02-20 21:53:20 +00:00
2016-02-22 15:00:20 +00:00
# Follow relations
has_many :active_relationships, class_name: 'Follow', foreign_key: 'account_id', dependent: :destroy
has_many :passive_relationships, class_name: 'Follow', foreign_key: 'target_account_id', dependent: :destroy
has_many :following, through: :active_relationships, source: :target_account
has_many :followers, through: :passive_relationships, source: :account
2016-09-05 15:46:36 +00:00
has_many :media_attachments, dependent: :destroy
scope :remote, -> { where.not(domain: nil) }
scope :local, -> { where(domain: nil) }
scope :without_followers, -> { where('(select count(f.id) from follows as f where f.target_account_id = accounts.id) = 0') }
scope :with_followers, -> { where('(select count(f.id) from follows as f where f.target_account_id = accounts.id) > 0') }
scope :expiring, -> (time) { where(subscription_expires_at: nil).or(where('subscription_expires_at < ?', time)).remote.with_followers }
2016-02-22 15:00:20 +00:00
def follow!(other_account)
self.active_relationships.where(target_account: other_account).first_or_create!(target_account: other_account)
2016-02-22 15:00:20 +00:00
end
def unfollow!(other_account)
follow = self.active_relationships.find_by(target_account: other_account)
follow.destroy unless follow.nil?
2016-02-22 15:00:20 +00:00
end
def following?(other_account)
following.include?(other_account)
end
def local?
self.domain.nil?
end
2016-02-22 17:10:30 +00:00
def acct
local? ? self.username : "#{self.username}@#{self.domain}"
end
def subscribed?
!self.subscription_expires_at.nil?
2016-02-22 17:10:30 +00:00
end
def favourited?(status)
(status.reblog? ? status.reblog : status).favourites.where(account: self).count > 0
end
def reblogged?(status)
(status.reblog? ? status.reblog : status).reblogs.where(account: self).count > 0
end
2016-02-22 15:00:20 +00:00
def keypair
self.private_key.nil? ? OpenSSL::PKey::RSA.new(self.public_key) : OpenSSL::PKey::RSA.new(self.private_key)
end
2016-02-20 21:53:20 +00:00
def subscription(webhook_url)
OStatus2::Subscription.new(self.remote_url, secret: self.secret, lease_seconds: 86400 * 30, webhook: webhook_url, hub: self.hub_url)
2016-02-20 21:53:20 +00:00
end
2016-02-26 14:28:08 +00:00
def ping!(atom_url, hubs)
return unless local? && !Rails.env.development?
OStatus2::Publication.new(atom_url, hubs).publish
end
def avatar_remote_url=(url)
unless self[:avatar_remote_url] == url
self.avatar = URI.parse(url)
end
self[:avatar_remote_url] = url
end
def object_type
:person
end
def to_param
self.username
end
def self.find_local!(username)
2016-09-04 19:06:04 +00:00
self.find_remote!(username, nil)
end
def self.find_remote!(username, domain)
table = self.arel_table
2016-09-04 19:06:04 +00:00
self.where(table[:username].matches(username)).where(domain: domain).take!
end
def self.find_local(username)
self.find_local!(username)
rescue ActiveRecord::RecordNotFound
nil
end
2016-09-04 19:06:04 +00:00
def self.find_remote(username, domain)
self.find_remote!(username, domain)
rescue ActiveRecord::RecordNotFound
nil
end
def self.following_map(target_account_ids, account_id)
Follow.where(target_account_id: target_account_ids).where(account_id: account_id).map { |f| [f.target_account_id, true] }.to_h
end
def self.followed_by_map(target_account_ids, account_id)
Follow.where(account_id: target_account_ids).where(target_account_id: account_id).map { |f| [f.account_id, true] }.to_h
end
2016-02-26 14:28:08 +00:00
before_create do
if local?
keypair = OpenSSL::PKey::RSA.new(Rails.env.test? ? 1024 : 2048)
2016-02-26 14:28:08 +00:00
self.private_key = keypair.to_pem
self.public_key = keypair.public_key.to_pem
end
end
2016-02-20 21:53:20 +00:00
end