pyssb/ssb/util.py

30 lines
690 B
Python
Raw Normal View History

2017-08-05 10:24:46 +00:00
import os
import yaml
2017-08-05 16:21:14 +00:00
from base64 import b64decode, b64encode
2017-05-25 10:47:01 +00:00
2017-08-05 10:24:46 +00:00
from nacl.signing import SigningKey
2017-05-25 10:47:01 +00:00
2017-08-05 10:24:46 +00:00
class ConfigException(Exception):
pass
2017-05-25 10:47:01 +00:00
2017-08-05 16:21:14 +00:00
def tag(key):
"""Create tag from publick key."""
return b'@' + b64encode(bytes(key)) + b'.ed25519'
2017-08-05 10:24:46 +00:00
def load_ssb_secret():
"""Load SSB keys from ~/.ssb"""
with open(os.path.expanduser('~/.ssb/secret')) as f:
config = yaml.load(f, Loader=yaml.SafeLoader)
2017-05-25 10:47:01 +00:00
2017-08-05 10:24:46 +00:00
if config['curve'] != 'ed25519':
raise ConfigException('Algorithm not known: ' + config['curve'])
2017-05-25 10:47:01 +00:00
2017-08-05 10:24:46 +00:00
server_prv_key = b64decode(config['private'][:-8])
return {
'keypair': SigningKey(server_prv_key[:32]),
'id': config['id']
}