Add domain specification

lib/better_errors/middleware.rb

@@ -29,13 +29,19 @@ class Middleware
     #
     # Set to `{ "127.0.0.1/8", "::1/128" }` by default.
     ALLOWED_IPS = Set.new
+    ALLOWED_DOMAINS = Set.new
 
     # Adds an address to the set of IP addresses allowed to access Better
     # Errors.
     def self.allow_ip!(addr)
       ALLOWED_IPS << IPAddr.new(addr)
     end
 
+    def self.allow_domain!(addr)
+      ALLOWED_DOMAINS << addr
+    end
+
+
     allow_ip! "127.0.0.0/8"
     allow_ip! "::1/128" rescue nil # windows ruby doesn't have ipv6 support
 
@@ -53,7 +59,7 @@ def initialize(app, handler = ErrorPage)
     # @param [Hash] env
     # @return [Array]
     def call(env)
-      if allow_ip? env
+      if allow_domain?(env) || allow_ip?(env)
         better_errors_call env
       else
         @app.call env
@@ -69,6 +75,11 @@ def allow_ip?(env)
       ALLOWED_IPS.any? { |subnet| subnet.include? ip }
     end
 
+    def allow_domain?(env)
+      request = Rack::Request.new(env)
+      ALLOWED_DOMAINS.include? request.host
+    end
+
     def better_errors_call(env)
       case env["PATH_INFO"]
       when %r{/__better_errors/(?<id>.+?)/(?<method>\w+)\z}