#!/usr/bin/env python2
from pisi.db.filesdb import FilesDB
import pisi.api
import magic
import os
import re
import subprocess

valid_dyn = ""

v_dyn = re.compile(r"ELF (64|32)\-bit LSB shared object,")
v_bin = re.compile(r"ELF (64|32)\-bit LSB executable,")
shared_lib = re.compile(r".*Shared library: \[(.*)\].*")
r_path = re.compile(r".*Library rpath: \[(.*)\].*")
r_soname = re.compile(r".*Library soname: \[(.*)\].*")

def get_soname(path):
    output = subprocess.check_output("/usr/bin/readelf -d {}".format(path), shell=True)

    for line in output.split("\n"):
        line = line.strip()
        g = r_soname.match(line)
        if g:
            return g.group(1)
    return None

def accumulate_dependencies(path, provided, emul32=False):
    output = subprocess.check_output("/usr/bin/readelf -d {}".format(path), shell=True)

    check_deps = set()
    r_paths = set()

    valid_libs = set()

    if emul32:
        valid_libs.update(["/usr/lib32", "/lib32"])
    else:
        # Currently on Solus this is the same thing as /usr/lib.
        valid_libs.update(["/usr/lib64", "/lib64"])

    for line in output.split("\n"):
        line = line.strip()
        g = shared_lib.match(line)
        if g:
            lib = g.group(1)
            if lib in provided:
                print("\nSkipping internally provided so: {}\n".format(lib))
                continue
            check_deps.add(lib)
            continue
        r = r_path.match(line)
        if r:
            r_paths.add(r.group(1))

    print("Deps: {}".format(", ".join(check_deps)))

    dirname = os.path.dirname(path)

    filter_deps = [x for x in check_deps for y in r_paths if os.path.exists(os.path.join(y, x)) or os.path.exists(os.path.join(dirname, x))]
    print("Filtered by rpath: {}".format(", ".join(filter_deps)))

    print("Got %d of rpath:" % len(filter_deps))

    ret_deps = filter(lambda s: s not in filter_deps, check_deps)
    print("Remaining deps: {}".format(", ".join(ret_deps)))

    full_paths = [os.path.join(y,x) for x in ret_deps for y in valid_libs if os.path.exists(os.path.join(y,x))]
    print("Full paths is now: {}".format(", ".join(full_paths)))
    return full_paths

def is_dynamic_binary(path):
    if not os.path.exists(path) or not os.path.isfile(path):
        return False
    try:
        mg = magic.from_file(path)
    except Exception, e:
        return False
    if v_bin.match(mg):
        return True
    if v_dyn.match(mg):
        return True
    return False

def clean_path(p):
    if not p[0] == '/':
        return "/%s" % p

def main():
    le_files = set()

    packages = ["firefox"]

    provided = set()
    want_depends = set()

    deps = set()

    for pkg in packages:
        (stuff,files,repo) = pisi.api.info_name(pkg, True)

        for f in files.list:
            f = clean_path(f.path)

            if not is_dynamic_binary(f):
                continue
            soname = get_soname(f)
            if soname is not None:
                print("\n\nAdding %s to provided\n\n" % soname)
                provided.add(soname)
            want_depends.add(f)

    for want in want_depends:
        mg = magic.from_file(want)
        emul32 = mg.startswith("ELF 32")
        deps.update(accumulate_dependencies(want, provided, emul32))

    print("\n\n")
    print("Dependencies")
    for i in deps:
        print("  -> %s" % i)

if __name__ == "__main__":
    main()