/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/builtins.py

Switched to build_tree instead of revert

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/builtins.py
 
1
# Copyright (C) 2004, 2005 by Canonical Ltd
 
2
 
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
 
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
 
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
"""builtin bzr commands"""
 
18
 
 
19
 
 
20
import errno
 
21
import os
 
22
from shutil import rmtree
 
23
import sys
 
24
 
 
25
import bzrlib
 
26
import bzrlib.branch
 
27
from bzrlib.branch import Branch
 
28
import bzrlib.bzrdir as bzrdir
 
29
from bzrlib.commands import Command, display_command
 
30
from bzrlib.revision import common_ancestor
 
31
import bzrlib.errors as errors
 
32
from bzrlib.errors import (BzrError, BzrCheckError, BzrCommandError, 
 
33
                           NotBranchError, DivergedBranches, NotConflicted,
 
34
                           NoSuchFile, NoWorkingTree, FileInWrongBranch)
 
35
from bzrlib.log import show_one_log
 
36
from bzrlib.merge import Merge3Merger
 
37
from bzrlib.option import Option
 
38
from bzrlib.revisionspec import RevisionSpec
 
39
import bzrlib.trace
 
40
from bzrlib.trace import mutter, note, log_error, warning, is_quiet
 
41
from bzrlib.transport.local import LocalTransport
 
42
from bzrlib.workingtree import WorkingTree
 
43
 
 
44
 
 
45
def tree_files(file_list, default_branch=u'.'):
 
46
    try:
 
47
        return internal_tree_files(file_list, default_branch)
 
48
    except FileInWrongBranch, e:
 
49
        raise BzrCommandError("%s is not in the same branch as %s" %
 
50
                             (e.path, file_list[0]))
 
51
 
 
52
def internal_tree_files(file_list, default_branch=u'.'):
 
53
    """\
 
54
    Return a branch and list of branch-relative paths.
 
55
    If supplied file_list is empty or None, the branch default will be used,
 
56
    and returned file_list will match the original.
 
57
    """
 
58
    if file_list is None or len(file_list) == 0:
 
59
        return WorkingTree.open_containing(default_branch)[0], file_list
 
60
    tree = WorkingTree.open_containing(file_list[0])[0]
 
61
    new_list = []
 
62
    for filename in file_list:
 
63
        try:
 
64
            new_list.append(tree.relpath(filename))
 
65
        except errors.PathNotChild:
 
66
            raise FileInWrongBranch(tree.branch, filename)
 
67
    return tree, new_list
 
68
 
 
69
 
 
70
# TODO: Make sure no commands unconditionally use the working directory as a
 
71
# branch.  If a filename argument is used, the first of them should be used to
 
72
# specify the branch.  (Perhaps this can be factored out into some kind of
 
73
# Argument class, representing a file in a branch, where the first occurrence
 
74
# opens the branch?)
 
75
 
 
76
class cmd_status(Command):
 
77
    """Display status summary.
 
78
 
 
79
    This reports on versioned and unknown files, reporting them
 
80
    grouped by state.  Possible states are:
 
81
 
 
82
    added
 
83
        Versioned in the working copy but not in the previous revision.
 
84
 
 
85
    removed
 
86
        Versioned in the previous revision but removed or deleted
 
87
        in the working copy.
 
88
 
 
89
    renamed
 
90
        Path of this file changed from the previous revision;
 
91
        the text may also have changed.  This includes files whose
 
92
        parent directory was renamed.
 
93
 
 
94
    modified
 
95
        Text has changed since the previous revision.
 
96
 
 
97
    unchanged
 
98
        Nothing about this file has changed since the previous revision.
 
99
        Only shown with --all.
 
100
 
 
101
    unknown
 
102
        Not versioned and not matching an ignore pattern.
 
103
 
 
104
    To see ignored files use 'bzr ignored'.  For details in the
 
105
    changes to file texts, use 'bzr diff'.
 
106
 
 
107
    If no arguments are specified, the status of the entire working
 
108
    directory is shown.  Otherwise, only the status of the specified
 
109
    files or directories is reported.  If a directory is given, status
 
110
    is reported for everything inside that directory.
 
111
 
 
112
    If a revision argument is given, the status is calculated against
 
113
    that revision, or between two revisions if two are provided.
 
114
    """
 
115
    
 
116
    # TODO: --no-recurse, --recurse options
 
117
    
 
118
    takes_args = ['file*']
 
119
    takes_options = ['all', 'show-ids', 'revision']
 
120
    aliases = ['st', 'stat']
 
121
    
 
122
    @display_command
 
123
    def run(self, all=False, show_ids=False, file_list=None, revision=None):
 
124
        tree, file_list = tree_files(file_list)
 
125
            
 
126
        from bzrlib.status import show_status
 
127
        show_status(tree.branch, show_unchanged=all, show_ids=show_ids,
 
128
                    specific_files=file_list, revision=revision)
 
129
 
 
130
 
 
131
class cmd_cat_revision(Command):
 
132
    """Write out metadata for a revision.
 
133
    
 
134
    The revision to print can either be specified by a specific
 
135
    revision identifier, or you can use --revision.
 
136
    """
 
137
 
 
138
    hidden = True
 
139
    takes_args = ['revision_id?']
 
140
    takes_options = ['revision']
 
141
    
 
142
    @display_command
 
143
    def run(self, revision_id=None, revision=None):
 
144
 
 
145
        if revision_id is not None and revision is not None:
 
146
            raise BzrCommandError('You can only supply one of revision_id or --revision')
 
147
        if revision_id is None and revision is None:
 
148
            raise BzrCommandError('You must supply either --revision or a revision_id')
 
149
        b = WorkingTree.open_containing(u'.')[0].branch
 
150
        if revision_id is not None:
 
151
            sys.stdout.write(b.repository.get_revision_xml(revision_id))
 
152
        elif revision is not None:
 
153
            for rev in revision:
 
154
                if rev is None:
 
155
                    raise BzrCommandError('You cannot specify a NULL revision.')
 
156
                revno, rev_id = rev.in_history(b)
 
157
                sys.stdout.write(b.repository.get_revision_xml(rev_id))
 
158
    
 
159
 
 
160
class cmd_revno(Command):
 
161
    """Show current revision number.
 
162
 
 
163
    This is equal to the number of revisions on this branch."""
 
164
    takes_args = ['location?']
 
165
    @display_command
 
166
    def run(self, location=u'.'):
 
167
        print Branch.open_containing(location)[0].revno()
 
168
 
 
169
 
 
170
class cmd_revision_info(Command):
 
171
    """Show revision number and revision id for a given revision identifier.
 
172
    """
 
173
    hidden = True
 
174
    takes_args = ['revision_info*']
 
175
    takes_options = ['revision']
 
176
    @display_command
 
177
    def run(self, revision=None, revision_info_list=[]):
 
178
 
 
179
        revs = []
 
180
        if revision is not None:
 
181
            revs.extend(revision)
 
182
        if revision_info_list is not None:
 
183
            for rev in revision_info_list:
 
184
                revs.append(RevisionSpec(rev))
 
185
        if len(revs) == 0:
 
186
            raise BzrCommandError('You must supply a revision identifier')
 
187
 
 
188
        b = WorkingTree.open_containing(u'.')[0].branch
 
189
 
 
190
        for rev in revs:
 
191
            revinfo = rev.in_history(b)
 
192
            if revinfo.revno is None:
 
193
                print '     %s' % revinfo.rev_id
 
194
            else:
 
195
                print '%4d %s' % (revinfo.revno, revinfo.rev_id)
 
196
 
 
197
    
 
198
class cmd_add(Command):
 
199
    """Add specified files or directories.
 
200
 
 
201
    In non-recursive mode, all the named items are added, regardless
 
202
    of whether they were previously ignored.  A warning is given if
 
203
    any of the named files are already versioned.
 
204
 
 
205
    In recursive mode (the default), files are treated the same way
 
206
    but the behaviour for directories is different.  Directories that
 
207
    are already versioned do not give a warning.  All directories,
 
208
    whether already versioned or not, are searched for files or
 
209
    subdirectories that are neither versioned or ignored, and these
 
210
    are added.  This search proceeds recursively into versioned
 
211
    directories.  If no names are given '.' is assumed.
 
212
 
 
213
    Therefore simply saying 'bzr add' will version all files that
 
214
    are currently unknown.
 
215
 
 
216
    Adding a file whose parent directory is not versioned will
 
217
    implicitly add the parent, and so on up to the root. This means
 
218
    you should never need to explictly add a directory, they'll just
 
219
    get added when you add a file in the directory.
 
220
 
 
221
    --dry-run will show which files would be added, but not actually 
 
222
    add them.
 
223
    """
 
224
    takes_args = ['file*']
 
225
    takes_options = ['no-recurse', 'dry-run', 'verbose']
 
226
 
 
227
    def run(self, file_list, no_recurse=False, dry_run=False, verbose=False):
 
228
        import bzrlib.add
 
229
 
 
230
        if dry_run:
 
231
            if is_quiet():
 
232
                # This is pointless, but I'd rather not raise an error
 
233
                action = bzrlib.add.add_action_null
 
234
            else:
 
235
                action = bzrlib.add.add_action_print
 
236
        elif is_quiet():
 
237
            action = bzrlib.add.add_action_add
 
238
        else:
 
239
            action = bzrlib.add.add_action_add_and_print
 
240
 
 
241
        added, ignored = bzrlib.add.smart_add(file_list, not no_recurse, 
 
242
                                              action)
 
243
        if len(ignored) > 0:
 
244
            for glob in sorted(ignored.keys()):
 
245
                match_len = len(ignored[glob])
 
246
                if verbose:
 
247
                    for path in ignored[glob]:
 
248
                        print "ignored %s matching \"%s\"" % (path, glob)
 
249
                else:
 
250
                    print "ignored %d file(s) matching \"%s\"" % (match_len,
 
251
                                                              glob)
 
252
            print "If you wish to add some of these files, please add them"\
 
253
                " by name."
 
254
 
 
255
 
 
256
class cmd_mkdir(Command):
 
257
    """Create a new versioned directory.
 
258
 
 
259
    This is equivalent to creating the directory and then adding it.
 
260
    """
 
261
    takes_args = ['dir+']
 
262
 
 
263
    def run(self, dir_list):
 
264
        for d in dir_list:
 
265
            os.mkdir(d)
 
266
            wt, dd = WorkingTree.open_containing(d)
 
267
            wt.add([dd])
 
268
            print 'added', d
 
269
 
 
270
 
 
271
class cmd_relpath(Command):
 
272
    """Show path of a file relative to root"""
 
273
    takes_args = ['filename']
 
274
    hidden = True
 
275
    
 
276
    @display_command
 
277
    def run(self, filename):
 
278
        tree, relpath = WorkingTree.open_containing(filename)
 
279
        print relpath
 
280
 
 
281
 
 
282
class cmd_inventory(Command):
 
283
    """Show inventory of the current working copy or a revision.
 
284
 
 
285
    It is possible to limit the output to a particular entry
 
286
    type using the --kind option.  For example; --kind file.
 
287
    """
 
288
    takes_options = ['revision', 'show-ids', 'kind']
 
289
    
 
290
    @display_command
 
291
    def run(self, revision=None, show_ids=False, kind=None):
 
292
        if kind and kind not in ['file', 'directory', 'symlink']:
 
293
            raise BzrCommandError('invalid kind specified')
 
294
        tree = WorkingTree.open_containing(u'.')[0]
 
295
        if revision is None:
 
