import os
import shutil
import zipfile
import re
from datetime import datetime

def calculate_file_percentages(directory, file_counts):
    code_extensions = ['.html', '.css', '.js', '.py']  # Add more extensions if needed
    
    # Count the occurrences of each file extension
    for filename in os.listdir(directory):
        if os.path.isfile(os.path.join(directory, filename)):
            _, extension = os.path.splitext(filename)
            if extension:
                extension = extension.lower()  # Convert extension to lowercase
                file_counts[extension] = file_counts.get(extension, 0) + 1
                
                if extension in code_extensions:
                    file_path = os.path.join(directory, filename)
                    with open(file_path, 'r', encoding='utf-8') as file:
                        content = file.read()
                        if contains_code(content, extension):
                            file_counts[extension] += 1

def contains_code(content, extension):
    if extension == '.html':
        css_pattern = r'<style[.\s\S]*?</style>'
        return re.search(css_pattern, content, re.IGNORECASE)
    elif extension == '.css':
        # Modify this pattern to match your CSS content detection requirements
        css_pattern = r'\w+\s*{\s*\w+\s*:\s*\w+\s*;.*?}'
        return re.search(css_pattern, content)
    elif extension == '.js':
        js_pattern = r'<script[.\s\S]*?</script>'
        return re.search(js_pattern, content, re.IGNORECASE)
    elif extension == '.py':
        # Modify this pattern to match your Python content detection requirements
        py_pattern = r'\bimport\b|\bfrom\b'
        return re.search(py_pattern, content)
    else:
        return False

def find_files_with_extensions(extensions):
    script_directory = os.path.dirname(os.path.abspath(__file__))
    found_files = []
    for root, _, files in os.walk(script_directory):
        for file in files:
            if file.endswith(tuple(extensions)):
                found_files.append(os.path.join(root, file))
    return found_files

def get_non_ignored_files(files, ignore_folder):
    def is_inside_ignore_folder(file_path, ignore_folder):
        ignore_folder = os.path.abspath(ignore_folder)
        file_path = os.path.abspath(file_path)
        return file_path.startswith(ignore_folder)

    non_ignored_files = []
    for file in files:
        abs_file_path = os.path.abspath(file)
        if not is_inside_ignore_folder(abs_file_path, ignore_folder) \
                and not is_inside_ignore_folder(abs_file_path, 'prevvers') \
                and not is_inside_ignore_folder(abs_file_path, 'futures') \
                and not abs_file_path.endswith('exp'):
            non_ignored_files.append(file)
    return non_ignored_files


def convert_to_txt(files):
    txt_files = []
    for file in files:
        if os.path.exists(file):  # Check if the source file exists
            new_filename = os.path.splitext(file)[0] + ".txt"
            try:
                os.rename(file, new_filename)
                txt_files.append(new_filename)
            except Exception as e:
                print(f"Error renaming file {file} to {new_filename}: {e}")
        else:
            print(f"Source file {file} does not exist. Skipping.")
    return txt_files


def duplicate_files(files, output_folder):
    if not os.path.exists(output_folder):
        os.mkdir(output_folder)

    duplicated_files = []
    for file in files:
        # Exclude files named "freerepo.py"
        if os.path.basename(file) != "freerepo.py":
            duplicated_file = os.path.join(output_folder, os.path.basename(file) + ".duplicate")
            shutil.copy2(file, duplicated_file)
            duplicated_files.append(duplicated_file)
    return duplicated_files



def is_inside_ignore_folder(file_path, ignore_folder):
    if os.path.isabs(file_path) and os.path.isabs(ignore_folder):
        return os.path.commonpath([file_path, ignore_folder]) == ignore_folder
    elif not os.path.isabs(file_path) and not os.path.isabs(ignore_folder):
        return os.path.commonpath([file_path, os.path.abspath(ignore_folder)]) == os.path.abspath(ignore_folder)
    else:
        return False


