~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/commands.py

  • Committer: Robert Collins
  • Date: 2005-10-10 08:27:26 UTC
  • mfrom: (1185.15.9)
  • Revision ID: robertc@robertcollins.net-20051010082726-b709d6577691ba69
mergeĀ fromĀ upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2004, 2005 by Canonical Ltd
 
2
 
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
#
 
12
 
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
18
# TODO: probably should say which arguments are candidates for glob
19
19
# expansion on windows and do that at the command level.
20
20
 
 
21
# TODO: Help messages for options.
 
22
 
21
23
# TODO: Define arguments by objects, rather than just using names.
22
24
# Those objects can specify the expected type of the argument, which
23
 
# would help with validation and shell completion.  They could also provide
24
 
# help/explanation for that argument in a structured way.
25
 
 
26
 
# TODO: Specific "examples" property on commands for consistent formatting.
 
25
# would help with validation and shell completion.
27
26
 
28
27
# TODO: "--profile=cum", to change sort order.  Is there any value in leaving
29
28
# the profile output behind so it can be interactively examined?
30
29
 
 
30
import sys
31
31
import os
32
 
import sys
33
 
 
34
 
from bzrlib.lazy_import import lazy_import
35
 
lazy_import(globals(), """
36
 
import codecs
37
 
import errno
38
32
from warnings import warn
 
33
from inspect import getdoc
39
34
 
40
35
import bzrlib
41
 
from bzrlib import (
42
 
    debug,
43
 
    errors,
44
 
    option,
45
 
    osutils,
46
 
    registry,
47
 
    trace,
48
 
    win32utils,
49
 
    )
50
 
""")
51
 
 
52
 
from bzrlib.symbol_versioning import (
53
 
    deprecated_function,
54
 
    deprecated_method,
55
 
    )
56
 
# Compatibility
57
 
from bzrlib.option import Option
58
 
 
 
36
import bzrlib.trace
 
37
from bzrlib.trace import mutter, note, log_error, warning
 
38
from bzrlib.errors import BzrError, BzrCheckError, BzrCommandError, NotBranchError
 
39
from bzrlib.revisionspec import RevisionSpec
 
40
from bzrlib import BZRDIR
59
41
 
60
42
plugin_cmds = {}
61
43
 
62
44
 
63
 
def register_command(cmd, decorate=False):
64
 
    """Utility function to help register a command
65
 
 
66
 
    :param cmd: Command subclass to register
67
 
    :param decorate: If true, allow overriding an existing command
68
 
        of the same name; the old command is returned by this function.
69
 
        Otherwise it is an error to try to override an existing command.
70
 
    """
 
45
def register_command(cmd):
 
46
    "Utility function to help register a command"
71
47
    global plugin_cmds
72
48
    k = cmd.__name__
73
49
    if k.startswith("cmd_"):
74
50
        k_unsquished = _unsquish_command_name(k)
75
51
    else:
76
52
        k_unsquished = k
77
 
    if k_unsquished not in plugin_cmds:
78
 
        plugin_cmds[k_unsquished] = cmd
79
 
        ## trace.mutter('registered plugin command %s', k_unsquished)
80
 
        if decorate and k_unsquished in builtin_command_names():
81
 
            return _builtin_commands()[k_unsquished]
82
 
    elif decorate:
83
 
        result = plugin_cmds[k_unsquished]
84
 
        plugin_cmds[k_unsquished] = cmd
85
 
        return result
 
53
    if not plugin_cmds.has_key(k_unsquished):
 
54
        plugin_cmds[k_unsquished] = cmd
 
55
        mutter('registered plugin command %s', k_unsquished)      
86
56
    else:
87
 
        trace.log_error('Two plugins defined the same command: %r' % k)
88
 
        trace.log_error('Not loading the one in %r' % sys.modules[cmd.__module__])
89
 
        trace.log_error('Previously this command was registered from %r' %
90
 
                        sys.modules[plugin_cmds[k_unsquished].__module__])
 
57
        log_error('Two plugins defined the same command: %r' % k)
 
58
        log_error('Not loading the one in %r' % sys.modules[cmd.__module__])
91
59
 
92
60
 
93
61
def _squish_command_name(cmd):
95
63
 
96
64
 
97
65
def _unsquish_command_name(cmd):
 
66
    assert cmd.startswith("cmd_")
98
67
    return cmd[4:].replace('_','-')
99
68
 
100
69
 
 
70
def _parse_revision_str(revstr):
 
71
    """This handles a revision string -> revno.
 
72
 
 
73
    This always returns a list.  The list will have one element for
 
74
    each revision specifier supplied.
 
75
 
 
76
    >>> _parse_revision_str('234')
 
77
    [<RevisionSpec_int 234>]
 
78
    >>> _parse_revision_str('234..567')
 
79
    [<RevisionSpec_int 234>, <RevisionSpec_int 567>]
 
80
    >>> _parse_revision_str('..')
 
81
    [<RevisionSpec None>, <RevisionSpec None>]
 
82
    >>> _parse_revision_str('..234')
 
83
    [<RevisionSpec None>, <RevisionSpec_int 234>]
 
84
    >>> _parse_revision_str('234..')
 
85
    [<RevisionSpec_int 234>, <RevisionSpec None>]
 
86
    >>> _parse_revision_str('234..456..789') # Maybe this should be an error
 
87
    [<RevisionSpec_int 234>, <RevisionSpec_int 456>, <RevisionSpec_int 789>]
 
88
    >>> _parse_revision_str('234....789') #Error ?
 
89
    [<RevisionSpec_int 234>, <RevisionSpec None>, <RevisionSpec_int 789>]
 
90
    >>> _parse_revision_str('revid:test@other.com-234234')
 
91
    [<RevisionSpec_revid revid:test@other.com-234234>]
 
92
    >>> _parse_revision_str('revid:test@other.com-234234..revid:test@other.com-234235')
 
93
    [<RevisionSpec_revid revid:test@other.com-234234>, <RevisionSpec_revid revid:test@other.com-234235>]
 
94
    >>> _parse_revision_str('revid:test@other.com-234234..23')
 
95
    [<RevisionSpec_revid revid:test@other.com-234234>, <RevisionSpec_int 23>]
 
96
    >>> _parse_revision_str('date:2005-04-12')
 
97
    [<RevisionSpec_date date:2005-04-12>]
 
98
    >>> _parse_revision_str('date:2005-04-12 12:24:33')
 
99
    [<RevisionSpec_date date:2005-04-12 12:24:33>]
 
100
    >>> _parse_revision_str('date:2005-04-12T12:24:33')
 
101
    [<RevisionSpec_date date:2005-04-12T12:24:33>]
 
102
    >>> _parse_revision_str('date:2005-04-12,12:24:33')
 
103
    [<RevisionSpec_date date:2005-04-12,12:24:33>]
 
104
    >>> _parse_revision_str('-5..23')
 
105
    [<RevisionSpec_int -5>, <RevisionSpec_int 23>]
 
106
    >>> _parse_revision_str('-5')
 
107
    [<RevisionSpec_int -5>]
 
108
    >>> _parse_revision_str('123a')
 
109
    Traceback (most recent call last):
 
110
      ...
 
111
    BzrError: No namespace registered for string: '123a'
 
112
    >>> _parse_revision_str('abc')
 
113
    Traceback (most recent call last):
 
114
      ...
 
115
    BzrError: No namespace registered for string: 'abc'
 
116
    >>> _parse_revision_str('branch:../branch2')
 
117
    [<RevisionSpec_branch branch:../branch2>]
 
118
    """
 
