~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/commands.py

  • Committer: Andrew Bennetts
  • Date: 2011-02-03 01:02:16 UTC
  • mto: This revision was merged to the branch mainline in revision 5641.
  • Revision ID: andrew.bennetts@canonical.com-20110203010216-yi9stzobqcj2wivf
Update doc index links to What's New in 2.4 (rather than 2.3).

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
#! /usr/bin/python
2
 
 
3
 
 
4
 
# Copyright (C) 2004, 2005 by Martin Pool
5
 
# Copyright (C) 2005 by Canonical Ltd
6
 
 
7
 
 
 
1
# Copyright (C) 2005-2010 Canonical Ltd
 
2
#
8
3
# This program is free software; you can redistribute it and/or modify
9
4
# it under the terms of the GNU General Public License as published by
10
5
# the Free Software Foundation; either version 2 of the License, or
11
6
# (at your option) any later version.
12
 
 
 
7
#
13
8
# This program is distributed in the hope that it will be useful,
14
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
15
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16
11
# GNU General Public License for more details.
17
 
 
 
12
#
18
13
# You should have received a copy of the GNU General Public License
19
14
# along with this program; if not, write to the Free Software
20
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21
 
 
22
 
"""Bazaar-NG -- a free distributed version-control tool
23
 
 
24
 
**WARNING: THIS IS AN UNSTABLE DEVELOPMENT VERSION**
25
 
 
26
 
Current limitation include:
27
 
 
28
 
* Metadata format is not stable yet -- you may need to
29
 
  discard history in the future.
30
 
 
31
 
* No handling of subdirectories, symlinks or any non-text files.
32
 
 
33
 
* Insufficient error handling.
34
 
 
35
 
* Many commands unimplemented or partially implemented.
36
 
 
37
 
* Space-inefficient storage.
38
 
 
39
 
* No merge operators yet.
40
 
 
41
 
Interesting commands::
42
 
 
43
 
  bzr help
44
 
       Show summary help screen
45
 
  bzr version
46
 
       Show software version/licence/non-warranty.
47
 
  bzr init
48
 
       Start versioning the current directory
49
 
  bzr add FILE...
50
 
       Make files versioned.
51
 
  bzr log
52
 
       Show revision history.
53
 
  bzr diff
54
 
       Show changes from last revision to working copy.
55
 
  bzr commit -m 'MESSAGE'
56
 
       Store current state as new revision.
57
 
  bzr export REVNO DESTINATION
58
 
       Export the branch state at a previous version.
59
 
  bzr status
60
 
       Show summary of pending changes.
61
 
  bzr remove FILE...
62
 
       Make a file not versioned.
63
 
"""
64
 
 
65
 
# not currently working:
66
 
#  bzr info
67
 
#       Show some information about this branch.
68
 
 
69
 
 
70
 
 
71
 
__copyright__ = "Copyright 2005 Canonical Development Ltd."
72
 
__author__ = "Martin Pool <mbp@canonical.com>"
73
 
__docformat__ = "restructuredtext en"
74
 
__version__ = '0.0.0'
75
 
 
76
 
 
77
 
import sys, os, random, time, sha, sets, types, re, shutil, tempfile
78
 
import traceback, socket, fnmatch, difflib
79
 
from os import path
80
 
from sets import Set
81
 
from pprint import pprint
82
 
from stat import *
83
 
from glob import glob
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
 
 
18
# TODO: Define arguments by objects, rather than just using names.
 
19
# Those objects can specify the expected type of the argument, which
 
20
# would help with validation and shell completion.  They could also provide
 
21
# help/explanation for that argument in a structured way.
 
22
 
 
23
# TODO: Specific "examples" property on commands for consistent formatting.
 
24
 
 
25
import os
 
26
import sys
 
27
 
 
28
from bzrlib.lazy_import import lazy_import
 
29
lazy_import(globals(), """
 
30
import codecs
 
31
import errno
 
32
import threading
 
33
from warnings import warn
84
34
 
85
35
import bzrlib
86
 
from bzrlib.store import ImmutableStore
87
 
from bzrlib.trace import mutter, note, log_error
88
 
from bzrlib.errors import bailout, BzrError
89
 
from bzrlib.osutils import quotefn, pumpfile, isdir, isfile
90
 
from bzrlib.tree import RevisionTree, EmptyTree, WorkingTree, Tree
91
 
from bzrlib.revision import Revision
92
 
from bzrlib import Branch, Inventory, InventoryEntry, ScratchBranch, BZRDIR, \
93
 
     format_date
94
 
 
95
 
BZR_DIFF_FORMAT = "## Bazaar-NG diff, format 0 ##\n"
96
 
BZR_PATCHNAME_FORMAT = 'cset:sha1:%s'
97
 
 
98
 
## standard representation
99
 
NONE_STRING = '(none)'
100
 
EMPTY = 'empty'
101
 
 
102
 
 
103
 
## TODO: Perhaps a different version of inventory commands that
104
 
## returns iterators...
105
 
 
106
 
## TODO: Perhaps an AtomicFile class that writes to a temporary file and then renames.
107
 
 
108
 
## TODO: Some kind of locking on branches.  Perhaps there should be a
109
 
## parameter to the branch object saying whether we want a read or
110
 
## write lock; release it from destructor.  Perhaps don't even need a
111
 
## read lock to look at immutable objects?
112
 
 
113
 
## TODO: Perhaps make UUIDs predictable in test mode to make it easier
114
 
## to compare output?
115
 
 
116
 
## TODO: Some kind of global code to generate the right Branch object
117
 
## to work on.  Almost, but not quite all, commands need one, and it
118
 
## can be taken either from their parameters or their working
119
 
## directory.
120
 
 
121
 
 
122
 
 
123
 
def cmd_status(all=False):
124
 
    """Display status summary.
125
 
 
126
 
    For each file there is a single line giving its file state and name.
127
 
    The name is that in the current revision unless it is deleted or
128
 
    missing, in which case the old name is shown.
129
 
 
130
 
    :todo: Don't show unchanged files unless ``--all`` is given?
131
 
    """
132
 
    Branch('.').show_status(show_all=all)
133
 
 
134
 
 
135
 
 
136
 
######################################################################
137
 
# examining history
138
 
def cmd_get_revision(revision_id):
139
 
    Branch('.').get_revision(revision_id).write_xml(sys.stdout)
140
 
 
141
 
 
142
 
def cmd_get_file_text(text_id):
143
 
    """Get contents of a file by hash."""
144
 
    sf = Branch('.').text_store[text_id]
145
 
    pumpfile(sf, sys.stdout)
146
 
 
147
 
 
148
 
 
149
 
######################################################################
150
 
# commands
151
 
    
152
 
 
153
 
def cmd_revno():
154
 
    """Show number of revisions on this branch"""
155
 
    print Branch('.').revno()
156
 
    
157
 
 
158
 
def cmd_add(file_list, verbose=False):
159
 
    """Add specified files.
160
 
    
161
 
    Fails if the files are already added.
162
 
    """
163
 
    Branch('.').add(file_list, verbose=verbose)
164
 
 
165
 
 
166
 
def cmd_inventory(revision=None):
167
 
    """Show inventory of the current working copy."""
168
 
    ## TODO: Also optionally show a previous inventory
169
 
    ## TODO: Format options
170
 
    b = Branch('.')
171
 
    if revision == None:
172
 
        inv = b.read_working_inventory()
173
 
    else:
174
 
        inv = b.get_revision_inventory(b.lookup_revision(revision))
175
 
        
176
 
    for path, entry in inv.iter_entries():
177
 
        print '%-50s %s' % (entry.file_id, path)
178
 
 
179
 
 
180
 
 
181
 
def cmd_info():
182
 
    b = Branch('.')
183
 
    print 'branch format:', b.controlfile('branch-format', 'r').readline().rstrip('\n')
184
 
 
185
 
    def plural(n, base='', pl=None):
186
 
        if n == 1:
187
 
            return base
188
 
        elif pl is not None:
189
 
            return pl
190
 
        else:
191
 
            return 's'
192
 
 
193
 
    count_version_dirs = 0
194
 
 
195
 
    count_status = {'A': 0, 'D': 0, 'M': 0, 'R': 0, '?': 0, 'I': 0, '.': 0}
196
 
    for st_tup in bzrlib.diff_trees(b.basis_tree(), b.working_tree()):
197
 
        fs = st_tup[0]
198
 
        count_status[fs] += 1
199
 
        if fs not in ['I', '?'] and st_tup[4] == 'directory':
200
 
            count_version_dirs += 1
201
 
 
202
 
    print
203
 
    print 'in the working tree:'
204
 
    for name, fs in (('unchanged', '.'),
205
 
                     ('modified', 'M'), ('added', 'A'), ('removed', 'D'),
206
 
                     ('renamed', 'R'), ('unknown', '?'), ('ignored', 'I'),
207
 
                     ):
208
 
        print '  %5d %s' % (count_status[fs], name)
209
 
    print '  %5d versioned subdirector%s' % (count_version_dirs,
210
 
                                             plural(count_version_dirs, 'y', 'ies'))
211
 
 
212
 
    print
213
 
    print 'branch history:'
214
 
    history = b.revision_history()
215
 
    revno = len(history)
216
 
    print '  %5d revision%s' % (revno, plural(revno))
217
 
    committers = Set()
218
 
    for rev in history:
219
 
        committers.add(b.get_revision(rev).committer)
220
 
    print '  %5d committer%s' % (len(committers), plural(len(committers)))
221
 
    if revno > 0:
222
 
        firstrev = b.get_revision(history[0])
223
 
        age = int((time.time() - firstrev.timestamp) / 3600 / 24)
224
 
        print '  %5d day%s old' % (age, plural(age))
225
 
        print '  first revision: %s' % format_date(firstrev.timestamp,
226
 
                                                 firstrev.timezone)
227
 
 
228
 
        lastrev = b.get_revision(history[-1])
229
 
        print '  latest revision: %s' % format_date(lastrev.timestamp,
230
 
                                                    lastrev.timezone)
231
 
        
232
 
    
233
 
 
234
 
 
235
 
def cmd_remove(file_list, verbose=False):
236
 
    Branch('.').remove(file_list, verbose=verbose)
237
 
 
238
 
 
239
 
 
240
 
def cmd_file_id(filename):
241
 
    i = Branch('.').read_working_inventory().path2id(filename)
242
 
    if i is None:
243
 
        bailout("%s is not a versioned file" % filename)
244
 
    else:
245
 
        print i
246
 
 
247
 
 
248
 
def cmd_find_filename(fileid):
249
 
    n = find_filename(fileid)
250
 
    if n is None:
251
 
        bailout("%s is not a live file id" % fileid)
252
 
    else:
253
 
        print n
254
 
 
255
 
 
256
 
def cmd_revision_history():
257
 
    for patchid in Branch('.').revision_history():
258
 
        print patchid
259
 
 
260
 
 
261
 
 
262
 
def cmd_init():
263
 
    # TODO: Check we're not already in a working directory?  At the
264
 
    # moment you'll get an ugly error.
265
 
    
266
 
    # TODO: What if we're in a subdirectory of a branch?  Would like
267
 
    # to allow that, but then the parent may need to understand that
268
 
    # the children have disappeared, or should they be versioned in
269
 
    # both?
270
 
 
271
 
    # TODO: Take an argument/option for branch name.
272
 
    Branch('.', init=True)
273
 
 
274
 
 
275
 
def cmd_diff(revision=None):
276
 
    """Show diff from basis to working copy.
277
 
 
278
 
    :todo: Take one or two revision arguments, look up those trees,
279
 
           and diff them.
280
 
 
281
 
    :todo: Allow diff across branches.
282
 
 
283
 
    :todo: Mangle filenames in diff to be more relevant.
284
 
 
285
 
    :todo: Shouldn't be in the cmd function.
286
 
    """
287
 
 
288
 
    b = Branch('.')
289
 
 
290
 
    if revision == None:
291
 
        old_tree = b.basis_tree()
292
 
    else:
293
 
        old_tree = b.revision_tree(b.lookup_revision(revision))
294
 
        
295
 
    new_tree = b.working_tree()
296
 
    old_inv = old_tree.inventory
297
 
    new_inv = new_tree.inventory
298
 
 
299
 
    # TODO: Options to control putting on a prefix or suffix, perhaps as a format string
300
 
    old_label = ''
301
 
    new_label = ''
302
 
 
303
 
    DEVNULL = '/dev/null'
304
 
    # Windows users, don't panic about this filename -- it is a
305
 
    # special signal to GNU patch that the file should be created or
306
 
    # deleted respectively.
307
 
 
308
 
    # TODO: Generation of pseudo-diffs for added/deleted files could
309
 
    # be usefully made into a much faster special case.
310
 
 
311
 
    # TODO: Better to return them in sorted order I think.
312
 
    
313
 
    for file_state, fid, old_name, new_name, kind in bzrlib.diff_trees(old_tree, new_tree):
314
 
        d = None
315
 
 
316
 
        # Don't show this by default; maybe do it if an option is passed
317
 
        # idlabel = '      {%s}' % fid
318
 
        idlabel = ''
319
 
 
320
 
        # FIXME: Something about the diff format makes patch unhappy
321
 
        # with newly-added files.
322
 
 
323
 
        def diffit(*a, **kw):
324
 
            sys.stdout.writelines(difflib.unified_diff(*a, **kw))
325
 
            print
326
 
        
327
 
        if file_state in ['.', '?', 'I']:
328
 
            continue
329
 
        elif file_state == 'A':
330
 
            print '*** added %s %r' % (kind, new_name)
331
 
            if kind == 'file':
332
 
                diffit([],
333
 
                       new_tree.get_file(fid).readlines(),
334
 
                       fromfile=DEVNULL,
335
 
                       tofile=new_label + new_name + idlabel)
336
 
        elif file_state == 'D':
337
 
            assert isinstance(old_name, types.StringTypes)
338
 
            print '*** deleted %s %r' % (kind, old_name)
339
 
            if kind == 'file':
340
 
                diffit(old_tree.get_file(fid).readlines(), [],
341
 
                       fromfile=old_label + old_name + idlabel,
342
 
                       tofile=DEVNULL)
343
 
        elif file_state in ['M', 'R']:
344
 
            if file_state == 'M':
345
 
                assert kind == 'file'
346
 
                assert old_name == new_name
347
 
                print '*** modified %s %r' % (kind, new_name)
348
 
            elif file_state == 'R':
349
 
                print '*** renamed %s %r => %r' % (kind, old_name, new_name)
350
 
 
351
 
            if kind == 'file':
352
 
                diffit(old_tree.get_file(fid).readlines(),
353
 
                       new_tree.get_file(fid).readlines(),
354
 
                       fromfile=old_label + old_name + idlabel,
355
 
                       tofile=new_label + new_name)
356
 
        else:
357
 
            bailout("can't represent state %s {%s}" % (file_state, fid))
358
 
 
359
 
 
360
 
 
361
 
def cmd_log(timezone='original'):
362
 
    """Show log of this branch.
363
 
 
364
 
    :todo: Options for utc; to show ids; to limit range; etc.
365
 
    """
366
 
    Branch('.').write_log(show_timezone=timezone)
367
 
 
368
 
 
369
 
def cmd_ls(revision=None, verbose=False):
370
 
    """List files in a tree.
371
 
 
372
 
    :todo: Take a revision or remote path and list that tree instead.
373
 
    """
374
 
    b = Branch('.')
375
 
    if revision == None:
376
 
        tree = b.working_tree()
377
 
    else:
378
 
        tree = b.revision_tree(b.lookup_revision(revision))
379
 
        
380
 
    for fp, fc, kind, fid in tree.list_files():
 
36
from bzrlib import (
 
37
    cleanup,
 
38
    cmdline,
 
39
    debug,
 
40
    errors,
 
41
    option,
 
42
    osutils,
 
43
    trace,
 
44
    ui,
 
45
    win32utils,
 
46
    )
 
47
""")
 
