#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# publish.py - part of the FDroid server tools
# Copyright (C) 2010-12, Ciaran Gultnieks, ciaran@ciarang.com
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import sys
import os
import shutil
import subprocess
import re
import zipfile
import tarfile
import md5
import glob
from optparse import OptionParser

import common
from common import BuildException

#Read configuration...
execfile('config.py')

# Parse command line...
parser = OptionParser()
parser.add_option("-v", "--verbose", action="store_true", default=False,
                  help="Spew out even more information than normal")
parser.add_option("-p", "--package", default=None,
                  help="Publish only the specified package")
(options, args) = parser.parse_args()

log_dir = 'logs'
if not os.path.isdir(log_dir):
    print "Creating log directory"
    os.makedirs(log_dir)

tmp_dir = 'tmp'
if not os.path.isdir(tmp_dir):
    print "Creating temporary directory"
    os.makedirs(tmp_dir)

output_dir = 'repo'
if not os.path.isdir(output_dir):
    print "Creating output directory"
    os.makedirs(output_dir)

unsigned_dir = 'unsigned'
if not os.path.isdir(unsigned_dir):
    print "No unsigned directory - nothing to do"
    sys.exit(0)

for apkfile in sorted(glob.glob(os.path.join(unsigned_dir, '*.apk'))):

    apkfilename = os.path.basename(apkfile)
    i = apkfilename.rfind('_')
    if i == -1:
        raise BuildException("Invalid apk name")
    appid = apkfilename[:i]
    print "Processing " + appid

    if not options.package or options.package == appid:

        # Figure out the key alias name we'll use. Only the first 8
        # characters are significant, so we'll use the first 8 from
        # the MD5 of the app's ID and hope there are no collisions.
        # If a collision does occur later, we're going to have to
        # come up with a new alogrithm, AND rename all existing keys
        # in the keystore!
        if keyaliases.has_key(appid):
            # For this particular app, the key alias is overridden...
            keyalias = keyaliases[appid]
        else:
            m = md5.new()
            m.update(appid)
            keyalias = m.hexdigest()[:8]
        print "Key alias: " + keyalias

        # See if we already have a key for this application, and
        # if not generate one...
        p = subprocess.Popen(['keytool', '-list',
            '-alias', keyalias, '-keystore', keystore,
            '-storepass', keystorepass], stdout=subprocess.PIPE)
        output = p.communicate()[0]
        if p.returncode !=0:
            print "Key does not exist - generating..."
            p = subprocess.Popen(['keytool', '-genkey',
                '-keystore', keystore, '-alias', keyalias,
                '-keyalg', 'RSA', '-keysize', '2048',
                '-validity', '10000',
                '-storepass', keystorepass, '-keypass', keypass,
                '-dname', keydname], stdout=subprocess.PIPE)
            output = p.communicate()[0]
            print output
            if p.returncode != 0:
                raise BuildException("Failed to generate key")

        # Sign the application...
        p = subprocess.Popen(['jarsigner', '-keystore', keystore,
            '-storepass', keystorepass, '-keypass', keypass,
                apkfile, keyalias], stdout=subprocess.PIPE)
        output = p.communicate()[0]
        print output
        if p.returncode != 0:
            raise BuildException("Failed to sign application")

        # Zipalign it...
        p = subprocess.Popen([os.path.join(sdk_path,'tools','zipalign'),
                            '-v', '4', apkfile,
                            os.path.join(output_dir, apkfilename)],
                            stdout=subprocess.PIPE)
        output = p.communicate()[0]
        print output
        if p.returncode != 0:
            raise BuildException("Failed to align application")
        os.remove(apkfile)

        # Move the source tarball into the output directory...
        tarfilename = apkfilename[:-4] + '_src.tar.gz'
        shutil.move(os.path.join(unsigned_dir, tarfilename),
                os.path.join(output_dir, tarfilename))

        print 'Published ' + apkfilename