shared key to string

.gitignore

@@ -1 +1,2 @@
 */_version.py
+__pycache__

unii/unii_connection.py

@@ -298,7 +298,7 @@ async def send(
             if self._writer is not None:
                 try:
                     with self._writer_lock:
-                        self._writer.write(self._shared_key.encode('utf-8'))
+                        self._writer.write(message.to_bytes(self._shared_key))
                         await self._writer.drain()
                         self.last_message_sent = datetime.now()
                         # logger.debug("Last message sent: %s", self.last_message_sent)

unii/unii_message.py

@@ -250,7 +250,7 @@ def to_bytes(self, shared_key: str | None = None) -> bytes:
             # logger.debug("Initial value: 0x%s", initial_value.hex())
 
             aes = AES.new(
-                shared_key, AES.MODE_CTR, initial_value=initial_value, nonce=b""
+                shared_key.encode(), AES.MODE_CTR, initial_value=initial_value, nonce=b""
             )
             payload = aes.encrypt(payload)
             # logger.debug("Encrypted Payload: 0x%s", payload.hex())
@@ -405,6 +405,6 @@ def __init__(self, message: bytes, shared_key: str | None = None):
         self.data = data
 
     def _decrypt(self, shared_key: str, initial_value: bytes, payload: bytes):
-        aes = AES.new(shared_key, AES.MODE_CTR, initial_value=initial_value, nonce=b"")
+        aes = AES.new(shared_key.encode(), AES.MODE_CTR, initial_value=initial_value, nonce=b"")
 
         return aes.decrypt(payload)