#!/usr/bin/python3

import animation
import os
import pickle
import subprocess
import sisyphus.getfs


@animation.wait('resolving dependencies')
def pkg(pkgname):
    areBinaries = []
    areSources = []
    needsConfig = int()
    portageExec = subprocess.Popen(['emerge', '--quiet', '--pretend', '--getbinpkg', '--rebuilt-binaries', '--with-bdeps=y', '--misspell-suggestion=n', '--fuzzy-search=n'] + list(pkgname), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout, stderr = portageExec.communicate()

    for portageOutput in stderr.decode('utf-8').splitlines():
        if "The following keyword changes are necessary to proceed:" in portageOutput:
            needsConfig = int(1)

        if "The following mask changes are necessary to proceed:" in portageOutput:
            needsConfig = int(1)

        if "The following USE changes are necessary to proceed:" in portageOutput:
            needsConfig = int(1)

        if "The following REQUIRED_USE flag constraints are unsatisfied:" in portageOutput:
            needsConfig = int(1)

    for portageOutput in stdout.decode('utf-8').splitlines():
        if "[binary" in portageOutput:
            isBinary = portageOutput.split("]")[1].split("[")[0].strip(" ")
            areBinaries.append(isBinary)

        if "[ebuild" in portageOutput:
            isSource = portageOutput.split("]")[1].split("[")[0].strip(" ")
            areSources.append(isSource)

    pickle.dump([areBinaries,areSources,needsConfig], open(os.path.join(sisyphus.getfs.portageMetadataDir, "sisyphus_pkgdeps.pickle"), "wb"))


@animation.wait('resolving dependencies')
def world():
    areBinaries = []
    areSources = []
    needsConfig = int()
    portageExec = subprocess.Popen(['emerge', '--quiet', '--update', '--deep', '--newuse', '--pretend', '--getbinpkg', '--rebuilt-binaries', '--backtrack=100', '--with-bdeps=y', '--misspell-suggestion=n', '--fuzzy-search=n', '@world'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout, stderr = portageExec.communicate()

    for portageOutput in stderr.decode('utf-8').splitlines():
        if "The following keyword changes are necessary to proceed:" in portageOutput:
            needsConfig = int(1)

        if "The following mask changes are necessary to proceed:" in portageOutput:
            needsConfig = int(1)

        if "The following USE changes are necessary to proceed:" in portageOutput:
            needsConfig = int(1)

        if "The following REQUIRED_USE flag constraints are unsatisfied:" in portageOutput:
            needsConfig = int(1)

    for portageOutput in stdout.decode('utf-8').splitlines():
        if "[binary" in portageOutput:
            isBinary = portageOutput.split("]")[1].split("[")[0].strip(" ")
            areBinaries.append(isBinary)

        if "[ebuild" in portageOutput:
            isSource = portageOutput.split("]")[1].split("[")[0].strip(" ")
            areSources.append(isSource)

    pickle.dump([areBinaries,areSources,needsConfig], open(os.path.join(sisyphus.getfs.portageMetadataDir, "sisyphus_worlddeps.pickle"), "wb"))