Refactor: add abstract method sign_raw in class Account

aleph-im · Jun 19, 2024 · 2b3b63f · 2b3b63f
1 parent fe84521
commit 2b3b63f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/aleph/sdk/client/vmclient.py b/src/aleph/sdk/client/vmclient.py
@@ -8,15 +8,15 @@
 from jwcrypto import jwk
 from jwcrypto.jwa import JWA
 
-from aleph.sdk.chains.ethereum import ETHAccount
+from aleph.sdk.types import Account
 from aleph.sdk.utils import to_0x_hex
 
 logger = logging.getLogger(__name__)
 
 
 class VmClient:
-    def __init__(self, account: ETHAccount, domain: str = ""):
-        self.account: ETHAccount = account
+    def __init__(self, account: Account, domain: str = ""):
+        self.account: Account = account
         self.ephemeral_key: jwk.JWK = jwk.JWK.generate(kty="EC", crv="P-256")
         self.domain: str = domain
         self.pubkey_payload = self._generate_pubkey_payload()

diff --git a/src/aleph/sdk/types.py b/src/aleph/sdk/types.py
@@ -20,6 +20,8 @@ class Account(Protocol):
     @abstractmethod
     async def sign_message(self, message: Dict) -> Dict: ...
 
+    @abstractmethod
+    async def sign_raw(self, buffer: bytes) -> bytes: ...
     @abstractmethod
     def get_address(self) -> str: ...