Merge pull request #12 from 8ball030/feat/create_subaccount

feat: add-create-subaccount

Author: joshpwrk

.github/workflows/unit-test.yml

@@ -1,6 +1,8 @@
 name: Unit-test
 
-on: push
+on: 
+  push:
+  workflow_dispatch:
 jobs:
   build:
     name: Unit-test 🎯

derive_action_signing/module_data/create_subaccount.py

@@ -0,0 +1,37 @@
+from dataclasses import dataclass
+
+from eth_abi.abi import encode
+from .module_data import ModuleData
+from ..utils import decimal_to_big_int
+from web3 import Web3
+
+
+@dataclass
+class CreateSubAccountDetails:
+    amount: int
+    base_asset_address: str
+    sub_asset_address: str
+
+    def to_eth_tx_params(self):
+        return (
+            decimal_to_big_int(self.amount),
+            Web3.to_checksum_address(self.base_asset_address),
+            Web3.to_checksum_address(self.sub_asset_address),
+        )
+
+
+@dataclass
+class CreateSubAccountData(ModuleData):
+    amount: int
+    asset_name: str
+    margin_type: str
+    create_account_details: CreateSubAccountDetails
+
+    def to_abi_encoded(self):
+        return encode(
+            ['uint256', 'address', 'address'],
+            self.create_account_details.to_eth_tx_params(),
+        )
+
+    def to_json(self):
+        return {}

derive_action_signing/module_data/module_data.py

@@ -1,10 +1,20 @@
+"""
+Base class for all module data classes
+"""
+from abc import ABC, abstractmethod
 from dataclasses import dataclass
 
 
 @dataclass
-class ModuleData:
+class ModuleData(ABC):
+    """Abstract Base class for all module data classes"""
+
+    @abstractmethod
     def to_abi_encoded(self):
-        pass
+        """Return the data in ABI encoded format"""
+        raise NotImplementedError
 
+    @abstractmethod
     def to_json(self):
-        pass
+        """Return the data in JSON format"""
+        raise NotImplementedError