* Validate internal actor * Use “internal.actor” by default for the server actor username * Fix instance actor username on the fly if it includes ':' * Change actor name from internal.actor to mastodon.internal
		
			
				
	
	
		
			67 lines
		
	
	
		
			1.6 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			67 lines
		
	
	
		
			1.6 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| module AccountFinderConcern
 | |
|   extend ActiveSupport::Concern
 | |
| 
 | |
|   class_methods do
 | |
|     def find_local!(username)
 | |
|       find_local(username) || raise(ActiveRecord::RecordNotFound)
 | |
|     end
 | |
| 
 | |
|     def find_remote!(username, domain)
 | |
|       find_remote(username, domain) || raise(ActiveRecord::RecordNotFound)
 | |
|     end
 | |
| 
 | |
|     def representative
 | |
|       actor = Account.find(-99).tap(&:ensure_keys!)
 | |
|       actor.update!(username: 'mastodon.internal') if actor.username.include?(':')
 | |
|       actor
 | |
|     rescue ActiveRecord::RecordNotFound
 | |
|       Account.create!(id: -99, actor_type: 'Application', locked: true, username: 'mastodon.internal')
 | |
|     end
 | |
| 
 | |
|     def find_local(username)
 | |
|       find_remote(username, nil)
 | |
|     end
 | |
| 
 | |
|     def find_remote(username, domain)
 | |
|       AccountFinder.new(username, domain).account
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   class AccountFinder
 | |
|     attr_reader :username, :domain
 | |
| 
 | |
|     def initialize(username, domain)
 | |
|       @username = username
 | |
|       @domain = domain
 | |
|     end
 | |
| 
 | |
|     def account
 | |
|       scoped_accounts.order(id: :asc).take
 | |
|     end
 | |
| 
 | |
|     private
 | |
| 
 | |
|     def scoped_accounts
 | |
|       Account.unscoped.tap do |scope|
 | |
|         scope.merge! with_usernames
 | |
|         scope.merge! matching_username
 | |
|         scope.merge! matching_domain
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     def with_usernames
 | |
|       Account.where.not(Account.arel_table[:username].lower.eq '')
 | |
|     end
 | |
| 
 | |
|     def matching_username
 | |
|       Account.where(Account.arel_table[:username].lower.eq username.to_s.downcase)
 | |
|     end
 | |
| 
 | |
|     def matching_domain
 | |
|       Account.where(Account.arel_table[:domain].lower.eq(domain.nil? ? nil : domain.to_s.downcase))
 | |
|     end
 | |
|   end
 | |
| end
 |