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

wayfinder_paths/mcp/utils.py

@@ -0,0 +1,129 @@
+from __future__ import annotations
+
+import hashlib
+import json
+import os
+from decimal import ROUND_DOWN, Decimal, InvalidOperation, getcontext
+from pathlib import Path
+from typing import Any
+
+import yaml
+
+getcontext().prec = 78
+
+
+def ok(result: Any) -> dict[str, Any]:
+    return {"ok": True, "result": result}
+
+
+def err(code: str, message: str, details: Any | None = None) -> dict[str, Any]:
+    return {
+        "ok": False,
+        "error": {"code": str(code), "message": str(message), "details": details},
+    }
+
+
+def repo_root() -> Path:
+    cur = Path(__file__).resolve()
+    for parent in [cur, *cur.parents]:
+        if (parent / "pyproject.toml").exists():
+            return parent
+    return Path.cwd()
+
+
+def read_yaml(path: Path) -> dict[str, Any]:
+    data = yaml.safe_load(path.read_text())
+    return data if isinstance(data, dict) else {}
+
+
+def read_text_excerpt(path: Path, *, max_chars: int = 1200) -> str | None:
+    try:
+        text = path.read_text(encoding="utf-8", errors="replace").strip()
+    except OSError:
+        return None
+    if not text:
+        return None
+    if len(text) <= max_chars:
+        return text
+    return text[: max_chars - 3] + "..."
+
+
+def load_json_file(path: Path) -> Any:
+    try:
+        return json.loads(path.read_text())
+    except Exception:
+        return None
+
+
+def load_config_json() -> dict[str, Any]:
+    cfg_path = os.getenv("WAYFINDER_CONFIG_PATH", "config.json")
+    path = (
+        repo_root() / cfg_path if not Path(cfg_path).is_absolute() else Path(cfg_path)
+    )
+    parsed = load_json_file(path)
+    return parsed if isinstance(parsed, dict) else {}
+
+
+def wallets_path() -> Path:
+    cfg = load_config_json()
+    system = cfg.get("system") if isinstance(cfg.get("system"), dict) else {}
+    candidate = (
+        (system or {}).get("wallets_path")
+        or os.getenv("WALLETS_PATH")
+        or "wallets.json"
+    )
+    p = Path(candidate)
+    if not p.is_absolute():
+        p = repo_root() / p
+    return p
+
+
+def load_wallets() -> list[dict[str, Any]]:
+    cfg = load_config_json()
+    if isinstance(cfg.get("wallets"), list):
+        return [w for w in cfg["wallets"] if isinstance(w, dict)]
+
+    p = wallets_path()
+    if p.exists():
+        parsed = load_json_file(p)
+        if isinstance(parsed, list):
+            return [w for w in parsed if isinstance(w, dict)]
+        if isinstance(parsed, dict) and isinstance(parsed.get("wallets"), list):
+            return [w for w in parsed["wallets"] if isinstance(w, dict)]
+    return []
+
+
+def find_wallet_by_label(label: str) -> dict[str, Any] | None:
+    want = str(label).strip()
+    if not want:
+        return None
+    for w in load_wallets():
+        if str(w.get("label", "")).strip() == want:
+            return w
+    return None
+
+
+def normalize_address(addr: str | None) -> str | None:
+    if not addr:
+        return None
+    a = str(addr).strip()
+    return a if a else None
+
+
+def parse_amount_to_raw(amount: str, decimals: int) -> int:
+    try:
+        d = Decimal(str(amount).strip())
+    except (InvalidOperation, ValueError) as exc:
+        raise ValueError(f"Invalid amount: {amount}") from exc
+    if d <= 0:
+        raise ValueError("Amount must be positive")
+    scale = Decimal(10) ** int(decimals)
+    raw = (d * scale).to_integral_value(rounding=ROUND_DOWN)
+    if raw <= 0:
+        raise ValueError("Amount is too small after decimal scaling")
+    return int(raw)
+
+
+def sha256_json(obj: Any) -> str:
+    payload = json.dumps(obj, sort_keys=True, separators=(",", ":"), ensure_ascii=False)
+    return "sha256:" + hashlib.sha256(payload.encode("utf-8")).hexdigest()

wayfinder_paths/policies/enso.py

@@ -1,7 +1,6 @@
+from wayfinder_paths.core.constants.contracts import ENSO_ROUTER
 from wayfinder_paths.policies.util import allow_functions
 
