import { CodeModelCli, CommandParameter } from "./CodeModelCli"
import { ModuleOption } from "../ModuleMap";
import { EscapeString } from "../Helpers";

export function GenerateAzureCliParams(model: CodeModelCli) : string[] {
    var output: string[] = [];

    output.push("# --------------------------------------------------------------------------------------------");
    output.push("# Copyright (c) Microsoft Corporation. All rights reserved.");
    output.push("# Licensed under the MIT License. See License.txt in the project root for license information.");
    output.push("# --------------------------------------------------------------------------------------------");
    output.push("# pylint: disable=line-too-long");
    output.push("# pylint: disable=too-many-lines");
    output.push("# pylint: disable=too-many-statements");   
    output.push("");
    output.push("from knack.arguments import CLIArgumentType");
    output.push("from azure.cli.core.commands.parameters import (");
    output.push("    tags_type,");
    //output.push("    get_resource_name_completion_list,");
    //output.push("    quotes,");
    output.push("    get_three_state_flag,");
    output.push("    get_enum_type,");
    output.push("    resource_group_name_type,");
    output.push("    get_location_type");
    output.push(")");
    output.push("from azure.cli.core.commands.validators import get_default_location_from_resource_group");
    output.push("");
    output.push("");
    output.push("def load_arguments(self, _):");
    output.push("    name_arg_type = CLIArgumentType(options_list=('--name', '-n'), metavar='NAME')");

    do
    {
        // this is a hack, as everything can be produced from main module now
        if (model.ModuleName.endsWith("_info"))
            continue;

            //output.push("    apimanagement_name_type = CLIArgumentType(options_list='--apimanagement-name-name', help='Name of the Apimanagement.', id_part='name')");
        //output.push("");
        //output.push("    with self.argument_context('apimanagement') as c:");
        //output.push("        c.argument('tags', tags_type)");
        //output.push("        c.argument('location', validator=get_default_location_from_resource_group)");
        //output.push("        c.argument('apimanagement_name', apimanagement_name_type, options_list=['--name', '-n'])");

        //output.push("    with self.argument_context('" + model.GetCliCommand() + "') as c:");
        //output.push("        c.argument('tags', tags_type)");
        //output.push("        c.argument('location', validator=get_default_location_from_resource_group)");
        //output.push("        c.argument('" + model.GetCliCommand() + "_name', name_arg_type, options_list=['--name', '-n'])");
        
        let options: ModuleOption[] = model.ModuleOptions;
        let methods: string[] = model.GetCliCommandMethods();
        for (let mi = 0; mi < methods.length; mi++)
        {
            let method: string = methods[mi];
            output.push("");
            output.push("    with self.argument_context('" + model.GetCliCommand() + " " + method + "') as c:");

            let ctx = model.GetCliCommandContext(method);
            let params: CommandParameter[] = ctx.Parameters;

            params.forEach(element => {
                let parameterName: string = element.Name.split("-").join("_");
                let argument = "        c.argument('" + parameterName + "'";

                // this is to handle names like "format", "type", etc
                if (parameterName == "type" || parameterName == "format")
                {
                    argument = "        c.argument('_" + parameterName + "'";
                    argument += ", options_list=['--" + parameterName + "']";
                }

                if (element.Type == "boolean")
                {
                    argument += ", arg_type=get_three_state_flag()";
                }
                else if ((element.EnumValues.length > 0) && !element.IsList)
                {
                    argument += ", arg_type=get_enum_type([";

                    element.EnumValues.forEach(element => {
                        if (!argument.endsWith("[")) argument += ", ";
                        argument += "'" + element + "'";
                    });
                    argument += "])";
                }
                if (parameterName == "resource_group")
                {
                    argument += ", resource_group_name_type)";
                }
                else if (parameterName == "tags")
                {
                    argument += ", tags_type)";
                }
                else if (parameterName == "location")
                {
                    argument += ", arg_type=get_location_type(self.cli_ctx))";
                }
                else
                {
                    argument += ", id_part=None, help='" + EscapeString(element.Help) + "')"; 
                }
                output.push(argument);
            });
        }
    } while (model.NextModule());;



    //output.push("    apimanagement_name_type = CLIArgumentType(options_list='--apimanagement-name-name', help='Name of the Apimanagement.', id_part='name')");
    output.push("");
    output.push("    with self.argument_context('apimanagement') as c:");
    output.push("        c.argument('tags', tags_type)");
    output.push("        c.argument('location', validator=get_default_location_from_resource_group)");
    output.push("        c.argument('apimanagement_name', name_arg_type, options_list=['--name', '-n'])");
    output.push("");

    return output;
}