~jelmer/bzr-builddeb/multiple-upstream-tarballs

#    repack_tarball.py -- Repack files/dirs in to tarballs.

#    Copyright (C) 2007 James Westby <jw+debian@jameswestby.net>

#

#    This file is part of bzr-builddeb.

#

#    bzr-builddeb is free software; you can redistribute it and/or modify

#    it under the terms of the GNU General Public License as published by

#    the Free Software Foundation; either version 2 of the License, or

#    (at your option) any later version.

#

#    bzr-builddeb 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 General Public License for more details.

#

#    You should have received a copy of the GNU General Public License

#    along with bzr-builddeb; if not, write to the Free Software

#    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

#

import gzip

import os

from StringIO import StringIO

import tarfile

import bz2

import sha

import zipfile

from bzrlib.errors import (

                           FileExists,

                           BzrCommandError,

                           NotADirectory,

                           )

from bzrlib.transport import get_transport

from bzrlib import urlutils

def _get_file_from_location(location):

    base_dir, path = urlutils.split(location)

    transport = get_transport(base_dir)

    return transport.get(path)

def repack_tarball(orig_name, new_name, target_dir=None):

  """Repack the file/dir named to a .tar.gz with the chosen name.

  This function takes a named file of either .tar.gz, .tar .tgz .tar.bz2 

  or .zip type, or a directory, and creates the file named in the second

  argument in .tar.gz format.

  If target_dir is specified then that directory will be created if it

  doesn't exist, and the new_name will be interpreted relative to that

  directory.

  The source must exist, and the target cannot exist.

  :param orig_name: the curent name of the file/dir

  :type orig_name: string

  :param new_name: the desired name of the tarball

  :type new_name: string

  :keyword target_dir: the directory to consider new_name relative to, and

                       will be created if non-existant.

  :type target_dir: string

  :return: None

  :throws NoSuchFile: if orig_name doesn't exist.

  :throws NotADirectory: if target_dir exists and is not a directory.

  :throws FileExists: if the target filename (after considering target_dir)

                      exists, and is not identical to the source.

  :throws BzrCommandError: if the source isn't supported for repacking.

  """

  if target_dir is not None:

    if not os.path.exists(target_dir):

      os.mkdir(target_dir)

    else:

      if not os.path.isdir(target_dir):

        raise NotADirectory(target_dir)

    new_name = os.path.join(target_dir, new_name)

  old_contents = None

  if isinstance(orig_name, unicode):

    orig_name = orig_name.encode('utf-8')

  if isinstance(new_name, unicode):

    new_name = new_name.encode('utf-8')

  if os.path.exists(new_name):

    if not orig_name.endswith('.tar.gz'):

      raise FileExists(new_name)

    trans_file = _get_file_from_location(orig_name)

    try:

      old_contents = trans_file.read()

    finally:

      trans_file.close()

    orig_sha = sha.sha(old_contents).hexdigest()

    f = open(new_name)

    try:

      new_sha = sha.sha(f.read()).hexdigest()

    finally:

      f.close()

    if orig_sha != new_sha:

      raise FileExists(new_name)

    return

  if os.path.isdir(orig_name):

    tar = tarfile.open(new_name, 'w:gz')

    try:

      tar.add(orig_name, os.path.basename(orig_name))

    finally:

      tar.close()

  else:

    if old_contents is None:

      trans_file = _get_file_from_location(orig_name)

      try:

        old_contents = trans_file.read()

      finally:

        trans_file.close()

    base_dir, path = urlutils.split(orig_name)

    transport = get_transport(base_dir)

    trans_file = transport.get(path)

    if orig_name.endswith('.tar.gz') or orig_name.endswith('.tgz'):

      dest = open(new_name, 'wb')

      try:

        dest.write(old_contents)

      finally:

        dest.close()

    elif orig_name.endswith('.tar'):

      gz = gzip.GzipFile(new_name, 'w')

      try:

        gz.write(old_contents)

      finally:

        gz.close()

    elif orig_name.endswith('.tar.bz2'):

      old_tar_content_decompressed = bz2.decompress(old_contents)

      gz = gzip.GzipFile(new_name, 'w')

      try:

        gz.write(old_tar_content_decompressed)

      finally:

        gz.close()

    elif orig_name.endswith('.zip') or zipfile.is_zipfile(orig_name):

      import time

      old_contents_f = StringIO(old_contents)

      zip = zipfile.ZipFile(old_contents_f, "r")

      try:

        tar = tarfile.open(new_name, 'w:gz')

        try:

          for info in zip.infolist():

            tarinfo = tarfile.TarInfo(info.filename)

            tarinfo.size = info.file_size

            tarinfo.mtime = time.mktime(info.date_time + (0, 1, -1))

            if info.filename.endswith("/"):

                tarinfo.mode = 0755

                tarinfo.type = tarfile.DIRTYPE

            else:

                tarinfo.mode = 0644

                tarinfo.type = tarfile.REGTYPE

            contents = StringIO(zip.read(info.filename))

            tar.addfile(tarinfo, contents)

        finally:

          tar.close()

      finally:

        zip.close()

    else:

      raise BzrCommandError('Unsupported format for repack: %s' % orig_name)

# vim: ts=2 sts=2 sw=2