#!/usr/bin/env python3

# The script is used for finding PRs that have a SVN revision that
# mentiones the PR and are not closed.  It's done by iterating all
# comments of a PR and finding SVN commit entries.

"""
Sample output of the script:
Bugzilla URL page size: 50
HINT: bugs with following comment are ignored: Can the bug be marked as resolved?

Bug URL                                              SVN commits                   known-to-fail                           known-to-work                           Bug summary                                                 
https://gcc.gnu.org/bugzilla/show_bug.cgi?id=88129   trunk                                                                                                         Two blockage insns are emited in the function epilogue      
https://gcc.gnu.org/bugzilla/show_bug.cgi?id=88122   trunk                                                                                                         [9 Regression] g++ ICE: internal compiler error: Segmentation fault
https://gcc.gnu.org/bugzilla/show_bug.cgi?id=88084   trunk                                                                                                         basic_string_view::copy doesn't use Traits::copy            
https://gcc.gnu.org/bugzilla/show_bug.cgi?id=88083   trunk                                                                                                         ICE in find_constant_pool_ref_1, at config/s390/s390.c:8231 
...
Bugzilla lists:
https://gcc.gnu.org/bugzilla/buglist.cgi?bug_id=88129,88122,88084,88083,88074,88073,88071,88070,88051,88018,87985,87955,87926,87917,87913,87898,87895,87874,87871,87855,87853,87826,87824,87819,87818,87799,87793,87789,87788,87787,87754,87725,87674,87665,87649,87647,87645,87625,87611,87610,87598,87593,87582,87566,87556,87547,87544,87541,87537,87528
https://gcc.gnu.org/bugzilla/buglist.cgi?bug_id=87486
"""

import argparse
import json

import requests

base_url = 'https://gcc.gnu.org/bugzilla/rest.cgi/'
statuses = ['UNCONFIRMED', 'ASSIGNED', 'SUSPENDED', 'NEW', 'WAITING', 'REOPENED']
regex = '(.*\[)([0-9\./]*)( [rR]egression])(.*)'
closure_question = 'Can the bug be marked as resolved?'
start_page = 20
url_page_size = 50

def get_branches_by_comments(comments):
    versions = set()
    for c in comments:
        text = c['text']
        if 'URL: https://gcc.gnu.org/viewcvs' in text:
            version = 'trunk'
            for l in text.split('\n'):
                if 'branches/gcc-' in l:
                    parts = l.strip().split('/')
                    parts = parts[1].split('-')
                    assert len(parts) == 3
                    versions.add(parts[1])
            versions.add(version)
    return versions

def get_bugs(api_key, query):
    u = base_url + 'bug'
    r = requests.get(u, params = query)
    return r.json()['bugs']

def search(api_key):
    chunk = 1000
    ids = []
    print('%-53s%-30s%-40s%-40s%-60s' % ('Bug URL', 'SVN commits', 'known-to-fail', 'known-to-work', 'Bug summary'))
    for i in range(start_page, 0, -1):
        # print('offset: %d' % (i * chunk), flush = True)
        bugs = get_bugs(api_key, {'api_key': api_key, 'bug_status': statuses, 'limit': chunk, 'offset': i * chunk})
        for b in sorted(bugs, key = lambda x: x['id'], reverse = True):
            id = b['id']

            fail = b['cf_known_to_fail']
            work = b['cf_known_to_work']

            u = base_url + 'bug/' + str(id) + '/comment'
            r = requests.get(u, params = {'api_key': api_key} ).json()
            keys = list(r['bugs'].keys())
            assert len(keys) == 1
            comments = r['bugs'][keys[0]]['comments']
            for c in comments:
                if closure_question in c['text']:
                    continue

            branches = get_branches_by_comments(comments)
            if len(branches):
                branches_str = ','.join(sorted(list(branches)))
                print('%-53s%-30s%-40s%-40s%-60s' % ('https://gcc.gnu.org/bugzilla/show_bug.cgi?id=%d' % id, branches_str, fail, work, b['summary']))
                ids.append(id)

    # print all URL lists
    print('\nBugzilla lists:')
    while len(ids) > 0:
        print('https://gcc.gnu.org/bugzilla/buglist.cgi?bug_id=%s' % ','.join([str(x) for x in ids[:url_page_size]]))
        ids = ids[url_page_size:]

print('Bugzilla URL page size: %d' % url_page_size)
print('HINT: bugs with following comment are ignored: %s\n' % closure_question)

parser = argparse.ArgumentParser(description='')
parser.add_argument('api_key', help = 'API key')

args = parser.parse_args()
search(args.api_key)