mdw@git.distorted.org.uk Git - stgit/blob - stgit/main.py

   1 """Basic quilt-like functionality
   2 """
   3
   4 __copyright__ = """
   5 Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>
   6
   7 This program is free software; you can redistribute it and/or modify
   8 it under the terms of the GNU General Public License version 2 as
   9 published by the Free Software Foundation.
  10
  11 This program is distributed in the hope that it will be useful,
  12 but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 GNU General Public License for more details.
  15
  16 You should have received a copy of the GNU General Public License
  17 along with this program; if not, write to the Free Software
  18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  19 """
  20
  21 import sys, os
  22 from optparse import OptionParser, make_option
  23
  24 from stgit.utils import *
  25 from stgit import stack, git, gitmergeonefile
  26 from stgit.version import version
  27 from stgit.config import config
  28 from stgit.commands.common import *
  29
  30 # The commands
  31 import stgit.commands.add
  32 import stgit.commands.applied
  33 import stgit.commands.branch
  34 import stgit.commands.delete
  35 import stgit.commands.diff
  36 import stgit.commands.clean
  37 import stgit.commands.clone
  38 import stgit.commands.commit
  39 import stgit.commands.export
  40 import stgit.commands.files
  41 import stgit.commands.fold
  42 import stgit.commands.id
  43 import stgit.commands.imprt
  44 import stgit.commands.init
  45 import stgit.commands.mail
  46 import stgit.commands.new
  47 import stgit.commands.patches
  48 import stgit.commands.pick
  49 import stgit.commands.pop
  50 import stgit.commands.pull
  51 import stgit.commands.push
  52 import stgit.commands.refresh
  53 import stgit.commands.rename
  54 import stgit.commands.resolved
  55 import stgit.commands.rm
  56 import stgit.commands.series
  57 import stgit.commands.show
  58 import stgit.commands.status
  59 import stgit.commands.top
  60 import stgit.commands.unapplied
  61 import stgit.commands.uncommit
  62
  63
  64 #
  65 # The commands map
  66 #
  67 commands = {
  68     'add':      stgit.commands.add,
  69     'applied':  stgit.commands.applied,
  70     'branch':   stgit.commands.branch,
  71     'delete':   stgit.commands.delete,
  72     'diff':     stgit.commands.diff,
  73     'clean':    stgit.commands.clean,
  74     'clone':    stgit.commands.clone,
  75     'commit':   stgit.commands.commit,
  76     'export':   stgit.commands.export,
  77     'files':    stgit.commands.files,
  78     'fold':     stgit.commands.fold,
  79     'id':       stgit.commands.id,
  80     'import':   stgit.commands.imprt,
  81     'init':     stgit.commands.init,
  82     'mail':     stgit.commands.mail,
  83     'new':      stgit.commands.new,
  84     'patches':  stgit.commands.patches,
  85     'pick':     stgit.commands.pick,
  86     'pop':      stgit.commands.pop,
  87     'pull':     stgit.commands.pull,
  88     'push':     stgit.commands.push,
  89     'refresh':  stgit.commands.refresh,
  90     'rename':   stgit.commands.rename,
  91     'resolved': stgit.commands.resolved,
  92     'rm':       stgit.commands.rm,
  93     'series':   stgit.commands.series,
  94     'show':     stgit.commands.show,
  95     'status':   stgit.commands.status,
  96     'top':      stgit.commands.top,
  97     'unapplied':stgit.commands.unapplied,
  98     'uncommit': stgit.commands.uncommit,
  99     }
 100
 101 def print_help():
 102     print 'usage: %s <command> [options]' % os.path.basename(sys.argv[0])
 103     print
 104     print 'commands:'
 105     print '  help        print the detailed command usage'
 106     print '  version     display version information'
 107     print '  copyright   display copyright information'
 108     print
 109
 110     cmds = commands.keys()
 111     cmds.sort()
 112     for cmd in cmds:
 113         print '  ' + cmd + ' ' * (12 - len(cmd)) + commands[cmd].help
 114
 115 #
 116 # The main function (command dispatcher)
 117 #
 118 def main():
 119     """The main function
 120     """
 121     prog = os.path.basename(sys.argv[0])
 122
 123     if len(sys.argv) < 2:
 124         print >> sys.stderr, 'Unknown command'
 125         print >> sys.stderr, \
 126               '  Try "%s --help" for a list of supported commands' % prog
 127         sys.exit(1)
 128
 129     cmd = sys.argv[1]
 130
 131     if cmd in ['-h', '--help']:
 132         if len(sys.argv) == 3 and sys.argv[2] in commands:
 133             cmd = sys.argv[2]
 134             sys.argv[2] = '--help'
 135         else:
 136             print_help()
 137             sys.exit(0)
 138     if cmd == 'help':
 139         if len(sys.argv) == 3 and not sys.argv[2] in ['-h', '--help']:
 140             cmd = sys.argv[2]
 141             if not cmd in commands:
 142                 print >> sys.stderr, '%s help: "%s" command unknown' \
 143                       % (prog, cmd)
 144                 sys.exit(1)
 145
 146             sys.argv[0] += ' %s' % cmd
 147             command = commands[cmd]
 148             parser = OptionParser(usage = command.usage,
 149                                   option_list = command.options)
 150             parser.print_help()
 151         else:
 152             print 'usage: %s help <command>' % prog
 153
 154         sys.exit(0)
 155     if cmd in ['-v', '--version', 'version']:
 156         print 'Stacked GIT %s' % version
 157         os.system('git --version')
 158         print 'Python version %s' % sys.version
 159         sys.exit(0)
 160     if cmd in ['copyright']:
 161         print __copyright__
 162         sys.exit(0)
 163     if not cmd in commands:
 164         print >> sys.stderr, 'Unknown command: %s' % cmd
 165         print >> sys.stderr, '  Try "%s help" for a list of supported ' \
 166               'commands' % prog
 167         sys.exit(1)
 168
 169     # re-build the command line arguments
 170     sys.argv[0] += ' %s' % cmd
 171     del(sys.argv[1])
 172
 173     command = commands[cmd]
 174     usage = command.usage.split('\n')[0].strip()
 175     parser = OptionParser(usage = usage, option_list = command.options)
 176     options, args = parser.parse_args()
 177     try:
 178         # 'clone' doesn't expect an already initialised GIT tree. A Series
 179         # object will be created after the GIT tree is cloned
 180         if cmd != 'clone':
 181             if hasattr(options, 'branch') and options.branch:
 182                 command.crt_series = stack.Series(options.branch)
 183             else:
 184                 command.crt_series = stack.Series()
 185             stgit.commands.common.crt_series = command.crt_series
 186
 187         command.func(parser, options, args)
 188     except (IOError, CmdException, stack.StackException, git.GitException,
 189             gitmergeonefile.GitMergeException), err:
 190         print >> sys.stderr, '%s %s: %s' % (prog, cmd, err)
 191         sys.exit(2)
 192     except KeyboardInterrupt:
 193         sys.exit(1)
 194
 195     sys.exit(0)