Check if shield exsists

HazAT · Mar 15, 2017 · da744f2 · da744f2
1 parent 5c09662
commit da744f2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/lib/badge/base.rb b/lib/badge/base.rb
@@ -1,6 +1,6 @@
 module Badge
 
-	VERSION = "0.8.2"
+	VERSION = "0.8.3"
 	DESCRIPTION = "Add a badge overlay to your app icon"
 
 	def self.root

diff --git a/lib/badge/runner.rb b/lib/badge/runner.rb
@@ -77,8 +77,11 @@ def run(path, options)
           UI.message "Did nothing... Enable --verbose for more info.".red
         end
 
-        File.delete(shield) if File.exist?(shield)
-        File.delete("#{shield.path}.png") if File.exist?("#{shield.path}.png")
+        if shield
+          File.delete(shield) if File.exist?(shield)
+          File.delete("#{shield.path}.png") if File.exist?("#{shield.path}.png")
+        end
+
       else
         UI.error "Could not find any app icons...".red
       end