def generate_html_with_iframes_and_p_elements(files, output_file, branches_html_file, file_percentages):
    with open(output_file, "w") as f:
        f.write('<!--This is the readme, edit and fill in with your details or redo it completely :) -->')
        f.write('<html>\n')
        f.write('<head>\n')
        f.write('<style type="text/css">\n')
        
        # Add CSS styles for headers and paragraphs
        f.write('h1, h2, p {\n')
        f.write('    text-align: center;\n')
        f.write('}\n')
        
        # Add CSS styles for spacer
        f.write('.spacer {\n')
        f.write('    padding-top: 40px;\n')
        f.write('    text-size: 0px;\n')
        f.write('}\n')
        
        # Add CSS styles for links
        f.write('a {\n')
        f.write('    color: black;\n')
        f.write('    text-decoration: none;\n')
        f.write('}\n')
        
        # Add CSS styles for body
        f.write('body {\n')
        f.write('    background-color: green;\n')
        f.write('    margin: 0;\n')
        f.write('    font-family: Arial, sans-serif;\n')
        f.write('}\n')
        
        # Add CSS styles for select dropdown
        f.write('select {\n')
        f.write('    margin: 10px;\n')
        f.write('    padding: 5px;\n')
        f.write('    font-size: 16px;\n')
        f.write('}\n')
        
        f.write('</style>\n')
        f.write('</head>\n')
        f.write('<body>\n')
        f.write('<h1>Your Repo</h1>\n')
        f.write('<h2>Edit the yourrepo.html to change this details.</h2>\n')
        f.write('<p class="spacer2"></p>\n')
        f.write('<h1>Lorem ipsum dolor sit amet, consectetur adipiscing elit. Fusce interdum vitae eros in luctus.\n')
        f.write('  <br>Suspendisse potenti. In hac habitasse platea dictumst.\n')
        f.write('  <br>Pellentesque tincidunt iaculis turpis et luctus.</h1>\n')
        f.write('<p class="spacer"></p>\n')
        f.write('<h1>* COMING SOON *</h1>\n')
        f.write('<h2>Sed et libero sed quam vulputate ultrices.</h2>\n')
        f.write('<!-- END OF README -->\n')

        f.write('<select id="branches" onchange="location = this.value;">\n')
        f.write(f'  <option value="{os.path.basename(branches_html_file)}">Master</option>\n')
        f.write(f'  <option value="ex1.html">EXP Branch</option>\n')  
        f.write('</select>\n')

        f.write(f'<!-- Uncomment the following lines to display commit_message.html -->\n')
        f.write(f'<!-- <p>commit_message.html</p>\n')
        f.write(f'<!-- <iframe src="commit_message.html" width="100%" height="400px"></iframe> -->\n')
        
        f.write(f'<!-- File Percentages -->\n')
        f.write(f'<p>Code Data:</p>\n')
        f.write(f'<iframe src="file_percentages.html" width="100%" height="400px"></iframe>\n')
        
        for file in files:
            filename = os.path.basename(file)
            file_name_without_extension = os.path.splitext(filename)[0]
            f.write(f'<p>{file_name_without_extension}</p>\n')
            f.write(f'<iframe src="{filename}" width="100%" height="400px"></iframe>\n')

        f.write(f'<p><a href="{os.path.join("yourrepo.zip")}" download><button>Download This Repo</button></a></p>\n')
        f.write('</body>\n')
        f.write('</html>')

def generate_ex_html(ex_html_file):
    with open(ex_html_file, "w") as f:
        f.write("<html>\n")
        f.write("<head>\n")
        f.write("<title>Ex1</title>\n")
        f.write("</head>\n")
        f.write("<body>\n")
        f.write('<h1>This is the Ex1 HTML page</h1>\n')

        f.write('<button onclick="location.href=\'yourrepo.html\'">Go back to Your Repo</button>\n')

        f.write("</body>\n")
        f.write("</html>\n")

def generate_commit_msg_html(commit_msg_file, commit_msg, file_percentages):
    current_date = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    with open(commit_msg_file, "w") as f:
        f.write("<html>\n")
        f.write("<head>\n")
        f.write("<title>Commit Message</title>\n")
        f.write("</head>\n")
        f.write("<body>\n")
        f.write(f"<p>Date: {current_date}</p>\n")
        f.write(f"<p>Commit msg: {commit_msg}</p>\n")
        
        f.write("<p>File Percentages:</p>\n")
        f.write(f"<pre>{file_percentages}</pre>\n")
        
        f.write("</body>\n")
        f.write("</html>\n")

