#!/usr/bin/python

import sys
import os
import koji
import logging
import urlgrabber.grabber as grabber
import urlgrabber.progress as progress
import urllib2
import time
import random
import string
import rpm 
import shutil

LOCALKOJIHUB = 'https://s390.koji.fedoraproject.org/kojihub'
REMOTEKOJIHUB = 'https://koji.fedoraproject.org/kojihub'
PACKAGEURL = 'http://kojipkgs.fedoraproject.org/'

# Should probably set these from a koji config file
SERVERCA = os.path.expanduser('~/.fedora-server-ca.cert')
CLIENTCA = os.path.expanduser('~/.fedora-upload-ca.cert')
CLIENTCERT = os.path.expanduser('~/.fedora.cert')

workpath = '/tmp/koji-import'

loglevel = logging.DEBUG
logging.basicConfig(format='%(levelname)s: %(message)s',
                    level=loglevel)

pg = progress.TextMeter()

def _unique_path(prefix):
    """Create a unique path fragment by appending a path component
    to prefix.  The path component will consist of a string of letter and numbers
    that is unlikely to be a duplicate, but is not guaranteed to be unique."""
    # Use time() in the dirname to provide a little more information when
    # browsing the filesystem.
    # For some reason repr(time.time()) includes 4 or 5
    # more digits of precision than str(time.time())
    return '%s/%r.%s' % (prefix, time.time(),
                      ''.join([random.choice(string.ascii_letters) for i in range(8)]))

def isNoarch(rpms):
    if not rpms:
        return False
    noarch = False
    for rpminfo in rpms:
        if rpminfo['arch'] == 'noarch':
            #note that we've seen a noarch rpm
            noarch = True
        elif rpminfo['arch'] != 'src':
            return False
    return noarch

def tagSuccessful(nvr, tag):
    """tag completed builds into final tags"""
    localkojisession.tagBuildBypass(tag, nvr)
    print "tagged %s to %s" % (nvr, tag)

def _downloadURL(url, destf):
    """Download a url and save it to a file"""
    file = grabber.urlopen(url, progress_obj = pg, text = "%s" % (destf))

    out = os.open(destf, os.O_WRONLY|os.O_CREAT|os.O_TRUNC, 0666)
    try:
        while 1:
            buf = file.read(4096)
            if not buf:
                break
            os.write(out, buf)
    finally:
        os.close(out)
        file.close()

def _importURL(url, fn):
    """Import an rpm directly from a url"""
    serverdir = _unique_path('build-recent')
    #TODO - would be possible, using uploadFile directly, to upload without writing locally.
    #for now, though, just use uploadWrapper
    koji.ensuredir(workpath)
    dst = "%s/%s" % (workpath, fn)
    print "Downloading %s to %s..." % (url, dst)
    _downloadURL(url, dst)
    #fsrc = urllib2.urlopen(url)
    #fdst = file(dst, 'w')
    #shutil.copyfileobj(fsrc, fdst)
    #fsrc.close()
    #fdst.close()
    print "Uploading %s..." % dst
    localkojisession.uploadWrapper(dst, serverdir, blocksize=65536)
    localkojisession.importRPM(serverdir, fn)

def importBuild(rpms, buildinfo, tag=None):
    '''import a build from remote hub'''
    for rpminfo in rpms:
        if rpminfo['arch'] == 'src':
            srpm = rpminfo
    pathinfo = koji.PathInfo(PACKAGEURL)

    build_url = pathinfo.build(buildinfo)
    url = "%s/%s" % (pathinfo.build(buildinfo), pathinfo.rpm(srpm))
    fname = "%s.src.rpm" % buildinfo['nvr']
    try:
        _importURL(url, fname)
    except:
	logging.info("Importing %s failed" % fname)
	return False
    else:
        for rpminfo in rpms:
	    if rpminfo['arch'] == 'src':
    		#already imported above
        	continue
    	    relpath = pathinfo.rpm(rpminfo)
    	    url = "%s/%s" % (build_url, relpath)
    	    logging.debug("url: %s" % url)
    	    fname = os.path.basename(relpath)
    	    logging.debug("fname: %s" % fname)
	    try:
    		_importURL(url, fname)
	    except:
		logging.info("Importing %s failed" % fname)
		return False

	tagSuccessful(buildinfo['nvr'], tag)
	return True

# get packages from command line
if len(sys.argv) > 2:
    tag = sys.argv[1]
    pkgs = sys.argv[2:]
else:
    print("Import a build from primary koji to secondary")
    print("Usage: %s <tag> <build> ..." % sys.argv[0])
    exit(0)

# setup the koji session
logging.info('Setting up koji session')
localkojisession = koji.ClientSession(LOCALKOJIHUB)
remotekojisession = koji.ClientSession(REMOTEKOJIHUB)
localkojisession.ssl_login(CLIENTCERT, CLIENTCA, SERVERCA)
remotekojisession.ssl_login(CLIENTCERT, CLIENTCA, SERVERCA)

for pkg in pkgs:
    buildinfo = remotekojisession.getBuild(pkg)

    logging.info("got build %s" % buildinfo['nvr'])

    rpms = remotekojisession.listRPMs(buildinfo['id'])
    if isNoarch(rpms):
	buildinfo = remotekojisession.getBuild(buildinfo['id'])
	importBuild(rpms, buildinfo, tag=tag)
    else:
	logging.error("not noarch")