import os
import re
import shutil
import zipfile


def replacetext(file_name: str, search_text: str, replace_text: str) -> None:
    """
    Helper function to replace the color in the file.
    Can be used to replace any text in the file.

    Args:
        file_name (str): The file to replace the text in.
        search_text (str): The text to be replaced.
        replace_text (str): The text to replace with.

    Returns:
        None
    """
    with open(file_name, 'r+') as f:
        file = f.read()
        file = re.sub(search_text, replace_text, file)
        f.seek(0)
        f.write(file)
        f.truncate()


def zipdir(path, ziph):
    """
    Takes in a oath of a directory and zips it in a ziph.
    Util to zip a directory.
    Thanks https://stackoverflow.com/questions/46229764/python-zip-multiple-directories-into-one-zip-file
    """
    for root, _, files in os.walk(path):
        for file in files:
            ziph.write(os.path.join(root, file),
                       os.path.relpath(os.path.join(root, file),
                                       os.path.join(path, '..')))


def zip_multiple_folders(dir_list, zip_name, remove = True):
    zipf = zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED)
    for dir in dir_list:
        zipdir(dir, zipf)
        if remove:
            shutil.rmtree(dir)
    zipf.close()