def calculate_file_percentages(directory, file_counts):
    code_extensions = ['.html', '.css', '.js', '.py']  # Add more extensions if needed
    
    for filename in os.listdir(directory):
        if os.path.isfile(os.path.join(directory, filename)):
            if filename != 'freerepo.py':  # Exclude 'freerepo.py' from the calculation
                _, extension = os.path.splitext(filename)
                if extension:
                    extension = extension.lower()  # Convert extension to lowercase
                    file_counts[extension] = file_counts.get(extension, 0) + 1

def generate_file_percentages_html(file_counts, output_file):
    total_files = sum(file_counts.values())
    with open(output_file, "w") as f:
        f.write("<html>\n")
        f.write("<head>\n")
        f.write("<title>File Percentages</title>\n")
        f.write("</head>\n")
        f.write("<body>\n")
        f.write("<h1>File Percentages</h1>\n")
        
        for extension, count in file_counts.items():
            percentage = (count / total_files) * 100
            f.write(f"<p>{extension}: {percentage:.2f}%</p>\n")
        
        f.write("</body>\n")
        f.write("</html>\n")

def main():
    print("Choose an option:")
    print("1. Create Repo")
    print("2. Update Repo")

    option = input("Enter the option number (1 or 2): ")
    output_folder = "YourRepo"

    if option == "1":
        extensions_to_find = [".md", ".html", ".css", ".js", ".py"]

        matching_files = find_files_with_extensions(extensions_to_find)

        if not os.path.exists(output_folder):
            os.mkdir(output_folder)

        ignore_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), "ignore"))
        non_ignored_files = get_non_ignored_files(matching_files, ignore_folder)

        duplicated_files = duplicate_files(non_ignored_files, output_folder)

        txt_files = convert_to_txt(duplicated_files)

        output_html_file = os.path.join(output_folder, "yourrepo.html")
        branches_html_file = os.path.join(output_folder, "branches.html")

        file_counts = {}
        calculate_file_percentages(os.path.dirname(os.path.abspath(__file__)), file_counts)
        total_files = sum(file_counts.values())
        
        file_percentages_html_file = os.path.join(output_folder, "file_percentages.html")
        generate_file_percentages_html(file_counts, file_percentages_html_file)

        generate_html_with_iframes_and_p_elements(txt_files, output_html_file, branches_html_file, file_percentages_html_file)
        generate_ex_html(os.path.join(output_folder, "ex1.html"))

        with zipfile.ZipFile(os.path.join(output_folder, "yourrepo.zip"), "w") as zipf:
            for file in non_ignored_files:
                # Exclude files named "freerepo.py" from the zip archive
                if not os.path.basename(file) == "freerepo.py":
                    zipf.write(file, os.path.relpath(file, start=output_folder))



        print(f"HTML file '{output_html_file}' created with iframes, <p> elements, and file percentages.")
        print("The generated files are stored in the 'YourRepo' folder.")

    elif option == "2":
        extensions_to_find = [".md", ".html", ".css", ".js", ".py"]

        matching_files = find_files_with_extensions(extensions_to_find)

        ignore_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), "ignore"))
        non_ignored_files = get_non_ignored_files(matching_files, ignore_folder)

        duplicated_files = duplicate_files(non_ignored_files, output_folder)

        txt_files = convert_to_txt(duplicated_files)

        commit_msg_html_file = os.path.join(output_folder, "commit_message.html")

        file_counts = {}
        calculate_file_percentages(os.path.dirname(os.path.abspath(__file__)), file_counts)
        total_files = sum(file_counts.values())
        file_percentages = ""
        for extension, count in file_counts.items():
            percentage = (count / total_files) * 100
            file_percentages += f"{extension}: {percentage:.2f}%\n"

        commit_msg = input("Enter the commit message: ")
        generate_commit_msg_html(commit_msg_html_file, commit_msg, file_percentages)

        print(f"HTML file '{commit_msg_html_file}' created with the specified commit message and file percentages.")
        print("The generated files are stored in the 'YourRepo' folder.")
        
        
        with zipfile.ZipFile(os.path.join(output_folder, "yourrepo.zip"), "w") as zipf:
            for file in non_ignored_files:
                # Exclude files named "freerepo.py" from the zip archive
                if not os.path.basename(file) == "freerepo.py":
                    zipf.write(file, os.path.relpath(file, start=output_folder))



if __name__ == "__main__":
    while True:
        main()

        run_again = input("Do you want to run another script? (y/n): ")
        if run_again.lower() != "y":
            break

    print("Goodbye!")