From b4efbd01ffa74dd24a0e24df536229e8c08ef31a Mon Sep 17 00:00:00 2001
From: mirkobrombin <send@mirko.pm>
Date: Mon, 5 Sep 2022 13:35:36 +0200
Subject: [PATCH] add flatpak, snap, apt backend

also connect some widgets
---
 ubuntu_smoother/gtk/window.ui         | 10 +++---
 ubuntu_smoother/meson.build           |  3 ++
 ubuntu_smoother/models/__init__.py    |  0
 ubuntu_smoother/models/config.py      |  7 +++++
 ubuntu_smoother/models/meson.build    | 10 ++++++
 ubuntu_smoother/models/preset.py      |  6 ++++
 ubuntu_smoother/utils/__init__.py     |  0
 ubuntu_smoother/utils/apt.py          | 44 +++++++++++++++++++++++++++
 ubuntu_smoother/utils/checks.py       | 10 ++++++
 ubuntu_smoother/utils/configurator.py | 39 ++++++++++++++++++++++++
 ubuntu_smoother/utils/flatpak.py      | 25 +++++++++++++++
 ubuntu_smoother/utils/meson.build     | 13 ++++++++
 ubuntu_smoother/utils/snap.py         | 18 +++++++++++
 ubuntu_smoother/window.py             | 25 +++++++++++++++
 14 files changed, 205 insertions(+), 5 deletions(-)
 create mode 100644 ubuntu_smoother/models/__init__.py
 create mode 100644 ubuntu_smoother/models/config.py
 create mode 100644 ubuntu_smoother/models/meson.build
 create mode 100644 ubuntu_smoother/models/preset.py
 create mode 100644 ubuntu_smoother/utils/__init__.py
 create mode 100644 ubuntu_smoother/utils/apt.py
 create mode 100644 ubuntu_smoother/utils/checks.py
 create mode 100644 ubuntu_smoother/utils/configurator.py
 create mode 100644 ubuntu_smoother/utils/flatpak.py
 create mode 100644 ubuntu_smoother/utils/meson.build
 create mode 100644 ubuntu_smoother/utils/snap.py

diff --git a/ubuntu_smoother/gtk/window.ui b/ubuntu_smoother/gtk/window.ui
index 80618d5..03aedd8 100644
--- a/ubuntu_smoother/gtk/window.ui
+++ b/ubuntu_smoother/gtk/window.ui
@@ -38,7 +38,7 @@
                 <property name="valign">fill</property>
                 <property name="hexpand">true</property>
                 <child>
-                  <object class="GtkButton">
+                  <object class="GtkButton" id="btn_start">
                     <property name="label">Let's Start</property>
                     <property name="halign">center</property>
                     <style>
@@ -65,7 +65,7 @@
                             <property name="title">Snap</property>
                             <property name="subtitle">Will replace GNOME Software with Snap store if the only package manager.</property>
                             <child>
-                              <object class="GtkSwitch">
+                              <object class="GtkSwitch" id="switch_snap">
                                 <property name="valign">center</property>
                               </object>
                             </child>
@@ -76,7 +76,7 @@
                             <property name="title">Flatpak</property>
                             <property name="subtitle">Will also configure the Flathub repository.</property>
                             <child>
-                              <object class="GtkSwitch">
+                              <object class="GtkSwitch" id="switch_flatpak">
                                 <property name="valign">center</property>
                               </object>
                             </child>
@@ -92,7 +92,7 @@
                             <property name="title">Apport</property>
                             <property name="subtitle">Do you want to keep the Bug Reporter utility?</property>
                             <child>
-                              <object class="GtkSwitch">
+                              <object class="GtkSwitch" id="switch_apport">
                                 <property name="valign">center</property>
                               </object>
                             </child>
@@ -103,7 +103,7 @@
                   </object>
                 </child>
                 <child>
-                  <object class="GtkButton">
+                  <object class="GtkButton" id="btn_save">
                     <property name="label">Save Changes</property>
                     <property name="halign">center</property>
                     <style>
diff --git a/ubuntu_smoother/meson.build b/ubuntu_smoother/meson.build
index 11b2058..517e770 100644
--- a/ubuntu_smoother/meson.build
+++ b/ubuntu_smoother/meson.build
@@ -25,6 +25,9 @@ configure_file(
   install_dir: get_option('bindir')
 )
 
