183 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			183 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
class ProcessFeedService < BaseService
 | 
						|
  # Create local statuses from an Atom feed
 | 
						|
  # @param [String] body Atom feed
 | 
						|
  # @param [Account] account Account this feed belongs to
 | 
						|
  def call(body, account)
 | 
						|
    xml = Nokogiri::XML(body)
 | 
						|
    update_remote_profile_service.(xml.at_xpath('/xmlns:feed/xmlns:author'), account) unless xml.at_xpath('/xmlns:feed').nil?
 | 
						|
    xml.xpath('//xmlns:entry').each { |entry| process_entry(account, entry) }
 | 
						|
  end
 | 
						|
 | 
						|
  private
 | 
						|
 | 
						|
  def process_entry(account, entry)
 | 
						|
    return unless [:note, :comment, :activity].include? object_type(entry)
 | 
						|
 | 
						|
    status = Status.find_by(uri: activity_id(entry))
 | 
						|
 | 
						|
    # If we already have a post and the verb is now "delete", we gotta delete it and move on!
 | 
						|
    if !status.nil? && verb(entry) == :delete
 | 
						|
      delete_post!(status)
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    return unless status.nil?
 | 
						|
 | 
						|
    status = Status.new(uri: activity_id(entry), url: activity_link(entry), account: account, text: content(entry), created_at: published(entry), updated_at: updated(entry))
 | 
						|
 | 
						|
    if verb(entry) == :share
 | 
						|
      add_reblog!(entry, status)
 | 
						|
    elsif verb(entry) == :post
 | 
						|
      if thread_id(entry).nil?
 | 
						|
        add_post!(entry, status)
 | 
						|
      else
 | 
						|
        add_reply!(entry, status)
 | 
						|
      end
 | 
						|
    end
 | 
						|
 | 
						|
    # If we added a status, go through accounts it mentions and create respective relations
 | 
						|
    unless status.new_record?
 | 
						|
      record_remote_mentions(status, entry.xpath('./xmlns:link[@rel="mentioned"]'))
 | 
						|
      DistributionWorker.perform_async(status.id)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def record_remote_mentions(status, links)
 | 
						|
    # Here we have to do a reverse lookup of local accounts by their URL!
 | 
						|
    # It's not pretty at all! I really wish all these protocols sticked to
 | 
						|
    # using acct:username@domain only! It would make things so much easier
 | 
						|
    # and tidier
 | 
						|
 | 
						|
    links.each do |mention_link|
 | 
						|
      href = Addressable::URI.parse(mention_link.attribute('href').value)
 | 
						|
 | 
						|
      if href.host == Rails.configuration.x.local_domain
 | 
						|
        # A local user is mentioned
 | 
						|
        mentioned_account = Account.find_local(href.path.gsub('/users/', ''))
 | 
						|
 | 
						|
        unless mentioned_account.nil?
 | 
						|
          mentioned_account.mentions.where(status: status).first_or_create(status: status)
 | 
						|
          NotificationMailer.mention(mentioned_account, status).deliver_later
 | 
						|
        end
 | 
						|
      else
 | 
						|
        # What to do about remote user?
 | 
						|
        # Are we supposed to do a search in the database by URL?
 | 
						|
        # We could technically open the URL, look for LRDD tags, get webfinger that way,
 | 
						|
        # finally acquire the acct:username@domain form, and then check DB
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def add_post!(_entry, status)
 | 
						|
    status.save!
 | 
						|
  end
 | 
						|
 | 
						|
  def add_reblog!(entry, status)
 | 
						|
    status.reblog = find_original_status(entry, target_id(entry))
 | 
						|
 | 
						|
    if status.reblog.nil?
 | 
						|
      status.reblog = fetch_remote_status(entry)
 | 
						|
    end
 | 
						|
 | 
						|
    if !status.reblog.nil?
 | 
						|
      status.save!
 | 
						|
      NotificationMailer.reblog(status.reblog, status.account).deliver_later if status.reblog.local?
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def add_reply!(entry, status)
 | 
						|
    status.thread = find_original_status(entry, thread_id(entry))
 | 
						|
    status.save!
 | 
						|
  end
 | 
						|
 | 
						|
  def delete_post!(status)
 | 
						|
    status.destroy!
 | 
						|
  end
 | 
						|
 | 
						|
  def find_original_status(_xml, id)
 | 
						|
    return nil if id.nil?
 | 
						|
 | 
						|
    if local_id?(id)
 | 
						|
      Status.find(unique_tag_to_local_id(id, 'Status'))
 | 
						|
    else
 | 
						|
      Status.find_by(uri: id)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def fetch_remote_status(xml)
 | 
						|
    username = xml.at_xpath('./activity:object/xmlns:author/xmlns:name').content
 | 
						|
    url      = xml.at_xpath('./activity:object/xmlns:author/xmlns:uri').content
 | 
						|
    domain   = Addressable::URI.parse(url).host
 | 
						|
    account  = Account.find_by(username: username, domain: domain)
 | 
						|
 | 
						|
    if account.nil?
 | 
						|
      account = follow_remote_account_service.("#{username}@#{domain}", false)
 | 
						|
      return nil if account.nil?
 | 
						|
    end
 | 
						|
 | 
						|
    Status.new(account: account, uri: target_id(xml), text: target_content(xml), url: target_url(xml))
 | 
						|
  end
 | 
						|
 | 
						|
  def published(xml)
 | 
						|
    xml.at_xpath('./xmlns:published').content
 | 
						|
  end
 | 
						|
 | 
						|
  def updated(xml)
 | 
						|
    xml.at_xpath('./xmlns:updated').content
 | 
						|
  end
 | 
						|
 | 
						|
  def content(xml)
 | 
						|
    xml.at_xpath('./xmlns:content').content
 | 
						|
  end
 | 
						|
 | 
						|
  def thread_id(xml)
 | 
						|
    xml.at_xpath('./thr:in-reply-to').attribute('ref').value
 | 
						|
  rescue
 | 
						|
    nil
 | 
						|
  end
 | 
						|
 | 
						|
  def target_id(xml)
 | 
						|
    xml.at_xpath('.//activity:object/xmlns:id').content
 | 
						|
  rescue
 | 
						|
    nil
 | 
						|
  end
 | 
						|
 | 
						|
  def activity_id(xml)
 | 
						|
    xml.at_xpath('./xmlns:id').content
 | 
						|
  end
 | 
						|
 | 
						|
  def activity_link(xml)
 | 
						|
    xml.at_xpath('./xmlns:link[@rel="alternate"]').attribute('href').value
 | 
						|
  rescue
 | 
						|
    ''
 | 
						|
  end
 | 
						|
 | 
						|
  def target_content(xml)
 | 
						|
    xml.at_xpath('.//activity:object/xmlns:content').content
 | 
						|
  end
 | 
						|
 | 
						|
  def target_url(xml)
 | 
						|
    xml.at_xpath('.//activity:object/xmlns:link[@rel="alternate"]').attribute('href').value
 | 
						|
  end
 | 
						|
 | 
						|
  def object_type(xml)
 | 
						|
    xml.at_xpath('./activity:object-type').content.gsub('http://activitystrea.ms/schema/1.0/', '').to_sym
 | 
						|
  rescue
 | 
						|
    :note
 | 
						|
  end
 | 
						|
 | 
						|
  def verb(xml)
 | 
						|
    xml.at_xpath('./activity:verb').content.gsub('http://activitystrea.ms/schema/1.0/', '').to_sym
 | 
						|
  rescue
 | 
						|
    :post
 | 
						|
  end
 | 
						|
 | 
						|
  def follow_remote_account_service
 | 
						|
    @follow_remote_account_service ||= FollowRemoteAccountService.new
 | 
						|
  end
 | 
						|
 | 
						|
  def update_remote_profile_service
 | 
						|
    @update_remote_profile_service ||= UpdateRemoteProfileService.new
 | 
						|
  end
 | 
						|
end
 |