import decman
from decman.plugins import pacman

import config


class VirtualMachines(decman.Module):
    def __init__(self):
        """Set up virtual machines"""
        super().__init__("virtualmachines")

    @pacman.packages
    def pkgs(self) -> set[str]:
        return {
            "dnsmasq",
            "libayatana-appindicator",
            "libvirt-python",
            "lvm2",
            "qemu-base",
            "vde2",
            "virt-manager",
            "virt-viewer",
        }

    def on_enable(self, store: decman.Store):
        # TODO: Everywhere, make user configurable
        decman.prg(["usermod", "-G", "libvirt", "-a", config.user])