import os

import sys

import pgpy

from Crypto.Util.number import long_to_bytes

rootkey_passphrase = "test"

subkey_passphrase = "example subkey passphrase"

rootkey_ascii_armor = """\

(rootkey, _) = pgpy.PGPKey.from_blob(rootkey_ascii_armor)

assert rootkey.is_protected

assert rootkey.is_unlocked is False

try:

print('Load these raw key values to OnlyKey by using the OnlyKey App --> Advanced -> Add Private Key')

with rootkey.unlock(rootkey_passphrase):

# rootkey is now unlocked

assert rootkey.is_unlocked

print('rootkey is now unlocked')

print('rootkey type %s', rootkey._key._pkalg)

if 'RSA' in rootkey._key._pkalg._name_:

print('rootkey value:')

#Parse rsa pgp key

primary_keyp = long_to_bytes(rootkey._key.keymaterial.p)

primary_keyq = long_to_bytes(rootkey._key.keymaterial.q)

print(("".join(["%02x" % c for c in primary_keyp])) + ("".join(["%02x" % c for c in primary_keyq])))

print('rootkey size =', (len(primary_keyp)+len(primary_keyq))*8, 'bits')

print('subkey values:')

for subkey, value in rootkey._children.items():

print('subkey id', subkey)

sub_keyp = long_to_bytes(value._key.keymaterial.p)

sub_keyq = long_to_bytes(value._key.keymaterial.q)

print('subkey value')

print(("".join(["%02x" % c for c in sub_keyp])) + ("".join(["%02x" % c for c in sub_keyq])))

print('subkey size =', (len(primary_keyp)+len(primary_keyq))*8, 'bits')

else:

print('rootkey value:')

#Parse ed25519 pgp key

primary_key = long_to_bytes(rootkey._key.keymaterial.s)

print("".join(["%02x" % c for c in primary_key]))

print('subkey values:')

for subkey, value in rootkey._children.items():

print('subkey id', subkey)

sub_key = long_to_bytes(value._key.keymaterial.s)

print('subkey value')

print("".join(["%02x" % c for c in sub_key]))

except:

print('Unlocking root key failed, attempting rootless subkey unlock.')

try:

print('subkey key values:')

for subkey, value in rootkey._children.items():

assert value.is_protected

assert value.is_unlocked is False

with value.unlock(subkey_passphrase):

# subkey is now unlocked

assert value.is_unlocked

print('subkey is now unlocked')

print('subkey id', subkey)

if 'RSA' in value._key._pkalg._name_:

sub_keyp = long_to_bytes(value._key.keymaterial.p)

sub_keyq = long_to_bytes(value._key.keymaterial.q)

print('subkey value')

print(("".join(["%02x" % c for c in sub_keyp])) + ("".join(["%02x" % c for c in sub_keyq])))

else:

sub_key = long_to_bytes(value._key.keymaterial.s)

print('subkey value')

print("".join(["%02x" % c for c in sub_key]))

# subkey is no longer unlocked

assert value.is_unlocked is False

except:

print('Unlocking failed')

assert rootkey.is_unlocked is False