#!/usr/bin/env python
#
# Copyright (C) 2011 - Soren Hansen
# Copyright (C) 2013 - Red Hat, Inc.

# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.



import calendar
import datetime
import glob
import json
import optparse
import os
import os.path
from pprint import pprint
import prettytable
import sys

import utils


optparser = optparse.OptionParser()
optparser.add_option('-p', '--project', default='projects/nova.json',
        help='JSON file describing the project to generate stats for')
optparser.add_option('-a', '--all', action='store_true',
        help='Generate stats across all known projects (*.json)')
optparser.add_option('-d', '--days', type='int', default=14,
        help='Number of days to consider')
optparser.add_option('-u', '--user', default='russellb', help='gerrit user')
optparser.add_option('-k', '--key', default=None, help='ssh key for gerrit')

options, args = optparser.parse_args()


projects = utils.get_projects_info(options.project, options.all)

if not projects:
    print "Please specify a project."
    sys.exit(1)

all_changes = utils.get_changes(projects, options.user, options.key)

reviews = []

for change in all_changes:
#    print json.dumps(change, sort_keys=True, indent=4)
    for patchset in change.get('patchSets', []):
        for review in patchset.get('approvals', []):
            reviews += [review]

cut_off = datetime.datetime.now() - datetime.timedelta(days=options.days)
ts = calendar.timegm(cut_off.timetuple())
reviews = filter(lambda x:x['grantedOn'] > ts, reviews)

def round_to_day(ts):
    SECONDS_PER_DAY = 60*60*24
    return (ts / (SECONDS_PER_DAY)) * SECONDS_PER_DAY

reviewers = {}
for review in reviews:
    if review['type'] != 'CRVW':
        # Only count code reviews.  Don't add another for Approved, which is
        # type 'APRV'
        continue

    reviewer = review['by'].get('username', 'unknown')
    reviewers.setdefault(reviewer,
            {'votes': {'-2': 0, '-1': 0, '1': 0, '2': 0}})
    reviewers[reviewer]['total'] = reviewers[reviewer].get('total', 0) + 1
    cur = reviewers[reviewer]['votes'][review['value']]
    reviewers[reviewer]['votes'][review['value']] = cur + 1

#print json.dumps(reviewers, sort_keys=True, indent=4)

reviewers = [(v, k) for k, v in reviewers.iteritems()
             if k.lower() not in ('jenkins', 'smokestack')]
reviewers.sort(reverse=True)

if options.all:
    print 'Reviews for the last %d days in projects: %s' % (options.days,
            [project['name'] for project in projects])
else:
    print 'Reviews for the last %d days in %s' % (options.days, projects[0]['name'])
if options.all:
    print '** -- Member of at least one core reviewer team'
else:
    print '** -- %s-core team member' % projects[0]['name']
table = prettytable.PrettyTable(('Reviewer', 'Reviews (-2|-1|+1|+2) (+/- ratio)'))
total = 0
for k, v in reviewers:
    in_core_team = False
    for project in projects:
        if v in project['core-team']:
            in_core_team = True
            break
    name = '%s%s' % (v, ' **' if in_core_team else '')
    plus = float(k['votes']['2'] + k['votes']['1'])
    minus = float(k['votes']['-2'] + k['votes']['-1'])
    ratio = (plus / (plus + minus)) * 100
    r = '%d (%d|%d|%d|%d) (%.1f%%)' % (k['total'],
            k['votes']['-2'], k['votes']['-1'],
            k['votes']['1'], k['votes']['2'], ratio)
    table.add_row((name, r))
    total += k['total']
print table
print '\nTotal reviews: %d' % total
print 'Total reviewers: %d' % len(reviewers)