48
 
 
49
from bzrlib.hooks import HookPoint, Hooks
 
50
# Compatibility - Option used to be in commands.
 
51
from bzrlib.option import Option
 
52
from bzrlib.plugin import disable_plugins, load_plugins
 
53
from bzrlib import registry
 
54
from bzrlib.symbol_versioning import (
 
55
    deprecated_function,
 
56
    deprecated_in,
 
57
    deprecated_method,
 
58
    )
 
59
 
 
60
 
 
61
class CommandInfo(object):
 
62
    """Information about a command."""
 
63
 
 
64
    def __init__(self, aliases):
 
65
        """The list of aliases for the command."""
 
66
        self.aliases = aliases
 
67
 
 
68
    @classmethod
 
69
    def from_command(klass, command):
 
70
        """Factory to construct a CommandInfo from a command."""
 
71
        return klass(command.aliases)
 
72
 
 
73
 
 
74
class CommandRegistry(registry.Registry):
 
75
    """Special registry mapping command names to command classes.
 
76
    
 
77
    :ivar overridden_registry: Look in this registry for commands being
 
78
        overridden by this registry.  This can be used to tell plugin commands
 
79
        about the builtin they're decorating.
 
80
    """
 
81
 
 
82
    def __init__(self):
 
83
        registry.Registry.__init__(self)
 
84
        self.overridden_registry = None
 
85
        # map from aliases to the real command that implements the name
 
86
        self._alias_dict = {}
 
87
 
 
88
    def get(self, command_name):
 
89
        real_name = self._alias_dict.get(command_name, command_name)
 
90
        return registry.Registry.get(self, real_name)
 
91
 
 
92
    @staticmethod
 
93
    def _get_name(command_name):
 
94
        if command_name.startswith("cmd_"):
 
95
            return _unsquish_command_name(command_name)
 
96
        else:
 
97
            return command_name
 
98
 
 
99
    def register(self, cmd, decorate=False):
 
100
        """Utility function to help register a command
 
101
 
 
102
        :param cmd: Command subclass to register
 
103
        :param decorate: If true, allow overriding an existing command
 
104
            of the same name; the old command is returned by this function.
 
105
            Otherwise it is an error to try to override an existing command.
 
106
        """
 
107
        k = cmd.__name__
 
108
        k_unsquished = self._get_name(k)
 
109
        try:
 
110
            previous = self.get(k_unsquished)
 
111
        except KeyError:
 
112
            previous = None
 
113
            if self.overridden_registry:
 
114
                try:
 
115
                    previous = self.overridden_registry.get(k_unsquished)
 
116
                except KeyError:
 
117
                    pass
 
118
        info = CommandInfo.from_command(cmd)
 
119
        try:
 
120
            registry.Registry.register(self, k_unsquished, cmd,
 
121
                                       override_existing=decorate, info=info)
 
122
        except KeyError:
 
123
            trace.warning('Two plugins defined the same command: %r' % k)
 
124
            trace.warning('Not loading the one in %r' %
 
125
                sys.modules[cmd.__module__])
 
126
            trace.warning('Previously this command was registered from %r' %
 
127
                sys.modules[previous.__module__])
 
128
        for a in cmd.aliases:
 
129
            self._alias_dict[a] = k_unsquished
 
130
        return previous
 
131
 
 
132
    def register_lazy(self, command_name, aliases, module_name):
 
133
        """Register a command without loading its module.
 
134
 
 
135
        :param command_name: The primary name of the command.
 
136
        :param aliases: A list of aliases for the command.
 
137
        :module_name: The module that the command lives in.
 
138
        """
 
139
        key = self._get_name(command_name)
 
140
        registry.Registry.register_lazy(self, key, module_name, command_name,
 
141
                                        info=CommandInfo(aliases))
 
142
        for a in aliases:
 
143
            self._alias_dict[a] = key
 
144
 
 
145
 
 
146
plugin_cmds = CommandRegistry()
 
147
builtin_command_registry = CommandRegistry()
 
148
plugin_cmds.overridden_registry = builtin_command_registry
 
149
 
 
150
 
 
151
def register_command(cmd, decorate=False):
 
