useless-doc 0.1.0 → 0.1.1

data/lib/useless/doc/rack/application.rb

@@ -12,18 +12,32 @@ require 'useless/doc/rack/ui'
 module Useless
   module Doc
     module Rack
-      module Application
-        def self.call(env)
-          ::Rack::Builder.app do
-            use Low::Rack::RackErrors
-            use Low::Rack::RequestLogger
-            use ::Rack::CommonLogger
-            use Useless::Doc::Rack::UI
-            use Useless::Doc::Rack::Stylesheet
-            use Useless::Doc::Rack::Transform
+      class Application
+        def initialize(*subdomains)
+          @subdomains = subdomains
+        end
+
+        def call(env)
+          app.call(env)
+        end
+
+        private
+
+        def app
+          @app ||= begin
+            subdomains = @subdomains
+
+            ::Rack::Builder.app do
+              use Low::Rack::RackErrors
+              use Low::Rack::RequestLogger
+              use ::Rack::CommonLogger
+              use Useless::Doc::Rack::UI
+              use Useless::Doc::Rack::Stylesheet
+              use Useless::Doc::Rack::Transform
 
-            run Useless::Doc::Rack::Proxy.new
-          end.call(env)
+              run Useless::Doc::Rack::Proxy.new(subdomains)
+            end
+          end
         end
       end
     end

data/lib/useless/doc/rack/proxy.rb

@@ -16,10 +16,16 @@ module Useless
       # OPTIONS request to some-api.useless.io/some/resource.
       #
       # If there is no corresponding endpoint, the proxy will respond with a
-      # 404.
+      # 404. If the requested subdomain is not in the list of supported
+      # subdomains specified at initialization, the proxy will also respond
+      # with a 404.
       #
       class Proxy
 
+        def initialize(subdomains)
+          @subdomains = subdomains
+        end
+
         def self.transform_url(url)
           uri = URI(url)
           new_host = uri.host.gsub(/\.doc\./, '.')
@@ -28,20 +34,31 @@ module Useless
 
         def call(env)
           request = ::Rack::Request.new(env)
-          url = Proxy.transform_url(request.url)
 
-          if json = retrieve_resource(url)
-            [200, {'Content-Type' => 'application/json'}, [json]]
+          if valid_subdomain?(request.url)
+            url = Proxy.transform_url(request.url)
+
+            if json = retrieve_resource(url)
+              [200, {'Content-Type' => 'application/json'}, [json]]
+            else
+              [404, {'Content-Type' => 'text/plain'}, ['Documentation JSON is missing.']]
+            end
           else
-            [404, {'Content-Type' => 'text/plain'}, ['Documentation JSON is missing.']]
+            [404, {'Content-Type' => 'text/plain'}, ['Unknown subdomain.']]
           end
         end
 
+        private
+
         def retrieve_resource(url)
           response = Typhoeus.options url, headers: { 'Accept' => 'application/json' }
           response.response_body
         end
 
+        def valid_subdomain?(url)
+          @subdomains.include?(url[/((?:\w|\-)+)\.doc\./, 1])
+        end
+
         # +Proxy::Stub+ retrieves JSON from the spec/documents directory for
         # easy UI testing.
         #

data/lib/useless/doc/version.rb

@@ -1,5 +1,5 @@
 module Useless
   module Doc
-    VERSION = '0.1.0'
+    VERSION = '0.1.1'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: useless-doc
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
   prerelease: 
 platform: ruby
 authors:
@@ -151,7 +151,6 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
-- config.ru
 - lib/useless/doc.rb
 - lib/useless/doc/action.rb
 - lib/useless/doc/body.rb

data/config.ru

@@ -1,4 +0,0 @@
-$:.push File.dirname(__FILE__) + '/lib'
-require 'useless/doc/rack/application'
-
-run Useless::Doc::Rack::Application