← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/builtins.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2006-02-22 07:59:56 UTC
  • mfrom: (1553.5.33 bzr.mbp.locks)
  • Revision ID: pqm@pqm.ubuntu.com-20060222075956-fb281c427e571da6
add LockDir and related fixes

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/builtins.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
 
 
 
17
"""builtin bzr commands"""
 
18
 
 
19
 
 
20
import errno
 
21
import os
 
22
from shutil import rmtree
18
23
import sys
19
 
import os
20
24
 
21
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
22
40
import bzrlib.trace
23
 
from bzrlib.trace import mutter, note, log_error, warning
24
 
from bzrlib.errors import BzrError, BzrCheckError, BzrCommandError, NotBranchError
25
 
from bzrlib.errors import DivergedBranches
26
 
from bzrlib.branch import Branch
27
 
from bzrlib import BZRDIR
28
 
from bzrlib.commands import Command
29
 
 
 
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?)
30
77
 
31
78
class cmd_status(Command):
32
79
    """Display status summary.
67
114
    If a revision argument is given, the status is calculated against
68
115
    that revision, or between two revisions if two are provided.
69
116
    """
70
 
    # XXX: FIXME: bzr status should accept a -r option to show changes
71
 
    # relative to a revision, or between revisions
72
 
 
 
117
    
 
118
    # TODO: --no-recurse, --recurse options
 
119
    
73
120
    takes_args = ['file*']
74
 
    takes_options = ['all', 'show-ids']
 
121
    takes_options = ['all', 'show-ids', 'revision']
75
122
    aliases = ['st', 'stat']
76
123
    
 
124
    @display_command
77
125
    def run(self, all=False, show_ids=False, file_list=None, revision=None):
78
 
        if file_list:
79
 
            b = Branch.open_containing(file_list[0])
80
 
            file_list = [b.relpath(x) for x in file_list]
81
 
            # special case: only one path was given and it's the root
82
 
            # of the branch
83
 
            if file_list == ['']:
84
 
                file_list = None
85
 
        else:
86
 
            b = Branch.open_containing('.')
 
126
        tree, file_list = tree_files(file_list)
87
127
            
88
128
        from bzrlib.status import show_status
89
 
        show_status(b, show_unchanged=all, show_ids=show_ids,
 
129
        show_status(tree.branch, show_unchanged=all, show_ids=show_ids,
90
130
                    specific_files=file_list, revision=revision)
91
131
 
92
132
 
101
141
    takes_args = ['revision_id?']
102
142
    takes_options = ['revision']
103
143
    
 
144
    @display_command
104
145
    def run(self, revision_id=None, revision=None):
105
 
        from bzrlib.revisionspec import RevisionSpec
106
146
 
107
147
        if revision_id is not None and revision is not None:
108
148
            raise BzrCommandError('You can only supply one of revision_id or --revision')
109
149
        if revision_id is None and revision is None:
110
150
            raise BzrCommandError('You must supply either --revision or a revision_id')
111
 
        b = Branch.open_containing('.')
 
151
        b = WorkingTree.open_containing(u'.')[0].branch
112
152
        if revision_id is not None:
113
 
            sys.stdout.write(b.get_revision_xml_file(revision_id).read())
 
153
            sys.stdout.write(b.repository.get_revision_xml(revision_id))
114
154
        elif revision is not None:
115
155
            for rev in revision:
116
156
                if rev is None:
117
157
                    raise BzrCommandError('You cannot specify a NULL revision.')
118
158
                revno, rev_id = rev.in_history(b)
119
 
                sys.stdout.write(b.get_revision_xml_file(rev_id).read())
 
159
                sys.stdout.write(b.repository.get_revision_xml(rev_id))
120
160
    
121
161
 
122
162
class cmd_revno(Command):
123
163
    """Show current revision number.
124
164
 
125
165
    This is equal to the number of revisions on this branch."""
126
 
    def run(self):
127
 
        print Branch.open_containing('.').revno()
 
166
    takes_args = ['location?']
 
167
    @display_command
 
168
    def run(self, location=u'.'):
 
169
        print Branch.open_containing(location)[0].revno()
128
170
 
129
171
 
130
172
class cmd_revision_info(Command):
133
175
    hidden = True
134
176
    takes_args = ['revision_info*']
135
177
    takes_options = ['revision']
 
178
    @display_command
136
179
    def run(self, revision=None, revision_info_list=[]):
137
 
        from bzrlib.revisionspec import RevisionSpec
138
180
 
139
181
        revs = []
140
182
        if revision is not None:
145
187
        if len(revs) == 0:
146
188
            raise BzrCommandError('You must supply a revision identifier')
147
189
 
148
 
        b = Branch.open_containing('.')
 
190
        b = WorkingTree.open_containing(u'.')[0].branch
149
191
 
150
192
        for rev in revs:
151
193
            revinfo = rev.in_history(b)
177
219
    implicitly add the parent, and so on up to the root. This means
178
220
    you should never need to explictly add a directory, they'll just
179
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.
180
225
    """
181
226
    takes_args = ['file*']
182
 
    takes_options = ['no-recurse', 'quiet']
183
 
    
184
 
    def run(self, file_list, no_recurse=False, quiet=False):
185
 
        from bzrlib.add import smart_add, add_reporter_print, add_reporter_null
186
 
        if quiet:
187
 
            reporter = add_reporter_null
 
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
188
240
        else:
189
 
            reporter = add_reporter_print
190
 
        smart_add(file_list, not no_recurse, reporter)
 
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."
191
256
 
192
257
 
193
258
class cmd_mkdir(Command):
198
263
    takes_args = ['dir+']
199
264
 
200
265
    def run(self, dir_list):
201
 
        b = None
202
 
        
203
266
        for d in dir_list:
204
267
            os.mkdir(d)
205
 
            if not b:
206
 
                b = Branch.open_containing(d)
207
 
            b.add([d])
 
268
            wt, dd = WorkingTree.open_containing(d)
 
269
            wt.add([dd])
208
270
            print 'added', d
209
271
 
210
272
 
213
275
    takes_args = ['filename']
214
276
    hidden = True
215
277
    
 
278
    @display_command
216
279
    def run(self, filename):
217
 
        print Branch.open_containing(filename).relpath(filename)
218
 
 
 
280
        tree, relpath = WorkingTree.open_containing(filename)
 
281
        print relpath
219
282
 
220
283
 
221
284
class cmd_inventory(Command):
222
 
    """Show inventory of the current working copy or a revision."""
223
 
    takes_options = ['revision', 'show-ids']
 
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']
224
291
    
225
 
    def run(self, revision=None, show_ids=False):
226
 
        b = Branch.open_containing('.')
 
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]
227
297
        if revision is None:
228
 
            inv = b.read_working_inventory()
 
298
            inv = tree.read_working_inventory()
229
299
        else:
230
300
            if len(revision) > 1:
231
301
                raise BzrCommandError('bzr inventory --revision takes'
232
302
                    ' exactly one revision identifier')
233
 
            inv = b.get_revision_inventory(revision[0].in_history(b).rev_id)
 
303
            inv = tree.branch.repository.get_revision_inventory(
 
304
                revision[0].in_history(tree.branch).rev_id)
234
305
 
235
306
        for path, entry in inv.entries():
 
307
            if kind and kind != entry.kind:
 
308
                continue
236
309
            if show_ids:
237
310
                print '%-50s %s' % (path, entry.file_id)
238
311
            else:
249
322
    """
250
323
    takes_args = ['source$', 'dest']
251
324
    def run(self, source_list, dest):
252
 
        b = Branch.open_containing('.')
253
 
 
 
325
        tree, source_list = tree_files(source_list)
254
326
        # TODO: glob expansion on windows?
255
 
        b.move([b.relpath(s) for s in source_list], b.relpath(dest))
 
327
        tree.move(source_list, tree.relpath(dest))
256
328
 
257
329
 
258
330
class cmd_rename(Command):
272
344
    takes_args = ['from_name', 'to_name']
273
345
    
274
346
    def run(self, from_name, to_name):
275
 
        b = Branch.open_containing('.')
276
 
        b.rename_one(b.relpath(from_name), b.relpath(to_name))
277
 
 
 
347
        tree, (from_name, to_name) = tree_files((from_name, to_name))
 
348
        tree.rename_one(from_name, to_name)
278
349
 
279
350
 
280
351
class cmd_mv(Command):
294
365
    def run(self, names_list):
295
366
        if len(names_list) < 2:
296
367
            raise BzrCommandError("missing file argument")
297
 
        b = Branch.open_containing(names_list[0])
298
 
 
299
 
        rel_names = [b.relpath(x) for x in names_list]
 
368
        tree, rel_names = tree_files(names_list)
300
369
        
301
370
        if os.path.isdir(names_list[-1]):
302
371
            # move into existing directory
303
 
            for pair in b.move(rel_names[:-1], rel_names[-1]):
 
372
            for pair in tree.move(rel_names[:-1], rel_names[-1]):
304
373
                print "%s => %s" % pair
305
374
        else:
306
375
            if len(names_list) != 2:
307
376
                raise BzrCommandError('to mv multiple files the destination '
308
377
                                      'must be a versioned directory')
309
 
            b.rename_one(rel_names[0], rel_names[1])
 
378
            tree.rename_one(rel_names[0], rel_names[1])
310
379
            print "%s => %s" % (rel_names[0], rel_names[1])
311
380
            
312
381
    
313
 
 
314
 
 
315
382
class cmd_pull(Command):
316
383
    """Pull any changes from another branch into the current one.
317
384
 
318
 
    If the location is omitted, the last-used location will be used.
319
 
    Both the revision history and the working directory will be
320
 
    updated.
 
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.
321
388
 
322
389
    This command only works on branches that have not diverged.  Branches are
323
390
    considered diverged if both branches have had commits without first
324
391
    pulling from the other.
325
392
 
326
393
    If branches have diverged, you can use 'bzr merge' to pull the text changes
327
 
    from one into the other.
 
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.
328
399
    """
329
 
    takes_options = ['remember']
 
400
    takes_options = ['remember', 'overwrite', 'revision', 'verbose']
330
401
    takes_args = ['location?']
331
402
 
332
 
    def run(self, location=None, remember=False):
333
 
        from bzrlib.merge import merge
334
 
        import tempfile
335
 
        from shutil import rmtree
336
 
        import errno
337
 
        
338
 
        br_to = Branch.open_containing('.')
339
 
        stored_loc = br_to.get_parent()
 
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()
340
407
        if location is None:
341
408
            if stored_loc is None:
342
409
                raise BzrCommandError("No pull location known or specified.")
343
410
            else:
344
411
                print "Using saved location: %s" % stored_loc
345
412
                location = stored_loc
346
 
        cache_root = tempfile.mkdtemp()
 
413
 
347
414
        br_from = Branch.open(location)
348
 
        br_from.lock_read()
349
 
        try:
350
 
            br_from.setup_caching(cache_root)
351
 
            location = br_from.base
352
 
            old_revno = br_to.revno()
353
 
            old_revision_history = br_to.revision_history()
 
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:
354
512
            try:
355
 
                br_to.update_revisions(br_from)
356
 
            except DivergedBranches:
357
 
                raise BzrCommandError("These branches have diverged."
358
 
                    "  Try merge.")
359
 
            new_revision_history = br_to.revision_history()
360
 
            if new_revision_history != old_revision_history:
361
 
                merge(('.', -1), ('.', old_revno), check_clean=False)
362
 
            if stored_loc is None or remember:
363
 
                br_to.set_parent(location)
364
 
        finally:
365
 
            br_from.unlock()
366
 
            rmtree(cache_root)
 
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,))
367
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)
368
535
 