296
            inv = tree.read_working_inventory()
 
297
        else:
 
298
            if len(revision) > 1:
 
299
                raise BzrCommandError('bzr inventory --revision takes'
 
300
                    ' exactly one revision identifier')
 
301
            inv = tree.branch.repository.get_revision_inventory(
 
302
                revision[0].in_history(tree.branch).rev_id)
 
303
 
 
304
        for path, entry in inv.entries():
 
305
            if kind and kind != entry.kind:
 
306
                continue
 
307
            if show_ids:
 
308
                print '%-50s %s' % (path, entry.file_id)
 
309
            else:
 
310
                print path
 
311
 
 
312
 
 
313
class cmd_move(Command):
 
314
    """Move files to a different directory.
 
315
 
 
316
    examples:
 
317
        bzr move *.txt doc
 
318
 
 
319
    The destination must be a versioned directory in the same branch.
 
320
    """
 
321
    takes_args = ['source$', 'dest']
 
322
    def run(self, source_list, dest):
 
323
        tree, source_list = tree_files(source_list)
 
324
        # TODO: glob expansion on windows?
 
325
        tree.move(source_list, tree.relpath(dest))
 
326
 
 
327
 
 
328
class cmd_rename(Command):
 
329
    """Change the name of an entry.
 
330
 
 
331
    examples:
 
332
      bzr rename frob.c frobber.c
 
333
      bzr rename src/frob.c lib/frob.c
 
334
 
 
335
    It is an error if the destination name exists.
 
336
 
 
337
    See also the 'move' command, which moves files into a different
 
338
    directory without changing their name.
 
339
    """
 
340
    # TODO: Some way to rename multiple files without invoking 
 
341
    # bzr for each one?"""
 
342
    takes_args = ['from_name', 'to_name']
 
343
    
 
344
    def run(self, from_name, to_name):
 
345
        tree, (from_name, to_name) = tree_files((from_name, to_name))
 
346
        tree.rename_one(from_name, to_name)
 
347
 
 
348
 
 
349
class cmd_mv(Command):
 
350
    """Move or rename a file.
 
351
 
 
352
    usage:
 
353
        bzr mv OLDNAME NEWNAME
 
354
        bzr mv SOURCE... DESTINATION
 
355
 
 
356
    If the last argument is a versioned directory, all the other names
 
357
    are moved into it.  Otherwise, there must be exactly two arguments
 
358
    and the file is changed to a new name, which must not already exist.
 
359
 
 
360
    Files cannot be moved between branches.
 
361
    """
 
362
    takes_args = ['names*']
 
363
    def run(self, names_list):
 
364
        if len(names_list) < 2:
 
365
            raise BzrCommandError("missing file argument")
 
366
        tree, rel_names = tree_files(names_list)
 
367
        
 
368
        if os.path.isdir(names_list[-1]):
 
369
            # move into existing directory
 
370
            for pair in tree.move(rel_names[:-1], rel_names[-1]):
 
371
                print "%s => %s" % pair
 
372
        else:
 
373
            if len(names_list) != 2:
 
374
                raise BzrCommandError('to mv multiple files the destination '
 
375
                                      'must be a versioned directory')
 
376
            tree.rename_one(rel_names[0], rel_names[1])
 
377
            print "%s => %s" % (rel_names[0], rel_names[1])
 
378
            
 
379
    
 
380
class cmd_pull(Command):
 
381
    """Pull any changes from another branch into the current one.
 
382
 
 
383
    If there is no default location set, the first pull will set it.  After
 
384
    that, you can omit the location to use the default.  To change the
 
385
    default, use --remember.
 
386
 
 
387
    This command only works on branches that have not diverged.  Branches are
 
388
    considered diverged if both branches have had commits without first
 
389
    pulling from the other.
 
390
 
 
391
    If branches have diverged, you can use 'bzr merge' to pull the text changes
 
392
    from one into the other.  Once one branch has merged, the other should
 
393
    be able to pull it again.
 
394
 
 
395
    If you want to forget your local changes and just update your branch to
 
396
    match the remote one, use --overwrite.
 
397
    """
 
398
    takes_options = ['remember', 'overwrite', 'revision', 'verbose']
 
399
    takes_args = ['location?']
 
400
 
 
401
    def run(self, location=None, remember=False, overwrite=False, revision=None, verbose=False):
 
402
        # FIXME: too much stuff is in the command class        
 
403
        tree_to = WorkingTree.open_containing(u'.')[0]
 
404
        stored_loc = tree_to.branch.get_parent()
 
405
        if location is None:
 
406
            if stored_loc is None:
 
407
                raise BzrCommandError("No pull location known or specified.")
 
408
            else:
 
409
                print "Using saved location: %s" % stored_loc
 
410
                location = stored_loc
 
411
 
 
412
        br_from = Branch.open(location)
 
413
        br_to = tree_to.branch
 
414
 
 
415
        if revision is None:
 
416
            rev_id = None
 
417
        elif len(revision) == 1:
 
418
            rev_id = revision[0].in_history(br_from).rev_id
 
419
        else:
 
420
            raise BzrCommandError('bzr pull --revision takes one value.')
 
421
 
 
422
        old_rh = br_to.revision_history()
 
423
        count = tree_to.pull(br_from, overwrite, rev_id)
 
424
 
 
425
        if br_to.get_parent() is None or remember:
 
426
            br_to.set_parent(location)
 
427
        note('%d revision(s) pulled.' % (count,))
 
428
 
 
429
        if verbose:
 
430
            new_rh = tree_to.branch.revision_history()
 
431
            if old_rh != new_rh:
 
432
                # Something changed
 
433
                from bzrlib.log import show_changed_revisions
 
434
                show_changed_revisions(tree_to.branch, old_rh, new_rh)
 
435
 
 
436
 
 
437
class cmd_push(Command):
 
438
    """Push this branch into another branch.
 
439
    
 
440
    The remote branch will not have its working tree populated because this
 
441
    is both expensive, and may not be supported on the remote file system.
 
442
    
 
443
    Some smart servers or protocols *may* put the working tree in place.
 
444
 
 
445
    If there is no default push location set, the first push will set it.
 
446
    After that, you can omit the location to use the default.  To change the
 
447
    default, use --remember.
 
448
 
 
449
    This command only works on branches that have not diverged.  Branches are
 
450
    considered diverged if the branch being pushed to is not an older version
 
451
    of this branch.
 
452
 
 
453
    If branches have diverged, you can use 'bzr push --overwrite' to replace
 
454
    the other branch completely.
 
455
    
 
456
    If you want to ensure you have the different changes in the other branch,
 
457
    do a merge (see bzr help merge) from the other branch, and commit that
 
458
    before doing a 'push --overwrite'.
 
459
    """
 
460
    takes_options = ['remember', 'overwrite', 
 
461
                     Option('create-prefix', 
 
462
                            help='Create the path leading up to the branch '
 
463
                                 'if it does not already exist')]
 
464
    takes_args = ['location?']
 
465
 
 
466
    def run(self, location=None, remember=False, overwrite=False,
 
467
            create_prefix=False, verbose=False):
 
468
        # FIXME: Way too big!  Put this into a function called from the
 
469
        # command.
 
470
        from bzrlib.transport import get_transport
 
471
        
 
472
        tree_from = WorkingTree.open_containing(u'.')[0]
 
473
        br_from = tree_from.branch
 
474
        stored_loc = tree_from.branch.get_push_location()
 
475
        if location is None:
 
476
            if stored_loc is None:
 
477
                raise BzrCommandError("No push location known or specified.")
 
478
            else:
 
479
                print "Using saved location: %s" % stored_loc
 
480
                location = stored_loc
 
481
        try:
 
482
            br_to = Branch.open(location)
 
483
        except NotBranchError:
 
484
            # create a branch.
 
485
            transport = get_transport(location).clone('..')
 
486
            if not create_prefix:
 
487
                try:
 
488
                    transport.mkdir(transport.relpath(location))
 
489
                except NoSuchFile:
 
490
                    raise BzrCommandError("Parent directory of %s "
 
491
                                          "does not exist." % location)
 
492
            else:
 
493
                current = transport.base
 
494
                needed = [(transport, transport.relpath(location))]
 
495
                while needed:
 
496
                    try:
 
497
                        transport, relpath = needed[-1]
 
498
                        transport.mkdir(relpath)
 
499
                        needed.pop()
 
500
                    except NoSuchFile:
 
501
                        new_transport = transport.clone('..')
 
502
                        needed.append((new_transport,
 
503
                                       new_transport.relpath(transport.base)))
 
504
                        if new_transport.base == transport.base:
 
505
                            raise BzrCommandError("Could not creeate "
 
506
                                                  "path prefix.")
 
507
            if isinstance(transport, LocalTransport):
 
508
                br_to = WorkingTree.create_standalone(location).branch
 
509
            else:
 
510
                br_to = Branch.create(location)
 
511
        old_rh = br_to.revision_history()
 
512
        try:
 
513
            try:
 
514
                tree_to = br_to.working_tree()
 
515
            except NoWorkingTree:
 
516
                # TODO: This should be updated for branches which don't have a
 
517
                # working tree, as opposed to ones where we just couldn't 
 
518
                # update the tree.
 
519
                warning('Unable to update the working tree of: %s' % (br_to.base,))
 
520
                count = br_to.pull(br_from, overwrite)
 
521
            else:
 
522
                count = tree_to.pull(br_from, overwrite)
 
523
        except DivergedBranches:
 
524
            raise BzrCommandError("These branches have diverged."
 
525
                                  "  Try a merge then push with overwrite.")
 
526
        if br_from.get_push_location() is None or remember:
 
527
            br_from.set_push_location(location)
 
528
        note('%d revision(s) pushed.' % (count,))
 
529
 
 
530
        if verbose:
 
531
            new_rh = br_to.revision_history()
 
532
            if old_rh != new_rh:
 
533
                # Something changed
 
534
                from bzrlib.log import show_changed_revisions
 
535
                show_changed_revisions(br_to, old_rh, new_rh)
 
536
 
 
537
 
 
538
class cmd_branch(Command):
 
539
    """Create a new copy of a branch.
 
540
 
 
541
    If the TO_LOCATION is omitted, the last component of the FROM_LOCATION will
 
542
    be used.  In other words, "branch ../foo/bar" will attempt to create ./bar.
 
543
 
 
544
    To retrieve the branch as of a particular revision, supply the --revision
 
545
    parameter, as in "branch foo/bar -r 5".
 
546
 
 
547
    --basis is to speed up branching from remote branches.  When specified, it
 
548
    copies all the file-contents, inventory and revision data from the basis
 
549
    branch before copying anything from the remote branch.
 
550
    """
 
551
    takes_args = ['from_location', 'to_location?']
 
552
    takes_options = ['revision', 'basis']
 
553
    aliases = ['get', 'clone']
 
554
 
 
555
    def run(self, from_location, to_location=None, revision=None, basis=None):
 
556
        if revision is None:
 
557
            revision = [None]
 
558
        elif len(revision) > 1:
 
559
            raise BzrCommandError(
 
560
                'bzr branch --revision takes exactly 1 revision value')
 
561
        try:
 
562
            br_from = Branch.open(from_location)
 
563
        except OSError, e:
 
564
            if e.errno == errno.ENOENT:
 
565
                raise BzrCommandError('Source location "%s" does not'
 
566
                                      ' exist.' % to_location)
 
567
            else:
 
568
                raise
 
569
        br_from.lock_read()
 
570
        try:
 
