lib/bundler/audit/cli.rb in mrjoy-bundler-audit-0.3.2 vs lib/bundler/audit/cli.rb in mrjoy-bundler-audit-0.3.3

@@ -18,16 +18,17 @@
 #
 
 require 'bundler/audit/scanner'
 require 'bundler/audit/version'
 
+require 'thor'
 require 'bundler'
 require 'bundler/vendored_thor'
 
 module Bundler
   module Audit
-    class CLI < Thor
+    class CLI < ::Thor
 
       default_task :check
       map '--version' => :version
 
       desc 'check', 'Checks the Gemfile.lock for insecure dependencies'
@@ -72,13 +73,13 @@
         puts "#{File.basename($0)} #{VERSION} (advisories: #{database.size})"
       end
 
       protected
 
-      def say(string="", color=nil)
+      def say(message="", color=nil)
         color = nil unless $stdout.tty?
-        super(string, color)
+        super(message.to_s, color)
       end
 
       def print_warning(message)
         say message, :yellow
       end
@@ -88,11 +89,11 @@
 
         say "Advisory: ", :red
         say advisory.id
 
         say "Criticality: ", :red
-        say *CRITICALITY_MAP[advisory.criticality]
+        say *(CRITICALITY_MAP[advisory.criticality] || "Unknown")
 
         say "URL: ", :red
         say advisory.url
 
         print_advisory_details advisory
@@ -131,15 +132,13 @@
           say "Solution: ", :red
           say "remove or disable this gem until a patch is available!", [:red, :bold]
         end
       end
 
-      CRITICALITY_MAP = Hash({
+      CRITICALITY_MAP = {
         :low    => ["Low"],
         :medium => ["Medium", :yellow],
         :high   => ["High", [:red, :bold]],
-      }) do |data, key|
-        "Unknown"
-      end
+      }
     end
   end
 end