wayfinder-paths 0.1.22__py3-none-any.whl → 0.1.24__py3-none-any.whl

wayfinder_paths/mcp/tools/strategies.py

@@ -0,0 +1,188 @@
+from __future__ import annotations
+
+import asyncio
+import importlib
+from pathlib import Path
+from typing import Any, Literal
+
+from wayfinder_paths.core.config import CONFIG
+from wayfinder_paths.core.engine.manifest import load_strategy_manifest
+from wayfinder_paths.core.strategies.Strategy import Strategy
+from wayfinder_paths.core.utils.evm_helpers import resolve_private_key_for_from_address
+from wayfinder_paths.core.utils.web3 import get_transaction_chain_id, web3_from_chain_id
+from wayfinder_paths.mcp.utils import err, ok, repo_root
+
+
+def _strategy_dir(name: str) -> Path:
+    return repo_root() / "wayfinder_paths" / "strategies" / name
+
+
+def _load_strategy_class(strategy_name: str) -> type[Strategy]:
+    manifest_path = _strategy_dir(strategy_name) / "manifest.yaml"
+    if not manifest_path.exists():
+        raise FileNotFoundError(f"Missing manifest.yaml for strategy: {strategy_name}")
+    manifest = load_strategy_manifest(str(manifest_path))
+    module_path, class_name = manifest.entrypoint.rsplit(".", 1)
+    module = importlib.import_module(module_path)
+    return getattr(module, class_name)
+
+
+def _get_strategy_config(strategy_name: str) -> dict[str, Any]:
+    config = dict(CONFIG.get("strategy", {}))
+    wallets = {w["label"]: w for w in CONFIG.get("wallets", [])}
+
+    if "main_wallet" not in config and "main" in wallets:
+        config["main_wallet"] = {"address": wallets["main"]["address"]}
+    if "strategy_wallet" not in config and strategy_name in wallets:
+        config["strategy_wallet"] = {"address": wallets[strategy_name]["address"]}
+
+    by_addr = {w["address"].lower(): w for w in CONFIG.get("wallets", [])}
+    for key in ("main_wallet", "strategy_wallet"):
+        if wallet := config.get(key):
+            if entry := by_addr.get(wallet.get("address", "").lower()):
+                if pk := entry.get("private_key") or entry.get("private_key_hex"):
+                    wallet["private_key_hex"] = pk
+    return config
+
+
+def _create_signing_callback(address: str, config: dict[str, Any]):
+    async def sign(transaction: dict) -> str:
+        pk = resolve_private_key_for_from_address(address, config)
+        async with web3_from_chain_id(get_transaction_chain_id(transaction)) as web3:
+            return web3.eth.account.sign_transaction(
+                transaction, pk
+            ).raw_transaction.hex()
+
+    return sign
+
+
+async def run_strategy(
+    *,
+    strategy: str,
+    action: Literal[
+        "status",
+        "analyze",
+        "snapshot",
+        "policy",
+        "deposit",
+        "update",
+        "withdraw",
+        "exit",
+    ],
+    amount_usdc: float = 1000.0,
+    amount: float | None = None,
+) -> dict[str, Any]:
+    if not strategy.strip():
+        return err("invalid_request", "strategy is required")
+
+    try:
+        strategy_class = _load_strategy_class(strategy)
+    except Exception as exc:  # noqa: BLE001
+        return err("not_found", str(exc))
+
+    if action == "policy":
+        pol = getattr(strategy_class, "policies", None)
+        if not callable(pol):
+            return ok({"strategy": strategy, "action": action, "output": []})
+        try:
+            res = pol()  # type: ignore[misc]
+            if asyncio.iscoroutine(res):
+                res = await res
+            return ok({"strategy": strategy, "action": action, "output": res})
+        except Exception as exc:  # noqa: BLE001
+            return err("strategy_error", str(exc))
+
+    config = _get_strategy_config(strategy)
+
+    def signing_cb(key: str):
+        if addr := config.get(key, {}).get("address"):
+            return _create_signing_callback(addr, config)
+        return None
+
+    try:
+        strategy_obj = strategy_class(
+            config,
+            main_wallet_signing_callback=signing_cb("main_wallet"),
+            strategy_wallet_signing_callback=signing_cb("strategy_wallet"),
+        )
+    except TypeError:
+        try:
+            strategy_obj = strategy_class(config=config)
+        except TypeError:
+            strategy_obj = strategy_class()
+
+    try:
+        if hasattr(strategy_obj, "setup"):
+            await strategy_obj.setup()
+
+        if action == "status":
+            out = await strategy_obj.status()
+            return ok({"strategy": strategy, "action": action, "output": out})
+
+        if action == "analyze":
+            if hasattr(strategy_obj, "analyze"):
+                out = await strategy_obj.analyze(deposit_usdc=amount_usdc)
+                return ok({"strategy": strategy, "action": action, "output": out})
+            return err("not_supported", "Strategy does not support analyze()")
+
+        if action == "snapshot":
+            if hasattr(strategy_obj, "build_batch_snapshot"):
+                out = await strategy_obj.build_batch_snapshot(
+                    score_deposit_usdc=amount_usdc
+                )
+                return ok({"strategy": strategy, "action": action, "output": out})
+            return err(
+                "not_supported", "Strategy does not support build_batch_snapshot()"
+            )
+
+        if action == "deposit":
+            if amount is None:
+                return err("invalid_request", "amount required for deposit")
+            success, msg = await strategy_obj.deposit(amount=amount)
+            return ok(
+                {
+                    "strategy": strategy,
+                    "action": action,
+                    "success": success,
+                    "message": msg,
+                }
+            )
+
+        if action == "update":
+            success, msg = await strategy_obj.update()
+            return ok(
+                {
+                    "strategy": strategy,
+                    "action": action,
+                    "success": success,
+                    "message": msg,
+                }
+            )
+
+        if action == "withdraw":
+            success, msg = await strategy_obj.withdraw(amount=amount)
+            return ok(
+                {
+                    "strategy": strategy,
+                    "action": action,
+                    "success": success,
+                    "message": msg,
+                }
+            )
+
+        if action == "exit":
+            if hasattr(strategy_obj, "exit"):
+                success, msg = await strategy_obj.exit()
+                return ok(
+                    {
+                        "strategy": strategy,
+                        "action": action,
+                        "success": success,
+                        "message": msg,
+                    }
+                )
+            return err("not_supported", "Strategy does not support exit()")
+
+        return err("invalid_request", f"Unknown action: {action}")
+    except Exception as exc:  # noqa: BLE001
+        return err("strategy_error", str(exc))

