import invoke

import services.plex.tasks
import services.deluge.tasks
import services.traefik.tasks

ns = invoke.Collection()

PYINFRA_COMMON_PREFIX = "pyinfra -vvv pyinfra/inventory.py"

@invoke.task()
def system_updates(ctx):
    ctx.run(f"{PYINFRA_COMMON_PREFIX} pyinfra/system_updates.py")


@invoke.task()
def system_reboot(ctx):
    ctx.run(f"{PYINFRA_COMMON_PREFIX} pyinfra/reboot.py")


server = invoke.Collection("server")

server.add_task(system_updates, name="update")
server.add_task(system_reboot, name="reboot")

ns.add_collection(server)

ns.add_collection(services.plex.tasks.ns)
ns.add_collection(services.traefik.tasks.ns)
ns.add_collection(services.deluge.tasks.ns)