tol 0.0.2 → 0.0.3

data/lib/tol/database.rb

@@ -33,17 +33,37 @@ class Database
   def download(heroku_app)
     puts "Downloading database for #{heroku_app.underline}".foreground(:green)
 
-    puts "1. Detecting local database settings.".foreground(:yellow)
+    puts "Step 1. Detecting local database settings.".foreground(:yellow)
     @settings = Tol::RailsApp.new.database_settings["development"]
 
-    puts "2. Capturing database on Heroku. Please wait.".foreground(:yellow)
-    db = `heroku pgbackups:capture --app #{heroku_app} --expire`
-    
-    puts "3. Downloading. Please wait.".foreground(:yellow)
-    url = `heroku pgbackups:url --app #{heroku_app}`
-    download = `curl -o /tmp/#{heroku_app}.dump '#{url}' > /dev/null 2>&1`
+    choose do |menu|
+      puts "Step 2. Which version of the database should I download?".foreground(:green)
+      
+      menu.prompt = "Please pick up database version?"
+      
+      local_file = "/tmp/#{heroku_app}.dump"
+      if File.exists?(local_file)
+        downloaded_at = File.mtime(local_file).strftime("%b %e, %l:%M %p")
+        menu.choice "Local File (Fastest) - downloaded at #{downloaded_at}"
+      end
+
+      menu.choice "Most Recent Snapshot (Fast)" do
+        puts "... Downloading. Please wait.".foreground(:yellow)
+        url = `heroku pgbackups:url --app #{heroku_app}`
+        download = `curl -o /tmp/#{heroku_app}.dump '#{url}' > /dev/null 2>&1`
+      end
+
+      menu.choice "New Snapshot (Slowest)" do
+        puts "... Capturing database on Heroku. Please wait".foreground(:yellow)
+        db = `heroku pgbackups:capture --app #{heroku_app} --expire`
 
-    puts "4. Importing Database.".foreground(:yellow)
+        puts "... Downloading. Please wait.".foreground(:yellow)
+        url = `heroku pgbackups:url --app #{heroku_app}`
+        download = `curl -o /tmp/#{heroku_app}.dump '#{url}' > /dev/null 2>&1`
+      end
+    end
+    
+    puts "Step 3. Importing Database.".foreground(:yellow)
     puts "-> drop old database"
     dropdb           = "dropdb"
     dropdb          += " -h #{@settings['host']}"      if @settings["host"]
@@ -72,9 +92,6 @@ class Database
     restore_command += " /tmp/#{heroku_app}.dump > /dev/null 2>&1"
     restore          = `/bin/bash -c '#{restore_command}'`
 
-    puts "5. Cleaning up.".foreground(:yellow)
-    clean_up = `rm /tmp/#{heroku_app}.dump`
-
     puts "DONE".foreground(:green)
   end
 

data/lib/tol/version.rb

@@ -1,3 +1,3 @@
 module Tol
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tol
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-27 00:00:00.000000000 Z
+date: 2013-06-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pry