from coordinator import ProtocolAdapter, Command, AssetState

class MyCustomAdapter(ProtocolAdapter):
    def send_command(self, command: Command) -> bool:
        """Translate and send. Return True on success."""
        payload = self._translate(command)
        return self._send(payload)

    def read_state(self, asset_id: str) -> AssetState:
        """Read current device state."""
        raw = self._read_registers(asset_id)
        return self._parse_state(raw)

from coordinator import OCPPAdapter

adapter = OCPPAdapter("ocpp_site_1")

from coordinator.base import Command, CommandType

cmd = Command(
    asset_id="ast_charger_001",
    command_type=CommandType.SET_CHARGE_RATE,
    parameters={"power_kw": 15.0, "connector_id": 1},
    scheduled_at=now
)

adapter.send_command(cmd)

# After send_command, the OCPP payload is attached:
print(cmd.parameters["ocpp_action"])
# "SetChargingProfile"

print(cmd.parameters["ocpp_payload"])
# {
#   "connectorId": 1,
#   "csChargingProfiles": {
#     "chargingSchedule": {
#       "chargingRateUnit": "W",
#       "chargingSchedulePeriod": [{"startPeriod": 0, "limit": 15000.0}]
#     }
#   }
# }

from coordinator import ModbusAdapter

# Default register map
adapter = ModbusAdapter("modbus_bms_1")

# Custom register map for specific BMS
adapter = ModbusAdapter("modbus_bms_1", register_map={
    "charge_rate_kw": 30001,
    "discharge_rate_kw": 30002,
    "power_limit_kw": 30003,
    "soc_target_percent": 30004,
})

cmd = Command(
    asset_id="ast_batt_001",
    command_type=CommandType.SET_DISCHARGE_RATE,
    parameters={"power_kw": 50.0},
    scheduled_at=now
)

adapter.send_command(cmd)

print(cmd.parameters["modbus_writes"])
# [(40002, 50.0)]

from coordinator import AdapterRegistry, OCPPAdapter, ModbusAdapter

registry = AdapterRegistry()

# Register adapters
registry.register("ast_charger_001", OCPPAdapter("ocpp_1"))
registry.register("ast_charger_002", OCPPAdapter("ocpp_1"))
registry.register("ast_batt_001", ModbusAdapter("modbus_1"))

# Look up by asset ID
adapter = registry.get("ast_charger_001")

# Check registration
"ast_charger_001" in registry  # True
len(registry)                   # 3

# Remove
registry.unregister("ast_charger_002")

registry.get("ast_unknown")
# KeyError: "No protocol adapter registered for asset 'ast_unknown'.
#            Registered assets: ['ast_charger_001', 'ast_batt_001']"

from coordinator import ProtocolAdapter, Command, AssetState, AssetType

class DNP3Adapter(ProtocolAdapter):
    def __init__(self, adapter_id: str, outstation_address: int):
        super().__init__(adapter_id)
        self.outstation_address = outstation_address

    def send_command(self, command: Command) -> bool:
        # Translate to DNP3 direct operate
        dnp3_payload = {
            "function": "DIRECT_OPERATE",
            "address": self.outstation_address,
            "index": self._command_to_index(command.command_type),
            "value": command.parameters.get("power_kw", 0),
        }
        command.parameters["dnp3_payload"] = dnp3_payload
        return True

    def read_state(self, asset_id: str) -> AssetState:
        return AssetState(
            asset_id=asset_id,
            asset_type=AssetType.INVERTER,
            current_state="RUNNING",
        )

# Register it
registry.register("ast_inverter_001", DNP3Adapter("dnp3_1", outstation_address=10))