import boto3, sys, datetime, socket, re


def _get_boto_session(profile_name: str) -> boto3.Session:
    """
    Get the boto3 session. Works with profile configured in runtime.

    :param profile_name: The configured AWS profile name. Pass in None to use default credentials chain.
    :returns: The Boto session.
    """
    if profile_name == None:
        session = boto3.Session(region_name="eu-central-1")
        return session
    else:
        session = boto3.Session(profile_name=profile_name)
        return session


def _parse_one_image(line: str) -> str:
    if re.match("^ *!\[", line):
        percent = re.sub("^ *!\[", "", line.split("](")[0])
        return (
            '<img style="margin-left: 58px; margin-right:58px; " width="'
            + percent
            + '%" align="middle" src="'
            + line.split("(")[1].replace(")", "").strip()
            + '"/>'
        )
    else:
        return line


def _parse_one_href(line: str) -> str:
    if not re.match("^ *!\[", line) and re.match("^.*\[.*\]\(.*\)", line):
        anchor = (
            '<a href="'
            + re.sub("\).*$", "", line.split("](")[1])
            + '" target="_blank">'
            + re.sub("^.*\[", "", line.split("](")[0])
            + "</a>"
        )
        return re.sub("\[.*\)", anchor, line)
    else:
        return line


def _parse_one_code_pv1(line: str) -> str:
    ret_tab = "<p>" + ("&nbsp;" * 17)
    if re.match("^.*`.*`", line):
        code = (
            '<code onclick="copyText(this)">' + line.split("`")[1].strip() + "</code>"
        )
        if ">Step" in line:
            return re.sub("`.*`", ret_tab + code, line)
        else:
            return re.sub("`.*`", code, line)
    else:
        return line


def _parse_one_code(line: str, pos: int) -> str:
    if re.match("^.*`", line) and not re.match("^ *```", line):
        new_pos = pos + 1
        if pos % 2 == 0:
            return _parse_one_code(line.replace("`", "</samp>", 1), new_pos)
        else:
            return _parse_one_code(
                line.replace("`", '<samp onclick="copyText(this)">', 1), new_pos
            )
    else:
        return line


def _process_command_begin() -> str:
    command_begin = '<div class="code-area"><pre><div onclick="copyText(this)">'
    return command_begin


def _process_command_end() -> str:
    command_end = "</div></pre></div><br/>"
    return command_end


def _process_command(line: str) -> str:
    return "<p>" + line.replace(" ", "&nbsp;") + "</p>"


def _process_sql_begin() -> str:
    sql_begin = '<pre><div onclick="copyText(this)" class="language-sql">'
    return sql_begin


def _process_sql_end() -> str:
    sql_end = "</pre></div><br/>"
    return sql_end


def _process_sql(line: str) -> str:
    # return "<p>" + line.replace(" ", "&nbsp;") + "</p>"
    return line + "\n"


def _process_output_begin() -> str:
    output_begin = '<div class="output-area"><pre><div>'
    return output_begin


def _process_output_end() -> str:
    output_end = "</div></pre></div><br/>"
    return output_end


def _process_output(line: str) -> str:
    return "<p>" + line.replace(" ", "&nbsp;") + "</p>"


