lib/dl/dl.rb in dl-0.1.0 vs lib/dl/dl.rb in dl-1.0.0

@@ -1,14 +1,18 @@
 module Dl
-  def self.download(url)
+  def self.download(url, raw = false)
     session = Patron::Session.new
 
     puts "Downloading #{url}..."
 
     response = session.get url
-    filename = response.url.split('/').pop
 
-    puts "Downloaded to file #{filename}!"
 
-    File.new(filename, 'w').write(response.body)
+    if raw == true
+      puts response.body
+    else
+      filename = response.url.split('/').pop
+      puts "Downloaded to file #{filename}!"
+      File.new(filename, 'w').write(response.body)
+    end
   end
 end