152
    """Register a plugin command.
 
153
 
 
154
    Should generally be avoided in favor of lazy registration. 
 
155
    """
 
156
    global plugin_cmds
 
157
    return plugin_cmds.register(cmd, decorate)
 
158
 
 
159
 
 
160
def _squish_command_name(cmd):
 
161
    return 'cmd_' + cmd.replace('-', '_')
 
162
 
 
163
 
 
164
def _unsquish_command_name(cmd):
 
165
    return cmd[4:].replace('_','-')
 
166
 
 
167
 
 
168
@deprecated_function(deprecated_in((2, 2, 0)))
 
169
def _builtin_commands():
 
170
    """Return a dict of {name: cmd_class} for builtin commands.
 
171
 
 
172
    :deprecated: Use the builtin_command_registry registry instead
 
173
    """
 
174
    # return dict(name: cmd_class)
 
175
    return dict(builtin_command_registry.items())
 
176
 
 
177
 
 
178
def _register_builtin_commands():
 
179
    if builtin_command_registry.keys():
 
180
        # only load once
 
181
        return
 
182
    import bzrlib.builtins
 
183
    for cmd_class in _scan_module_for_commands(bzrlib.builtins).values():
 
184
        builtin_command_registry.register(cmd_class)
 
185
    bzrlib.builtins._register_lazy_builtins()
 
186
 
 
187
 
 
188
def _scan_module_for_commands(module):
 
189
    r = {}
 
190
    for name, obj in module.__dict__.iteritems():
 
191
        if name.startswith("cmd_"):
 
192
            real_name = _unsquish_command_name(name)
 
193
            r[real_name] = obj
 
194
    return r
 
195
 
 
196
 
 
197
def _list_bzr_commands(names):
 
198
    """Find commands from bzr's core and plugins.
 
199
    
 
200
    This is not the public interface, just the default hook called by all_command_names.
 
201
    """
 
202
    # to eliminate duplicates
 
203
    names.update(builtin_command_names())
 
204
    names.update(plugin_command_names())
 
205
    return names
 
206
 
 
207
 
 
208
def all_command_names():
 
209
    """Return a set of all command names."""
 
210
    names = set()
 
211
    for hook in Command.hooks['list_commands']:
 
212
        names = hook(names)
 
213
        if names is None:
 
214
            raise AssertionError(
 
215
                'hook %s returned None' % Command.hooks.get_hook_name(hook))
 
216
    return names
 
217
 
 
218
 
 
219
def builtin_command_names():
 
220
    """Return list of builtin command names.
 
221
    
 
222
    Use of all_command_names() is encouraged rather than builtin_command_names
 
223
    and/or plugin_command_names.
 
224
    """
 
225
    _register_builtin_commands()
 
226
    return builtin_command_registry.keys()
 
227
 
 
228
 
 
229
def plugin_command_names():
 
230
    """Returns command names from commands registered by plugins."""
 
231
    return plugin_cmds.keys()
 
232
 
 
233
 
 
234
def get_cmd_object(cmd_name, plugins_override=True):
 
235
    """Return the command object for a command.
 
236
 
 
237
    plugins_override
 
238
        If true, plugin commands can override builtins.
 
239
    """
 
240
    try:
 
241
        return _get_cmd_object(cmd_name, plugins_override)
 
242
    except KeyError:
 
243
        raise errors.BzrCommandError('unknown command "%s"' % cmd_name)
 
244
 
 
245
 
 
246
def _get_cmd_object(cmd_name, plugins_override=True, check_missing=True):
 
247
    """Get a command object.
 
248
 
 
249
    :param cmd_name: The name of the command.
 
250
    :param plugins_override: Allow plugins to override builtins.
 
251
    :param check_missing: Look up commands not found in the regular index via
 
252
        the get_missing_command hook.
 
253
    :return: A Command object instance
 
254
    :raises KeyError: If no command is found.
 
255
    """
 
256
    # We want only 'ascii' command names, but the user may have typed
 
257
    # in a Unicode name. In that case, they should just get a
 
258
    # 'command not found' error later.
 
259
    # In the future, we may actually support Unicode command names.
 
260
    cmd = None
 
261
    # Get a command
 
262
    for hook in Command.hooks['get_command']:
 
263
        cmd = hook(cmd, cmd_name)
 
264
        if cmd is not None and not plugins_override and not cmd.plugin_name():
 
265
            # We've found a non-plugin command, don't permit it to be
 
266
            # overridden.
 
267
            break
 
268
    if cmd is None and check_missing:
 
269
        for hook in Command.hooks['get_missing_command']:
 
270
            cmd = hook(cmd_name)
 
271
            if cmd is not None:
 
272
                break
 
273
    if cmd is None:
 
274
        # No command found.
 
275
        raise KeyError
 
276
    # Allow plugins to extend commands
 
277
    for hook in Command.hooks['extend_command']:
 
278
        hook(cmd)
 
279
    return cmd
 
280
 
 
281
 
 
282
def _try_plugin_provider(cmd_name):
 
283
    """Probe for a plugin provider having cmd_name."""
 
284
    try:
 
285
        plugin_metadata, provider = probe_for_provider(cmd_name)
 
286
        raise errors.CommandAvailableInPlugin(cmd_name,
 
287
            plugin_metadata, provider)
 
288
    except errors.NoPluginAvailable:
 
289
        pass
 
290
 
 
291
 
 
292
def probe_for_provider(cmd_name):
 
293
    """Look for a provider for cmd_name.
 
294
 
 
295
    :param cmd_name: The command name.
 
296
    :return: plugin_metadata, provider for getting cmd_name.
 
297
    :raises NoPluginAvailable: When no provider can supply the plugin.
 
298
    """
 
299
    # look for providers that provide this command but aren't installed
 
300
    for provider in command_providers_registry:
 
301
        try:
 
302
            return provider.plugin_for_command(cmd_name), provider
 
303
        except errors.NoPluginAvailable:
 
304
            pass
 
305
    raise errors.NoPluginAvailable(cmd_name)
 
306
 
 
307
 
 
308
def _get_bzr_command(cmd_or_None, cmd_name):
 
309
    """Get a command from bzr's core."""
 
310
    try:
 
311
        cmd_class = builtin_command_registry.get(cmd_name)
 
312
    except KeyError:
 
313
        pass
 
314
    else:
 
315
        return cmd_class()
 
316
    return cmd_or_None
 
317
 
 
318
 
 
319
def _get_external_command(cmd_or_None, cmd_name):
 
320
    """Lookup a command that is a shell script."""
 
321
    # Only do external command lookups when no command is found so far.
 
322
    if cmd_or_None is not None:
 
323
        return cmd_or_None
 
324
    from bzrlib.externalcommand import ExternalCommand
 
325
    cmd_obj = ExternalCommand.find_command(cmd_name)
 
326
    if cmd_obj:
 
327
        return cmd_obj
 
328
 
 
329
 
 
330
def _get_plugin_command(cmd_or_None, cmd_name):
 
331
    """Get a command from bzr's plugins."""
 
332
    try:
 
333
        return plugin_cmds.get(cmd_name)()
 
334
    except KeyError:
 
335
        pass
 
336
    for key in plugin_cmds.keys():
 
337
        info = plugin_cmds.get_info(key)
 
338
        if cmd_name in info.aliases:
 
339
            return plugin_cmds.get(key)()
 
340
    return cmd_or_None
 
341
 
 
342
 
 
343
class Command(object):
 
344
    """Base class for commands.
 
345
 
 
346
    Commands are the heart of the command-line bzr interface.
 
347
 
 
348
    The command object mostly handles the mapping of command-line
 
349
    parameters into one or more bzrlib operations, and of the results
 
350
    into textual output.
 
351
 
 
352
    Commands normally don't have any state.  All their arguments are
 
353
    passed in to the run method.  (Subclasses may take a different
 
354
    policy if the behaviour of the instance needs to depend on e.g. a
 
355
    shell plugin and not just its Python class.)
 
356
 
 
357
    The docstring for an actual command should give a single-line
 
358
    summary, then a complete description of the command.  A grammar
 
359
    description will be inserted.
 
360
 
 
361
    aliases
 
362
        Other accepted names for this command.
 
363
 
 
364
    takes_args
 
365
        List of argument forms, marked with whether they are optional,
 
366
        repeated, etc.
 
367
 
 
368
                Examples:
 
369
 
 
370
                ['to_location', 'from_branch?', 'file*']
 
371
 
 
372
                'to_location' is required
 
373
                'from_branch' is optional
 
374
                'file' can be specified 0 or more times
 
375
 
 
376
    takes_options
 
377
        List of options that may be given for this command.  These can
 
378
        be either strings, referring to globally-defined options,
 
379
        or option objects.  Retrieve through options().
 
380
 
 
381
    hidden
 
382
        If true, this command isn't advertised.  This is typically
 
383
        for commands intended for expert users.
 
384
 
 
385
    encoding_type
 
386
        Command objects will get a 'outf' attribute, which has been
 
387
        setup to properly handle encoding of unicode strings.
 
388
        encoding_type determines what will happen when characters cannot
 
389
        be encoded
 
390
            strict - abort if we cannot decode
 
391
            replace - put in a bogus character (typically '?')
 
392
            exact - do not encode sys.stdout
 
393
 
 
394
            NOTE: by default on Windows, sys.stdout is opened as a text
 
395
            stream, therefore LF line-endings are converted to CRLF.
 
396
            When a command uses encoding_type = 'exact', then
 
397
            sys.stdout is forced to be a binary stream, and line-endings
 
398
            will not mangled.
 
399
 
 
400
    :cvar hooks: An instance of CommandHooks.
 
401
    :ivar __doc__: The help shown by 'bzr help command' for this command.
 
402
        This is set by assigning explicitly to __doc__ so that -OO can
 
403
        be used::
 
404
 
 
405
        class Foo(Command):
 
406
            __doc__ = "My help goes here"
 
407
    """
 
408
    aliases = []
 
409
    takes_args = []
 
410
    takes_options = []
 
411
    encoding_type = 'strict'
 
412
 
 
413
    hidden = False
 
