import argparse, pygame, pathlib, os
from lib.pgfw.pgfw import extension

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("input_path", type=pathlib.Path)
    parser.add_argument("output_path", type=pathlib.Path)
    parser.add_argument("shift", type=int)
    arguments = parser.parse_args()
    pygame.display.set_mode((200, 100))
    pygame.init()
    for root, dirs, files in os.walk(arguments.input_path):
        for name in files:
            if name.endswith(".png"):
                input_file_path = os.path.join(root, name)
                shifted = extension.get_hue_shifted_surface(pygame.image.load(input_file_path).convert_alpha(), arguments.shift)
                output_file_path = pathlib.Path(input_file_path.replace(str(arguments.input_path), str(arguments.output_path)))
                os.makedirs(output_file_path.parent, exist_ok=True)
                pygame.image.save(shifted, str(output_file_path))
                print(f"saved hue shifted {input_file_path} to {output_file_path}")