snapshots 0.0.1 → 0.0.2

data/VERSION

@@ -1 +1 @@
-0.0.1
+0.0.2

data/lib/snapshots/app.rb

@@ -7,9 +7,14 @@ module Snapshots
       :banner => 'rails_root',
       :group  => 'Global'
     
+    class_option :environment, :default => (ENV['RAILS_ENV'] || 'development'), :type => :string,
+      :desc   => 'Rails environment',
+      :banner => 'env',
+      :group  => 'Global'
+    
     desc 'list', 'List all snapshots for a rails app.'
     def list
-      goto_rails(self.options.app) do
+      goto_rails do
         
         Snapshots.find.each do |snapshot|
           snapshot =~ /snapshot_(\d+)\.rb$/
@@ -22,7 +27,7 @@ module Snapshots
     
     desc 'dump', 'Make a new snapshot.'
     def dump
-      goto_rails(self.options.app) do
+      goto_rails do
         load_environment!
         
         begin
@@ -38,7 +43,7 @@ module Snapshots
     
     desc 'load VERSION', 'Restore a snapshot.'
     def load(version)
-      goto_rails(self.options.app) do
+      goto_rails do
         load_environment!
         
         begin
@@ -54,7 +59,7 @@ module Snapshots
     
     desc 'install', 'Install rake tasks in rails app.'
     def install
-      goto_rails(self.options.app) do
+      goto_rails do
         File.open('lib/tasks/snapshots.rake', 'w+') do |f|
           f.write <<-EOR
 begin
@@ -72,11 +77,16 @@ EOR
   private
     
     def load_environment!
+      Object.const_set('RAILS_ROOT', rails_root)  rescue nil
+      ENV['RAILS_ENV'] = environment
       $rails_rake_task = true
       require(File.join(RAILS_ROOT, 'config', 'environment'))
     end
     
-    def goto_rails(root_path, &proc)
+    def rails_root
+      return @rails_root if @rails_root
+      
+      root_path = self.options.app
       root_path = File.expand_path(root_path)
       
       unless File.directory?(root_path) and File.file?(File.join(root_path, 'config', 'environment.rb'))
@@ -84,8 +94,15 @@ EOR
         exit(1)
       end
       
-      Object.const_set('RAILS_ROOT', root_path)
-      Dir.chdir(root_path, &proc)
+      @rails_root = root_path
+    end
+    
+    def environment
+      self.options.environment
+    end
+    
+    def goto_rails(&proc)
+      Dir.chdir(rails_root, &proc)
     end
     
   end

data/snapshots.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{snapshots}
-  s.version = "0.0.1"
+  s.version = "0.0.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Simon Menke"]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: snapshots
 version: !ruby/object:Gem::Version 
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors: 
 - Simon Menke