414
 
 
415
    def __init__(self):
 
416
        """Construct an instance of this command."""
 
417
        # List of standard options directly supported
 
418
        self.supported_std_options = []
 
419
        self._setup_run()
 
420
 
 
421
    def add_cleanup(self, cleanup_func, *args, **kwargs):
 
422
        """Register a function to call after self.run returns or raises.
 
423
 
 
424
        Functions will be called in LIFO order.
 
425
        """
 
426
        self._operation.add_cleanup(cleanup_func, *args, **kwargs)
 
427
 
 
428
    def cleanup_now(self):
 
429
        """Execute and empty pending cleanup functions immediately.
 
430
 
 
431
        After cleanup_now all registered cleanups are forgotten.  add_cleanup
 
432
        may be called again after cleanup_now; these cleanups will be called
 
433
        after self.run returns or raises (or when cleanup_now is next called).
 
434
 
 
435
        This is useful for releasing expensive or contentious resources (such
 
436
        as write locks) before doing further work that does not require those
 
437
        resources (such as writing results to self.outf). Note though, that
 
438
        as it releases all resources, this may release locks that the command
 
439
        wants to hold, so use should be done with care.
 
440
        """
 
441
        self._operation.cleanup_now()
 
442
 
 
443
    @deprecated_method(deprecated_in((2, 1, 0)))
 
444
    def _maybe_expand_globs(self, file_list):
 
445
        """Glob expand file_list if the platform does not do that itself.
 
446
 
 
447
        Not used anymore, now that the bzr command-line parser globs on
 
448
        Windows.
 
449
 
 
450
        :return: A possibly empty list of unicode paths.
 
451
 
 
452
        Introduced in bzrlib 0.18.
 
453
        """
 
454
        return file_list
 
455
 
 
456
    def _usage(self):
 
457
        """Return single-line grammar for this command.
 
458
 
 
459
        Only describes arguments, not options.
 
460
        """
 
461
        s = 'bzr ' + self.name() + ' '
 
462
        for aname in self.takes_args:
 
463
            aname = aname.upper()
 
464
            if aname[-1] in ['$', '+']:
 
465
                aname = aname[:-1] + '...'
 
466
            elif aname[-1] == '?':
 
467
                aname = '[' + aname[:-1] + ']'
 
468
            elif aname[-1] == '*':
 
469
                aname = '[' + aname[:-1] + '...]'
 
470
            s += aname + ' '
 
471
        s = s[:-1]      # remove last space
 
472
        return s
 
473
 
 
474
    def get_help_text(self, additional_see_also=None, plain=True,
 
475
                      see_also_as_links=False, verbose=True):
 
476
        """Return a text string with help for this command.
 
477
 
 
478
        :param additional_see_also: Additional help topics to be
 
479
            cross-referenced.
 
480
        :param plain: if False, raw help (reStructuredText) is
 
481
            returned instead of plain text.
 
482
        :param see_also_as_links: if True, convert items in 'See also'
 
483
            list to internal links (used by bzr_man rstx generator)
 
484
        :param verbose: if True, display the full help, otherwise
 
485
            leave out the descriptive sections and just display
 
486
            usage help (e.g. Purpose, Usage, Options) with a
 
487
            message explaining how to obtain full help.
 
488
        """
 
489
        doc = self.help()
 
490
        if not doc:
 
491
            doc = "No help for this command."
 
492
 
 
493
        # Extract the summary (purpose) and sections out from the text
 
494
        purpose,sections,order = self._get_help_parts(doc)
 
495
 
 
496
        # If a custom usage section was provided, use it
 
497
        if sections.has_key('Usage'):
 
498
            usage = sections.pop('Usage')
 
499
        else:
 
500
            usage = self._usage()
 
501
 
 
502
        # The header is the purpose and usage
 
503
        result = ""
 
504
        result += ':Purpose: %s\n' % purpose
 
505
        if usage.find('\n') >= 0:
 
506
            result += ':Usage:\n%s\n' % usage
 
507
        else:
 
508
            result += ':Usage:   %s\n' % usage
 
509
        result += '\n'
 
510
 
 
511
        # Add the options
 
512
        #
 
513
        # XXX: optparse implicitly rewraps the help, and not always perfectly,
 
514
        # so we get <https://bugs.launchpad.net/bzr/+bug/249908>.  -- mbp
 
515
        # 20090319
 
516
        options = option.get_optparser(self.options()).format_option_help()
 
517
        # XXX: According to the spec, ReST option lists actually don't support 
 
518
        # options like --1.9 so that causes syntax errors (in Sphinx at least).
 
519
        # As that pattern always appears in the commands that break, we trap
 
520
        # on that and then format that block of 'format' options as a literal
 
521
        # block.
 
522
        if not plain and options.find('  --1.9  ') != -1:
 
523
            options = options.replace(' format:\n', ' format::\n\n', 1)
 
524
        if options.startswith('Options:'):
 
525
            result += ':' + options
 
526
        elif options.startswith('options:'):
 
527
            # Python 2.4 version of optparse
 
528
            result += ':Options:' + options[len('options:'):]
 
529
        else:
 
530
            result += options
 
531
        result += '\n'
 
532
 
381
533
        if verbose:
382
 
            if kind == 'directory':
383
 
                kindch = '/'
384
 
            elif kind == 'file':
385
 
                kindch = ''
 
534
            # Add the description, indenting it 2 spaces
 
535
            # to match the indentation of the options
 
536
            if sections.has_key(None):
 
537
                text = sections.pop(None)
 
538
                text = '\n  '.join(text.splitlines())
 
539
                result += ':%s:\n  %s\n\n' % ('Description',text)
 
540
 
 
541
            # Add the custom sections (e.g. Examples). Note that there's no need
 
542
            # to indent these as they must be indented already in the source.
 
543
            if sections:
 
544
                for label in order:
 
545
                    if sections.has_key(label):
 
546
                        result += ':%s:\n%s\n' % (label,sections[label])
 
547
                result += '\n'
 
548
        else:
 
549
            result += ("See bzr help %s for more details and examples.\n\n"
 
550
                % self.name())
 
551
 
 
552
        # Add the aliases, source (plug-in) and see also links, if any
 
553
        if self.aliases:
 
554
            result += ':Aliases:  '
 
555
            result += ', '.join(self.aliases) + '\n'
 
556
        plugin_name = self.plugin_name()
 
557
        if plugin_name is not None:
 
558
            result += ':From:     plugin "%s"\n' % plugin_name
 
559
        see_also = self.get_see_also(additional_see_also)
 
560
        if see_also:
 
561
            if not plain and see_also_as_links:
 
562
                see_also_links = []
 
563
                for item in see_also:
 
564
                    if item == 'topics':
 
565
                        # topics doesn't have an independent section
 
566
                        # so don't create a real link
 
567
                        see_also_links.append(item)
 
568
                    else:
 
569
                        # Use a Sphinx link for this entry
 
570
                        link_text = ":doc:`%s <%s-help>`" % (item, item)
 
571
                        see_also_links.append(link_text)
 
572
                see_also = see_also_links
 
573
            result += ':See also: '
 
574
            result += ', '.join(see_also) + '\n'
 
575
 
 
576
        # If this will be rendered as plain text, convert it
 
577
        if plain:
 
578
            import bzrlib.help_topics
 
579
            result = bzrlib.help_topics.help_as_plain_text(result)
 
580
        return result
 
581
 
 
582
    @staticmethod
 
583
    def _get_help_parts(text):
 
584
        """Split help text into a summary and named sections.
 
585
 
 
586
        :return: (summary,sections,order) where summary is the top line and
 
587
            sections is a dictionary of the rest indexed by section name.
 
588
            order is the order the section appear in the text.
 
589
            A section starts with a heading line of the form ":xxx:".
 
590
            Indented text on following lines is the section value.
 
591
            All text found outside a named section is assigned to the
 
592
            default section which is given the key of None.
 
593
        """
 
594
        def save_section(sections, order, label, section):
 
595
            if len(section) > 0:
 
596
                if sections.has_key(label):
 
597
                    sections[label] += '\n' + section
 
598
                else:
 
599
                    order.append(label)
 
600
                    sections[label] = section
 
601
 
 
602
        lines = text.rstrip().splitlines()
 
603
        summary = lines.pop(0)
 
604
        sections = {}
 
605
        order = []
 
606
        label,section = None,''
 
607
        for line in lines:
 
608
            if line.startswith(':') and line.endswith(':') and len(line) > 2:
 
609
                save_section(sections, order, label, section)
 
610
                label,section = line[1:-1],''
 
611
            elif (label is not None) and len(line) > 1 and not line[0].isspace():
 
612
                save_section(sections, order, label, section)
 
613
                label,section = None,line
386
614
            else:
387
 
                kindch = '???'
388
 
                
389
 
            print '%-8s %s%s' % (fc, fp, kindch)
 
615
                if len(section) > 0:
 
616
                    section += '\n' + line
 
617
                else:
 
618
                    section = line
 
619
        save_section(sections, order, label, section)
 
620
        return summary, sections, order
 
621
 
 
622
    def get_help_topic(self):
 
623
        """Return the commands help topic - its name."""
 
624
        return self.name()
 
625
 
 
626
    def get_see_also(self, additional_terms=None):
 
627
        """Return a list of help topics that are related to this command.
 
628
 
 
629
        The list is derived from the content of the _see_also attribute. Any
 
630
        duplicates are removed and the result is in lexical order.
 
631
        :param additional_terms: Additional help topics to cross-reference.
 
632
        :return: A list of help topics.
 
633
        """
 
634
        see_also = set(getattr(self, '_see_also', []))
 
635
        if additional_terms:
 
636
            see_also.update(additional_terms)
 
637
        return sorted(see_also)
 
638
 
 
639
    def options(self):
 
640
        """Return dict of valid options for this command.
 
641
 
 
642
        Maps from long option name to option object."""
 
643
        r = Option.STD_OPTIONS.copy()
 
644
        std_names = r.keys()
 
645
        for o in self.takes_options:
 
646
            if isinstance(o, basestring):
 
647
                o = option.Option.OPTIONS[o]
 
