Add error class and rescue/raise common errors

README.md

@@ -80,17 +80,17 @@ puts discoverer.response # returns HTTP::Response
 
 ### Exception Handling
 
-There are several exceptions that may be raised by webmention-endpoint-ruby's underlying dependencies:
+There are several exceptions that may be raised by webmention-endpoint-ruby's underlying dependencies. These errors are raised as subclasses of `Webmention::Endpoint::Error` (which itself is a subclass of `StandardError`).
 
-- `Addressable::URI::InvalidURIError`
-- `HTTP::ConnectionError`
-- `HTTP::TimeoutError`
-- `HTTP::Redirector::TooManyRedirectsError`
+From [sporkmonger/addressable](/~https://github.com/sporkmonger/addressable):
 
-You'll want to make sure your code gracefully handles these exceptions. For more on each of these exceptions, see:
+- `Webmention::Endpoint::InvalidURIError`
 
-- [sporkmonger/addressable](/~https://github.com/sporkmonger/addressable) – Source code for the Addressable Ruby gem
-- [httprb/http](/~https://github.com/httprb/http) – Source code for the HTTP Ruby gem
+From [httprb/http](/~https://github.com/httprb/http):
+
+- `Webmention::Endpoint::ConnectionError`
+- `Webmention::Endpoint::TimeoutError`
+- `Webmention::Endpoint::TooManyRedirectsError`
 
 ## Contributing
 

lib/webmention/endpoint.rb

@@ -3,6 +3,7 @@
 require 'nokogiri'
 
 require 'webmention/endpoint/version'
+require 'webmention/endpoint/error'
 require 'webmention/endpoint/discover'
 
 module Webmention

lib/webmention/endpoint/discover.rb

@@ -25,6 +25,8 @@ def initialize(url)
 
         @url = url
         @uri = Addressable::URI.parse(url)
+      rescue Addressable::URI::InvalidURIError => error
+        raise InvalidURIError, error
       end
 
       def endpoint
@@ -36,6 +38,12 @@ def response
           connect: 10,
           read: 10
         ).get(uri)
+      rescue HTTP::ConnectionError => error
+        raise ConnectionError, error
+      rescue HTTP::TimeoutError => error
+        raise TimeoutError, error
+      rescue HTTP::Redirector::TooManyRedirectsError => error
+        raise TooManyRedirectsError, error
       end
 
       private
@@ -49,6 +57,8 @@ def absolutize(base:, relative:)
         return relative if relative_uri.absolute?
 
         (base_uri + relative_uri).to_s
+      rescue Addressable::URI::InvalidURIError => error
+        raise InvalidURIError, error
       end
 
       def endpoint_from_body

lib/webmention/endpoint/error.rb

@@ -0,0 +1,15 @@
+module Webmention
+  module Endpoint
+    class Error < StandardError; end
+
+    class ArgumentError < Error; end
+
+    class ConnectionError < Error; end
+
+    class InvalidURIError < Error; end
+
+    class TimeoutError < Error; end
+
+    class TooManyRedirectsError < Error; end
+  end
+end

lib/webmention/endpoint/version.rb

@@ -1,5 +1,5 @@
 module Webmention
   module Endpoint
-    VERSION = '0.1.0'.freeze
+    VERSION = '0.2.0'.freeze
   end
 end

spec/lib/webmention/endpoint_spec.rb

@@ -8,7 +8,13 @@
       let(:message) { 'url must be a String (given NilClass)' }
 
       it 'raises an ArgumentError' do
-        expect { described_class.discover(nil) }.to raise_error(ArgumentError, message)
+        expect { described_class.discover(nil) }.to raise_error(Webmention::Endpoint::ArgumentError, message)
+      end
+    end
+
+    context 'when given an invalid URL' do
+      it 'raises an InvalidURIError' do
+        expect { described_class.discover('http:') }.to raise_error(Webmention::Endpoint::InvalidURIError)
       end
     end