369
536
 
370
537
class cmd_branch(Command):
385
552
    aliases = ['get', 'clone']
386
553
 
387
554
    def run(self, from_location, to_location=None, revision=None, basis=None):
388
 
        from bzrlib.clone import copy_branch
389
 
        import tempfile
390
 
        import errno
391
 
        from shutil import rmtree
392
 
        cache_root = tempfile.mkdtemp()
393
555
        if revision is None:
394
556
            revision = [None]
395
557
        elif len(revision) > 1:
405
567
                raise
406
568
        br_from.lock_read()
407
569
        try:
408
 
            br_from.setup_caching(cache_root)
409
570
            if basis is not None:
410
 
                basis_branch = Branch.open_containing(basis)
 
571
                basis_dir = bzrdir.BzrDir.open_containing(basis)[0]
411
572
            else:
412
 
                basis_branch = None
 
573
                basis_dir = None
413
574
            if len(revision) == 1 and revision[0] is not None:
414
575
                revision_id = revision[0].in_history(br_from)[1]
415
576
            else:
416
 
                revision_id = None
 
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()
417
581
            if to_location is None:
418
582
                to_location = os.path.basename(from_location.rstrip("/\\"))
 
583
                name = None
 
584
            else:
 
585
                name = os.path.basename(to_location) + '\n'
419
586
            try:
420
587
                os.mkdir(to_location)
421
588
            except OSError, e:
428
595
                else:
429
596
                    raise
430
597
            try:
431
 
                copy_branch(br_from, to_location, revision_id, basis_branch)
 
598
                dir = br_from.bzrdir.sprout(to_location, revision_id, basis_dir)
 
599
                branch = dir.open_branch()
432
600
            except bzrlib.errors.NoSuchRevision:
433
601
                rmtree(to_location)
434
 
                msg = "The branch %s has no revision %d." % (from_location, revision[0])
 
602
                msg = "The branch %s has no revision %s." % (from_location, revision[0])
435
603
                raise BzrCommandError(msg)
436
604
            except bzrlib.errors.UnlistableBranch:
 
605
                rmtree(to_location)
437
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())
438
612
        finally:
439
613
            br_from.unlock()
440
 
            rmtree(cache_root)
 
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)
441
661
 
442
662
 
443
663
class cmd_renames(Command):
448
668
    # TODO: Only show renames under dir, rather than in the whole branch.
449
669
    takes_args = ['dir?']
450
670
 
451
 
    def run(self, dir='.'):
452
 
        b = Branch.open_containing(dir)
453
 
        old_inv = b.basis_tree().inventory
454
 
        new_inv = b.read_working_inventory()
 
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()
455
676
 
456
677
        renames = list(bzrlib.tree.find_renames(old_inv, new_inv))
457
678
        renames.sort()
459
680
            print "%s => %s" % (old_name, new_name)        
460
681
 
461
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
 
462
710
class cmd_info(Command):
463
711
    """Show statistical information about a branch."""
464
712
    takes_args = ['branch?']
465
713
    
 
714
    @display_command
466
715
    def run(self, branch=None):
467
 
        import info
468
 
        b = Branch.open_containing(branch)
469
 
        info.show_info(b)
 
716
        import bzrlib.info
 
717
        bzrlib.info.show_bzrdir_info(bzrdir.BzrDir.open_containing(branch)[0])
470
718
 
471
719
 
472
720
class cmd_remove(Command):
480
728
    aliases = ['rm']
481
729
    
482
730
    def run(self, file_list, verbose=False):
483
 
        b = Branch.open_containing(file_list[0])
484
 
        b.remove([b.relpath(f) for f in file_list], verbose=verbose)
 
731
        tree, file_list = tree_files(file_list)
 
732
        tree.remove(file_list, verbose=verbose)
485
733
 
486
734
 
487
735
class cmd_file_id(Command):
493
741
    """
494
742
    hidden = True
495
743
    takes_args = ['filename']
 
744
    @display_command
496
745
    def run(self, filename):
497
 
        b = Branch.open_containing(filename)
498
 
        i = b.inventory.path2id(b.relpath(filename))
 
746
        tree, relpath = WorkingTree.open_containing(filename)
 
747
        i = tree.inventory.path2id(relpath)
499
748
        if i == None:
500
749
            raise BzrError("%r is not a versioned file" % filename)
501
750
        else:
509
758
    starting at the branch root."""
510
759
    hidden = True
511
760
    takes_args = ['filename']
 
761
    @display_command
512
762
    def run(self, filename):
513
 
        b = Branch.open_containing(filename)
514
 
        inv = b.inventory
515
 
        fid = inv.path2id(b.relpath(filename))
 
763
        tree, relpath = WorkingTree.open_containing(filename)
 
764
        inv = tree.inventory
 
765
        fid = inv.path2id(relpath)
516
766
        if fid == None:
517
767
            raise BzrError("%r is not a versioned file" % filename)
518
768
        for fip in inv.get_idpath(fid):
522
772
class cmd_revision_history(Command):
523
773
    """Display list of revision ids on this branch."""
524
774
    hidden = True
 
775
    @display_command
525
776
    def run(self):
526
 
        for patchid in Branch.open_containing('.').revision_history():
 
777
        branch = WorkingTree.open_containing(u'.')[0].branch
 
778
        for patchid in branch.revision_history():
527
779
            print patchid
528
780
 
529
781
 
530
782
class cmd_ancestry(Command):
531
783
    """List all revisions merged into this branch."""
532
784
    hidden = True
 
785
    @display_command
533
786
    def run(self):
534
 
        b = find_branch('.')
535
 
        for revision_id in b.get_ancestry(b.last_revision()):
 
787
        tree = WorkingTree.open_containing(u'.')[0]
 
788
        b = tree.branch
 
789
        # FIXME. should be tree.last_revision
 
790
        for revision_id in b.repository.get_ancestry(b.last_revision()):
536
791
            print revision_id
537
792
 
538
793
 
539
 
class cmd_directories(Command):
540
 
    """Display list of versioned directories in this branch."""
541
 
    def run(self):
542
 
        for name, ie in Branch.open_containing('.').read_working_inventory().directories():
543
 
            if name == '':
544
 
                print '.'
545
 
            else:
546
 
                print name
547
 
 
548
 
 
549
794
class cmd_init(Command):
550
795
    """Make a directory into a versioned branch.
551
796
 
555
800
    Recipe for importing a tree of files:
556
801
        cd ~/project
557
802
        bzr init
558
 
        bzr add -v .
 
803
        bzr add .
559
804
        bzr status
560
805
        bzr commit -m 'imported project'
561
806
    """
562
 
    def run(self):
563
 
        Branch.initialize('.')
 
807
    takes_args = ['location?']
 
808
    def run(self, location=None):
 
809
        from bzrlib.branch import Branch
 
810
        if location is None:
 
811
            location = u'.'
 
812
        else:
 
813
            # The path has to exist to initialize a
 
814
            # branch inside of it.
 
815
            # Just using os.mkdir, since I don't
 
816
            # believe that we want to create a bunch of
 
817
            # locations if the user supplies an extended path
 
818
            if not os.path.exists(location):
 
819
                os.mkdir(location)
 
820
        bzrdir.BzrDir.create_standalone_workingtree(location)
564
821
 
565
822
 
566
823
class cmd_diff(Command):
592
849
    takes_options = ['revision', 'diff-options']
593
850
    aliases = ['di', 'dif']
594
851
 
 
852
    @display_command
595
853
    def run(self, revision=None, file_list=None, diff_options=None):
596
854
        from bzrlib.diff import show_diff
597
 
 
598
 
        if file_list:
599
 
            b = Branch.open_containing(file_list[0])
600
 
            file_list = [b.relpath(f) for f in file_list]
601
 
            if file_list == ['']:
602
 
                # just pointing to top-of-tree
603
 
                file_list = None
604
 
        else:
605
 
            b = Branch.open_containing('.')
606
 
 
 
855
        try:
 
856
            tree, file_list = internal_tree_files(file_list)
 
857
            b = None
 
858
            b2 = None
 
859
        except FileInWrongBranch:
 
860
            if len(file_list) != 2:
 
861
                raise BzrCommandError("Files are in different branches")
 
862
 
 
863
            b, file1 = Branch.open_containing(file_list[0])
 
864
            b2, file2 = Branch.open_containing(file_list[1])
 
865
            if file1 != "" or file2 != "":
 
866
                # FIXME diff those two files. rbc 20051123
 
867
                raise BzrCommandError("Files are in different branches")
 
