#!/usr/bin/env ruby
require 'optparse'

require_relative '../../lib/rnp'

options = {armored: false}
parser = OptionParser.new do |opts|
  opts.banner = "Usage: #{$0} [options]"
  opts.on('-a', '--armored', 'Input file is ASCII armored') do
    options[:armored] = true
  end
  opts.on('-h', '--help', 'Print this help') do
    puts opts
    exit
  end
end

def print_pubkey(pubkey, indent=0)
    puts "#{' ' * indent}Version: #{pubkey[:version]}"
    puts "#{' ' * indent}Creation Time: #{Time.at(pubkey[:birthtime])}"
    puts "#{' ' * indent}Algorithm: #{pubkey[:alg]}"
    case pubkey[:alg]
    when :PGP_PKA_RSA
      n = LibRNP::bn2hex(pubkey[:key][:rsa][:n])
      e = LibRNP::bn2hex(pubkey[:key][:rsa][:e])
      puts "#{' ' * indent}n: 0x#{n}"
      puts "#{' ' * indent}e: 0x#{e}"
    end
end

def print_seckey(seckey, indent=0)
  puts "#{' ' * (indent+2)}[Public Key]"
  print_pubkey(seckey[:pubkey], indent + 2)
  puts "#{' ' * indent}string-to-key usage: #{seckey[:s2k_usage]}"
  puts "#{' ' * indent}string-to-key specifier: #{seckey[:s2k_specifier]}"
  puts "#{' ' * indent}Symmetric algorithm: #{seckey[:alg]}"
  puts "#{' ' * indent}Hash algorithm: #{seckey[:hash_alg]}"
end

parser.parse!
armored = options[:armored] ? 1 : 0

pgpio = LibRNP::PGPIO.new
stdout_fp = LibC::fdopen($stdout.to_i, 'w')
stderr_fp = LibC::fdopen($stderr.to_i, 'w')
pgpio[:outs] = stdout_fp
pgpio[:errs] = stderr_fp
pgpio[:res] = stdout_fp

mem_ptr = LibC::calloc(1, LibRNP::PGPMemory.size)
mem = LibRNP::PGPMemory.new(mem_ptr)

$stdin.binmode
data = $stdin.read
data_buf = FFI::MemoryPointer.new(:uint8, data.bytesize)
data_buf.put_bytes(0, data)
LibRNP::pgp_memory_add(mem, data_buf, data_buf.size)

keyring_mem = LibC::calloc(1, LibRNP::PGPKeyring.size)
keyring = LibRNP::PGPKeyring.new(keyring_mem)
if 1 != LibRNP::pgp_keyring_read_from_mem(pgpio, keyring, armored, mem)
  puts 'Failed to load keyring'
  exit 1
end
keycount = LibRNP::dynarray_count(keyring, 'key')
puts "Loaded #{keycount} key(s)"

(0..keycount - 1).each {|keyn|
  key = LibRNP::dynarray_get_item(keyring, 'key', LibRNP::PGPKey, keyn)
  puts "[Key ##{keyn}]"
  uidcount = LibRNP::dynarray_count(key, 'uid')
  print "User ids: "
  puts LibRNP::dynarray_get_item(key, 'uid', :string, 0)
  (1..uidcount - 1).each {|uidn|
    print '          '
    puts LibRNP::dynarray_get_item(key, 'uid', :string, uidn)
  }
  puts "Subpackets:  #{LibRNP::dynarray_count(key, 'packet')}"
  puts "Subsigs:     #{LibRNP::dynarray_count(key, 'subsig')}"
  puts "Revocations: #{LibRNP::dynarray_count(key, 'revoke')}"
  printf "Key Flags: 0x%02X\n", key[:key_flags]
  case key[:type]
  when :PGP_PTAG_CT_PUBLIC_KEY
    puts '  [Public Key]'
    pubkey = key[:key][:pubkey]
    print_pubkey(pubkey, 2)
  when :PGP_PTAG_CT_SECRET_KEY
    puts '  [Secret Key]'
    seckey = key[:key][:seckey]
    print_seckey(seckey, 2)
  end
  puts ''
}

LibRNP::pgp_memory_free(mem)