Require grit inside begin-rescue and fail gracefully with error message

flowdock · Feb 10, 2012 · 48b3275 · 48b3275
1 parent 7de4430
commit 48b3275
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/flowdock/capistrano.rb b/lib/flowdock/capistrano.rb
@@ -1,5 +1,4 @@
 require 'flowdock'
-require 'grit'
 require 'digest/md5'
 require 'cgi'
 
@@ -22,8 +21,11 @@
     task :set_flowdock_api do
       set :rails_env, variables.include?(:stage) ? stage : ENV['RAILS_ENV']
       begin
+        require 'grit'
         set :repo, Grit::Repo.new(".")
         config = Grit::Config.new(repo)
+      rescue LoadError
+        puts "Flowdock: you need to have Grit gem installed: #{e.to_s}"
       rescue => e
         puts "Flowdock: error in fetching your git repository information: #{e.to_s}"
       end