53 lines
		
	
	
	
		
			1.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			53 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
 | |
| 
 | |
|   def sync!
 | |
|     add_to_graph
 | |
|   end
 | |
| 
 | |
|   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
 |