feedhandlers extended

modules/aster_auth.py

@@ -0,0 +1,108 @@
+import requests
+from dotenv import load_dotenv
+import os
+import time
+import threading
+import urllib
+
+from eth_account.messages import encode_typed_data
+from eth_account import Account
+
+load_dotenv()
+
+user = os.getenv("RABBY_WALLET")
+signer = os.getenv("ASTER_API_WALLET_ADDRESS")
+private_key = os.getenv("ASTER_API_PRIVATE_KEY")
+
+_last_ms = 0
+_i = 0
+
+def post_authenticated_url(req: dict) -> dict:
+    typed_data = {
+    "types": {
+        "EIP712Domain": [
+        {"name": "name", "type": "string"},
+        {"name": "version", "type": "string"},
+        {"name": "chainId", "type": "uint256"},
+        {"name": "verifyingContract", "type": "address"}
+        ],
+        "Message": [
+        { "name": "msg", "type": "string" }
+        ]
+    },
+    "primaryType": "Message",
+    "domain": {
+        "name": "AsterSignTransaction",
+        "version": "1",
+        "chainId": 1666,
+        "verifyingContract": "0x0000000000000000000000000000000000000000"
+    },
+    "message": {
+        "msg": "$msg"
+    }
+    }
+    headers = {
+        'Content-Type': 'application/x-www-form-urlencoded',
+        'User-Agent': 'PythonApp/1.0'
+    }
+    host = 'https://fapi.asterdex.com'
+
+
+    def get_nonce():
+        _nonce_lock = threading.Lock()
+        global _last_ms, _i
+        with _nonce_lock:
+            now_ms = int(time.time())
+
+            if now_ms == _last_ms:
+                _i += 1
+            else:
+                _last_ms = now_ms
+                _i = 0
+
+            return now_ms * 1_000_000 + _i
+
+    def sign_typed_data(data: dict, private_key: str):
+        """Sign EIP-712 typed data using encode_typed_data."""
+        message = encode_typed_data(
+            domain_data=data["domain"],
+            message_types={"Message": data["types"]["Message"]},
+            message_data=data["message"],
+        )
+        return Account.sign_message(message, private_key=private_key)
+
+    def send_by_url(req):
+        my_dict = req['params'].copy()
+        url = host + req['url']
+        method = req['method']
+
+        my_dict['nonce'] = str(get_nonce())
+        my_dict['user'] = user
+        my_dict['signer'] = signer
+
+        param = urllib.parse.urlencode(my_dict)
+
+        typed_data['message']['msg'] = param
+        signed = sign_typed_data(typed_data, private_key)
+
+        full_url = url + '?' + param + '&signature=' + signed.signature.hex()
+        # print(full_url)
+
+        if method == 'GET':
+            res = requests.get(full_url, headers=headers)
+            # print(res.status_code, res.text)
+            return res.json()
+        elif method == 'POST':
+            res = requests.post(full_url, headers=headers)
+            # print(res.status_code, res.text)
+            return res.json()
+        elif method == 'PUT':
+            res = requests.put(full_url, headers=headers)
+            # print(res.status_code, res.text)
+            return res.json()
+        elif method == 'DELETE':
+            res = requests.delete(full_url, headers=headers)
+            # print(res.status_code, res.text)
+            return res.json()
+
+    return send_by_url(req=req)