1 files changed, 66 insertions, 0 deletions
diff --git a/src/backend/revdepsolve.py b/src/backend/revdepsolve.py
new file mode 100644
index 0000000..bba3932
--- /dev/null
+++ b/src/backend/revdepsolve.py
@@ -0,0 +1,66 @@
+#!/usr/bin/python3
+
+import animation
+import os
+import pickle
+import re
+import signal
+import subprocess
+import sys
+import sisyphus.getfs
+
+
+def sigint_handler(signal, frame):
+    sys.exit(0)
+
+
+signal.signal(signal.SIGINT, sigint_handler)
+
+
+@animation.wait('resolving reverse dependencies')
+def start(pkgname=None, depclean=False, unmerge=False):
+    pattern = r'(\b[a-zA-Z0-9-_]+/[a-zA-Z0-9-_]+):\s+([0-9]+(\.[0-9]+){0,4})'
+    rm_list = []
+
+    args = ['--quiet', '--pretend', '--verbose']
+    args += ['--unmerge'] if unmerge else ['--depclean']
+    if pkgname:
+        args += list(pkgname)
+
+    is_installed = int(1)
+    is_needed = int(0)
+    is_vague = int(0)
+
+    try:
+        p_exe = subprocess.Popen(
+            ['emerge'] + args, stdout=subprocess.PIPE, stderr=subprocess.PIPE
+        )
+
+        stdout, stderr = p_exe.communicate()
+
+        for p_out in stdout.decode('utf-8').splitlines():
+            match = re.search(pattern, p_out)
+            if match:
+                to_remove = f"{match.group(1)}-{match.group(2)}"
+                rm_list.append(to_remove)
+
+            if any(key in p_out for key in ["pulled in by:", "required"]):
+                is_needed = int(1)
+
+        for p_out in stderr.decode('utf-8').splitlines():
+            if any(key in p_out for key in ["Couldn't find", "to depclean."]):
+                is_installed = int(0)
+
+            if any(key in p_out for key in ["short ebuild name", "is ambiguous"]):
+                is_vague = int(1)
+
+    except KeyboardInterrupt:
+        p_exe.terminate()
+        try:
+            p_exe.wait(1)
+        except subprocess.TimeoutExpired:
+            p_exe.kill()
+        sys.exit()
+
+    pickle.dump([is_installed, is_needed, is_vague, rm_list], open(
+        os.path.join(sisyphus.getfs.p_mtd_dir, "sisyphus_pkgrevdeps.pickle"), "wb"))