#!/usr/bin/python -u # vim: set fileencoding=utf-8 : # # (C) 2010 Guido Guenther # This program 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. # # 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 General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA # # Based on the aa-create-git-repo and dom-new-git-repo shell scripts """Create a remote repo based on the current one""" # TODO: allow to add hooks by default import sys import os, os.path import urlparse import subprocess import tty, termios import gbp.deb as du from gbp.command_wrappers import (CommandExecFailed, PristineTar, GitCommand, GitFetch) from gbp.config import (GbpOptionParser, GbpOptionGroup) from gbp.errors import GbpError from gbp.git import (GitRepositoryError, GitRepository) import gbp.log def print_config(remote, branches): print """[remote "%(name)s"] url = %(url)s fetch = +refs/heads/*:refs/remotes/%(name)s/*""" % remote for branch in branches: print " push = %s" % branch for branch in branches: print """[branch "%s"] remote = %s merge = refs/heads/%s""" % (branch, remote['name'], branch) def parse_remote(remote_url, name, pkg): """Sanity check our remote URL""" frags = urlparse.urlparse(remote_url) if frags.scheme not in ['ssh', 'git+ssh']: raise GbpError, "Remote URL must use ssh protocol." if not '%(pkg)s' in remote_url and not remote_url.endswith(".git"): raise GbpError, "Remote URL needs to contain either a repository name or '%(pkg)s'" remote = { 'pkg' : pkg, 'url' : remote_url % { 'pkg': pkg }, 'dir' : frags.path % { 'pkg': pkg }, 'host': frags.netloc, 'name': name} return remote def read_yn(): fd = sys.stdin.fileno() old_settings = termios.tcgetattr(fd) try: tty.setraw(sys.stdin.fileno()) ch = sys.stdin.read(1) finally: termios.tcsetattr(fd, termios.TCSADRAIN, old_settings) if ch in ( 'y', 'Y' ): return True else: return False def setup_branch_tracking(remote, branches): gitRemoteAdd = GitCommand('remote', ['add']) gitRemoteAdd([remote['name'], remote['url']]) GitFetch()([remote['name']]) gitTrackRemote = GitCommand('branch', ['--set-upstream']) for branch in branches: gitTrackRemote(['%s' % branch, '%s/%s' % (remote['name'], branch)]) def push_branches(remote, branches): gitPush = GitCommand('push') gitPush([remote['url']] + branches) gitPush([remote['url'], '--tags']) def main(argv): retval = 0 changelog = 'debian/changelog' parser = GbpOptionParser(command=os.path.basename(argv[0]), prefix='', usage='%prog [options] - create a remote repository') branch_group = GbpOptionGroup(parser, "branch options", "branch layout and tracking options") branch_group.add_config_file_option(option_name="remote-url-pattern", dest="remote_url") parser.add_option_group(branch_group) branch_group.add_config_file_option(option_name="upstream-branch", dest="upstream_branch") branch_group.add_config_file_option(option_name="debian-branch", dest="debian_branch") branch_group.add_boolean_config_file_option(option_name="pristine-tar", dest="pristine_tar") branch_group.add_boolean_config_file_option(option_name="track", dest='track') parser.add_option("-v", "--verbose", action="store_true", dest="verbose", default=False, help="verbose command execution") parser.add_config_file_option(option_name="color", dest="color", type='tristate') parser.add_option("--remote-name", dest="name", default="origin", help="The name of the remote, default is 'origin'") (options, args) = parser.parse_args(argv) gbp.log.setup(options.color, options.verbose) try: repo = GitRepository(os.path.curdir) except GitRepositoryError: gbp.log.err("%s is not a git repository" % (os.path.abspath('.'))) return 1 try: branches = [] for branch in [ options.debian_branch, options.upstream_branch ]: if repo.has_branch(branch): branches += [ branch ] if repo.has_branch(PristineTar.branch) and options.pristine_tar: branches += [ PristineTar.branch ] try: cp = du.parse_changelog(filename=changelog) pkg = cp['Source'] except gbp.deb.NoChangelogError: pkg = None if not pkg: gbp.log.warn("Couldn't parse changelog, will use directory name.") pkg = os.path.basename(os.path.abspath(os.path.curdir)) pkg = os.path.splitext(pkg)[0] remote = parse_remote(options.remote_url, options.name, pkg) if repo.has_remote_repo(options.name): raise GbpError, "You already have a remote name '%s' defined for this repository." % options.name gbp.log.info("Shall I create a repository for '%(pkg)s' at '%(url)s' now? (y/n)?" % remote) if not read_yn(): raise GbpError, "Aborted." # Create and run the remote script ssh = 'ssh %(host)s sh' % remote remote_script = """ cat < description EOF""" % remote if options.verbose: print remote_script p1 = subprocess.Popen([remote_script], stdout=subprocess.PIPE, shell=True) p2 = subprocess.Popen([ssh], stdin=p1.stdout, shell=True) p2.communicate() if p2.returncode: raise GbpError, "Error creating remote repository" push_branches(remote, branches) if options.track: setup_branch_tracking(remote, branches) else: gbp.log.info("You can now add:") print_config(remote, branches) gbp.log.info("to your .git/config to 'gbp-pull' and 'git push' in the future.") except CommandExecFailed: retval = 1 except GbpError, err: if len(err.__str__()): gbp.log.err(err) retval = 1 return retval if __name__ == '__main__': sys.exit(main(sys.argv)) # vim:et:ts=4:sw=4:et:sts=4:ai:set list listchars=tab\:»·,trail\:·: