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="ap-southeast-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

    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/>"
            )
            # print(title)
        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>'
    )
    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(
                    "<h3>Click <kbd>CREATE LAB</kbd> and wait for the countdown timer to finish to access the lab resources</h3><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>'
            )
            print(
                '<button class="mui-btn mui-btn--small mui-btn--primary" onclick="activateNext('
                + str(paneIds)
                + ')">Next</button>'
            )
            print(
                '</div><div class="scrollable-div mui-tabs__pane" id="e'
                + pl.replace("<br/><h3>Exercise ", "").replace("</h3>", "").strip()
                + '"><br/>'
            )
        else:
            if "<img" in pl:
                print(
                    pl.replace("monitor1_address", m_public_ip).replace(
                        "trainer_name", trainer_name
                    )
                )
            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>',
                    )
                print(
                    pl.replace("monitor1_address", m_public_ip)
                    .replace("trainer_name", trainer_name)
                    .replace("us-west-2", "ap-southeast-1")
                    .replace("oregon", "singapore")
                )
    print("</div>")  # For tab closing
    # print(
    #     '<br/><button class="mui-btn mui-btn--small mui-btn--primary" onclick="defaultPosition()">Return to Top</button>'
    # )
    # print("<span>&nbsp;&nbsp;&nbsp;</span>")
    # print(
    #     '<button class="mui-btn mui-btn--small mui-btn--primary" onclick="activateNext('
    #     + str(paneIds)
    #     + ')">Next</button>'
    # )
    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"):
                    step_2_needed = "Setup Phase 2 In-Progress ..."
                cost = 0.0
                cost = round(up_time * c5_xlarge_minute_price, 2)
                if name != "lab-control":
                    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:
        nodes_table += (
            "<br/><p>Reporting at - (UTC) "
            + str(datetime.datetime.now())
            + " with "
            + str(len([*instances_dict]))
            + " nodes.</p>"
        )
    if m_public_ip != "monitor1_address":
        nodes_table += "<p>"
        if "None" not in pd_dashboard_url:
            nodes_table += (
                '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,
    )