import pandas as pd
import subprocess
import argparse
import sys
import json

# This Python script will read an Excel file containing a list of devices for a site (using --serials *file name*)
# It will then create a IoT Edge device for each of the line in the file
# The user need to precise which IoT Hub is used (using --env *PROD or DEV*)
# The user need provide the site name, which will be added as a tag (using --site *site name*)
# The user need to precise which configuration of the modules should be used, according to the running firmware on the devices (using --version *version number*)
# Example:
# python create_devices_list.py --serials DANISH.xlsx --env PROD --site DANISH --version 1.5_patch
# will create all the IoT Edge devices using the serial number found in the DANISH.xlsx file
#   on the PROD IoT Hub
#   tag each new device with a site = DANISH
#   configure the modules for each devices with the corresponding version found in moxa_ac_template_1.5_patch.json
#   each device will have also be given a number = i where i is incremented automatically according to the number of device in the Excel file
#   the provided Excel file will be modified, adding a column connection_string for each of the devices created with their corresponding connection string
#   the output is therefore SENSITIVE

def generate_commands(serials_file, env, site, version):
    # Read serial numbers from Excel file
    df = pd.read_excel(serials_file)
    df = df[df['device_name'].notnull()]
    serials = df['device_name'].tolist()

    # Initialize number
    number = 1

    # List to store connection strings
    connection_strings = []

    # Generate commands for each serial number
    for serial in serials:
        # Replace placeholders with actual values
        print(serial, end=" ")
        device_id = f"DIGIT-{serial}"
        print(device_id, end=" ")
        tags = f'{{"deviceId":"{device_id}","site":"{site}","number":"{number}"}}'
        content = f"moxa_ac_template_{version}.json"
        # Construct command strings
        create_command = f"az iot hub device-identity create --device-id {device_id} --hub-name IotHub-CUBE-{env} --edge-enabled"
        twin_update_command = f"az iot hub device-twin update --device-id {device_id} --hub-name IotHub-CUBE-{env} --set tags='{tags}'"
        set_modules_command = f"az iot edge set-modules --device-id {device_id} --hub-name IotHub-CUBE-{env} --content {content}"

        # Execute create command and get primary key
        create_output = subprocess.check_output(create_command, shell=True)
        create_output_json = json.loads(create_output.decode('utf-8'))
        primary_key = create_output_json['authentication']['symmetricKey']['primaryKey']
        print(primary_key, end=" ")
        
        # Generate connection string
        connection_string = f"HostName=IotHub-CUBE-{env}.azure-devices.net;DeviceId={device_id};SharedAccessKey={primary_key}"
        print(connection_string)
        connection_strings.append(connection_string)
        
        tags_output = subprocess.run(twin_update_command, shell=True)
        modules_output = subprocess.run(set_modules_command, shell=True)
        # Increment number
        number += 1

    # Add connection strings to DataFrame
    df['connection_string'] = connection_strings

    # Save DataFrame to Excel file
    df.to_excel(serials_file, index=False)

if __name__ == "__main__":
    # Parse command line arguments
    parser = argparse.ArgumentParser(description='Create devices list')
    parser.add_argument('--serials', required=True, help='Excel file containing serial numbers')
    parser.add_argument('--env', required=True, help='Environment (PROD or DEV)')
    parser.add_argument('--site', required=True, help='Site name')
    parser.add_argument('--version', required=True, help='Version number')
    args = parser.parse_args()

    if len(sys.argv) == 1:
        parser.print_help()
        sys.exit(1)

    # Generate commands
    generate_commands(args.serials, args.env, args.site, args.version)