mm-eth 0.5.12__py3-none-any.whl → 0.5.15__py3-none-any.whl

mm_eth/rpc_async.py

@@ -0,0 +1,170 @@
+import json
+import string
+from collections.abc import Sequence
+from typing import Any
+
+import ens.utils
+import eth_utils
+import websockets
+from mm_std import DataResult, http_request
+
+DEFAULT_TIMEOUT = 7.0
+
+
+async def rpc_call(
+    node: str,
+    method: str,
+    params: Sequence[object],
+    timeout: float,
+    proxy: str | None,
+    id_: int = 1,
+) -> DataResult[Any]:
+    data = {"jsonrpc": "2.0", "method": method, "params": params, "id": id_}
+    if node.startswith("http"):
+        return await _http_call(node, data, timeout, proxy)
+    return await _ws_call(node, data, timeout)
+
+
+async def _http_call(node: str, data: dict[str, object], timeout: float, proxy: str | None) -> DataResult[Any]:
+    res = await http_request(node, method="POST", proxy=proxy, timeout=timeout, json=data)
+    if res.is_error():
+        return res.to_data_result_err()
+    try:
+        parsed_body = res.parse_json_body()
+        err = parsed_body.get("error", {}).get("message", "")
+        if err:
+            return res.to_data_result_err(f"service_error: {err}")
+        if "result" in parsed_body:
+            return res.to_data_result_ok(parsed_body["result"])
+        return res.to_data_result_err("unknown_response")
+    except Exception as err:
+        return res.to_data_result_err(f"exception: {err}")
+
+
+async def _ws_call(node: str, data: dict[str, object], timeout: float) -> DataResult[Any]:
+    try:
+        async with websockets.connect(node, timeout=timeout) as ws:
+            await ws.send(json.dumps(data))
+            response = json.loads(await ws.recv())
+
+        err = response.get("error", {}).get("message", "")
+        if err:
+            return DataResult(err=f"service_error: {err}", data=response)
+        if "result" in response:
+            return DataResult(ok=response["result"], data=response)
+        return DataResult(err="unknown_response", data=response)
+    except TimeoutError:
+        return DataResult(err="timeout")
+    except Exception as err:
+        return DataResult(err=f"exception: {err}")
+
+
+async def eth_block_number(node: str, timeout: float = DEFAULT_TIMEOUT, proxy: str | None = None) -> DataResult[int]:
+    return (await rpc_call(node, "eth_blockNumber", [], timeout, proxy)).map(_hex_str_to_int)
+
+
+async def eth_get_balance(node: str, address: str, timeout: float = DEFAULT_TIMEOUT, proxy: str | None = None) -> DataResult[int]:
+    return (await rpc_call(node, "eth_getBalance", [address, "latest"], timeout, proxy)).map(_hex_str_to_int)
+
+
+async def erc20_balance(
+    node: str, token_address: str, user_address: str, timeout: float = DEFAULT_TIMEOUT, proxy: str | None = None
+) -> DataResult[int]:
+    data = "0x70a08231000000000000000000000000" + user_address[2:]
+    params = [{"to": token_address, "data": data}, "latest"]
+    return (await rpc_call(node, "eth_call", params, timeout, proxy)).map(_hex_str_to_int)
+
+
+async def erc20_name(
+    node: str, token_address: str, timeout: float = DEFAULT_TIMEOUT, proxy: str | None = None
+) -> DataResult[str]:
+    params = [{"to": token_address, "data": "0x06fdde03"}, "latest"]
+    return (await rpc_call(node, "eth_call", params, timeout, proxy)).map(_normalize_str)
+
+
+async def erc20_symbol(
+    node: str, token_address: str, timeout: float = DEFAULT_TIMEOUT, proxy: str | None = None
+) -> DataResult[str]:
+    params = [{"to": token_address, "data": "0x95d89b41"}, "latest"]
+    return (await rpc_call(node, "eth_call", params, timeout, proxy)).map(_normalize_str)
+
+
+async def erc20_decimals(
+    node: str, token_address: str, timeout: float = DEFAULT_TIMEOUT, proxy: str | None = None
+) -> DataResult[int]:
+    params = [{"to": token_address, "data": "0x313ce567"}, "latest"]
+    res = await rpc_call(node, "eth_call", params, timeout, proxy)
+    if res.is_err():
+        return res
+    try:
+        if res.unwrap() == "0x":
+            return DataResult(err="no_decimals", data=res.data)
+        value = res.unwrap()
+        result = eth_utils.to_int(hexstr=value[0:66]) if len(value) > 66 else eth_utils.to_int(hexstr=value)
+        return DataResult(ok=result, data=res.data)
+    except Exception as e:
+        return DataResult(err=f"exception: {e}", data=res.data)
+
+
+ENS_REGISTRY_ADDRESS: str = "0x00000000000C2E074eC69A0dFb2997BA6C7d2e1e"
+FUNC_SELECTOR_RESOLVER: str = "0x0178b8bf"  # resolver(bytes32)
+FUNC_SELECTOR_NAME: str = "0x691f3431"  # name(bytes32)
+
+
+async def ens_name(node: str, address: str, timeout: float = DEFAULT_TIMEOUT, proxy: str | None = None) -> DataResult[str | None]:
+    checksum_addr = eth_utils.to_checksum_address(address)
+    reverse_name = checksum_addr.lower()[2:] + ".addr.reverse"
+    name_hash_hex = ens.utils.normal_name_to_hash(reverse_name).hex()
+
+    resolver_data = FUNC_SELECTOR_RESOLVER + name_hash_hex
+
+    resolver_params = [{"to": ENS_REGISTRY_ADDRESS, "data": resolver_data}, "latest"]
+
+    resolver_res = await rpc_call(node, method="eth_call", params=resolver_params, timeout=timeout, proxy=proxy)
+    if resolver_res.is_err():
+        return resolver_res
+
+    if resolver_res.is_ok() and len(resolver_res.unwrap()) != 66:
+        return DataResult(ok=None, data={"revolver_response": resolver_res.dict()})
+
+    resolver_address = eth_utils.to_checksum_address("0x" + resolver_res.unwrap()[-40:])
+
+    name_data: str = FUNC_SELECTOR_NAME + name_hash_hex
+    name_params = [{"to": resolver_address, "data": name_data}, "latest"]
+
+    name_res: DataResult[str] = await rpc_call(node, "eth_call", name_params, timeout=timeout, proxy=proxy)
+
+    if name_res.is_err():
+        return DataResult(
+            err=name_res.unwrap_err(), data={"resolver_response": resolver_res.dict(), "name_response": name_res.dict()}
+        )
+
+    if name_res.unwrap() == "0x":
+        return DataResult(
+            ok=None,
+            data={"resolver_response": resolver_res.dict(), "name_response": name_res.dict()},
+            ok_is_none=True,
+        )
+
+    try:
+        hex_data = name_res.unwrap()
+        length_hex = hex_data[66:130]
+        str_len = int(length_hex, 16) * 2
+        name_hex = hex_data[130 : 130 + str_len]
+        return DataResult(
+            ok=bytes.fromhex(name_hex).decode("utf-8"),
+            data={"resolver_response": resolver_res.dict(), "name_response": name_res.dict()},
+        )
+    except Exception as e:
+        return DataResult(
+            err="exception",
+            data={"resolver_response": resolver_res.dict(), "name_response": name_res.dict(), "exception": str(e)},
+        )
+
+
+def _hex_str_to_int(value: str) -> int:
+    return int(value, 16)
+
+
+def _normalize_str(value: str) -> str:
+    return "".join(filter(lambda x: x in string.printable, eth_utils.to_text(hexstr=value))).strip()

