From f0dff8eb742441901f6a88bf0ce078efe60e4085 Mon Sep 17 00:00:00 2001
From: Steve Kemp <steve@steve.org.uk>
Date: Tue, 13 Nov 2012 20:35:07 +0000
Subject:   HTTP-fetching of macros works.

---
 bin/custodian-enqueue | 68 +++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 66 insertions(+), 2 deletions(-)

(limited to 'bin/custodian-enqueue')

diff --git a/bin/custodian-enqueue b/bin/custodian-enqueue
index a462b0a..6e5627a 100755
--- a/bin/custodian-enqueue
+++ b/bin/custodian-enqueue
@@ -11,6 +11,9 @@
 require 'beanstalk-client'
 require 'getoptlong'
 require 'json'
+require 'net/http'
+require 'net/https'
+require 'uri'
 
 
 
@@ -72,6 +75,61 @@ class MonitorConfig
 
 
 
+
+  #
+  # Retrieve a HTTP page from the web - this is used for macro-expansion
+  #
+  # NOTE:  This came from sentinel.
+  #
+  def getURL (uri_str)
+    begin
+      uri_str = 'http://' + uri_str unless uri_str.match(/^http/)
+      url = URI.parse(uri_str)
+      http = Net::HTTP.new(url.host, url.port)
+      http.open_timeout = 3
+      http.read_timeout = 3
+
+      if (url.scheme == "https")
+        http.use_ssl = true
+        http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+      end
+
+      response = nil
+
+      if nil == url.query
+        response = http.start { http.get(url.path) }
+      else
+        response = http.start { http.get("#{url.path}?#{url.query}") }
+      end
+
+
+      if ( response.code.to_i != 200 )
+        puts "Status code of #{uri_str} was #{response.code}"
+        puts "ABORTING"
+        exit( 0 )
+      end
+
+      case response
+      when Net::HTTPRedirection
+      then
+        newURL = response['location'].match(/^http/)?
+        response['Location']:uri_str+response['Location']
+        return( getURL(newURL) )
+      else
+        return response.body
+      end
+
+    rescue Errno::EHOSTUNREACH => ex
+      raise ex, "no route to host"
+    rescue Timeout::Error => ex
+      raise ex, "timeout"
+    rescue Errno::ECONNREFUSED => ex
+      raise ex, "connection refused"
+    end
+  end
+
+
+
   #
   #  Define a macro, from the configuration file.
   #
@@ -93,8 +151,12 @@ class MonitorConfig
       #
       #  HTTP-fetch
       #
-      val.push( "steve")
-      val.push("kemp")
+      uri = $1.dup.chomp(".")
+
+      text = getURL(uri)
+      text.split( /[\r\n]/ ).each do |line|
+        val.push( line ) if ( line.length() > 0)
+      end
 
     elsif ( line =~ /\s(is|are)\s+(.*)\.+$/ )
 
@@ -408,3 +470,5 @@ if __FILE__ == $0 then
   mon = MonitorConfig.new( ENV['FILE'] )
   mon.parse_file();
 end
+
+
-- 
cgit v1.2.1