#!/usr/bin/env ruby
    require 'dbus'
    require 'awesome_print'
    require 'filesize'
    require 'colorize'
     
    loop = DBus::Main.new
    bus = DBus::SystemBus.instance
     
    udisk2   = bus['org.freedesktop.UDisks2']
    device = nil
     
    EXPECTED_INTERFACES = %w[org.freedesktop.UDisks2.PartitionTable org.freedesktop.UDisks2.Block]
     
    manager = udisk2['/org/freedesktop/UDisks2']['org.freedesktop.DBus.ObjectManager']
    manager.on_signal 'InterfacesAdded' do |path|
    	interfaces = udisk2[path].interfaces
    	next unless (EXPECTED_INTERFACES - interfaces).empty?
    	block = udisk2[path]['org.freedesktop.UDisks2.Block']
     
    	path = path.split('/').last
     
    	drive = block['Drive']
    	drive = udisk2[drive]['org.freedesktop.UDisks2.Drive']
    	id = drive['Id']
    	size = Filesize.new drive['Size']
     
    	device = { path: path, id: id, size: size }
     
    	loop.quit
    end
     
    iso = ARGV.first
    size = File.size iso
    puts "#{'Plug a drive'.colorize :red} to begin to write #{iso.colorize :green} [#{Filesize.new(size).pretty.colorize :yellow}]"
     
    loop << bus
    loop.run
     
    print "Erase #{device[:id].colorize :red} [#{device[:size].pretty.colorize :yellow}]? [y/N] "
    answer = STDIN.gets.chomp
     
    exit -1 unless answer.downcase == 'y'
    device = File.join '/dev/', device[:path]
    current = 0
    chunk = Filesize.from('4MiB').to_i
     
    File.open iso, 'rb' do |iso|
    	File.open device, 'wb' do |device|
    		while read = iso.read(chunk) do
    			current += read.size
    			device.write read
    			print "\r[%6.2f%%] %d/%d" % [current.to_f / size * 100, current, size]
    		end
    		puts
    		puts 'Finalizing...'.colorize :yellow
    	end
    end
    puts 'Done!'.colorize :green