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(folder_name):
    print(f"Running docker-compose for {folder_name}")
    os.chdir(CLR_TEST_DIR)
    subprocess.call(["docker", "compose", "down", "--volumes"])
    subprocess.call(["docker", "compose", "up", "--build", "--abort-on-container-exit", "--exit-code-from", "clearing_system"])
    os.chdir("..")


def main():
    folders = [i for i in sorted(os.listdir(DATA_ROOT_DIR)) if re.search("\\d{1,}.\\d{1,}.*", i)]
    shutil.rmtree(f"{CLR_TEST_DIR}/clearing-all/clearing_testing_utils/store/expected")
    shutil.rmtree(f"{CLR_TEST_DIR}/clearing-all/clearing_testing_utils/store/input")
    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}/{folders[0]}/db/", f"{CLR_TEST_DIR}/db/", dirs_exist_ok=True)
    shutil.copytree(f"{DATA_ROOT_DIR}/{folders[0]}/clearing_testing_utils/", f"{CLR_TEST_DIR}/clearing-all/clearing_testing_utils/", dirs_exist_ok=True)
    run_docker(folders[0])
    # 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(folder)


if __name__=='__main__':
    main()