import configparser
import bugzilla
import pathlib
import sys
import logging
import subprocess
from click import progressbar
assert sys.version_info[0] > 2, 'Needs Python 3'
# XXX Update this after branching!
RAWHIDE = '32'
LOGGER = logging.getLogger()
LOGGER.setLevel(logging.DEBUG)
fh = logging.FileHandler('bugzilla.log')
fh.setLevel(logging.DEBUG)
LOGGER.addHandler(fh)
# Get credentials from config
config_path = pathlib.Path('./close_retired_bugzillas.cfg')
if not config_path.exists():
    config_path = pathlib.Path('/etc/close_retired_bugzillas.cfg')
if not config_path.exists():
    raise RuntimeError('Create a config file as ./close_retired_bugzillas.cfg '
                       'or /etc/close_retired_bugzillas.cfg')
config = configparser.ConfigParser()
config.read(config_path)
URL = config['bugzilla'].get('url', 'https://bugzilla.redhat.com')
USERNAME = config['bugzilla']['username']
PASSWORD = config['bugzilla'].get('password', '')
FEDORA = (config['bugzilla'].get('fedora', '').upper().replace('F', '')
          or RAWHIDE)
TEMPLATE = f"""Automation has figured out the package is retired in Fedora {FEDORA}.
If you like it to be unretired, please open a ticket at https://pagure.io/releng/new_issue?template=package_unretirement
"""  # noqa
if USERNAME == '-interactive-':
    bzapi = bugzilla.Bugzilla(URL)
    if not bzapi.logged_in:
        print(f'Log in to Bugzilla ta {URL}')
        bzapi.interactive_login()
else:
    bzapi = bugzilla.Bugzilla(URL, user=USERNAME, password=PASSWORD)
def open_bugz(version=FEDORA):
    if version == RAWHIDE:
        version = 'rawhide'
    query = bzapi.build_query(product='Fedora', version=version)
    query['status'] = [
        'NEW',
        'ASSIGNED',
        'POST',
        'MODIFIED',
        'ON_QA',
        'VERIFIED',
        'RELEASE_PENDING',
    ]
    return bzapi.query(query)
def is_retired(component, *, version=FEDORA):
    if component == 'distribution' or ' ' in component:
        return False
    if version == 'rawhide':
        version = RAWHIDE
    tag = f'f{version}'
    cmd = ('koji', 'list-pkgs', '--show-blocked',
           '--tag', tag, '--package', component)
    p = subprocess.run(cmd, stdout=subprocess.PIPE, text=True)
    return '[BLOCKED]' in p.stdout
def close_bugzillas(bug_ids, comment=TEMPLATE):
    update = bzapi.build_update(comment=comment,
                                status='CLOSED', resolution='WONTFIX')
    try:
        bzapi.update_bugs(bug_ids, update)
    except Exception:
        LOGGER.exception()
print('Gathering bugz, this can take several minutes...')
bugz = open_bugz()
components = set(bug.component for bug in bugz)
print(f'There are {len(bugz)} unclosed bugz for {len(components)} components')
print('Checking which components are retired...')
retired = set()
with progressbar(components, item_show_func=lambda c: c or 'Finished!') as cb:
    for component in cb:
        if is_retired(component):
            retired.add(component)
bug_ids = [b.id for b in bugz if b.component in retired]
print(f'There are {len(retired)} retired components, '
      f'closing their {len(bug_ids)} bugzillas.')
close_bugzillas(bug_ids)