Cheaper metal cascade

railties/lib/rails/rack/cascade.rb

@@ -0,0 +1,31 @@
+require 'active_support/ordered_hash'
+
+module Rails
+  module Rack
+    # Try a request on several apps; return the first non-404 response.
+    class Cascade
+      attr_reader :apps
+
+      def initialize(apps)
+        @apps = ActiveSupport::OrderedHash.new
+        apps.each { |app| add app }
+      end
+
+      def call(env)
+        @apps.keys.each do |app|
+          result = app.call(env)
+          return result unless result[0].to_i == 404
+        end
+        Metal::NotFoundResponse
+      end
+
+      def add(app)
+        @apps[app] = true
+      end
+
+      def include?(app)
+        @apps.include?(app)
+      end
+    end
+  end
+end

railties/lib/rails/rack/metal.rb

@@ -1,17 +1,27 @@
+require 'rails/rack/cascade'
+
 module Rails
   module Rack
-    class Metal
+    module Metal
-      def self.new(app)
+      NotFoundResponse = [404, {}, []].freeze
-        apps = Dir["#{Rails.root}/app/metal/*.rb"].map do |file|
+      NotFound = lambda { NotFoundResponse }
-          File.basename(file, '.rb').camelize.constantize
-        end
-        apps << app
-        ::Rack::Cascade.new(apps)
-      end
 
-      NotFound = lambda { |env|
+      class << self
-        [404, {"Content-Type" => "text/html"}, "Not Found"]
+        def new(app)
-      }
+          Cascade.new(builtins + [app])
+        end
+
+        def builtins
+          base = "#{Rails.root}/app/metal"
+          matcher = /\A#{Regexp.escape(base)}\/(.*)\.rb\Z/
+
+          Dir["#{base}/**/*.rb"].sort.map do |file|
+            file.sub!(matcher, '\1')
+            require file
+            file.classify.constantize
+          end
+        end
+      end
     end
   end
 end