571
            if basis is not None:
 
572
                basis_dir = bzrdir.BzrDir.open_containing(basis)[0]
 
573
            else:
 
574
                basis_dir = None
 
575
            if len(revision) == 1 and revision[0] is not None:
 
576
                revision_id = revision[0].in_history(br_from)[1]
 
577
            else:
 
578
                # FIXME - wt.last_revision, fallback to branch, fall back to
 
579
                # None or perhaps NULL_REVISION to mean copy nothing
 
580
                # RBC 20060209
 
581
                revision_id = br_from.last_revision()
 
582
            if to_location is None:
 
583
                to_location = os.path.basename(from_location.rstrip("/\\"))
 
584
                name = None
 
585
            else:
 
586
                name = os.path.basename(to_location) + '\n'
 
587
            try:
 
588
                os.mkdir(to_location)
 
589
            except OSError, e:
 
590
                if e.errno == errno.EEXIST:
 
591
                    raise BzrCommandError('Target directory "%s" already'
 
592
                                          ' exists.' % to_location)
 
593
                if e.errno == errno.ENOENT:
 
594
                    raise BzrCommandError('Parent of "%s" does not exist.' %
 
595
                                          to_location)
 
596
                else:
 
597
                    raise
 
598
            try:
 
599
                dir = br_from.bzrdir.sprout(to_location, revision_id, basis_dir)
 
600
                branch = dir.open_branch()
 
601
            except bzrlib.errors.NoSuchRevision:
 
602
                rmtree(to_location)
 
603
                msg = "The branch %s has no revision %s." % (from_location, revision[0])
 
604
                raise BzrCommandError(msg)
 
605
            except bzrlib.errors.UnlistableBranch:
 
606
                rmtree(to_location)
 
607
                msg = "The branch %s cannot be used as a --basis"
 
608
                raise BzrCommandError(msg)
 
609
            if name:
 
610
                branch.control_files.put_utf8('branch-name', name)
 
611
 
 
612
            note('Branched %d revision(s).' % branch.revno())
 
613
        finally:
 
614
            br_from.unlock()
 
615
 
 
616
 
 
617
class cmd_checkout(Command):
 
618
    """Create a new checkout of an existing branch.
 
619
 
 
620
    If the TO_LOCATION is omitted, the last component of the BRANCH_LOCATION will
 
621
    be used.  In other words, "checkout ../foo/bar" will attempt to create ./bar.
 
622
 
 
623
    To retrieve the branch as of a particular revision, supply the --revision
 
624
    parameter, as in "checkout foo/bar -r 5". Note that this will be immediately
 
625
    out of date [so you cannot commit] but it may be useful (i.e. to examine old
 
626
    code.)
 
627
 
 
628
    --basis is to speed up checking out from remote branches.  When specified, it
 
629
    uses the inventory and file contents from the basis branch in preference to the
 
630
    branch being checked out. [Not implemented yet.]
 
631
    """
 
632
    takes_args = ['branch_location', 'to_location?']
 
633
    takes_options = ['revision'] # , 'basis']
 
634
 
 
635
    def run(self, branch_location, to_location=None, revision=None, basis=None):
 
636
        if revision is None:
 
637
            revision = [None]
 
638
        elif len(revision) > 1:
 
639
            raise BzrCommandError(
 
640
                'bzr checkout --revision takes exactly 1 revision value')
 
641
        source = Branch.open(branch_location)
 
642
        if len(revision) == 1 and revision[0] is not None:
 
643
            revision_id = revision[0].in_history(source)[1]
 
644
        else:
 
645
            revision_id = None
 
646
        if to_location is None:
 
647
            to_location = os.path.basename(branch_location.rstrip("/\\"))
 
648
        try:
 
649
            os.mkdir(to_location)
 
650
        except OSError, e:
 
651
            if e.errno == errno.EEXIST:
 
652
                raise BzrCommandError('Target directory "%s" already'
 
653
                                      ' exists.' % to_location)
 
654
            if e.errno == errno.ENOENT:
 
655
                raise BzrCommandError('Parent of "%s" does not exist.' %
 
656
                                      to_location)
 
657
            else:
 
658
                raise
 
659
        checkout = bzrdir.BzrDirMetaFormat1().initialize(to_location)
 
660
        bzrlib.branch.BranchReferenceFormat().initialize(checkout, source)
 
661
        checkout.create_workingtree(revision_id)
 
662
 
 
663
 
 
664
class cmd_renames(Command):
 
665
    """Show list of renamed files.
 
666
    """
 
667
    # TODO: Option to show renames between two historical versions.
 
668
 
 
669
    # TODO: Only show renames under dir, rather than in the whole branch.
 
670
    takes_args = ['dir?']
 
671
 
 
672
    @display_command
 
673
    def run(self, dir=u'.'):
 
674
        tree = WorkingTree.open_containing(dir)[0]
 
675
        old_inv = tree.basis_tree().inventory
 
676
        new_inv = tree.read_working_inventory()
 
677
 
 
678
        renames = list(bzrlib.tree.find_renames(old_inv, new_inv))
 
679
        renames.sort()
 
680
        for old_name, new_name in renames:
 
681
            print "%s => %s" % (old_name, new_name)        
 
682
 
 
683
 
 
684
class cmd_update(Command):
 
685
    """Update a tree to have the latest code committed to its branch.
 
686
    
 
687
    This will perform a merge into the working tree, and may generate
 
688
    conflicts. If you have any uncommitted changes, you will still 
 
689
    need to commit them after the update.
 
690
    """
 
691
    takes_args = ['dir?']
 
692
 
 
693
    def run(self, dir='.'):
 
694
        tree = WorkingTree.open_containing(dir)[0]
 
695
        tree.lock_write()
 
696
        try:
 
697
            if tree.last_revision() == tree.branch.last_revision():
 
698
                note("Tree is up to date.")
 
699
                return
 
700
            conflicts = tree.update()
 
701
            note('Updated to revision %d.' %
 
702
                 (tree.branch.revision_id_to_revno(tree.last_revision()),))
 
703
            if conflicts != 0:
 
704
                return 1
 
705
            else:
 
706
                return 0
 
707
        finally:
 
708
            tree.unlock()
 
709
 
 
710
 
 
711
class cmd_info(Command):
 
712
    """Show statistical information about a branch."""
 
713
    takes_args = ['branch?']
 
714
    
 
715
    @display_command
 
716
    def run(self, branch=None):
 
717
        import info
 
718
        b = WorkingTree.open_containing(branch)[0].branch
 
719
        info.show_info(b)
 
720
 
 
721
 
 
722
class cmd_remove(Command):
 
723
    """Make a file unversioned.
 
724
 
 
725
    This makes bzr stop tracking changes to a versioned file.  It does
 
726
    not delete the working copy.
 
727
    """
 
728
    takes_args = ['file+']
 
729
    takes_options = ['verbose']
 
730
    aliases = ['rm']
 
731
    
 
732
    def run(self, file_list, verbose=False):
 
733
        tree, file_list = tree_files(file_list)
 
734
        tree.remove(file_list, verbose=verbose)
 
735
 
 
736
 
 
737
class cmd_file_id(Command):
 
738
    """Print file_id of a particular file or directory.
 
739
 
 
740
    The file_id is assigned when the file is first added and remains the
 
741
    same through all revisions where the file exists, even when it is
 
742
    moved or renamed.
 
743
    """
 
744
    hidden = True
 
745
    takes_args = ['filename']
 
746
    @display_command
 
747
    def run(self, filename):
 
748
        tree, relpath = WorkingTree.open_containing(filename)
 
749
        i = tree.inventory.path2id(relpath)
 
750
        if i == None:
 
751
            raise BzrError("%r is not a versioned file" % filename)
 
752
        else:
 
753
            print i
 
754
 
 
755
 
 
756
class cmd_file_path(Command):
 
757
    """Print path of file_ids to a file or directory.
 
758
 
 
759
    This prints one line for each directory down to the target,
 
760
    starting at the branch root."""
 
761
    hidden = True
 
762
    takes_args = ['filename']
 
763
    @display_command
 
764
    def run(self, filename):
 
765
        tree, relpath = WorkingTree.open_containing(filename)
 
766
        inv = tree.inventory
 
767
        fid = inv.path2id(relpath)
 
768
        if fid == None:
 
769
            raise BzrError("%r is not a versioned file" % filename)
 
770
        for fip in inv.get_idpath(fid):
 
771
            print fip
 
772
 
 
773
 
 
774
class cmd_revision_history(Command):
 
775
    """Display list of revision ids on this branch."""
 
776
    hidden = True
 
777
    @display_command
 
778
    def run(self):
 
779
        branch = WorkingTree.open_containing(u'.')[0].branch
 
780
        for patchid in branch.revision_history():
 
781
            print patchid
 
782
 
 
783
 
 
784
class cmd_ancestry(Command):
 
785
    """List all revisions merged into this branch."""
 
786
    hidden = True
 
787
    @display_command
 
788
    def run(self):
 
789
        tree = WorkingTree.open_containing(u'.')[0]
 
790
        b = tree.branch
 
791
        # FIXME. should be tree.last_revision
 
792
        for revision_id in b.repository.get_ancestry(b.last_revision()):
 
793
            print revision_id
 
794
 
 
795
 
 
796
class cmd_init(Command):
 
797
    """Make a directory into a versioned branch.
 
798
 
 
799
    Use this to create an empty branch, or before importing an
 
800
    existing project.
 
801
 
 
802
    Recipe for importing a tree of files:
 
803
        cd ~/project
 
804
        bzr init
 
805
        bzr add .
 
806
        bzr status
 
807
        bzr commit -m 'imported project'
 
808
    """
 
809
    takes_args = ['location?']
 
810
    def run(self, location=None):
 
811
        from bzrlib.branch import Branch
 
812
        if location is None:
 
813
            location = u'.'
 
814
        else:
 
815
            # The path has to exist to initialize a
 
816
            # branch inside of it.
 
817
            # Just using os.mkdir, since I don't
 
818
            # believe that we want to create a bunch of
 
819
            # locations if the user supplies an extended path
 
820
            if not os.path.exists(location):
 
821
                os.mkdir(location)
 
822
        WorkingTree.create_standalone(location)
 
823
 
 
824
 
 
825
class cmd_diff(Command):
 
826
    """Show differences in working tree.
 
827
    
 
828
    If files are listed, only the changes in those files are listed.
 
829
    Otherwise, all changes for the tree are listed.
 
830
 
 
831
    examples:
 
832
        bzr diff
 
833
        bzr diff -r1
 
834
        bzr diff -r1..2
 
835
    """
 
836
    # TODO: Allow diff across branches.
 
837
    # TODO: Option to use external diff command; could be GNU diff, wdiff,
 
838
    #       or a graphical diff.
 
839
 
 
840
    # TODO: Python difflib is not exactly the same as unidiff; should
 
841
    #       either fix it up or prefer to use an external diff.
 
842
 
 
843
    # TODO: If a directory is given, diff everything under that.
 
844
 
 
845
    # TODO: Selected-file diff is inefficient and doesn't show you
 
846
    #       deleted files.
 
847
 
 
848
    # TODO: This probably handles non-Unix newlines poorly.
 
849
    
 
850
    takes_args = ['file*']
 
851
    takes_options = ['revision', 'diff-options']
 
852
    aliases = ['di', 'dif']
 
853
 
 
854
    @display_command
 
855
    def run(self, revision=None, file_list=None, diff_options=None):
 
856
        from bzrlib.diff import show_diff
 
