mirror of https://github.com/mastodon/goldfinger
57 lines
1.3 KiB
Ruby
57 lines
1.3 KiB
Ruby
require 'addressable'
|
|
require 'nokogiri'
|
|
|
|
module Goldfinger
|
|
class Client
|
|
include Goldfinger::Utils
|
|
|
|
def initialize(uri)
|
|
@uri = uri
|
|
end
|
|
|
|
def finger
|
|
ssl = true
|
|
|
|
begin
|
|
_, template = perform_get(url(ssl))
|
|
rescue HTTP::Error
|
|
if ssl
|
|
ssl = false
|
|
retry
|
|
else
|
|
raise Goldfinger::NotFoundError
|
|
end
|
|
end
|
|
|
|
headers, body = perform_get(url_from_template(template))
|
|
|
|
raise Goldfinger::Error, "Invalid response mime type: #{headers.get(HTTP::Headers::CONTENT_TYPE).first}" unless ['application/jrd+json', 'application/xrd+xml'].include?(headers.get(HTTP::Headers::CONTENT_TYPE).first)
|
|
|
|
Goldfinger::Result.new(headers, body)
|
|
rescue HTTP::Error
|
|
raise Goldfinger::NotFoundError
|
|
rescue OpenSSL::SSL::SSLError
|
|
raise Goldfinger::SSLError
|
|
end
|
|
|
|
private
|
|
|
|
def url(ssl = true)
|
|
"http#{'s' if ssl}://#{domain}/.well-known/host-meta"
|
|
end
|
|
|
|
def url_from_template(template)
|
|
xml = Nokogiri::XML(template)
|
|
links = xml.xpath('//xmlns:Link[@rel="lrdd"]')
|
|
|
|
raise Goldfinger::NotFoundError if links.empty?
|
|
|
|
links.first.attribute('template').value.gsub('{uri}', @uri)
|
|
end
|
|
|
|
def domain
|
|
@uri.split('@').last
|
|
end
|
|
end
|
|
end
|