868
            file_list = None
607
869
        if revision is not None:
608
 
            if len(revision) == 1:
609
 
                show_diff(b, revision[0], specific_files=file_list,
610
 
                          external_diff_options=diff_options)
 
870
            if b2 is not None:
 
871
                raise BzrCommandError("Can't specify -r with two branches")
 
872
            if (len(revision) == 1) or (revision[1].spec is None):
 
873
                return show_diff(tree.branch, revision[0], specific_files=file_list,
 
874
                                 external_diff_options=diff_options)
611
875
            elif len(revision) == 2:
612
 
                show_diff(b, revision[0], specific_files=file_list,
613
 
                          external_diff_options=diff_options,
614
 
                          revision2=revision[1])
 
876
                return show_diff(tree.branch, revision[0], specific_files=file_list,
 
877
                                 external_diff_options=diff_options,
 
878
                                 revision2=revision[1])
615
879
            else:
616
880
                raise BzrCommandError('bzr diff --revision takes exactly one or two revision identifiers')
617
881
        else:
618
 
            show_diff(b, None, specific_files=file_list,
619
 
                      external_diff_options=diff_options)
620
 
 
621
 
        
 
882
            if b is not None:
 
883
                return show_diff(b, None, specific_files=file_list,
 
884
                                 external_diff_options=diff_options, b2=b2)
 
885
            else:
 
886
                return show_diff(tree.branch, None, specific_files=file_list,
 
887
                                 external_diff_options=diff_options)
622
888
 
623
889
 
624
890
class cmd_deleted(Command):
630
896
    # directories with readdir, rather than stating each one.  Same
631
897
    # level of effort but possibly much less IO.  (Or possibly not,
632
898
    # if the directories are very large...)
 
899
    @display_command
633
900
    def run(self, show_ids=False):
634
 
        b = Branch.open_containing('.')
635
 
        old = b.basis_tree()
636
 
        new = b.working_tree()
 
901
        tree = WorkingTree.open_containing(u'.')[0]
 
902
        old = tree.basis_tree()
637
903
        for path, ie in old.inventory.iter_entries():
638
 
            if not new.has_id(ie.file_id):
 
904
            if not tree.has_id(ie.file_id):
639
905
                if show_ids:
640
906
                    print '%-50s %s' % (path, ie.file_id)
641
907
                else:
645
911
class cmd_modified(Command):
646
912
    """List files modified in working tree."""
647
913
    hidden = True
 
914
    @display_command
648
915
    def run(self):
649
916
        from bzrlib.delta import compare_trees
650
917
 
651
 
        b = Branch.open_containing('.')
652
 
        td = compare_trees(b.basis_tree(), b.working_tree())
 
918
        tree = WorkingTree.open_containing(u'.')[0]
 
919
        td = compare_trees(tree.basis_tree(), tree)
653
920
 
654
921
        for path, id, kind, text_modified, meta_modified in td.modified:
655
922
            print path
659
926
class cmd_added(Command):
660
927
    """List files added in working tree."""
661
928
    hidden = True
 
929
    @display_command
662
930
    def run(self):
663
 
        b = Branch.open_containing('.')
664
 
        wt = b.working_tree()
665
 
        basis_inv = b.basis_tree().inventory
 
931
        wt = WorkingTree.open_containing(u'.')[0]
 
932
        basis_inv = wt.basis_tree().inventory
666
933
        inv = wt.inventory
667
934
        for file_id in inv:
668
935
            if file_id in basis_inv:
669
936
                continue
670
937
            path = inv.id2path(file_id)
671
 
            if not os.access(b.abspath(path), os.F_OK):
 
938
            if not os.access(bzrlib.osutils.abspath(path), os.F_OK):
672
939
                continue
673
940
            print path
674
941
                
680
947
    The root is the nearest enclosing directory with a .bzr control
681
948
    directory."""
682
949
    takes_args = ['filename?']
 
950
    @display_command
683
951
    def run(self, filename=None):
684
952
        """Print the branch root."""
685
 
        b = Branch.open_containing(filename)
686
 
        print b.base
 
953
        tree = WorkingTree.open_containing(filename)[0]
 
954
        print tree.basedir
687
955
 
688
956
 
689
957
class cmd_log(Command):
690
958
    """Show log of this branch.
691
959
 
692
 
    To request a range of logs, you can use the command -r begin:end
693
 
    -r revision requests a specific revision, -r :end or -r begin: are
 
960
    To request a range of logs, you can use the command -r begin..end
 
961
    -r revision requests a specific revision, -r ..end or -r begin.. are
694
962
    also valid.
695
 
 
696
 
    --message allows you to give a regular expression, which will be evaluated
697
 
    so that only matching entries will be displayed.
698
963
    """
699
964
 
700
965
    # TODO: Make --revision support uuid: and hash: [future tag:] notation.
701
966
 
702
967
    takes_args = ['filename?']
703
 
    takes_options = ['forward', 'timezone', 'verbose', 'show-ids', 'revision',
704
 
                     'long', 'message', 'short',]
705
 
    
 
968
    takes_options = [Option('forward', 
 
969
                            help='show from oldest to newest'),
 
970
                     'timezone', 'verbose', 
 
971
                     'show-ids', 'revision',
 
972
                     'log-format',
 
973
                     'line', 'long', 
 
974
                     Option('message',
 
975
                            help='show revisions whose message matches this regexp',
 
976
                            type=str),
 
977
                     'short',
 
978
                     ]
 
979
    @display_command
706
980
    def run(self, filename=None, timezone='original',
707
981
            verbose=False,
708
982
            show_ids=False,
709
983
            forward=False,
710
984
            revision=None,
 
985
            log_format=None,
711
986
            message=None,
712
987
            long=False,
713
 
            short=False):
 
988
            short=False,
 
989
            line=False):
714
990
        from bzrlib.log import log_formatter, show_log
715
991
        import codecs
716
 
 
 
992
        assert message is None or isinstance(message, basestring), \
 
993
            "invalid message argument %r" % message
717
994
        direction = (forward and 'forward') or 'reverse'
718
995
        
 
996
        # log everything
 
997
        file_id = None
719
998
        if filename:
720
 
            b = Branch.open_containing(filename)
721
 
            fp = b.relpath(filename)
722
 
            if fp:
723
 
                file_id = b.read_working_inventory().path2id(fp)
724
 
            else:
725
 
                file_id = None  # points to branch root
 
999
            # find the file id to log:
 
1000
 
 
1001
            dir, fp = bzrdir.BzrDir.open_containing(filename)
 
1002
            b = dir.open_branch()
 
1003
            if fp != '':
 
1004
                try:
 
1005
                    # might be a tree:
 
1006
                    inv = dir.open_workingtree().inventory
 
1007
                except (errors.NotBranchError, errors.NotLocalUrl):
 
1008
                    # either no tree, or is remote.
 
1009
                    inv = b.basis_tree().inventory
 
1010
                file_id = inv.path2id(fp)
726
1011
        else:
727
 
            b = Branch.open_containing('.')
728
 
            file_id = None
 
1012
            # local dir only
 
1013
            # FIXME ? log the current subdir only RBC 20060203 
 
1014
            dir, relpath = bzrdir.BzrDir.open_containing('.')
 
1015
            b = dir.open_branch()
729
1016
 
730
1017
        if revision is None:
731
1018
            rev1 = None
733
1020
        elif len(revision) == 1:
734
1021
            rev1 = rev2 = revision[0].in_history(b).revno
735
1022
        elif len(revision) == 2:
736
 
            rev1 = revision[0].in_history(b).revno
737
 
            rev2 = revision[1].in_history(b).revno
 
1023
            if revision[0].spec is None:
 
1024
                # missing begin-range means first revision
 
1025
                rev1 = 1
 
1026
            else:
 
1027
                rev1 = revision[0].in_history(b).revno
 
1028
 
 
1029
            if revision[1].spec is None:
 
1030
                # missing end-range means last known revision
 
1031
                rev2 = b.revno()
 
1032
            else:
 
1033
                rev2 = revision[1].in_history(b).revno
738
1034
        else:
739
1035
            raise BzrCommandError('bzr log --revision takes one or two values.')
740
1036
 
741
 
        if rev1 == 0:
742
 
            rev1 = None
743
 
        if rev2 == 0:
744
 
            rev2 = None
 
1037
        # By this point, the revision numbers are converted to the +ve
 
1038
        # form if they were supplied in the -ve form, so we can do
 
1039
        # this comparison in relative safety
 
1040
        if rev1 > rev2:
 
1041
            (rev2, rev1) = (rev1, rev2)
745
1042
 
746
 
        mutter('encoding log as %r' % bzrlib.user_encoding)
 
1043
        mutter('encoding log as %r', bzrlib.user_encoding)
747
1044
 
748
1045
        # use 'replace' so that we don't abort if trying to write out
749
1046
        # in e.g. the default C locale.
750
1047
        outf = codecs.getwriter(bzrlib.user_encoding)(sys.stdout, errors='replace')
751
1048
 
752
 
        if not short:
753
 
            log_format = 'long'
754
 
        else:
755
 
            log_format = 'short'
 
1049
        if (log_format == None):
 
1050
            default = bzrlib.config.BranchConfig(b).log_format()
 
1051
            log_format = get_log_format(long=long, short=short, line=line, default=default)
 
1052
 
756
1053
        lf = log_formatter(log_format,
757
1054
                           show_ids=show_ids,
758
1055
                           to_file=outf,
768
1065
                 search=message)
769
1066
 
770
1067
 
 
1068
def get_log_format(long=False, short=False, line=False, default='long'):
 
1069
    log_format = default
 
1070
    if long:
 
1071
        log_format = 'long'
 
1072
    if short:
 
1073
        log_format = 'short'
 
1074
    if line:
 
1075
        log_format = 'line'
 
1076
    return log_format
 
1077
 
771
1078
 
772
1079
class cmd_touching_revisions(Command):
773
1080
    """Return revision-ids which affected a particular file.
