import os
from os.path import join
import subprocess


def test_sentry_admin(setup_backup_restore_env_variables):
    sentry_admin_sh = os.path.join(os.getcwd(), "sentry-admin.sh")
    output = subprocess.run(
        [sentry_admin_sh, "--help"], check=True, capture_output=True, encoding="utf8"
    ).stdout
    assert "Usage: ./sentry-admin.sh" in output
    assert "SENTRY_DOCKER_IO_DIR" in output

    output = subprocess.run(
        [sentry_admin_sh, "permissions", "--help"],
        check=True,
        capture_output=True,
        encoding="utf8",
    ).stdout
    assert "Usage: ./sentry-admin.sh permissions" in output


def test_01_backup(setup_backup_restore_env_variables):
    # Docker was giving me permission issues when trying to create this file and write to it even after giving read + write access
    # to group and owner. Instead, try creating the empty file and then give everyone write access to the backup file
    file_path = os.path.join(os.getcwd(), "sentry", "backup.json")
    sentry_admin_sh = os.path.join(os.getcwd(), "sentry-admin.sh")
    open(file_path, "a", encoding="utf8").close()
    os.chmod(file_path, 0o666)
    assert os.path.getsize(file_path) == 0
    subprocess.run(
        [
            sentry_admin_sh,
            "export",
            "global",
            "/sentry-admin/backup.json",
            "--no-prompt",
        ],
        check=True,
    )
    assert os.path.getsize(file_path) > 0


def test_02_import(setup_backup_restore_env_variables):
    # Bring postgres down and recreate the docker volume
    subprocess.run(["docker", "compose", "--ansi", "never", "down"], check=True)
    # We reset all DB-related volumes here and not just Postgres although the backups
    # are only for Postgres. The reason is to get a "clean slate" as we need the Kafka
    # and Clickhouse volumes to be back to their initial state as well (without any events)
    # We cannot just rm and create them as they still need the migrations.
    for name in ("postgres", "clickhouse", "kafka"):
        subprocess.run(["docker", "volume", "rm", f"sentry-{name}"], check=True)
        subprocess.run(["docker", "volume", "create", f"sentry-{name}"], check=True)
        subprocess.run(
            [
                "rsync",
                "-aW",
                "--super",
                "--numeric-ids",
                "--no-compress",
                "--mkpath",
                join(os.environ["RUNNER_TEMP"], "volumes", f"sentry-{name}", ""),
                f"/var/lib/docker/volumes/sentry-{name}/",
            ],
            check=True,
            capture_output=True,
        )

    sentry_admin_sh = os.path.join(os.getcwd(), "sentry-admin.sh")
    subprocess.run(
        [
            sentry_admin_sh,
            "import",
            "global",
            "/sentry-admin/backup.json",
            "--no-prompt",
        ],
        check=True,
    )
    # TODO: Check something actually restored here like the test user we have from earlier