119
    import re
 
120
    old_format_re = re.compile('\d*:\d*')
 
121
    m = old_format_re.match(revstr)
 
122
    revs = []
 
123
    if m:
 
124
        warning('Colon separator for revision numbers is deprecated.'
 
125
                ' Use .. instead')
 
126
        for rev in revstr.split(':'):
 
127
            if rev:
 
128
                revs.append(RevisionSpec(int(rev)))
 
129
            else:
 
130
                revs.append(RevisionSpec(None))
 
131
    else:
 
132
        next_prefix = None
 
133
        for x in revstr.split('..'):
 
134
            if not x:
 
135
                revs.append(RevisionSpec(None))
 
136
            elif x[-1] == ':':
 
137
                # looks like a namespace:.. has happened
 
138
                next_prefix = x + '..'
 
139
            else:
 
140
                if next_prefix is not None:
 
141
                    x = next_prefix + x
 
142
                revs.append(RevisionSpec(x))
 
143
                next_prefix = None
 
144
        if next_prefix is not None:
 
145
            revs.append(RevisionSpec(next_prefix))
 
146
    return revs
 
147
 
 
148
 
101
149
def _builtin_commands():
102
150
    import bzrlib.builtins
103
151
    r = {}
104
152
    builtins = bzrlib.builtins.__dict__
105
153
    for name in builtins:
106
154
        if name.startswith("cmd_"):
107
 
            real_name = _unsquish_command_name(name)
 
155
            real_name = _unsquish_command_name(name)        
108
156
            r[real_name] = builtins[name]
109
157
    return r
 
158
 
110
159
            
111
160
 
112
161
def builtin_command_names():
138
187
    plugins_override
139
188
        If true, plugin commands can override builtins.
140
189
    """
141
 
    try:
142
 
        return _get_cmd_object(cmd_name, plugins_override)
143
 
    except KeyError:
144
 
        raise errors.BzrCommandError('unknown command "%s"' % cmd_name)
145
 
 
146
 
 
147
 
def _get_cmd_object(cmd_name, plugins_override=True):
148
 
    """Worker for get_cmd_object which raises KeyError rather than BzrCommandError."""
149
190
    from bzrlib.externalcommand import ExternalCommand
150
191
 
151
 
    # We want only 'ascii' command names, but the user may have typed
152
 
    # in a Unicode name. In that case, they should just get a
153
 
    # 'command not found' error later.
154
 
    # In the future, we may actually support Unicode command names.
 
192
    cmd_name = str(cmd_name)            # not unicode
155
193
 
156
194
    # first look up this command under the specified name
157
195
    cmds = _get_cmd_dict(plugins_override=plugins_override)
169
207
    if cmd_obj:
170
208
        return cmd_obj
171
209
 
172
 
    # look for plugins that provide this command but aren't installed
173
 
    for provider in command_providers_registry:
174
 
        try:
175
 
            plugin_metadata = provider.plugin_for_command(cmd_name)
176
 
        except errors.NoPluginAvailable:
177
 
            pass
178
 
        else:
179
 
            raise errors.CommandAvailableInPlugin(cmd_name, 
180
 
                                                  plugin_metadata, provider)
181
 
 
182
 
    raise KeyError
 
210
    raise BzrCommandError("unknown command %r" % cmd_name)
183
211
 
184
212
 
185
213
class Command(object):
207
235
        List of argument forms, marked with whether they are optional,
208
236
        repeated, etc.
209
237
 
210
 
                Examples:
211
 
 
212
 
                ['to_location', 'from_branch?', 'file*']
213
 
 
214
 
                'to_location' is required
215
 
                'from_branch' is optional
216
 
                'file' can be specified 0 or more times
217
 
 
218
238
    takes_options
219
 
        List of options that may be given for this command.  These can
220
 
        be either strings, referring to globally-defined options,
221
 
        or option objects.  Retrieve through options().
 
239
        List of options that may be given for this command.
222
240
 
223
241
    hidden
224
242
        If true, this command isn't advertised.  This is typically
225
243
        for commands intended for expert users.
226
 
 
227
 
    encoding_type
228
 
        Command objects will get a 'outf' attribute, which has been
229
 
        setup to properly handle encoding of unicode strings.
230
 
        encoding_type determines what will happen when characters cannot
231
 
        be encoded
232
 
            strict - abort if we cannot decode
233
 
            replace - put in a bogus character (typically '?')
234
 
            exact - do not encode sys.stdout
235
 
 
236
 
            NOTE: by default on Windows, sys.stdout is opened as a text
237
 
            stream, therefore LF line-endings are converted to CRLF.
238
 
            When a command uses encoding_type = 'exact', then
239
 
            sys.stdout is forced to be a binary stream, and line-endings
240
 
            will not mangled.
241
 
 
242
244
    """
