1"""
2Wallet skill exports — for use in task scripts via core.skill_tools.
3 
4Usage:
5    from core.skill_tools import wallet
6    info = wallet.wallet_info()
7    bal = wallet.wallet_balance(chain="base")
8    all_bal = wallet.wallet_get_all_balances()
9 
10Delegates to /app/tools/wallet core functions for single-source maintenance.
11"""
12 
13import asyncio
14 
15try:
16    import nest_asyncio
17    nest_asyncio.apply()
18except ImportError:
19    pass
20 
21# Import core functions from /app/tools/wallet
22from tools.wallet import (
23    _wallet_request,
24    _is_fly_machine,
25    _get_wallet_addresses,
26    _validate_and_clean_rules,
27    DEBANK_CHAIN_MAP,
28)
29from core.http_client import proxied_get
30 
31EVM_CHAINS = list(DEBANK_CHAIN_MAP.keys())
32 
33 
34def _run(coro):
35    """Run async in sync context (works even if event loop is running)."""
36    try:
37        loop = asyncio.get_event_loop()
38        return loop.run_until_complete(coro)
39    except RuntimeError:
40        return asyncio.run(coro)
41 
42 
43# ── Info ─────────────────────────────────────────────────────────────────────
44 
45def wallet_info():
46    """Get all wallet addresses."""
47    return _run(_wallet_request("GET", "/agent/wallet"))
48 
49 
50# ── Balances ─────────────────────────────────────────────────────────────────
51 
52def wallet_balance(chain: str, address: str = "", asset: str = ""):
53    """Get EVM balance on a chain (via DeBank or wallet-service fallback)."""
54    import os
55    async def _impl():
56        if chain not in EVM_CHAINS:
57            return {"error": f"Invalid chain '{chain}'. Must be one of: {', '.join(EVM_CHAINS)}"}
58        debank_key = os.environ.get("DEBANK_API_KEY", "")
59        if debank_key:
60            evm_address = address
61            if not evm_address:
62                addrs = await _get_wallet_addresses()
63                evm_address = addrs.get("evm", "")
64            if not evm_address:
65                return {"error": "Could not determine EVM wallet address"}
66            resp = proxied_get(
67                "https://pro-openapi.debank.com/v1/user/token_list",
68                params={"id": evm_address, "chain_id": DEBANK_CHAIN_MAP.get(chain), "is_all": "false"},
69                headers={"AccessKey": debank_key},
70            )
71            resp.raise_for_status()
72            return {"address": evm_address, "chain": chain, "tokens": resp.json(), "source": "debank"}
73        else:
74            params = [f"chain_type=ethereum&chain={chain}"]
75            if asset: params.append(f"asset={asset}")
76            return await _wallet_request("GET", f"/agent/balance?{'&'.join(params)}")
77    return _run(_impl())
78 
79 
80def wallet_sol_balance(address: str = "", asset: str = ""):
81    """Get Solana balance."""
82    import os
83    async def _impl():
84        birdeye_key = os.environ.get("BIRDEYE_API_KEY", "")
85        if birdeye_key:
86            sol_address = address
87            if not sol_address:
88                addrs = await _get_wallet_addresses()
89                sol_address = addrs.get("sol", "")
90            if not sol_address:
91                return {"error": "Could not determine Solana wallet address"}
92            resp = proxied_get(
93                "https://public-api.birdeye.so/wallet/v2/net-worth",
94                params={"wallet": sol_address},
95                headers={"X-API-KEY": birdeye_key, "x-chain": "solana", "accept": "application/json"},
96            )
97            resp.raise_for_status()
98            return {"address": sol_address, "source": "birdeye", "data": resp.json()}
99        else:
100            params = ["chain_type=solana"]
101            if asset: params.append(f"asset={asset}")
102            return await _wallet_request("GET", f"/agent/balance?{'&'.join(params)}")
103    return _run(_impl())
104 
105 
106def wallet_get_all_balances(evm_address: str = "", sol_address: str = ""):
107    """Get balances across all chains."""
108    async def _impl():
109        ea, sa = evm_address, sol_address
110        if not ea or not sa:
111            if _is_fly_machine():
112                try:
113                    addrs = await _get_wallet_addresses()
114                    ea = ea or addrs.get("evm", "")
115                    sa = sa or addrs.get("sol", "")
116                except Exception:
117                    pass
118        # Delegate to per-chain balance calls
119        result = {}
120        if ea:
121            for chain in EVM_CHAINS:
122                try:
123                    data = wallet_balance(chain, ea)
124                    if "error" not in data:
125                        result[chain] = data
126                except Exception:
127                    pass
128        if sa:
129            try:
130                data = wallet_sol_balance(sa)
131                if "error" not in data:
132                    result["solana"] = data
133            except Exception:
134                pass
135        return result
136    return _run(_impl())
137 
138 
139# ── Transfers ────────────────────────────────────────────────────────────────
140 
141def wallet_transfer(to, amount, chain_id=1, data="", **kw):
142    """Sign and broadcast EVM tx."""
143    async def _impl():
144        body = {"to": to, "amount": str(amount), "chain_id": chain_id}
145        if data: body["data"] = data
146        for k in ("gas_limit", "gas_price", "max_fee_per_gas", "max_priority_fee_per_gas", "nonce"):
147            if kw.get(k): body[k] = kw[k]
148        if kw.get("tx_type") is not None: body["tx_type"] = kw["tx_type"]
149        return await _wallet_request("POST", "/agent/transfer", body)
150    return _run(_impl())
151 
152 
153def wallet_sign_transaction(to, amount, chain_id=1, data="", **kw):
154    """Sign EVM tx without broadcasting."""
155    async def _impl():
156        body = {"to": to, "amount": str(amount), "chain_id": chain_id}
157        if data: body["data"] = data
158        for k in ("gas_limit", "gas_price", "max_fee_per_gas", "max_priority_fee_per_gas", "nonce"):
159            if kw.get(k): body[k] = kw[k]
160        if kw.get("tx_type") is not None: body["tx_type"] = kw["tx_type"]
161        return await _wallet_request("POST", "/agent/sign-transaction", body)
162    return _run(_impl())
163 
164 
165def wallet_sign(message):
166    """EIP-191 personal_sign."""
167    return _run(_wallet_request("POST", "/agent/sign", {"message": message}))
168 
169 
170def wallet_sign_typed_data(domain, types, primaryType, message):
171    """Sign EIP-712 typed data."""
172    return _run(_wallet_request("POST", "/agent/sign-typed-data", {
173        "domain": domain, "types": types, "primaryType": primaryType, "message": message,
174    }))
175 
176 
177def wallet_transactions(chain="ethereum", asset="", limit=20):
178    """EVM tx history."""
179    qs = f"?chain_type=ethereum&chain={chain}&asset={asset or 'eth'}&limit={limit}"
180    return _run(_wallet_request("GET", f"/agent/transactions{qs}"))
181 
182 
183# ── Solana ───────────────────────────────────────────────────────────────────
184 
185def wallet_sol_transfer(transaction, caip2="solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp"):
186    """Sign and broadcast Solana tx."""
187    return _run(_wallet_request("POST", "/agent/sol/transfer", {
188        "transaction": transaction, "caip2": caip2,
189    }))
190 
191 
192def wallet_sol_sign_transaction(transaction):
193    """Sign Solana tx without broadcasting."""
194    return _run(_wallet_request("POST", "/agent/sol/sign-transaction", {"transaction": transaction}))
195 
196 
197def wallet_sol_sign(message):
198    """Sign message with Solana wallet."""
199    return _run(_wallet_request("POST", "/agent/sol/sign", {"message": message}))
200 
201 
202def wallet_sol_transactions(chain="solana", asset="sol", limit=20):
203    """Solana tx history."""
204    qs = f"?chain_type=solana&chain={chain}&asset={asset}&limit={limit}"
205    return _run(_wallet_request("GET", f"/agent/transactions{qs}"))
206 
207 
208# ── Policy ───────────────────────────────────────────────────────────────────
209 
210def wallet_get_policy(chain_type="ethereum"):
211    """Get current policy status."""
212    return _run(_wallet_request("GET", f"/agent/policy?chain_type={chain_type}"))
213 
214 
215def validate_and_clean_rules(rules, chain_type):
216    """Validate rules (re-export from system)."""
217    return _validate_and_clean_rules(rules, chain_type)
218