import os
import re
import shutil

def replace_title(save_to_path, title):
    # Define input and output file names
    input_file_name = os.path.join(save_to_path, "template.tex")
    output_file_name = os.path.join(save_to_path , "main.tex")

    # Open the input file and read its content
    with open(input_file_name, 'r') as infile:
        content = infile.read()
    content = content.replace(r"\title{TITLE} ", f"\\title{{{title}}} ")

    # Open the output file and write the modified content
    with open(output_file_name, 'w') as outfile:
        outfile.write(content)


# return all string in \cite{...} \citet{...} or \citep{...}.

# check if citations are in bibtex.

# replace citations

# sometimes the output may include thebibliography and bibitem . remove all of it.

# return all .png and replace it using placeholder.

def find_tex_files(directory_path):
    tex_files = []

    for filename in os.listdir(directory_path):
        if filename.endswith(".tex"):
            tex_files.append(filename)

    return tex_files

def find_figure_names(tex_file_path):
    # Regular expression pattern to find \includegraphics commands
    pattern = r'\\includegraphics.*?{(.*?)}'
    with open(tex_file_path, 'r') as file:
        content = file.read()
    # Find all matches in the file content
    matches = re.findall(pattern, content)
    # Matches will be a list of figure names
    return matches

def create_copies(output_dir):
    tex_files = find_tex_files(output_dir)
    for tex_file in tex_files:
        path =  os.path.join(output_dir, tex_file)
        all_figs = find_figure_names(path)
        for fig in all_figs:
            original_fig = os.path.join(output_dir, "fig.png")
            target_fig = os.path.join(output_dir, fig)
            shutil.copy2(original_fig, target_fig)






if __name__ == "__main__":
    auto_draft = os.path.dirname(os.getcwd())
    directory_path = "outputs/outputs_20230520_110413/"
    path = os.path.join(auto_draft, directory_path)

    create_copies(path)