from modules import printer
from shutil import copyfile
from glob import glob
import os


def get_dir_size(directory, files):
    total_size = 0
    for folder, folders, files in os.walk(directory):
        for file in files:
            if not os.path.islink(f"{folder}/{file}"):
                total_size += os.path.getsize(f"{folder}/{file}")
    return total_size


def get_compression(orig, comp):
    processed_files = []
    for folder, folders, files in os.walk(comp):
        for file in files:
            processed_files.append(file)

    try:
        comp = 100 - int((get_dir_size(comp, processed_files) / get_dir_size(orig, processed_files)) * 100)
        if comp < 0:
            printer.warning(f'Compression: {comp}%')
            printer.warning("The resulting files are larger than the original ones!")
        else:
            printer.info(f'Compression: {comp}%')
    except ZeroDivisionError:
        printer.warning("Nothing compressed!")


def get_compression_status(orig_folder):
    orig_folder_len = 0
    comp_folder_len = 0

    for folder, folders, files in os.walk(orig_folder):
        orig_folder_len += len(files)

    for folder, folders, files in os.walk(f'{orig_folder}_compressed'):
        for file in files:
            if not os.path.splitext(file)[1].count(" (copy)"):
                comp_folder_len += 1

    if orig_folder_len == comp_folder_len:
        printer.info("Success!")
        get_compression(orig_folder, f"{orig_folder}_compressed")
    else:
        printer.warning("Some files failed to compress!")
        get_compression(orig_folder, f"{orig_folder}_compressed")


def add_unprocessed_files(orig_folder):
    for folder, folders, files in os.walk(orig_folder):
        for file in files:
            new_folder = f"{folder}".replace(orig_folder, f"{orig_folder}_compressed")
            if len(glob(f"{folder}/{os.path.splitext(file)[0]}*")) != 1:
                if len(glob(f"{new_folder}/{file}")):
                    copyfile(f"{folder}/{file}", f"{new_folder}/{file} (copy)")
                    printer.warning(f'Duplicate file has been found! Check manually this files - "{file}", "{file} (copy)"')
                else:
                    copyfile(f"{folder}/{file}", f"{new_folder}/{file}")
                    printer.info(f"File {file} copied to compressed folder.")
            else:
                if not len(glob(f"{folder}_compressed/{os.path.splitext(file)[0]}*")):
                    copyfile(f"{folder}/{file}", f"{new_folder}/{file}")
                    printer.info(f"File {file} copied to compressed folder.")


def help_message():
    return "Usage: ffmpeg-comp {folder}"