775
1082
    A more user-friendly interface is "bzr log FILE"."""
776
1083
    hidden = True
777
1084
    takes_args = ["filename"]
 
1085
    @display_command
778
1086
    def run(self, filename):
779
 
        b = Branch.open_containing(filename)
780
 
        inv = b.read_working_inventory()
781
 
        file_id = inv.path2id(b.relpath(filename))
 
1087
        tree, relpath = WorkingTree.open_containing(filename)
 
1088
        b = tree.branch
 
1089
        inv = tree.read_working_inventory()
 
1090
        file_id = inv.path2id(relpath)
782
1091
        for revno, revision_id, what in bzrlib.log.find_touching_revisions(b, file_id):
783
1092
            print "%6d %s" % (revno, what)
784
1093
 
788
1097
    """
789
1098
    # TODO: Take a revision or remote path and list that tree instead.
790
1099
    hidden = True
791
 
    def run(self, revision=None, verbose=False):
792
 
        b = Branch.open_containing('.')
793
 
        if revision == None:
794
 
            tree = b.working_tree()
795
 
        else:
796
 
            tree = b.revision_tree(revision.in_history(b).rev_id)
 
1100
    takes_options = ['verbose', 'revision',
 
1101
                     Option('non-recursive',
 
1102
                            help='don\'t recurse into sub-directories'),
 
1103
                     Option('from-root',
 
1104
                            help='Print all paths from the root of the branch.'),
 
1105
                     Option('unknown', help='Print unknown files'),
 
1106
                     Option('versioned', help='Print versioned files'),
 
1107
                     Option('ignored', help='Print ignored files'),
 
1108
 
 
1109
                     Option('null', help='Null separate the files'),
 
1110
                    ]
 
1111
    @display_command
 
1112
    def run(self, revision=None, verbose=False, 
 
1113
            non_recursive=False, from_root=False,
 
1114
            unknown=False, versioned=False, ignored=False,
 
1115
            null=False):
 
1116
 
 
1117
        if verbose and null:
 
1118
            raise BzrCommandError('Cannot set both --verbose and --null')
 
1119
        all = not (unknown or versioned or ignored)
 
1120
 
 
1121
        selection = {'I':ignored, '?':unknown, 'V':versioned}
 
1122
 
 
1123
        tree, relpath = WorkingTree.open_containing(u'.')
 
1124
        if from_root:
 
1125
            relpath = u''
 
1126
        elif relpath:
 
1127
            relpath += '/'
 
1128
        if revision is not None:
 
1129
            tree = tree.branch.repository.revision_tree(
 
1130
                revision[0].in_history(tree.branch).rev_id)
797
1131
        for fp, fc, kind, fid, entry in tree.list_files():
798
 
            if verbose:
799
 
                kindch = entry.kind_character()
800
 
                print '%-8s %s%s' % (fc, fp, kindch)
801
 
            else:
802
 
                print fp
803
 
 
 
1132
            if fp.startswith(relpath):
 
1133
                fp = fp[len(relpath):]
 
1134
                if non_recursive and '/' in fp:
 
1135
                    continue
 
1136
                if not all and not selection[fc]:
 
1137
                    continue
 
1138
                if verbose:
 
1139
                    kindch = entry.kind_character()
 
1140
                    print '%-8s %s%s' % (fc, fp, kindch)
 
1141
                elif null:
 
1142
                    sys.stdout.write(fp)
 
1143
                    sys.stdout.write('\0')
 
1144
                    sys.stdout.flush()
 
1145
                else:
 
1146
                    print fp
804
1147
 
805
1148
 
806
1149
class cmd_unknowns(Command):
807
1150
    """List unknown files."""
 
1151
    @display_command
808
1152
    def run(self):
809
1153
        from bzrlib.osutils import quotefn
810
 
        for f in Branch.open_containing('.').unknowns():
 
1154
        for f in WorkingTree.open_containing(u'.')[0].unknowns():
811
1155
            print quotefn(f)
812
1156
 
813
1157
 
814
 
 
815
1158
class cmd_ignore(Command):
816
1159
    """Ignore a command or pattern.
817
1160
 
837
1180
        from bzrlib.atomicfile import AtomicFile
838
1181
        import os.path
839
1182
 
840
 
        b = Branch.open_containing('.')
841
 
        ifn = b.abspath('.bzrignore')
 
1183
        tree, relpath = WorkingTree.open_containing(u'.')
 
1184
        ifn = tree.abspath('.bzrignore')
842
1185
 
843
1186
        if os.path.exists(ifn):
844
1187
            f = open(ifn, 'rt')
863
1206
        finally:
864
1207
            f.close()
865
1208
 
866
 
        inv = b.working_tree().inventory
 
1209
        inv = tree.inventory
867
1210
        if inv.path2id('.bzrignore'):
868
1211
            mutter('.bzrignore is already versioned')
869
1212
        else:
870
1213
            mutter('need to make new .bzrignore file versioned')
871
 
            b.add(['.bzrignore'])
872
 
 
 
1214
            tree.add(['.bzrignore'])
873
1215
 
874
1216
 
875
1217
class cmd_ignored(Command):
876
1218
    """List ignored files and the patterns that matched them.
877
1219
 
878
1220
    See also: bzr ignore"""
 
1221
    @display_command
879
1222
    def run(self):
880
 
        tree = Branch.open_containing('.').working_tree()
 
1223
        tree = WorkingTree.open_containing(u'.')[0]
881
1224
        for path, file_class, kind, file_id, entry in tree.list_files():
882
1225
            if file_class != 'I':
883
1226
                continue
895
1238
    hidden = True
896
1239
    takes_args = ['revno']
897
1240
    
 
1241
    @display_command
898
1242
    def run(self, revno):
899
1243
        try:
900
1244
            revno = int(revno)
901
1245
        except ValueError:
902
1246
            raise BzrCommandError("not a valid revision-number: %r" % revno)
903
1247
 
904
 
        print Branch.open_containing('.').get_rev_id(revno)
 
1248
        print WorkingTree.open_containing(u'.')[0].branch.get_rev_id(revno)
905
1249
 
906
1250
 
907
1251
class cmd_export(Command):
914
1258
    is found exports to a directory (equivalent to --format=dir).
915
1259
 
916
1260
    Root may be the top directory for tar, tgz and tbz2 formats. If none
917
 
    is given, the top directory will be the root name of the file."""
918
 
    # TODO: list known exporters
 
1261
    is given, the top directory will be the root name of the file.
 
1262
 
 
1263
    Note: export of tree with non-ascii filenames to zip is not supported.
 
1264
 
 
1265
     Supported formats       Autodetected by extension
 
1266
     -----------------       -------------------------
 
1267
         dir                            -
 
1268
         tar                          .tar
 
1269
         tbz2                    .tar.bz2, .tbz2
 
1270
         tgz                      .tar.gz, .tgz
 
1271
         zip                          .zip
 
1272
    """
919
1273
    takes_args = ['dest']
920
1274
    takes_options = ['revision', 'format', 'root']
921
1275
    def run(self, dest, revision=None, format=None, root=None):
922
1276
        import os.path
923
 
        b = Branch.open_containing('.')
 
1277
        from bzrlib.export import export
 
1278
        tree = WorkingTree.open_containing(u'.')[0]
 
1279
        b = tree.branch
924
1280
        if revision is None:
 
1281
            # should be tree.last_revision  FIXME
925
1282
            rev_id = b.last_revision()
926
1283
        else:
927
1284
            if len(revision) != 1:
928
1285
                raise BzrError('bzr export --revision takes exactly 1 argument')
929
1286
            rev_id = revision[0].in_history(b).rev_id
930
 
        t = b.revision_tree(rev_id)
931
 
        arg_root, ext = os.path.splitext(os.path.basename(dest))
932
 
        if ext in ('.gz', '.bz2'):
933
 
            new_root, new_ext = os.path.splitext(arg_root)
934
 
            if new_ext == '.tar':
935
 
                arg_root = new_root
936
 
                ext = new_ext + ext
937
 
        if root is None:
938
 
            root = arg_root
939
 
        if not format:
940
 
            if ext in (".tar",):
941
 
                format = "tar"
942
 
            elif ext in (".tar.gz", ".tgz"):
943
 
                format = "tgz"
944
 
            elif ext in (".tar.bz2", ".tbz2"):
945
 
                format = "tbz2"
946
 
            else:
947
 
                format = "dir"
948
 
        t.export(dest, format, root)
 
1287
        t = b.repository.revision_tree(rev_id)
 
1288
        try:
 
1289
            export(t, dest, format, root)
 
1290
        except errors.NoSuchExportFormat, e:
 
1291
            raise BzrCommandError('Unsupported export format: %s' % e.format)
949
1292
 
950
1293
 
951
1294
class cmd_cat(Command):
954
1297
    takes_options = ['revision']
955
1298
    takes_args = ['filename']
956
1299
 
 
1300
    @display_command
957
1301
    def run(self, filename, revision=None):
958
 
        if revision is None:
959
 
            raise BzrCommandError("bzr cat requires a revision number")
960
 
        elif len(revision) != 1:
 
1302
        if revision is not None and len(revision) != 1:
961
1303
            raise BzrCommandError("bzr cat --revision takes exactly one number")
962
 
        b = Branch.open_containing('.')
963
 
        b.print_file(b.relpath(filename), revision[0].in_history(b).revno)
 
1304
        tree = None
 
1305
        try:
 
1306
            tree, relpath = WorkingTree.open_containing(filename)
 
1307
            b = tree.branch
 
1308
        except NotBranchError:
 
1309
            pass
 
1310
 
 
1311
        if tree is None:
 
1312
            b, relpath = Branch.open_containing(filename)
 
1313
        if revision is None:
 
1314
            revision_id = b.last_revision()
 
1315
        else:
 
1316
            revision_id = revision[0].in_history(b).rev_id
 
1317
        b.print_file(relpath, revision_id)
964
1318
 
965
1319
 
966
1320
class cmd_local_time_offset(Command):
967
1321
    """Show the offset in seconds from GMT to local time."""
968
1322
    hidden = True    
 
1323
    @display_command
969
1324
    def run(self):
970
1325
        print bzrlib.osutils.local_time_offset()
971
1326
 
986
1341
    """
