bin/ronin-sql in ronin-sql-0.2.4 vs bin/ronin-sql in ronin-sql-1.0.0

@@ -1,14 +1,27 @@
 #!/usr/bin/env ruby
 
 require 'rubygems'
 
-lib_dir = File.expand_path(File.join(File.dirname(__FILE__),'..','lib'))
-unless $LOAD_PATH.include?(lib_dir)
-  $LOAD_PATH << lib_dir
+root_dir = File.expand_path(File.join(File.dirname(__FILE__),'..'))
+if File.directory?(File.join(root_dir,'.git'))
+  Dir.chdir(root_dir) do |path|
+    require 'bundler'
+
+    begin
+      Bundler.setup(:default)
+    rescue Bundler::BundlerError => e
+      STDERR.puts e.message
+      STDERR.puts "Run `bundle install` to install missing gems"
+      exit e.status_code
+    end
+  end
 end
 
-require 'ronin/ui/command_line/commands/console'
+lib_dir = File.join(root_dir,'lib')
+$LOAD_PATH << lib_dir unless $LOAD_PATH.include?(lib_dir)
+
 require 'ronin/ui/console'
+require 'ronin/ui/command_line/commands/console'
 
 Ronin::UI::Console.auto_load << 'ronin/sql'
-Ronin::UI::CommandLine::Commands::Console.start
+Ronin::UI::CLI::Commands::Console.start