# frozen_string_literal: true require 'cgi' ## # The UriFormatter handles URIs from user-input and escaping. # # uf = Gem::UriFormatter.new 'example.com' # # p uf.normalize #=> 'http://example.com' class Gem::UriFormatter ## # The URI to be formatted. attr_reader :uri ## # Creates a new URI formatter for +uri+. def initialize(uri) @uri = uri end ## # Escapes the #uri for use as a CGI parameter def escape return unless @uri CGI.escape @uri end ## # Normalize the URI by adding "http://" if it is missing. def normalize (@uri =~ /^(https?|ftp|file):/i) ? @uri : "http://#{@uri}" end ## # Unescapes the #uri which came from a CGI parameter def unescape return unless @uri CGI.unescape @uri end end