import os

from fabric import Connection
from invoke import task


@task()
def deploy(c):
    """Make sure proxy_user is set to your neuf username."""
    project_path = "/opt/mdb"
    proxy_user = os.getenv("DEPLOY_USER", os.getenv("USER"))

    c = Connection(
        host="gitdeploy@sega.neuf.no",
        gateway=Connection("login.neuf.no", user=proxy_user),
    )

    with (
        c.cd(project_path),
        c.prefix("source {}/.venv/bin/activate".format(project_path)),
    ):
        c.run("git pull")  # Get source
        c.run("pip install -U uv")
        c.run("uv sync")  # install deps in virtualenv
        c.run("uv run manage.py collectstatic --noinput")  # Collect static
        c.run("uv run manage.py migrate")  # Run DB migrations

    # Reload gunicorn
    c.sudo("/usr/bin/supervisorctl pid mdb.neuf.no | xargs kill -HUP", shell=False)