243
245
    aliases = []
 
246
    
244
247
    takes_args = []
245
248
    takes_options = []
246
 
    encoding_type = 'strict'
247
249
 
248
250
    hidden = False
249
251
    
251
253
        """Construct an instance of this command."""
252
254
        if self.__doc__ == Command.__doc__:
253
255
            warn("No help message set for %r" % self)
254
 
        # List of standard options directly supported
255
 
        self.supported_std_options = []
256
 
 
257
 
    def _maybe_expand_globs(self, file_list):
258
 
        """Glob expand file_list if the platform does not do that itself.
259
 
        
260
 
        :return: A possibly empty list of unicode paths.
261
 
 
262
 
        Introduced in bzrlib 0.18.
263
 
        """
264
 
        if not file_list:
265
 
            file_list = []
266
 
        if sys.platform == 'win32':
267
 
            file_list = win32utils.glob_expand(file_list)
268
 
        return list(file_list)
269
 
 
270
 
    def _usage(self):
271
 
        """Return single-line grammar for this command.
272
 
 
273
 
        Only describes arguments, not options.
274
 
        """
275
 
        s = 'bzr ' + self.name() + ' '
276
 
        for aname in self.takes_args:
277
 
            aname = aname.upper()
278
 
            if aname[-1] in ['$', '+']:
279
 
                aname = aname[:-1] + '...'
280
 
            elif aname[-1] == '?':
281
 
                aname = '[' + aname[:-1] + ']'
282
 
            elif aname[-1] == '*':
283
 
                aname = '[' + aname[:-1] + '...]'
284
 
            s += aname + ' '
285
 
        s = s[:-1]      # remove last space
286
 
        return s
287
 
 
288
 
    def get_help_text(self, additional_see_also=None, plain=True,
289
 
                      see_also_as_links=False):
290
 
        """Return a text string with help for this command.
291
 
        
292
 
        :param additional_see_also: Additional help topics to be
293
 
            cross-referenced.
294
 
        :param plain: if False, raw help (reStructuredText) is
295
 
            returned instead of plain text.
296
 
        :param see_also_as_links: if True, convert items in 'See also'
297
 
            list to internal links (used by bzr_man rstx generator)
298
 
        """
299
 
        doc = self.help()
300
 
        if doc is None:
301
 
            raise NotImplementedError("sorry, no detailed help yet for %r" % self.name())
302
 
 
303
 
        # Extract the summary (purpose) and sections out from the text
304
 
        purpose,sections = self._get_help_parts(doc)
305
 
 
306
 
        # If a custom usage section was provided, use it
307
 
        if sections.has_key('Usage'):
308
 
            usage = sections.pop('Usage')
309
 
        else:
310
 
            usage = self._usage()
311
 
 
312
 
        # The header is the purpose and usage
313
 
        result = ""
314
 
        result += ':Purpose: %s\n' % purpose
315
 
        if usage.find('\n') >= 0:
316
 
            result += ':Usage:\n%s\n' % usage
317
 
        else:
318
 
            result += ':Usage:   %s\n' % usage
319
 
        result += '\n'
320
 
 
321
 
        # Add the options
322
 
        options = option.get_optparser(self.options()).format_option_help()
323
 
        if options.startswith('Options:'):
324
 
            result += ':' + options
325
 
        elif options.startswith('options:'):
326
 
            # Python 2.4 version of optparse
327
 
            result += ':Options:' + options[len('options:'):]
328
 
        else:
329
 
            result += options
330
 
        result += '\n'
331
 
 
332
 
        # Add the description, indenting it 2 spaces
333
 
        # to match the indentation of the options
334
 
        if sections.has_key(None):
335
 
            text = sections.pop(None)
336
 
            text = '\n  '.join(text.splitlines())
337
 
            result += ':%s:\n  %s\n\n' % ('Description',text)
338
 
 
339
 
        # Add the custom sections (e.g. Examples). Note that there's no need
340
 
        # to indent these as they must be indented already in the source.
341
 
        if sections:
342
 
            labels = sorted(sections.keys())
343
 
            for label in labels:
344
 
                result += ':%s:\n%s\n\n' % (label,sections[label])
345
 
 
346
 
        # Add the aliases, source (plug-in) and see also links, if any
347
 
        if self.aliases:
348
 
            result += ':Aliases:  '
349
 
            result += ', '.join(self.aliases) + '\n'
350
 
        plugin_name = self.plugin_name()
351
 
        if plugin_name is not None:
352
 
            result += ':From:     plugin "%s"\n' % plugin_name
353
 
        see_also = self.get_see_also(additional_see_also)
354
 
        if see_also:
355
 
            if not plain and see_also_as_links:
356
 
                see_also_links = []
357
 
                for item in see_also:
358
 
                    if item == 'topics':
359
 
                        # topics doesn't have an independent section
360
 
                        # so don't create a real link
361
 
                        see_also_links.append(item)
362
 
                    else:
363
 
                        # Use a reST link for this entry
364
 
                        see_also_links.append("`%s`_" % (item,))
365
 
                see_also = see_also_links
366
 
            result += ':See also: '
367
 
            result += ', '.join(see_also) + '\n'
368
 
 
369
 
        # If this will be rendered as plain text, convert it
370
 
        if plain:
371
 
            import bzrlib.help_topics
372
 
            result = bzrlib.help_topics.help_as_plain_text(result)
373
 
        return result
374
 
 
375
 
    @staticmethod
376
 
    def _get_help_parts(text):
377
 
        """Split help text into a summary and named sections.
378
 
 
379
 
        :return: (summary,sections) where summary is the top line and
380
 
            sections is a dictionary of the rest indexed by section name.
381
 
            A section starts with a heading line of the form ":xxx:".
382
 
            Indented text on following lines is the section value.
383
 
            All text found outside a named section is assigned to the
384
 
            default section which is given the key of None.
385
 
        """
