From 3227be1bd9a291f35a2ab5744e40a8b494fe6a22 Mon Sep 17 00:00:00 2001
From: Tom Cowell <tc@eightsquarestudio.com>
Date: Wed, 7 Dec 2011 18:16:10 +0000
Subject: [PATCH] Update lib/sinatra/activerecord/rake.rb

---
 lib/sinatra/activerecord/rake.rb | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/lib/sinatra/activerecord/rake.rb b/lib/sinatra/activerecord/rake.rb
index 37ff5d6..f96b626 100644
--- a/lib/sinatra/activerecord/rake.rb
+++ b/lib/sinatra/activerecord/rake.rb
@@ -9,6 +9,12 @@
       ENV["VERSION"] ? ENV["VERSION"].to_i : nil
     )
   end
+  
+  desc "rollback your DB use STEPS to specify more than 1"
+  task :rollback do
+    steps = ENV['STEPS'] || 1
+    ActiveRecord::Migrator.rollback('db/migrate',steps)
+  end
 
   desc "create an ActiveRecord migration in ./db/migrate"
   task :create_migration do