diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/hcl/app.rb | 4 | ||||
| -rw-r--r-- | lib/hcl/timesheet_resource.rb | 9 |
2 files changed, 13 insertions, 0 deletions
diff --git a/lib/hcl/app.rb b/lib/hcl/app.rb index ac1b992..4185a67 100644 --- a/lib/hcl/app.rb +++ b/lib/hcl/app.rb @@ -61,6 +61,10 @@ module HCl rescue SocketError => e STDERR.puts "Connection failed. (#{e.message})" exit 1 + rescue TimesheetResource::ThrottleFailure => e + STDERR.puts "Too many requests, retrying in #{e.retry_after+5} seconds..." + sleep e.retry_after+5 + run rescue TimesheetResource::AuthFailure => e STDERR.puts "Unable to authenticate: #{e}" request_config diff --git a/lib/hcl/timesheet_resource.rb b/lib/hcl/timesheet_resource.rb index 69eeb75..d4624ad 100644 --- a/lib/hcl/timesheet_resource.rb +++ b/lib/hcl/timesheet_resource.rb @@ -17,6 +17,13 @@ module HCl class TimesheetResource class Failure < StandardError; end class AuthFailure < StandardError; end + class ThrottleFailure < StandardError + attr_reader :retry_after + def initialize response + @retry_after = response.headers['Retry-After'].to_i + super "Too many requests! Try again in #{@retry_after} seconds." + end + end def self.configure opts = nil if opts @@ -68,6 +75,8 @@ module HCl response.body when Net::HTTPFound raise Failure, "Redirected! Perhaps your ssl configuration variable is set incorrectly?" + when Net::HTTPServiceUnavailable + raise ThrottleFailure, response when Net::HTTPUnauthorized raise AuthFailure, "Login failed." else |