987
1342
    # TODO: Run hooks on tree to-be-committed, and after commit.
988
1343
 
989
 
    # TODO: Strict commit that fails if there are unknown or deleted files.
 
1344
    # TODO: Strict commit that fails if there are deleted files.
 
1345
    #       (what does "deleted files" mean ??)
 
1346
 
990
1347
    # TODO: Give better message for -s, --summary, used by tla people
991
1348
 
992
1349
    # XXX: verbose currently does nothing
993
1350
 
994
1351
    takes_args = ['selected*']
995
 
    takes_options = ['message', 'file', 'verbose', 'unchanged']
 
1352
    takes_options = ['message', 'verbose', 
 
1353
                     Option('unchanged',
 
1354
                            help='commit even if nothing has changed'),
 
1355
                     Option('file', type=str, 
 
1356
                            argname='msgfile',
 
1357
                            help='file containing commit message'),
 
1358
                     Option('strict',
 
1359
                            help="refuse to commit if there are unknown "
 
1360
                            "files in the working tree."),
 
1361
                     ]
996
1362
    aliases = ['ci', 'checkin']
997
1363
 
998
1364
    def run(self, message=None, file=None, verbose=True, selected_list=None,
999
 
            unchanged=False):
1000
 
        from bzrlib.errors import PointlessCommit, ConflictsInTree
1001
 
        from bzrlib.msgeditor import edit_commit_message
 
1365
            unchanged=False, strict=False):
 
1366
        from bzrlib.errors import (PointlessCommit, ConflictsInTree,
 
1367
                StrictCommitFailed)
 
1368
        from bzrlib.msgeditor import edit_commit_message, \
 
1369
                make_commit_message_template
1002
1370
        from bzrlib.status import show_status
1003
 
        from cStringIO import StringIO
1004
 
 
1005
 
        b = Branch.open_containing('.')
1006
 
        if selected_list:
1007
 
            selected_list = [b.relpath(s) for s in selected_list]
1008
 
 
1009
 
            
 
1371
        from tempfile import TemporaryFile
 
1372
        import codecs
 
1373
 
 
1374
        # TODO: Need a blackbox test for invoking the external editor; may be
 
1375
        # slightly problematic to run this cross-platform.
 
1376
 
 
1377
        # TODO: do more checks that the commit will succeed before 
 
1378
        # spending the user's valuable time typing a commit message.
 
1379
        #
 
1380
        # TODO: if the commit *does* happen to fail, then save the commit 
 
1381
        # message to a temporary file where it can be recovered
 
1382
        tree, selected_list = tree_files(selected_list)
1010
1383
        if message is None and not file:
1011
 
            catcher = StringIO()
1012
 
            show_status(b, specific_files=selected_list,
1013
 
                        to_file=catcher)
1014
 
            message = edit_commit_message(catcher.getvalue())
1015
 
 
 
1384
            template = make_commit_message_template(tree, selected_list)
 
1385
            message = edit_commit_message(template)
1016
1386
            if message is None:
1017
1387
                raise BzrCommandError("please specify a commit message"
1018
1388
                                      " with either --message or --file")
1027
1397
                raise BzrCommandError("empty commit message specified")
1028
1398
            
1029
1399
        try:
1030
 
            b.commit(message,
1031
 
                     specific_files=selected_list,
1032
 
                     allow_pointless=unchanged)
 
1400
            tree.commit(message, specific_files=selected_list,
 
1401
                        allow_pointless=unchanged, strict=strict)
1033
1402
        except PointlessCommit:
1034
1403
            # FIXME: This should really happen before the file is read in;
1035
1404
            # perhaps prepare the commit; get the message; then actually commit
1038
1407
        except ConflictsInTree:
1039
1408
            raise BzrCommandError("Conflicts detected in working tree.  "
1040
1409
                'Use "bzr conflicts" to list, "bzr resolve FILE" to resolve.')
 
1410
        except StrictCommitFailed:
 
1411
            raise BzrCommandError("Commit refused because there are unknown "
 
1412
                                  "files in the working tree.")
 
1413
        note('Committed revision %d.' % (tree.branch.revno(),))
1041
1414
 
1042
1415
 
1043
1416
class cmd_check(Command):
1046
1419
    This command checks various invariants about the branch storage to
1047
1420
    detect data corruption or bzr bugs.
1048
1421
    """
1049
 
    takes_args = ['dir?']
 
1422
    takes_args = ['branch?']
 
1423
    takes_options = ['verbose']
1050
1424
 
1051
 
    def run(self, dir='.'):
 
1425
    def run(self, branch=None, verbose=False):
1052
1426
        from bzrlib.check import check
1053
 
 
1054
 
        check(Branch.open_containing(dir))
 
1427
        if branch is None:
 
1428
            tree = WorkingTree.open_containing()[0]
 
1429
            branch = tree.branch
 
1430
        else:
 
1431
            branch = Branch.open(branch)
 
1432
        check(branch, verbose)
1055
1433
 
1056
1434
 
1057
1435
class cmd_scan_cache(Command):
1059
1437
    def run(self):
1060
1438
        from bzrlib.hashcache import HashCache
1061
1439
 
1062
 
        c = HashCache('.')
 
1440
        c = HashCache(u'.')
1063
1441
        c.read()
1064
1442
        c.scan()
1065
1443
            
1071
1449
 
1072
1450
        if c.needs_write:
1073
1451
            c.write()
1074
 
            
 
1452
 
 
1453
 
 
1454
def get_format_type(typestring):
 
1455
    """Parse and return a format specifier."""
 
1456
    if typestring == "metadir":
 
1457
        return bzrdir.BzrDirMetaFormat1()
 
1458
    if typestring == "knit":
 
1459
        format = bzrdir.BzrDirMetaFormat1()
 
1460
        format.repository_format = bzrlib.repository.RepositoryFormatKnit1()
 
1461
        return format
 
1462
    msg = "No known bzr-dir format %s. Supported types are: metadir\n" %\
 
1463
        (typestring)
 
1464
    raise BzrCommandError(msg)
1075
1465
 
1076
1466
 
1077
1467
class cmd_upgrade(Command):
1078
1468
    """Upgrade branch storage to current format.
1079
1469
 
1080
1470
    The check command or bzr developers may sometimes advise you to run
1081
 
    this command.
1082
 
 
1083
 
    This version of this command upgrades from the full-text storage
1084
 
    used by bzr 0.0.8 and earlier to the weave format (v5).
 
1471
    this command. When the default format has changed you may also be warned
 
1472
    during other operations to upgrade.
1085
1473
    """
1086
 
    takes_args = ['dir?']
1087
 
 
1088
 
    def run(self, dir='.'):
 
1474
    takes_args = ['url?']
 
1475
    takes_options = [
 
1476
                     Option('format', 
 
1477
                            help='Upgrade to a specific format rather than the'
 
1478
                                 ' current default format. Currently this '
 
1479
                                 ' option only accepts =metadir',
 
1480
                            type=get_format_type),
 
1481
                    ]
 
1482
 
 
1483
 
 
1484
    def run(self, url='.', format=None):
1089
1485
        from bzrlib.upgrade import upgrade
1090
 
        upgrade(dir)
 
1486
        upgrade(url, format)
1091
1487
 
1092
1488
 
1093
1489
class cmd_whoami(Command):
1094
1490
    """Show bzr user id."""
1095
1491
    takes_options = ['email']
1096
1492
    
 
1493
    @display_command
1097
1494
    def run(self, email=False):
1098
1495
        try:
1099
 
            b = bzrlib.branch.Branch.open_containing('.')
 
1496
            b = WorkingTree.open_containing(u'.')[0].branch
 
1497
            config = bzrlib.config.BranchConfig(b)
1100
1498
        except NotBranchError:
1101
 
            b = None
 
1499
            config = bzrlib.config.GlobalConfig()
1102
1500
        
1103
1501
        if email:
1104
 
            print bzrlib.config.user_email(b)
1105
 
        else:
1106
 
            print bzrlib.config.username(b)
 
1502
            print config.user_email()
 
1503
        else:
 
1504
            print config.username()
 
1505
 
 
1506
 
 
1507
class cmd_nick(Command):
 
1508
    """Print or set the branch nickname.  
 
1509
 
 
1510
    If unset, the tree root directory name is used as the nickname
 
1511
    To print the current nickname, execute with no argument.  
 
1512
    """
 
1513
    takes_args = ['nickname?']
 
1514
    def run(self, nickname=None):
 
1515
        branch = Branch.open_containing(u'.')[0]
 
1516
        if nickname is None:
 
1517
            self.printme(branch)
 
1518
        else:
 
1519
            branch.nick = nickname
 
1520
 
 
1521
    @display_command
 
1522
    def printme(self, branch):
 
1523
        print branch.nick 
1107
1524
 
1108
1525
 
1109
1526
class cmd_selftest(Command):
1112
1529
    This creates temporary test directories in the working directory,
1113
1530
    but not existing data is affected.  These directories are deleted
1114
1531
    if the tests pass, or left behind to help in debugging if they
1115
 
    fail.
 
1532
    fail and --keep-output is specified.
1116
1533
    
1117
1534
    If arguments are given, they are regular expressions that say
1118
 
    which tests should run."""
 
1535
    which tests should run.
 
1536
 
 
1537
    If the global option '--no-plugins' is given, plugins are not loaded
 
1538
    before running the selftests.  This has two effects: features provided or
 
1539
    modified by plugins will not be tested, and tests provided by plugins will
 
1540
    not be run.
 
1541
 
 
1542
    examples:
 
1543
        bzr selftest ignore
 
1544
        bzr --no-plugins selftest -v
 