857
        try:
 
858
            tree, file_list = internal_tree_files(file_list)
 
859
            b = None
 
860
            b2 = None
 
861
        except FileInWrongBranch:
 
862
            if len(file_list) != 2:
 
863
                raise BzrCommandError("Files are in different branches")
 
864
 
 
865
            b, file1 = Branch.open_containing(file_list[0])
 
866
            b2, file2 = Branch.open_containing(file_list[1])
 
867
            if file1 != "" or file2 != "":
 
868
                # FIXME diff those two files. rbc 20051123
 
869
                raise BzrCommandError("Files are in different branches")
 
870
            file_list = None
 
871
        if revision is not None:
 
872
            if b2 is not None:
 
873
                raise BzrCommandError("Can't specify -r with two branches")
 
874
            if (len(revision) == 1) or (revision[1].spec is None):
 
875
                return show_diff(tree.branch, revision[0], specific_files=file_list,
 
876
                                 external_diff_options=diff_options)
 
877
            elif len(revision) == 2:
 
878
                return show_diff(tree.branch, revision[0], specific_files=file_list,
 
879
                                 external_diff_options=diff_options,
 
880
                                 revision2=revision[1])
 
881
            else:
 
882
                raise BzrCommandError('bzr diff --revision takes exactly one or two revision identifiers')
 
883
        else:
 
884
            if b is not None:
 
885
                return show_diff(b, None, specific_files=file_list,
 
886
                                 external_diff_options=diff_options, b2=b2)
 
887
            else:
 
888
                return show_diff(tree.branch, None, specific_files=file_list,
 
889
                                 external_diff_options=diff_options)
 
890
 
 
891
 
 
892
class cmd_deleted(Command):
 
893
    """List files deleted in the working tree.
 
894
    """
 
895
    # TODO: Show files deleted since a previous revision, or
 
896
    # between two revisions.
 
897
    # TODO: Much more efficient way to do this: read in new
 
898
    # directories with readdir, rather than stating each one.  Same
 
899
    # level of effort but possibly much less IO.  (Or possibly not,
 
900
    # if the directories are very large...)
 
901
    @display_command
 
902
    def run(self, show_ids=False):
 
903
        tree = WorkingTree.open_containing(u'.')[0]
 
904
        old = tree.basis_tree()
 
905
        for path, ie in old.inventory.iter_entries():
 
906
            if not tree.has_id(ie.file_id):
 
907
                if show_ids:
 
908
                    print '%-50s %s' % (path, ie.file_id)
 
909
                else:
 
910
                    print path
 
911
 
 
912
 
 
913
class cmd_modified(Command):
 
914
    """List files modified in working tree."""
 
915
    hidden = True
 
916
    @display_command
 
917
    def run(self):
 
918
        from bzrlib.delta import compare_trees
 
919
 
 
920
        tree = WorkingTree.open_containing(u'.')[0]
 
921
        td = compare_trees(tree.basis_tree(), tree)
 
922
 
 
923
        for path, id, kind, text_modified, meta_modified in td.modified:
 
924
            print path
 
925
 
 
926
 
 
927
 
 
928
class cmd_added(Command):
 
929
    """List files added in working tree."""
 
930
    hidden = True
 
931
    @display_command
 
932
    def run(self):
 
933
        wt = WorkingTree.open_containing(u'.')[0]
 
934
        basis_inv = wt.basis_tree().inventory
 
935
        inv = wt.inventory
 
936
        for file_id in inv:
 
937
            if file_id in basis_inv:
 
938
                continue
 
939
            path = inv.id2path(file_id)
 
940
            if not os.access(bzrlib.osutils.abspath(path), os.F_OK):
 
941
                continue
 
942
            print path
 
943
                
 
944
        
 
945
 
 
946
class cmd_root(Command):
 
947
    """Show the tree root directory.
 
948
 
 
949
    The root is the nearest enclosing directory with a .bzr control
 
950
    directory."""
 
951
    takes_args = ['filename?']
 
952
    @display_command
 
953
    def run(self, filename=None):
 
954
        """Print the branch root."""
 
955
        tree = WorkingTree.open_containing(filename)[0]
 
956
        print tree.basedir
 
957
 
 
958
 
 
959
class cmd_log(Command):
 
960
    """Show log of this branch.
 
961
 
 
962
    To request a range of logs, you can use the command -r begin..end
 
963
    -r revision requests a specific revision, -r ..end or -r begin.. are
 
964
    also valid.
 
965
    """
 
966
 
 
967
    # TODO: Make --revision support uuid: and hash: [future tag:] notation.
 
968
 
 
969
    takes_args = ['filename?']
 
970
    takes_options = [Option('forward', 
 
971
                            help='show from oldest to newest'),
 
972
                     'timezone', 'verbose', 
 
973
                     'show-ids', 'revision',
 
974
                     'line', 'long', 
 
975
                     Option('message',
 
976
                            help='show revisions whose message matches this regexp',
 
977
                            type=str),
 
978
                     'short',
 
979
                     ]
 
980
    @display_command
 
981
    def run(self, filename=None, timezone='original',
 
982
            verbose=False,
 
983
            show_ids=False,
 
984
            forward=False,
 
985
            revision=None,
 
986
            message=None,
 
987
            long=False,
 
988
            short=False,
 
989
            line=False):
 
990
        from bzrlib.log import log_formatter, show_log
 
991
        import codecs
 
992
        assert message is None or isinstance(message, basestring), \
 
993
            "invalid message argument %r" % message
 
994
        direction = (forward and 'forward') or 'reverse'
 
995
        
 
996
        # log everything
 
997
        file_id = None
 
998
        if filename:
 
999
            # find the file id to log:
 
1000
 
 
1001
            dir, fp = bzrdir.BzrDir.open_containing(filename)
 
1002
            b = dir.open_branch()
 
1003
            if fp != '':
 
1004
                try:
 
1005
                    # might be a tree:
 
1006
                    inv = dir.open_workingtree().inventory
 
1007
                except (errors.NotBranchError, errors.NotLocalUrl):
 
1008
                    # either no tree, or is remote.
 
1009
                    inv = b.basis_tree().inventory
 
1010
                file_id = inv.path2id(fp)
 
1011
        else:
 
1012
            # local dir only
 
1013
            # FIXME ? log the current subdir only RBC 20060203 
 
1014
            dir, relpath = bzrdir.BzrDir.open_containing('.')
 
1015
            b = dir.open_branch()
 
1016
 
 
1017
        if revision is None:
 
1018
            rev1 = None
 
1019
            rev2 = None
 
1020
        elif len(revision) == 1:
 
1021
            rev1 = rev2 = revision[0].in_history(b).revno
 
1022
        elif len(revision) == 2:
 
1023
            rev1 = revision[0].in_history(b).revno
 
1024
            if revision[1].spec is None:
 
1025
                # missing end-range means last known revision
 
1026
                rev2 = b.revno()
 
1027
            else:
 
1028
                rev2 = revision[1].in_history(b).revno
 
1029
        else:
 
1030
            raise BzrCommandError('bzr log --revision takes one or two values.')
 
1031
 
 
1032
        # By this point, the revision numbers are converted to the +ve
 
1033
        # form if they were supplied in the -ve form, so we can do
 
1034
        # this comparison in relative safety
 
1035
        if rev1 > rev2:
 
1036
            (rev2, rev1) = (rev1, rev2)
 
1037
 
 
1038
        mutter('encoding log as %r', bzrlib.user_encoding)
 
1039
 
 
1040
        # use 'replace' so that we don't abort if trying to write out
 
1041
        # in e.g. the default C locale.
 
1042
        outf = codecs.getwriter(bzrlib.user_encoding)(sys.stdout, errors='replace')
 
1043
 
 
1044
        log_format = get_log_format(long=long, short=short, line=line)
 
1045
        lf = log_formatter(log_format,
 
1046
                           show_ids=show_ids,
 
1047
                           to_file=outf,
 
1048
                           show_timezone=timezone)
 
1049
 
 
1050
        show_log(b,
 
1051
                 lf,
 
1052
                 file_id,
 
1053
                 verbose=verbose,
 
1054
                 direction=direction,
 
1055
                 start_revision=rev1,
 
1056
                 end_revision=rev2,
 
1057
                 search=message)
 
1058
 
 
1059
 
 
1060
def get_log_format(long=False, short=False, line=False, default='long'):
 
1061
    log_format = default
 
1062
    if long:
 
1063
        log_format = 'long'
 
1064
    if short:
 
1065
        log_format = 'short'
 
1066
    if line:
 
1067
        log_format = 'line'
 
1068
    return log_format
 
1069
 
 
1070
 
 
1071
class cmd_touching_revisions(Command):
 
1072
    """Return revision-ids which affected a particular file.
 
1073
 
 
1074
    A more user-friendly interface is "bzr log FILE"."""
 
1075
    hidden = True
 
1076
    takes_args = ["filename"]
 
1077
    @display_command
 
1078
    def run(self, filename):
 
1079
        tree, relpath = WorkingTree.open_containing(filename)
 
1080
        b = tree.branch
 
1081
        inv = tree.read_working_inventory()
 
1082
        file_id = inv.path2id(relpath)
 
1083
        for revno, revision_id, what in bzrlib.log.find_touching_revisions(b, file_id):
 
1084
            print "%6d %s" % (revno, what)
 
1085
 
 
1086
 
 
1087
class cmd_ls(Command):
 
1088
    """List files in a tree.
 
1089
    """
 
1090
    # TODO: Take a revision or remote path and list that tree instead.
 
1091
    hidden = True
 
1092
    takes_options = ['verbose', 'revision',
 
1093
                     Option('non-recursive',
 
1094
                            help='don\'t recurse into sub-directories'),
 
1095
                     Option('from-root',
 
1096
                            help='Print all paths from the root of the branch.'),
 
1097
                     Option('unknown', help='Print unknown files'),
 
1098
                     Option('versioned', help='Print versioned files'),
 
1099
                     Option('ignored', help='Print ignored files'),
 
1100
 
 
1101
                     Option('null', help='Null separate the files'),
 
1102
                    ]
 
1103
    @display_command
 
1104
    def run(self, revision=None, verbose=False, 
 
1105
            non_recursive=False, from_root=False,
 
1106
            unknown=False, versioned=False, ignored=False,
 
1107
            null=False):
 
1108
 
 
1109
        if verbose and null:
 
1110
            raise BzrCommandError('Cannot set both --verbose and --null')
 
1111
        all = not (unknown or versioned or ignored)
 
1112
 
 
1113
        selection = {'I':ignored, '?':unknown, 'V':versioned}
 
1114
 
 
1115
        tree, relpath = WorkingTree.open_containing(u'.')
 
1116
        if from_root:
 
1117
            relpath = u''
 
1118
        elif relpath:
 
1119
            relpath += '/'
 
1120
        if revision is not None:
 
1121
            tree = tree.branch.repository.revision_tree(
 
1122
                revision[0].in_history(tree.branch).rev_id)
 
1123
        for fp, fc, kind, fid, entry in tree.list_files():
 
1124
            if fp.startswith(relpath):
 
1125
                fp = fp[len(relpath):]
 
1126
                if non_recursive and '/' in fp:
 
1127
                    continue
 
1128
                if not all and not selection[fc]:
 
1129
                    continue
 
1130
                if verbose:
 
1131
                    kindch = entry.kind_character()
 
1132
                    print '%-8s %s%s' % (fc, fp, kindch)
 
1133
                elif null:
 
1134
                    sys.stdout.write(fp)
 