def _parse_and_print_md(
    normal_font_size: str,
    language: str,
    lab_guide_markdown_file: str,
    m_public_ip: str,
    trainer_name: str,
    pd_dashboard_url: str,
    grafana_dashboard_url: str,
    nodes_table: str,
):
    print("<div>")
    f = open(
        "/usr/share/httpd/lab_control_scripts/lib/"
        + language
        + "/"
        + lab_guide_markdown_file,
        "r",
        encoding="UTF-8",
    )
    lines = f.read()
    f.close()
    parsed_lines = []
    paneIds = ["overview"]
    in_sql = False
    in_command = False
    in_output = False
    exercise_title = ""
    for line in lines.splitlines(False):
        parsed_line = _parse_one_image(line)
        if in_command:
            if re.match("^ *``` *$", parsed_line):
                parsed_lines += [_process_command_end()]
                in_command = False
            else:
                parsed_lines += [_process_command(line)]
        elif in_output:
            if re.match("^ *``` *$", parsed_line):
                parsed_lines += [_process_output_end()]
                in_output = False
            else:
                parsed_lines += [_process_output(line)]
        elif in_sql:
            if re.match("^ *``` *$", parsed_line):
                parsed_lines += [_process_sql_end()]
                in_sql = False
            else:
                parsed_lines += [_process_sql(line)]
        elif re.match("^# ", line):
            title = (
                '<br/><h3 class="mui--text-center">'
                + parsed_line.replace("# ", "", 1).strip()
                + "</h3><br/>"
            )
            exercise_title = parsed_line.replace("# ", "", 1).strip()
        elif re.match("^### ", parsed_line):
            parsed_lines += [
                "<br/><h3>" + parsed_line.replace("### ", "", 1).strip() + "</h3>"
            ]
        elif re.match("^## ", parsed_line):
            parsed_lines += [
                "<br/><h3>" + parsed_line.replace("## ", "", 1).strip() + "</h3>"
            ]
        elif re.match("^\+ ", parsed_line):
            tl = (
                '<li style="font-size:'
                + normal_font_size
                + 'px">'
                + parsed_line.replace("+ ", "", 1).strip()
                + "</li>"
            )
            ptl = _parse_one_code(_parse_one_href(tl), 1)
            parsed_lines += [ptl]
        elif re.match("^ +\+ ", parsed_line):
            tl = (
                '<li style="margin-left: 36px; font-size:'
                + normal_font_size
                + 'px">'
                + parsed_line.replace("+ ", "", 1).strip()
                + "</li>"
            )
            ptl = _parse_one_code(_parse_one_href(tl), 1)
            parsed_lines += [ptl]
        elif re.match("^[0-9]*\. ", parsed_line):
            tl = (
                '<li style="font-size:'
                + normal_font_size
                + 'px">'
                + "<b>Step "
                + re.findall("^([0-9]+\.)", parsed_line)[0]
                + "</b> "
                + re.sub("^[0-9]*\. ", "", parsed_line, 1).strip()
                + "</li><br/>"
            )
            ptl = _parse_one_code(_parse_one_href(tl), 1)
            parsed_lines += [ptl]
        elif re.match("^ *```command", parsed_line):
            parsed_lines += [_process_command_begin()]
            in_command = True
        elif re.match("^ *```output", parsed_line):
            parsed_lines += [_process_output_begin()]
            in_output = True
        elif re.match("^ *```sql", parsed_line):
            parsed_lines += [_process_sql_begin()]
            in_sql = True
        elif re.match("^ *``` *$", parsed_line):
            if in_command:
                parsed_lines += [_process_command_end()]
                in_command = False
            if in_output:
                parsed_lines += [_process_output_end()]
                in_output = False
            if in_sql:
                parsed_lines += [_process_sql_end()]
                in_sql = False
        else:
            ptl = (
                '<div class="overview-desc">'
                + _parse_one_code(_parse_one_href(parsed_line), 1)
                + "</div><br/>"
            )
            parsed_lines += [ptl]
    print('<div style="white-space:normal;">')
    print('<ul class="mui-tabs__bar" style="background-color:#f0f0f0">')
    print(
        '<li class="mui--is-active"><a data-mui-toggle="tab" data-mui-controls="overview" style="cursor: pointer;" onclick="activate(0)"><b>Overview</b></a></li>'
    )
    overview_tab = True
    idx = 0
    for line in lines.splitlines(False):
        if line.startswith("## Exercise "):
            idx += 1
            paneId = "e" + line.replace("## Exercise ", "").strip()
            print(
                '<li><a data-mui-toggle="tab" data-mui-controls="'
                + paneId
                + '" style="cursor: pointer;" onclick="activate('
                + str(idx)
                + ')">'
                + "<b>"
                + line.replace("## ", "").strip()
                + "</b>"
                + "</a></li>"
            )
            paneIds += [paneId]
    print("</ul>")
    print(
        '<div class="scrollable-div mui-tabs__pane mui--is-active" id="overview"><br/>'
    )
    single_nodes_table = True
    single_create_lab_hint = True
    for pl in parsed_lines:
        if pl.startswith("<br/><h3>Overview") and single_nodes_table:
            if len(nodes_table) > 50:
                print(nodes_table)
                single_nodes_table = False
            elif single_create_lab_hint:
                print(
                    "<h4>Click <button class=\"mui-btn mui-btn--small mui-btn--primary\" onclick=\"lab_start_exercises('frankfurt','Creating lab ...','Preparing Exercises. Please do NOT close or refresh your browser.','Please proceed with your hands-on practices by following the Exercises section in the student guide.', 'Auto terminated at')\">CREATE LAB</button> and allow approximately 6 minutes for the countdown timer to finish. Upon completion, you will <span style=\"color: Red\">unlock the detailed exercise tabs</span> above and receive the information of the lab environment.</h4><br/>"
                )
                single_create_lab_hint = False
        elif pl.startswith("<br/><h3>Exercise "):
            print(
                '<button class="mui-btn mui-btn--small mui-btn--primary" onclick="defaultPosition()">Back to Top</button>'
            )
            if len(nodes_table) > 50:
                print(
                    '<button class="mui-btn mui-btn--small mui-btn--primary" onclick="activateNext('
                    + str(paneIds)
                    + ')">Next</button>'
                )
            else:
                print(
                    "<button class=\"mui-btn mui-btn--small mui-btn--primary\" onclick=\"lab_start_exercises('frankfurt','Creating lab ...','Preparing Exercises. Please do NOT close or refresh your browser.','Please proceed with your hands-on practices by following the Exercises section in the student guide.', 'Auto terminated at')\">CREATE LAB</button>"
                )
            if len(nodes_table) > 50:
                print(
                    '</div><div class="scrollable-div mui-tabs__pane" id="e'
                    + pl.replace("<br/><h3>Exercise ", "").replace("</h3>", "").strip()
                    + '"><br/>'
                )
            else:
                print("</div>")
                break
        else:
            if "<img" in pl:
                if overview_tab or len(nodes_table) > 50:
                    print(pl)
                else:
                    None
            else:
                if "None" not in pd_dashboard_url:
                    pl = pl.replace(
                        "TiDB Dashboard",
                        '<a href="'
                        + pd_dashboard_url
                        + '" target="_blank">TiDB Dashboard</a>',
                    )
                if grafana_dashboard_url != None:
                    pl = pl.replace(
                        "Grafana Dashboard",
                        '<a href="'
                        + grafana_dashboard_url
                        + '" target="_blank">Grafana Dashboard</a>',
                    )
                if len(nodes_table) > 50:
                    print(
                        pl.replace("monitor1_address", m_public_ip)
                        .replace("trainer_name", trainer_name)
                        .replace("us-west-2", "eu-central-1")
                        .replace("oregon", "frankfurt")
                    )
                else:
                    print(
                        pl.replace("us-west-2", "eu-central-1").replace(
                            "oregon", "frankfurt"
                        )
                    )
    if len(nodes_table) > 50:
        print(
            '<button class="mui-btn mui-btn--small mui-btn--primary" onclick="defaultPosition()">Back to Top</button>'
        )
        print(
            "<button class=\"mui-btn mui-btn--small mui-btn--danger\" onclick=\"lab_delete_stack('frankfurt',' Terminating lab ...','Cleanup in-progress. It requires approximately 4 minutes. Click REFRESH button to verify if all lab nodes have been terminated.','Click CREATE LAB to start. If the nodes table above is NOT empty, you can skip the setup action.', false)\">TERMINATE LAB</button>"
        )  # For terminate resources
    print("</div>")  # For tab closing
    print("</div>")  # For panel closing
    print("</div>")  # For routine section closing


