Wallet

"""
Wallet Skill — Multi-chain wallet (EVM + Solana)
Balances, transfers, signing, policy management.
Delegates to /app/tools/wallet for core functions.
"""
import logging
from typing import List

logger = logging.getLogger(__name__)


def register(api) -> List[str]:
    """Register all wallet tools with the Agent framework."""
    registered = []

    try:
        from .wallet import (
            WalletInfoTool,
            WalletBalanceTool,
            WalletSolBalanceTool,
            WalletGetAllBalancesTool,
            WalletTransferTool,
            WalletSignTransactionTool,
            WalletSignTool,
            WalletSignTypedDataTool,
            WalletTransactionsTool,
            WalletSolTransferTool,
            WalletSolSignTransactionTool,
            WalletSolSignTool,
            WalletSolTransactionsTool,
            WalletGetPolicyTool,
            WalletProposePolicyTool,
        )

        tools = [
            WalletInfoTool(),
            WalletBalanceTool(),
            WalletSolBalanceTool(),
            WalletGetAllBalancesTool(),
            WalletTransferTool(),
            WalletSignTransactionTool(),
            WalletSignTool(),
            WalletSignTypedDataTool(),
            WalletTransactionsTool(),
            WalletSolTransferTool(),
            WalletSolSignTransactionTool(),
            WalletSolSignTool(),
            WalletSolTransactionsTool(),
            WalletGetPolicyTool(),
            WalletProposePolicyTool(),
        ]

        for tool in tools:
            api.register_tool(tool)
            registered.append(tool.name)

    except Exception as e:
        logger.error(f"Failed to register wallet tools: {e}", exc_info=True)

    return registered


"""
Wallet skill exports — for use in task scripts via core.skill_tools.

Usage:
    from core.skill_tools import wallet
    info = wallet.wallet_info()
    bal = wallet.wallet_balance(chain="base")
    all_bal = wallet.wallet_get_all_balances()

Delegates to /app/tools/wallet core functions for single-source maintenance.
"""

import asyncio

try:
    import nest_asyncio
    nest_asyncio.apply()
except ImportError:
    pass

# Import core functions from /app/tools/wallet
from tools.wallet import (
    _wallet_request,
    _is_fly_machine,
    _get_wallet_addresses,
    _validate_and_clean_rules,
    DEBANK_CHAIN_MAP,
)
from core.http_client import proxied_get

EVM_CHAINS = list(DEBANK_CHAIN_MAP.keys())


def _run(coro):
    """Run async in sync context (works even if event loop is running)."""
    try:
        loop = asyncio.get_event_loop()
        return loop.run_until_complete(coro)
    except RuntimeError:
        return asyncio.run(coro)


# ── Info ─────────────────────────────────────────────────────────────────────

def wallet_info():
    """Get all wallet addresses."""
    return _run(_wallet_request("GET", "/agent/wallet"))


# ── Balances ─────────────────────────────────────────────────────────────────

def wallet_balance(chain: str, address: str = "", asset: str = ""):
    """Get EVM balance on a chain (via DeBank or wallet-service fallback)."""
    import os
    async def _impl():
        if chain not in EVM_CHAINS:
            return {"error": f"Invalid chain '{chain}'. Must be one of: {', '.join(EVM_CHAINS)}"}
        debank_key = os.environ.get("DEBANK_API_KEY", "")
        if debank_key:
            evm_address = address
            if not evm_address:
                addrs = await _get_wallet_addresses()
                evm_address = addrs.get("evm", "")
            if not evm_address:
                return {"error": "Could not determine EVM wallet address"}
            resp = proxied_get(
                "https://pro-openapi.debank.com/v1/user/token_list",
                params={"id": evm_address, "chain_id": DEBANK_CHAIN_MAP.get(chain), "is_all": "false"},
                headers={"AccessKey": debank_key},
            )
            resp.raise_for_status()
            return {"address": evm_address, "chai