diff --git a/lib/ronin/repos/cli/commands/install.rb b/lib/ronin/repos/cli/commands/install.rb index 9896ee5..ea39e40 100644 --- a/lib/ronin/repos/cli/commands/install.rb +++ b/lib/ronin/repos/cli/commands/install.rb @@ -63,7 +63,7 @@ class Install < Command def run(uri) log_info "Installing repository from #{uri} ..." cache_dir.install(uri) - rescue CommandFailed => error + rescue CommandNotInstalled, CommandFailed => error print_error(error.message) exit(-1) end diff --git a/lib/ronin/repos/cli/commands/update.rb b/lib/ronin/repos/cli/commands/update.rb index d86faaf..a8b1873 100644 --- a/lib/ronin/repos/cli/commands/update.rb +++ b/lib/ronin/repos/cli/commands/update.rb @@ -77,7 +77,7 @@ def run(name=nil) begin repo.update - rescue CommandFailed => error + rescue CommandNotInstalled, CommandFailed => error log_error("failed to update repository #{repo}: #{error.message}") end end