class Unshorter

Attributes

max_level[R]

Public Class Methods

new(max_level = 5) click to toggle source

# File lib/unshorter.rb, line 9
def initialize(max_level = 5)
  @max_level = max_level
end

Public Instance Methods

call(url) click to toggle source

# File lib/unshorter.rb, line 13
def call(url)
  url = "http://#{url}" if url !~ %r{^(http|https)://}
  original_uri = uri = URI(url)
  max_level.times do
    response = Net::HTTP.get_response(uri)

    if %w[301 302].include?(response.code)
      uri = URI(response["location"])
      next
    end

    if original_uri.host.sub(/www\./, "") == uri.host
      break response.body.scan(%r{location=['|"]([a-z:/\.\d]+)['|"]}).flatten.fetch(0, nil)
    end

    break uri.to_s
  end
end