axis-synome 0.1.0.dev59__tar.gz → 0.1.0.dev60__tar.gz

{axis_synome-0.1.0.dev59 → axis_synome-0.1.0.dev60}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: axis-synome
-Version: 0.1.0.dev59
+Version: 0.1.0.dev60
 Summary: Axis specification modules (entities, formulas, validators)
 Author-email: Archon Tech <hello@archontech.ai>
 License: MIT

{axis_synome-0.1.0.dev59 → axis_synome-0.1.0.dev60}/pyproject.toml

@@ -24,6 +24,9 @@ runtime = [
 Repository = "https://github.com/archon-research/next-gen-atlas"
 Documentation = "https://github.com/archon-research/next-gen-atlas"
 
+[project.scripts]
+export_entities = "axis_synome.export_entities:main"
+
 [project.entry-points."flake8.extension"]
 AXS = "axis_synome.spec_validator.flake8_plugin:SpecSubsetChecker"
 

{axis_synome-0.1.0.dev59 → axis_synome-0.1.0.dev60}/src/axis_synome/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.1.0.dev59'
-__version_tuple__ = version_tuple = (0, 1, 0, 'dev59')
+__version__ = version = '0.1.0.dev60'
+__version_tuple__ = version_tuple = (0, 1, 0, 'dev60')
 
 __commit_id__ = commit_id = None

axis_synome-0.1.0.dev60/src/axis_synome/export_entities.py

@@ -0,0 +1,233 @@
+from __future__ import annotations
+
+import argparse
+import importlib.metadata
+import json
+import re
+import subprocess
+from pathlib import Path
+from typing import Any
+
+from pydantic import BaseModel, ConfigDict
+
+from axis_synome.spec.asc.entities.alm_proxies import (
+    STL_ALM_PROXY_ADDRESS_OVERRIDES,
+    AlmProxy,
+)
+from axis_synome.spec.asc.entities.assets_by_prime import ASSETS_BY_PRIME, PrimeName
+from axis_synome.spec.asc.entities.networks import STL_CHAIN_BY_NETWORK
+from axis_synome.spec.asc.entities.protocol_sets import (
+    ATOKEN_PROTOCOLS,
+    ERC4626_PROTOCOLS,
+    L1_PSM_PROTOCOLS,
+    L2_PSM_PROTOCOLS,
+    UNISWAP_STYLE_PROTOCOLS,
+    Protocol,
+)
+
+
+class AxisSynomeContract(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    version: str
+    axis_synome_git_commit: str
+    axis_synome: dict[str, Any]
+
+
+def _slug(text: str) -> str:
+    return re.sub(r"[^a-z0-9]+", "-", text.lower()).strip("-")
+
+
+def _prime_to_star(prime: PrimeName | str) -> str:
+    text = prime.value if isinstance(prime, PrimeName) else str(prime)
+    return text.lower().replace(" ", "-")
+
+
+def _token_type(protocol: Protocol, chain: str, token_name: str) -> str:
+    if protocol in ATOKEN_PROTOCOLS:
+        return "atoken"
+
+    if protocol in ERC4626_PROTOCOLS:
+        return "erc4626"
+
+    if protocol == Protocol.CURVE:
+        return "curve"
+
+    if protocol in UNISWAP_STYLE_PROTOCOLS:
+        if "LP" in token_name.upper():
+            return "uni_v3_lp"
+        return "uni_v3_pool"
+
+    if protocol == Protocol.CENTRIFUGE:
+        if chain == "plume":
+            return "centrifuge_feeder"
+        return "centrifuge"
+
+    if protocol == Protocol.ANCHORAGE:
+        return "anchorage"
+
+    if protocol in L1_PSM_PROTOCOLS | L2_PSM_PROTOCOLS:
+        return "psm3"
+
+    return "erc20"
+
+
+def _allocation_type(protocol: Protocol) -> str:
+    if protocol in L1_PSM_PROTOCOLS | L2_PSM_PROTOCOLS:
+        return "psm3"
+    return "allocation"
+
+
+def _version_from_package_metadata() -> str:
+    return importlib.metadata.version("axis-synome")
+
+
+def _axis_synome_git_commit() -> str:
+    repo_root = Path(__file__).resolve().parents[2]
+    try:
+        return subprocess.check_output(
+            ["git", "rev-parse", "HEAD"], cwd=repo_root, text=True
+        ).strip()
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        return "unknown"
+
+
+def build_axis_synome_contract(
+    version: str | None = None,
+    axis_synome_git_commit: str | None = None,
+) -> AxisSynomeContract:
+    proxy_lookup: dict[tuple[str, str], str] = {}
+
+    for member in AlmProxy:
+        deployment = member.value
+        chain = STL_CHAIN_BY_NETWORK.get(deployment.network)
+        if chain is None:
+            continue
+
+        star = _prime_to_star(deployment.prime.name)
+        proxy_lookup[(star, chain)] = str(deployment.address).lower()
+
+    for (prime, network), address in STL_ALM_PROXY_ADDRESS_OVERRIDES.items():
+        chain = STL_CHAIN_BY_NETWORK.get(network)
+        if chain is None:
+            continue
+
+        star = _prime_to_star(prime.name)
+        # Explicit overrides for STL compatibility when these keys already exist.
+        proxy_lookup[(star, chain)] = str(address).lower()
+
+    entries: list[dict[str, Any]] = []
+    for prime_name, assets in ASSETS_BY_PRIME.items():
+        star = _prime_to_star(prime_name)
+        if star not in {"spark", "grove"}:
+            continue
+
+        for asset in assets:
+            chain = STL_CHAIN_BY_NETWORK.get(asset.network)
+            if chain is None:
+                continue
+
+            wallet = proxy_lookup.get((star, chain))
+            if wallet is None:
+                continue
+
+            entries.append({
+                "contract_address": str(asset.address).lower(),
+                "wallet_address": wallet,
+                "asset_address": str(asset.underlying_asset_address).lower(),
+                "star": star,
+                "chain": chain,
+                "protocol": _slug(asset.protocol.value),
+                "allocation_type": _allocation_type(asset.protocol),
+                "token_type": _token_type(asset.protocol, chain, asset.token.value),
+                "created_at_block": None,
+            })
+
+    assets_by_prime_export: dict[str, list[dict[str, Any]]] = {}
+    for entry in entries:
+        assets_by_prime_export.setdefault(entry["star"], []).append(entry)
+
+    alm_proxy_export: dict[str, dict[str, dict[str, str]]] = {}
+    for (star, chain), address in proxy_lookup.items():
+        alm_proxy_export.setdefault(star, {})[chain] = {
+            "star": star,
+            "chain": chain,
+            "address": address,
+        }
+
+    return AxisSynomeContract(
+        version=version or _version_from_package_metadata(),
+        axis_synome_git_commit=axis_synome_git_commit or _axis_synome_git_commit(),
+        axis_synome={
+            "spec": {
+                "asc": {
+                    "entities": {
+                        "assets_by_prime": {"ASSETS_BY_PRIME": assets_by_prime_export},
+                        "alm_proxies": {"AlmProxy": alm_proxy_export},
+                    }
+                }
+            }
+        },
+    )
+
+
+def export_axis_synome_contract(
+    data_path: Path,
+    schema_path: Path,
+    version: str | None,
+    axis_synome_git_commit: str | None,
+) -> None:
+    config = build_axis_synome_contract(
+        version=version,
+        axis_synome_git_commit=axis_synome_git_commit,
+    )
+
+    data_path.parent.mkdir(parents=True, exist_ok=True)
+    schema_path.parent.mkdir(parents=True, exist_ok=True)
+
+    data_path.write_text(config.model_dump_json(indent=2), encoding="utf-8")
+
+    schema = AxisSynomeContract.model_json_schema()
+    schema_path.write_text(json.dumps(schema, indent=2), encoding="utf-8")
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(
+        description="Export axis-synome entities contract JSON and pydantic schema."
+    )
+    parser.add_argument(
+        "--out",
+        type=Path,
+        default=Path("generated/stl/axis_synome_entities.json"),
+        help="Output JSON data path",
+    )
+    parser.add_argument(
+        "--schema-out",
+        type=Path,
+        default=Path("generated/stl/axis_synome_entities.schema.json"),
+        help="Output JSON schema path",
+    )
+    parser.add_argument(
+        "--version",
+        type=str,
+        default=None,
+        help="Version string embedded in exported config (default resolves from pyproject metadata)",
+    )
+    parser.add_argument(
+        "--git-commit",
+        type=str,
+        default=None,
+        help="axis-synome git commit hash embedded in exported config (default resolves from git)",
+    )
+
+    args = parser.parse_args()
+    export_axis_synome_contract(
+        args.out,
+        args.schema_out,
+        version=args.version,
+        axis_synome_git_commit=args.git_commit,
+    )
+
+
+if __name__ == "__main__":
+    main()

{axis_synome-0.1.0.dev59 → axis_synome-0.1.0.dev60}/src/axis_synome/spec/asc/entities/alm_proxies.py

@@ -207,3 +207,10 @@ class AlmProxy(Enum):
     GROVE_PLASMA = GROVE_PLASMA_ALM_PROXY
     GROVE_PLUME = GROVE_PLUME_ALM_PROXY
     GROVE_MONAD = GROVE_MONAD_ALM_PROXY
+
+
+# Temporary STL compatibility overrides while tracker config migrates.
+STL_ALM_PROXY_ADDRESS_OVERRIDES: Final[dict[tuple[PrimeAgent, Network], str]] = {
+    (PrimeAgent.SPARK, Network.ETHEREUM_MAINNET): "0x3300f198988e4c9c63f75df86de36421f06af8c4",
+    (PrimeAgent.GROVE, Network.ETHEREUM_MAINNET): "0x1369f7b2b38c76b6478c0f0e66d94923421891ba",
+}

{axis_synome-0.1.0.dev59 → axis_synome-0.1.0.dev60}/src/axis_synome/spec/asc/entities/networks.py

@@ -15,3 +15,17 @@ class Network(Enum):
 
 # L2 networks: all networks that are not Ethereum mainnet.
 L2_NETWORKS = {Network.BASE, Network.ARBITRUM, Network.OPTIMISM, Network.UNICHAIN}
+
+
+# Chain labels consumed by STL compatibility checks.
+STL_CHAIN_BY_NETWORK = {
+    Network.ETHEREUM_MAINNET: "mainnet",
+    Network.BASE: "base",
+    Network.ARBITRUM: "arbitrum",
+    Network.OPTIMISM: "optimism",
+    Network.UNICHAIN: "unichain",
+    Network.AVALANCHE: "avalanche-c",
+    Network.PLASMA: "plasma",
+    Network.PLUME: "plume",
+    Network.MONAD: "monad",
+}

{axis_synome-0.1.0.dev59 → axis_synome-0.1.0.dev60}/src/axis_synome/spec/asc/entities/protocol_sets.py

@@ -80,3 +80,32 @@ POOL_PROTOCOLS = {Protocol.CURVE, Protocol.UNISWAP}
 GUNI_PROTOCOLS = {Protocol.GUNI}
 
 GUNI_RESTING_FEE_TIERS = {0.0001, 0.0005}
+
+
+# STL exporter protocol classifications.
+ATOKEN_PROTOCOLS = {
+    Protocol.AAVE,
+    Protocol.AAVE_V2,
+    Protocol.AAVE_V3,
+    Protocol.AAVE_CORE,
+    Protocol.AAVE_CORE_V3,
+    Protocol.AAVE_PRIME,
+    Protocol.AAVE_HORIZON,
+    Protocol.SPARKLEND,
+    Protocol.SPARKLEND_PROTOCOL,
+}
+
+ERC4626_PROTOCOLS = {
+    Protocol.MORPHO,
+    Protocol.MORPHO_BLUE_ERC4626_VAULT,
+    Protocol.MAPLE,
+    Protocol.FLUID,
+    Protocol.FLUID_FINANCE_ERC4626_VAULT,
+    Protocol.SPARK_SAVINGS_PROTOCOL,
+    Protocol.STEAKHOUSE_PYUSD_MORPHO_VAULT,
+    Protocol.GROVE_STEAKHOUSE_AUSD_MORPHO_VAULT,
+    Protocol.GROVE_STEAKHOUSE_USDC_HIGH_YIELD_VAULT_V2,
+    Protocol.ARKIS,
+}
+
+UNISWAP_STYLE_PROTOCOLS = {Protocol.UNISWAP, Protocol.UNISWAP_V3, Protocol.GUNI}

{axis_synome-0.1.0.dev59 → axis_synome-0.1.0.dev60}/src/axis_synome.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: axis-synome
-Version: 0.1.0.dev59
+Version: 0.1.0.dev60
 Summary: Axis specification modules (entities, formulas, validators)
 Author-email: Archon Tech <hello@archontech.ai>
 License: MIT

{axis_synome-0.1.0.dev59 → axis_synome-0.1.0.dev60}/src/axis_synome.egg-info/SOURCES.txt

@@ -5,6 +5,7 @@ pyproject.toml
 uv.lock
 src/axis_synome/__init__.py
 src/axis_synome/_version.py
+src/axis_synome/export_entities.py
 src/axis_synome.egg-info/PKG-INFO
 src/axis_synome.egg-info/SOURCES.txt
 src/axis_synome.egg-info/dependency_links.txt
@@ -115,6 +116,7 @@ tests/axis_synome/risk_capital/formulas/test_loss_given_default.py
 tests/axis_synome/risk_capital/smart_contract_rrc/__init__.py
 tests/axis_synome/risk_capital/smart_contract_rrc/test_exceptions.py
 tests/axis_synome/risk_capital/smart_contract_rrc/test_smart_contract_risk.py
+tests/axis_synome/scripts/test_export_entities.py
 tests/axis_synome/spec_support/__init__.py
 tests/axis_synome/spec_support/references/__init__.py
 tests/axis_synome/spec_support/references/test_graph.py

{axis_synome-0.1.0.dev59 → axis_synome-0.1.0.dev60}/src/axis_synome.egg-info/entry_points.txt

@@ -1,2 +1,5 @@
+[console_scripts]
+export_entities = axis_synome.export_entities:main
+
 [flake8.extension]
 AXS = axis_synome.spec_validator.flake8_plugin:SpecSubsetChecker

axis_synome-0.1.0.dev60/tests/axis_synome/scripts/test_export_entities.py

@@ -0,0 +1,76 @@
+import json
+import subprocess
+import sys
+from pathlib import Path
+
+
+def _run_export(repo_root: Path, out_path: Path, schema_path: Path) -> None:
+    subprocess.run(
+        [
+            sys.executable,
+            "-m",
+            "axis_synome.export_entities",
+            "--out",
+            str(out_path),
+            "--schema-out",
+            str(schema_path),
+            "--version",
+            "test-version",
+            "--git-commit",
+            "test-commit",
+        ],
+        check=True,
+        cwd=repo_root / "axis-synome",
+    )
+
+
+def test_export_entities_outputs_contract_and_schema(tmp_path: Path):
+    repo_root = Path(__file__).resolve().parents[4]
+    out_path = tmp_path / "axis_synome_entities.json"
+    schema_path = tmp_path / "axis_synome_entities.schema.json"
+
+    _run_export(repo_root, out_path, schema_path)
+
+    contract = json.loads(out_path.read_text(encoding="utf-8"))
+    schema = json.loads(schema_path.read_text(encoding="utf-8"))
+
+    assert contract["version"] == "test-version"
+    assert contract["axis_synome_git_commit"] == "test-commit"
+    assert schema["title"] == "AxisSynomeContract"
+
+
+def test_export_entities_applies_extra_proxy_overrides(tmp_path: Path):
+    repo_root = Path(__file__).resolve().parents[4]
+    out_path = tmp_path / "axis_synome_entities.json"
+    schema_path = tmp_path / "axis_synome_entities.schema.json"
+
+    _run_export(repo_root, out_path, schema_path)
+
+    contract = json.loads(out_path.read_text(encoding="utf-8"))
+    entities = contract["axis_synome"]["spec"]["asc"]["entities"]
+    proxies = entities["alm_proxies"]["AlmProxy"]
+
+    assert proxies["spark"]["mainnet"]["address"] == "0x3300f198988e4c9c63f75df86de36421f06af8c4"
+    assert proxies["grove"]["mainnet"]["address"] == "0x1369f7b2b38c76b6478c0f0e66d94923421891ba"
+
+    spark_mainnet_entries = [
+        entry
+        for entry in entities["assets_by_prime"]["ASSETS_BY_PRIME"]["spark"]
+        if entry["chain"] == "mainnet"
+    ]
+    grove_mainnet_entries = [
+        entry
+        for entry in entities["assets_by_prime"]["ASSETS_BY_PRIME"]["grove"]
+        if entry["chain"] == "mainnet"
+    ]
+
+    assert spark_mainnet_entries
+    assert grove_mainnet_entries
+    assert all(
+        entry["wallet_address"] == "0x3300f198988e4c9c63f75df86de36421f06af8c4"
+        for entry in spark_mainnet_entries
+    )
+    assert all(
+        entry["wallet_address"] == "0x1369f7b2b38c76b6478c0f0e66d94923421891ba"
+        for entry in grove_mainnet_entries
+    )