/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

Add a test environment for InterRepository objects, and remove the fetch corner case tests from test_repository.

Show diffs side-by-side

added added

removed removed

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