386
 
        def save_section(sections, label, section):
387
 
            if len(section) > 0:
388
 
                if sections.has_key(label):
389
 
                    sections[label] += '\n' + section
390
 
                else:
391
 
                    sections[label] = section
392
 
 
393
 
        lines = text.rstrip().splitlines()
394
 
        summary = lines.pop(0)
395
 
        sections = {}
396
 
        label,section = None,''
397
 
        for line in lines:
398
 
            if line.startswith(':') and line.endswith(':') and len(line) > 2:
399
 
                save_section(sections, label, section)
400
 
                label,section = line[1:-1],''
401
 
            elif (label is not None) and len(line) > 1 and not line[0].isspace():
402
 
                save_section(sections, label, section)
403
 
                label,section = None,line
404
 
            else:
405
 
                if len(section) > 0:
406
 
                    section += '\n' + line
407
 
                else:
408
 
                    section = line
409
 
        save_section(sections, label, section)
410
 
        return summary, sections
411
 
 
412
 
    def get_help_topic(self):
413
 
        """Return the commands help topic - its name."""
414
 
        return self.name()
415
 
 
416
 
    def get_see_also(self, additional_terms=None):
417
 
        """Return a list of help topics that are related to this command.
418
 
        
419
 
        The list is derived from the content of the _see_also attribute. Any
420
 
        duplicates are removed and the result is in lexical order.
421
 
        :param additional_terms: Additional help topics to cross-reference.
422
 
        :return: A list of help topics.
423
 
        """
424
 
        see_also = set(getattr(self, '_see_also', []))
425
 
        if additional_terms:
426
 
            see_also.update(additional_terms)
427
 
        return sorted(see_also)
428
 
 
429
 
    def options(self):
430
 
        """Return dict of valid options for this command.
431
 
 
432
 
        Maps from long option name to option object."""
433
 
        r = Option.STD_OPTIONS.copy()
434
 
        std_names = r.keys()
435
 
        for o in self.takes_options:
436
 
            if isinstance(o, basestring):
437
 
                o = option.Option.OPTIONS[o]
438
 
            r[o.name] = o
439
 
            if o.name in std_names:
440
 
                self.supported_std_options.append(o.name)
441
 
        return r
442
 
 
443
 
    def _setup_outf(self):
444
 
        """Return a file linked to stdout, which has proper encoding."""
445
 
        # Originally I was using self.stdout, but that looks
446
 
        # *way* too much like sys.stdout
447
 
        if self.encoding_type == 'exact':
448
 
            # force sys.stdout to be binary stream on win32
449
 
            if sys.platform == 'win32':
450
 
                fileno = getattr(sys.stdout, 'fileno', None)
451
 
                if fileno:
452
 
                    import msvcrt
453
 
                    msvcrt.setmode(fileno(), os.O_BINARY)
454
 
            self.outf = sys.stdout
455
 
            return
456
 
 
457
 
        output_encoding = osutils.get_terminal_encoding()
458
 
 
459
 
        self.outf = codecs.getwriter(output_encoding)(sys.stdout,
460
 
                        errors=self.encoding_type)
461
 
        # For whatever reason codecs.getwriter() does not advertise its encoding
462
 
        # it just returns the encoding of the wrapped file, which is completely
463
 
        # bogus. So set the attribute, so we can find the correct encoding later.
464
 
        self.outf.encoding = output_encoding
465
 
 
466
 
    def run_argv_aliases(self, argv, alias_argv=None):
467
 
        """Parse the command line and run with extra aliases in alias_argv."""
468
 
        if argv is None:
469
 
            warn("Passing None for [] is deprecated from bzrlib 0.10",
470
 
                 DeprecationWarning, stacklevel=2)
471
 
            argv = []
472
 
        args, opts = parse_args(self, argv, alias_argv)
473
 
 
474
 
        # Process the standard options
 
256
 
 
257
 
 
258
    def run_argv(self, argv):
 
259
        """Parse command line and run."""
 
260
        args, opts = parse_args(argv)
 
261
 
475
262
        if 'help' in opts:  # e.g. bzr add --help
476
 
            sys.stdout.write(self.get_help_text())
 
263
            from bzrlib.help import help_on_command
 
264
            help_on_command(self.name())
477
265
            return 0
478
 
        trace.set_verbosity_level(option._verbosity_level)
479
 
        if 'verbose' in self.supported_std_options:
480
 
            opts['verbose'] = trace.is_verbose()
481
 
        elif opts.has_key('verbose'):
482
 
            del opts['verbose']
483
 
        if 'quiet' in self.supported_std_options:
484
 
            opts['quiet'] = trace.is_quiet()
485
 
        elif opts.has_key('quiet'):
486
 
            del opts['quiet']
 
266
 
 
267
        # check options are reasonable
 
268
        allowed = self.takes_options
 
269
        for oname in opts:
 
270
            if oname not in allowed:
 
271
                raise BzrCommandError("option '--%s' is not allowed for command %r"
 
272
                                      % (oname, self.name()))
487
273
 
488
274
        # mix arguments and options into one dictionary
489
275
        cmdargs = _match_argform(self.name(), self.takes_args, args)
494
280
        all_cmd_args = cmdargs.copy()
495
281
        all_cmd_args.update(cmdopts)
496
282
 
497
 
        self._setup_outf()
498
 
 
499
283
        return self.run(**all_cmd_args)
500
284
 
 
285
    
501
286
    def run(self):
502
287
        """Actually run the command.
503
288
 
508
293
        shell error code if not.  It's OK for this method to allow
509
294
        an exception to raise up.
510
295
        """
511
 
        raise NotImplementedError('no implementation of command %r'
512
 
                                  % self.name())
 
296
        raise NotImplementedError()
 
297
 
513
298
 
514
299
    def help(self):
515
300
        """Return help message for this class."""
516
 
        from inspect import getdoc
517
301
        if self.__doc__ is Command.__doc__:
518
302
            return None
519
303
        return getdoc(self)
521
305
    def name(self):
522
306
        return _unsquish_command_name(self.__class__.__name__)
523
307
 
