#!/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"))