chiark / gitweb /
~mdw / stgit / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
Check bottom and invariants
[stgit] / stgit / main.py
diff --git a/stgit/main.py b/stgit/main.py
index d0e7230ad7933f46006320c81b295c6d1486a0d3..95e25f8795648d33052b2607ebb73d79ec7999a6 100644 (file)
--- a/stgit/main.py
+++ b/stgit/main.py
@@ -22,10 +22,8 @@ import sys, os
 from optparse import OptionParser
 
 import stgit.commands
 from optparse import OptionParser
 
 import stgit.commands
-from stgit.stack import Series, StackException
-from stgit.git import GitException
-from stgit.commands.common import CmdException
-from stgit.gitmergeonefile import GitMergeException
+from stgit.out import *
+from stgit import utils
 
 #
 # The commands map
 
 #
 # The commands map
@@ -33,26 +31,50 @@ from stgit.gitmergeonefile import GitMergeException
 class Commands(dict):
     """Commands class. It performs on-demand module loading
     """
 class Commands(dict):
     """Commands class. It performs on-demand module loading
     """
+    def canonical_cmd(self, key):
+        """Return the canonical name for a possibly-shortenned
+        command name.
+        """
+        candidates = [cmd for cmd in self.keys() if cmd.startswith(key)]
+
+        if not candidates:
+            out.error('Unknown command: %s' % key,
+                      'Try "%s help" for a list of supported commands' % prog)
+            sys.exit(utils.STGIT_GENERAL_ERROR)
+        elif len(candidates) > 1:
+            out.error('Ambiguous command: %s' % key,
+                      'Candidates are: %s' % ', '.join(candidates))
+            sys.exit(utils.STGIT_GENERAL_ERROR)
+
+        return candidates[0]
+        
     def __getitem__(self, key):
     def __getitem__(self, key):
