fix CtString type

Author: Spencer Miller

coti/crypto_utils.py

@@ -5,7 +5,7 @@
 from cryptography.hazmat.primitives.asymmetric import padding
 from cryptography.hazmat.primitives.asymmetric import rsa
 from eth_keys import keys
-from .types import ItString, ItUint
+from .types import CtString, CtUint, ItString, ItUint
 
 block_size = AES.block_size
 address_size = 20
@@ -152,7 +152,7 @@ def build_string_input_text(plaintext: int, user_aes_key: str, sender_address: s
     return input_text
 
 
-def decrypt_uint(ciphertext: int, user_aes_key: str) -> int:
+def decrypt_uint(ciphertext: CtUint, user_aes_key: str) -> int:
     # Convert ct to bytes (big-endian)
     byte_array = ciphertext.to_bytes(32, byteorder='big')
 
@@ -169,7 +169,7 @@ def decrypt_uint(ciphertext: int, user_aes_key: str) -> int:
     return decrypted_uint
 
 
-def decrypt_string(ciphertext: int, user_aes_key: str) -> str:
+def decrypt_string(ciphertext: CtString, user_aes_key: str) -> str:
     if 'value' in ciphertext or hasattr(ciphertext, 'value'): # format when reading ciphertext from an event
         __ciphertext = ciphertext['value']
     elif isinstance(ciphertext, tuple): # format when reading ciphertext from state variable

coti/types.py

@@ -4,8 +4,7 @@
 
 CtUint: TypeAlias = int
 
-class CtString(TypedDict):
-    value: List[int]
+CtString: TypeAlias = tuple[List[int]]
 
 class ItBool(TypedDict):
     ciphertext: int
@@ -15,6 +14,9 @@ class ItUint(TypedDict):
     ciphertext: int
     signature: bytes
 
-class ItString(TypedDict):
-    ciphertext: CtString
+class ItStringCiphertext(TypedDict):
+    value: List[int]
+
+class ItString(TypedDict): 
+    ciphertext: ItStringCiphertext
     signature: List[bytes]