1545
    """
1119
1546
    # TODO: --list should give a list of all available tests
 
1547
 
 
1548
    # NB: this is used from the class without creating an instance, which is
 
1549
    # why it does not have a self parameter.
 
1550
    def get_transport_type(typestring):
 
1551
        """Parse and return a transport specifier."""
 
1552
        if typestring == "sftp":
 
1553
            from bzrlib.transport.sftp import SFTPAbsoluteServer
 
1554
            return SFTPAbsoluteServer
 
1555
        if typestring == "memory":
 
1556
            from bzrlib.transport.memory import MemoryServer
 
1557
            return MemoryServer
 
1558
        msg = "No known transport type %s. Supported types are: sftp\n" %\
 
1559
            (typestring)
 
1560
        raise BzrCommandError(msg)
 
1561
 
1120
1562
    hidden = True
1121
1563
    takes_args = ['testspecs*']
1122
 
    takes_options = ['verbose']
1123
 
    def run(self, testspecs_list=None, verbose=False):
 
1564
    takes_options = ['verbose',
 
1565
                     Option('one', help='stop when one test fails'),
 
1566
                     Option('keep-output', 
 
1567
                            help='keep output directories when tests fail'),
 
1568
                     Option('transport', 
 
1569
                            help='Use a different transport by default '
 
1570
                                 'throughout the test suite.',
 
1571
                            type=get_transport_type),
 
1572
                    ]
 
1573
 
 
1574
    def run(self, testspecs_list=None, verbose=False, one=False,
 
1575
            keep_output=False, transport=None):
1124
1576
        import bzrlib.ui
1125
 
        from bzrlib.selftest import selftest
 
1577
        from bzrlib.tests import selftest
1126
1578
        # we don't want progress meters from the tests to go to the
1127
1579
        # real output; and we don't want log messages cluttering up
1128
1580
        # the real logs.
1135
1587
            else:
1136
1588
                pattern = ".*"
1137
1589
            result = selftest(verbose=verbose, 
1138
 
                              pattern=pattern)
 
1590
                              pattern=pattern,
 
1591
                              stop_on_failure=one, 
 
1592
                              keep_output=keep_output,
 
1593
                              transport=transport)
1139
1594
            if result:
1140
1595
                bzrlib.trace.info('tests passed')
1141
1596
            else:
1145
1600
            bzrlib.ui.ui_factory = save_ui
1146
1601
 
1147
1602
 
 
1603
def _get_bzr_branch():
 
1604
    """If bzr is run from a branch, return Branch or None"""
 
1605
    import bzrlib.errors
 
1606
    from bzrlib.branch import Branch
 
1607
    from bzrlib.osutils import abspath
 
1608
    from os.path import dirname
 
1609
    
 
1610
    try:
 
1611
        branch = Branch.open(dirname(abspath(dirname(__file__))))
 
1612
        return branch
 
1613
    except bzrlib.errors.BzrError:
 
1614
        return None
 
1615
    
 
1616
 
1148
1617
def show_version():
1149
1618
    print "bzr (bazaar-ng) %s" % bzrlib.__version__
1150
1619
    # is bzrlib itself in a branch?
1151
 
    bzrrev = bzrlib.get_bzr_revision()
1152
 
    if bzrrev:
1153
 
        print "  (bzr checkout, revision %d {%s})" % bzrrev
 
1620
    branch = _get_bzr_branch()
 
1621
    if branch:
 
1622
        rh = branch.revision_history()
 
1623
        revno = len(rh)
 
1624
        print "  bzr checkout, revision %d" % (revno,)
 
1625
        print "  nick: %s" % (branch.nick,)
 
1626
        if rh:
 
1627
            print "  revid: %s" % (rh[-1],)
1154
1628
    print bzrlib.__copyright__
1155
1629
    print "http://bazaar-ng.org/"
1156
1630
    print
1161
1635
 
1162
1636
class cmd_version(Command):
1163
1637
    """Show version of bzr."""
 
1638
    @display_command
1164
1639
    def run(self):
1165
1640
        show_version()
1166
1641
 
1167
1642
class cmd_rocks(Command):
1168
1643
    """Statement of optimism."""
1169
1644
    hidden = True
 
1645
    @display_command
1170
1646
    def run(self):
1171
1647
        print "it sure does!"
1172
1648
 
1179
1655
    takes_args = ['branch', 'other']
1180
1656
    hidden = True
1181
1657
    
 
1658
    @display_command
1182
1659
    def run(self, branch, other):
1183
1660
        from bzrlib.revision import common_ancestor, MultipleRevisionSources
1184
1661
        
1185
 
        branch1 = Branch.open_containing(branch)
1186
 
        branch2 = Branch.open_containing(other)
 
1662
        branch1 = Branch.open_containing(branch)[0]
 
1663
        branch2 = Branch.open_containing(other)[0]
1187
1664
 
1188
1665
        history_1 = branch1.revision_history()
1189
1666
        history_2 = branch2.revision_history()
1191
1668
        last1 = branch1.last_revision()
1192
1669
        last2 = branch2.last_revision()
1193
1670
 
1194
 
        source = MultipleRevisionSources(branch1, branch2)
 
1671
        source = MultipleRevisionSources(branch1.repository, 
 
1672
                                         branch2.repository)
1195
1673
        
1196
1674
        base_rev_id = common_ancestor(last1, last2, source)
1197
1675
 
1238
1716
    --force is given.
1239
1717
    """
1240
1718
    takes_args = ['branch?']
1241
 
    takes_options = ['revision', 'force', 'merge-type']
 
1719
    takes_options = ['revision', 'force', 'merge-type', 'reprocess',
 
1720
                     Option('show-base', help="Show base revision text in "
 
1721
                            "conflicts")]
1242
1722
 
1243
 
    def run(self, branch=None, revision=None, force=False, 
1244
 
            merge_type=None):
1245
 
        from bzrlib.merge import merge
1246
 
        from bzrlib.merge_core import ApplyMerge3
 
1723
    def run(self, branch=None, revision=None, force=False, merge_type=None,
 
1724
            show_base=False, reprocess=False):
1247
1725
        if merge_type is None:
1248
 
            merge_type = ApplyMerge3
 
1726
            merge_type = Merge3Merger
1249
1727
        if branch is None:
1250
 
            branch = Branch.open_containing('.').get_parent()
 
1728
            branch = WorkingTree.open_containing(u'.')[0].branch.get_parent()
1251
1729
            if branch is None:
1252
1730
                raise BzrCommandError("No merge location known or specified.")
1253
1731
            else:
1258
1736
        else:
1259
1737
            if len(revision) == 1:
1260
1738
                base = [None, None]
1261
 
                other = [branch, revision[0].in_history(branch).revno]
 
1739
                other_branch = Branch.open_containing(branch)[0]
 
1740
                revno = revision[0].in_history(other_branch).revno
 
1741
                other = [branch, revno]
1262
1742
            else:
1263
1743
                assert len(revision) == 2
1264
1744
                if None in revision:
1265
1745
                    raise BzrCommandError(
1266
1746
                        "Merge doesn't permit that revision specifier.")
1267
 
                b = Branch.open(branch)
 
1747
                b = Branch.open_containing(branch)[0]
1268
1748
 
1269
1749
                base = [branch, revision[0].in_history(b).revno]
1270
1750
                other = [branch, revision[1].in_history(b).revno]
1271
1751
 
1272
1752
        try:
1273
 
            merge(other, base, check_clean=(not force), merge_type=merge_type)
 
1753
            conflict_count = merge(other, base, check_clean=(not force),
 
1754
                                   merge_type=merge_type, reprocess=reprocess,
 
1755
                                   show_base=show_base, 
 
1756
                                   pb=bzrlib.ui.ui_factory.progress_bar())
 
1757
            if conflict_count != 0:
 
1758
                return 1
 
1759
            else:
 
1760
                return 0
1274
1761
        except bzrlib.errors.AmbiguousBase, e:
1275
1762
            m = ("sorry, bzr can't determine the right merge base yet\n"
1276
1763
                 "candidates are:\n  "
1281
1768
            log_error(m)
1282
1769
 
1283
1770
 
 
1771
class cmd_remerge(Command):
 
1772
    """Redo a merge.
 
1773
    """
 
1774
    takes_args = ['file*']
 
1775
    takes_options = ['merge-type', 'reprocess',
 
1776
                     Option('show-base', help="Show base revision text in "
 
1777
                            "conflicts")]
 
1778
 
 
1779
    def run(self, file_list=None, merge_type=None, show_base=False,
 
1780
            reprocess=False):
 
1781
        from bzrlib.merge import merge_inner, transform_tree
 
1782
        if merge_type is None:
 
1783
            merge_type = Merge3Merger
 
1784
        tree, file_list = tree_files(file_list)
 
1785
        tree.lock_write()
 
1786
        try:
 
1787
            pending_merges = tree.pending_merges() 
 
1788
            if len(pending_merges) != 1:
 
1789
                raise BzrCommandError("Sorry, remerge only works after normal"
 
1790
                                      + " merges.  Not cherrypicking or"
 
1791
                                      + "multi-merges.")
 
1792
            repository = tree.branch.repository
 
1793
            base_revision = common_ancestor(tree.branch.last_revision(), 
 
1794
                                            pending_merges[0], repository)
 
1795
            base_tree = repository.revision_tree(base_revision)
 
1796
            other_tree = repository.revision_tree(pending_merges[0])
 
1797
            interesting_ids = None
 
1798
            if file_list is not None:
 
1799
                interesting_ids = set()
 
1800
                for filename in file_list:
 
1801
                    file_id = tree.path2id(filename)
 
1802
                    interesting_ids.add(file_id)
 
1803
                    if tree.kind(file_id) != "directory":
 
1804
                        continue
 
1805
                    
 
1806
                    for name, ie in tree.inventory.iter_entries(file_id):
 
1807
                        interesting_ids.add(ie.file_id)
 
1808
            transform_tree(tree, tree.basis_tree(), interesting_ids)
 
1809
            if file_list is None:
 
1810
                restore_files = list(tree.iter_conflicts())
 
1811
            else:
 
1812
                restore_files = file_list
 
1813
            for filename in restore_files:
 
1814
                try:
 
1815
                    restore(tree.abspath(filename))
 
1816
                except NotConflicted:
 
1817
                    pass
 
1818
            conflicts =  merge_inner(tree.branch, other_tree, base_tree, 
 
1819
                                     interesting_ids = interesting_ids, 
 
1820
                                     other_rev_id=pending_merges[0], 
 
1821
                                     merge_type=merge_type, 
 
1822
                                     show_base=show_base,
 
1823
                                     reprocess=reprocess)
 
1824
        finally:
 
1825
            tree.unlock()
 
1826
        if conflicts > 0:
 
1827
            return 1
 
1828
        else:
 
1829
            return 0
 
1830
 
1284
1831
class cmd_revert(Command):
1285
1832
    """Reverse all changes since the last commit.
1286
1833
 
1293
1840
    aliases = ['merge-revert']
1294
1841
 
1295
1842
    def run(self, revision=None, no_backup=False, file_list=None):
1296
 
        from bzrlib.merge import merge
1297
1843
        from bzrlib.commands import parse_spec
1298
 
 
1299
1844
        if file_list is not None:
1300
1845
            if len(file_list) == 0:
1301
1846
                raise BzrCommandError("No files specified")
 
1847
        else:
 
1848
            file_list = []
 
1849
        
 
1850
        tree, file_list = tree_files(file_list)
1302
1851
        if revision is None:
1303
 
            revno = -1
 
1852
            # FIXME should be tree.last_revision
 
1853
            rev_id = tree.branch.last_revision()
1304
1854
        elif len(revision) != 1:
1305
1855
            raise BzrCommandError('bzr revert --revision takes exactly 1 argument')
1306
1856
        else:
1307
 
            b = Branch.open_containing('.')
1308
 
            revno = revision[0].in_history(b).revno
1309
 
        merge(('.', revno), parse_spec('.'),
1310
 
              check_clean=False,
1311
 
              ignore_zero=True,
1312
 
              backup_files=not no_backup,
1313
 
              file_list=file_list)
1314
 
        if not file_list:
1315
 
            Branch.open_containing('.').set_pending_merges([])
 
1857
            rev_id = revision[0].in_history(tree.branch).rev_id
 
1858
        tree.revert(file_list, tree.branch.repository.revision_tree(rev_id),
 
1859
                    not no_backup, bzrlib.ui.ui_factory.progress_bar())
1316
1860
 
1317
1861
 
1318
1862
class cmd_assert_fail(Command):
1326
1870
    """Show help on a command or other topic.
1327
1871
 
1328
1872
    For a list of all available commands, say 'bzr help commands'."""
1329
 
    takes_options = ['long']
 
1873
    takes_options = [Option('long', 'show help on all commands')]
1330
1874
    takes_args = ['topic?']
1331
1875
    aliases = ['?']
1332
1876
    
 
1877
    @display_command
1333
1878
    def run(self, topic=None, long=False):
1334
1879
        import help
1335
1880
        if topic is None and long:
1345
1890
    aliases = ['s-c']
1346
1891
    hidden = True
1347
1892
    
 
1893
    @display_command
1348
1894
    def run(self, context=None):
1349
1895
        import shellcomplete
1350
1896
        shellcomplete.shellcomplete(context)
1359
1905
    def run(self, from_branch, to_branch):
1360
1906
        from bzrlib.fetch import Fetcher
1361
1907
        from bzrlib.branch import Branch
1362
 
        from_b = Branch(from_branch)
1363
 
        to_b = Branch(to_branch)
 
1908
        from_b = Branch.open(from_branch)
 
1909
        to_b = Branch.open(to_branch)
1364
1910
        Fetcher(to_b, from_b)
1365
 
        
1366
1911
 
1367
1912
 
1368
1913
class cmd_missing(Command):
1369
 
    """What is missing in this branch relative to other branch.
1370
 
    """
1371
 
    # TODO: rewrite this in terms of ancestry so that it shows only
1372
 
    # unmerged things
1373
 
    
1374
 
    takes_args = ['remote?']
1375
 
    aliases = ['mis', 'miss']
1376
 
    # We don't have to add quiet to the list, because 
1377
 
    # unknown options are parsed as booleans
1378
 
    takes_options = ['verbose', 'quiet']
1379
 
 
1380
 
    def run(self, remote=None, verbose=False, quiet=False):
1381
 
        from bzrlib.errors import BzrCommandError
1382
 
        from bzrlib.missing import show_missing
1383
 
 
1384
 
        if verbose and quiet:
1385
 
            raise BzrCommandError('Cannot pass both quiet and verbose')
1386
 
 
1387
 
        b = Branch.open_containing('.')
1388
 
        parent = b.get_parent()
1389
 
        if remote is None:
1390
 
            if parent is None:
 
1914
    """Show unmerged/unpulled revisions between two branches.
 
1915
 
 
1916
    OTHER_BRANCH may be local or remote."""
 
1917
    takes_args = ['other_branch?']
 
1918
    takes_options = [Option('reverse', 'Reverse the order of revisions'),
 
1919
                     Option('mine-only', 
 
1920
                            'Display changes in the local branch only'),
 
1921
                     Option('theirs-only', 
 
1922
                            'Display changes in the remote branch only'), 
 
1923
                     'log-format',
 
1924
                     'line',
 
1925
                     'long', 
 
1926
                     'short',
 
1927
                     'show-ids',
 
1928
                     'verbose'
 
1929
                     ]
 
1930
 
 
1931
    def run(self, other_branch=None, reverse=False, mine_only=False,
 
1932
            theirs_only=False, log_format=None, long=False, short=False, line=False, 
 
1933
            show_ids=False, verbose=False):
 
1934
        from bzrlib.missing import find_unmerged, iter_log_data
 
1935
        from bzrlib.log import log_formatter
 
1936
        local_branch = bzrlib.branch.Branch.open_containing(u".")[0]
 
1937
        parent = local_branch.get_parent()
 
1938
        if other_branch is None:
 
1939
            other_branch = parent
 
1940
            if other_branch is None:
1391
1941
                raise BzrCommandError("No missing location known or specified.")
1392
 
            else:
1393
 
                if not quiet:
1394
 
                    print "Using last location: %s" % parent
1395
 
                remote = parent
1396
 
        elif parent is None:
1397
 
            # We only update parent if it did not exist, missing
1398
 
            # should not change the parent
1399
 
            b.set_parent(remote)
1400
 
        br_remote = Branch.open_containing(remote)
1401
 
        return show_missing(b, br_remote, verbose=verbose, quiet=quiet)
 
1942
            print "Using last location: " + local_branch.get_parent()
 
1943
        remote_branch = bzrlib.branch.Branch.open(other_branch)
 
1944
        local_extra, remote_extra = find_unmerged(local_branch, remote_branch)
 
1945
        if (log_format == None):
 
1946
            default = bzrlib.config.BranchConfig(local_branch).log_format()
 
1947
            log_format = get_log_format(long=long, short=short, line=line, default=default)
 
1948
        lf = log_formatter(log_format, sys.stdout,
 
1949
                           show_ids=show_ids,
 
1950
                           show_timezone='original')
 
1951
        if reverse is False:
 
1952
            local_extra.reverse()
 
1953
            remote_extra.reverse()
 
1954
        if local_extra and not theirs_only:
 
1955
            print "You have %d extra revision(s):" % len(local_extra)
 
1956
            for data in iter_log_data(local_extra, local_branch.repository,
 
1957
                                      verbose):
 
1958
                lf.show(*data)
 
1959
            printed_local = True
 
1960
        else:
 
1961
            printed_local = False
 
1962
        if remote_extra and not mine_only:
 
1963
            if printed_local is True:
 
1964
                print "\n\n"
 
1965
            print "You are missing %d revision(s):" % len(remote_extra)
 
1966
            for data in iter_log_data(remote_extra, remote_branch.repository, 
 
1967
                                      verbose):
 
1968
                lf.show(*data)
 
1969
        if not remote_extra and not local_extra:
 
1970
            status_code = 0
 
1971
            print "Branches are up to date."
 
1972
        else:
 
1973
            status_code = 1
 
1974
        if parent is None and other_branch is not None:
 
1975
            local_branch.set_parent(other_branch)
 
1976
        return status_code
1402
1977
 
1403
1978
 
1404
1979
class cmd_plugins(Command):
1405
1980
    """List plugins"""
1406
1981
    hidden = True
 
1982
    @display_command
1407
1983
    def run(self):
1408
1984
        import bzrlib.plugin
1409
1985
        from inspect import getdoc
1410
 
        for plugin in bzrlib.plugin.all_plugins:
 
1986
        for name, plugin in bzrlib.plugin.all_plugins().items():
1411
1987
            if hasattr(plugin, '__path__'):
1412
1988
                print plugin.__path__[0]
1413
1989
            elif hasattr(plugin, '__file__'):
1424
2000
    """Show testament (signing-form) of a revision."""
1425
2001
    takes_options = ['revision', 'long']
1426
2002
    takes_args = ['branch?']
1427
 
    def run(self, branch='.', revision=None, long=False):
 
2003
    @display_command
 
2004
    def run(self, branch=u'.', revision=None, long=False):
1428
2005
        from bzrlib.testament import Testament
1429
 
        b = Branch.open_containing(branch)
 
2006
        b = WorkingTree.open_containing(branch)[0].branch
1430
2007
        b.lock_read()
1431
2008
        try:
1432
2009
            if revision is None:
1433
2010
                rev_id = b.last_revision()
1434
2011
            else:
1435
2012
                rev_id = revision[0].in_history(b).rev_id
1436
 
            t = Testament.from_revision(b, rev_id)
 
2013
            t = Testament.from_revision(b.repository, rev_id)
1437
2014
            if long:
1438
2015
                sys.stdout.writelines(t.as_text_lines())
1439
2016
            else:
1445
2022
class cmd_annotate(Command):
1446
2023
    """Show the origin of each line in a file.
1447
2024
 
1448
 
    This prints out the given file with an annotation on the 
1449
 
    left side indicating which revision, author and date introduced the 
1450
 
    change.
 
2025
    This prints out the given file with an annotation on the left side
 
2026
    indicating which revision, author and date introduced the change.
 
2027
 
 
2028
    If the origin is the same for a run of consecutive lines, it is 
 
2029
    shown only at the top, unless the --all option is given.
1451
2030
    """
1452
2031
    # TODO: annotate directories; showing when each file was last changed
1453
2032
    # TODO: annotate a previous version of a file
 
2033
    # TODO: if the working copy is modified, show annotations on that 
 
2034
    #       with new uncommitted lines marked
1454
2035
    aliases = ['blame', 'praise']
1455
2036
    takes_args = ['filename']
 
2037
    takes_options = [Option('all', help='show annotations on all lines'),
 
2038
                     Option('long', help='show date in annotations'),
 
2039
                     ]
1456
2040
 
1457
 
    def run(self, filename):
 
2041
    @display_command
 
2042
    def run(self, filename, all=False, long=False):
1458
2043
        from bzrlib.annotate import annotate_file
1459
 
        b = Branch.open_containing(filename)
1460
 
        b.lock_read()
 
2044
        tree, relpath = WorkingTree.open_containing(filename)
 
2045
        branch = tree.branch
 
2046
        branch.lock_read()
1461
2047
        try:
1462
 
            rp = b.relpath(filename)
1463
 
            tree = b.revision_tree(b.last_revision())
1464
 
            file_id = tree.inventory.path2id(rp)
 
2048
            file_id = tree.inventory.path2id(relpath)
 
2049
            tree = branch.repository.revision_tree(branch.last_revision())
1465
2050
            file_version = tree.inventory[file_id].revision
1466
 
            annotate_file(b, file_version, file_id, sys.stdout)
 
2051
            annotate_file(branch, file_version, file_id, long, all, sys.stdout)
1467
2052
        finally:
1468
 
            b.unlock()
 
2053
            branch.unlock()
 
2054
 
 
2055
 
 
2056
class cmd_re_sign(Command):
 
2057
    """Create a digital signature for an existing revision."""
 
2058
    # TODO be able to replace existing ones.
 
2059
 
 
2060
    hidden = True # is this right ?
 
2061
    takes_args = ['revision_id*']
 
2062
    takes_options = ['revision']
 
2063
    
 
2064
    def run(self, revision_id_list=None, revision=None):
 
2065
        import bzrlib.config as config
 
2066
        import bzrlib.gpg as gpg
 
2067
        if revision_id_list is not None and revision is not None:
 
2068
            raise BzrCommandError('You can only supply one of revision_id or --revision')
 
2069
        if revision_id_list is None and revision is None:
 
2070
            raise BzrCommandError('You must supply either --revision or a revision_id')
 
2071
        b = WorkingTree.open_containing(u'.')[0].branch
 
2072
        gpg_strategy = gpg.GPGStrategy(config.BranchConfig(b))
 
2073
        if revision_id_list is not None:
 
2074
            for revision_id in revision_id_list:
 
2075
                b.repository.sign_revision(revision_id, gpg_strategy)
 
2076
        elif revision is not None:
 
2077
            if len(revision) == 1:
 
2078
                revno, rev_id = revision[0].in_history(b)
 
2079
                b.repository.sign_revision(rev_id, gpg_strategy)
 
2080
            elif len(revision) == 2:
 
2081
                # are they both on rh- if so we can walk between them
 
2082
                # might be nice to have a range helper for arbitrary
 
2083
                # revision paths. hmm.
 
2084
                from_revno, from_revid = revision[0].in_history(b)
 
2085
                to_revno, to_revid = revision[1].in_history(b)
 
2086
                if to_revid is None:
 
2087
                    to_revno = b.revno()
 
2088
                if from_revno is None or to_revno is None:
 
2089
                    raise BzrCommandError('Cannot sign a range of non-revision-history revisions')
 
2090
                for revno in range(from_revno, to_revno + 1):
 
2091
                    b.repository.sign_revision(b.get_rev_id(revno), 
 
2092
                                               gpg_strategy)
 
2093
            else:
 
2094
                raise BzrCommandError('Please supply either one revision, or a range.')
 
2095
 
 
2096
 
 
2097
class cmd_uncommit(bzrlib.commands.Command):
 
2098
    """Remove the last committed revision.
 
2099
 
 
2100
    By supplying the --all flag, it will not only remove the entry 
 
2101
    from revision_history, but also remove all of the entries in the
 
2102
    stores.
 
2103
 
 
2104
    --verbose will print out what is being removed.
 
2105
    --dry-run will go through all the motions, but not actually
 
2106
    remove anything.
 
2107
    
 
2108
    In the future, uncommit will create a changeset, which can then
 
2109
    be re-applied.
 
2110
 
 
2111
    TODO: jam 20060108 Add an option to allow uncommit to remove unreferenced
 
2112
              information in 'branch-as-repostory' branches.
 
2113
    TODO: jam 20060108 Add the ability for uncommit to remove unreferenced
 
2114
              information in shared branches as well.
 
2115
    """
 
2116
    takes_options = ['verbose', 'revision',
 
2117
                    Option('dry-run', help='Don\'t actually make changes'),
 
2118
                    Option('force', help='Say yes to all questions.')]
 
2119
    takes_args = ['location?']
 
2120
    aliases = []
 
2121
 
 
2122
    def run(self, location=None, 
 
2123
            dry_run=False, verbose=False,
 
2124
            revision=None, force=False):
 
2125
        from bzrlib.branch import Branch
 
2126
        from bzrlib.log import log_formatter
 
2127
        import sys
 
2128
        from bzrlib.uncommit import uncommit
 
2129
 
 
2130
        if location is None:
 
2131
            location = u'.'
 
2132
        b, relpath = Branch.open_containing(location)
 
2133
 
 
2134
        if revision is None:
 
2135
            revno = b.revno()
 
2136
            rev_id = b.last_revision()
 
2137
        else:
 
2138
            revno, rev_id = revision[0].in_history(b)
 
2139
        if rev_id is None:
 
2140
            print 'No revisions to uncommit.'
 
2141
 
 
2142
        for r in range(revno, b.revno()+1):
 
2143
            rev_id = b.get_rev_id(r)
 
2144
            lf = log_formatter('short', to_file=sys.stdout,show_timezone='original')
 
2145
            lf.show(r, b.repository.get_revision(rev_id), None)
 
2146
 
 
2147
        if dry_run:
 
2148
            print 'Dry-run, pretending to remove the above revisions.'
 
2149
            if not force:
 
2150
                val = raw_input('Press <enter> to continue')
 
2151
        else:
 
2152
            print 'The above revision(s) will be removed.'
 
2153
            if not force:
 
2154
                val = raw_input('Are you sure [y/N]? ')
 
2155
                if val.lower() not in ('y', 'yes'):
 
2156
                    print 'Canceled'
 
2157
                    return 0
 
2158
 
 
2159
        uncommit(b, dry_run=dry_run, verbose=verbose,
 
2160
                revno=revno)
 
2161
 
 
2162
 
 
2163
def merge(other_revision, base_revision,
 
2164
          check_clean=True, ignore_zero=False,
 
2165
          this_dir=None, backup_files=False, merge_type=Merge3Merger,
 
2166
          file_list=None, show_base=False, reprocess=False, 
 
2167
          pb=DummyProgress()):
 
2168
    """Merge changes into a tree.
 
2169
 
 
2170
    base_revision
 
2171
        list(path, revno) Base for three-way merge.  
 
2172
        If [None, None] then a base will be automatically determined.
 
2173
    other_revision
 
2174
        list(path, revno) Other revision for three-way merge.
 
2175
    this_dir
 
2176
        Directory to merge changes into; '.' by default.
 
2177
    check_clean
 
2178
        If true, this_dir must have no uncommitted changes before the
 
2179
        merge begins.
 
2180
    ignore_zero - If true, suppress the "zero conflicts" message when 
 
2181
        there are no conflicts; should be set when doing something we expect
 
2182
        to complete perfectly.
 
2183
    file_list - If supplied, merge only changes to selected files.
 
2184
 
 
2185
    All available ancestors of other_revision and base_revision are
 
2186
    automatically pulled into the branch.
 
2187
 
 
2188
    The revno may be -1 to indicate the last revision on the branch, which is
 
2189
    the typical case.
 
2190
 
 
2191
    This function is intended for use from the command line; programmatic
 
2192
    clients might prefer to call merge.merge_inner(), which has less magic 
 
2193
    behavior.
 
2194
    """
 
2195
    from bzrlib.merge import Merger
 
2196
    if this_dir is None:
 
2197
        this_dir = u'.'
 
2198
    this_tree = WorkingTree.open_containing(this_dir)[0]
 
2199
    if show_base and not merge_type is Merge3Merger:
 
2200
        raise BzrCommandError("Show-base is not supported for this merge"
 
2201
                              " type. %s" % merge_type)
 
2202
    if reprocess and not merge_type is Merge3Merger:
 
2203
        raise BzrCommandError("Reprocess is not supported for this merge"
 
2204
                              " type. %s" % merge_type)
 
2205
    if reprocess and show_base:
 
2206
        raise BzrCommandError("Cannot reprocess and show base.")
 
2207
    merger = Merger(this_tree.branch, this_tree=this_tree, pb=pb)
 
2208
    merger.check_basis(check_clean)
 
2209
    merger.set_other(other_revision)
 
2210
    merger.set_base(base_revision)
 
2211
    if merger.base_rev_id == merger.other_rev_id:
 
2212
        note('Nothing to do.')
 
2213
        return 0
 
2214
    merger.backup_files = backup_files
 
2215
    merger.merge_type = merge_type 
 
2216
    merger.set_interesting_files(file_list)
 
2217
    merger.show_base = show_base 
 
2218
    merger.reprocess = reprocess
 
2219
    conflicts = merger.do_merge()
 
2220
    merger.set_pending()
 
2221
    return conflicts
 
2222
 
1469
2223
 
1470
2224
# these get imported and then picked up by the scan for cmd_*
1471
2225
# TODO: Some more consistent way to split command definitions across files;
1472
2226
# we do need to load at least some information about them to know of 
1473
2227
# aliases.
1474
 
from bzrlib.conflicts import cmd_resolve, cmd_conflicts
 
2228
from bzrlib.conflicts import cmd_resolve, cmd_conflicts, restore
 
2229
from bzrlib.sign_my_commits import cmd_sign_my_commits