pyssb/ssb/util.py

"""Utility functions"""

from base64 import b64decode, b64encode
import os
from typing import TypedDict

from nacl.signing import SigningKey, VerifyKey
import yaml


class SSBSecret(TypedDict):
    """Dictionary type to hold an SSB secret identity"""

    keypair: SigningKey
    id: str


class ConfigException(Exception):
    """Exception to raise if there is a problem with the configuration data"""


def tag(key: VerifyKey) -> bytes:
    """Create tag from public key."""

    return b"@" + b64encode(bytes(key)) + b".ed25519"


def load_ssb_secret() -> SSBSecret:
    """Load SSB keys from ~/.ssb"""

    with open(os.path.expanduser("~/.ssb/secret"), encoding="utf-8") as f:
        config = yaml.load(f, Loader=yaml.SafeLoader)

    if config["curve"] != "ed25519":
        raise ConfigException("Algorithm not known: " + config["curve"])

    server_prv_key = b64decode(config["private"][:-8])

    return {"keypair": SigningKey(server_prv_key[:32]), "id": config["id"]}