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.goto
  43 import stgit.commands.id
  44 import stgit.commands.imprt
  45 import stgit.commands.init
  46 import stgit.commands.log
  47 import stgit.commands.mail
  48 import stgit.commands.new
  49 import stgit.commands.patches
  50 import stgit.commands.pick
  51 import stgit.commands.pop
  52 import stgit.commands.pull
  53 import stgit.commands.push
  54 import stgit.commands.refresh
  55 import stgit.commands.rename
  56 import stgit.commands.resolved
  57 import stgit.commands.rm
  58 import stgit.commands.series
  59 import stgit.commands.show
  60 import stgit.commands.status
  61 import stgit.commands.top
  62 import stgit.commands.unapplied
  63 import stgit.commands.uncommit
  64
  65
  66 #
  67 # The commands map
  68 #
  69 commands = {
  70     'add':      stgit.commands.add,
  71     'applied':  stgit.commands.applied,
  72     'branch':   stgit.commands.branch,
  73     'delete':   stgit.commands.delete,
  74     'diff':     stgit.commands.diff,
  75     'clean':    stgit.commands.clean,
  76     'clone':    stgit.commands.clone,
  77     'commit':   stgit.commands.commit,
  78     'export':   stgit.commands.export,
  79     'files':    stgit.commands.files,
  80     'fold':     stgit.commands.fold,
  81     'goto':     stgit.commands.goto,
  82     'id':       stgit.commands.id,
  83     'import':   stgit.commands.imprt,
  84     'init':     stgit.commands.init,
  85     'log':      stgit.commands.log,
  86     'mail':     stgit.commands.mail,
  87     'new':      stgit.commands.new,
  88     'patches':  stgit.commands.patches,
  89     'pick':     stgit.commands.pick,
  90     'pop':      stgit.commands.pop,
  91     'pull':     stgit.commands.pull,
  92     'push':     stgit.commands.push,
  93     'refresh':  stgit.commands.refresh,
  94     'rename':   stgit.commands.rename,
  95     'resolved': stgit.commands.resolved,
  96     'rm':       stgit.commands.rm,
  97     'series':   stgit.commands.series,
  98     'show':     stgit.commands.show,
  99     'status':   stgit.commands.status,
 100     'top':      stgit.commands.top,
 101     'unapplied':stgit.commands.unapplied,
 102     'uncommit': stgit.commands.uncommit,
 103     }
 104
 105 # classification: repository, stack, patch, working copy
 106 repocommands = (
 107     'branch',
 108     'clone',
 109     'id',
 110     'pull'
 111     )
 112 stackcommands = (
 113     'applied',
 114     'clean',
 115     'commit',
 116     'goto',
 117     'init',
 118     'pop',
 119     'push',
 120     'series',
 121     'top',
 122     'unapplied',
 123     'uncommit'
 124     )
 125 patchcommands = (
 126     'delete',
 127     'export',
 128     'files',
 129     'fold',
 130     'import',
 131     'log',
 132     'mail',
 133     'new',
 134     'pick',
 135     'refresh',
 136     'rename',
 137     'show'
 138     )
 139 wccommands = (
 140     'add',
 141     'diff',
 142     'patches',
 143     'resolved',
 144     'rm',
 145     'status'
 146     )
 147
 148 def _print_helpstring(cmd):
 149     print '  ' + cmd + ' ' * (12 - len(cmd)) + commands[cmd].help
 150
 151 def print_help():
 152     print 'usage: %s <command> [options]' % os.path.basename(sys.argv[0])
 153     print
 154     print 'Generic commands:'
 155     print '  help        print the detailed command usage'
 156     print '  version     display version information'
 157     print '  copyright   display copyright information'
 158     # unclassified commands if any
 159     cmds = commands.keys()
 160     cmds.sort()
 161     for cmd in cmds:
 162         if not cmd in repocommands and not cmd in stackcommands \
 163                and not cmd in patchcommands and not cmd in wccommands:
 164             _print_helpstring(cmd)
 165     print
 166
 167     print 'Repository commands:'
 168     for cmd in repocommands:
 169         _print_helpstring(cmd)
 170     print
 171
 172     print 'Stack commands:'
 173     for cmd in stackcommands:
 174         _print_helpstring(cmd)
 175     print
 176
 177     print 'Patch commands:'
 178     for cmd in patchcommands:
 179         _print_helpstring(cmd)
 180     print
 181
 182     print 'Working-copy commands:'
 183     for cmd in wccommands:
 184         _print_helpstring(cmd)
 185
 186 #
 187 # The main function (command dispatcher)
 188 #
 189 def main():
 190     """The main function
 191     """
 192     prog = os.path.basename(sys.argv[0])
 193
 194     if len(sys.argv) < 2:
 195         print >> sys.stderr, 'Unknown command'
 196         print >> sys.stderr, \
 197               '  Try "%s --help" for a list of supported commands' % prog
 198         sys.exit(1)
 199
 200     cmd = sys.argv[1]
 201
 202     if cmd in ['-h', '--help']:
 203         if len(sys.argv) == 3 and sys.argv[2] in commands:
 204             cmd = sys.argv[2]
 205             sys.argv[2] = '--help'
 206         else:
 207             print_help()
 208             sys.exit(0)
 209     if cmd == 'help':
 210         if len(sys.argv) == 3 and not sys.argv[2] in ['-h', '--help']:
 211             cmd = sys.argv[2]
 212             if not cmd in commands:
 213                 print >> sys.stderr, '%s help: "%s" command unknown' \
 214                       % (prog, cmd)
 215                 sys.exit(1)
 216
 217             sys.argv[0] += ' %s' % cmd
 218             command = commands[cmd]
 219             parser = OptionParser(usage = command.usage,
 220                                   option_list = command.options)
 221             parser.print_help()
 222         else:
 223             print 'usage: %s help <command>' % prog
 224
 225         sys.exit(0)
 226     if cmd in ['-v', '--version', 'version']:
 227         print 'Stacked GIT %s' % version
 228         os.system('git --version')
 229         print 'Python version %s' % sys.version
 230         sys.exit(0)
 231     if cmd in ['copyright']:
 232         print __copyright__
 233         sys.exit(0)
 234     if not cmd in commands:
 235         print >> sys.stderr, 'Unknown command: %s' % cmd
 236         print >> sys.stderr, '  Try "%s help" for a list of supported ' \
 237               'commands' % prog
 238         sys.exit(1)
 239
 240     # re-build the command line arguments
 241     sys.argv[0] += ' %s' % cmd
 242     del(sys.argv[1])
 243
 244     command = commands[cmd]
 245     usage = command.usage.split('\n')[0].strip()
 246     parser = OptionParser(usage = usage, option_list = command.options)
 247     options, args = parser.parse_args()
 248     try:
 249         # 'clone' doesn't expect an already initialised GIT tree. A Series
 250         # object will be created after the GIT tree is cloned
 251         if cmd != 'clone':
 252             if hasattr(options, 'branch') and options.branch:
 253                 command.crt_series = stack.Series(options.branch)
 254             else:
 255                 command.crt_series = stack.Series()
 256             stgit.commands.common.crt_series = command.crt_series
 257
 258         command.func(parser, options, args)
 259     except (IOError, CmdException, stack.StackException, git.GitException,
 260             gitmergeonefile.GitMergeException), err:
 261         print >> sys.stderr, '%s %s: %s' % (prog, cmd, err)
 262         sys.exit(2)
 263     except KeyboardInterrupt:
 264         sys.exit(1)
 265
 266     sys.exit(0)