#!/bin/python3

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

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 = arg_path
printer.orig_folder = 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 compressor.get_file_type(file):
                case "audio":
                    compressor.compress_audio(folder, file, target_folder)
                case "image":
                    compressor.compress_image(folder, file, target_folder)
                case "video":
                    compressor.compress_video(folder, file, target_folder)
                case "unknown":
                    compressor.compress(folder, file, target_folder)
utils.get_compression_status(orig_folder)