+subdir('utils')
+subdir('models')
+
 ubuntu_smoother_sources = [
   '__init__.py',
   'main.py',
diff --git a/ubuntu_smoother/models/__init__.py b/ubuntu_smoother/models/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/ubuntu_smoother/models/config.py b/ubuntu_smoother/models/config.py
new file mode 100644
index 0000000..ca8c1e0
--- /dev/null
+++ b/ubuntu_smoother/models/config.py
@@ -0,0 +1,7 @@
+class Config:
+
+    def __init__(self, snap: bool, flatpak: bool, apport: bool):
+        self.snap = snap
+        self.flatpak = flatpak
+        self.apport = apport
+        
\ No newline at end of file
diff --git a/ubuntu_smoother/models/meson.build b/ubuntu_smoother/models/meson.build
new file mode 100644
index 0000000..ed366b7
--- /dev/null
+++ b/ubuntu_smoother/models/meson.build
@@ -0,0 +1,10 @@
+pkgdatadir = join_paths(get_option('prefix'), get_option('datadir'), meson.project_name())
+modelsdir = join_paths(pkgdatadir, 'ubuntu_smoother/models')
+
+sources = [
+  '__init__.py',
+  'config.py',
+  'preset.py',
+]
+
+install_data(sources, install_dir: modelsdir)
\ No newline at end of file
diff --git a/ubuntu_smoother/models/preset.py b/ubuntu_smoother/models/preset.py
new file mode 100644
index 0000000..8fc22fe
--- /dev/null
+++ b/ubuntu_smoother/models/preset.py
@@ -0,0 +1,6 @@
+class Preset:
+
+    snap: bool = True
+    flatpak: bool = True
+    apport: bool = True
+    
\ No newline at end of file
diff --git a/ubuntu_smoother/utils/__init__.py b/ubuntu_smoother/utils/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/ubuntu_smoother/utils/apt.py b/ubuntu_smoother/utils/apt.py
new file mode 100644
index 0000000..e64d303
--- /dev/null
+++ b/ubuntu_smoother/utils/apt.py
@@ -0,0 +1,44 @@
+import subprocess
+
+
+class Apt:
+
+    @staticmethod
+    def install(self, packages: list):
+        subprocess.run(
+            ['sudo', 'apt', 'install'] + packages,
+            env={'DEBIAN_FRONTEND': 'noninteractive'},
+            check=True
+        )
+
+    @staticmethod
+    def remove(self, packages: list):
+        subprocess.run(
+            ['sudo', 'apt', 'remove'] + packages,
+            env={'DEBIAN_FRONTEND': 'noninteractive'},
+            check=True
+        )
+
+    @staticmethod
+    def purge(self, packages: list):
+        subprocess.run(
+            ['sudo', 'apt', 'purge'] + packages,
+            env={'DEBIAN_FRONTEND': 'noninteractive'},
+            check=True
+        )
+
+    @staticmethod
+    def update(self):
+        subprocess.run(
+            ['sudo', 'apt', 'update'],
+            env={'DEBIAN_FRONTEND': 'noninteractive'},
+            check=True
+        )
+
+    @staticmethod
+    def upgrade(self):
+        subprocess.run(
+            ['sudo', 'apt', 'upgrade'],
+            env={'DEBIAN_FRONTEND': 'noninteractive'},
+            check=True
+        )
diff --git a/ubuntu_smoother/utils/checks.py b/ubuntu_smoother/utils/checks.py
new file mode 100644
index 0000000..fafb79b
--- /dev/null
+++ b/ubuntu_smoother/utils/checks.py
@@ -0,0 +1,10 @@
+import shutil
+
+def is_snap_installed():
+    return shutil.which('snap') is not None
+
+def is_flatpak_installed():
+    return shutil.which('flatpak') is not None
+
+def is_apport_installed():
+    return shutil.which('apport') is not None
diff --git a/ubuntu_smoother/utils/configurator.py b/ubuntu_smoother/utils/configurator.py
new file mode 100644
index 0000000..4887fcc
--- /dev/null
+++ b/ubuntu_smoother/utils/configurator.py
@@ -0,0 +1,39 @@
+from ubuntu_smoother.utils import checks
+from ubuntu_smoother.utils.apt import Apt
+from ubuntu_smoother.utils.flatpak import Flatpak
+from ubuntu_smoother.utils.snap import Snap
+
+
+class Configurator:
+
+    def __init__(self, config: 'Config'):
+        self.config = config
+
+    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 __enable_snap(self):
+        pkgs = []
+        if not checks.is_snap_installed():
+            pkgs += ['snapd', 'gnome-software-plugin-snap']
+        Apt.install(pkgs)
+        Apt.update()
+
+        if not self.config.flatpak:
+            Snap.install(['snap-store'])
+            
+    def __disable_snap(self):
+        if checks.is_snap_installed():
+            Apt.purge(['snapd'])
+            
+    def __enable_flatpak(self):
+        if not checks.is_flatpak_installed():
+            Apt.install(['flatpak'])
+            Flatpak.add_repo("https://flathub.org/repo/flathub.flatpakrepo")
+        Apt.update()
+
+    def __disable_flatpak(self):
+        if checks.is_flatpak_installed():
+            Apt.purge(['flatpak'])
diff --git a/ubuntu_smoother/utils/flatpak.py b/ubuntu_smoother/utils/flatpak.py
new file mode 100644
index 0000000..ba0368c
--- /dev/null
+++ b/ubuntu_smoother/utils/flatpak.py
@@ -0,0 +1,25 @@
+import subprocess
+
+
+class Flatpak:
+
+    @staticmethod
+    def install(self, packages: list):
+        subprocess.run(
+            ['flatpak', 'install', '--user'] + packages,
+            check=True
+        )
+
+    @staticmethod
+    def remove(self, packages: list):
+        subprocess.run(
+            ['flatpak', 'remove', '--user'] + packages,
+            check=True
+        )
+
+    @staticmethod
+    def add_repo(self, repo: str):
+        subprocess.run(
+            ['flatpak', 'remote-add', '--user', '--if-not-exists', repo],
+            check=True
+        )
diff --git a/ubuntu_smoother/utils/meson.build b/ubuntu_smoother/utils/meson.build
new file mode 100644
index 0000000..7151ce0
--- /dev/null
+++ b/ubuntu_smoother/utils/meson.build
@@ -0,0 +1,13 @@
+pkgdatadir = join_paths(get_option('prefix'), get_option('datadir'), meson.project_name())
+utilsdir = join_paths(pkgdatadir, 'ubuntu_smoother/utils')
+
+sources = [
+  '__init__.py',
+  'checks.py',
+  'apt.py',
+  'flatpak.py',
+  'snap.py',
+  'configurator.py',
+]
+
+install_data(sources, install_dir: utilsdir)
\ No newline at end of file
diff --git a/ubuntu_smoother/utils/snap.py b/ubuntu_smoother/utils/snap.py
new file mode 100644
index 0000000..ccd16ba
--- /dev/null
+++ b/ubuntu_smoother/utils/snap.py
@@ -0,0 +1,18 @@
+import subprocess
+
+
+class Snap:
+
+    @staticmethod
+    def install(self, packages: list):
+        subprocess.run(
+            ['snap', 'install'] + packages,
+            check=True
+        )
+
+    @staticmethod
+    def remove(self, packages: list):
+        subprocess.run(
+            ['snap', 'remove'] + packages,
+            check=True
+        )
diff --git a/ubuntu_smoother/window.py b/ubuntu_smoother/window.py
index 57fcda3..44f6829 100644
--- a/ubuntu_smoother/window.py
+++ b/ubuntu_smoother/window.py
@@ -16,10 +16,35 @@
 
 from gi.repository import Gtk, Gio, Adw
 
+from ubuntu_smoother.models.preset import Preset
+from ubuntu_smoother.models.config import Config
+
 
 @Gtk.Template(resource_path='/pm/mirko/UbuntuSmoother/gtk/window.ui')
 class UbuntuSmootherWindow(Adw.ApplicationWindow):
     __gtype_name__ = 'UbuntuSmootherWindow'
 
+    carousel = Gtk.Template.Child()
+    btn_start = Gtk.Template.Child()
+    btn_save = Gtk.Template.Child()
+    switch_snap = Gtk.Template.Child()
+    switch_flatpak = Gtk.Template.Child()
+    switch_apport = Gtk.Template.Child()
+
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
+        self.__buiild_ui()
+        self.__connect_signals()
+
+    def __buiild_ui(self):
+        self.switch_snap.set_active(Preset.snap)
+        self.switch_flatpak.set_active(Preset.flatpak)
+        self.switch_apport.set_active(Preset.apport)
+
+    def __connect_signals(self):
+        self.btn_start.connect('clicked', self.__on_btn_start_clicked)
+
+    def __on_btn_start_clicked(self, widget):
+        index = int(self.carousel.get_position())
+        next_page = self.carousel.get_nth_page(index + 1)
+        self.carousel.scroll_to(next_page, True)