648
            r[o.name] = o
 
649
            if o.name in std_names:
 
650
                self.supported_std_options.append(o.name)
 
651
        return r
 
652
 
 
653
    def _setup_outf(self):
 
654
        """Return a file linked to stdout, which has proper encoding."""
 
655
        self.outf = ui.ui_factory.make_output_stream(
 
656
            encoding_type=self.encoding_type)
 
657
 
 
658
    def run_argv_aliases(self, argv, alias_argv=None):
 
659
        """Parse the command line and run with extra aliases in alias_argv."""
 
660
        args, opts = parse_args(self, argv, alias_argv)
 
661
 
 
662
        # Process the standard options
 
663
        if 'help' in opts:  # e.g. bzr add --help
 
664
            sys.stdout.write(self.get_help_text())
 
665
            return 0
 
666
        if 'usage' in opts:  # e.g. bzr add --usage
 
667
            sys.stdout.write(self.get_help_text(verbose=False))
 
668
            return 0
 
669
        trace.set_verbosity_level(option._verbosity_level)
 
670
        if 'verbose' in self.supported_std_options:
 
671
            opts['verbose'] = trace.is_verbose()
 
672
        elif opts.has_key('verbose'):
 
673
            del opts['verbose']
 
674
        if 'quiet' in self.supported_std_options:
 
675
            opts['quiet'] = trace.is_quiet()
 
676
        elif opts.has_key('quiet'):
 
677
            del opts['quiet']
 
678
 
 
679
        # mix arguments and options into one dictionary
 
680
        cmdargs = _match_argform(self.name(), self.takes_args, args)
 
681
        cmdopts = {}
 
682
        for k, v in opts.items():
 
683
            cmdopts[k.replace('-', '_')] = v
 
684
 
 
685
        all_cmd_args = cmdargs.copy()
 
686
        all_cmd_args.update(cmdopts)
 
687
 
 
688
        self._setup_outf()
 
689
 
 
690
        try:
 
691
            return self.run(**all_cmd_args)
 
692
        finally:
 
693
            # reset it, so that other commands run in the same process won't
 
694
            # inherit state
 
695
            trace.set_verbosity_level(0)
 
696
 
 
697
    def _setup_run(self):
 
698
        """Wrap the defined run method on self with a cleanup.
 
699
 
 
700
        This is called by __init__ to make the Command be able to be run
 
701
        by just calling run(), as it could be before cleanups were added.
 
702
 
 
703
        If a different form of cleanups are in use by your Command subclass,
 
704
        you can override this method.
 
705
        """
 
706
        class_run = self.run
 
707
        def run(*args, **kwargs):
 
708
            self._operation = cleanup.OperationWithCleanups(class_run)
 
709
            try:
 
710
                return self._operation.run_simple(*args, **kwargs)
 
711
            finally:
 
712
                del self._operation
 
713
        self.run = run
 
714
 
 
715
    @deprecated_method(deprecated_in((2, 2, 0)))
 
716
    def run_direct(self, *args, **kwargs):
 
717
        """Deprecated thunk from bzrlib 2.1."""
 
718
        return self.run(*args, **kwargs)
 
719
 
 
720
    def run(self):
 
721
        """Actually run the command.
 
722
 
 
723
        This is invoked with the options and arguments bound to
 
724
        keyword parameters.
 
725
 
 
726
        Return 0 or None if the command was successful, or a non-zero
 
727
        shell error code if not.  It's OK for this method to allow
 
728
        an exception to raise up.
 
729
 
 
730
        This method is automatically wrapped by Command.__init__ with a 
 
731
        cleanup operation, stored as self._operation. This can be used
 
732
        via self.add_cleanup to perform automatic cleanups at the end of
 
733
        run().
 
734
 
 
735
        The argument for run are assembled by introspection. So for instance,
 
736
        if your command takes an argument files, you would declare::
 
737
 
 
738
            def run(self, files=None):
 
739
                pass
 
740
        """
 
741
        raise NotImplementedError('no implementation of command %r'
 
742
                                  % self.name())
 
743
 
 
744
    def help(self):
 
745
        """Return help message for this class."""
 
746
        from inspect import getdoc
 
747
        if self.__doc__ is Command.__doc__:
 
748
            return None
 
749
        return getdoc(self)
 
750
 
 
751
    def name(self):
 
752
        return _unsquish_command_name(self.__class__.__name__)
 
753
 
 
754
    def plugin_name(self):
 
755
        """Get the name of the plugin that provides this command.
 
756
 
 
757
        :return: The name of the plugin or None if the command is builtin.
 
758
        """
 
759
        mod_parts = self.__module__.split('.')
 
760
        if len(mod_parts) >= 3 and mod_parts[1] == 'plugins':
 
761
            return mod_parts[2]
390
762
        else:
391
 
            print fp
392
 
    
393
 
    
394
 
 
395
 
def cmd_unknowns():
396
 
    """List unknown files"""
397
 
    for f in Branch('.').unknowns():
398
 
        print quotefn(f)
399
 
 
400
 
 
401
 
def cmd_lookup_revision(revno):
402
 
    try:
403
 
        revno = int(revno)
404
 
    except ValueError:
405
 
        bailout("usage: lookup-revision REVNO",
406
 
                ["REVNO is a non-negative revision number for this branch"])
407
 
 
408
 
    print Branch('.').lookup_revision(revno) or NONE_STRING
409
 
 
410
 
 
411
 
 
412
 
def cmd_export(revno, dest):
413
 
    """Export past revision to destination directory."""
414
 
    b = Branch('.')
415
 
    rh = b.lookup_revision(int(revno))
416
 
    t = b.revision_tree(rh)
417
 
    t.export(dest)
418
 
 
419
 
 
420
 
 
421
 
######################################################################
422
 
# internal/test commands
423
 
 
424
 
 
425
 
def cmd_uuid():
426
 
    """Print a newly-generated UUID."""
427
 
    print uuid()
428
 
 
429
 
 
430
 
 
431
 
def cmd_local_time_offset():
432
 
    print bzrlib.osutils.local_time_offset()
433
 
 
434
 
 
435
 
 
436
 
def cmd_commit(message, verbose=False):
437
 
    Branch('.').commit(message, verbose=verbose)
438
 
 
439
 
 
440
 
def cmd_check():
441
 
    """Check consistency of the branch."""
442
 
    check()
443
 
 
444
 
 
445
 
def cmd_is(pred, *rest):
446
 
    """Test whether PREDICATE is true."""
447
 
    try:
448
 
        cmd_handler = globals()['assert_' + pred.replace('-', '_')]
449
 
    except KeyError:
450
 
        bailout("unknown predicate: %s" % quotefn(pred))
451
 
        
452
 
    try:
453
 
        cmd_handler(*rest)
454
 
    except BzrCheckError:
455
 
        # by default we don't print the message so that this can
456
 
        # be used from shell scripts without producing noise
457
 
        sys.exit(1)
458
 
 
459
 
 
460
 
def cmd_username():
461
 
    print bzrlib.osutils.username()
462
 
 
463
 
 
464
 
def cmd_user_email():
465
 
    print bzrlib.osutils.user_email()
466
 
 
467
 
 
468
 
def cmd_gen_revision_id():
469
 
    import time
470
 
    print bzrlib.branch._gen_revision_id(time.time())
471
 
 
472
 
 
473
 
def cmd_doctest():
474
 
    """Run internal doctest suite"""
475
 
    ## -v, if present, is seen by doctest; the argument is just here
476
 
    ## so our parser doesn't complain
477
 
 
478
 
    ## TODO: --verbose option
479
 
    
480
 
    import bzr, doctest, bzrlib.store
481
 
    bzrlib.trace.verbose = False
482
 
    doctest.testmod(bzr)
483
 
    doctest.testmod(bzrlib.store)
484
 
    doctest.testmod(bzrlib.inventory)
485
 
    doctest.testmod(bzrlib.branch)
486
 
    doctest.testmod(bzrlib.osutils)
487
 
    doctest.testmod(bzrlib.tree)
488
 
 
489
 
    # more strenuous tests;
490
 
    import bzrlib.tests
491
 
    doctest.testmod(bzrlib.tests)
492
 
 
493
 
 
494
 
######################################################################
495
 
# help
496
 
 
497
 
 
498
 
def cmd_help():
499
 
    # TODO: Specific help for particular commands
500
 
    print __doc__
501
 
 
502
 
 
503
 
def cmd_version():
504
 
    print "bzr (bazaar-ng) %s" % __version__
505
 
    print __copyright__
506
 
    print "http://bazaar-ng.org/"
507
 
    print
508
 
    print \
509
 
"""bzr comes with ABSOLUTELY NO WARRANTY.  bzr is free software, and
510
 
you may use, modify and redistribute it under the terms of the GNU 
511
 
General Public License version 2 or later."""
512
 
 
513
 
 
514
 
def cmd_rocks():
515
 
    """Statement of optimism."""
516
 
    print "it sure does!"
517
 
 
518
 
 
519
 
 
520
 
######################################################################
521
 
# main routine
522
 
 
523
 
 
524
 
# list of all available options; the rhs can be either None for an
525
 
# option that takes no argument, or a constructor function that checks
526
 
# the type.
527
 
OPTIONS = {
528
 
    'all':                    None,
529
 
    'help':                   None,
530
 
    'message':                unicode,
531
 
    'revision':               int,
532
 
    'show-ids':               None,
533
 
    'timezone':               str,
534
 
    'verbose':                None,
535
 
    'version':                None,
536
 
    }
537
 
 
538
 
SHORT_OPTIONS = {
539
 
    'm':                      'message',
540
 
    'r':                      'revision',
541
 
    'v':                      'verbose',
542
 
}
543
 
 
544
 
# List of options that apply to particular commands; commands not
545
 
# listed take none.
546
 
cmd_options = {
547
 
    'add':                    ['verbose'],
548
 
    'commit':                 ['message', 'verbose'],
549
 
    'diff':                   ['revision'],
550
 
    'inventory':              ['revision'],
551
 
    'log':                    ['show-ids', 'timezone'],
552
 
    'ls':                     ['revision', 'verbose'],
553
 
    'remove':                 ['verbose'],
554
 
    'status':                 ['all'],
555
 
    }