1135
                    sys.stdout.write('\0')
 
1136
                    sys.stdout.flush()
 
1137
                else:
 
1138
                    print fp
 
1139
 
 
1140
 
 
1141
class cmd_unknowns(Command):
 
1142
    """List unknown files."""
 
1143
    @display_command
 
1144
    def run(self):
 
1145
        from bzrlib.osutils import quotefn
 
1146
        for f in WorkingTree.open_containing(u'.')[0].unknowns():
 
1147
            print quotefn(f)
 
1148
 
 
1149
 
 
1150
class cmd_ignore(Command):
 
1151
    """Ignore a command or pattern.
 
1152
 
 
1153
    To remove patterns from the ignore list, edit the .bzrignore file.
 
1154
 
 
1155
    If the pattern contains a slash, it is compared to the whole path
 
1156
    from the branch root.  Otherwise, it is compared to only the last
 
1157
    component of the path.  To match a file only in the root directory,
 
1158
    prepend './'.
 
1159
 
 
1160
    Ignore patterns are case-insensitive on case-insensitive systems.
 
1161
 
 
1162
    Note: wildcards must be quoted from the shell on Unix.
 
1163
 
 
1164
    examples:
 
1165
        bzr ignore ./Makefile
 
1166
        bzr ignore '*.class'
 
1167
    """
 
1168
    # TODO: Complain if the filename is absolute
 
1169
    takes_args = ['name_pattern']
 
1170
    
 
1171
    def run(self, name_pattern):
 
1172
        from bzrlib.atomicfile import AtomicFile
 
1173
        import os.path
 
1174
 
 
1175
        tree, relpath = WorkingTree.open_containing(u'.')
 
1176
        ifn = tree.abspath('.bzrignore')
 
1177
 
 
1178
        if os.path.exists(ifn):
 
1179
            f = open(ifn, 'rt')
 
1180
            try:
 
1181
                igns = f.read().decode('utf-8')
 
1182
            finally:
 
1183
                f.close()
 
1184
        else:
 
1185
            igns = ''
 
1186
 
 
1187
        # TODO: If the file already uses crlf-style termination, maybe
 
1188
        # we should use that for the newly added lines?
 
1189
 
 
1190
        if igns and igns[-1] != '\n':
 
1191
            igns += '\n'
 
1192
        igns += name_pattern + '\n'
 
1193
 
 
1194
        try:
 
1195
            f = AtomicFile(ifn, 'wt')
 
1196
            f.write(igns.encode('utf-8'))
 
1197
            f.commit()
 
1198
        finally:
 
1199
            f.close()
 
1200
 
 
1201
        inv = tree.inventory
 
1202
        if inv.path2id('.bzrignore'):
 
1203
            mutter('.bzrignore is already versioned')
 
1204
        else:
 
1205
            mutter('need to make new .bzrignore file versioned')
 
1206
            tree.add(['.bzrignore'])
 
1207
 
 
1208
 
 
1209
class cmd_ignored(Command):
 
1210
    """List ignored files and the patterns that matched them.
 
1211
 
 
1212
    See also: bzr ignore"""
 
1213
    @display_command
 
1214
    def run(self):
 
1215
        tree = WorkingTree.open_containing(u'.')[0]
 
1216
        for path, file_class, kind, file_id, entry in tree.list_files():
 
1217
            if file_class != 'I':
 
1218
                continue
 
1219
            ## XXX: Slightly inefficient since this was already calculated
 
1220
            pat = tree.is_ignored(path)
 
1221
            print '%-50s %s' % (path, pat)
 
1222
 
 
1223
 
 
1224
class cmd_lookup_revision(Command):
 
1225
    """Lookup the revision-id from a revision-number
 
1226
 
 
1227
    example:
 
1228
        bzr lookup-revision 33
 
1229
    """
 
1230
    hidden = True
 
1231
    takes_args = ['revno']
 
1232
    
 
1233
    @display_command
 
1234
    def run(self, revno):
 
1235
        try:
 
1236
            revno = int(revno)
 
1237
        except ValueError:
 
1238
            raise BzrCommandError("not a valid revision-number: %r" % revno)
 
1239
 
 
1240
        print WorkingTree.open_containing(u'.')[0].branch.get_rev_id(revno)
 
1241
 
 
1242
 
 
1243
class cmd_export(Command):
 
1244
    """Export past revision to destination directory.
 
1245
 
 
1246
    If no revision is specified this exports the last committed revision.
 
1247
 
 
1248
    Format may be an "exporter" name, such as tar, tgz, tbz2.  If none is
 
1249
    given, try to find the format with the extension. If no extension
 
1250
    is found exports to a directory (equivalent to --format=dir).
 
1251
 
 
1252
    Root may be the top directory for tar, tgz and tbz2 formats. If none
 
1253
    is given, the top directory will be the root name of the file.
 
1254
 
 
1255
    Note: export of tree with non-ascii filenames to zip is not supported.
 
1256
 
 
1257
     Supported formats       Autodetected by extension
 
1258
     -----------------       -------------------------
 
1259
         dir                            -
 
1260
         tar                          .tar
 
1261
         tbz2                    .tar.bz2, .tbz2
 
1262
         tgz                      .tar.gz, .tgz
 
1263
         zip                          .zip
 
1264
    """
 
1265
    takes_args = ['dest']
 
1266
    takes_options = ['revision', 'format', 'root']
 
1267
    def run(self, dest, revision=None, format=None, root=None):
 
1268
        import os.path
 
1269
        from bzrlib.export import export
 
1270
        tree = WorkingTree.open_containing(u'.')[0]
 
1271
        b = tree.branch
 
1272
        if revision is None:
 
1273
            # should be tree.last_revision  FIXME
 
1274
            rev_id = b.last_revision()
 
1275
        else:
 
1276
            if len(revision) != 1:
 
1277
                raise BzrError('bzr export --revision takes exactly 1 argument')
 
1278
            rev_id = revision[0].in_history(b).rev_id
 
1279
        t = b.repository.revision_tree(rev_id)
 
1280
        try:
 
1281
            export(t, dest, format, root)
 
1282
        except errors.NoSuchExportFormat, e:
 
1283
            raise BzrCommandError('Unsupported export format: %s' % e.format)
 
1284
 
 
1285
 
 
1286
class cmd_cat(Command):
 
1287
    """Write a file's text from a previous revision."""
 
1288
 
 
1289
    takes_options = ['revision']
 
1290
    takes_args = ['filename']
 
1291
 
 
1292
    @display_command
 
1293
    def run(self, filename, revision=None):
 
1294
        if revision is not None and len(revision) != 1:
 
1295
            raise BzrCommandError("bzr cat --revision takes exactly one number")
 
1296
        tree = None
 
1297
        try:
 
1298
            tree, relpath = WorkingTree.open_containing(filename)
 
1299
            b = tree.branch
 
1300
        except NotBranchError:
 
1301
            pass
 
1302
 
 
1303
        if tree is None:
 
1304
            b, relpath = Branch.open_containing(filename)
 
1305
        if revision is None:
 
1306
            revision_id = b.last_revision()
 
1307
        else:
 
1308
            revision_id = revision[0].in_history(b).rev_id
 
1309
        b.print_file(relpath, revision_id)
 
1310
 
 
1311
 
 
1312
class cmd_local_time_offset(Command):
 
1313
    """Show the offset in seconds from GMT to local time."""
 
1314
    hidden = True    
 
1315
    @display_command
 
1316
    def run(self):
 
1317
        print bzrlib.osutils.local_time_offset()
 
1318
 
 
1319
 
 
1320
 
 
1321
class cmd_commit(Command):
 
1322
    """Commit changes into a new revision.
 
1323
    
 
1324
    If no arguments are given, the entire tree is committed.
 
1325
 
 
1326
    If selected files are specified, only changes to those files are
 
1327
    committed.  If a directory is specified then the directory and everything 
 
1328
    within it is committed.
 
1329
 
 
1330
    A selected-file commit may fail in some cases where the committed
 
1331
    tree would be invalid, such as trying to commit a file in a
 
1332
    newly-added directory that is not itself committed.
 
1333
    """
 
1334
    # TODO: Run hooks on tree to-be-committed, and after commit.
 
1335
 
 
1336
    # TODO: Strict commit that fails if there are deleted files.
 
1337
    #       (what does "deleted files" mean ??)
 
1338
 
 
1339
    # TODO: Give better message for -s, --summary, used by tla people
 
1340
 
 
1341
    # XXX: verbose currently does nothing
 
1342
 
 
1343
    takes_args = ['selected*']
 
1344
    takes_options = ['message', 'verbose', 
 
1345
                     Option('unchanged',
 
1346
                            help='commit even if nothing has changed'),
 
1347
                     Option('file', type=str, 
 
1348
                            argname='msgfile',
 
1349
                            help='file containing commit message'),
 
1350
                     Option('strict',
 
1351
                            help="refuse to commit if there are unknown "
 
1352
                            "files in the working tree."),
 
1353
                     ]
 
1354
    aliases = ['ci', 'checkin']
 
1355
 
 
1356
    def run(self, message=None, file=None, verbose=True, selected_list=None,
 
1357
            unchanged=False, strict=False):
 
1358
        from bzrlib.errors import (PointlessCommit, ConflictsInTree,
 
1359
                StrictCommitFailed)
 
1360
        from bzrlib.msgeditor import edit_commit_message, \
 
1361
                make_commit_message_template
 
1362
        from bzrlib.status import show_status
 
1363
        from tempfile import TemporaryFile
 
1364
        import codecs
 
1365
 
 
1366
        # TODO: Need a blackbox test for invoking the external editor; may be
 
1367
        # slightly problematic to run this cross-platform.
 
1368
 
 
1369
        # TODO: do more checks that the commit will succeed before 
 
1370
        # spending the user's valuable time typing a commit message.
 
1371
        #
 
1372
        # TODO: if the commit *does* happen to fail, then save the commit 
 
1373
        # message to a temporary file where it can be recovered
 
1374
        tree, selected_list = tree_files(selected_list)
 
1375
        if message is None and not file:
 
1376
            template = make_commit_message_template(tree, selected_list)
 
1377
            message = edit_commit_message(template)
 
1378
            if message is None:
 
1379
                raise BzrCommandError("please specify a commit message"
 
1380
                                      " with either --message or --file")
 
1381
        elif message and file:
 
1382
            raise BzrCommandError("please specify either --message or --file")
 
1383
        
 
1384
        if file:
 
1385
            import codecs
 
1386
            message = codecs.open(file, 'rt', bzrlib.user_encoding).read()
 
1387
 
 
1388
        if message == "":
 
1389
                raise BzrCommandError("empty commit message specified")
 
1390
            
 
1391
        try:
 
1392
            tree.commit(message, specific_files=selected_list,
 
1393
                        allow_pointless=unchanged, strict=strict)
 
1394
        except PointlessCommit:
 
1395
            # FIXME: This should really happen before the file is read in;
 
1396
            # perhaps prepare the commit; get the message; then actually commit
 
1397
            raise BzrCommandError("no changes to commit",
 
1398
                                  ["use --unchanged to commit anyhow"])
 
1399
        except ConflictsInTree:
 
1400
            raise BzrCommandError("Conflicts detected in working tree.  "
 
1401
                'Use "bzr conflicts" to list, "bzr resolve FILE" to resolve.')
 
1402
        except StrictCommitFailed:
 
1403
            raise BzrCommandError("Commit refused because there are unknown "
 
1404
                                  "files in the working tree.")
 
