1 files changed, 107 insertions, 0 deletions
diff --git a/src/backend/rmpkgsrc.py b/src/backend/rmpkgsrc.py
new file mode 100644
index 0000000..17d5a12
--- /dev/null
+++ b/src/backend/rmpkgsrc.py
@@ -0,0 +1,107 @@
+#!/usr/bin/python3
+
+import atexit
+import io
+import os
+import pickle
+import signal
+import subprocess
+import sys
+import sisyphus.checkenv
+import sisyphus.getclr
+import sisyphus.getfs
+import sisyphus.killemerge
+import sisyphus.solverevdeps
+import sisyphus.syncdb
+
+
+def sigint_handler(signal, frame):
+    sys.exit(0)
+
+
+signal.signal(signal.SIGINT, sigint_handler)
+
+
+def start(pkgname, depclean=False, gfx_ui=False, unmerge=False):
+    args = ['--quiet', '--depclean']
+
+    if not sisyphus.checkenv.root() and (unmerge or depclean):
+        print(sisyphus.getclr.bright_red +
+              "\nYou need root permissions to do this!\n" + sisyphus.getclr.reset)
+        sys.exit()
+    else:
+        if gfx_ui:
+            sisyphus.solverevdeps.start.__wrapped__(pkgname)
+        else:
+            sisyphus.solverevdeps.start(pkgname)
+
+        is_needed = pickle.load(
+            open(os.path.join(sisyphus.getfs.p_mtd_dir, "sisyphus_pkgrevdeps.pickle"), "rb"))
+
+    if is_needed != 0:
+        if gfx_ui:
+            p_exe = subprocess.Popen(['emerge'] + args + ['--pretend', '--verbose'] + list(
+                pkgname), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            # kill portage if the program dies or it's terminated by the user
+            atexit.register(sisyphus.killemerge.start, p_exe)
+
+            for p_out in io.TextIOWrapper(p_exe.stdout, encoding="utf-8"):
+                print(p_out.rstrip())
+
+            p_exe.wait()
+            print("\nWon't uninstall! Other packages depend on " + str(pkgname))
+        else:
+            p_exe = subprocess.Popen(
+                ['emerge'] + args + ['--pretend', '--verbose'] + list(pkgname))
+            try:
+                p_exe.wait()
+            except KeyboardInterrupt:
+                p_exe.terminate()
+                try:
+                    p_exe.wait(1)
+                except subprocess.TimeoutExpired:
+                    p_exe.kill()
+                sys.exit()
+            print(sisyphus.getclr.bright_red +
+                  "\nWon't uninstall! Other packages depend on " + sisyphus.getclr.reset + str(pkgname))
+            print(sisyphus.getclr.bright_red + "Use the " + sisyphus.getclr.reset + sisyphus.getclr.green + "'--force'" +
+                  sisyphus.getclr.reset + sisyphus.getclr.bright_red + " option to override at your own risk!\n" + sisyphus.getclr.reset)
+    else:
+        if unmerge:
+            p_exe = subprocess.Popen(
+                ['emerge', '--quiet', '--unmerge', '--ask'] + list(pkgname))
+            try:
+                p_exe.wait()
+                sisyphus.syncdb.lcl_tbl()
+            except KeyboardInterrupt:
+                p_exe.terminate()
+                try:
+                    p_exe.wait(1)
+                except subprocess.TimeoutExpired:
+                    p_exe.kill()
+                sys.exit()
+        elif depclean:
+            if gfx_ui:
+                p_exe = subprocess.Popen(
+                    ['emerge'] + args + pkgname, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+                # kill portage if the program dies or it's terminated by the user
+                atexit.register(sisyphus.killemerge.start, p_exe)
+
+                for p_out in io.TextIOWrapper(p_exe.stdout, encoding="utf-8"):
+                    print(p_out.rstrip())
+
+                p_exe.wait()
+                sisyphus.syncdb.lcl_tbl()
+            else:
+                p_exe = subprocess.Popen(
+                    ['emerge'] + args + ['--ask'] + list(pkgname))
+                try:
+                    p_exe.wait()
+                    sisyphus.syncdb.lcl_tbl()
+                except KeyboardInterrupt:
+                    p_exe.terminate()
+                    try:
+                        p_exe.wait(1)
+                    except subprocess.TimeoutExpired:
+                        p_exe.kill()
+                    sys.exit()