Merge pull request #5 from lukewwww/main

Get latest block height

Author: lukewwww

README.md

@@ -1,5 +1,8 @@
 # Crynux MCP Server
 
+[![](https://dcbadge.limes.pink/api/server/https://discord.gg/zmU9GRwU6f)](https://discord.gg/zmU9GRwU6f)
+[![PyPI - Version](https://img.shields.io/pypi/v/crynux-mcp?style=for-the-badge)](https://pypi.org/project/crynux-mcp/)
+
 MCP server for Crynux Network operations, designed for LLM clients (such as Cursor, VS Code, and Claude Desktop) to perform most Crynux Network/Crynux Portal actions through standardized MCP tools.
 
 ## Features
@@ -11,6 +14,7 @@ MCP server for Crynux Network operations, designed for LLM clients (such as Curs
 ### Blockchain
 
 - Native CNX balance query on Crynux L2 networks
+- Latest block height query on Crynux L2 networks
 - Native CNX transfer
 - Beneficial address query and on-chain update
 - Node staking query

docs/tools.md

@@ -19,6 +19,7 @@ This document contains detailed input and output fields for each MCP tool.
 | Action | Description |
 |---|---|
 | [`get_balance`](#get_balance) | Query native CNX balance for an address. |
+| [`get_latest_block_number`](#get_latest_block_number) | Query latest on-chain block height for a network. |
 | [`transfer_native`](#transfer_native) | Send native CNX with local signer key. |
 | [`get_beneficial_address`](#get_beneficial_address) | Query node beneficial address. |
 | [`set_beneficial_address`](#set_beneficial_address) | Submit transaction to set beneficial address. |
@@ -56,6 +57,14 @@ Output fields:
 - `balance_wei`
 - `symbol` (`CNX`)
 
+## get_latest_block_number
+
+Inputs:
+- `network`: optional (`dymension` or `near`). Defaults to configured default network.
+
+Output fields:
+- `block_number`: latest finalized block number returned by RPC.
+
 ## transfer_native
 
 Inputs:

src/crynux_mcp/blockchain/evm_client.py

@@ -10,6 +10,7 @@
 from crynux_mcp.blockchain.schemas import (
     BalanceResult,
     BeneficialAddressResult,
+    LatestBlockNumberResult,
     NodeCreditsResult,
     NodeStakingInfoResult,
     SetBeneficialAddressResult,
@@ -85,6 +86,9 @@ def get_balance(self, address: str) -> BalanceResult:
             symbol=self.chain.native_currency.symbol,
         )
 
+    def get_latest_block_number(self) -> LatestBlockNumberResult:
+        return LatestBlockNumberResult(block_number=int(self.w3.eth.block_number))
+
     def transfer_native(
         self,
         private_key: str,

src/crynux_mcp/blockchain/schemas.py

@@ -14,6 +14,11 @@ class BalanceResult:
     symbol: str
 
 
+@dataclass(frozen=True)
+class LatestBlockNumberResult:
+    block_number: int
+
+
 @dataclass(frozen=True)
 class TransferResult:
     from_address: str

src/crynux_mcp/server.py

@@ -88,6 +88,15 @@ def handle_get_balance(
         raise _execution_error(exc, {"network": network, "address": address, "key_name": key_name}) from exc
 
 
+def handle_get_latest_block_number(network: str | None) -> dict[str, Any]:
+    try:
+        chain = registry.resolve(network)
+        client = EvmClient(chain)
+        return _to_response_payload(client.get_latest_block_number())
+    except Exception as exc:  # noqa: BLE001
+        raise _execution_error(exc, {"network": network}) from exc
+
+
 def handle_transfer_native(
     network: str | None,
     to: str,
@@ -550,6 +559,12 @@ def get_balance(
     return handle_get_balance(network=network, address=address, key_name=key_name)
 
 
+@mcp.tool()
+def get_latest_block_number(network: str | None) -> dict[str, Any]:
+    """Get latest block number on a selected Crynux EVM network."""
+    return handle_get_latest_block_number(network=network)
+
+
 @mcp.tool()
 def transfer_native(
     network: str | None,

tests/test_tools_contract.py

@@ -3,6 +3,7 @@
 from crynux_mcp.server import (
     handle_get_balance,
     handle_get_beneficial_address,
+    handle_get_latest_block_number,
     handle_get_node_credits,
     handle_get_node_staking_info,
     handle_set_beneficial_address,
@@ -38,6 +39,31 @@ def resolve(self, _network):
     assert "chain_id" not in payload
 
 
+def test_handle_get_latest_block_number_shape(monkeypatch) -> None:  # type: ignore[no-untyped-def]
+    @dataclass(frozen=True)
+    class FakeResult:
+        block_number: int = 123456
+
+    class FakeClient:
+        def __init__(self, _chain) -> None:
+            pass
+
+        def get_latest_block_number(self):
+            return FakeResult()
+
+    class FakeRegistry:
+        def resolve(self, _network):
+            return object()
+
+    monkeypatch.setattr("crynux_mcp.server.EvmClient", FakeClient)
+    monkeypatch.setattr("crynux_mcp.server.registry", FakeRegistry())
+
+    payload = handle_get_latest_block_number(network="dymension")
+    assert payload["block_number"] == 123456
+    assert "network" not in payload
+    assert "chain_id" not in payload
+
+
 def test_handle_transfer_sanitizes_private_key_error(monkeypatch) -> None:  # type: ignore[no-untyped-def]
     class FakeClient:
         def __init__(self, _chain) -> None: