import { Example } from "./Example"

export function GenerateExamplePythonRest(model: Example) : string[] {
    var output: string[] = [];

    output.push("# " + model.Name);
    output.push("#");
    output.push("# This script expects that the following environment vars are set:");
    output.push("#");
    output.push("# AZURE_TENANT: your Azure Active Directory tenant id or domain");
    output.push("# AZURE_CLIENT_ID: your Azure Active Directory Application Client ID");
    output.push("# AZURE_SECRET: your Azure Active Directory Application Secret");
    output.push("# AZURE_SUBSCRIPTION_ID: your Azure Subscription Id");
    output.push("");
    output.push("import os");
    output.push("import traceback");
    output.push("from azure.common.credentials import ServicePrincipalCredentials");
    output.push("from msrestazure.azure_exceptions import CloudError");
    output.push("from msrestazure.azure_configuration import AzureConfiguration");
    output.push("from msrest.service_client import ServiceClient");
    output.push("from msrest.polling import LROPoller");
    output.push("from msrestazure.polling.arm_polling import ARMPolling");
    output.push("from msrest.pipeline import ClientRawResponse");
    output.push("import uuid");
    output.push("");
    output.push("SUBSCRIPTION_ID = os.environ['AZURE_SUBSCRIPTION_ID']");

    var vars =  model.Variables;
    for (var v in vars)
    {
        output.push(vars[v].name.toUpperCase() + " = \"" + ToCamelCase(vars[v].value.split("_NAME")[0].toLowerCase()) + "\"");   
    }
    output.push("");

    var json: string[] = GetExampleBodyJson(model.GetExampleBody());
    for (var lidx in json)
    {
        var line: string = json[lidx]; 
        if (line.startsWith("{"))
        {
            output.push("BODY = " + line);
        }
        else
        {
            output.push(line);
        }
    }

    output.push("");
    output.push("API_VERSION = '" + model.GetExampleApiVersion() + "'");
    output.push("");
    output.push("def get_credentials():");
    output.push("    credentials = ServicePrincipalCredentials(");
    output.push("        client_id=os.environ['AZURE_CLIENT_ID'],");
    output.push("        secret=os.environ['AZURE_SECRET'],");
    output.push("        tenant=os.environ['AZURE_TENANT']");
    output.push("    )");
    output.push("    return credentials");
    if (model.IsExampleLongRunning())
    {
        output.push("");
        output.push("");
        output.push("def wait_poller(service_client, operation_config, response):");
        output.push("    def get_long_running_output(response):");
        output.push("        return response");
        output.push("    poller = LROPoller(service_client,");
        output.push("                        ClientRawResponse(None, response),");
        output.push("                        get_long_running_output,");
        output.push("                        ARMPolling(30, **operation_config))");
        output.push("    try:");
        output.push("        poller.wait(timeout=600)");
        output.push("        response = poller.result()");
        output.push("    except Exception as exc:");
        output.push("        raise");
        output.push("    return response");
    }
    output.push("");
    output.push("");
    output.push("def run_example():");
    output.push("    credentials = get_credentials()");
    output.push("");
    output.push("    config = AzureConfiguration('https://management.azure.com')");
    output.push("    service_client = ServiceClient(credentials, config)");
    output.push("");
    output.push("    query_parameters = {}");
    output.push("    query_parameters['api-version'] = API_VERSION");
    output.push("");
    output.push("    header_parameters = {}");
    if (model.ExampleHasBody())
    {
        output.push("    header_parameters['Content-Type'] = 'application/json; charset=utf-8'");
    }
    if (model.IsExampleLongRunning())
    {
        output.push("    header_parameters['x-ms-client-request-id'] = str(uuid.uuid1())");
    }
    output.push("");
    output.push("    operation_config = {}");
    output.push("    request = service_client." + model.Method + "(" + ConvertUrl(model.Url) + ", query_parameters)");
    output.push("    response = service_client.send(request, header_parameters, BODY, **operation_config)");
    if (model.IsExampleLongRunning())
    {
        output.push("");
        output.push("    if response.status_code == 202:");
        output.push("        response = wait_poller(service_client, operation_config, response)");
    }
    output.push("    print(response.text)");
    output.push("");
    output.push("");
    output.push("if __name__ == \"__main__\":");
    output.push("    run_example()");

    return output;
}

function ToCamelCase(v: string)
{
    v = v.toLowerCase().replace(/[^A-Za-z0-9]/g, ' ').split(' ')
    .reduce((result, word) => result + capitalize(word.toLowerCase()))
    return v.charAt(0).toLowerCase() + v.slice(1)
}

function capitalize(v: string) {
    return v.charAt(0).toUpperCase() + v.toLowerCase().slice(1);
}

function GetExampleBodyJson(body: any): string[]
{
    var json: string = "{}";

    if (body != null)
    {
        //this.ExampleNormalize(body);
        json = JSON.stringify(body, null, "  ");
    }

    // XXX check if \n is sufficient
    var lines: string[] = json.split("\n");

    for (var i = 0; i < lines.length; i++)
    {
        var l: string = lines[i];
        if (lines[i].endsWith(": true"))
        {
            l = l.replace(": true", ": True");
        }
        else if (lines[i].endsWith(": true,"))
        {
            l = l.replace(": true,", ": True,");
        }
        else if (lines[i].endsWith(": false"))
        {
            l = l.replace(": false", ": False");
        }
        else if (lines[i].endsWith(": false,"))
        {
            l = l.replace(": false,", ": False,");
        }

        // XXX - will this work?
        if (l.indexOf("/subscription") >= 0)
        {
            var idx: number = 0;
            while ((idx = l.indexOf("{{", idx)) > 0)
            {
                var start: number = idx;
                var end: number = l.indexOf("}}", idx) + 2;
                var part: string = l.substring(start, end);
                var name: string = part.substring(2, part.length - 2).trim();
                var isLast: boolean = l[end + 2] == '"';

                if (!isLast)
                {
                    l = l.replace(part, "\" + " + name.toUpperCase() + " + \"");
                }
                else
                {
                    l = l.replace(part + "\"", "\" + " + name.toUpperCase());
                }
                idx = end + 2;
            }
        }

        lines[i] = l;
    }
    return lines;
}

function ConvertUrl(sourceUrl: string): string
{
    var parts: string[] = sourceUrl.split("/");
    var url = "\"";

    for (var i: number = 0; i < parts.length; i++)
    {
        var part: string = parts[i];
        var last: boolean = (i == parts.length - 1);

        if (part.startsWith("{{"))
        {
            var varName: string = part.substring(3, part.length - 6).toUpperCase();

            //if (varName == "SUBSCRIPTION_ID")
            //{
            //    varName = varName.ToLower();
            //}

            // close and reopen quotes, add add variable name in between
            url += "\" + " + varName + (last ? "" : " + \"/");
        }
        else
        {
            url += part + (last ? "\"" : "/");
        }
    }
    return url;
}