524
 
    def plugin_name(self):
525
 
        """Get the name of the plugin that provides this command.
526
308
 
527
 
        :return: The name of the plugin or None if the command is builtin.
528
 
        """
529
 
        mod_parts = self.__module__.split('.')
530
 
        if len(mod_parts) >= 3 and mod_parts[1] == 'plugins':
531
 
            return mod_parts[2]
 
309
def parse_spec(spec):
 
310
    """
 
311
    >>> parse_spec(None)
 
312
    [None, None]
 
313
    >>> parse_spec("./")
 
314
    ['./', None]
 
315
    >>> parse_spec("../@")
 
316
    ['..', -1]
 
317
    >>> parse_spec("../f/@35")
 
318
    ['../f', 35]
 
319
    >>> parse_spec('./@revid:john@arbash-meinel.com-20050711044610-3ca0327c6a222f67')
 
320
    ['.', 'revid:john@arbash-meinel.com-20050711044610-3ca0327c6a222f67']
 
321
    """
 
322
    if spec is None:
 
323
        return [None, None]
 
324
    if '/@' in spec:
 
325
        parsed = spec.split('/@')
 
326
        assert len(parsed) == 2
 
327
        if parsed[1] == "":
 
328
            parsed[1] = -1
532
329
        else:
533
 
            return None
534
 
 
535
 
 
536
 
def parse_args(command, argv, alias_argv=None):
 
330
            try:
 
331
                parsed[1] = int(parsed[1])
 
332
            except ValueError:
 
333
                pass # We can allow stuff like ./@revid:blahblahblah
 
334
            else:
 
335
                assert parsed[1] >=0
 
336
    else:
 
337
        parsed = [spec, None]
 
338
    return parsed
 
339
 
 
340
 
 
341
# list of all available options; the rhs can be either None for an
 
342
# option that takes no argument, or a constructor function that checks
 
343
# the type.
 
344
OPTIONS = {
 
345
    'all':                    None,
 
346
    'basis':                  str,
 
347
    'diff-options':           str,
 
348
    'help':                   None,
 
349
    'file':                   unicode,
 
350
    'force':                  None,
 
351
    'format':                 unicode,
 
352
    'forward':                None,
 
353
    'message':                unicode,
 
354
    'no-recurse':             None,
 
355
    'profile':                None,
 
356
    'revision':               _parse_revision_str,
 
357
    'short':                  None,
 
358
    'show-ids':               None,
 
359
    'timezone':               str,
 
360
    'verbose':                None,
 
361
    'version':                None,
 
362
    'email':                  None,
 
363
    'unchanged':              None,
 
364
    'update':                 None,
 
365
    'long':                   None,
 
366
    'root':                   str,
 
367
    'no-backup':              None,
 
368
    'pattern':                str,
 
369
    'remember':               None,
 
370
    }
 
371
 
 
372
SHORT_OPTIONS = {
 
373
    'F':                      'file', 
 
374
    'h':                      'help',
 
375
    'm':                      'message',
 
376
    'r':                      'revision',
 
377
    'v':                      'verbose',
 
378
    'l':                      'long',
 
379
}
 
380
 
 
381
 
 
382
def parse_args(argv):
537
383
    """Parse command line.
538
384
    
539
385
    Arguments and options are parsed at this level before being passed
540
386
    down to specific command handlers.  This routine knows, from a
541
387
    lookup table, something about the available options, what optargs
542
388
    they take, and which commands will accept them.
 
389
 
 
390
    >>> parse_args('--help'.split())
 
391
    ([], {'help': True})
 
392
    >>> parse_args('help -- --invalidcmd'.split())
 
393
    (['help', '--invalidcmd'], {})
 
394
    >>> parse_args('--version'.split())
 
395
    ([], {'version': True})
 
396
    >>> parse_args('status --all'.split())
 
397
    (['status'], {'all': True})
 
398
    >>> parse_args('commit --message=biter'.split())
 
399
    (['commit'], {'message': u'biter'})
 
400
    >>> parse_args('log -r 500'.split())
 
401
    (['log'], {'revision': [<RevisionSpec_int 500>]})
 
402
    >>> parse_args('log -r500..600'.split())
 
403
    (['log'], {'revision': [<RevisionSpec_int 500>, <RevisionSpec_int 600>]})
 
404
    >>> parse_args('log -vr500..600'.split())
 
405
    (['log'], {'verbose': True, 'revision': [<RevisionSpec_int 500>, <RevisionSpec_int 600>]})
 
406
    >>> parse_args('log -rrevno:500..600'.split()) #the r takes an argument
 
407
    (['log'], {'revision': [<RevisionSpec_revno revno:500>, <RevisionSpec_int 600>]})
543
408
    """
544
 
    # TODO: make it a method of the Command?
545
 
    parser = option.get_optparser(command.options())
546
 
    if alias_argv is not None:
547
 
        args = alias_argv + argv
548
 
    else:
549
 
        args = argv
550
 
 
551
 
    options, args = parser.parse_args(args)
552
 
    opts = dict([(k, v) for k, v in options.__dict__.iteritems() if
553
 
                 v is not option.OptionParser.DEFAULT_VALUE])
 
409
    args = []
 
410
    opts = {}
 
411
 
 
412
    argsover = False
 
413
    while argv:
 
414
        a = argv.pop(0)
 
415
        if not argsover and a[0] == '-':
 
416
            # option names must not be unicode
 
417
            a = str(a)
 
418
            optarg = None
 
419
            if a[1] == '-':
 
420
                if a == '--':
 
421
                    # We've received a standalone -- No more flags
 
422
                    argsover = True
 
423
                    continue
 
424
                mutter("  got option %r" % a)
 
425
                if '=' in a:
 
426
                    optname, optarg = a[2:].split('=', 1)
 
427
                else:
 
428
                    optname = a[2:]
 
429
                if optname not in OPTIONS:
 
430
                    raise BzrError('unknown long option %r' % a)
 
431
            else:
 
432
                shortopt = a[1:]
 
433
                if shortopt in SHORT_OPTIONS:
 
434
                    # Multi-character options must have a space to delimit
 
