Download imgur albums all at once

imgur-grab.rb

#!/usr/bin/env ruby

require "net/http"

if ARGV.length == 0 then
	puts "Usage: #{__FILE__.split('/').last} album loc [albumN locN ...]"
	exit
end

while ARGV.length > 0
	album = ARGV.shift
	folder = ARGV.shift || "."
	album_url = (album =~ /http/) ? album : "http://imgur.com/a/#{album}"

	system("mkdir #{folder}") unless File.directory? folder

	list = Net::HTTP.get(URI.parse(album_url)).split("\n").select { |line|
		line.scan('<a href="/download/').length > 0
	}.map { |line|
		line.sub(/^[^"]*"/,'').sub(/".*$/,'')
	}
	list.each_with_index { |link, i|
		puts "#{link.split('/').last} (#{i+1}/#{list.length})"
		filename = link.split('/').last + ".jpeg"
		data = Net::HTTP.get(URI.parse("http://imgur.com"+link)) 
		File.open(folder+"/"+filename, "w") { |f|
			f.puts data
		}
	}
end