import asyncio
import logging

# Uncomment lines below to use library from local dev
import sys
import os

sys.path.insert(1, os.path.join(os.path.dirname(__file__), ".."))

from aiorussound.rio import RussoundRIOClient
from aiorussound import RussoundTcpConnectionHandler
from aiorussound.rio.models import CallbackType

HOST = "192.168.20.17"
PORT = 4999


async def on_state_change(client: RussoundRIOClient, callback_type: CallbackType):
    """Called when new information is received."""
    print(f"Callback Type: {callback_type} {client.is_connected()}")
    print(client.controllers[1].zones[1].status)


async def main():
    """Subscribe demo entrypoint."""
    conn_handler = RussoundTcpConnectionHandler(HOST, PORT)
    client = RussoundRIOClient(conn_handler)

    await client.register_state_update_callbacks(on_state_change)
    await client.connect()
    await client.load_zone_source_metadata()

    for s_id, source in client.sources.items():
        print(f"Found source {s_id} - {source.name}")

    for c_id, controller in client.controllers.items():
        print(f"Found controller {c_id} - {controller.mac_address}")
        for z_id, zone in controller.zones.items():
            print(f"Found zone {z_id} - {zone.name}")
    print(client.state)

    # Play media using the unit's front controls or Russound app
    await asyncio.sleep(30)
    await client.disconnect()


if __name__ == "__main__":
    logging.basicConfig(level=logging.DEBUG)
    asyncio.run(main())