-ENSO_ROUTER = "0xf75584ef6673ad213a685a1b58cc0330b8ea22cf"
-
 
 async def enso_swap():
     return await allow_functions(

wayfinder_paths/policies/hyper_evm.py

@@ -1,9 +1,12 @@
+from wayfinder_paths.core.constants.contracts import (
+    HYPERCORE_SENTINEL_ADDRESS,
+    HYPERCORE_SENTINEL_VALUE,
+    HYPEREVM_WHYPE,
+)
 from wayfinder_paths.policies.evm import native_transfer
 from wayfinder_paths.policies.util import allow_functions
 
-WHYPE_TOKEN = "0x5555555555555555555555555555555555555555"
-HYPERCORE_SENTINEL_ADDRESS = "0x2222222222222222222222222222222222222222"
-HYPERCORE_SENTINEL_VALUE = 100_000_000_000
+WHYPE_TOKEN = HYPEREVM_WHYPE
 
 
 def hypecore_sentinel_deposit():

wayfinder_paths/policies/hyperlend.py

@@ -1,7 +1,6 @@
+from wayfinder_paths.core.constants.contracts import HYPERLEND_POOL
 from wayfinder_paths.policies.util import allow_functions
 
-HYPERLEND_POOL = "0x00A89d7a5A02160f20150EbEA7a2b5E4879A1A8b"
-
 
 async def hyperlend_supply_and_withdraw():
     return await allow_functions(

wayfinder_paths/policies/hyperliquid.py

@@ -24,7 +24,7 @@ def any_hyperliquid_user_payload():
                 "field": "chainId",
                 "field_source": "ethereum_typed_data_domain",
                 "operator": "eq",
-                "value": "421614",
+                "value": "42161",
             }
         ],
     }

wayfinder_paths/policies/lifi.py

@@ -0,0 +1,18 @@
+from wayfinder_paths.policies.util import allow_functions
+
+LIFI_ROUTERS: dict[int:str] = {999: "0x0a0758d937d1059c356D4714e57F5df0239bce1A"}
+LIFI_GENERIC = "0x31a9b1835864706Af10103b31Ea2b79bdb995F5F"
+
+
+async def lifi_swap(chain_id):
+    # NOTE: we get the abi from the base contract as it is a generic abi used everywhere
+    # and not all chains have this ABI published
+    return await allow_functions(
+        policy_name="Allow LIFI Swap",
+        abi_chain_id=8453,
+        address=LIFI_ROUTERS[chain_id],
+        function_names=[
+            "swapTokensMultipleV3ERC20ToERC20",
+        ],
+        abi_address_override=LIFI_GENERIC,
+    )

wayfinder_paths/policies/moonwell.py

@@ -1,12 +1,17 @@
+from wayfinder_paths.core.constants.contracts import (
+    BASE_WETH,
+    MOONWELL_COMPTROLLER,
+    MOONWELL_M_USDC,
+    MOONWELL_M_WETH,
+    MOONWELL_M_WSTETH,
+)
 from wayfinder_paths.policies.util import allow_functions
 
-WETH = "0x4200000000000000000000000000000000000006"
-
-M_USDC = "0xEdc817A28E8B93B03976FBd4a3dDBc9f7D176c22"
-M_WETH = "0x628ff693426583D9a7FB391E54366292F509D457"
-M_WSTETH = "0x627Fe393Bc6EdDA28e99AE648fD6fF362514304b"
-
-COMPTROLLER = "0xfbb21d0380bee3312b33c4353c8936a0f13ef26c"
+WETH = BASE_WETH
+M_USDC = MOONWELL_M_USDC
+M_WETH = MOONWELL_M_WETH
+M_WSTETH = MOONWELL_M_WSTETH
+COMPTROLLER = MOONWELL_COMPTROLLER
 
 
 async def weth_deposit():

wayfinder_paths/policies/prjx.py

@@ -1,8 +1,6 @@
+from wayfinder_paths.core.constants.contracts import PRJX_NPM, PRJX_ROUTER
 from wayfinder_paths.policies.util import allow_functions
 
-PRJX_ROUTER = "0x1ebdfc75ffe3ba3de61e7138a3e8706ac841af9b"
-PRJX_NPM = "0xeAd19AE861c29bBb2101E834922B2FEee69B9091"
-
 
 async def prjx_swap():
     return await allow_functions(

wayfinder_paths/policies/util.py

@@ -2,7 +2,11 @@ from wayfinder_paths.core.utils.evm_helpers import get_abi_filtered
 
 
 async def allow_functions(
-    policy_name: str, abi_chain_id: int, address: str, function_names: list[str]
+    policy_name: str,
+    abi_chain_id: int,
+    address: str,
+    function_names: list[str],
+    abi_address_override=None,
 ):
     # Note: ChainID is just for fetching ABI, doesn't appear in the final policy. Doesn't bind a strict chain.
     return {
@@ -19,7 +23,9 @@ async def allow_functions(
             {
                 "field_source": "ethereum_calldata",
                 "field": "function_name",
-                "abi": await get_abi_filtered(abi_chain_id, address, function_names),
+                "abi": await get_abi_filtered(
+                    abi_chain_id, abi_address_override or address, function_names
+                ),
                 "operator": "in",
                 "value": function_names,
             },