import decman
from decman.plugins import pacman


class VirtualMachines(decman.Module):
    def __init__(self, user: str):
        """Set up virtual machines"""
        self._user = user
        super().__init__("virtualmachines")

    @pacman.packages
    def pkgs(self) -> set[str]:
        return {
            "dnsmasq",
            "ebtables",
            "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", self._user])