435
                    # their value
 
436
                    optname = SHORT_OPTIONS[shortopt]
 
437
                else:
 
438
                    # Single character short options, can be chained,
 
439
                    # and have their value appended to their name
 
440
                    shortopt = a[1:2]
 
441
                    if shortopt not in SHORT_OPTIONS:
 
442
                        # We didn't find the multi-character name, and we
 
443
                        # didn't find the single char name
 
444
                        raise BzrError('unknown short option %r' % a)
 
445
                    optname = SHORT_OPTIONS[shortopt]
 
446
 
 
447
                    if a[2:]:
 
448
                        # There are extra things on this option
 
449
                        # see if it is the value, or if it is another
 
450
                        # short option
 
451
                        optargfn = OPTIONS[optname]
 
452
                        if optargfn is None:
 
453
                            # This option does not take an argument, so the
 
454
                            # next entry is another short option, pack it back
 
455
                            # into the list
 
456
                            argv.insert(0, '-' + a[2:])
 
457
                        else:
 
458
                            # This option takes an argument, so pack it
 
459
                            # into the array
 
460
                            optarg = a[2:]
 
461
            
 
462
            if optname in opts:
 
463
                # XXX: Do we ever want to support this, e.g. for -r?
 
464
                raise BzrError('repeated option %r' % a)
 
465
                
 
466
            optargfn = OPTIONS[optname]
 
467
            if optargfn:
 
468
                if optarg == None:
 
469
                    if not argv:
 
470
                        raise BzrError('option %r needs an argument' % a)
 
471
                    else:
 
472
                        optarg = argv.pop(0)
 
473
                opts[optname] = optargfn(optarg)
 
474
            else:
 
475
                if optarg != None:
 
476
                    raise BzrError('option %r takes no argument' % optname)
 
477
                opts[optname] = True
 
478
        else:
 
479
            args.append(a)
 
480
 
554
481
    return args, opts
555
482
 
556
483
 
 
484
 
 
485
 
557
486
def _match_argform(cmd, takes_args, args):
558
487
    argdict = {}
559
488
 
571
500
                argdict[argname + '_list'] = None
572
501
        elif ap[-1] == '+':
573
502
            if not args:
574
 
                raise errors.BzrCommandError("command %r needs one or more %s"
575
 
                                             % (cmd, argname.upper()))
 
503
                raise BzrCommandError("command %r needs one or more %s"
 
504
                        % (cmd, argname.upper()))
576
505
            else:
577
506
                argdict[argname + '_list'] = args[:]
578
507
                args = []
579
508
        elif ap[-1] == '$': # all but one
580
509
            if len(args) < 2:
581
 
                raise errors.BzrCommandError("command %r needs one or more %s"
582
 
                                             % (cmd, argname.upper()))
 
510
                raise BzrCommandError("command %r needs one or more %s"
 
511
                        % (cmd, argname.upper()))
583
512
            argdict[argname + '_list'] = args[:-1]
584
 
            args[:-1] = []
 
513
            args[:-1] = []                
585
514
        else:
586
515
            # just a plain arg
587
516
            argname = ap
588
517
            if not args:
589
 
                raise errors.BzrCommandError("command %r requires argument %s"
590
 
                               % (cmd, argname.upper()))
 
518
                raise BzrCommandError("command %r requires argument %s"
 
519
                        % (cmd, argname.upper()))
591
520
            else:
592
521
                argdict[argname] = args.pop(0)
593
522
            
594
523
    if args:
595
 
        raise errors.BzrCommandError("extra argument to command %s: %s"
596
 
                                     % (cmd, args[0]))
 
524
        raise BzrCommandError("extra argument to command %s: %s"
 
525
                              % (cmd, args[0]))
597
526
 
598
527
    return argdict
599
528
 
600
 
def apply_coveraged(dirname, the_callable, *args, **kwargs):
601
 
    # Cannot use "import trace", as that would import bzrlib.trace instead of
602
 
    # the standard library's trace.
603
 
    trace = __import__('trace')
604
 
 
605
 
    tracer = trace.Trace(count=1, trace=0)
606
 
    sys.settrace(tracer.globaltrace)
607
 
 
608
 
    ret = the_callable(*args, **kwargs)
609
 
 
610
 
    sys.settrace(None)
611
 
    results = tracer.results()
612
 
    results.write_results(show_missing=1, summary=False,
613
 
                          coverdir=dirname)
614
529
 
615
530
 
616
531
def apply_profiled(the_callable, *args, **kwargs):
636
551
        os.remove(pfname)
637
552
 
638
553
 
639
 
def apply_lsprofiled(filename, the_callable, *args, **kwargs):
640
 
    from bzrlib.lsprof import profile
641
 
    ret, stats = profile(the_callable, *args, **kwargs)
642
 
    stats.sort()
643
 
    if filename is None:
644
 
        stats.pprint()
645
 
    else:
646
 
        stats.save(filename)
647
 
        trace.note('Profile data written to "%s".', filename)
648
 
    return ret
649
 
 
650
 
 
651
 
def shlex_split_unicode(unsplit):
652
 
    import shlex
653
 
    return [u.decode('utf-8') for u in shlex.split(unsplit.encode('utf-8'))]
654
 
 
655
 
 
656
 
def get_alias(cmd, config=None):
657
 
    """Return an expanded alias, or None if no alias exists.
658
 
 
659
 
    cmd
660
 
        Command to be checked for an alias.
661
 
    config
662
 
        Used to specify an alternative config to use,
663
 
        which is especially useful for testing.
664
 
        If it is unspecified, the global config will be used.
665
 
    """
666
 
    if config is None:
667
 
        import bzrlib.config
668
 
        config = bzrlib.config.GlobalConfig()
669
 
    alias = config.get_alias(cmd)
670
 
    if (alias):
671
 
        return shlex_split_unicode(alias)
672
 
    return None
