# coding=utf-8
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# --------------------------------------------------------------------------------------------

"""
Load CLI commands
"""

from azext_iot import (
    iothub_ops,
    iotdps_ops,
)


def load_command_table(self, _):
    """
    Load CLI commands
    """
    with self.command_group("iot hub", command_type=iothub_ops) as cmd_group:
        cmd_group.command("query", "iot_query")
        cmd_group.command("invoke-device-method", "iot_device_method")
        cmd_group.command("invoke-module-method", "iot_device_module_method")
        cmd_group.command("generate-sas-token", "iot_get_sas_token")
        cmd_group.command("monitor-events", "iot_hub_monitor_events")
        cmd_group.command("monitor-feedback", "iot_hub_monitor_feedback")

    with self.command_group(
        "iot hub device-identity", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.command("create", "iot_device_create")
        cmd_group.show_command("show", "iot_device_show")
        cmd_group.command(
            "list",
            "iot_device_twin_list",
            # deprecate_info=self.deprecate(redirect='iot device-twin list')
        )
        cmd_group.command("delete", "iot_device_delete")
        cmd_group.generic_update_command(
            "update",
            getter_name="iot_device_show",
            custom_func_type=iothub_ops,
            setter_name="iot_device_update",
            custom_func_name="update_iot_device_custom"
        )
        cmd_group.command("renew-key", "iot_device_key_regenerate")
        cmd_group.command("import", "iot_device_import")
        cmd_group.command("export", "iot_device_export")

    with self.command_group(
        "iot hub device-identity children", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.show_command("add", "iot_device_children_add")
        cmd_group.show_command("remove", "iot_device_children_remove")
        cmd_group.show_command("list", "iot_device_children_list")

    with self.command_group(
        "iot hub device-identity parent", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.show_command("show", "iot_device_get_parent")
        cmd_group.show_command("set", "iot_device_set_parent")

    with self.command_group(
        "iot hub device-identity connection-string", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.show_command("show", "iot_get_device_connection_string")

    with self.command_group(
        "iot hub module-identity", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.command("create", "iot_device_module_create")
        cmd_group.show_command("show", "iot_device_module_show")
        cmd_group.command("list", "iot_device_module_list")
        cmd_group.command("delete", "iot_device_module_delete")
        cmd_group.generic_update_command(
            "update",
            getter_name="iot_device_module_show",
            setter_name="iot_device_module_update",
        )
        cmd_group.command("renew-key", "iot_device_module_key_regenerate")

    with self.command_group(
        "iot hub module-identity connection-string", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.show_command("show", "iot_get_module_connection_string")

    with self.command_group(
        "iot hub module-twin", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.show_command("show", "iot_device_module_twin_show")
        cmd_group.command("replace", "iot_device_module_twin_replace")
        cmd_group.generic_update_command(
            "update",
            getter_name="iot_device_module_twin_show",
            setter_name="iot_device_module_twin_update",
            custom_func_name="iot_twin_update_custom",
            custom_func_type=iothub_ops,
        )

    with self.command_group(
        "iot hub device-twin", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.show_command("show", "iot_device_twin_show")
        cmd_group.command("list", "iot_device_twin_list")
        cmd_group.command("replace", "iot_device_twin_replace")
        cmd_group.generic_update_command(
            "update",
            getter_name="iot_device_twin_show",
            setter_name="iot_device_twin_update",
            custom_func_name="iot_twin_update_custom",
            custom_func_type=iothub_ops,
        )

    with self.command_group(
        "iot hub configuration", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.command("show-metric", "iot_hub_configuration_metric_show")
        cmd_group.command("create", "iot_hub_configuration_create")
        cmd_group.show_command("show", "iot_hub_configuration_show")
        cmd_group.command("list", "iot_hub_configuration_list")
        cmd_group.command("delete", "iot_hub_configuration_delete")
        cmd_group.generic_update_command(
            "update",
            getter_name="iot_hub_configuration_show",
            setter_name="iot_hub_configuration_update",
        )

    with self.command_group(
        "iot hub distributed-tracing", command_type=iothub_ops, is_preview=True
    ) as cmd_group:
        cmd_group.show_command("show", "iot_hub_distributed_tracing_show")
        cmd_group.command("update", "iot_hub_distributed_tracing_update")

    with self.command_group(
        "iot hub connection-string", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.show_command("show", "iot_hub_connection_string_show")

    with self.command_group("iot edge", command_type=iothub_ops) as cmd_group:
        cmd_group.command("set-modules", "iot_edge_set_modules")
        cmd_group.command("export-modules", "iot_edge_export_modules")

    with self.command_group(
        "iot edge deployment", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.command("show-metric", "iot_edge_deployment_metric_show")
        cmd_group.command("create", "iot_edge_deployment_create")
        cmd_group.show_command("show", "iot_hub_configuration_show")
        cmd_group.command("list", "iot_edge_deployment_list")
        cmd_group.command("delete", "iot_hub_configuration_delete")
        cmd_group.generic_update_command(
            "update",
            getter_name="iot_hub_configuration_show",
            setter_name="iot_hub_configuration_update",
        )

    with self.command_group("iot device", command_type=iothub_ops) as cmd_group:
        cmd_group.command("send-d2c-message", "iot_device_send_message")
        cmd_group.command("simulate", "iot_simulate_device", is_experimental=True)
        cmd_group.command("upload-file", "iot_device_upload_file")

    with self.command_group(
        "iot device c2d-message", command_type=iothub_ops
    ) as cmd_group:
        cmd_group.command("complete", "iot_c2d_message_complete")
        cmd_group.command("abandon", "iot_c2d_message_abandon")
        cmd_group.command("reject", "iot_c2d_message_reject")
        cmd_group.command("receive", "iot_c2d_message_receive")
        cmd_group.command("send", "iot_c2d_message_send")
        cmd_group.command("purge", "iot_c2d_message_purge")

    with self.command_group("iot dps", command_type=iotdps_ops) as cmd_group:
        cmd_group.command(
            "compute-device-key",
            "iot_dps_compute_device_key",
            deprecate_info=self.deprecate(
                redirect='iot dps enrollment-group compute-device-key', hide=True
            )
        )

    with self.command_group("iot dps enrollment", command_type=iotdps_ops) as cmd_group:
        cmd_group.command("create", "iot_dps_device_enrollment_create")
        cmd_group.command("list", "iot_dps_device_enrollment_list")
        cmd_group.show_command("show", "iot_dps_device_enrollment_get")
        cmd_group.command("update", "iot_dps_device_enrollment_update")
        cmd_group.command("delete", "iot_dps_device_enrollment_delete")

    with self.command_group(
        "iot dps enrollment registration", command_type=iotdps_ops
    ) as cmd_group:
        cmd_group.show_command("show", "iot_dps_registration_get")
        cmd_group.command("delete", "iot_dps_registration_delete")

    with self.command_group(
        "iot dps enrollment-group", command_type=iotdps_ops
    ) as cmd_group:
        cmd_group.command("create", "iot_dps_device_enrollment_group_create")
        cmd_group.command("list", "iot_dps_device_enrollment_group_list")
        cmd_group.show_command("show", "iot_dps_device_enrollment_group_get")
        cmd_group.command("update", "iot_dps_device_enrollment_group_update")
        cmd_group.command("delete", "iot_dps_device_enrollment_group_delete")
        cmd_group.command(
            "compute-device-key", "iot_dps_compute_device_key"
        )

    with self.command_group(
        "iot dps enrollment-group registration", command_type=iotdps_ops
    ) as cmd_group:
        cmd_group.command("list", "iot_dps_registration_list")
        cmd_group.show_command("show", "iot_dps_registration_get")
        cmd_group.command("delete", "iot_dps_registration_delete")

    with self.command_group(
        "iot dps registration",
        command_type=iotdps_ops,
        deprecate_info=self.deprecate(redirect='iot dps enrollment-group registration', hide=True)
    ) as cmd_group:
        cmd_group.command("list", "iot_dps_registration_list")
        cmd_group.show_command("show", "iot_dps_registration_get")
        cmd_group.command("delete", "iot_dps_registration_delete")

    with self.command_group(
        "iot dps connection-string",
        command_type=iotdps_ops,
    ) as cmd_group:
        cmd_group.show_command("show", "iot_dps_connection_string_show")