You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
import os
|
|
import time
|
|
import logging
|
|
|
|
from vanilla_first_setup.utils import checks
|
|
from vanilla_first_setup.utils.apt import Apt
|
|
from vanilla_first_setup.utils.flatpak import Flatpak
|
|
from vanilla_first_setup.utils.snap import Snap
|
|
|
|
|
|
logger = logging.getLogger("FirstSetup::Configurator")
|
|
|
|
|
|
class Configurator:
|
|
|
|
def __init__(self, config: 'Config', fake: bool = False):
|
|
self.config = config
|
|
self.fake = fake
|
|
|
|
def apply(self):
|
|
self.__enable_snap() if self.config.snap else self.__disable_snap()
|
|
self.__enable_flatpak() if self.config.flatpak else self.__disable_flatpak()
|
|
self.__enable_apport() if self.config.apport else self.__disable_apport()
|
|
|
|
def __fake(self, msg: str):
|
|
time.sleep(1)
|
|
logger.info(f"Fake: {msg}")
|
|
|
|
def __enable_snap(self):
|
|
if self.fake:
|
|
return self.__fake("Fake: Snap enabled")
|
|
|
|
if not checks.is_snap_installed():
|
|
Apt.install(['snapd', 'gnome-software-plugin-snap'])
|
|
Apt.update()
|
|
|
|
if not self.config.flatpak:
|
|
Snap.install(['snap-store'])
|
|
|
|
def __disable_snap(self):
|
|
if self.fake:
|
|
return self.__fake("Fake: Snap disabled")
|
|
|
|
if checks.is_snap_installed():
|
|
Apt.purge(['snapd'])
|
|
|
|
def __enable_flatpak(self):
|
|
if self.fake:
|
|
return self.__fake("Fake: Flatpak enabled")
|
|
|
|
if not checks.is_flatpak_installed():
|
|
Apt.install(['flatpak'])
|
|
Apt.update()
|
|
Flatpak.add_repo("https://flathub.org/repo/flathub.flatpakrepo")
|
|
|
|
def __disable_flatpak(self):
|
|
if self.fake:
|
|
return self.__fake("Fake: Flatpak disabled")
|
|
|
|
if checks.is_flatpak_installed():
|
|
Apt.purge(['flatpak'])
|
|
|
|
def __enable_apport(self):
|
|
if self.fake:
|
|
return self.__fake("Fake: Apport enabled")
|
|
|
|
if not checks.is_apport_installed():
|
|
Apt.install(['apport'])
|
|
Apt.update()
|
|
|
|
def __disable_apport(self):
|
|
if self.fake:
|
|
return self.__fake("Fake: Apport disabled")
|
|
|
|
if checks.is_apport_installed():
|
|
Apt.purge(['apport'])
|
|
|
|
def __disable_on_startup(self):
|
|
if self.fake:
|
|
return self.__fake("Fake: Disable on startup")
|
|
|
|
autostart_file = os.path.expanduser("~/.config/autostart/vanilla-first-setup.desktop")
|
|
if os.path.exists(autostart_file):
|
|
os.remove(autostart_file)
|
|
|
|
def __enable_on_startup(self):
|
|
if self.fake:
|
|
return self.__fake("Fake: Enable on startup")
|
|
|
|
autostart_file = os.path.expanduser("~/.config/autostart/vanilla-first-setup.desktop")
|
|
if not os.path.exists(autostart_file):
|
|
with open(autostart_file, "w") as f:
|
|
f.write("[Desktop Entry]")
|
|
f.write("Type=Application")
|
|
f.write("Name=Vanilla OS First Setup")
|
|
f.write("Exec=vanilla-first-setup")
|
|
f.write("Terminal=false")
|
|
f.write("X-GNOME-Autostart-enabled=true")
|
|
|