556
 
 
557
 
 
558
 
cmd_args = {
559
 
    'init':                   [],
560
 
    'add':                    ['file+'],
561
 
    'commit':                 [],
562
 
    'diff':                   [],
563
 
    'file-id':                ['filename'],
564
 
    'get-file-text':          ['text_id'],
565
 
    'get-inventory':          ['inventory_id'],
566
 
    'get-revision':           ['revision_id'],
567
 
    'get-revision-inventory': ['revision_id'],
568
 
    'log':                    [],
569
 
    'lookup-revision':        ['revno'],
570
 
    'export':                 ['revno', 'dest'],
571
 
    'remove':                 ['file+'],
572
 
    'status':                 [],
573
 
    }
574
 
 
575
 
 
576
 
def parse_args(argv):
 
763
            return None
 
764
 
 
765
 
 
766
class CommandHooks(Hooks):
 
767
    """Hooks related to Command object creation/enumeration."""
 
768
 
 
769
    def __init__(self):
 
770
        """Create the default hooks.
 
771
 
 
772
        These are all empty initially, because by default nothing should get
 
773
        notified.
 
774
        """
 
775
        Hooks.__init__(self)
 
776
        self.create_hook(HookPoint('extend_command',
 
777
            "Called after creating a command object to allow modifications "
 
778
            "such as adding or removing options, docs etc. Called with the "
 
779
            "new bzrlib.commands.Command object.", (1, 13), None))
 
780
        self.create_hook(HookPoint('get_command',
 
781
            "Called when creating a single command. Called with "
 
782
            "(cmd_or_None, command_name). get_command should either return "
 
783
            "the cmd_or_None parameter, or a replacement Command object that "
 
784
            "should be used for the command. Note that the Command.hooks "
 
785
            "hooks are core infrastructure. Many users will prefer to use "
 
786
            "bzrlib.commands.register_command or plugin_cmds.register_lazy.",
 
787
            (1, 17), None))
 
788
        self.create_hook(HookPoint('get_missing_command',
 
789
            "Called when creating a single command if no command could be "
 
790
            "found. Called with (command_name). get_missing_command should "
 
791
            "either return None, or a Command object to be used for the "
 
792
            "command.", (1, 17), None))
 
793
        self.create_hook(HookPoint('list_commands',
 
794
            "Called when enumerating commands. Called with a set of "
 
795
            "cmd_name strings for all the commands found so far. This set "
 
796
            " is safe to mutate - e.g. to remove a command. "
 
797
            "list_commands should return the updated set of command names.",
 
798
            (1, 17), None))
 
799
 
 
800
Command.hooks = CommandHooks()
 
801
 
 
802
 
 
803
def parse_args(command, argv, alias_argv=None):
577
804
    """Parse command line.
578
 
    
 
805
 
579
806
    Arguments and options are parsed at this level before being passed
580
807
    down to specific command handlers.  This routine knows, from a
581
808
    lookup table, something about the available options, what optargs
582
809
    they take, and which commands will accept them.
583
 
 
584
 
    >>> parse_args('--help'.split())
585
 
    ([], {'help': True})
586
 
    >>> parse_args('--version'.split())
587
 
    ([], {'version': True})
588
 
    >>> parse_args('status --all'.split())
589
 
    (['status'], {'all': True})
590
 
    >>> parse_args('commit --message=biter'.split())
591
 
    (['commit'], {'message': u'biter'})
592
810
    """
593
 
    args = []
594
 
    opts = {}
595
 
 
596
 
    # TODO: Maybe handle '--' to end options?
597
 
 
598
 
    while argv:
599
 
        a = argv.pop(0)
600
 
        if a[0] == '-':
601
 
            optarg = None
602
 
            if a[1] == '-':
603
 
                mutter("  got option %r" % a)
604
 
                if '=' in a:
605
 
                    optname, optarg = a[2:].split('=', 1)
606
 
                else:
607
 
                    optname = a[2:]
608
 
                if optname not in OPTIONS:
609
 
                    bailout('unknown long option %r' % a)
610
 
            else:
611
 
                shortopt = a[1:]
612
 
                if shortopt not in SHORT_OPTIONS:
613
 
                    bailout('unknown short option %r' % a)
614
 
                optname = SHORT_OPTIONS[shortopt]
615
 
            
616
 
            if optname in opts:
617
 
                # XXX: Do we ever want to support this, e.g. for -r?
618
 
                bailout('repeated option %r' % a)
619
 
                
620
 
            optargfn = OPTIONS[optname]
621
 
            if optargfn:
622
 
                if optarg == None:
623
 
                    if not argv:
624
 
                        bailout('option %r needs an argument' % a)
625
 
                    else:
626
 
                        optarg = argv.pop(0)
627
 
                opts[optname] = optargfn(optarg)
628
 
                mutter("    option argument %r" % opts[optname])
629
 
            else:
630
 
                if optarg != None:
631
 
                    bailout('option %r takes no argument' % optname)
632
 
                opts[optname] = True
633
 
        else:
634
 
            args.append(a)
635
 
 
 
811
    # TODO: make it a method of the Command?
 
812
    parser = option.get_optparser(command.options())
 
813
    if alias_argv is not None:
 
814
        args = alias_argv + argv
 
815
    else:
 
816
        args = argv
 
817
 
 
818
    # for python 2.5 and later, optparse raises this exception if a non-ascii
 
819
    # option name is given.  See http://bugs.python.org/issue2931
 
820
    try:
 
821
        options, args = parser.parse_args(args)
 
822
    except UnicodeEncodeError,e:
 
823
        raise errors.BzrCommandError('Only ASCII permitted in option names')
 
824
 
 
825
    opts = dict([(k, v) for k, v in options.__dict__.iteritems() if
 
826
                 v is not option.OptionParser.DEFAULT_VALUE])
636
827
    return args, opts
637
828
 
638
829
 
639
 
 
640
 
def _match_args(cmd, args):
641
 
    """Check non-option arguments match required pattern.
642
 
 
643
 
    >>> _match_args('status', ['asdasdsadasd'])
644
 
    Traceback (most recent call last):
645
 
    ...
646
 
    BzrError: ("extra arguments to command status: ['asdasdsadasd']", [])
647
 
    >>> _match_args('add', ['asdasdsadasd'])
648
 
    {'file_list': ['asdasdsadasd']}
649
 
    >>> _match_args('add', 'abc def gj'.split())
650
 
    {'file_list': ['abc', 'def', 'gj']}
651
 
    """
652
 
    # match argument pattern
653
 
    argform = cmd_args.get(cmd, [])
 
830
def _match_argform(cmd, takes_args, args):
654
831
    argdict = {}
655
 
    # TODO: Need a way to express 'cp SRC... DEST', where it matches
656
 
    # all but one.
657
832
 
658
 
    for ap in argform:
 
833
    # step through args and takes_args, allowing appropriate 0-many matches
 
834
    for ap in takes_args:
659
835
        argname = ap[:-1]
660
836
        if ap[-1] == '?':
661
 
            assert 0
662
 
        elif ap[-1] == '*':
663
 
            assert 0
 
837
            if args:
 
838
                argdict[argname] = args.pop(0)
 
839
        elif ap[-1] == '*': # all remaining arguments
 
840
            if args:
 
841
                argdict[argname + '_list'] = args[:]
 
842
                args = []
 
843
            else:
 
844
                argdict[argname + '_list'] = None
664
845
        elif ap[-1] == '+':
665
846
            if not args:
666
 
                bailout("command %r needs one or more %s"
667
 
                        % (cmd, argname.upper()))
 
847
                raise errors.BzrCommandError("command %r needs one or more %s"
 
848
                                             % (cmd, argname.upper()))
668
849
            else:
669
850
                argdict[argname + '_list'] = args[:]
670
851
                args = []
 
852
        elif ap[-1] == '$': # all but one
 
853
            if len(args) < 2:
 
854
                raise errors.BzrCommandError("command %r needs one or more %s"
 
855
                                             % (cmd, argname.upper()))
 
856
            argdict[argname + '_list'] = args[:-1]
 
857
            args[:-1] = []
671
858
        else:
672
859
            # just a plain arg
673
860
            argname = ap
674
861
            if not args:
675
 
                bailout("command %r requires argument %s"
676
 
                        % (cmd, argname.upper()))
 
862
                raise errors.BzrCommandError("command %r requires argument %s"
 
863
                               % (cmd, argname.upper()))
677
864
            else:
678
865
                argdict[argname] = args.pop(0)
679
 
            
 
866
 
680
867
    if args:
681
 
        bailout("extra arguments to command %s: %r"
682
 
                % (cmd, args))
 
868
        raise errors.BzrCommandError("extra argument to command %s: %s"
 
869
                                     % (cmd, args[0]))
683
870
 
684
871
    return argdict
685
872
 
686
 
 
687
 
 
688
 
def run_bzr(argv):
 
873
def apply_coveraged(dirname, the_callable, *args, **kwargs):
 
874
    # Cannot use "import trace", as that would import bzrlib.trace instead of
 
875
    # the standard library's trace.
 
876
    trace = __import__('trace')
 
877
 
 
878
    tracer = trace.Trace(count=1, trace=0)
 
879
    sys.settrace(tracer.globaltrace)
 
880
    threading.settrace(tracer.globaltrace)
 
881
 
 
882
    try:
 
883
        return exception_to_return_code(the_callable, *args, **kwargs)
 
884
    finally:
 
885
        sys.settrace(None)
 
886
        results = tracer.results()
 
887
        results.write_results(show_missing=1, summary=False,
 
888
                              coverdir=dirname)
 
889
 
 
890
 
 
891
def apply_profiled(the_callable, *args, **kwargs):
 
892
    import hotshot
 
893
    import tempfile
 
894
    import hotshot.stats
 
895
    pffileno, pfname = tempfile.mkstemp()
 
896
    try:
 
897
        prof = hotshot.Profile(pfname)
 
898
        try:
 
