/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

  • Committer: Robert Collins
  • Date: 2006-02-24 23:13:20 UTC
  • mto: (1587.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 1588.
  • Revision ID: robertc@robertcollins.net-20060224231320-dbaf879d3070bfd7
Replace the slow topo_sort routine with a much faster one for non trivial datasets.

Show diffs side-by-side

added added

removed removed

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