def _check(
    session: boto3.Session,
    trainer_name: str,
    lab_guide_markdown_file: str,
    language: str,
    normal_font_size: str,
    pd_dashboard_url: str,
):
    grafana_dashboard_url = None
    step_2_needed = None
    nodes_table = ""
    # AWS pricing for SIN
    # c5_xlarge_instance_minute_price = 0.003266666666666667
    # ebs_gp2_100gb_minute_price = 0.00023148148148148146

    # AWS pricing for NRT
    # c5_xlarge_instance_minute_price = 0.0035666666666666668
    # ebs_gp2_100gb_minute_price = 0.00023148148148148146

    # AWS pricing for PAX
    c5_xlarge_instance_minute_price = 0.0028333333333333335
    ebs_gp2_100gb_minute_price = 0.00023148148148148146

    c5_xlarge_minute_price = (
        c5_xlarge_instance_minute_price + ebs_gp2_100gb_minute_price
    )
    ec2_client = session.client("ec2")
    sum_cost = 0.0
    instances_dict = {}
    response = None
    real_reservations = []
    if trainer_name == "all":
        response = ec2_client.describe_instances(
            Filters=[
                {
                    "Name": "instance-state-name",
                    "Values": [
                        "running",
                    ],
                },
            ],
            DryRun=False,
            MaxResults=500,
        )
        real_reservations += [response["Reservation"]]
        while "NextToken" in response and response["NextToken"] != None:
            nt = response["NextToken"]
            response = ec2_client.describe_instances(
                Filters=[
                    {
                        "Name": "instance-state-name",
                        "Values": [
                            "running",
                        ],
                    },
                ],
                DryRun=False,
                MaxResults=500,
                NextToken=nt,
            )
            real_reservations += [response["Reservation"]]
    else:
        response = ec2_client.describe_instances(
            Filters=[
                {
                    "Name": "instance-state-name",
                    "Values": [
                        "running",
                    ],
                },
                {
                    "Name": "tag:trainer",
                    "Values": [
                        trainer_name,
                    ],
                },
            ],
            DryRun=False,
            MaxResults=500,
        )
        real_reservations += [response["Reservations"]]
        while "NextToken" in response and response["NextToken"] != None:
            nt = response["NextToken"]
            response = ec2_client.describe_instances(
                Filters=[
                    {
                        "Name": "instance-state-name",
                        "Values": [
                            "running",
                        ],
                    },
                    {
                        "Name": "tag:trainer",
                        "Values": [
                            trainer_name,
                        ],
                    },
                ],
                DryRun=False,
                MaxResults=500,
                NextToken=nt,
            )
            real_reservations += [response["Reservations"]]
    for reservations in real_reservations:
        for reservation in reservations:
            for instances in reservation["Instances"]:
                up_time = round(
                    (
                        datetime.datetime.now(datetime.timezone.utc)
                        - instances["UsageOperationUpdateTime"]
                    ).seconds
                    / 60
                )
                name = None
                trainer = None
                student = None
                node_role = None
                for tag in instances["Tags"]:
                    if tag["Key"] == "Name":
                        name = tag["Value"]
                    if tag["Key"] == "trainer":
                        trainer = tag["Value"]
                    if tag["Key"] == "student":
                        student = tag["Value"]
                    if tag["Key"] == "role":
                        node_role = tag["Value"]
                if (
                    name == None
                    or trainer == None
                    # or student == None
                    or node_role == None
                ) and (name != "lab-control-v2"):
                    step_2_needed = "Setup Phase 2 In-Progress ..."
                cost = 0.0
                cost = round(up_time * c5_xlarge_minute_price, 2)
                if name != "lab-control-v2":
                    sum_cost += cost
                    instances_dict.update(
                        {
                            (trainer if trainer != None else " ")
                            + ":"
                            + (student if student != None else " ")
                            + ":"
                            + (node_role if node_role != None else " ")
                            + ":"
                            + instances["InstanceId"]: "<tr><td>"
                            + instances["InstanceId"].rjust(19, " ")
                            + "</td><td>"
                            + instances["PublicIpAddress"].ljust(16, " ")
                            + "</td><td>"
                            + instances["PrivateIpAddress"].ljust(16, " ")
                            + "</td><td>"
                            + (name.ljust(13, " ") if name != None else " ")
                            + "</td><td>"
                            + (trainer.ljust(10, " ") if trainer != None else " ")
                            + "</td><td>"
                            + (student.ljust(7, " ") if student != None else " ")
                            + "</td><td>"
                            + (node_role.ljust(9, " ") if node_role != None else " ")
                            + "</td><td>"
                            + str(up_time).ljust(14, " ")
                            + "</td></tr>"
                        }
                    )
                else:
                    cost = str(cost) + " (covered by RI)"

    if step_2_needed != None:
        print("<h3>" + step_2_needed + "</h3>")

    trainer_name_cache = "NONE"
    student_name_cache = "NONE"
    for key in sorted([*instances_dict]):
        trainer_name = key.split(":")[0]
        student_name = key.split(":")[1]
        if trainer_name != trainer_name_cache or student_name != student_name_cache:
            nodes_table += "<table>"
            # nodes_table += ("<caption>Available Nodes for", trainer_name, "</caption>")
            nodes_table += "<thead><tr>"
            nodes_table += "<th>Instance ID</th><th>Public IP</th><th>Private IP</th><th>Suggested Role</th><th>Instructor</th><th>Student</th><th>Name</th><th>Up Time (mins)</th>"
            nodes_table += "</tr></thead>"
            nodes_table += "<tbody>"
            trainer_name_cache = trainer_name
            student_name_cache = student_name
        nodes_table += instances_dict[key]
    nodes_table += "</tbody>"
    nodes_table += "</table>"
    m_public_ip = "monitor1_address"
    for key in sorted([*instances_dict]):
        trainer_name = key.split(":")[0]
        student_name = key.split(":")[1]

        trainer_name_cache = trainer_name
        student_name_cache = student_name
        row = instances_dict[key]
        if "monitor1" in row:
            m_public_ip = row.split("</td><td>")[1].strip()
    if len([*instances_dict]) > 0:
        None
    if m_public_ip != "monitor1_address":
        nodes_table += "<p>"
        if "None" not in pd_dashboard_url:
            nodes_table += (
                '<br/>TiDB Dashboard - <a href="'
                + pd_dashboard_url
                + '" target="_blank">'
                + pd_dashboard_url
                + "</a><p>"
            )
        monitor_ip = m_public_ip.strip()
        grafana_port = 3000
        timeout_seconds = 1
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.settimeout(timeout_seconds)
        result = sock.connect_ex((monitor_ip, int(grafana_port)))
        if result == 0:
            grafana_dashboard_url = "http://" + monitor_ip + ":3000"
            nodes_table += (
                'Grafana Dashboard - <a href="http://'
                + monitor_ip
                + ':3000" target="_blank">http://'
                + monitor_ip
                + ":3000</a>"
            )
        sock.close()
        nodes_table += "</p><br/>"
    else:
        None
    # print("</div>")
    _parse_and_print_md(
        normal_font_size,
        language,
        lab_guide_markdown_file,
        m_public_ip,
        trainer_name,
        pd_dashboard_url,
        grafana_dashboard_url,
        nodes_table,
    )


if __name__ == "__main__":
    trainer_name = sys.argv[1]
    pd_dashboard_url = sys.argv[2]
    lab_name = sys.argv[3]
    session = _get_boto_session(None)
    _check(
        session,
        trainer_name,
        lab_name + "-lab-guide.md",
        "english",
        "16",
        pd_dashboard_url,
    )