about summary refs log tree commit diff
path: root/app/lib/provider_discovery.rb
blob: 3bec7211bdd184f48b838ea41c2444bee85fb95b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# frozen_string_literal: true

class ProviderDiscovery < OEmbed::ProviderDiscovery
  class << self
    def get(url, **options)
      provider = discover_provider(url, options)

      options.delete(:html)

      provider.get(url, options)
    end

    def discover_provider(url, **options)
      format = options[:format]

      html = if options[:html]
               Nokogiri::HTML(options[:html])
             else
               Request.new(:get, url).perform do |res|
                 raise OEmbed::NotFound, url if res.code != 200 || res.mime_type != 'text/html'
                 Nokogiri::HTML(res.body_with_limit)
               end
             end

      if format.nil? || format == :json
        provider_endpoint ||= html.at_xpath('//link[@type="application/json+oembed"]')&.attribute('href')&.value
        format ||= :json if provider_endpoint
      end

      if format.nil? || format == :xml
        provider_endpoint ||= html.at_xpath('//link[@type="text/xml+oembed"]')&.attribute('href')&.value
        format ||= :xml if provider_endpoint
      end

      raise OEmbed::NotFound, url if provider_endpoint.nil?
      begin
        provider_endpoint = Addressable::URI.parse(provider_endpoint)
        provider_endpoint.query = nil
        provider_endpoint = provider_endpoint.to_s
      rescue Addressable::URI::InvalidURIError
        raise OEmbed::NotFound, url
      end

      OEmbed::Provider.new(provider_endpoint, format)
    end
  end
end