1405
        note('Committed revision %d.' % (tree.branch.revno(),))
 
1406
 
 
1407
 
 
1408
class cmd_check(Command):
 
1409
    """Validate consistency of branch history.
 
1410
 
 
1411
    This command checks various invariants about the branch storage to
 
1412
    detect data corruption or bzr bugs.
 
1413
    """
 
1414
    takes_args = ['branch?']
 
1415
    takes_options = ['verbose']
 
1416
 
 
1417
    def run(self, branch=None, verbose=False):
 
1418
        from bzrlib.check import check
 
1419
        if branch is None:
 
1420
            tree = WorkingTree.open_containing()[0]
 
1421
            branch = tree.branch
 
1422
        else:
 
1423
            branch = Branch.open(branch)
 
1424
        check(branch, verbose)
 
1425
 
 
1426
 
 
1427
class cmd_scan_cache(Command):
 
1428
    hidden = True
 
1429
    def run(self):
 
1430
        from bzrlib.hashcache import HashCache
 
1431
 
 
1432
        c = HashCache(u'.')
 
1433
        c.read()
 
1434
        c.scan()
 
1435
            
 
1436
        print '%6d stats' % c.stat_count
 
1437
        print '%6d in hashcache' % len(c._cache)
 
1438
        print '%6d files removed from cache' % c.removed_count
 
1439
        print '%6d hashes updated' % c.update_count
 
1440
        print '%6d files changed too recently to cache' % c.danger_count
 
1441
 
 
1442
        if c.needs_write:
 
1443
            c.write()
 
1444
            
 
1445
 
 
1446
 
 
1447
class cmd_upgrade(Command):
 
1448
    """Upgrade branch storage to current format.
 
1449
 
 
1450
    The check command or bzr developers may sometimes advise you to run
 
1451
    this command. When the default format has changed you may also be warned
 
1452
    during other operations to upgrade.
 
1453
    """
 
1454
    takes_args = ['url?']
 
1455
 
 
1456
    def run(self, url='.'):
 
1457
        from bzrlib.upgrade import upgrade
 
1458
        upgrade(url)
 
1459
 
 
1460
 
 
1461
class cmd_whoami(Command):
 
1462
    """Show bzr user id."""
 
1463
    takes_options = ['email']
 
1464
    
 
1465
    @display_command
 
1466
    def run(self, email=False):
 
1467
        try:
 
1468
            b = WorkingTree.open_containing(u'.')[0].branch
 
1469
            config = bzrlib.config.BranchConfig(b)
 
1470
        except NotBranchError:
 
1471
            config = bzrlib.config.GlobalConfig()
 
1472
        
 
1473
        if email:
 
1474
            print config.user_email()
 
1475
        else:
 
1476
            print config.username()
 
1477
 
 
1478
 
 
1479
class cmd_nick(Command):
 
1480
    """Print or set the branch nickname.  
 
1481
 
 
1482
    If unset, the tree root directory name is used as the nickname
 
1483
    To print the current nickname, execute with no argument.  
 
1484
    """
 
1485
    takes_args = ['nickname?']
 
1486
    def run(self, nickname=None):
 
1487
        branch = Branch.open_containing(u'.')[0]
 
1488
        if nickname is None:
 
1489
            self.printme(branch)
 
1490
        else:
 
1491
            branch.nick = nickname
 
1492
 
 
1493
    @display_command
 
1494
    def printme(self, branch):
 
1495
        print branch.nick 
 
1496
 
 
1497
 
 
1498
class cmd_selftest(Command):
 
1499
    """Run internal test suite.
 
1500
    
 
1501
    This creates temporary test directories in the working directory,
 
1502
    but not existing data is affected.  These directories are deleted
 
1503
    if the tests pass, or left behind to help in debugging if they
 
1504
    fail and --keep-output is specified.
 
1505
    
 
1506
    If arguments are given, they are regular expressions that say
 
1507
    which tests should run.
 
1508
 
 
1509
    If the global option '--no-plugins' is given, plugins are not loaded
 
1510
    before running the selftests.  This has two effects: features provided or
 
1511
    modified by plugins will not be tested, and tests provided by plugins will
 
1512
    not be run.
 
1513
 
 
1514
    examples:
 
1515
        bzr selftest ignore
 
1516
        bzr --no-plugins selftest -v
 
1517
    """
 
1518
    # TODO: --list should give a list of all available tests
 
1519
 
 
1520
    # NB: this is used from the class without creating an instance, which is
 
1521
    # why it does not have a self parameter.
 
1522
    def get_transport_type(typestring):
 
1523
        """Parse and return a transport specifier."""
 
1524
        if typestring == "sftp":
 
1525
            from bzrlib.transport.sftp import SFTPAbsoluteServer
 
1526
            return SFTPAbsoluteServer
 
1527
        if typestring == "memory":
 
1528
            from bzrlib.transport.memory import MemoryServer
 
1529
            return MemoryServer
 
1530
        msg = "No known transport type %s. Supported types are: sftp\n" %\
 
1531
            (typestring)
 
1532
        raise BzrCommandError(msg)
 
1533
 
 
1534
    hidden = True
 
1535
    takes_args = ['testspecs*']
 
1536
    takes_options = ['verbose',
 
1537
                     Option('one', help='stop when one test fails'),
 
1538
                     Option('keep-output', 
 
1539
                            help='keep output directories when tests fail'),
 
1540
                     Option('transport', 
 
1541
                            help='Use a different transport by default '
 
1542
                                 'throughout the test suite.',
 
1543
                            type=get_transport_type),
 
1544
                    ]
 
1545
 
 
1546
    def run(self, testspecs_list=None, verbose=False, one=False,
 
1547
            keep_output=False, transport=None):
 
1548
        import bzrlib.ui
 
1549
        from bzrlib.tests import selftest
 
1550
        # we don't want progress meters from the tests to go to the
 
1551
        # real output; and we don't want log messages cluttering up
 
1552
        # the real logs.
 
1553
        save_ui = bzrlib.ui.ui_factory
 
1554
        bzrlib.trace.info('running tests...')
 
1555
        try:
 
1556
            bzrlib.ui.ui_factory = bzrlib.ui.SilentUIFactory()
 
1557
            if testspecs_list is not None:
 
1558
                pattern = '|'.join(testspecs_list)
 
1559
            else:
 
1560
                pattern = ".*"
 
1561
            result = selftest(verbose=verbose, 
 
1562
                              pattern=pattern,
 
1563
                              stop_on_failure=one, 
 
1564
                              keep_output=keep_output,
 
1565
                              transport=transport)
 
1566
            if result:
 
1567
                bzrlib.trace.info('tests passed')
 
1568
            else:
 
1569
                bzrlib.trace.info('tests failed')
 
1570
            return int(not result)
 
1571
        finally:
 
1572
            bzrlib.ui.ui_factory = save_ui
 
1573
 
 
1574
 
 
1575
def _get_bzr_branch():
 
1576
    """If bzr is run from a branch, return Branch or None"""
 
1577
    import bzrlib.errors
 
1578
    from bzrlib.branch import Branch
 
1579
    from bzrlib.osutils import abspath
 
1580
    from os.path import dirname
 
1581
    
 
1582
    try:
 
1583
        branch = Branch.open(dirname(abspath(dirname(__file__))))
 
1584
        return branch
 
1585
    except bzrlib.errors.BzrError:
 
1586
        return None
 
1587
    
 
1588
 
 
1589
def show_version():
 
1590
    print "bzr (bazaar-ng) %s" % bzrlib.__version__
 
1591
    # is bzrlib itself in a branch?
 
1592
    branch = _get_bzr_branch()
 
1593
    if branch:
 
1594
        rh = branch.revision_history()
 
1595
        revno = len(rh)
 
1596
        print "  bzr checkout, revision %d" % (revno,)
 
1597
        print "  nick: %s" % (branch.nick,)
 
1598
        if rh:
 
1599
            print "  revid: %s" % (rh[-1],)
 
1600
    print bzrlib.__copyright__
 
1601
    print "http://bazaar-ng.org/"
 
1602
    print
 
1603
    print "bzr comes with ABSOLUTELY NO WARRANTY.  bzr is free software, and"
 
1604
    print "you may use, modify and redistribute it under the terms of the GNU"
 
1605
    print "General Public License version 2 or later."
 
1606
 
 
1607
 
 
1608
class cmd_version(Command):
 
1609
    """Show version of bzr."""
 
1610
    @display_command
 
1611
    def run(self):
 
1612
        show_version()
 
1613
 
 
1614
class cmd_rocks(Command):
 
1615
    """Statement of optimism."""
 
1616
    hidden = True
 
1617
    @display_command
 
1618
    def run(self):
 
1619
        print "it sure does!"
 
1620
 
 
1621
 
 
1622
class cmd_find_merge_base(Command):
 
1623
    """Find and print a base revision for merging two branches.
 
1624
    """
 
1625
    # TODO: Options to specify revisions on either side, as if
 
1626
    #       merging only part of the history.
 
1627
    takes_args = ['branch', 'other']
 
1628
    hidden = True
 
1629
    
 
1630
    @display_command
 
1631
    def run(self, branch, other):
 
1632
        from bzrlib.revision import common_ancestor, MultipleRevisionSources
 
1633
        
 
1634
        branch1 = Branch.open_containing(branch)[0]
 
1635
        branch2 = Branch.open_containing(other)[0]
 
1636
 
 
1637
        history_1 = branch1.revision_history()
 
1638
        history_2 = branch2.revision_history()
 
1639
 
 
1640
        last1 = branch1.last_revision()
 
1641
        last2 = branch2.last_revision()
 
1642
 
 
1643
        source = MultipleRevisionSources(branch1.repository, 
 
1644
                                         branch2.repository)
 
1645
        
 
1646
        base_rev_id = common_ancestor(last1, last2, source)
 
1647
 
 
1648
        print 'merge base is revision %s' % base_rev_id
 
1649
        
 
1650
        return
 
1651
 
 
1652
        if base_revno is None:
 
1653
            raise bzrlib.errors.UnrelatedBranches()
 
1654
 
 
1655
        print ' r%-6d in %s' % (base_revno, branch)
 
1656
 
 
1657
        other_revno = branch2.revision_id_to_revno(base_revid)
 
1658
        
 
1659
        print ' r%-6d in %s' % (other_revno, other)
 
1660
 
 
1661
 
 
1662
 
 
1663
class cmd_merge(Command):
 
1664
    """Perform a three-way merge.
 
1665
    
 
1666
    The branch is the branch you will merge from.  By default, it will
 
1667
    merge the latest revision.  If you specify a revision, that
 
1668
    revision will be merged.  If you specify two revisions, the first
 
1669
    will be used as a BASE, and the second one as OTHER.  Revision
 
1670
    numbers are always relative to the specified branch.
 
1671
 
 
1672
    By default bzr will try to merge in all new work from the other
 
1673
    branch, automatically determining an appropriate base.  If this
 
1674
    fails, you may need to give an explicit base.
 
1675
    
 
1676
    Examples:
 
1677
 
 
1678
    To merge the latest revision from bzr.dev
 
1679
    bzr merge ../bzr.dev
 
1680
 
 
1681
    To merge changes up to and including revision 82 from bzr.dev
 
1682
    bzr merge -r 82 ../bzr.dev
 
1683
 
 
1684
    To merge the changes introduced by 82, without previous changes:
 
1685
    bzr merge -r 81..82 ../bzr.dev
 
1686
    
 
1687
    merge refuses to run if there are any uncommitted changes, unless
 
1688
    --force is given.
 
1689
    """
 
