module Efi

# File lib/efi.rb, line 5
def self.device_info(disk)
  print '.'
  Plist::parse_xml(`diskutil info -plist #{disk}`)
end

# File lib/efi.rb, line 10
def self.efi_partitions(plist)
  parts = []
  plist['AllDisksAndPartitions'].select do |d|
     d['Partitions']
  end.each do |d|
    d['Partitions'].each do |p|
      next if not p['Content'] == 'EFI'
      p['DiskInfo'] = device_info(d['DeviceIdentifier'])
      p['PartitionInfo'] = device_info(p['DeviceIdentifier'])
      parts << p
    end
  end
  parts
end

# File lib/efi.rb, line 25
def self.run
  puts
  puts "Available EFI partitions:"
  print "   loading"
  ep = efi_partitions Plist::parse_xml(`diskutil list -plist`)
  print "\r"
  ep.each_with_index do |p, i|
    if p['PartitionInfo']['MountPoint'] != ""
      puts "   #{i+1}) #{p['DeviceIdentifier']} (#{p['DiskInfo']['MediaName']}) mounted on #{p['PartitionInfo']['MountPoint']}"
    else
      puts "   #{i+1}) #{p['DeviceIdentifier']} (#{p['DiskInfo']['MediaName']})"
    end
  end
  puts "   Ctrl-C) exit"
  puts
  print "Select option: "
  
  n = nil
  begin
    n = gets.to_i
  rescue SignalException => e
    puts "\rSelect option: exit"
    exit
  rescue Exception => e
    puts e
    retry
  end
  
  partition_info = ep[n.abs - 1]['PartitionInfo']
  
  if partition_info['MountPoint'].empty?
    print "Mounting partition #{partition_info['DeviceIdentifier']}.."
    `diskutil mount #{partition_info['DeviceIdentifier']}`
    partition_info = device_info ep[n.abs - 1]['DeviceIdentifier']
    puts
  end
  
  if n < 0
    puts "Unmounting #{partition_info['DeviceIdentifier']}..."
    `diskutil unmount #{partition_info['DeviceIdentifier']}`
  else
    puts "Opening #{partition_info['MountPoint']} in Finder..."
    `open '#{partition_info['MountPoint']}'`
  end
  
  puts "Done"
end