#!/usr/bin/env python3

from modules import configloader
from modules import compressor
from modules import printer
from modules import utils
from datetime import datetime
import shutil
import sys
import os


def get_file_type(filename):
    audio_ext = ['.aac', '.flac', '.m4a', '.mp3', '.ogg', '.opus', '.raw', '.wav', '.wma']
    image_ext = ['.apng', '.avif', '.bmp', '.tga', '.tiff', '.dds', '.svg', '.webp', '.jpg', '.jpeg', '.png']
    video_ext = ['.3gp' '.amv', '.avi', '.gif', '.m2t', '.m4v', '.mkv', '.mov', '.mp4', '.m4v', '.mpeg', '.mpv',
                 '.webm', '.ogv']

    if os.path.splitext(filename)[1] in audio_ext:
        return "audio"
    elif os.path.splitext(filename)[1] in image_ext:
        return "image"
    elif os.path.splitext(filename)[1] in video_ext:
        return "video"
    else:
        return "unknown"


if __name__ == "__main__":
    start_time = datetime.now()
    printer.win_ascii_esc()
    try:
        if sys.argv[1][len(sys.argv[1])-1] == "/":
            arg_path = sys.argv[1][:len(sys.argv[1])-1]
        else:
            arg_path = sys.argv[1]
    except IndexError:
        print(utils.help_message())
        exit()

    orig_folder = os.path.abspath(arg_path)
    printer.orig_folder = os.path.abspath(arg_path)

    printer.bar_init(orig_folder)

    if os.path.exists(f"{orig_folder}_compressed"):
        shutil.rmtree(f"{orig_folder}_compressed")

    printer.info("Creating folders...")
    for folder, folders, files in os.walk(orig_folder):
        if not os.path.exists(folder.replace(orig_folder, f"{orig_folder}_compressed")):
            os.mkdir(folder.replace(orig_folder, f"{orig_folder}_compressed"))

        printer.info(f"Compressing \"{folder.replace(orig_folder, orig_folder.split('/').pop())}\" folder...")
        target_folder = folder.replace(orig_folder, f"{orig_folder}_compressed")
        for file in os.listdir(folder):
            if os.path.isfile(f'{folder}/{file}'):
                match get_file_type(file):
                    case "audio":
                        comp_file = compressor.compress_audio(folder, file, target_folder,
                                                              configloader.config['AUDIO']['Extension'])
                    case "image":
                        comp_file = compressor.compress_image(folder, file, target_folder,
                                                              configloader.config['IMAGE']['Extension'])
                    case "video":
                        comp_file = compressor.compress_video(folder, file, target_folder,
                                                              configloader.config['VIDEO']['Extension'])
                    case "unknown":
                        comp_file = compressor.compress(folder, file, target_folder)

                if configloader.config['FFMPEG']['MimicMode']:
                    try:
                        os.rename(comp_file, f'{folder}/{file}'.replace(orig_folder, f"{orig_folder}_compressed"))
                    except FileNotFoundError:
                        pass

    utils.get_compression_status(orig_folder)
    utils.sys_pause()
    print(f"Time taken: {datetime.now() - start_time}")