wayfinder_paths/mcp/tools/tokens.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from typing import Any, Literal
+
+from wayfinder_paths.core.clients.TokenClient import TokenClient
+from wayfinder_paths.mcp.utils import err, ok
+
+
+async def tokens(
+    action: Literal["resolve", "gas", "fuzzy"],
+    query: str | None = None,
+    chain_code: str | None = None,
+) -> dict[str, Any]:
+    client = TokenClient()
+    try:
+        if action == "resolve":
+            q = (query or "").strip()
+            if not q:
+                return err(
+                    "invalid_request", "query is required for tokens(action=resolve)"
+                )
+            token = await client.get_token_details(q)
+            return ok({"token": token})
+
+        if action == "gas":
+            cc = (chain_code or "").strip()
+            if not cc:
+                return err(
+                    "invalid_request", "chain_code is required for tokens(action=gas)"
+                )
+            token = await client.get_gas_token(cc)
+            return ok({"token": token})
+
+        if action == "fuzzy":
+            q = (query or "").strip()
+            if not q:
+                return err(
+                    "invalid_request", "query is required for tokens(action=fuzzy)"
+                )
+            cc = (chain_code or "").strip() or None
+            result = await client.fuzzy_search(q, chain=cc)
+            return ok(result)
+
+        return err("invalid_request", f"Unknown tokens action: {action}")
+    except Exception as exc:  # noqa: BLE001
+        return err("token_error", str(exc))

wayfinder_paths/mcp/tools/wallets.py

