import argparse
from hoymiles_wifi.inverter import Inverter
import time

def get_real_data_new(inverter):
    return inverter.get_real_data_new()

def get_real_data_hms(inverter):
    return inverter.get_real_data_hms()

def get_real_data(inverter):
    return inverter.get_real_data()

def get_config(inverter):
    return inverter.get_config()

def network_info(inverter):
    return inverter.network_info()

def app_information_data(inverter):
    return inverter.app_information_data()

def app_get_hist_power(inverter):
    return inverter.app_get_hist_power()

def set_power_limit(inverter):

    RED = '\033[91m'
    END = '\033[0m'

    print(RED + "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! WARNING !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" + END)
    print(RED + "!!! Danger zone! This will change the power limit of the inverter. !!!" + END)
    print(RED + "!!!  Please be careful and make sure you know what you are doing.  !!!" + END)
    print(RED + "!!!          Only proceed if you know what you are doing.          !!!" + END)
    print(RED + "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! WARNING !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" + END)
    print("")


    cont = input("Do you want to continue? (y/n): ")
    if(cont != 'y'):
        return

    power_limit = int(input("Enter the new power limit (0-100): "))

    if(power_limit < 0 or power_limit > 100):
        print("Error. Invalid power limit!")
        return
    
    print(f"Setting power limit to {power_limit}%")
    cont = input("Are you sure? (y/n): ") 

    if(cont != 'y'):
        return

    return inverter.set_power_limit(power_limit)


def set_wifi(inverter):
    wifi_ssid = input("Enter the new wifi SSID: ").strip()
    wifi_password = input("Enter the new wifi password: ").strip()
    print(f'Setting wifi to "{wifi_ssid}"')
    print(f'Setting wifi password to "{wifi_password}"')
    cont = input("Are you sure? (y/n): ")
    if(cont != 'y'):
        return
    return inverter.set_wifi(wifi_ssid, wifi_password)

def print_invalid_command(command):
    print(f"Invalid command: {command}")

def main():
    parser = argparse.ArgumentParser(description="Hoymiles HMS Monitoring")
    parser.add_argument("--host", type=str, required=True, help="IP address or hostname of the inverter")
    parser.add_argument("command", type=str, choices=['get-real-data-new', 'get-real-data-hms', 'get-real-data', 'get-config', 'network-info', 'app-information-data', 'app-get-hist-power', 'set-power-limit', 'set-wifi'], help="Command to execute")
    args = parser.parse_args()

    inverter = Inverter(args.host)

    # Execute the specified command using a switch case
    switch = {
        'get-real-data-new': get_real_data_new,
        'get-real-data-hms': get_real_data_hms,
        'get-real-data': get_real_data,
        'get-config': get_config,
        'network-info': network_info,
        'app-information-data': app_information_data,
        'app-get-hist-power': app_get_hist_power,
        'set-power-limit': set_power_limit,
        'set-wifi': set_wifi,
    }

    command_func = switch.get(args.command, print_invalid_command)
    response = command_func(inverter)

    if response:
        print(f"{args.command.capitalize()} Response: {response}")
    else:
        print(f"Unable to retrieve {args.command.replace('_', ' ')}")


if __name__ == "__main__":
    main()