1 files changed, 108 insertions, 0 deletions
diff --git a/ebuildgen/linkdeps.py b/ebuildgen/linkdeps.py
new file mode 100644
index 0000000..655af1c
--- /dev/null
+++ b/ebuildgen/linkdeps.py
@@ -0,0 +1,108 @@
+import os
+from subprocess import getstatusoutput
+from urllib.request import urlopen
+import gentoopm
+
+def deptopackage(dep,addpaths):
+    #return pfltopackage(dep,addpaths)
+    return qfiletopackage(dep,addpaths)
+
+def qfiletopackage(dep,addpaths):
+    """Converts supplied deps with additional include paths to portage packages
+
+    This uses qfile to quess which package certain files belongs to.
+    """
+
+    print(dep)
+    (statuscode,outstr) = getstatusoutput('echo "" | `gcc -print-prog-name=cc1` -v -q')
+    #"`gcc -print-prog-name=cc1plus` -v" for cpp
+    outlst = outstr.split("\n")
+    incpaths = []
+    for item in outlst:
+        if item[:2] == " /":
+            incpaths += [item[1:]]
+    incpaths += addpaths
+    depname = os.path.split(dep)[1]
+
+    (statuscode,packagestr) = getstatusoutput("qfile -C " + depname)
+    if not statuscode == 0:
+        package = pfltopackage(dep,incpaths)
+
+    else:
+        packagelst = packagestr.split()
+        package = []
+        n = 0
+        for depfile in packagelst[1::2]:
+            for incpath in incpaths:
+                if depfile.strip("()") == (incpath + "/" + dep):
+                    package.append(packagelst[n])
+            n += 2
+
+        if len(package) > 1:
+            print("more than one matching package were found!")
+
+        if not package:
+            package = pfltopackage(dep,incpaths)
+
+    print(package)
+    #check if package exists
+    pm=gentoopm.get_package_manager()
+    if package:
+        #does the package exist in this computers package manager?
+        if pm.stack.filter(package[0]):
+            return package[0]
+        else:
+            print("No package named: " + package[0] + " found localy, ignoring")
+            return []
+    else:
+        return package
+
+def pfltopackage(dep,addpaths):
+    """This uses the online ply database to guess packages
+
+    """
+
+    print(dep)
+    incpaths = addpaths
+
+    url_lines = []
+    depname = os.path.split(dep)[1]
+    matching_packages = set()
+    all_packages = set()
+
+    url = urlopen("http://www.portagefilelist.de/index.php/Special:PFLQuery2?file="
+            + depname + "&searchfile=lookup&lookup=file&txt")
+
+    for line in url:
+        url_lines += [line.decode("utf-8").split()]
+
+    #First line does not contain any useful information, skip it
+    url_lines = url_lines[1:]
+    #structure of lines: [portage_category, package, path, file, misc, version]
+
+    for line in url_lines:
+        all_packages.add(line[0] + "/" + line[1])
+        #check if path is correct
+        for path in incpaths:
+            if line[2] + "/" + line[3] == path + "/" + dep:
+                matching_packages.add(line[0] + "/" + line[1])
+
+    if len(matching_packages) > 1:
+        print("More than one matching package for dep found!\nPicking the last one...")
+
+    if not matching_packages:
+        print("no matching package found within the include paths!")
+        if len(all_packages) == 1:
+            print("but only one package matches the headerfile, picking that one")
+            matching_packages = all_packages
+        elif all_packages:
+            print("file not found was: " + dep)
+            print("a dummy dep will be placed in the ebuild, fix it!")
+            matching_packages = ["dummy_for_" + dep]
+        else:
+            print("No package supplies the headerfile, ignoring...")
+            return []
+
+    return [matching_packages.pop()]
+
+#qfiletopackage("jack/ringbuffer.h",[])