from __future__ import annotations

from typing import List

from deltaforge_skeleton.core import PlanDelta


class ADMMCoordinator:
    """Lightweight, production-friendly ADMM-lite coordinator for cross-venue coherence.

    This is intentionally tiny: it augments an existing PlanDelta with a detectable
    ADMM coordination step. It is safe to disable or remove this in environments
    that do not require cross-venue coordination.
    """

    def __init__(self, rho: float = 1.0):
        self.rho = rho

    def coordinate(self, plan: PlanDelta, venues: List[str]) -> PlanDelta:
        if not plan or not plan.steps:
            return plan
        # Minimal ADMM-like annotation to indicate coordination happened
        venues_str = ",".join(venues) if venues else "[]"
        plan.steps.append(f"ADMM_COORDINATE between [{venues_str}] with rho={self.rho}")
        return plan