import os
import shutil
out_subs_path = os.path.join(os.getcwd(), "subs")
in_subs_path = os.path.join(out_subs_path, "orig")

MEDIA_TYPES = [".mp4", ".mkv", ".m4v", ".avi", ".webm"]
SUBS_TYPES = [".srt", ".ass", ".vtt"]

media_names = []
subs_name_exts = []
subs_exts = []

for element in os.scandir(): # use dir indexing to rename
    if not element.is_file(follow_symlinks=False): continue

    ext_start: int = element.name.rfind(".")
    ext: str = element.name[ext_start: ]

    if ext in MEDIA_TYPES:
        media_names.append(element.name[ :ext_start])


for element in os.scandir(in_subs_path):
    if not element.is_file(follow_symlinks=False): continue

    ext_start: int = element.name.rfind(".")
    ext: str = element.name[ext_start: ]

    if ext in SUBS_TYPES:
         subs_name_exts.append(element.name)
         subs_exts.append(ext)


if len(media_names) <= len(subs_name_exts):
    for element in os.scandir(out_subs_path): # ooga booga clean up past names
        if not element.is_file(follow_symlinks=False): continue
        os.remove(element.path)

    for i, name in enumerate(media_names):
        shutil.copyfile(os.path.join(in_subs_path, subs_name_exts[i]), os.path.join(out_subs_path, media_names[i] + subs_exts[i]))