899
            ret = prof.runcall(exception_to_return_code, the_callable, *args,
 
900
                **kwargs) or 0
 
901
        finally:
 
902
            prof.close()
 
903
        stats = hotshot.stats.load(pfname)
 
904
        stats.strip_dirs()
 
905
        stats.sort_stats('cum')   # 'time'
 
906
        ## XXX: Might like to write to stderr or the trace file instead but
 
907
        ## print_stats seems hardcoded to stdout
 
908
        stats.print_stats(20)
 
909
        return ret
 
910
    finally:
 
911
        os.close(pffileno)
 
912
        os.remove(pfname)
 
913
 
 
914
 
 
915
def exception_to_return_code(the_callable, *args, **kwargs):
 
916
    """UI level helper for profiling and coverage.
 
917
 
 
918
    This transforms exceptions into a return value of 3. As such its only
 
919
    relevant to the UI layer, and should never be called where catching
 
920
    exceptions may be desirable.
 
921
    """
 
922
    try:
 
923
        return the_callable(*args, **kwargs)
 
924
    except (KeyboardInterrupt, Exception), e:
 
925
        # used to handle AssertionError and KeyboardInterrupt
 
926
        # specially here, but hopefully they're handled ok by the logger now
 
927
        exc_info = sys.exc_info()
 
928
        exitcode = trace.report_exception(exc_info, sys.stderr)
 
929
        if os.environ.get('BZR_PDB'):
 
930
            print '**** entering debugger'
 
931
            tb = exc_info[2]
 
932
            import pdb
 
933
            if sys.version_info[:2] < (2, 6):
 
934
                # XXX: we want to do
 
935
                #    pdb.post_mortem(tb)
 
936
                # but because pdb.post_mortem gives bad results for tracebacks
 
937
                # from inside generators, we do it manually.
 
938
                # (http://bugs.python.org/issue4150, fixed in Python 2.6)
 
939
 
 
940
                # Setup pdb on the traceback
 
941
                p = pdb.Pdb()
 
942
                p.reset()
 
943
                p.setup(tb.tb_frame, tb)
 
944
                # Point the debugger at the deepest frame of the stack
 
945
                p.curindex = len(p.stack) - 1
 
946
                p.curframe = p.stack[p.curindex][0]
 
947
                # Start the pdb prompt.
 
948
                p.print_stack_entry(p.stack[p.curindex])
 
949
                p.execRcLines()
 
950
                p.cmdloop()
 
951
            else:
 
952
                pdb.post_mortem(tb)
 
953
        return exitcode
 
954
 
 
955
 
 
956
def apply_lsprofiled(filename, the_callable, *args, **kwargs):
 
957
    from bzrlib.lsprof import profile
 
958
    ret, stats = profile(exception_to_return_code, the_callable, *args, **kwargs)
 
959
    stats.sort()
 
960
    if filename is None:
 
961
        stats.pprint()
 
962
    else:
 
963
        stats.save(filename)
 
964
        trace.note('Profile data written to "%s".', filename)
 
965
    return ret
 
966
 
 
967
 
 
968
@deprecated_function(deprecated_in((2, 2, 0)))
 
969
def shlex_split_unicode(unsplit):
 
970
    return cmdline.split(unsplit)
 
971
 
 
972
 
 
973
def get_alias(cmd, config=None):
 
974
    """Return an expanded alias, or None if no alias exists.
 
975
 
 
976
    cmd
 
977
        Command to be checked for an alias.
 
978
    config
 
979
        Used to specify an alternative config to use,
 
980
        which is especially useful for testing.
 
981
        If it is unspecified, the global config will be used.
 
982
    """
 
983
    if config is None:
 
984
        import bzrlib.config
 
985
        config = bzrlib.config.GlobalConfig()
 
986
    alias = config.get_alias(cmd)
 
987
    if (alias):
 
988
        return cmdline.split(alias)
 
989
    return None
 
990
 
 
991
 
 
992
def run_bzr(argv, load_plugins=load_plugins, disable_plugins=disable_plugins):
689
993
    """Execute a command.
690
994
 
691
 
    This is similar to main(), but without all the trappings for
692
 
    logging and error handling.
693
 
    """
694
 
    try:
695
 
        args, opts = parse_args(argv[1:])
696
 
        if 'help' in opts:
697
 
            # TODO: pass down other arguments in case they asked for
698
 
            # help on a command name?
699
 
            cmd_help()
700
 
            return 0
701
 
        elif 'version' in opts:
702
 
            cmd_version()
703
 
            return 0
704
 
        cmd = args.pop(0)
705
 
    except IndexError:
706
 
        log_error('usage: bzr COMMAND\n')
707
 
        log_error('  try "bzr help"\n')
708
 
        return 1
709
 
            
710
 
    try:
711
 
        cmd_handler = globals()['cmd_' + cmd.replace('-', '_')]
712
 
    except KeyError:
713
 
        bailout("unknown command " + `cmd`)
714
 
 
715
 
    # TODO: special --profile option to turn on the Python profiler
716
 
 
717
 
    # check options are reasonable
718
 
    allowed = cmd_options.get(cmd, [])
719
 
    for oname in opts:
720
 
        if oname not in allowed:
721
 
            bailout("option %r is not allowed for command %r"
722
 
                    % (oname, cmd))
723
 
 
724
 
    cmdargs = _match_args(cmd, args)
725
 
    cmdargs.update(opts)
726
 
 
727
 
    ret = cmd_handler(**cmdargs) or 0
728
 
 
729
 
 
730
 
 
731
 
def main(argv):
732
 
    ## TODO: Handle command-line options; probably know what options are valid for
733
 
    ## each command
734
 
 
735
 
    ## TODO: If the arguments are wrong, give a usage message rather
736
 
    ## than just a backtrace.
737
 
 
738
 
    try:
739
 
        # TODO: Lift into separate function in trace.py
740
 
        # TODO: Also show contents of /etc/lsb-release, if it can be parsed.
741
 
        #       Perhaps that should eventually go into the platform library?
742
 
        # TODO: If the file doesn't exist, add a note describing it.
743
 
        t = bzrlib.trace._tracefile
744
 
        t.write('-' * 60 + '\n')
745
 
        t.write('bzr invoked at %s\n' % format_date(time.time()))
746
 
        t.write('  by %s on %s\n' % (bzrlib.osutils.username(), socket.getfqdn()))
747
 
        t.write('  arguments: %r\n' % argv)
748
 
 
749
 
        starttime = os.times()[4]
750
 
 
751
 
        import platform
752
 
        t.write('  platform: %s\n' % platform.platform())
753
 
        t.write('  python: %s\n' % platform.python_version())
754
 
 
755
 
        ret = run_bzr(argv)
756
 
        
757
 
        times = os.times()
758
 
        mutter("finished, %.3fu/%.3fs cpu, %.3fu/%.3fs cum"
759
 
               % times[:4])
760
 
        mutter("    %.3f elapsed" % (times[4] - starttime))
761
 
 
762
 
        return ret
763
 
    except BzrError, e:
764
 
        log_error('bzr: error: ' + e.args[0] + '\n')
765
 
        if len(e.args) > 1:
766
 
            for h in e.args[1]:
767
 
                log_error('  ' + h + '\n')
768
 
        return 1
 
995
    :param argv: The command-line arguments, without the program name from
 
996
        argv[0] These should already be decoded. All library/test code calling
 
