from azure.iot.hub import IoTHubRegistryManager
from azure.iot.hub.protocol.models import QuerySpecification
from azure.iot.hub.models import CloudToDeviceMethod, CloudToDeviceMethodResult
from dotenv import load_dotenv

import json
import os

load_dotenv()

# Authenticate to your Azure account
# CONNECTION_STRING = str(os.getenv("CONNECTION_STRING_SAFT_PROD"))
CONNECTION_STRING = str(os.getenv("CONNECTION_STRING_INOX_PROD"))
if CONNECTION_STRING == "":
    print("Provide a connection string for the Iot Hub before running the script!")
    exit(13)

SITE_NAME = "LIBERTY"


registry_manager = IoTHubRegistryManager.from_connection_string(CONNECTION_STRING)
query_spec = QuerySpecification(query="SELECT * FROM devices WHERE IS_DEFINED(tags.site) AND tags.site = '" + SITE_NAME + "' AND capabilities.iotEdge = true")

query_result = registry_manager.query_iot_hub(query_spec)

devices = []
for item in query_result.items:
    deviceId = str(item.device_id)
    site = str(item.tags['site'])
    number = int(item.tags['number'])
    cloud_version = str(item.tags['version'])
    devices.append([deviceId, site, number, cloud_version])

ordered_devices = sorted(devices, key = lambda x: (x[1], x[2]))

for i in ordered_devices:
    device_info = registry_manager.get_device(i[0])
    primary_key = device_info.authentication.symmetric_key.primary_key
    connection_string = f"HostName={CONNECTION_STRING.split(';')[0].split('=')[1]};DeviceId={i[0]};SharedAccessKey={primary_key}"
    print(i[1], i[2], i[0])
    print(connection_string)