Updated the Bundler auto-detection boilerplate code.

ronin-rb · Aug 15, 2010 · 3ecaac2 · 3ecaac2
1 parent 4df2dd7
commit 3ecaac2
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/bin/ronin-scanners b/bin/ronin-scanners
@@ -3,15 +3,17 @@
 require 'rubygems'
 
 root_dir = File.expand_path(File.join(File.dirname(__FILE__),'..'))
-if File.directory?(File.join(root_dir,'.bundle'))
-  require 'bundler'
+if File.directory?(File.join(root_dir,'.git'))
+  Dir.chdir(root_dir) do |path|
+    require 'bundler'
 
-  begin
-    Bundler.setup(:runtime)
-  rescue Bundler::BundlerError => e
-    STDERR.puts e.message
-    STDERR.puts "Run `bundle install` to install missing gems"
-    exit e.status_code
+    begin
+      Bundler.setup(:runtime)
+    rescue Bundler::BundlerError => e
+      STDERR.puts e.message
+      STDERR.puts "Run `bundle install` to install missing gems"
+      exit e.status_code
+    end
   end
 end