+"""Branch command
+"""
+
+__copyright__ = """
+Copyright (C) 2005, Chuck Lever <cel@netapp.com>
+
+This program is free software; you can redistribute it and/or modify
+it under the terms of the GNU General Public License version 2 as
+published by the Free Software Foundation.
+
+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
+"""
+
+import sys, os
+from optparse import OptionParser, make_option
+
+from stgit.commands.common import *
+from stgit.utils import *
+from stgit import stack, git
+
+
+help = 'manage development branches'
+usage = """%prog [options] branch-name [commit-id]
+
+Create, list, switch between, rename, or delete development branches
+within a git repository. By default, a single branch called 'master'
+is always created in a new repository. This subcommand allows you to
+manage several patch series in the same repository.
+
+When displaying the branches, the names can be prefixed with
+'s' (StGIT managed) or 'p' (protected)."""
+
+options = [make_option('-c', '--create',
+ help = 'create a new development branch',
+ action = 'store_true'),
+ make_option('--delete',
+ help = 'delete an existing development branch',
+ action = 'store_true'),
+ make_option('--force',
+ help = 'force a delete when the series is not empty',
+ action = 'store_true'),
+ make_option('-l', '--list',
+ help = 'list branches contained in this repository',
+ action = 'store_true'),
+ make_option('-r', '--rename',
+ help = 'rename an existing development branch',
+ action = 'store_true')]
+
+
+def is_current_branch(branch_name):
+ return git.get_head_file() == branch_name
+
+def print_branch(branch_name):
+ initialized = ' '
+ current = ' '
+ if os.path.isdir(os.path.join(git.base_dir, 'patches', branch_name)):
+ initialized = 's'
+ if is_current_branch(branch_name):
+ current = '>'
+ print '%s %s\t%s' % (current, initialized, branch_name)
+
+def delete_branch(doomed_name, force = False):
+ if is_current_branch(doomed_name) and doomed_name != 'master':
+ git.switch_branch('master')
+
+ stack.Series(doomed_name).delete(force)
+
+ if doomed_name != 'master':
+ git.delete_branch(doomed_name)
+
+ print 'Branch "%s" has been deleted.' % doomed_name
+
+def rename_branch(from_name, to_name):
+ if from_name == 'master':
+ raise CmdException, 'Renaming the master branch is not allowed'
+
+ to_patchdir = os.path.join(git.base_dir, 'patches', to_name)
+ if os.path.isdir(to_patchdir):
+ raise CmdException, '"%s" already exists' % to_patchdir
+ to_base = os.path.join(git.base_dir, 'refs', 'bases', to_name)
+ if os.path.isfile(to_base):
+ raise CmdException, '"%s" already exists' % to_base
+
+ git.rename_branch(from_name, to_name)
+
+ from_patchdir = os.path.join(git.base_dir, 'patches', from_name)
+ if os.path.isdir(from_patchdir):
+ os.rename(from_patchdir, to_patchdir)
+ from_base = os.path.join(git.base_dir, 'refs', 'bases', from_name)
+ if os.path.isfile(from_base):
+ os.rename(from_base, to_base)
+
+ print 'Renamed branch "%s" as "%s".' % (from_name, to_name)
+
+def func(parser, options, args):
+
+ if options.create:
+
+ if len(args) == 0 or len(args) > 2:
+ parser.error('incorrect number of arguments')
+ tree_id = None
+ if len(args) == 2:
+ tree_id = args[1]
+
+ git.create_branch(args[0], tree_id)
+ stack.Series(args[0]).init()
+
+ print 'Branch "%s" created.' % args[0]
+ return
+
+ elif options.delete:
+
+ if len(args) != 1:
+ parser.error('incorrect number of arguments')
+ delete_branch(args[0], options.force)
+ return
+
+ elif options.list:
+
+ if len(args) != 0:
+ parser.error('incorrect number of arguments')
+
+ branches = os.listdir(os.path.join(git.base_dir, 'refs', 'heads'))
+ branches.sort()
+
+ print 'Available branches:'
+ for i in branches:
+ print_branch(i)
+ return
+
+ elif options.rename:
+
+ if len(args) != 2:
+ parser.error('incorrect number of arguments')
+ rename_branch(args[0], args[1])
+ return
+
+ elif len(args) == 1:
+
+ print 'Switching to branch "%s"...' % args[0],
+ sys.stdout.flush()
+
+ git.switch_branch(args[0])
+
+ print 'done'
+ return
+
+ # default action: print the current branch
+ if len(args) != 0:
+ parser.error('incorrect number of arguments')
+
+ print git.get_head_file()
~mdw / stgit / commitdiff