From 3ab46c3160dbecaddb3f50b874cb891c67146b49 Mon Sep 17 00:00:00 2001
From: Mike Chlipala <mchlipala@gmail.com>
Date: Fri, 13 Jul 2012 14:04:32 -0400
Subject: [PATCH] simple rake tasks to migrate up and down

---
 Rakefile | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/Rakefile b/Rakefile
index 2a610fd..d213565 100644
--- a/Rakefile
+++ b/Rakefile
@@ -4,6 +4,7 @@ require 'rubygems'
 require 'bundler'
 Bundler.setup
 require 'wd_sinatra/app_loader'
+require 'active_record'
 
 root = File.expand_path(File.dirname(__FILE__))
 
@@ -33,8 +34,7 @@ end
 
 namespace :db do
   task :environment do
-    require 'active_record'
-    ActiveRecord::Base.establish_connection adapter: 'sqlite3', database: 'db/development.sqlite3'
+    ActiveRecord::Base.establish_connection adapter: "sqlite3", database: "db/#{ENV['RACK_ENV']}.sqlite3"
   end
 
   desc "migrate"
@@ -43,4 +43,10 @@ namespace :db do
     ActiveRecord::Migration.verbose = true
     ActiveRecord::Migrator.migrate("db/migrate")
   end
+
+  desc "rolls back the migration (use steps with STEP=n)"
+  task(rollback: :environment) do
+    step = ENV["STEP"] ? ENV["STEP"].to_i : 1
+    ActiveRecord::Migrator.rollback('db/migrate', step)
+  end
 end
\ No newline at end of file