import asyncio
import logging
import os
from pprint import pprint

from aiohttp import ClientResponse

from pyenphase.envoy import Envoy

logging.basicConfig(level=logging.DEBUG)


async def main() -> None:
    envoy = Envoy(os.environ.get("ENVOY_HOST", "envoy.local"))

    await envoy.setup()

    username = os.environ.get("ENVOY_USERNAME")
    password = os.environ.get("ENVOY_PASSWORD")
    token = os.environ.get("ENVOY_TOKEN")

    await envoy.authenticate(username=username, password=password, token=token)

    # Test https://enphase.com/download/iq-gateway-access-using-local-apis-or-local-ui-token-based-authentication-tech-brief endpoints

    print(await envoy.update())

    end_points = [
        "/ivp/livedata/status",
        "/api/v1/production",
        "/api/v1/production/inverters",
        "/production.json",
        "/production",
        "/ivp/meters",
        "/ivp/meters/readings",
        "/ivp/meters/reports/consumption",
        "/ivp/ensemble/inventory",
        "/ivp/ensemble/dry_contacts",
        "/ivp/ss/dry_contact_settings",
        "/ivp/pdm/device_data",
    ]

    for end_point in end_points:
        try:
            response: ClientResponse = await envoy.request(end_point)
        except Exception as e:
            print(e)
            continue
        json_dict = await response.json()
        print((end_point, "=" * 80))
        pprint(json_dict)
        print((end_point, "=" * 80))


asyncio.run(main())