doublezero-serviceability 0.0.2__py3-none-any.whl

doublezero_serviceability-0.0.2.dist-info/METADATA

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: doublezero-serviceability
+Version: 0.0.2
+Summary: DoubleZero Serviceability SDK
+Requires-Python: >=3.10
+Requires-Dist: borsh-incremental
+Requires-Dist: httpx>=0.27
+Requires-Dist: solana>=0.35
+Requires-Dist: solders>=0.21

doublezero_serviceability-0.0.2.dist-info/RECORD

@@ -0,0 +1,14 @@
+serviceability/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+serviceability/client.py,sha256=3FYGJqIgkRm-Kc7DrckXVPreIZQP_ozn9lt8mUDnNQY,2335
+serviceability/config.py,sha256=F04WS56QdiS1Uj_QnkkIsHaLi-RLLNS9BTxao-Gt5yo,690
+serviceability/pda.py,sha256=wWEruxwSAiVCzrZXdDd_hfR3udTMGZQgqxcp99_tHaY,739
+serviceability/rpc.py,sha256=G7GPRl0DM3x3p6B3NTdIgjDJTljwCat9-C_XKL5wXwM,1548
+serviceability/state.py,sha256=ekUD4rprfo4ipmBNWgUNBsCrjFJeFH0_wIKUU0-OeKA,23905
+serviceability/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+serviceability/tests/test_compat.py,sha256=H2N1iTJWxovhIIy1nJIPpTFhL3eiNNcoiT1DCPZExtA,4903
+serviceability/tests/test_enum_strings.py,sha256=hL-WoPRg2xa32UdxOq3GAyiyoEqQc2HyrC7winDoXgk,3448
+serviceability/tests/test_fixtures.py,sha256=9Tit9fyyADbE4KdPg5_IMv9VPk39Ge4obGGYLuW209Q,11132
+serviceability/tests/test_pda.py,sha256=9uOGU821Y98J6JdvZ6AiESvN9leYDT8K8vNmCqqfCGs,998
+doublezero_serviceability-0.0.2.dist-info/METADATA,sha256=SF8DeKOgoNBawRmDozmk611qzk3IiK1t1wozZzzPVbs,249
+doublezero_serviceability-0.0.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+doublezero_serviceability-0.0.2.dist-info/RECORD,,

doublezero_serviceability-0.0.2.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

serviceability/client.py

@@ -0,0 +1,85 @@
+"""RPC client for fetching serviceability program accounts."""
+
+from __future__ import annotations
+
+from typing import Protocol
+
+from solders.pubkey import Pubkey  # type: ignore[import-untyped]
+from solders.rpc.responses import GetAccountInfoResp  # type: ignore[import-untyped]
+
+from serviceability.config import PROGRAM_IDS, LEDGER_RPC_URLS
+from serviceability.rpc import new_rpc_client
+from serviceability.state import (
+    AccessPass,
+    Contributor,
+    Device,
+    Exchange,
+    GlobalConfig,
+    GlobalState,
+    Link,
+    Location,
+    MulticastGroup,
+    ProgramConfig,
+    User,
+)
+
+
+class SolanaClient(Protocol):
+    def get_account_info(self, pubkey: Pubkey) -> GetAccountInfoResp: ...
+
+
+class ProgramData:
+    """Aggregate of all serviceability program accounts."""
+
+    def __init__(self) -> None:
+        self.global_state: GlobalState | None = None
+        self.global_config: GlobalConfig | None = None
+        self.program_config: ProgramConfig | None = None
+        self.locations: list[Location] = []
+        self.exchanges: list[Exchange] = []
+        self.devices: list[Device] = []
+        self.links: list[Link] = []
+        self.users: list[User] = []
+        self.multicast_groups: list[MulticastGroup] = []
+        self.contributors: list[Contributor] = []
+        self.access_passes: list[AccessPass] = []
+
+
+class Client:
+    """Read-only client for serviceability program accounts."""
+
+    def __init__(
+        self,
+        solana_rpc: SolanaClient,
+        program_id: Pubkey,
+    ) -> None:
+        self._solana_rpc = solana_rpc
+        self._program_id = program_id
+
+    @classmethod
+    def from_env(cls, env: str) -> Client:
+        """Create a client configured for the given environment.
+
+        Args:
+            env: Environment name ("mainnet-beta", "testnet", "devnet", "localnet")
+        """
+        return cls(
+            new_rpc_client(LEDGER_RPC_URLS[env]),
+            Pubkey.from_string(PROGRAM_IDS[env]),
+        )
+
+    @classmethod
+    def mainnet_beta(cls) -> Client:
+        return cls.from_env("mainnet-beta")
+
+    @classmethod
+    def testnet(cls) -> Client:
+        return cls.from_env("testnet")
+
+    @classmethod
+    def devnet(cls) -> Client:
+        return cls.from_env("devnet")
+
+    @classmethod
+    def localnet(cls) -> Client:
+        return cls.from_env("localnet")

