0
@@ -374,6 +374,29 @@ module Sinatra
0
     end
0
     alias :header :headers        
0
 
0
+    # Set the content type of the response body (HTTP 'Content-Type' header).
0
+    #
0
+    # The +type+ argument may be an internet media type (e.g., 'text/html',
0
+    # 'application/xml+atom', 'image/png') or a Symbol key into the
0
+    # Rack::File::MIME_TYPES table.
0
+    #
0
+    # Media type parameters, such as "charset", may also be specified using the
0
+    # optional hash argument:
0
+    #
0
+    #   get '/foo.html' do
0
+    #     content_type 'text/html', :charset => 'utf-8'
0
+    #     "<h1>Hello World</h1>"
0
+    #   end
0
+    #
0
+    def content_type(type, params={})
0
+      type = options[:mime_types][type.to_s] if type.kind_of?(Symbol)
0
+      fail "Invalid or undefined media_type: #{type}" if type.nil?
0
+      if params.any?
0
+        params = params.collect { |kv| "%s=%s" % kv }.join(', ')
0
+        type = [ type, params ].join(";")
0
+      end
0
+      response.header['Content-Type'] = type
0
+    end
0
   end
0
 
0
   ##
0
@@ -466,6 +489,7 @@ module Sinatra
0
         :public => root + '/public',
0
         :sessions => false,
0
         :logging => true,
0
+        :mime_types => Rack::File::MIME_TYPES.dup,
0
         :app_file => $0,
0
         :error_logging => true,
0
         :raise_errors => false,
0
@@ -768,6 +792,11 @@ module Sinatra
0
         opts.each { |key| set(key, false) }
0
       end
0
 
0
+      # TODO: doc
0
+      def mime(ext, type)
0
+        options.mime_types[ext.to_s] = type
0
+      end
0
+
0
       # Determine whether the application is in the process of being
0
       # reloaded.
0
       def reloading?
0
@@ -786,7 +815,7 @@ module Sinatra
0
   module DelegatingDSL
0
     
0
     FORWARDABLE_METHODS = [ :get, :post, :put, :delete, :head, :error,
0
-                            :set, :set_option, :enable, :disable,
0
+                            :set, :set_option, :enable, :disable, :mime,
0
                             :configure, :configures, :use, :helpers ]
0
     
0
     FORWARDABLE_METHODS.each do |method|

0
@@ -111,4 +111,60 @@ context "A Sinatra::Application when falling" do
0
   
0
 end
0
 
0
+context "A Sinatra::Application with custom Content-Type" do
0
+  before do
0
+    @app = Sinatra::Application.new do
0
+      mime :nwa, 'rap/westcoast'
0
+
0
+      get '/shortcut' do
0
+        content_type :xml
0
+        '<xml />'
0
+      end
0
 
0
+      get '/full' do
0
+        content_type 'text/foo'
0
+        'hey'
0
+      end
0
+
0
+      get '/with_params' do
0
+        content_type 'foo/bar', :bla => 'a', :nas => 'awesome'
0
+        "shoot 'em up"
0
+      end
0
+
0
+      get '/a_classic' do
0
+        content_type :nwa
0
+        "one less bitch"
0
+      end
0
+    end
0
+
0
+    @request = Rack::MockRequest.new(@app)
0
+  end
0
+
0
+  specify "Content-Type should be easily settable using shortcut" do
0
+    @response = @request.get('/shortcut')
0
+    @response.should.be.ok
0
+    @response.headers['Content-Type'].should.equal 'text/xml'
0
+    @response.body.should.equal '<xml />'
0
+  end
0
+
0
+  specify "Content-Type should be settable using full mime-type" do
0
+    @response = @request.get('/full')
0
+    @response.should.be.ok
0
+    @response.headers['Content-Type'].should.equal 'text/foo'
0
+    @response.body.should.equal 'hey'
0
+  end
0
+
0
+  specify "#content_type should append any other param" do
0
+    @response = @request.get('/with_params')
0
+    @response.should.be.ok
0
+    @response.headers['Content-Type'].should.equal 'foo/bar;bla=a, nas=awesome'
0
+    @response.body.should.equal "shoot 'em up"
0
+  end
0
+
0
+  specify "can create its own shortcut" do
0
+    @response = @request.get('/a_classic')
0
+    @response.should.be.ok
0
+    @response.headers['Content-Type'].should.equal 'rap/westcoast'
0
+    @response.body.should.equal "one less bitch"
0
+  end
0
+end