tootlab-mastodon/app/services/follow_remote_account_service.rb

83 lines
2.4 KiB
Ruby
Raw Normal View History

2016-02-24 06:57:29 -05:00
class FollowRemoteAccountService < BaseService
# Find or create a local account for a remote user.
# When creating, look up the user's webfinger and fetch all
# important information from their feed
# @param [String] uri User URI in the form of username@domain
# @param [Boolean] subscribe Whether to initiate a PubSubHubbub subscription
# @return [Account]
def call(uri, subscribe = true)
2016-02-22 10:00:20 -05:00
username, domain = uri.split('@')
2016-02-20 16:53:20 -05:00
account = Account.where(username: username, domain: domain).first
2016-02-22 12:10:30 -05:00
if account.nil?
account = Account.new(username: username, domain: domain)
elsif account.subscribed?
return account
end
2016-02-20 16:53:20 -05:00
2016-02-22 12:10:30 -05:00
data = Goldfinger.finger("acct:#{uri}")
2016-02-20 16:53:20 -05:00
account.remote_url = data.link('http://schemas.google.com/g/2010#updates-from').href
account.salmon_url = data.link('salmon').href
account.url = data.link('http://webfinger.net/rel/profile-page').href
2016-02-20 16:53:20 -05:00
account.public_key = magic_key_to_pem(data.link('magic-public-key').href)
account.private_key = nil
feed = get_feed(account.remote_url)
hubs = feed.xpath('//xmlns:link[@rel="hub"]')
2016-02-22 12:10:30 -05:00
return nil if hubs.empty? || hubs.first.attribute('href').nil? || feed.at_xpath('/xmlns:feed/xmlns:author/xmlns:uri').nil?
2016-02-20 16:53:20 -05:00
2016-02-22 12:10:30 -05:00
account.uri = feed.at_xpath('/xmlns:feed/xmlns:author/xmlns:uri').content
2016-02-20 16:53:20 -05:00
account.hub_url = hubs.first.attribute('href').value
2016-02-22 12:10:30 -05:00
get_profile(feed, account)
2016-02-20 16:53:20 -05:00
account.save!
if subscribe
account.secret = SecureRandom.hex
account.verify_token = SecureRandom.hex
subscription = account.subscription(subscription_url(account))
subscription.subscribe
account.save!
end
2016-02-22 12:10:30 -05:00
return account
rescue Goldfinger::Error, HTTP::Error
2016-02-22 12:10:30 -05:00
nil
2016-02-20 16:53:20 -05:00
end
private
def get_feed(url)
response = http_client.get(Addressable::URI.parse(url))
Nokogiri::XML(response)
end
2016-02-22 12:10:30 -05:00
def get_profile(xml, account)
author = xml.at_xpath('/xmlns:feed/xmlns:author')
update_remote_profile_service.(author, account)
2016-02-22 12:10:30 -05:00
end
2016-02-20 16:53:20 -05:00
def magic_key_to_pem(magic_key)
_, modulus, exponent = magic_key.split('.')
modulus, exponent = [modulus, exponent].map { |n| Base64.urlsafe_decode64(n).bytes.inject(0) { |num, byte| (num << 8) | byte } }
key = OpenSSL::PKey::RSA.new
key.n = modulus
2016-02-22 10:00:20 -05:00
key.e = exponent
2016-02-20 16:53:20 -05:00
key.to_pem
end
def update_remote_profile_service
@update_remote_profile_service ||= UpdateRemoteProfileService.new
end
2016-02-20 16:53:20 -05:00
def http_client
HTTP
end
end