/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

Quick, urgent fix for common_ancestor performance.

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