53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
import os
|
|
import importlib.resources
|
|
|
|
from pyinfra.operations import apt, files, pip, server, systemd
|
|
|
|
|
|
# install into .local/lib
|
|
def install(unix_user, enable_linger=True) -> None:
|
|
remember_package = importlib.resources.files("remember_remember_bot")
|
|
working_directory = os.getcwd()
|
|
remote_venv_dir = f"{working_directory}/venv"
|
|
|
|
# install systemd units
|
|
fn = "remember-remember-bot"
|
|
service_file = remember_package / "pyinfra_assets" / f"{fn}.service.j2"
|
|
if enable_linger:
|
|
server.shell(
|
|
name=f"enable {unix_user}'s systemd units to auto-start at boot",
|
|
commands=[f"loginctl enable-linger {unix_user}"],
|
|
)
|
|
files.line(
|
|
name=f"{unix_user} .profile: ensure XDG_RUNTIME_DIR",
|
|
path=f"/home/{unix_user}/.profile",
|
|
line='export XDG_RUNTIME_DIR="/run/user/$UID"',
|
|
)
|
|
files.line(
|
|
name=f"{unix_user} .profile: ensure DBUS_SESSION_BUS_ADDRESS",
|
|
path=f"/home/{unix_user}/.profile",
|
|
line='export DBUS_SESSION_BUS_ADDRESS="unix:path=${XDG_RUNTIME_DIR}/bus"',
|
|
)
|
|
|
|
files.template(
|
|
name=f"Upload {fn}.service",
|
|
src=service_file,
|
|
dest=f"/home/{unix_user}/.config/systemd/user/{fn}.service",
|
|
execpath=f"{remote_venv_dir}/bin/{fn}",
|
|
working_directory=working_directory,
|
|
unix_user=unix_user,
|
|
)
|
|
systemd.service(
|
|
name=f"Setup {fn} service",
|
|
service=f"{fn}.service",
|
|
running=True,
|
|
enabled=True,
|
|
restarted=True,
|
|
daemon_reload=True,
|
|
user_mode=True,
|
|
)
|
|
|
|
|
|
unix_user = os.environ.get("REMEMBER_PYINFRA_USER", os.environ.get("USER"))
|
|
install(unix_user)
|