1690
    takes_args = ['branch?']
 
1691
    takes_options = ['revision', 'force', 'merge-type', 'reprocess',
 
1692
                     Option('show-base', help="Show base revision text in "
 
1693
                            "conflicts")]
 
1694
 
 
1695
    def run(self, branch=None, revision=None, force=False, merge_type=None,
 
1696
            show_base=False, reprocess=False):
 
1697
        if merge_type is None:
 
1698
            merge_type = Merge3Merger
 
1699
        if branch is None:
 
1700
            branch = WorkingTree.open_containing(u'.')[0].branch.get_parent()
 
1701
            if branch is None:
 
1702
                raise BzrCommandError("No merge location known or specified.")
 
1703
            else:
 
1704
                print "Using saved location: %s" % branch 
 
1705
        if revision is None or len(revision) < 1:
 
1706
            base = [None, None]
 
1707
            other = [branch, -1]
 
1708
        else:
 
1709
            if len(revision) == 1:
 
1710
                base = [None, None]
 
1711
                other_branch = Branch.open_containing(branch)[0]
 
1712
                revno = revision[0].in_history(other_branch).revno
 
1713
                other = [branch, revno]
 
1714
            else:
 
1715
                assert len(revision) == 2
 
1716
                if None in revision:
 
1717
                    raise BzrCommandError(
 
1718
                        "Merge doesn't permit that revision specifier.")
 
1719
                b = Branch.open_containing(branch)[0]
 
1720
 
 
1721
                base = [branch, revision[0].in_history(b).revno]
 
1722
                other = [branch, revision[1].in_history(b).revno]
 
1723
 
 
1724
        try:
 
1725
            conflict_count = merge(other, base, check_clean=(not force),
 
1726
                                   merge_type=merge_type, reprocess=reprocess,
 
1727
                                   show_base=show_base)
 
1728
            if conflict_count != 0:
 
1729
                return 1
 
1730
            else:
 
1731
                return 0
 
1732
        except bzrlib.errors.AmbiguousBase, e:
 
1733
            m = ("sorry, bzr can't determine the right merge base yet\n"
 
1734
                 "candidates are:\n  "
 
1735
                 + "\n  ".join(e.bases)
 
1736
                 + "\n"
 
1737
                 "please specify an explicit base with -r,\n"
 
1738
                 "and (if you want) report this to the bzr developers\n")
 
1739
            log_error(m)
 
1740
 
 
1741
 
 
1742
class cmd_remerge(Command):
 
1743
    """Redo a merge.
 
1744
    """
 
1745
    takes_args = ['file*']
 
1746
    takes_options = ['merge-type', 'reprocess',
 
1747
                     Option('show-base', help="Show base revision text in "
 
1748
                            "conflicts")]
 
1749
 
 
1750
    def run(self, file_list=None, merge_type=None, show_base=False,
 
1751
            reprocess=False):
 
1752
        from bzrlib.merge import merge_inner, transform_tree
 
1753
        if merge_type is None:
 
1754
            merge_type = Merge3Merger
 
1755
        tree, file_list = tree_files(file_list)
 
1756
        tree.lock_write()
 
1757
        try:
 
1758
            pending_merges = tree.pending_merges() 
 
1759
            if len(pending_merges) != 1:
 
1760
                raise BzrCommandError("Sorry, remerge only works after normal"
 
1761
                                      + " merges.  Not cherrypicking or"
 
1762
                                      + "multi-merges.")
 
1763
            repository = tree.branch.repository
 
1764
            base_revision = common_ancestor(tree.branch.last_revision(), 
 
1765
                                            pending_merges[0], repository)
 
1766
            base_tree = repository.revision_tree(base_revision)
 
1767
            other_tree = repository.revision_tree(pending_merges[0])
 
1768
            interesting_ids = None
 
1769
            if file_list is not None:
 
1770
                interesting_ids = set()
 
1771
                for filename in file_list:
 
1772
                    file_id = tree.path2id(filename)
 
1773
                    interesting_ids.add(file_id)
 
1774
                    if tree.kind(file_id) != "directory":
 
1775
                        continue
 
1776
                    
 
1777
                    for name, ie in tree.inventory.iter_entries(file_id):
 
1778
                        interesting_ids.add(ie.file_id)
 
1779
            transform_tree(tree, tree.basis_tree(), interesting_ids)
 
1780
            if file_list is None:
 
1781
                restore_files = list(tree.iter_conflicts())
 
1782
            else:
 
1783
                restore_files = file_list
 
1784
            for filename in restore_files:
 
1785
                try:
 
1786
                    restore(tree.abspath(filename))
 
1787
                except NotConflicted:
 
1788
                    pass
 
1789
            conflicts =  merge_inner(tree.branch, other_tree, base_tree, 
 
1790
                                     interesting_ids = interesting_ids, 
 
1791
                                     other_rev_id=pending_merges[0], 
 
1792
                                     merge_type=merge_type, 
 
1793
                                     show_base=show_base,
 
1794
                                     reprocess=reprocess)
 
1795
        finally:
 
1796
            tree.unlock()
 
1797
        if conflicts > 0:
 
1798
            return 1
 
1799
        else:
 
1800
            return 0
 
1801
 
 
1802
class cmd_revert(Command):
 
1803
    """Reverse all changes since the last commit.
 
1804
 
 
1805
    Only versioned files are affected.  Specify filenames to revert only 
 
1806
    those files.  By default, any files that are changed will be backed up
 
1807
    first.  Backup files have a '~' appended to their name.
 
1808
    """
 
1809
    takes_options = ['revision', 'no-backup']
 
1810
    takes_args = ['file*']
 
1811
    aliases = ['merge-revert']
 
1812
 
 
1813
    def run(self, revision=None, no_backup=False, file_list=None):
 
1814
        from bzrlib.commands import parse_spec
 
1815
        if file_list is not None:
 
1816
            if len(file_list) == 0:
 
1817
                raise BzrCommandError("No files specified")
 
1818
        else:
 
1819
            file_list = []
 
1820
        
 
1821
        tree, file_list = tree_files(file_list)
 
1822
        if revision is None:
 
1823
            # FIXME should be tree.last_revision
 
1824
            rev_id = tree.branch.last_revision()
 
1825
        elif len(revision) != 1:
 
1826
            raise BzrCommandError('bzr revert --revision takes exactly 1 argument')
 
1827
        else:
 
1828
            rev_id = revision[0].in_history(tree.branch).rev_id
 
1829
        tree.revert(file_list, tree.branch.repository.revision_tree(rev_id),
 
1830
                    not no_backup)
 
1831
 
 
1832
 
 
1833
class cmd_assert_fail(Command):
 
1834
    """Test reporting of assertion failures"""
 
1835
    hidden = True
 
1836
    def run(self):
 
1837
        assert False, "always fails"
 
1838
 
 
1839
 
 
1840
class cmd_help(Command):
 
1841
    """Show help on a command or other topic.
 
1842
 
 
1843
    For a list of all available commands, say 'bzr help commands'."""
 
1844
    takes_options = ['long']
 
1845
    takes_args = ['topic?']
 
1846
    aliases = ['?']
 
1847
    
 
1848
    @display_command
 
1849
    def run(self, topic=None, long=False):
 
1850
        import help
 
1851
        if topic is None and long:
 
1852
            topic = "commands"
 
1853
        help.help(topic)
 
1854
 
 
1855
 
 
1856
class cmd_shell_complete(Command):
 
1857
    """Show appropriate completions for context.
 
1858
 
 
1859
    For a list of all available commands, say 'bzr shell-complete'."""
 
1860
    takes_args = ['context?']
 
1861
    aliases = ['s-c']
 
1862
    hidden = True
 
1863
    
 
1864
    @display_command
 
1865
    def run(self, context=None):
 
1866
        import shellcomplete
 
1867
        shellcomplete.shellcomplete(context)
 
1868
 
 
1869
 
 
1870
class cmd_fetch(Command):
 
1871
    """Copy in history from another branch but don't merge it.
 
1872
 
 
1873
    This is an internal method used for pull and merge."""
 
1874
    hidden = True
 
1875
    takes_args = ['from_branch', 'to_branch']
 
1876
    def run(self, from_branch, to_branch):
 
1877
        from bzrlib.fetch import Fetcher
 
1878
        from bzrlib.branch import Branch
 
1879
        from_b = Branch.open(from_branch)
 
1880
        to_b = Branch.open(to_branch)
 
1881
        Fetcher(to_b, from_b)
 
1882
 
 
1883
 
 
1884
class cmd_missing(Command):
 
1885
    """Show unmerged/unpulled revisions between two branches.
 
1886
 
 
1887
    OTHER_BRANCH may be local or remote."""
 
1888
    takes_args = ['other_branch?']
 
1889
    takes_options = [Option('reverse', 'Reverse the order of revisions'),
 
1890
                     Option('mine-only', 
 
1891
                            'Display changes in the local branch only'),
 
1892
                     Option('theirs-only', 
 
1893
                            'Display changes in the remote branch only'), 
 
1894
                     'line',
 
1895
                     'long', 
 
1896
                     'short',
 
1897
                     'show-ids',
 
1898
                     'verbose'
 
1899
                     ]
 
1900
 
 
1901
    def run(self, other_branch=None, reverse=False, mine_only=False,
 
1902
            theirs_only=False, long=True, short=False, line=False, 
 
1903
            show_ids=False, verbose=False):
 
1904
        from bzrlib.missing import find_unmerged, iter_log_data
 
1905
        from bzrlib.log import log_formatter
 
1906
        local_branch = bzrlib.branch.Branch.open_containing(u".")[0]
 
1907
        parent = local_branch.get_parent()
 
1908
        if other_branch is None:
 
1909
            other_branch = parent
 
1910
            if other_branch is None:
 
1911
                raise BzrCommandError("No missing location known or specified.")
 
1912
            print "Using last location: " + local_branch.get_parent()
 
1913
        remote_branch = bzrlib.branch.Branch.open(other_branch)
 
1914
        local_extra, remote_extra = find_unmerged(local_branch, remote_branch)
 
1915
        log_format = get_log_format(long=long, short=short, line=line)
 
1916
        lf = log_formatter(log_format, sys.stdout,
 
1917
                           show_ids=show_ids,
 
1918
                           show_timezone='original')
 
1919
        if reverse is False:
 
1920
            local_extra.reverse()
 
1921
            remote_extra.reverse()
 
1922
        if local_extra and not theirs_only:
 
1923
            print "You have %d extra revision(s):" % len(local_extra)
 
1924
            for data in iter_log_data(local_extra, local_branch.repository,
 
1925
                                      verbose):
 
1926
                lf.show(*data)
 
1927
            printed_local = True
 
1928
        else:
 
1929
            printed_local = False
 
1930
        if remote_extra and not mine_only:
 
1931
            if printed_local is True:
 
1932
                print "\n\n"
 
1933
            print "You are missing %d revision(s):" % len(remote_extra)
 
1934
            for data in iter_log_data(remote_extra, remote_branch.repository, 
 
1935
                                      verbose):
 
1936
                lf.show(*data)
 
1937
        if not remote_extra and not local_extra:
 
1938
            status_code = 0
 
1939
            print "Branches are up to date."
 
1940
        else:
 
1941
            status_code = 1
 
1942
        if parent is None and other_branch is not None:
 
1943
            local_branch.set_parent(other_branch)
 
1944
        return status_code
 
