"""
!/usr/bin/env python

https://pypi.org/project/websockets/

@author Lachezar Todorov
"""

import asyncio
import encrypt_decrypt
from websockets import serve

class Bob():
    def __init__(self,privateKey):
        self.pk = privateKey

    async def decrypt(websocket, path):
        privateKey = "" #dummy variable to make code work, please change!
        async for message in websocket:
            message = await websocket.recv()
            decryptedMessage = encrypt_decrypt.encrypt_decrypt.decryption(message, privateKey)
            #await websocket.send(decryptedMessage)
            print(decryptedMessage)

    async def messageDecrypt(privateKey):
        #do something with privateKey to send it to decrypt
        
        async with serve(Bob.decrypt, "localhost", 8765):
            await asyncio.Future()  # run forever

    #asyncio.run(main())