maintenance file in json

Gemfile.lock

@@ -1,14 +1,14 @@
 PATH
   remote: .
   specs:
-    rack-maintenance (0.3.0)
+    rack-maintenance (1.0.0)
       rack (>= 1.0)
 
 GEM
   remote: http://rubygems.org/
   specs:
     diff-lcs (1.1.3)
-    rack (1.4.1)
+    rack (1.5.2)
     rake (0.9.2.2)
     rspec (2.11.0)
       rspec-core (~> 2.11.0)

lib/rack/maintenance.rb

@@ -14,14 +14,18 @@ def initialize(app, options={})
   def call(env)
     if maintenance? && path_in_app(env)
       data = File.read(file)
-      [ 503, { 'Content-Type' => 'text/html', 'Content-Length' => data.length.to_s }, [data] ]
+      [ 503, { 'Content-Type' => content_type, 'Content-Length' => data.length.to_s }, [data] ]
     else
       app.call(env)
     end
   end
 
 private ######################################################################
 
+  def content_type
+    file.end_with?('json') ? 'application/json' : 'text/html'
+  end
+
   def environment
     options[:env]
   end

spec/rack-maintenance_spec.rb

@@ -1,9 +1,9 @@
 require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 require 'fileutils'
 
-describe "RackMaintenance" do
+shared_examples "RackMaintenance" do
   let(:app) { Class.new { def call(env); end }.new }
-  let(:rack) { Rack::Maintenance.new(app, :file => "spec/maintenance.html") }
+  let(:rack) { Rack::Maintenance.new(app, :file => file_name) }
 
   context "without a :file option" do
     it "raises an error" do
@@ -22,11 +22,11 @@
 
   context "with maintenance file" do
     before do
-      FileUtils.touch 'spec/maintenance.html'
+      FileUtils.touch file_name
     end
 
     after do
-      FileUtils.rm 'spec/maintenance.html'
+      FileUtils.rm file_name
     end
 
     it "does not call the app" do
@@ -35,11 +35,11 @@
     end
 
     it "returns the maintenance response" do
-      rack.call({}).should eq [503, {"Content-Type"=>"text/html", "Content-Length"=>"0"}, [""]]
+      rack.call({}).should eq [503, {"Content-Type"=>content_type, "Content-Length"=>"0"}, [""]]
     end
 
     context "and :env option MAINTENANCE" do
-      let(:rack) { Rack::Maintenance.new(app, :file => "spec/maintenance.html", :env => "MAINTENANCE") }
+      let(:rack) { Rack::Maintenance.new(app, :file => file_name, :env => "MAINTENANCE") }
 
       context "outside MAINTENANCE env" do
         it "calls the app" do
@@ -53,6 +53,10 @@
           ENV['MAINTENANCE'] = "true"
         end
 
+        after do
+          ENV.delete("MAINTENANCE")
+        end
+
         it "does not call the app" do
           app.should_not_receive :call
           rack.call({})
@@ -68,3 +72,18 @@
     end
   end
 end
+
+describe "RackMaintenance with json maintenance file" do
+  it_behaves_like "RackMaintenance" do
+    let(:file_name) { "spec/maintenance.json" }
+    let(:content_type) { "application/json" }
+  end
+end
+
+describe "RackMaintenance with html maintenance file" do
+  it_behaves_like "RackMaintenance" do
+    let(:file_name) { "spec/maintenance.html" }
+    let(:content_type) { "text/html" }
+  end
+end
+