forked from treehouse/mastodon
50 lines
1.3 KiB
Ruby
50 lines
1.3 KiB
Ruby
class Follow < ApplicationRecord
|
|
include Streamable
|
|
|
|
belongs_to :account
|
|
belongs_to :target_account, class_name: 'Account'
|
|
|
|
validates :account, :target_account, presence: true
|
|
validates :account_id, uniqueness: { scope: :target_account_id }
|
|
|
|
def verb
|
|
destroyed? ? :unfollow : :follow
|
|
end
|
|
|
|
def target
|
|
target_account
|
|
end
|
|
|
|
def object_type
|
|
:person
|
|
end
|
|
|
|
def title
|
|
destroyed? ? "#{account.acct} is no longer following #{target_account.acct}" : "#{account.acct} started following #{target_account.acct}"
|
|
end
|
|
|
|
after_create :add_to_graph
|
|
after_destroy :remove_from_graph
|
|
|
|
private
|
|
|
|
def add_to_graph
|
|
neo = Neography::Rest.new
|
|
|
|
a = neo.create_unique_node('account_index', 'Account', account_id.to_s, account_id: account_id)
|
|
b = neo.create_unique_node('account_index', 'Account', target_account_id.to_s, account_id: target_account_id)
|
|
|
|
neo.create_unique_relationship('follow_index', 'Follow', id.to_s, 'follows', a, b)
|
|
rescue Neography::NeographyError, Excon::Error::Socket => e
|
|
Rails.logger.error e
|
|
end
|
|
|
|
def remove_from_graph
|
|
neo = Neography::Rest.new
|
|
rel = neo.get_relationship_index('follow_index', 'Follow', id.to_s)
|
|
neo.delete_relationship(rel)
|
|
rescue Neography::NeographyError, Excon::Error::Socket => e
|
|
Rails.logger.error e
|
|
end
|
|
end
|