@@ -0,0 +1,354 @@
+from __future__ import annotations
+
+import asyncio
+import importlib
+import time
+from typing import Any, Literal
+
+from wayfinder_paths.core.utils.wallets import make_random_wallet, write_wallet_to_json
+from wayfinder_paths.mcp.state.profile_store import WalletProfileStore
+from wayfinder_paths.mcp.utils import (
+    err,
+    find_wallet_by_label,
+    load_wallets,
+    normalize_address,
+    ok,
+    repo_root,
+    wallets_path,
+)
+
+PROTOCOL_ADAPTERS: dict[str, dict[str, Any]] = {
+    "hyperliquid": {
+        "module": "wayfinder_paths.adapters.hyperliquid_adapter.adapter",
+        "class": "HyperliquidAdapter",
+        "init_kwargs": {"simulation": True},
+        "method": "get_full_user_state",
+        "account_param": "account",
+        "extra_kwargs": {},
+    },
+    "hyperlend": {
+        "module": "wayfinder_paths.adapters.hyperlend_adapter.adapter",
+        "class": "HyperlendAdapter",
+        "init_kwargs": {},
+        "method": "get_full_user_state",
+        "account_param": "account",
+        "extra_kwargs": {"include_zero_positions": False},
+    },
+    "moonwell": {
+        "module": "wayfinder_paths.adapters.moonwell_adapter.adapter",
+        "class": "MoonwellAdapter",
+        "init_kwargs": {},
+        "method": "get_full_user_state",
+        "account_param": "account",
+        "extra_kwargs": {"include_zero_positions": False},
+    },
+    "boros": {
+        "module": "wayfinder_paths.adapters.boros_adapter.adapter",
+        "class": "BorosAdapter",
+        "init_kwargs": {},
+        "method": "get_full_user_state",
+        "account_param": "account",
+        "extra_kwargs": {},
+    },
+    "pendle": {
+        "module": "wayfinder_paths.adapters.pendle_adapter.adapter",
+        "class": "PendleAdapter",
+        "init_kwargs": {},
+        "method": "get_full_user_state",
+        "account_param": "account",
+        "extra_kwargs": {"chain": 42161, "include_zero_positions": False},
+    },
+}
+
+
+def _public_wallet_view(w: dict[str, Any]) -> dict[str, Any]:
+    return {"label": w.get("label"), "address": w.get("address")}
+
+
+def _resolve_wallet_address(
+    *, wallet_label: str | None, wallet_address: str | None
+) -> tuple[str | None, str | None]:
+    waddr = normalize_address(wallet_address)
+    if waddr:
+        return waddr, None
+
+    want = (wallet_label or "").strip()
+    if not want:
+        return None, None
+
+    w = find_wallet_by_label(want)
+    if not w:
+        return None, None
+
+    return normalize_address(w.get("address")), want
+
+
+async def _query_adapter(
+    protocol: str,
+    address: str,
+    include_zero_positions: bool = False,
+) -> dict[str, Any]:
+    config = PROTOCOL_ADAPTERS.get(protocol)
+    if not config:
+        return {
+            "protocol": protocol,
+            "ok": False,
+            "error": f"Unknown protocol: {protocol}",
+        }
+
+    start = time.time()
+    try:
+        module = importlib.import_module(config["module"])
+        adapter_class = getattr(module, config["class"])
+        adapter = adapter_class(**config["init_kwargs"])
+
+        method = getattr(adapter, config["method"])
+        kwargs = {config["account_param"]: address, **config["extra_kwargs"]}
+
+        if "include_zero_positions" in config["extra_kwargs"]:
+            kwargs["include_zero_positions"] = include_zero_positions
+
+        success, data = await method(**kwargs)
+        duration = time.time() - start
+
+        return {
+            "protocol": protocol,
+            "ok": bool(success),
+            "data": data if success else None,
+            "error": data if not success else None,
+            "duration_s": round(duration, 3),
+        }
+
+    except Exception as exc:
+        duration = time.time() - start
+        return {
+            "protocol": protocol,
+            "ok": False,
+            "error": str(exc),
+            "duration_s": round(duration, 3),
+        }
+
+
+async def wallets(
+    action: Literal["list", "create", "get", "annotate", "discover_portfolio"],
+    *,
+    label: str | None = None,
+    wallet_label: str | None = None,
+    wallet_address: str | None = None,
+    protocol: str | None = None,
+    annotate_action: str | None = None,
+    tool: str | None = None,
+    status: str | None = None,
+    chain_id: int | None = None,
+    details: dict[str, Any] | None = None,
+    idempotency_key: str | None = None,
+    protocols: list[str] | None = None,
+    parallel: bool = False,
+    include_zero_positions: bool = False,
+) -> dict[str, Any]:
+    p = wallets_path()
+    root = repo_root()
+    rel = str(p.relative_to(root)) if p.is_absolute() and root in p.parents else str(p)
+    store = WalletProfileStore.default()
+
+    if action == "list":
+        existing = load_wallets()
+        wallet_list = []
+        for w in existing:
+            view = _public_wallet_view(w)
+            # Enrich with tracked protocols (lowercase to match profile store)
+            addr = normalize_address(w.get("address"))
+            if addr:
+                tracked = store.get_protocols_for_wallet(addr.lower())
+                view["protocols"] = tracked
+            else:
+                view["protocols"] = []
+            wallet_list.append(view)
+        return ok({"wallets_path": rel, "wallets": wallet_list})
+
+    if action == "create":
+        existing = load_wallets()
+        want = (label or "").strip()
+        if not want:
+            return err(
+                "invalid_request", "label is required for wallets(action=create)"
+            )
+
+        for w in existing:
+            if str(w.get("label", "")).strip() == want:
+                return ok(
+                    {
+                        "wallets_path": rel,
+                        "wallets": [_public_wallet_view(x) for x in existing],
+                        "created": _public_wallet_view(w),
+                        "note": "Wallet label already existed; returning existing wallet.",
+                    }
+                )
+
+        w = make_random_wallet()
+        w["label"] = want
+        write_wallet_to_json(w, out_dir=p.parent, filename=p.name)
+
+        refreshed = load_wallets()
+        return ok(
+            {
+                "wallets_path": rel,
+                "wallets": [_public_wallet_view(x) for x in refreshed],
+                "created": _public_wallet_view(w),
+            }
+        )
+
+    if action == "get":
+        address, lbl = _resolve_wallet_address(
+            wallet_label=wallet_label or label, wallet_address=wallet_address
+        )
+        if not address:
+            return err(
+                "invalid_request",
+                "wallet_label or wallet_address is required",
+                {"wallet_label": wallet_label, "wallet_address": wallet_address},
+            )
+
+        profile = store.get_profile(address)
+        return ok(
+            {
+                "action": "get",
+                "address": address,
+                "label": lbl,
+                "profile": profile,
+                "found": profile is not None,
+            }
+        )
+
+    if action == "annotate":
+        address, lbl = _resolve_wallet_address(
+            wallet_label=wallet_label or label, wallet_address=wallet_address
+        )
+        if not address:
+            return err(
+                "invalid_request",
+                "wallet_label or wallet_address is required",
+            )
+        if not protocol:
+            return err("invalid_request", "protocol is required for annotate")
+        if not annotate_action:
+            return err("invalid_request", "annotate_action is required for annotate")
+        if not tool:
+            return err("invalid_request", "tool is required for annotate")
+        if not status:
+            return err("invalid_request", "status is required for annotate")
+
+        store.annotate(
+            address=address,
+            label=lbl,
+            protocol=protocol,
+            action=annotate_action,
+            tool=tool,
+            status=status,
+            chain_id=chain_id,
+            details=details,
+            idempotency_key=idempotency_key,
+        )
+
+        return ok(
+            {
+                "action": "annotate",
+                "address": address,
+                "protocol": protocol,
+                "annotated": True,
+            }
+        )
+
+    if action == "discover_portfolio":
+        address, lbl = _resolve_wallet_address(
+            wallet_label=wallet_label or label, wallet_address=wallet_address
+        )
+        if not address:
+            return err(
+                "invalid_request",
+                "wallet_label or wallet_address is required for discover_portfolio",
+            )
+
+        profile_protocols = store.get_protocols_for_wallet(address)
+
+        if protocols:
+            target_protocols = list(dict.fromkeys(protocols))  # dedupe, preserve order
+        else:
+            target_protocols = profile_protocols
+
+        supported_protocols = [p for p in target_protocols if p in PROTOCOL_ADAPTERS]
+        unsupported = [p for p in target_protocols if p not in PROTOCOL_ADAPTERS]
+
+        if not supported_protocols:
+            return ok(
+                {
+                    "action": "discover_portfolio",
+                    "address": address,
+                    "label": lbl,
+                    "profile_protocols": profile_protocols,
+                    "positions": [],
+                    "warning": "No supported protocols to query",
+                    "unsupported_protocols": unsupported,
+                }
+            )
+
+        if len(supported_protocols) >= 3 and not parallel:
+            return ok(
+                {
+                    "action": "discover_portfolio",
+                    "address": address,
+                    "label": lbl,
+                    "profile_protocols": profile_protocols,
+                    "supported_protocols": supported_protocols,
+                    "requires_confirmation": True,
+                    "warning": f"Found {len(supported_protocols)} protocols to query. "
+                    f"Set parallel=true for concurrent queries, or filter with protocols=[...] "
+                    f"to query specific protocols.",
+                    "protocols_to_query": supported_protocols,
+                }
+            )
+
+        start = time.time()
+        results: list[dict[str, Any]] = []
+
+        if parallel:
+            tasks = [
+                _query_adapter(proto, address, include_zero_positions)
+                for proto in supported_protocols
+            ]
+            results = await asyncio.gather(*tasks)
+        else:
+            for proto in supported_protocols:
+                result = await _query_adapter(proto, address, include_zero_positions)
+                results.append(result)
+
+        total_duration = time.time() - start
+        all_positions: list[dict[str, Any]] = []
+        for r in results:
+            if r.get("ok") and r.get("data"):
+                data = r["data"]
+                positions = data.get("positions", [])
+                if positions:
+                    for pos in positions:
+                        all_positions.append(
+                            {"protocol": r["protocol"], "position": pos}
+                        )
+                r["data"] = data
+
+        return ok(
+            {
+                "action": "discover_portfolio",
+                "address": address,
+                "label": lbl,
+                "profile_protocols": profile_protocols,
+                "queried_protocols": supported_protocols,
+                "results": results,
+                "positions_count": len(all_positions),
+                "positions_summary": all_positions[:10],
+                "total_duration_s": round(total_duration, 3),
+                "parallel": parallel,
+                "unsupported_protocols": unsupported if unsupported else None,
+            }
+        )
+
+    return err("invalid_request", f"Unknown action: {action}")