From 780bea3eac378f7b129e4d79948f3bc2548c86cb Mon Sep 17 00:00:00 2001
From: Patrick J Cherry <patrick@bytemark.co.uk>
Date: Wed, 4 Jul 2012 16:42:07 +0100
Subject: Moved HTTP requests into a generic http api client.

---
 lib/mauve/calendar_interface.rb | 135 ++++------------------------------------
 1 file changed, 11 insertions(+), 124 deletions(-)

(limited to 'lib/mauve/calendar_interface.rb')

diff --git a/lib/mauve/calendar_interface.rb b/lib/mauve/calendar_interface.rb
index c4417ba..5282d79 100644
--- a/lib/mauve/calendar_interface.rb
+++ b/lib/mauve/calendar_interface.rb
@@ -1,22 +1,14 @@
 # encoding: UTF-8
-require 'log4r'
-require 'net/http'
-require 'net/https'
-require 'uri'
+require 'mauve/generic_http_api_client'
 
 module Mauve
 
   # Interface to the Bytemark calendar.
   #
-  class CalendarInterface 
+  class CalendarInterface  < GenericHttpApiClient
     
     class << self
 
-      # @return [Log4r::Logger]
-      def logger
-        @logger ||= Log4r::Logger.new(self.to_s)
-      end
-
       def get_attendees(klass, at=Time.now)
         #
         # Returns nil if no calendar_url has been set.
@@ -26,7 +18,7 @@ module Mauve
         url = Configuration.current.bytemark_calendar_url.dup
 
         url.merge!(File.join(url.path, "/api/attendees/#{klass}/#{at.strftime("%Y-%m-%dT%H:%M:00")}"))
-        ans = do_get(url)
+        ans = do_get_yaml(url)
 
         return [] unless ans.is_a?(Array)
 
@@ -41,7 +33,7 @@ module Mauve
 
         url = Configuration.current.bytemark_calendar_url.dup
         url.merge!(File.join(url.path, "/api/bank_holidays/#{at.strftime("%Y-%m-%d")}"))
-        ans = do_get(url)
+        ans = do_get_yaml(url)
 
         return [] unless ans.is_a?(Array)
         ans.select{|x| x.is_a?(Date)}
@@ -71,119 +63,14 @@ module Mauve
 
       private
 
-      # Grab a URL from the wide web.
-      #
-      # @todo boot this in its own class since list of ips will need it too.
-      #
-      # @param [String] uri -- a URL
-      # @return [String or nil] -- the contents of the URI or nil if an error has been encountered.
-      #
-      def do_get (uri, limit = 11)
-
-        if 0 > limit
-          logger.warn("HTTP redirect too deep for #{uri}.")
-          return nil
-        end
-        
-        begin
-          uri = URI.parse(uri) unless uri.is_a?(URI::HTTP)
-
-          raise ArgumentError, "#{uri_str.inspect} doesn't look like an HTTP uri" unless uri.is_a?(URI::HTTP)
-
-          http = Net::HTTP.new(uri.host, uri.port)
-
-          #
-          # Five second timeouts.
-          #
-          http.open_timeout = http.read_timeout = Configuration.current.remote_http_timeout || 5
- 
-          if (uri.scheme == "https")
-            http.use_ssl     = true
-            http.ca_path     = "/etc/ssl/certs/" if File.directory?("/etc/ssl/certs")
-            http.verify_mode = Configuration.current.remote_https_verify_mode || OpenSSL::SSL::VERIFY_NONE
-          end
-
-          response = http.start { http.get(uri.request_uri()) }
-
-          if response.is_a?(Net::HTTPOK)
-            #
-            # Parse the string as YAML.
-            #
-            result = if response.body.is_a?(String)
-              begin
-                YAML.load(response.body)
-              rescue YAML::Error => err
-                logger.error "Caught #{ex.class.to_s} (#{ex.to_s}) whilst querying #{url.to_s}."
-                logger.debug err.backtrace.join("\n")
-                nil
-              end
-            else
-              nil
-            end
-
-            return result
-          elsif response.is_a?(Net::HTTPRedirection) and response.key?('Location')
-            location = response['Location']
-            
-            #
-            # Bodge locations..
-            #
-            if location =~ /^\//
-              location = uri.class.build([uri.userinfo, uri.host, uri.port, nil, nil, nil]).to_s + location
-            end
-
-            return do_get(location, limit-1) 
-
-          else
-            logger.warn("Request to #{uri.to_s} returned #{response.code} #{response.message}.")
-            return nil
-
-          end
-
-        rescue Timeout::Error => ex
-          logger.error("Timeout caught during fetch of #{uri.to_s}.")
-
-        rescue StandardError => ex
-          logger.error("#{ex.class} caught during fetch of #{uri.to_s}: #{ex.to_s}.")
-          logger.debug(ex.backtrace.join("\n"))
-
-        end
-
-        return nil
-      end
-
-      # This does HTTP fetches with a 5 minute cache
-      #
-      # @param [String] url
-      # @param [Time] cache_until
-      #
-      # @return [String or nil]
-      def do_get_with_cache(url, cache_until = Time.now + 5.minutes)
-        @cache ||= {}
-
-        if @cache.has_key?(url)
-          result, cached_until = @cache[url]
-
-          return result if cached_until > Time.now and not result.nil?
-        end
-
-        result = do_get(url)
-        @cache[url] = [result, cache_until] unless result.nil?
-
-        return result
-      end
-
-      #
-      # This should get called periodically.
-      #
-      def clean_cache
-
-        @cache.keys.select do |url|
-          result, cached_until = @cache[url]
-          @cache.delete(url) if !cached_until.is_a?(Time) or cached_until <= Time.now
-        end
+      def get_yaml(url)
+        resp = do_get(url)
 
-        @cache 
+        return (resp.is_a?(String) ? YAML.load(resp) : nil)
+      rescue StandardError => err
+        logger.error "Caught #{ex.class.to_s} (#{ex.to_s}) whilst querying #{url.to_s}."
+        logger.debug err.backtrace.join("\n")
+        nil
       end
 
     end
-- 
cgit v1.2.1