from pathlib import Path
import subprocess
import os
import time
import decimals_functions as decf

# import magick_pdf_to_png


currfile_dir = Path(__file__).parent
tex_template_path = currfile_dir / "decimals_booklet_template.tex"
texans_template_path = currfile_dir / "decimals_booklet_ans_template.tex"
tex_diagram_template_path = (currfile_dir /
                             "decimals_booklet_diagram_template.tex")

colbreak = "\columnbreak  % Break column after 30 questions \n"


def convert_to_pdf(tex_path, outputdir):
    tex_path = Path(tex_path).resolve()
    outputdir = Path(outputdir).resolve()
    # for testing
    # print(f"tex_path: {tex_path}")
    # print(f"outputdir: {outputdir}")
    try:
        # Generate the PDF
        subprocess.run(["latexmk", "-pdf", "-outdir=" + str(outputdir), str(tex_path)], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
        # # Clean auxiliary files after successful PDF generation
        subprocess.run(["latexmk", "-c", "-outdir=" + str(outputdir), str(tex_path)], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
        # for hosted remove stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL for debugging any errors
        # Remove the .tex file manually
        if tex_path.exists():
            os.remove(tex_path)
    except subprocess.CalledProcessError as e:
        print(f"Error: {e}")


# tex_keys = ["num1", "num2", "process", "numip", "numdp"]
tex_keys_q = ["answer"]

def make1_diagram(tex_diagram_template_txt, nump, numip, numdp):
    tex_diagram_template_txt_ans = tex_diagram_template_txt
    kv = decf.get_dec_dict(nump, numip, numdp)
    posttext = r"\vspace{-1pt}"

    for key, value in kv.items():
        tex_diagram_template_txt_ans = tex_diagram_template_txt_ans.replace(
            "<<" + key + ">>", value
        )

    for key, value in kv.items():
        if key in tex_keys_q:
            tex_diagram_template_txt = tex_diagram_template_txt.replace(
                "<<" + key + ">>", ""
            )
        else:
            tex_diagram_template_txt = tex_diagram_template_txt.replace(
                "<<" + key + ">>", value
            )
    # return tex_diagram_template_txt
    return tex_diagram_template_txt + posttext, tex_diagram_template_txt_ans + posttext


def get_title(nump):
    match nump:
        case 1:
            return "Addition"
        case 2:
            return "Subtraction"
        case 3:
            return "Multiplication"
        case 4:
            return "Addition and subtraction"
        case 5:
            return "Addition, subtraction and multiplication"


def main():
    nump = input("Enter 1, 2, or 3 for +, -, random for the process \n")
    if nump.strip().isdigit():
        nump = int(nump)
        if not nump in [1, 2]:
            nump = 3  # random by default
    else:
        nump = 3  # random by default
    # get title for part of heading indicating which process/es
    title = get_title(nump)
    #
    numip = input(
        "Enter 0, 1, 2, 3, or 4 for the number of places before the decimal point: \n"
    )
    if numip.strip().isdigit():
        numip = int(numip)
        if not numip in [0, 1, 2, 3, 4]:
            numip = 1  # 1 by default
    else:
        numip = 1  # 1 by default
    #
    numdp = input(
        "Enter 1, 2, 3, 4, or 5 for the number of decimal places: \n")
    if numdp.strip().isdigit():
        numdp = int(numdp)
        if not numdp in [1, 2, 3, 4, 5]:
            numdp = 1  # 1 by default
    else:
        numdp = 1  # 1 by default
    #
    #
    numq = input(
        "Enter the number of questions from 1 to 108, with 27 per page: \n")
    if numq.strip().isdigit():
        numq = int(numq)
        if not numq in range(1, 109):
            numq = 27  # 27 by default
    else:
        numq = 27  # 27 by default
    #
    filename = input(
        "Enter the base filename to be added to the prefix asd_: \n")
    if not filename:
        filename = "1"  # "asd_1_q and asd_1_ans as default file"
    # set names of files that are made
    # questions
    tex_output_path = currfile_dir / f"asdBk_{filename}_q.tex"

    # answers
    tex_output_path_ans = currfile_dir / f"asdBk_{filename}_ans.tex"

    # Read in the LaTeX template file
    with open(tex_template_path, "r") as infile:
        tex_template_txt = infile.read()
    # Read in the LaTeX template file for answers
    with open(texans_template_path, "r") as infile:
        tex_template_txt_ans = infile.read()
    # Read in the LaTeX diagram template file
    with open(tex_diagram_template_path, "r") as infile:
        tex_diagram_template_txt = infile.read()

    # <<diagrams>>
    # generate column text and column text for answers
    diagram_text = ""
    diagram_text_ans = ""
    for i in range(1, numq + 1):
        img_tex, img_tex_ans = make1_diagram(tex_diagram_template_txt, nump,
                                             numip, numdp)
        diagram_text += img_tex
        diagram_text_ans += img_tex_ans
        if i % 27 == 0 and i > 0 and i != numq:
            diagram_text += colbreak
            diagram_text_ans += colbreak

    # Replace the <<title>> placeholder in the LaTeX template
    tex_template_txt = tex_template_txt.replace("<<title>>", title)
    tex_template_txt_ans = tex_template_txt_ans.replace("<<title>>", title)
    # Replace the <<diagrams>> placeholder in the LaTeX template with the generated diagrams
    tex_template_txt = tex_template_txt.replace("<<diagrams>>", diagram_text)
    tex_template_txt_ans = tex_template_txt_ans.replace(
        "<<diagrams>>", diagram_text_ans)

    # Write the question tex to an output file
    with open(tex_output_path, "w") as outfile:
        outfile.write(tex_template_txt)

    # Write the answer tex to an output file
    with open(tex_output_path_ans, "w") as outfile:
        outfile.write(tex_template_txt_ans)

    # Wait for the file to be created
    time.sleep(1)
    # Convert the LaTeX files to PDFs
    convert_to_pdf(tex_output_path, currfile_dir)
    convert_to_pdf(tex_output_path_ans, currfile_dir)


if __name__ == "__main__":
    print("starting")
    main()
    print("finished")