import os
from types import SimpleNamespace

import yaml

from tests.regression.answer_quality.cli_utils import manage_data_directories
from tests.regression.answer_quality.cli_utils import set_env_variables
from tests.regression.answer_quality.cli_utils import start_docker_compose
from tests.regression.answer_quality.cli_utils import switch_to_commit


def load_config(config_filename: str) -> SimpleNamespace:
    current_dir = os.path.dirname(os.path.abspath(__file__))
    config_path = os.path.join(current_dir, config_filename)
    with open(config_path, "r") as file:
        return SimpleNamespace(**yaml.safe_load(file))


def main() -> None:
    config = load_config("search_test_config.yaml")
    if config.environment_name:
        env_name = config.environment_name
        print("launching danswer with environment name:", env_name)
    else:
        print("No env name defined. Not launching docker.")
        print(
            "Please define a name in the config yaml to start a new env "
            "or use an existing env"
        )
        return

    set_env_variables(
        config.model_server_ip,
        config.model_server_port,
        config.use_cloud_gpu,
        config.llm,
    )
    manage_data_directories(env_name, config.output_folder, config.use_cloud_gpu)
    if config.commit_sha:
        switch_to_commit(config.commit_sha)

    start_docker_compose(
        env_name, config.launch_web_ui, config.use_cloud_gpu, config.only_state
    )


if __name__ == "__main__":
    main()