serviceability/config.py

@@ -0,0 +1,15 @@
+"""Network configuration for the serviceability program."""
+
+PROGRAM_IDS = {
+    "mainnet-beta": "ser2VaTMAcYTaauMrTSfSrxBaUDq7BLNs2xfUugTAGv",
+    "testnet": "DZtnuQ839pSaDMFG5q1ad2V95G82S5EC4RrB3Ndw2Heb",
+    "devnet": "GYhQDKuESrasNZGyhMJhGYFtbzNijYhcrN9poSqCQVah",
+    "localnet": "7CTniUa88iJKUHTrCkB4TjAoG6TD7AMivhQeuqN2LPtX",
+}
+
+LEDGER_RPC_URLS = {
+    "mainnet-beta": "https://doublezero-mainnet-beta.rpcpool.com/db336024-e7a8-46b1-80e5-352dd77060ab",
+    "testnet": "https://doublezerolocalnet.rpcpool.com/8a4fd3f4-0977-449f-88c7-63d4b0f10f16",
+    "devnet": "https://doublezerolocalnet.rpcpool.com/8a4fd3f4-0977-449f-88c7-63d4b0f10f16",
+    "localnet": "http://localhost:8899",
+}

serviceability/pda.py

@@ -0,0 +1,20 @@
+"""PDA derivation for serviceability program accounts."""
+
+from solders.pubkey import Pubkey  # type: ignore[import-untyped]
+
+SEED_PREFIX = b"doublezero"
+SEED_GLOBAL_STATE = b"globalstate"
+SEED_GLOBAL_CONFIG = b"config"
+SEED_PROGRAM_CONFIG = b"programconfig"
+
+
+def derive_global_state_pda(program_id: Pubkey) -> tuple[Pubkey, int]:
+    return Pubkey.find_program_address([SEED_PREFIX, SEED_GLOBAL_STATE], program_id)
+
+
+def derive_global_config_pda(program_id: Pubkey) -> tuple[Pubkey, int]:
+    return Pubkey.find_program_address([SEED_PREFIX, SEED_GLOBAL_CONFIG], program_id)
+
+
+def derive_program_config_pda(program_id: Pubkey) -> tuple[Pubkey, int]:
+    return Pubkey.find_program_address([SEED_PREFIX, SEED_PROGRAM_CONFIG], program_id)

serviceability/rpc.py

@@ -0,0 +1,48 @@
+"""RPC client helpers with retry on rate limiting."""
+
+import time
+
+import httpx
+from solana.rpc.api import Client as SolanaHTTPClient  # type: ignore[import-untyped]
+
+_DEFAULT_MAX_RETRIES = 5
+
+
+class _RetryTransport(httpx.BaseTransport):
+    """HTTP transport that retries on 429 Too Many Requests."""
+
+    def __init__(
+        self,
+        wrapped: httpx.BaseTransport | None = None,
+        max_retries: int = _DEFAULT_MAX_RETRIES,
+    ) -> None:
+        self._wrapped = wrapped or httpx.HTTPTransport()
+        self._max_retries = max_retries
+
+    def handle_request(self, request: httpx.Request) -> httpx.Response:
+        for attempt in range(self._max_retries + 1):
+            response = self._wrapped.handle_request(request)
+            if response.status_code != 429 or attempt >= self._max_retries:
+                return response
+            response.close()
+            time.sleep((attempt + 1) * 2)
+        return response  # unreachable, but satisfies type checker
+
+
+def new_rpc_client(
+    url: str,
+    timeout: float = 30,
+    max_retries: int = _DEFAULT_MAX_RETRIES,
+) -> SolanaHTTPClient:
+    """Create a Solana RPC client with automatic retry on 429 responses."""
+    client = SolanaHTTPClient(url, timeout=timeout)
+    # Replace the underlying httpx session with one using retry transport.
+    transport = _RetryTransport(
+        wrapped=httpx.HTTPTransport(),
+        max_retries=max_retries,
+    )
+    client._provider.session = httpx.Client(
+        timeout=timeout,
+        transport=transport,
+    )
+    return client