673
 
 
674
 
 
675
554
def run_bzr(argv):
676
555
    """Execute a command.
677
556
 
680
559
    
681
560
    argv
682
561
       The command-line arguments, without the program name from argv[0]
683
 
       These should already be decoded. All library/test code calling
684
 
       run_bzr should be passing valid strings (don't need decoding).
685
562
    
686
563
    Returns a command status or raises an exception.
687
564
 
691
568
    --no-plugins
692
569
        Do not load plugin modules at all
693
570
 
694
 
    --no-aliases
695
 
        Do not allow aliases
696
 
 
697
571
    --builtin
698
572
        Only use builtin commands.  (Plugins are still allowed to change
699
573
        other behaviour.)
700
574
 
701
575
    --profile
702
 
        Run under the Python hotshot profiler.
703
 
 
704
 
    --lsprof
705
 
        Run under the Python lsprof profiler.
706
 
 
707
 
    --coverage
708
 
        Generate line coverage report in the specified directory.
 
576
        Run under the Python profiler.
709
577
    """
710
 
    argv = list(argv)
711
 
    trace.mutter("bzr arguments: %r", argv)
 
578
    # Load all of the transport methods
 
579
    import bzrlib.transport.local, bzrlib.transport.http
 
580
    
 
581
    argv = [a.decode(bzrlib.user_encoding) for a in argv]
712
582
 
713
 
    opt_lsprof = opt_profile = opt_no_plugins = opt_builtin =  \
714
 
                opt_no_aliases = False
715
 
    opt_lsprof_file = opt_coverage_dir = None
 
583
    opt_profile = opt_no_plugins = opt_builtin = False
716
584
 
717
585
    # --no-plugins is handled specially at a very early stage. We need
718
586
    # to load plugins before doing other command parsing so that they
719
587
    # can override commands, but this needs to happen first.
720
588
 
721
 
    argv_copy = []
722
 
    i = 0
723
 
    while i < len(argv):
724
 
        a = argv[i]
 
589
    for a in argv:
725
590
        if a == '--profile':
726
591
            opt_profile = True
727
 
        elif a == '--lsprof':
728
 
            opt_lsprof = True
729
 
        elif a == '--lsprof-file':
730
 
            opt_lsprof = True
731
 
            opt_lsprof_file = argv[i + 1]
732
 
            i += 1
733
592
        elif a == '--no-plugins':
734
593
            opt_no_plugins = True
735
 
        elif a == '--no-aliases':
736
 
            opt_no_aliases = True
737
594
        elif a == '--builtin':
738
595
            opt_builtin = True
739
 
        elif a == '--coverage':
740
 
            opt_coverage_dir = argv[i + 1]
741
 
            i += 1
742
 
        elif a.startswith('-D'):
743
 
            debug.debug_flags.add(a[2:])
744
596
        else:
745
 
            argv_copy.append(a)
746
 
        i += 1
 
597
            break
 
598
        argv.remove(a)
747
599
 
748
 
    argv = argv_copy
749
 
    if (not argv):
750
 
        from bzrlib.builtins import cmd_help
751
 
        cmd_help().run_argv_aliases([])
 
600
    if (not argv) or (argv[0] == '--help'):
 
601
        from bzrlib.help import help
 
602
        if len(argv) > 1:
 
603
            help(argv[1])
 
604
        else:
 
605
            help()
752
606
        return 0
753
607
 
754
608
    if argv[0] == '--version':
755
 
        from bzrlib.builtins import cmd_version
756
 
        cmd_version().run_argv_aliases([])
 
609
        from bzrlib.builtins import show_version
 
610
        show_version()
757
611
        return 0
758
612
        
759
613
    if not opt_no_plugins:
760
614
        from bzrlib.plugin import load_plugins
761
615
        load_plugins()
762
 
    else:
763
 
        from bzrlib.plugin import disable_plugins
764
 
        disable_plugins()
765
 
 
766
 
    alias_argv = None
767
 
 
768
 
    if not opt_no_aliases:
769
 
        alias_argv = get_alias(argv[0])
770
 
        if alias_argv:
771
 
            alias_argv = [a.decode(bzrlib.user_encoding) for a in alias_argv]
772
 
            argv[0] = alias_argv.pop(0)
773
 
 
774
 
    cmd = argv.pop(0)
775
 
    # We want only 'ascii' command names, but the user may have typed
776
 
    # in a Unicode name. In that case, they should just get a
777
 
    # 'command not found' error later.
 
616
 
 
617
    cmd = str(argv.pop(0))
778
618
 
779
619
    cmd_obj = get_cmd_object(cmd, plugins_override=not opt_builtin)
780
 
    run = cmd_obj.run_argv_aliases
781
 
    run_argv = [argv, alias_argv]
782
 
 
783
 
    try:
784
 
        if opt_lsprof:
785
 
            if opt_coverage_dir:
786
 
                trace.warning(
787
 
                    '--coverage ignored, because --lsprof is in use.')
788
 
            ret = apply_lsprofiled(opt_lsprof_file, run, *run_argv)
789
 
        elif opt_profile:
790
 
            if opt_coverage_dir:
791
 
                trace.warning(
792
 
                    '--coverage ignored, because --profile is in use.')
793
 
            ret = apply_profiled(run, *run_argv)
794
 
        elif opt_coverage_dir:
795
 
            ret = apply_coveraged(opt_coverage_dir, run, *run_argv)
796
 
        else:
797
 
            ret = run(*run_argv)
798
 
        if 'memory' in debug.debug_flags:
799
 
            try:
800
 
                status_file = file('/proc/%s/status' % os.getpid(), 'rb')
801
 
            except IOError:
802
 
                pass
803
 
            else:
804
 
                status = status_file.read()
805
 
                status_file.close()
806
 
                trace.note("Process status after command:")
807
 
                for line in status.splitlines():
808
 
                    trace.note(line)
809
 
        return ret or 0
810
 
    finally:
811
 
        # reset, in case we may do other commands later within the same process
812
 
        option._verbosity_level = 0
813
 
 
814
 
def display_command(func):
815
 
    """Decorator that suppresses pipe/interrupt errors."""
816
 
    def ignore_pipe(*args, **kwargs):
817
 
        try:
818
 
            result = func(*args, **kwargs)
819
 
            sys.stdout.flush()
