download.rb in bukkit-2.5.0

- old
+ new

@@ -1,6 +1,7 @@
 require 'rest-client'
+require 'fileutils'
 
 module Bukkit
 	class Server
 		# Download a file from a URI.
 		def self.download(uri, options = {}) # Options: { :filename => "filename.ext" }
@@ -9,10 +10,17 @@
 				filename = options[:filename]
 			else
 				filename = uri.split("\/").last
 			end
 
+			# Catch SIGINT if needed.
+			trap("SIGINT") {
+				puts "\nDownload failed.".red
+				FileUtils.rm(filename) if File.exists? filename
+				exit
+			}
+
 			# Give some friendly output.
 			puts "Downloading: ".yellow + filename
 			puts "       From: ".yellow + uri
 			puts "(This may take a while depending on your internet connection.)".light_yellow
 
@@ -21,9 +29,9 @@
 			File.open(filename, "wb") do |file|
 				file.write(data)
 			end
 			# => filename.ext
 
-			puts filename.light_green + " successfully downloaded!".green
+			puts filename.light_green + " successfully downloaded.".green
 		end
 	end
 end
\ No newline at end of file