#!/usr/bin/env python
#
#  License: GPLv3
#
#  Initial Author: Bonnie King <bonniek@fnal.gov>
#         Updates:
#                  Pat Riehecky <riehecky@fnal.gov>
#
'''Get list of repos from gitblit RPC, to grab CentOS sources'''
 
import optparse
import requests
import simplejson as json
import sys
 
RPCURL = "https://git.centos.org/rpc?req=LIST_REPOSITORIES"
 
def read_args():
    '''
        read in the command line args and set things up
    '''
 
    desc = '''Get list of git repositories from the GitBlit json RPC
           '''
 
    usage = "usage: %prog [options] "
    parser = optparse.OptionParser(usage=usage, description=desc)
    parser.add_option('-p', '--project', metavar="<PROJECTS>",
                      help='''project path (default 'rpms', could be 'all', 'core-sig'...)''',
                      default='rpms')
 
    parser.add_option('-b', '--branch', metavar="<branch name>",
                      help='Only list repos with this branch (default master)',
                      default = 'master')
 
    parser.add_option('-u', '--url', metavar="<URL>",
                      help='URL to check (default %s)' % (RPCURL),
                      default = RPCURL)
 
    (options, args) = parser.parse_args()
    return options
 
def get_repo_list(url, branch, projectpath):
    '''return a list of repo URLs'''
    try:
        req = requests.get(url)
    except requests.exceptions.RequestException as err_msg:
        print err_msg
        sys.exit(1)
 
    payload = req.text
    repos = json.loads(payload)
    branchname = 'refs/heads/' + branch
 
    for repo in repos.keys():
        if projectpath != 'all':
            if repos[repo]['projectPath'] != projectpath:
                del repos[repo]
                continue
        if branchname not in repos[repo]['availableRefs']:
            del repos[repo]
 
    return repos.keys()
 
def main():
    '''Broken out so it can be inherited if someone wants'''
    options = read_args()
    repos = get_repo_list(url=options.url, branch=options.branch, projectpath=options.project)
    if repos:
        print '\n'.join(repos)
    
if __name__ == "__main__":
    main()