From ed88b4acdb5a91074b965bf95550d8b4ad1b5731 Mon Sep 17 00:00:00 2001
From: Alex Schultz <aschultz@mirantis.com>
Date: Mon, 1 Jun 2015 16:38:36 -0500
Subject: [PATCH] Add proxy support for url_available

In order to support deployments that are configuring proxies for
their yum/apt repositories, we need to also support proxies for the
repo connectivity_tests task. This change adds the support to pass
in a proxy to use when checking the url availability.

Change-Id: Id363c12c35f35b07e280f196091f71cfe91b7311
Related-Bug: 1460169
Related-Bug: 1460108
---
 .../puppet/parser/functions/url_available.rb  | 56 ++++++++++++++++---
 1 file changed, 49 insertions(+), 7 deletions(-)

diff --git a/deployment/puppet/osnailyfacter/lib/puppet/parser/functions/url_available.rb b/deployment/puppet/osnailyfacter/lib/puppet/parser/functions/url_available.rb
index b854631083..80e092d118 100644
--- a/deployment/puppet/osnailyfacter/lib/puppet/parser/functions/url_available.rb
+++ b/deployment/puppet/osnailyfacter/lib/puppet/parser/functions/url_available.rb
@@ -5,40 +5,82 @@ require 'uri'
 
 Puppet::Parser::Functions::newfunction(:url_available, :doc => <<-EOS
 The url_available function attempts to make a http request to a url and throws
-a puppet error if the URL is unavailable.  The url_available function takes
-a single parameter that can be one of the following:
+a puppet error if the URL is unavailable.  The url_available function can take
+up to two paramters. The first paramter is the URL which is required and can be
+one of the following:
 1) String - a single url string
 3) Hash - a hash with the url set to the key of 'uri'
 2) Array - an array of url strings or an array of hashes that match the
 previous format.
 
+The second paramter is a proxy url. It should be a valid uri and will be parsed
+to ensure it is properly formated. It should be noted that if the url provided
+to the url_available function is a hash, the proxy can be specified using a key
+named 'proxy'. The hostname, port, username and password will be parsed from
+this uri to be used when issuing a request via the proxy.
+
 Examples:
+# no proxy
 url_available('http://www.google.com')
 url_available(['http://www.google.com', 'http://www.mirantis.com'])
 url_available({ 'uri' => 'http://www.google.com' })
 url_available([{ 'uri' => 'http://www.google.com' },
                { 'uri' => 'http://www.mirantis.com' }]
+# with a proxy
+url_available('http://www.google.com', 'http://proxy.example.com:3128/')
+url_available({ 'uri' => 'http://www.google.com',
+                'proxy' => 'http://proxy.example.com:3128'})
+
+# with a proxy with authentication
+url_available('http://www.google.com', 'http://user:pass@proxy.example.com:3128/')
+url_available({ 'uri'   => 'http://www.google.com',
+                'proxy' => 'http://user:pass@proxy.example.com:3128/'})
+
 EOS
 ) do |argv|
   url = argv[0]
+  http_proxy = argv[1]
+
+  def fetch(url, http_proxy = nil)
+    # proxy variables, set later if http_proxy is provided or there is a proxy
+    # element provided as part of the a url that is a Hash
+    proxy_host = nil
+    proxy_port = nil
+    proxy_user = nil
+    proxy_pass = nil
 
-  def fetch(url)
     # check the type of url being passed, if hash look for the uri key
     if url.instance_of? Hash
       if url.has_key?('uri')
         uri = url['uri']
       end
+      if url.has_key?('proxy')
+        http_proxy = url['proxy']
+      end
     elsif url.instance_of? String
       uri = url
     else
       raise Puppet::ParseError, "Invalid url type passed to the url_available
 function. Must be of type String or Hash."
     end
+
+    # attempt to parse the proxy if it's not nil or a blank string
+    unless [nil, ''].include?(http_proxy)
+      begin
+        proxy = URI.parse(http_proxy)
+        proxy_host = proxy.host
+        proxy_port = proxy.port
+        proxy_user, proxy_pass = proxy.userinfo.split(/:/) if proxy.userinfo
+      rescue Exception => e
+        puts "Unable to parse proxy settings from '#{http_proxy}', will ignore proxy setting. Error '#{e}'"
+      end
+    end
+
     puts "Checking #{uri}"
     begin
       out = Timeout::timeout(180) do
         u = URI.parse(uri)
-        http = Net::HTTP.new(u.host, u.port)
+        http = Net::HTTP.new(u.host, u.port, proxy_host, proxy_port, proxy_user, proxy_pass)
         http.open_timeout = 60
         http.read_timeout = 60
         request = Net::HTTP::Get.new(u.request_uri)
@@ -52,16 +94,16 @@ function. Must be of type String or Hash."
   end
 
   #Add sleep before checking for repos
-  #as Ubuntu waits 32 seconds for the 
+  #as Ubuntu waits 32 seconds for the
   #bridge to become ready
   sleep 32
   # if passed an array, iterate through the array an check each element
   if url.instance_of? Array
     url.each do |u|
-      fetch(u)
+      fetch(u, http_proxy)
     end
   else
-    fetch(url)
+    fetch(url, http_proxy)
   end
   return true
 end