import os
import shutil

from conpeek_setup import util


def run(config):
    util.print_black_light("Preparing rabbitmq file")

    output_directory = os.path.join(util.get_output_path(), "rabbitmq")
    template_rabbitmq_directory = os.path.join(util.get_templates_path(), "rabbitmq")

    shutil.rmtree(output_directory, ignore_errors=True)
    os.makedirs(output_directory, exist_ok=True)

    rabbitmq_directory = os.path.join(util.get_output_path(), "etc/rabbitmq")
    os.makedirs(rabbitmq_directory, exist_ok=True)

    files_to_copy = [
        "rabbit_init.sh"
    ]

    for file in files_to_copy:
        util.copy_file(os.path.join(template_rabbitmq_directory, file), output_directory)

    util.basic_on_location_sed(os.path.join(template_rabbitmq_directory, "rabbit_init.sh"), 'MASTER_PASSWORD', config["machine_secrets"]["master_password"])
    util.copy_file(os.path.join(template_rabbitmq_directory, "rabbitmq.config"), rabbitmq_directory)