Browse Source

Improve redirect handling in FetchAtomService

closed-social-glitch-2
Eugen Rochko 7 years ago
parent
commit
210362e665
2 changed files with 5 additions and 4 deletions
  1. +4
    -4
      app/services/fetch_atom_service.rb
  2. +1
    -0
      app/services/fetch_remote_account_service.rb

+ 4
- 4
app/services/fetch_atom_service.rb View File

@ -8,7 +8,7 @@ class FetchAtomService < BaseService
if response.mime_type == 'application/atom+xml'
return [url, fetch(url)]
elsif !response['Link'].blank?
return process_headers(response)
return process_headers(url, response)
else
return process_html(fetch(url))
end
@ -26,13 +26,13 @@ class FetchAtomService < BaseService
return [alternate_link['href'], fetch(alternate_link['href'])]
end
def process_headers(response)
def process_headers(url, response)
Rails.logger.debug "Processing link header"
link_header = LinkHeader.parse(response['Link'])
alternate_link = link_header.find_link(['rel', 'alternate'], ['type', 'application/atom+xml'])
return nil if alternate_link.nil?
return process_html(fetch(url)) if alternate_link.nil?
return [alternate_link.href, fetch(alternate_link.href)]
end
@ -41,6 +41,6 @@ class FetchAtomService < BaseService
end
def http_client
HTTP.timeout(:per_operation, write: 20, connect: 20, read: 50)
HTTP.timeout(:per_operation, write: 20, connect: 20, read: 50).follow
end
end

+ 1
- 0
app/services/fetch_remote_account_service.rb View File

@ -9,6 +9,7 @@ class FetchRemoteAccountService < BaseService
private
def process_atom(url, body)
xml = Nokogiri::XML(body)
url_parts = Addressable::URI.parse(url)
username = xml.at_xpath('//xmlns:author/xmlns:name').try(:content)
domain = url_parts.host

Loading…
Cancel
Save