from mathgenerator.mathgen import *

write_list = []
subjects = [
    'algebra', 'basic_math', 'calculus', 'computer_science', 'geometry',
    'misc', 'statistics'
]
wList = getGenList()


def array2markdown_table(string):
    string = string.replace("[[", "<table><tr><td>")
    string = string.replace("[", "<tr><td>")
    string = string.replace(", ", "</td><td>")
    string = string.replace("]]", "</td></tr></table>")
    string = string.replace("]", "</td></tr>")
    string = string.replace(" ", "")
    string = string.replace("\n", "")
    return string


def write_table_of_contents():
    lines = []

    tc_lines = [
        '* [Installation](#installation)\n',
        '* [Basic Usage](#basic-usage)\n',
        '  * [More Complicated Usage](#more-complicated-usage)\n'
        '* [Documentation](#documentation)\n',
        '* [List of Generators](#list-of-generators)\n',
    ]
    for subject in subjects:
        line = '  * [' + subject + '](#' + subject + ')\n'
        tc_lines.append(line)

    with open('../README.md', "r") as g:
        lines = g.readlines()

        upper_bound = lines.index('## Table of Contents\n')
        upper_lines = lines[:upper_bound + 1]
        lower_bound = lines.index('## Installation\n')
        lower_lines = lines[lower_bound - 1:]
        lines = []

        for upper_line in upper_lines:
            lines.append(upper_line)
        for tc_line in tc_lines:
            lines.append(tc_line)
        for lower_line in lower_lines:
            lines.append(lower_line)

    with open('../README.md', "w") as g:
        g.writelines(lines)


def gen_to_row_string(item):
    myGen = item[2]
    # NOTE: renamed 'sol' to 'solu' to make it look nicer
    print(f"Generating sample for {item[0]}")
    prob, solu = myGen()
    prob = str(prob).rstrip("\n")
    solu = str(solu).rstrip("\n")
    # edge case for matrixMultiplication
    if item[0] == 46:
        prob, solu = myGen(maxVal=10, max_dim=4)
        prob = str(prob).rstrip("\n")
        solu = str(solu).rstrip("\n")
        prob = array2markdown_table(prob)
        solu = array2markdown_table(solu)

    # NOTE: renamed 'def_name' to 'func_name' because it suits it more
    func_name = item[3]
    kwargs = ""
    kwargs_list = myGen.kwargs
    for kwarg in kwargs_list:
        kwargs += '`' + kwarg + '` '
    title = '[' + myGen.title + '](https://github.com/lukew3/mathgenerator/blob/main/mathgenerator/funcs/' + item[
        4] + '/' + func_name + '.py)'
    row = [myGen.id, title, prob, solu, func_name, kwargs]
    # tableLine = "| " + str(row[0]) + " | " + str(row[1]) + " | " + str(
    #    row[2]) + " | " + str(row[3]) + " | " + str(row[4]) + " |\n"
    tableLine = "| " + str(row[0]) + " | " + str(row[1]) + " | " + str(
        row[2]) + " | " + str(row[3]) + " | " + str(row[4]) + " | " + str(
            row[5]) + " |\n"
    print('added', item[1], '-', func_name, 'to the README.md')
    return tableLine


def make_table_header(name):
    lines = [
        '## ' + name + '\n',
        '| Id   | Skill | Example problem | Example Solution | Function Name | Kwargs |\n',
        '|------|-------|-----------------|------------------|---------------|--------|\n'
    ]
    for line in lines:
        write_list.append(line)


def write_subject_table(subject_name, full_gen_list):
    subject_list = []
    # Create list of generators in given subject
    for gen in full_gen_list:
        if gen[4] == subject_name:
            subject_list.append(gen)
    subject_list.sort()

    # Create table header
    make_table_header(subject_name)

    # Add each item to write_list
    for item in subject_list:
        write_list.append(gen_to_row_string(item))


def update_next_id():
    next_id = len(getGenList())
    new_lines = ""
    file = open('../CONTRIBUTING.md', 'r')
    for line in file:
        new_line = ""
        if "<!--Start next id-->" in line:
            new_line = f"      * Use this id: <!--Start next id-->{next_id}<!--End next id-->. (This is updated automatically after a new generator is added)\n"
        else:
            new_line = line
        new_lines += new_line
    file.close()
    write_file = open("../CONTRIBUTING.md", "w")
    write_file.write(new_lines)
    write_file.close()
    print("CONTRIBUTING.md updated")


def main():
    write_table_of_contents()
    for subject in subjects:
        write_subject_table(subject, wList)

    with open('../README.md', "r") as g:
        lines = g.readlines()

        line = lines.index('## List of Generators\n')
        lines = lines[:line + 1]

        for write_line in write_list:
            lines.append(write_line)

    with open('../README.md', "w") as g:
        g.writelines(lines)

    print("New README.md table generated")

    update_next_id()


if __name__ == "__main__":
    main()