From 68b2a8074ca7d6ee95c2382608559a698fd1be15 Mon Sep 17 00:00:00 2001
From: Lucas Noki <lucas-noki@gmx.de>
Date: Sun, 19 Jan 2020 21:52:29 +0100
Subject: [PATCH] Add functions to make swap persistent, add function to show
 swapiness

---
 modules/swap/module.py | 31 +++++++++++++++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

diff --git a/modules/swap/module.py b/modules/swap/module.py
index e810041..264e540 100644
--- a/modules/swap/module.py
+++ b/modules/swap/module.py
@@ -11,7 +11,9 @@ class SwapModule(AbstractModule):
 
     def run(self):
 
-        actions = ["Create/Resize", "Delete", "Get swap location", "Get swap size"]
+        actions = ["Create/Resize temp. swap", "Create/Resize persistent swap", "Delete swap",
+                   "Make temp. swap persistent", "Get swap location", "Get swap size",
+                   "Show swapiness"]
         menu = [
             {
                 'type': 'list',
@@ -21,8 +23,15 @@ class SwapModule(AbstractModule):
             }
         ]
         selected_action = prompt(menu)['action']
-        if selected_action == "Create/Resize":
+        if selected_action == "Create/Resize temp. swap":
             self.create_resize_swap()
+        elif selected_action == "Create/Resize persistent swap":
+            self.create_resize_swap()
+            self.make_swap_persistent()
+        elif selected_action == "Make temp. swap persistent":
+            self.make_swap_persistent()
+        elif selected_action == "Show swapiness":
+            self.show_swapiness()
         elif selected_action == "Delete":
             self.delete_swap()
         elif selected_action == "Get swap location":
@@ -63,8 +72,12 @@ class SwapModule(AbstractModule):
             print(output_free.strip())
 
         else:
+            swap_size = subprocess.check_output(['swapon', '--show']).decode("UTF-8")
+            li = list(swap_size.split())
+            final_list = li
             print("")
             print("Swap already installed! You can resize it!")
+            print("Curr. swap size: {}b".format(final_list[7]))
             resize = input("How big should your swap become(numbers in Gigabyte)? ")
             resize_swapfile = """sudo swapoff /swapfile && 
             sudo fallocate -l {}G /swapfile && 
@@ -74,6 +87,20 @@ class SwapModule(AbstractModule):
             output_free = subprocess.check_output(['free', '-h']).decode("UTF-8")
             print(output_free.strip())
 
+    def make_swap_persistent(self):
+        self.swap_location_check()
+        backup_fstab = "sudo cp /etc/fstab /etc/fstab.bak"
+        enable_persistence = "echo '{} none swap sw 0 0' | sudo tee -a /etc/fstab".format(self.swap_file)
+
+        subprocess.call(backup_fstab, shell=True)
+        subprocess.call(enable_persistence, shell=True)
+        print("Swap is now persistent!")
+
+    def show_swapiness(self):
+        get_swapiness = "cat /proc/sys/vm/swappiness"
+
+        subprocess.call(get_swapiness, shell=True)
+
     def delete_swap(self):
         self.swap_location_check()
         disable_swapfile = """sudo swapoff {0} &&