#!/usr/bin/python

# Copyright (c) 2010-2014, Benjamin Drung <bdrung@debian.org>
#
# Permission to use, copy, modify, and/or distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

import argparse
import os
import subprocess
import sys

# error codes
COMMAND_LINE_SYNTAX_ERROR = 1


def remove_recursive(path):
    """equivalent to rm -rf path"""
    if os.path.exists(path):
        for i in os.listdir(path):
            full_path = os.path.join(path, i)
            if os.path.isdir(full_path):
                remove_recursive(full_path)
            else:
                os.remove(full_path)
        os.rmdir(path)


def repack_xpi(package, upstream_version, xpi_file, verbose):
    # extract xpi file
    tmp_dir = "/tmp"
    extract_dir = package + "-" + upstream_version
    full_extract_dir = os.path.join(tmp_dir, extract_dir)
    remove_recursive(full_extract_dir)
    subprocess.check_call(["xpi-unpack", xpi_file, full_extract_dir])

    # check, if source 3.0 (quilt) format is used
    extension = ".gz"
    if os.path.isfile("debian/source/format"):
        source_format = open("debian/source/format").readline().strip()
        if source_format == "3.0 (quilt)":
            extension = ".bz2"

    # pack source
    tar_file = package + "_" + upstream_version + ".orig.tar"
    full_tar_file = os.path.realpath(os.path.join(os.path.dirname(xpi_file),
                                     tar_file))
    cmd = ["tar", "-ca", "-C", tmp_dir, "-f", full_tar_file + extension,
           extract_dir]
    if verbose:
        print " ".join(cmd)
    subprocess.check_call(cmd)

    # remove temporary directory
    remove_recursive(full_extract_dir)


def get_source_package_name(script_name):
    if not os.path.isfile("debian/control"):
        sys.stderr.write(script_name + ": Error: debian/control file is "
                         "missing. Please execute the script in a Debian "
                         "source package or provide a source package name.\n")
        sys.exit(1)
    lines = open("debian/control").readlines()
    package_lines = filter(lambda x: x.find("Source:") >= 0, lines)
    packages = map(lambda x: x[x.find(":")+1:].strip(), package_lines)
    return packages[0]


def usage(output):
    print >> output, """Usage: %s [options] <xpi-file>

Options:
  -p, --package=<value>              specify source package name
  -u, --upstream-version=<version>   specify the upstream version

General options:
  -h, --help                   display this help and exit
  -v, --verbose                print more information

See %s(1) for more info.""" % (os.path.basename(sys.argv[0]),
                               os.path.basename(sys.argv[0]))


def main():
    script_name = os.path.basename(sys.argv[0])
    epilog = "See {prog}(1) for more info.".format(prog=script_name)
    parser = argparse.ArgumentParser(epilog=epilog)
    parser.add_argument("xpi_file", metavar="<xpi-file>",
                        help=".xpi file that should be repacked")
    parser.add_argument("-p", "--package", help="specify source package name")
    parser.add_argument("-u", "--upstream-version", dest="version",
                        help="specify the upstream version")
    parser.add_argument("-v", "--verbose", action="store_true",
                        help="print more information")

    args = parser.parse_args()

    if not args.package:
        args.package = get_source_package_name(script_name)

    repack_xpi(args.package, args.version, args.xpi_file, args.verbose)

if __name__ == "__main__":
    main()