997
        run_bzr should be passing valid strings (don't need decoding).
 
998
    :param load_plugins: What function to call when triggering plugin loading.
 
999
        This function should take no arguments and cause all plugins to be
 
1000
        loaded.
 
1001
    :param disable_plugins: What function to call when disabling plugin
 
1002
        loading. This function should take no arguments and cause all plugin
 
1003
        loading to be prohibited (so that code paths in your application that
 
1004
        know about some plugins possibly being present will fail to import
 
1005
        those plugins even if they are installed.)
 
1006
    :return: Returns a command exit code or raises an exception.
 
1007
 
 
1008
    Special master options: these must come before the command because
 
1009
    they control how the command is interpreted.
 
1010
 
 
1011
    --no-plugins
 
1012
        Do not load plugin modules at all
 
1013
 
 
1014
    --no-aliases
 
1015
        Do not allow aliases
 
1016
 
 
1017
    --builtin
 
1018
        Only use builtin commands.  (Plugins are still allowed to change
 
1019
        other behaviour.)
 
1020
 
 
1021
    --profile
 
1022
        Run under the Python hotshot profiler.
 
1023
 
 
1024
    --lsprof
 
1025
        Run under the Python lsprof profiler.
 
1026
 
 
1027
    --coverage
 
1028
        Generate line coverage report in the specified directory.
 
1029
 
 
1030
    --concurrency
 
1031
        Specify the number of processes that can be run concurrently (selftest).
 
1032
    """
 
1033
    trace.mutter("bazaar version: " + bzrlib.__version__)
 
1034
    argv = list(argv)
 
1035
    trace.mutter("bzr arguments: %r", argv)
 
1036
 
 
1037
    opt_lsprof = opt_profile = opt_no_plugins = opt_builtin =  \
 
1038
                opt_no_aliases = False
 
1039
    opt_lsprof_file = opt_coverage_dir = None
 
1040
 
 
1041
    # --no-plugins is handled specially at a very early stage. We need
 
1042
    # to load plugins before doing other command parsing so that they
 
1043
    # can override commands, but this needs to happen first.
 
1044
 
 
1045
    argv_copy = []
 
1046
    i = 0
 
1047
    while i < len(argv):
 
1048
        a = argv[i]
 
1049
        if a == '--profile':
 
1050
            opt_profile = True
 
1051
        elif a == '--lsprof':
 
1052
            opt_lsprof = True
 
1053
        elif a == '--lsprof-file':
 
1054
            opt_lsprof = True
 
1055
            opt_lsprof_file = argv[i + 1]
 
1056
            i += 1
 
1057
        elif a == '--no-plugins':
 
1058
            opt_no_plugins = True
 
1059
        elif a == '--no-aliases':
 
1060
            opt_no_aliases = True
 
1061
        elif a == '--builtin':
 
1062
            opt_builtin = True
 
1063
        elif a == '--concurrency':
 
1064
            os.environ['BZR_CONCURRENCY'] = argv[i + 1]
 
1065
            i += 1
 
1066
        elif a == '--coverage':
 
1067
            opt_coverage_dir = argv[i + 1]
 
1068
            i += 1
 
1069
        elif a == '--profile-imports':
 
1070
            pass # already handled in startup script Bug #588277
 
1071
        elif a.startswith('-D'):
 
1072
            debug.debug_flags.add(a[2:])
 
1073
        else:
 
1074
            argv_copy.append(a)
 
1075
        i += 1
 
1076
 
 
1077
    debug.set_debug_flags_from_config()
 
1078
 
 
1079
    if not opt_no_plugins:
 
1080
        load_plugins()
 
1081
    else:
 
1082
        disable_plugins()
 
1083
 
 
1084
    argv = argv_copy
 
1085
    if (not argv):
 
1086
        get_cmd_object('help').run_argv_aliases([])
 
1087
        return 0
 
1088
 
 
1089
    if argv[0] == '--version':
 
1090
        get_cmd_object('version').run_argv_aliases([])
 
1091
        return 0
 
1092
 
 
1093
    alias_argv = None
 
1094
 
 
1095
    if not opt_no_aliases:
 
1096
        alias_argv = get_alias(argv[0])
 
1097
        if alias_argv:
 
1098
            argv[0] = alias_argv.pop(0)
 
1099
 
 
1100
    cmd = argv.pop(0)
 
1101
    cmd_obj = get_cmd_object(cmd, plugins_override=not opt_builtin)
 
1102
    run = cmd_obj.run_argv_aliases
 
1103
    run_argv = [argv, alias_argv]
 
1104
 
 
1105
    try:
 
1106
        # We can be called recursively (tests for example), but we don't want
 
1107
        # the verbosity level to propagate.
 
1108
        saved_verbosity_level = option._verbosity_level
 
1109
        option._verbosity_level = 0
 
1110
        if opt_lsprof:
 
1111
            if opt_coverage_dir:
 
1112
                trace.warning(
 
1113
                    '--coverage ignored, because --lsprof is in use.')
 
1114
            ret = apply_lsprofiled(opt_lsprof_file, run, *run_argv)
 
1115
        elif opt_profile:
 
1116
            if opt_coverage_dir:
 
1117
                trace.warning(
 
1118
                    '--coverage ignored, because --profile is in use.')
 
1119
            ret = apply_profiled(run, *run_argv)
 
1120
        elif opt_coverage_dir:
 
1121
            ret = apply_coveraged(opt_coverage_dir, run, *run_argv)
 
1122
        else:
 
1123
            ret = run(*run_argv)
 
1124
        return ret or 0
 
1125
    finally:
 
1126
        # reset, in case we may do other commands later within the same
 
1127
        # process. Commands that want to execute sub-commands must propagate
 
1128
        # --verbose in their own way.
 
1129
        if 'memory' in debug.debug_flags:
 
1130
            trace.debug_memory('Process status after command:', short=False)
 
1131
        option._verbosity_level = saved_verbosity_level
 
1132
 
 
1133
 
 
1134
def display_command(func):
 
1135
    """Decorator that suppresses pipe/interrupt errors."""
 
1136
    def ignore_pipe(*args, **kwargs):
 
1137
        try:
 
1138
            result = func(*args, **kwargs)
 
1139
            sys.stdout.flush()
 
1140
            return result
 
1141
        except IOError, e:
 
1142
            if getattr(e, 'errno', None) is None:
 
1143
                raise
 
1144
            if e.errno != errno.EPIPE:
 
1145
                # Win32 raises IOError with errno=0 on a broken pipe
 
1146
                if sys.platform != 'win32' or (e.errno not in (0, errno.EINVAL)):
 
1147
                    raise
 
1148
            pass
 
1149
        except KeyboardInterrupt:
 
1150
            pass
 
1151
    return ignore_pipe
 
1152
 
 
1153
 
 
1154
def install_bzr_command_hooks():
 
1155
    """Install the hooks to supply bzr's own commands."""
 
1156
    if _list_bzr_commands in Command.hooks["list_commands"]:
 
1157
        return
 
1158
    Command.hooks.install_named_hook("list_commands", _list_bzr_commands,
 
1159
        "bzr commands")
 
1160
    Command.hooks.install_named_hook("get_command", _get_bzr_command,
 
1161
        "bzr commands")
 
1162
    Command.hooks.install_named_hook("get_command", _get_plugin_command,
 
1163
        "bzr plugin commands")
 
1164
    Command.hooks.install_named_hook("get_command", _get_external_command,
 
1165
        "bzr external command lookup")
 
1166
    Command.hooks.install_named_hook("get_missing_command", _try_plugin_provider,
 
1167
        "bzr plugin-provider-db check")
 
1168
 
 
1169
 
 
1170
 
 
1171
def _specified_or_unicode_argv(argv):
 
1172
    # For internal or testing use, argv can be passed.  Otherwise, get it from
 
1173
    # the process arguments in a unicode-safe way.
 
1174
    if argv is None:
 
1175
        return osutils.get_unicode_argv()
 
1176
    else:
 
1177
        new_argv = []
 
1178
        try:
 
1179
            # ensure all arguments are unicode strings
 
1180
            for a in argv[1:]:
 
1181
                if isinstance(a, unicode):
 
1182
                    new_argv.append(a)
 
1183
                else:
 
1184
                    new_argv.append(a.decode('ascii'))
 
1185
        except UnicodeDecodeError:
 
1186
            raise errors.BzrError("argv should be list of unicode strings.")
 
1187
        return new_argv
 
1188
 
 
1189
 
 
1190
def main(argv=None):
 
1191
    """Main entry point of command-line interface.
 
1192
 
 
1193
    Typically `bzrlib.initialize` should be called first.
 
1194
 
 
1195
    :param argv: list of unicode command-line arguments similar to sys.argv.
 
1196
        argv[0] is script name usually, it will be ignored.
 
1197
        Don't pass here sys.argv because this list contains plain strings
 
1198
        and not unicode; pass None instead.
 
1199
 
 
1200
    :return: exit code of bzr command.
 
1201
    """
 
1202
    argv = _specified_or_unicode_argv(argv)
 
1203
    _register_builtin_commands()
 
1204
    ret = run_bzr_catch_errors(argv)
 
1205
    bzrlib.ui.ui_factory.log_transport_activity(
 
1206
        display=('bytes' in debug.debug_flags))
 
1207
    trace.mutter("return code %d", ret)
 
1208
    return ret
 
1209
 
 
1210
 
 
1211
def run_bzr_catch_errors(argv):
 
1212
    """Run a bzr command with parameters as described by argv.
 
1213
 
 
1214
    This function assumed that that UI layer is setup, that symbol deprecations
 
1215
    are already applied, and that unicode decoding has already been performed on argv.
 
1216
    """
 
1217
    # done here so that they're covered for every test run
 
1218
    install_bzr_command_hooks()
 
1219
    return exception_to_return_code(run_bzr, argv)
 
1220
 
 
1221
 
 
1222
def run_bzr_catch_user_errors(argv):
 
1223
    """Run bzr and report user errors, but let internal errors propagate.
 
1224
 
 
1225
    This is used for the test suite, and might be useful for other programs
 
1226
    that want to wrap the commandline interface.
 
1227
    """
 
1228
    # done here so that they're covered for every test run
 
1229
    install_bzr_command_hooks()
 
1230
    try:
 
1231
        return run_bzr(argv)
769
1232
    except Exception, e:
770
 
        log_error('bzr: exception: %s\n' % e)
771
 
        log_error('    see .bzr.log for details\n')
772
 
        traceback.print_exc(None, bzrlib.trace._tracefile)
773
 
        traceback.print_exc(None, sys.stderr)
774
 
        return 1
775
 
 
776
 
    # TODO: Maybe nicer handling of IOError?
777
 
 
778
 
 
779
 
 
780
 
if __name__ == '__main__':
781
 
    sys.exit(main(sys.argv))
782
 
    ##import profile
783
 
    ##profile.run('main(sys.argv)')
784
 
    
 
1233
        if (isinstance(e, (OSError, IOError))
 
1234
            or not getattr(e, 'internal_error', True)):
 
1235
            trace.report_exception(sys.exc_info(), sys.stderr)
 
1236
            return 3
 
1237
        else:
 
1238
            raise
 
1239
 
 
1240
 
 
1241
class HelpCommandIndex(object):
 
1242
    """A index for bzr help that returns commands."""
 
1243
 
 
1244
    def __init__(self):
 
1245
        self.prefix = 'commands/'
 
1246
 
 
1247
    def get_topics(self, topic):
 
1248
        """Search for topic amongst commands.
 
1249
 
 
1250
        :param topic: A topic to search for.
 
1251
        :return: A list which is either empty or contains a single
 
1252
            Command entry.
 
1253
        """
 
1254
        if topic and topic.startswith(self.prefix):
 
1255
            topic = topic[len(self.prefix):]
 
1256
        try:
 
1257
            cmd = _get_cmd_object(topic, check_missing=False)
 
1258
        except KeyError:
 
1259
            return []
 
1260
        else:
 
1261
            return [cmd]
 
1262
 
 
1263
 
 
1264
class Provider(object):
 
1265
    '''Generic class to be overriden by plugins'''
 
1266
 
 
1267
    def plugin_for_command(self, cmd_name):
 
1268
        '''Takes a command and returns the information for that plugin
 
1269
 
 
1270
        :return: A dictionary with all the available information
 
1271
        for the requested plugin
 
1272
        '''
 
1273
        raise NotImplementedError
 
1274
 
 
1275
 
 
1276
class ProvidersRegistry(registry.Registry):
 
1277
    '''This registry exists to allow other providers to exist'''
 
1278
 
 
1279
    def __iter__(self):
 
1280
        for key, provider in self.iteritems():
 
1281
            yield provider
 
1282
 
 
1283
command_providers_registry = ProvidersRegistry()