820
 
            return result
821
 
        except IOError, e:
822
 
            if getattr(e, 'errno', None) is None:
823
 
                raise
824
 
            if e.errno != errno.EPIPE:
825
 
                # Win32 raises IOError with errno=0 on a broken pipe
826
 
                if sys.platform != 'win32' or (e.errno not in (0, errno.EINVAL)):
827
 
                    raise
828
 
            pass
829
 
        except KeyboardInterrupt:
830
 
            pass
831
 
    return ignore_pipe
 
620
 
 
621
    if opt_profile:
 
622
        ret = apply_profiled(cmd_obj.run_argv, argv)
 
623
    else:
 
624
        ret = cmd_obj.run_argv(argv)
 
625
    return ret or 0
832
626
 
833
627
 
834
628
def main(argv):
835
629
    import bzrlib.ui
836
 
    from bzrlib.ui.text import TextUIFactory
837
 
    bzrlib.ui.ui_factory = TextUIFactory()
838
 
     
839
 
    # Is this a final release version? If so, we should suppress warnings
840
 
    if bzrlib.version_info[3] == 'final':
841
 
        from bzrlib import symbol_versioning
842
 
        symbol_versioning.suppress_deprecation_warnings(override=False)
843
 
    try:
844
 
        argv = [a.decode(bzrlib.user_encoding) for a in argv[1:]]
845
 
    except UnicodeDecodeError:
846
 
        raise errors.BzrError(("Parameter '%r' is unsupported by the current "
847
 
                                                            "encoding." % a))
848
 
    ret = run_bzr_catch_errors(argv)
849
 
    trace.mutter("return code %d", ret)
850
 
    return ret
 
630
    bzrlib.trace.log_startup(argv)
 
631
    bzrlib.ui.ui_factory = bzrlib.ui.TextUIFactory()
 
632
 
 
633
    return run_bzr_catch_errors(argv[1:])
851
634
 
852
635
 
853
636
def run_bzr_catch_errors(argv):
854
 
    # Note: The except clause logic below should be kept in sync with the
855
 
    # profile() routine in lsprof.py.
856
 
    try:
857
 
        return run_bzr(argv)
858
 
    except (KeyboardInterrupt, Exception), e:
859
 
        # used to handle AssertionError and KeyboardInterrupt
860
 
        # specially here, but hopefully they're handled ok by the logger now
861
 
        exitcode = trace.report_exception(sys.exc_info(), sys.stderr)
862
 
        if os.environ.get('BZR_PDB'):
863
 
            print '**** entering debugger'
864
 
            import pdb
865
 
            pdb.post_mortem(sys.exc_traceback)
866
 
        return exitcode
867
 
 
868
 
 
869
 
def run_bzr_catch_user_errors(argv):
870
 
    """Run bzr and report user errors, but let internal errors propagate.
871
 
 
872
 
    This is used for the test suite, and might be useful for other programs
873
 
    that want to wrap the commandline interface.
874
 
    """
875
 
    try:
876
 
        return run_bzr(argv)
 
637
    try:
 
638
        try:
 
639
            return run_bzr(argv)
 
640
        finally:
 
641
            # do this here inside the exception wrappers to catch EPIPE
 
642
            sys.stdout.flush()
 
643
    except BzrCommandError, e:
 
644
        # command line syntax error, etc
 
645
        log_error(str(e))
 
646
        return 1
 
647
    except BzrError, e:
 
648
        bzrlib.trace.log_exception()
 
649
        return 1
 
650
    except AssertionError, e:
 
651
        bzrlib.trace.log_exception('assertion failed: ' + str(e))
 
652
        return 3
 
653
    except KeyboardInterrupt, e:
 
654
        bzrlib.trace.log_exception('interrupted')
 
655
        return 2
877
656
    except Exception, e:
878
 
        if (isinstance(e, (OSError, IOError))
879
 
            or not getattr(e, 'internal_error', True)):
880
 
            trace.report_exception(sys.exc_info(), sys.stderr)
881
 
            return 3
882
 
        else:
883
 
            raise
884
 
 
885
 
 
886
 
class HelpCommandIndex(object):
887
 
    """A index for bzr help that returns commands."""
888
 
 
889
 
    def __init__(self):
890
 
        self.prefix = 'commands/'
891
 
 
892
 
    def get_topics(self, topic):
893
 
        """Search for topic amongst commands.
894
 
 
895
 
        :param topic: A topic to search for.
896
 
        :return: A list which is either empty or contains a single
897
 
            Command entry.
898
 
        """
899
 
        if topic and topic.startswith(self.prefix):
900
 
            topic = topic[len(self.prefix):]
901
 
        try:
902
 
            cmd = _get_cmd_object(topic)
903
 
        except KeyError:
904
 
            return []
905
 
        else:
906
 
            return [cmd]
907
 
 
908
 
 
909
 
class Provider(object):
910
 
    '''Generic class to be overriden by plugins'''
911
 
 
912
 
    def plugin_for_command(self, cmd_name):
913
 
        '''Takes a command and returns the information for that plugin
914
 
        
915
 
        :return: A dictionary with all the available information 
916
 
        for the requested plugin
917
 
        '''
918
 
        raise NotImplementedError
919
 
 
920
 
 
921
 
class ProvidersRegistry(registry.Registry):
922
 
    '''This registry exists to allow other providers to exist'''
923
 
 
924
 
    def __iter__(self):
925
 
        for key, provider in self.iteritems():
926
 
            yield provider
927
 
 
928
 
command_providers_registry = ProvidersRegistry()
929
 
 
 
657
        import errno
 
658
        if (isinstance(e, IOError) 
 
659
            and hasattr(e, 'errno')
 
660
            and e.errno == errno.EPIPE):
 
661
            bzrlib.trace.note('broken pipe')
 
662
            return 2
 
663
        else:
 
664
            ## import pdb
 
665
            ## pdb.pm()
 
666
            bzrlib.trace.log_exception()
 
667
            return 2
930
668
 
931
669
if __name__ == '__main__':
932
670
    sys.exit(main(sys.argv))