@agentlayer.tech/wallet 0.1.0

package/agent-wallet/agent_wallet/wallet_layer/base58.py

@@ -0,0 +1,44 @@
+"""Minimal base58 helpers to avoid extra runtime dependencies."""
+
+from __future__ import annotations
+
+_ALPHABET = "123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz"
+_ALPHABET_INDEX = {char: index for index, char in enumerate(_ALPHABET)}
+
+
+def b58encode(raw: bytes) -> str:
+    """Encode bytes into a base58 string."""
+    if not raw:
+        return ""
+
+    zeros = 0
+    for byte in raw:
+        if byte == 0:
+            zeros += 1
+        else:
+            break
+
+    value = int.from_bytes(raw, "big")
+    encoded = []
+    while value > 0:
+        value, remainder = divmod(value, 58)
+        encoded.append(_ALPHABET[remainder])
+
+    return ("1" * zeros) + "".join(reversed(encoded or ["1"]))
+
+
+def b58decode(value: str) -> bytes:
+    """Decode a base58 string into bytes."""
+    cleaned = value.strip()
+    if not cleaned:
+        return b""
+
+    number = 0
+    for char in cleaned:
+        if char not in _ALPHABET_INDEX:
+            raise ValueError(f"Invalid base58 character: {char!r}")
+        number = (number * 58) + _ALPHABET_INDEX[char]
+
+    decoded = number.to_bytes((number.bit_length() + 7) // 8, "big") if number else b""
+    leading_zeros = len(cleaned) - len(cleaned.lstrip("1"))
+    return (b"\x00" * leading_zeros) + decoded

package/agent-wallet/agent_wallet/wallet_layer/factory.py

@@ -0,0 +1,102 @@
+"""Factory helpers for agent wallet backends."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from agent_wallet.bootstrap import ensure_solana_wallet_ready, ensure_wallet_pin
+from agent_wallet.encrypted_storage import load_wallet_secret_material
+from agent_wallet.config import (
+    resolve_runtime_solana_rpc_config,
+    resolve_runtime_solana_swap_config,
+    resolve_solana_private_key,
+    settings,
+)
+from agent_wallet.wallet_layer.base import AgentWalletBackend, WalletBackendError
+from agent_wallet.wallet_layer.wdk_evm import WdkEvmLocalWalletBackend
+from agent_wallet.wallet_layer.solana import SolanaLocalKeypairSigner, SolanaWalletBackend
+from agent_wallet.wallet_layer.wdk_btc import WdkBtcLocalWalletBackend
+
+
+def _load_keypair_material() -> str | None:
+    secret = resolve_solana_private_key()
+    if secret:
+        return secret
+
+    ensured = ensure_solana_wallet_ready()
+    keypair_path = settings.solana_agent_keypair_path.strip() or (ensured["path"] if ensured else "")
+    if not keypair_path:
+        return None
+
+    path = Path(keypair_path).expanduser()
+    if not path.exists():
+        raise WalletBackendError(
+            f"Configured Solana keypair path does not exist: {path}"
+        )
+    secret_material, _ = load_wallet_secret_material(path)
+    return secret_material.strip()
+
+
+def create_wallet_backend() -> AgentWalletBackend | None:
+    """Build the configured wallet backend instance."""
+    backend = settings.agent_wallet_backend.strip().lower()
+    if not backend or backend == "none":
+        return None
+
+    if backend in {"solana", "solana_local", "solana-local"}:
+        secret_material = _load_keypair_material()
+        signer = (
+            SolanaLocalKeypairSigner.from_secret_material(secret_material)
+            if secret_material
+            else None
+        )
+        keypair_path = settings.solana_agent_keypair_path.strip()
+        if signer and keypair_path:
+            ensure_wallet_pin(
+                Path(keypair_path).expanduser(),
+                address=signer.address,
+                network=settings.solana_network,
+            )
+        configured_address = settings.solana_agent_public_key.strip() or None
+        rpc_config = resolve_runtime_solana_rpc_config(
+            settings.solana_network,
+            settings.solana_rpc_url,
+            settings.solana_rpc_urls,
+        )
+        swap_config = resolve_runtime_solana_swap_config(settings.solana_network)
+        return SolanaWalletBackend(
+            rpc_url=rpc_config["rpc_urls"],
+            commitment=settings.solana_commitment,
+            network=settings.solana_network,
+            signer=signer,
+            address=configured_address,
+            sign_only=settings.agent_wallet_sign_only,
+            rpc_provider_mode=str(rpc_config["mode"]),
+            rpc_provider=str(rpc_config["provider"]),
+            rpc_transport=str(rpc_config["transport"]),
+            swap_provider=str(swap_config["provider"]),
+            swap_transport=str(swap_config["transport"]),
+        )
+
+    if backend in {"wdk_btc_local", "wdk-btc-local", "btc_local", "btc-local"}:
+        return WdkBtcLocalWalletBackend(
+            service_url=settings.wdk_btc_service_url,
+            wallet_id=settings.wdk_btc_wallet_id,
+            network=settings.solana_network,
+            account_index=settings.wdk_btc_account_index,
+            sign_only=settings.agent_wallet_sign_only,
+        )
+
+    if backend in {"wdk_evm_local", "wdk-evm-local", "evm_local", "evm-local"}:
+        return WdkEvmLocalWalletBackend(
+            service_url=settings.wdk_evm_service_url,
+            wallet_id=settings.wdk_evm_wallet_id,
+            network=settings.solana_network,
+            account_index=settings.wdk_evm_account_index,
+            sign_only=settings.agent_wallet_sign_only,
+        )
+
+    raise WalletBackendError(
+        f"Unsupported agent wallet backend: {backend}. "
+        "Supported values: none, solana_local, wdk_btc_local, wdk_evm_local."
+    )