PySecretHandshake/examples/test_server.py

"""Example SHS server"""

import os
from asyncio import get_event_loop
from base64 import b64decode

import yaml
from nacl.signing import SigningKey

from secret_handshake import SHSServer

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


async def _on_connect(conn):
    async for msg in conn:
        print(msg)


async def main():
    """Main function to run"""

    server_keypair = SigningKey(b64decode(config["private"][:-8])[:32])
    server = SHSServer("localhost", 8008, server_keypair)
    server.on_connect(_on_connect)
    await server.listen()


if __name__ == "__main__":
    loop = get_event_loop()
    loop.run_until_complete(main())
    loop.run_forever()
    loop.close()