-        cmd_mod = self.get(key)
+        """Return the command python module name based.
+        """
+        global prog
+
+        cmd_mod = self.get(key) or self.get(self.canonical_cmd(key))
+            
         __import__('stgit.commands.' + cmd_mod)
         return getattr(stgit.commands, cmd_mod)
 
 commands = Commands({
     'add':              'add',
     'applied':          'applied',
         __import__('stgit.commands.' + cmd_mod)
         return getattr(stgit.commands, cmd_mod)
 
 commands = Commands({
     'add':              'add',
     'applied':          'applied',
-    'assimilate':       'assimilate',
     'branch':           'branch',
     'delete':           'delete',
     'diff':             'diff',
     'clean':            'clean',
     'clone':            'clone',
     'commit':           'commit',
     'branch':           'branch',
     'delete':           'delete',
     'diff':             'diff',
     'clean':            'clean',
     'clone':            'clone',
     'commit':           'commit',
+    'cp':              'copy',
+    'edit':             'edit',
     'export':           'export',
     'files':            'files',
     'float':            'float',
     'fold':             'fold',
     'goto':             'goto',
     'export':           'export',
     'files':            'files',
     'float':            'float',
     'fold':             'fold',
     'goto':             'goto',
+    'hide':             'hide',
     'id':               'id',
     'import':           'imprt',
     'init':             'init',
     'id':               'id',
     'import':           'imprt',
     'init':             'init',
@@ -64,42 +86,53 @@ commands = Commands({
     'pop':              'pop',
     'pull':             'pull',
     'push':             'push',
     'pop':              'pop',
     'pull':             'pull',
     'push':             'push',
+    'rebase':           'rebase',
     'refresh':          'refresh',
     'rename':           'rename',
     'refresh':          'refresh',
     'rename':           'rename',
+    'repair':           'repair',
     'resolved':         'resolved',
     'rm':               'rm',
     'series':           'series',
     'show':             'show',
     'resolved':         'resolved',
     'rm':               'rm',
     'series':           'series',
     'show':             'show',
+    'sink':             'sink',
     'status':           'status',
     'status':           'status',
+    'sync':             'sync',
     'top':              'top',
     'unapplied':        'unapplied',
     'top':              'top',
     'unapplied':        'unapplied',
-    'uncommit':         'uncommit'
+    'uncommit':         'uncommit',
+    'unhide':           'unhide'
     })
 
 # classification: repository, stack, patch, working copy
 repocommands = (
     })
 
 # classification: repository, stack, patch, working copy
 repocommands = (
-    'branch',
     'clone',
     'id',
     'clone',
     'id',
-    'pull'
     )
 stackcommands = (
     'applied',
     )
 stackcommands = (
     'applied',
-    'assimilate',
+    'branch',
     'clean',
     'commit',
     'float',
     'goto',
     'clean',
     'commit',
     'float',
     'goto',
+    'hide',
     'init',
     'init',
+    'patches',
     'pop',
     'pop',
+    'pull',
     'push',
     'push',
+    'rebase',
+    'repair',
     'series',
     'series',
+    'sink',
     'top',
     'unapplied',
     'top',
     'unapplied',
-    'uncommit'
+    'uncommit',
+    'unhide',
     )
 patchcommands = (
     'delete',
     )
 patchcommands = (
     'delete',
+    'edit',
     'export',
     'files',
     'fold',
     'export',
     'files',
     'fold',
@@ -110,15 +143,16 @@ patchcommands = (
     'pick',
     'refresh',
     'rename',
     'pick',
     'refresh',
     'rename',
-    'show'
+    'show',
+    'sync',
     )
 wccommands = (
     'add',
     )
 wccommands = (
     'add',
+    'cp',
     'diff',
     'diff',
-    'patches',
     'resolved',
     'rm',
     'resolved',
     'rm',
-    'status'
+    'status',
     )
 
 def _print_helpstring(cmd):
     )
 
 def _print_helpstring(cmd):
@@ -165,55 +199,53 @@ def print_help():
 def main():
     """The main function
     """
 def main():
     """The main function
     """
+    global prog
+
     prog = os.path.basename(sys.argv[0])
 
     if len(sys.argv) < 2:
         print >> sys.stderr, 'usage: %s <command>' % prog
         print >> sys.stderr, \
               '  Try "%s --help" for a list of supported commands' % prog
     prog = os.path.basename(sys.argv[0])
 
     if len(sys.argv) < 2:
         print >> sys.stderr, 'usage: %s <command>' % prog
         print >> sys.stderr, \
               '  Try "%s --help" for a list of supported commands' % prog
-        sys.exit(1)
+        sys.exit(utils.STGIT_GENERAL_ERROR)
 
     cmd = sys.argv[1]
 
     if cmd in ['-h', '--help']:
 
     cmd = sys.argv[1]
 
     if cmd in ['-h', '--help']:
-        if len(sys.argv) >= 3 and sys.argv[2] in commands:
-            cmd = sys.argv[2]
+        if len(sys.argv) >= 3:
+            cmd = commands.canonical_cmd(sys.argv[2])
             sys.argv[2] = '--help'
         else:
             print_help()
             sys.argv[2] = '--help'
         else:
             print_help()
-            sys.exit(0)
+            sys.exit(utils.STGIT_SUCCESS)
     if cmd == 'help':
         if len(sys.argv) == 3 and not sys.argv[2] in ['-h', '--help']:
     if cmd == 'help':
         if len(sys.argv) == 3 and not sys.argv[2] in ['-h', '--help']:
-            cmd = sys.argv[2]
+            cmd = commands.canonical_cmd(sys.argv[2])
             if not cmd in commands:
             if not cmd in commands:
-                print >> sys.stderr, '%s help: "%s" command unknown' \
-                      % (prog, cmd)
-                sys.exit(1)
+                out.error('%s help: "%s" command unknown' % (prog, cmd))
+                sys.exit(utils.STGIT_GENERAL_ERROR)
 
             sys.argv[0] += ' %s' % cmd
             command = commands[cmd]
             parser = OptionParser(usage = command.usage,
                                   option_list = command.options)
 
             sys.argv[0] += ' %s' % cmd
             command = commands[cmd]
             parser = OptionParser(usage = command.usage,
                                   option_list = command.options)
-            parser.print_help()
+            from pydoc import pager
+            pager(parser.format_help())
         else:
             print_help()
         else:
             print_help()
-        sys.exit(0)
+        sys.exit(utils.STGIT_SUCCESS)
     if cmd in ['-v', '--version', 'version']:
         from stgit.version import version
         print 'Stacked GIT %s' % version
         os.system('git --version')
         print 'Python version %s' % sys.version
     if cmd in ['-v', '--version', 'version']:
         from stgit.version import version
         print 'Stacked GIT %s' % version
         os.system('git --version')
         print 'Python version %s' % sys.version
-        sys.exit(0)
+        sys.exit(utils.STGIT_SUCCESS)
     if cmd in ['copyright']:
         print __copyright__
     if cmd in ['copyright']:
         print __copyright__
-        sys.exit(0)
-    if not cmd in commands:
-        print >> sys.stderr, 'Unknown command: %s' % cmd
-        print >> sys.stderr, '  Try "%s help" for a list of supported ' \
-              'commands' % prog
-        sys.exit(1)
+        sys.exit(utils.STGIT_SUCCESS)
 
     # re-build the command line arguments
 
     # re-build the command line arguments
+    cmd = commands.canonical_cmd(cmd)
     sys.argv[0] += ' %s' % cmd
     del(sys.argv[1])
 
     sys.argv[0] += ' %s' % cmd
     del(sys.argv[1])
 
@@ -221,22 +253,40 @@ def main():
     usage = command.usage.split('\n')[0].strip()
     parser = OptionParser(usage = usage, option_list = command.options)
     options, args = parser.parse_args()
     usage = command.usage.split('\n')[0].strip()
     parser = OptionParser(usage = usage, option_list = command.options)
     options, args = parser.parse_args()
+    directory = command.directory
+
+    # These modules are only used from this point onwards and do not
+    # need to be imported earlier
+    from stgit.exception import StgException
+    from stgit.config import config_setup
+    from ConfigParser import ParsingError, NoSectionError
+    from stgit.stack import Series
+
     try:
     try:
-        # 'clone' doesn't expect an already initialised GIT tree. A Series
-        # object will be created after the GIT tree is cloned
-        if cmd != 'clone':
+        debug_level = int(os.environ.get('STGIT_DEBUG_LEVEL', 0))
+    except ValueError:
+        out.error('Invalid STGIT_DEBUG_LEVEL environment variable')
+        sys.exit(utils.STGIT_GENERAL_ERROR)
+
+    try:
+        directory.setup()
+        config_setup()
+
+        # Some commands don't (always) need an initialized series.
+        if directory.needs_current_series:
             if hasattr(options, 'branch') and options.branch:
                 command.crt_series = Series(options.branch)
             else:
                 command.crt_series = Series()
             if hasattr(options, 'branch') and options.branch:
                 command.crt_series = Series(options.branch)
             else:
                 command.crt_series = Series()
-            stgit.commands.common.crt_series = command.crt_series
 
         command.func(parser, options, args)
 
         command.func(parser, options, args)
-    except (IOError, CmdException, StackException, GitException,
-            GitMergeException), err:
-        print >> sys.stderr, '%s %s: %s' % (prog, cmd, err)
-        sys.exit(2)
+    except (StgException, IOError, ParsingError, NoSectionError), err:
+        out.error(str(err), title = '%s %s' % (prog, cmd))
+        if debug_level > 0:
+            raise
+        else:
+            sys.exit(utils.STGIT_COMMAND_ERROR)
     except KeyboardInterrupt:
     except KeyboardInterrupt:
-        sys.exit(1)
+        sys.exit(utils.STGIT_GENERAL_ERROR)
 
 
-    sys.exit(0)
+    sys.exit(utils.STGIT_SUCCESS)
GIT patch-maintenance porcelain (local modifications)