import os
import re
import shutil
import subprocess


DATA_ROOT_DIR = os.environ['CLEARING_TEST_DIR']
CLR_TEST_DIR = os.environ['CLR_TEST_DIR']


def run_docker():
    os.chdir(CLR_TEST_DIR)
    subprocess.call(["docker", "compose", "down", "--volumes"])
    subprocess.call(["docker", "compose", "up", "--abort-on-container-exit", "--exit-code-from", "clearing_system"])
    

def main():
    folders = [i for i in sorted(os.listdir(DATA_ROOT_DIR)) if re.search("\\d{1,}.\\d{1,}.*", i)]
    for folder in folders:
        shutil.rmtree(f"{CLR_TEST_DIR}/clearing_all/clearing_testing_utils/store")
        os.remove(f"{CLR_TEST_DIR}/db/export.sql")
        os.remove(f"{CLR_TEST_DIR}/db/test-data.sql")
        shutil.copytree(f"{DATA_ROOT_DIR}/{folder}/db/", f"{CLR_TEST_DIR}/db/", dirs_exist_ok=True)
        shutil.copytree(f"{DATA_ROOT_DIR}/{folder}/clearing_testing_utils/", f"{CLR_TEST_DIR}/clearing_all/clearing_testing_utils/", dirs_exist_ok=True)
        run_docker()
        os.chdir("..")


if __name__=='__main__':
    main()