1 files changed, 73 insertions, 0 deletions
diff --git a/src/backend/depsolve.py b/src/backend/depsolve.py
new file mode 100644
index 0000000..34d70b8
--- /dev/null
+++ b/src/backend/depsolve.py
@@ -0,0 +1,73 @@
+#!/usr/bin/python3
+
+import animation
+import os
+import pickle
+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 dependencies')
+def start(pkgname=None, nodeps=False):
+    bin_list = []
+    src_list = []
+    is_vague = int()
+    need_cfg = int()
+
+    if pkgname:
+        args = ['--quiet', '--pretend', '--getbinpkg', '--rebuilt-binaries', '--misspell-suggestion=n',
+                '--fuzzy-search=n'] + (['--nodeps'] if nodeps else ['--with-bdeps=y']) + list(pkgname)
+    else:
+        args = ['--quiet', '--update', '--deep', '--newuse', '--pretend', '--getbinpkg', '--rebuilt-binaries',
+                '--backtrack=100', '--with-bdeps=y', '--misspell-suggestion=n', '--fuzzy-search=n', '@world']
+
+    p_exe = subprocess.Popen(
+        ['emerge'] + args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    try:
+        stdout, stderr = p_exe.communicate()
+
+        for p_out in stderr.decode('utf-8').splitlines():
+            if pkgname:
+                if any(key in p_out for key in ["short ebuild name",
+                                                "is ambiguous",
+                                                "there are no ebuilds to satisfy"]):  # likely very fragile
+                    is_vague = int(1)
+
+            if any(key in p_out for key in ["The following keyword changes are necessary to proceed:",
+                                            "The following mask changes are necessary to proceed:",
+                                            "The following USE changes are necessary to proceed:",
+                                            "The following REQUIRED_USE flag constraints are unsatisfied:",
+                                            "One of the following masked packages is required to complete your request:"]):  # likely very fragile
+                need_cfg = int(1)
+
+        for p_out in stdout.decode('utf-8').splitlines():
+            if "[binary" in p_out:
+                is_bin = p_out.split("]")[1].split("[")[0].strip(" ")
+                bin_list.append(is_bin)
+
+            if "[ebuild" in p_out:
+                is_src = p_out.split("]")[1].split("[")[0].strip(" ")
+                src_list.append(is_src)
+
+        if pkgname:
+            pickle.dump([bin_list, src_list, is_vague, need_cfg], open(
+                os.path.join(sisyphus.getfs.p_mtd_dir, "sisyphus_pkgdeps.pickle"), "wb"))
+        else:
+            pickle.dump([bin_list, src_list, is_vague, need_cfg], open(
+                os.path.join(sisyphus.getfs.p_mtd_dir, "sisyphus_worlddeps.pickle"), "wb"))
+    except KeyboardInterrupt:
+        p_exe.terminate()
+        try:
+            p_exe.wait(1)
+        except subprocess.TimeoutExpired:
+            p_exe.kill()
+        sys.exit()