1945
 
 
1946
 
 
1947
class cmd_plugins(Command):
 
1948
    """List plugins"""
 
1949
    hidden = True
 
1950
    @display_command
 
1951
    def run(self):
 
1952
        import bzrlib.plugin
 
1953
        from inspect import getdoc
 
1954
        for name, plugin in bzrlib.plugin.all_plugins().items():
 
1955
            if hasattr(plugin, '__path__'):
 
1956
                print plugin.__path__[0]
 
1957
            elif hasattr(plugin, '__file__'):
 
1958
                print plugin.__file__
 
1959
            else:
 
1960
                print `plugin`
 
1961
                
 
1962
            d = getdoc(plugin)
 
1963
            if d:
 
1964
                print '\t', d.split('\n')[0]
 
1965
 
 
1966
 
 
1967
class cmd_testament(Command):
 
1968
    """Show testament (signing-form) of a revision."""
 
1969
    takes_options = ['revision', 'long']
 
1970
    takes_args = ['branch?']
 
1971
    @display_command
 
1972
    def run(self, branch=u'.', revision=None, long=False):
 
1973
        from bzrlib.testament import Testament
 
1974
        b = WorkingTree.open_containing(branch)[0].branch
 
1975
        b.lock_read()
 
1976
        try:
 
1977
            if revision is None:
 
1978
                rev_id = b.last_revision()
 
1979
            else:
 
1980
                rev_id = revision[0].in_history(b).rev_id
 
1981
            t = Testament.from_revision(b.repository, rev_id)
 
1982
            if long:
 
1983
                sys.stdout.writelines(t.as_text_lines())
 
1984
            else:
 
1985
                sys.stdout.write(t.as_short_text())
 
1986
        finally:
 
1987
            b.unlock()
 
1988
 
 
1989
 
 
1990
class cmd_annotate(Command):
 
1991
    """Show the origin of each line in a file.
 
1992
 
 
1993
    This prints out the given file with an annotation on the left side
 
1994
    indicating which revision, author and date introduced the change.
 
1995
 
 
1996
    If the origin is the same for a run of consecutive lines, it is 
 
1997
    shown only at the top, unless the --all option is given.
 
1998
    """
 
1999
    # TODO: annotate directories; showing when each file was last changed
 
2000
    # TODO: annotate a previous version of a file
 
2001
    # TODO: if the working copy is modified, show annotations on that 
 
2002
    #       with new uncommitted lines marked
 
2003
    aliases = ['blame', 'praise']
 
2004
    takes_args = ['filename']
 
2005
    takes_options = [Option('all', help='show annotations on all lines'),
 
2006
                     Option('long', help='show date in annotations'),
 
2007
                     ]
 
2008
 
 
2009
    @display_command
 
2010
    def run(self, filename, all=False, long=False):
 
2011
        from bzrlib.annotate import annotate_file
 
2012
        tree, relpath = WorkingTree.open_containing(filename)
 
2013
        branch = tree.branch
 
2014
        branch.lock_read()
 
2015
        try:
 
2016
            file_id = tree.inventory.path2id(relpath)
 
2017
            tree = branch.repository.revision_tree(branch.last_revision())
 
2018
            file_version = tree.inventory[file_id].revision
 
2019
            annotate_file(branch, file_version, file_id, long, all, sys.stdout)
 
2020
        finally:
 
2021
            branch.unlock()
 
2022
 
 
2023
 
 
2024
class cmd_re_sign(Command):
 
2025
    """Create a digital signature for an existing revision."""
 
2026
    # TODO be able to replace existing ones.
 
2027
 
 
2028
    hidden = True # is this right ?
 
2029
    takes_args = ['revision_id?']
 
2030
    takes_options = ['revision']
 
2031
    
 
2032
    def run(self, revision_id=None, revision=None):
 
2033
        import bzrlib.config as config
 
2034
        import bzrlib.gpg as gpg
 
2035
        if revision_id is not None and revision is not None:
 
2036
            raise BzrCommandError('You can only supply one of revision_id or --revision')
 
2037
        if revision_id is None and revision is None:
 
2038
            raise BzrCommandError('You must supply either --revision or a revision_id')
 
2039
        b = WorkingTree.open_containing(u'.')[0].branch
 
2040
        gpg_strategy = gpg.GPGStrategy(config.BranchConfig(b))
 
2041
        if revision_id is not None:
 
2042
            b.repository.sign_revision(revision_id, gpg_strategy)
 
2043
        elif revision is not None:
 
2044
            if len(revision) == 1:
 
2045
                revno, rev_id = revision[0].in_history(b)
 
2046
                b.repository.sign_revision(rev_id, gpg_strategy)
 
2047
            elif len(revision) == 2:
 
2048
                # are they both on rh- if so we can walk between them
 
2049
                # might be nice to have a range helper for arbitrary
 
2050
                # revision paths. hmm.
 
2051
                from_revno, from_revid = revision[0].in_history(b)
 
2052
                to_revno, to_revid = revision[1].in_history(b)
 
2053
                if to_revid is None:
 
2054
                    to_revno = b.revno()
 
2055
                if from_revno is None or to_revno is None:
 
2056
                    raise BzrCommandError('Cannot sign a range of non-revision-history revisions')
 
2057
                for revno in range(from_revno, to_revno + 1):
 
2058
                    b.repository.sign_revision(b.get_rev_id(revno), 
 
2059
                                               gpg_strategy)
 
2060
            else:
 
2061
                raise BzrCommandError('Please supply either one revision, or a range.')
 
2062
 
 
2063
 
 
2064
class cmd_uncommit(bzrlib.commands.Command):
 
2065
    """Remove the last committed revision.
 
2066
 
 
2067
    By supplying the --all flag, it will not only remove the entry 
 
2068
    from revision_history, but also remove all of the entries in the
 
2069
    stores.
 
2070
 
 
2071
    --verbose will print out what is being removed.
 
2072
    --dry-run will go through all the motions, but not actually
 
2073
    remove anything.
 
2074
    
 
2075
    In the future, uncommit will create a changeset, which can then
 
2076
    be re-applied.
 
2077
 
 
2078
    TODO: jam 20060108 Add an option to allow uncommit to remove unreferenced
 
2079
              information in 'branch-as-repostory' branches.
 
2080
    TODO: jam 20060108 Add the ability for uncommit to remove unreferenced
 
2081
              information in shared branches as well.
 
2082
    """
 
2083
    takes_options = ['verbose', 'revision',
 
2084
                    Option('dry-run', help='Don\'t actually make changes'),
 
2085
                    Option('force', help='Say yes to all questions.')]
 
2086
    takes_args = ['location?']
 
2087
    aliases = []
 
2088
 
 
2089
    def run(self, location=None, 
 
2090
            dry_run=False, verbose=False,
 
2091
            revision=None, force=False):
 
2092
        from bzrlib.branch import Branch
 
2093
        from bzrlib.log import log_formatter
 
2094
        import sys
 
2095
        from bzrlib.uncommit import uncommit
 
2096
 
 
2097
        if location is None:
 
2098
            location = u'.'
 
2099
        b, relpath = Branch.open_containing(location)
 
2100
 
 
2101
        if revision is None:
 
2102
            revno = b.revno()
 
2103
            rev_id = b.last_revision()
 
2104
        else:
 
2105
            revno, rev_id = revision[0].in_history(b)
 
2106
        if rev_id is None:
 
2107
            print 'No revisions to uncommit.'
 
2108
 
 
2109
        for r in range(revno, b.revno()+1):
 
2110
            rev_id = b.get_rev_id(r)
 
2111
            lf = log_formatter('short', to_file=sys.stdout,show_timezone='original')
 
2112
            lf.show(r, b.repository.get_revision(rev_id), None)
 
2113
 
 
2114
        if dry_run:
 
2115
            print 'Dry-run, pretending to remove the above revisions.'
 
2116
            if not force:
 
2117
                val = raw_input('Press <enter> to continue')
 
2118
        else:
 
2119
            print 'The above revision(s) will be removed.'
 
2120
            if not force:
 
2121
                val = raw_input('Are you sure [y/N]? ')
 
2122
                if val.lower() not in ('y', 'yes'):
 
2123
                    print 'Canceled'
 
2124
                    return 0
 
2125
 
 
2126
        uncommit(b, dry_run=dry_run, verbose=verbose,
 
2127
                revno=revno)
 
2128
 
 
2129
 
 
2130
def merge(other_revision, base_revision,
 
2131
          check_clean=True, ignore_zero=False,
 
2132
          this_dir=None, backup_files=False, merge_type=Merge3Merger,
 
2133
          file_list=None, show_base=False, reprocess=False):
 
2134
    """Merge changes into a tree.
 
2135
 
 
2136
    base_revision
 
2137
        list(path, revno) Base for three-way merge.  
 
2138
        If [None, None] then a base will be automatically determined.
 
2139
    other_revision
 
2140
        list(path, revno) Other revision for three-way merge.
 
2141
    this_dir
 
2142
        Directory to merge changes into; '.' by default.
 
2143
    check_clean
 
2144
        If true, this_dir must have no uncommitted changes before the
 
2145
        merge begins.
 
2146
    ignore_zero - If true, suppress the "zero conflicts" message when 
 
2147
        there are no conflicts; should be set when doing something we expect
 
2148
        to complete perfectly.
 
2149
    file_list - If supplied, merge only changes to selected files.
 
2150
 
 
2151
    All available ancestors of other_revision and base_revision are
 
2152
    automatically pulled into the branch.
 
2153
 
 
2154
    The revno may be -1 to indicate the last revision on the branch, which is
 
2155
    the typical case.
 
2156
 
 
2157
    This function is intended for use from the command line; programmatic
 
2158
    clients might prefer to call merge.merge_inner(), which has less magic 
 
2159
    behavior.
 
2160
    """
 
2161
    from bzrlib.merge import Merger
 
2162
    if this_dir is None:
 
2163
        this_dir = u'.'
 
2164
    this_tree = WorkingTree.open_containing(this_dir)[0]
 
2165
    if show_base and not merge_type is Merge3Merger:
 
2166
        raise BzrCommandError("Show-base is not supported for this merge"
 
2167
                              " type. %s" % merge_type)
 
2168
    if reprocess and not merge_type is Merge3Merger:
 
2169
        raise BzrCommandError("Reprocess is not supported for this merge"
 
2170
                              " type. %s" % merge_type)
 
2171
    if reprocess and show_base:
 
2172
        raise BzrCommandError("Cannot reprocess and show base.")
 
2173
    merger = Merger(this_tree.branch, this_tree=this_tree)
 
2174
    merger.check_basis(check_clean)
 
2175
    merger.set_other(other_revision)
 
2176
    merger.set_base(base_revision)
 
2177
    if merger.base_rev_id == merger.other_rev_id:
 
2178
        note('Nothing to do.')
 
2179
        return 0
 
2180
    merger.backup_files = backup_files
 
2181
    merger.merge_type = merge_type 
 
2182
    merger.set_interesting_files(file_list)
 
2183
    merger.show_base = show_base 
 
2184
    merger.reprocess = reprocess
 
2185
    conflicts = merger.do_merge()
 
2186
    merger.set_pending()
 
2187
    return conflicts
 
2188
 
 
2189
 
 
2190
# these get imported and then picked up by the scan for cmd_*
 
2191
# TODO: Some more consistent way to split command definitions across files;
 
2192
# we do need to load at least some information about them to know of 
 
2193
# aliases.
 
2194
from bzrlib.conflicts import cmd_resolve, cmd_conflicts, restore