{mm_eth-0.5.12.dist-info → mm_eth-0.5.15.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: mm-eth
-Version: 0.5.12
+Version: 0.5.15
 Requires-Python: >=3.12
 Requires-Dist: aiohttp-socks~=0.10.1
-Requires-Dist: mm-crypto-utils>=0.2.10
-Requires-Dist: mm-std~=0.3.22
+Requires-Dist: mm-crypto-utils>=0.2.13
+Requires-Dist: mm-std~=0.3.29
 Requires-Dist: typer>=0.15.2
 Requires-Dist: web3~=7.10.0
 Requires-Dist: websocket-client~=1.8.0

{mm_eth-0.5.12.dist-info → mm_eth-0.5.15.dist-info}/RECORD

@@ -11,6 +11,7 @@ mm_eth/ethernodes.py,sha256=V4VVbC6Nr9jhwT7blxtLugXC5KfXqE8n-kP0VvGHbqo,3070
 mm_eth/json_encoder.py,sha256=S4oD-qfTVztMb4sRpY1puhBQwOBofTyQXWszmdXk4og,433
 mm_eth/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mm_eth/rpc.py,sha256=k0eHxo_Dp6G0fHQ_rD-QbwOJz5ngC6kxBjl5NEHnyw8,13832
+mm_eth/rpc_async.py,sha256=4bGMSygj_L782KSHtIpmPu9s3f-tJ3H5qzm0Q-V8op4,6678
 mm_eth/solc.py,sha256=dYRvT8PjZlLDZhNsc_-0790Eug_ZwU2G-iBfIdGj6wQ,1071
 mm_eth/tx.py,sha256=efSoMCoWkenbGdHo1_LX66_Edz1HvED5-J_i3wrHwMw,4051
 mm_eth/utils.py,sha256=FytG3U6h80mnUaKP8W2mPZ77EuOp4U7pVbPuoKI3wW4,8215
@@ -42,7 +43,7 @@ mm_eth/cli/cmd/wallet/private_key_cmd.py,sha256=Fv_2OLog1h32pIP7PJITwl_pHdy3BXva
 mm_eth/cli/examples/balances.toml,sha256=i_ALpiEcf8-0TFiUg1cgJhxxfHYeBl9x0b3tnUWjswU,421
 mm_eth/cli/examples/call_contract.toml,sha256=ZQWK-409V_vLIZ2bsRD5RCWPPzShPz2KJTTRQY4YaGw,248
 mm_eth/cli/examples/transfer.toml,sha256=8mWuphDquoSDJw-hb9VJqtonjmv3kJ5Ip8jJ4t5YnfM,1810
-mm_eth-0.5.12.dist-info/METADATA,sha256=Hj0xMr3BCxx3MxKT-e5C6cQMO9tHQ0XJhjFbA4teqDA,277
-mm_eth-0.5.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mm_eth-0.5.12.dist-info/entry_points.txt,sha256=aGhpsozl8NIrkuUcX5fSgURCcDhr3ShUdeTSIrJq4oc,46
-mm_eth-0.5.12.dist-info/RECORD,,
+mm_eth-0.5.15.dist-info/METADATA,sha256=jg5LHl1UakJEHjV9e6oY3zK2DeK3vBAjGxazDJdOfVg,277
+mm_eth-0.5.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mm_eth-0.5.15.dist-info/entry_points.txt,sha256=aGhpsozl8NIrkuUcX5fSgURCcDhr3ShUdeTSIrJq4oc,46
+mm_eth-0.5.15.dist-info/RECORD,,