/brz/remove-bazaar

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

« back to all changes in this revision

Viewing changes to bzrlib/builtins.py

  • Committer: Aaron Bentley
  • Date: 2006-08-16 04:04:30 UTC
  • mfrom: (1922 +trunk)
  • mto: This revision was merged to the branch mainline in revision 1934.
  • Revision ID: aaron.bentley@utoronto.ca-20060816040430-3d869d673bdde644
Merge bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/builtins.py
 
1
# Copyright (C) 2004, 2005, 2006 by Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
"""builtin bzr commands"""
 
18
 
 
19
 
 
20
import codecs
 
21
import errno
 
22
import os
 
23
import os.path
 
24
import sys
 
25
 
 
26
import bzrlib
 
27
from bzrlib import (
 
28
    branch,
 
29
    bundle,
 
30
    bzrdir,
 
31
    config,
 
32
    errors,
 
33
    ignores,
 
34
    log,
 
35
    osutils,
 
36
    repository,
 
37
    transport,
 
38
    ui,
 
39
    urlutils,
 
40
    )
 
41
from bzrlib.branch import Branch, BranchReferenceFormat
 
42
from bzrlib.bundle import read_bundle_from_url
 
43
from bzrlib.bundle.apply_bundle import install_bundle, merge_bundle
 
44
from bzrlib.conflicts import ConflictList
 
45
from bzrlib.commands import Command, display_command
 
46
from bzrlib.errors import (BzrError, BzrCheckError, BzrCommandError, 
 
47
                           NotBranchError, DivergedBranches, NotConflicted,
 
48
                           NoSuchFile, NoWorkingTree, FileInWrongBranch,
 
49
                           NotVersionedError, NotABundle)
 
50
from bzrlib.merge import Merge3Merger
 
51
from bzrlib.option import Option, EnumOption
 
52
from bzrlib.progress import DummyProgress, ProgressPhase
 
53
from bzrlib.revision import common_ancestor
 
54
from bzrlib.revisionspec import RevisionSpec
 
55
from bzrlib.trace import mutter, note, log_error, warning, is_quiet, info
 
56
from bzrlib.transport.local import LocalTransport
 
57
from bzrlib.workingtree import WorkingTree
 
58
 
 
59
 
 
60
def tree_files(file_list, default_branch=u'.'):
 
61
    try:
 
62
        return internal_tree_files(file_list, default_branch)
 
63
    except FileInWrongBranch, e:
 
64
        raise BzrCommandError("%s is not in the same branch as %s" %
 
65
                             (e.path, file_list[0]))
 
66
 
 
67
 
 
68
# XXX: Bad function name; should possibly also be a class method of
 
69
# WorkingTree rather than a function.
 
70
def internal_tree_files(file_list, default_branch=u'.'):
 
71
    """Convert command-line paths to a WorkingTree and relative paths.
 
72
 
 
73
    This is typically used for command-line processors that take one or
 
74
    more filenames, and infer the workingtree that contains them.
 
75
 
 
76
    The filenames given are not required to exist.
 
77
 
 
78
    :param file_list: Filenames to convert.  
 
79
 
 
80
    :param default_branch: Fallback tree path to use if file_list is empty or None.
 
81
 
 
82
    :return: workingtree, [relative_paths]
 
83
    """
 
84
    if file_list is None or len(file_list) == 0:
 
85
        return WorkingTree.open_containing(default_branch)[0], file_list
 
86
    tree = WorkingTree.open_containing(file_list[0])[0]
 
87
    new_list = []
 
88
    for filename in file_list:
 
89
        try:
 
90
            new_list.append(tree.relpath(filename))
 
91
        except errors.PathNotChild:
 
92
            raise FileInWrongBranch(tree.branch, filename)
 
93
    return tree, new_list
 
94
 
 
95
 
 
96
def get_format_type(typestring):
 
97
    """Parse and return a format specifier."""
 
98
    if typestring == "weave":
 
99
        return bzrdir.BzrDirFormat6()
 
100
    if typestring == "default":
 
101
        return bzrdir.BzrDirMetaFormat1()
 
102
    if typestring == "metaweave":
 
103
        format = bzrdir.BzrDirMetaFormat1()
 
104
        format.repository_format = repository.RepositoryFormat7()
 
105
        return format
 
106
    if typestring == "knit":
 
107
        format = bzrdir.BzrDirMetaFormat1()
 
108
        format.repository_format = repository.RepositoryFormatKnit1()
 
109
        return format
 
110
    msg = "Unknown bzr format %s. Current formats are: default, knit,\n" \
 
111
          "metaweave and weave" % typestring
 
112
    raise BzrCommandError(msg)
 
113
 
 
114
 
 
115
# TODO: Make sure no commands unconditionally use the working directory as a
 
116
# branch.  If a filename argument is used, the first of them should be used to
 
117
# specify the branch.  (Perhaps this can be factored out into some kind of
 
118
# Argument class, representing a file in a branch, where the first occurrence
 
119
# opens the branch?)
 
120
 
 
121
class cmd_status(Command):
 
122
    """Display status summary.
 
123
 
 
124
    This reports on versioned and unknown files, reporting them
 
125
    grouped by state.  Possible states are:
 
126
 
 
127
    added
 
128
        Versioned in the working copy but not in the previous revision.
 
129
 
 
130
    removed
 
131
        Versioned in the previous revision but removed or deleted
 
132
        in the working copy.
 
133
 
 
134
    renamed
 
135
        Path of this file changed from the previous revision;
 
136
        the text may also have changed.  This includes files whose
 
137
        parent directory was renamed.
 
138
 
 
139
    modified
 
140
        Text has changed since the previous revision.
 
141
 
 
142
    unknown
 
143
        Not versioned and not matching an ignore pattern.
 
144
 
 
145
    To see ignored files use 'bzr ignored'.  For details in the
 
146
    changes to file texts, use 'bzr diff'.
 
147
 
 
148
    If no arguments are specified, the status of the entire working
 
149
    directory is shown.  Otherwise, only the status of the specified
 
150
    files or directories is reported.  If a directory is given, status
 
151
    is reported for everything inside that directory.
 
152
 
 
153
    If a revision argument is given, the status is calculated against
 
154
    that revision, or between two revisions if two are provided.
 
155
    """
 
156
    
 
157
    # TODO: --no-recurse, --recurse options
 
158
    
 
159
    takes_args = ['file*']
 
160
    takes_options = ['show-ids', 'revision']
 
161
    aliases = ['st', 'stat']
 
162
 
 
163
    encoding_type = 'replace'
 
164
    
 
165
    @display_command
 
166
    def run(self, show_ids=False, file_list=None, revision=None):
 
167
        from bzrlib.status import show_tree_status
 
168
 
 
169
        tree, file_list = tree_files(file_list)
 
170
            
 
171
        show_tree_status(tree, show_ids=show_ids,
 
172
                         specific_files=file_list, revision=revision,
 
173
                         to_file=self.outf)
 
174
 
 
175
 
 
176
class cmd_cat_revision(Command):
 
177
    """Write out metadata for a revision.
 
178
    
 
179
    The revision to print can either be specified by a specific
 
180
    revision identifier, or you can use --revision.
 
181
    """
 
182
 
 
183
    hidden = True
 
184
    takes_args = ['revision_id?']
 
185
    takes_options = ['revision']
 
186
    # cat-revision is more for frontends so should be exact
 
187
    encoding = 'strict'
 
188
    
 
189
    @display_command
 
190
    def run(self, revision_id=None, revision=None):
 
191
 
 
192
        if revision_id is not None and revision is not None:
 
193
            raise BzrCommandError('You can only supply one of revision_id or --revision')
 
194
        if revision_id is None and revision is None:
 
195
            raise BzrCommandError('You must supply either --revision or a revision_id')
 
196
        b = WorkingTree.open_containing(u'.')[0].branch
 
197
 
 
198
        # TODO: jam 20060112 should cat-revision always output utf-8?
 
199
        if revision_id is not None:
 
200
            self.outf.write(b.repository.get_revision_xml(revision_id).decode('utf-8'))
 
201
        elif revision is not None:
 
202
            for rev in revision:
 
203
                if rev is None:
 
204
                    raise BzrCommandError('You cannot specify a NULL revision.')
 
205
                revno, rev_id = rev.in_history(b)
 
206
                self.outf.write(b.repository.get_revision_xml(rev_id).decode('utf-8'))
 
207
    
 
208
 
 
209
class cmd_revno(Command):
 
210
    """Show current revision number.
 
211
 
 
212
    This is equal to the number of revisions on this branch.
 
213
    """
 
214
 
 
215
    takes_args = ['location?']
 
216
 
 
217
    @display_command
 
218
    def run(self, location=u'.'):
 
219
        self.outf.write(str(Branch.open_containing(location)[0].revno()))
 
220
        self.outf.write('\n')
 
221
 
 
222
 
 
223
class cmd_revision_info(Command):
 
224
    """Show revision number and revision id for a given revision identifier.
 
225
    """
 
226
    hidden = True
 
227
    takes_args = ['revision_info*']
 
228
    takes_options = ['revision']
 
229
 
 
230
    @display_command
 
231
    def run(self, revision=None, revision_info_list=[]):
 
232
 
 
233
        revs = []
 
234
        if revision is not None:
 
235
            revs.extend(revision)
 
236
        if revision_info_list is not None:
 
237
            for rev in revision_info_list:
 
238
                revs.append(RevisionSpec(rev))
 
239
        if len(revs) == 0:
 
240
            raise BzrCommandError('You must supply a revision identifier')
 
241
 
 
242
        b = WorkingTree.open_containing(u'.')[0].branch
 
243
 
 
244
        for rev in revs:
 
245
            revinfo = rev.in_history(b)
 
246
            if revinfo.revno is None:
 
247
                print '     %s' % revinfo.rev_id
 
248
            else:
 
249
                print '%4d %s' % (revinfo.revno, revinfo.rev_id)
 
250
 
 
251
    
 
252
class cmd_add(Command):
 
253
    """Add specified files or directories.
 
254
 
 
255
    In non-recursive mode, all the named items are added, regardless
 
256
    of whether they were previously ignored.  A warning is given if
 
257
    any of the named files are already versioned.
 
258
 
 
259
    In recursive mode (the default), files are treated the same way
 
260
    but the behaviour for directories is different.  Directories that
 
261
    are already versioned do not give a warning.  All directories,
 
262
    whether already versioned or not, are searched for files or
 
263
    subdirectories that are neither versioned or ignored, and these
 
264
    are added.  This search proceeds recursively into versioned
 
265
    directories.  If no names are given '.' is assumed.
 
266
 
 
267
    Therefore simply saying 'bzr add' will version all files that
 
268
    are currently unknown.
 
269
 
 
270
    Adding a file whose parent directory is not versioned will
 
271
    implicitly add the parent, and so on up to the root. This means
 
272
    you should never need to explicitly add a directory, they'll just
 
273
    get added when you add a file in the directory.
 
274
 
 
275
    --dry-run will show which files would be added, but not actually 
 
276
    add them.
 
277
    """
 
278
    takes_args = ['file*']
 
279
    takes_options = ['no-recurse', 'dry-run', 'verbose']
 
280
    encoding_type = 'replace'
 
281
 
 
282
    def run(self, file_list, no_recurse=False, dry_run=False, verbose=False):
 
283
        import bzrlib.add
 
284
 
 
285
        action = bzrlib.add.AddAction(to_file=self.outf,
 
286
            should_print=(not is_quiet()))
 
287
 
 
288
        added, ignored = bzrlib.add.smart_add(file_list, not no_recurse, 
 
289
                                              action=action, save=not dry_run)
 
290
        if len(ignored) > 0:
 
291
            if verbose:
 
292
                for glob in sorted(ignored.keys()):
 
293
                    for path in ignored[glob]:
 
294
                        self.outf.write("ignored %s matching \"%s\"\n" 
 
295
                                        % (path, glob))
 
296
            else:
 
297
                match_len = 0
 
298
                for glob, paths in ignored.items():
 
299
                    match_len += len(paths)
 
300
                self.outf.write("ignored %d file(s).\n" % match_len)
 
301
            self.outf.write("If you wish to add some of these files,"
 
302
                            " please add them by name.\n")
 
303
 
 
304
 
 
305
class cmd_mkdir(Command):
 
306
    """Create a new versioned directory.
 
307
 
 
308
    This is equivalent to creating the directory and then adding it.
 
309
    """
 
310
 
 
311
    takes_args = ['dir+']
 
312
    encoding_type = 'replace'
 
313
 
 
314
    def run(self, dir_list):
 
315
        for d in dir_list:
 
316
            os.mkdir(d)
 
317
            wt, dd = WorkingTree.open_containing(d)
 
318
            wt.add([dd])
 
319
            self.outf.write('added %s\n' % d)
 
320
 
 
321
 
 
322
class cmd_relpath(Command):
 
323
    """Show path of a file relative to root"""
 
324
 
 
325
    takes_args = ['filename']
 
326
    hidden = True
 
327
    
 
328
    @display_command
 
329
    def run(self, filename):
 
330
        # TODO: jam 20050106 Can relpath return a munged path if
 
331
        #       sys.stdout encoding cannot represent it?
 
332
        tree, relpath = WorkingTree.open_containing(filename)
 
333
        self.outf.write(relpath)
 
334
        self.outf.write('\n')
 
335
 
 
336
 
 
337
class cmd_inventory(Command):
 
338
    """Show inventory of the current working copy or a revision.
 
339
 
 
340
    It is possible to limit the output to a particular entry
 
341
    type using the --kind option.  For example; --kind file.
 
342
    """
 
343
 
 
344
    takes_options = ['revision', 'show-ids', 'kind']
 
345
    
 
346
    @display_command
 
347
    def run(self, revision=None, show_ids=False, kind=None):
 
348
        if kind and kind not in ['file', 'directory', 'symlink']:
 
349
            raise BzrCommandError('invalid kind specified')
 
350
        tree = WorkingTree.open_containing(u'.')[0]
 
351
        if revision is None:
 
352
            inv = tree.read_working_inventory()
 
353
        else:
 
354
            if len(revision) > 1:
 
355
                raise BzrCommandError('bzr inventory --revision takes'
 
356
                    ' exactly one revision identifier')
 
357
            inv = tree.branch.repository.get_revision_inventory(
 
358
                revision[0].in_history(tree.branch).rev_id)
 
359
 
 
360
        for path, entry in inv.entries():
 
361
            if kind and kind != entry.kind:
 
362
                continue
 
363
            if show_ids:
 
364
                self.outf.write('%-50s %s\n' % (path, entry.file_id))
 
365
            else:
 
366
                self.outf.write(path)
 
367
                self.outf.write('\n')
 
368
 
 
369
 
 
370
class cmd_mv(Command):
 
371
    """Move or rename a file.
 
372
 
 
373
    usage:
 
374
        bzr mv OLDNAME NEWNAME
 
375
        bzr mv SOURCE... DESTINATION
 
376
 
 
377
    If the last argument is a versioned directory, all the other names
 
378
    are moved into it.  Otherwise, there must be exactly two arguments
 
379
    and the file is changed to a new name, which must not already exist.
 
380
 
 
381
    Files cannot be moved between branches.
 
382
    """
 
383
 
 
384
    takes_args = ['names*']
 
385
    aliases = ['move', 'rename']
 
386
    encoding_type = 'replace'
 
387
 
 
388
    def run(self, names_list):
 
389
        if names_list is None:
 
390
            names_list = []
 
391
 
 
392
        if len(names_list) < 2:
 
393
            raise BzrCommandError("missing file argument")
 
394
        tree, rel_names = tree_files(names_list)
 
395
        
 
396
        if os.path.isdir(names_list[-1]):
 
397
            # move into existing directory
 
398
            for pair in tree.move(rel_names[:-1], rel_names[-1]):
 
399
                self.outf.write("%s => %s\n" % pair)
 
400
        else:
 
401
            if len(names_list) != 2:
 
402
                raise BzrCommandError('to mv multiple files the destination '
 
403
                                      'must be a versioned directory')
 
404
            tree.rename_one(rel_names[0], rel_names[1])
 
405
            self.outf.write("%s => %s\n" % (rel_names[0], rel_names[1]))
 
406
            
 
407
    
 
408
class cmd_pull(Command):
 
409
    """Turn this branch into a mirror of another branch.
 
410
 
 
411
    This command only works on branches that have not diverged.  Branches are
 
412
    considered diverged if the destination branch's most recent commit is one
 
413
    that has not been merged (directly or indirectly) into the parent.
 
414
 
 
415
    If branches have diverged, you can use 'bzr merge' to integrate the changes
 
416
    from one into the other.  Once one branch has merged, the other should
 
417
    be able to pull it again.
 
418
 
 
419
    If you want to forget your local changes and just update your branch to
 
420
    match the remote one, use pull --overwrite.
 
421
 
 
422
    If there is no default location set, the first pull will set it.  After
 
423
    that, you can omit the location to use the default.  To change the
 
424
    default, use --remember. The value will only be saved if the remote
 
425
    location can be accessed.
 
426
    """
 
427
 
 
428
    takes_options = ['remember', 'overwrite', 'revision', 'verbose']
 
429
    takes_args = ['location?']
 
430
    encoding_type = 'replace'
 
431
 
 
432
    def run(self, location=None, remember=False, overwrite=False, revision=None, verbose=False):
 
433
        # FIXME: too much stuff is in the command class
 
434
        try:
 
435
            tree_to = WorkingTree.open_containing(u'.')[0]
 
436
            branch_to = tree_to.branch
 
437
        except NoWorkingTree:
 
438
            tree_to = None
 
439
            branch_to = Branch.open_containing(u'.')[0]
 
440
 
 
441
        reader = None
 
442
        if location is not None:
 
443
            try:
 
444
                reader = bundle.read_bundle_from_url(location)
 
445
            except NotABundle:
 
446
                pass # Continue on considering this url a Branch
 
447
 
 
448
        stored_loc = branch_to.get_parent()
 
449
        if location is None:
 
450
            if stored_loc is None:
 
451
                raise BzrCommandError("No pull location known or specified.")
 
452
            else:
 
453
                display_url = urlutils.unescape_for_display(stored_loc,
 
454
                        self.outf.encoding)
 
455
                self.outf.write("Using saved location: %s\n" % display_url)
 
456
                location = stored_loc
 
457
 
 
458
 
 
459
        if reader is not None:
 
460
            install_bundle(branch_to.repository, reader)
 
461
            branch_from = branch_to
 
462
        else:
 
463
            branch_from = Branch.open(location)
 
464
 
 
465
            if branch_to.get_parent() is None or remember:
 
466
                branch_to.set_parent(branch_from.base)
 
467
 
 
468
        rev_id = None
 
469
        if revision is None:
 
470
            if reader is not None:
 
471
                rev_id = reader.target
 
472
        elif len(revision) == 1:
 
473
            rev_id = revision[0].in_history(branch_from).rev_id
 
474
        else:
 
475
            raise BzrCommandError('bzr pull --revision takes one value.')
 
476
 
 
477
        old_rh = branch_to.revision_history()
 
478
        if tree_to is not None:
 
479
            count = tree_to.pull(branch_from, overwrite, rev_id)
 
480
        else:
 
481
            count = branch_to.pull(branch_from, overwrite, rev_id)
 
482
        note('%d revision(s) pulled.' % (count,))
 
483
 
 
484
        if verbose:
 
485
            new_rh = branch_to.revision_history()
 
486
            if old_rh != new_rh:
 
487
                # Something changed
 
488
                from bzrlib.log import show_changed_revisions
 
489
                show_changed_revisions(branch_to, old_rh, new_rh,
 
490
                                       to_file=self.outf)
 
491
 
 
492
 
 
493
class cmd_push(Command):
 
494
    """Update a mirror of this branch.
 
495
    
 
496
    The target branch will not have its working tree populated because this
 
497
    is both expensive, and is not supported on remote file systems.
 
498
    
 
499
    Some smart servers or protocols *may* put the working tree in place in
 
500
    the future.
 
501
 
 
502
    This command only works on branches that have not diverged.  Branches are
 
503
    considered diverged if the destination branch's most recent commit is one
 
504
    that has not been merged (directly or indirectly) by the source branch.
 
505
 
 
506
    If branches have diverged, you can use 'bzr push --overwrite' to replace
 
507
    the other branch completely, discarding its unmerged changes.
 
508
    
 
509
    If you want to ensure you have the different changes in the other branch,
 
510
    do a merge (see bzr help merge) from the other branch, and commit that.
 
511
    After that you will be able to do a push without '--overwrite'.
 
512
 
 
513
    If there is no default push location set, the first push will set it.
 
514
    After that, you can omit the location to use the default.  To change the
 
515
    default, use --remember. The value will only be saved if the remote
 
516
    location can be accessed.
 
517
    """
 
518
 
 
519
    takes_options = ['remember', 'overwrite', 'verbose',
 
520
                     Option('create-prefix', 
 
521
                            help='Create the path leading up to the branch '
 
522
                                 'if it does not already exist')]
 
523
    takes_args = ['location?']
 
524
    encoding_type = 'replace'
 
525
 
 
526
    def run(self, location=None, remember=False, overwrite=False,
 
527
            create_prefix=False, verbose=False):
 
528
        # FIXME: Way too big!  Put this into a function called from the
 
529
        # command.
 
530
        
 
531
        br_from = Branch.open_containing('.')[0]
 
532
        stored_loc = br_from.get_push_location()
 
533
        if location is None:
 
534
            if stored_loc is None:
 
535
                raise BzrCommandError("No push location known or specified.")
 
536
            else:
 
537
                display_url = urlutils.unescape_for_display(stored_loc,
 
538
                        self.outf.encoding)
 
539
                self.outf.write("Using saved location: %s\n" % display_url)
 
540
                location = stored_loc
 
541
 
 
542
        to_transport = transport.get_transport(location)
 
543
        location_url = to_transport.base
 
544
 
 
545
        old_rh = []
 
546
        try:
 
547
            dir_to = bzrdir.BzrDir.open(location_url)
 
548
            br_to = dir_to.open_branch()
 
549
        except NotBranchError:
 
550
            # create a branch.
 
551
            to_transport = to_transport.clone('..')
 
552
            if not create_prefix:
 
553
                try:
 
554
                    relurl = to_transport.relpath(location_url)
 
555
                    mutter('creating directory %s => %s', location_url, relurl)
 
556
                    to_transport.mkdir(relurl)
 
557
                except NoSuchFile:
 
558
                    raise BzrCommandError("Parent directory of %s "
 
559
                                          "does not exist." % location)
 
560
            else:
 
561
                current = to_transport.base
 
562
                needed = [(to_transport, to_transport.relpath(location_url))]
 
563
                while needed:
 
564
                    try:
 
565
                        to_transport, relpath = needed[-1]
 
566
                        to_transport.mkdir(relpath)
 
567
                        needed.pop()
 
568
                    except NoSuchFile:
 
569
                        new_transport = to_transport.clone('..')
 
570
                        needed.append((new_transport,
 
571
                                       new_transport.relpath(to_transport.base)))
 
572
                        if new_transport.base == to_transport.base:
 
573
                            raise BzrCommandError("Could not create "
 
574
                                                  "path prefix.")
 
575
            dir_to = br_from.bzrdir.clone(location_url,
 
576
                revision_id=br_from.last_revision())
 
577
            br_to = dir_to.open_branch()
 
578
            count = len(br_to.revision_history())
 
579
            # We successfully created the target, remember it
 
580
            if br_from.get_push_location() is None or remember:
 
581
                br_from.set_push_location(br_to.base)
 
582
        else:
 
583
            # We were able to connect to the remote location, so remember it
 
584
            # we don't need to successfully push because of possible divergence.
 
585
            if br_from.get_push_location() is None or remember:
 
586
                br_from.set_push_location(br_to.base)
 
587
            old_rh = br_to.revision_history()
 
588
            try:
 
589
                try:
 
590
                    tree_to = dir_to.open_workingtree()
 
591
                except errors.NotLocalUrl:
 
592
                    warning('This transport does not update the working '
 
593
                            'tree of: %s' % (br_to.base,))
 
594
                    count = br_to.pull(br_from, overwrite)
 
595
                except NoWorkingTree:
 
596
                    count = br_to.pull(br_from, overwrite)
 
597
                else:
 
598
                    count = tree_to.pull(br_from, overwrite)
 
599
            except DivergedBranches:
 
600
                raise BzrCommandError("These branches have diverged."
 
601
                                      "  Try a merge then push with overwrite.")
 
602
        note('%d revision(s) pushed.' % (count,))
 
603
 
 
604
        if verbose:
 
605
            new_rh = br_to.revision_history()
 
606
            if old_rh != new_rh:
 
607
                # Something changed
 
608
                from bzrlib.log import show_changed_revisions
 
609
                show_changed_revisions(br_to, old_rh, new_rh,
 
610
                                       to_file=self.outf)
 
611
 
 
612
 
 
613
class cmd_branch(Command):
 
614
    """Create a new copy of a branch.
 
615
 
 
616
    If the TO_LOCATION is omitted, the last component of the FROM_LOCATION will
 
617
    be used.  In other words, "branch ../foo/bar" will attempt to create ./bar.
 
618
 
 
619
    To retrieve the branch as of a particular revision, supply the --revision
 
620
    parameter, as in "branch foo/bar -r 5".
 
621
 
 
622
    --basis is to speed up branching from remote branches.  When specified, it
 
623
    copies all the file-contents, inventory and revision data from the basis
 
624
    branch before copying anything from the remote branch.
 
625
    """
 
626
    takes_args = ['from_location', 'to_location?']
 
627
    takes_options = ['revision', 'basis']
 
628
    aliases = ['get', 'clone']
 
629
 
 
630
    def run(self, from_location, to_location=None, revision=None, basis=None):
 
631
        if revision is None:
 
632
            revision = [None]
 
633
        elif len(revision) > 1:
 
634
            raise BzrCommandError(
 
635
                'bzr branch --revision takes exactly 1 revision value')
 
636
        try:
 
637
            br_from = Branch.open(from_location)
 
638
        except OSError, e:
 
639
            if e.errno == errno.ENOENT:
 
640
                raise BzrCommandError('Source location "%s" does not'
 
641
                                      ' exist.' % to_location)
 
642
            else:
 
643
                raise
 
644
        br_from.lock_read()
 
645
        try:
 
646
            if basis is not None:
 
647
                basis_dir = bzrdir.BzrDir.open_containing(basis)[0]
 
648
            else:
 
649
                basis_dir = None
 
650
            if len(revision) == 1 and revision[0] is not None:
 
651
                revision_id = revision[0].in_history(br_from)[1]
 
652
            else:
 
653
                # FIXME - wt.last_revision, fallback to branch, fall back to
 
654
                # None or perhaps NULL_REVISION to mean copy nothing
 
655
                # RBC 20060209
 
656
                revision_id = br_from.last_revision()
 
657
            if to_location is None:
 
658
                to_location = os.path.basename(from_location.rstrip("/\\"))
 
659
                name = None
 
660
            else:
 
661
                name = os.path.basename(to_location) + '\n'
 
662
 
 
663
            to_transport = transport.get_transport(to_location)
 
664
            try:
 
665
                to_transport.mkdir('.')
 
666
            except errors.FileExists:
 
667
                raise BzrCommandError('Target directory "%s" already'
 
668
                                      ' exists.' % to_location)
 
669
            except errors.NoSuchFile:
 
670
                raise BzrCommandError('Parent of "%s" does not exist.' %
 
671
                                      to_location)
 
672
            try:
 
673
                # preserve whatever source format we have.
 
674
                dir = br_from.bzrdir.sprout(to_transport.base,
 
675
                        revision_id, basis_dir)
 
676
                branch = dir.open_branch()
 
677
            except errors.NoSuchRevision:
 
678
                to_transport.delete_tree('.')
 
679
                msg = "The branch %s has no revision %s." % (from_location, revision[0])
 
680
                raise BzrCommandError(msg)
 
681
            except errors.UnlistableBranch:
 
682
                osutils.rmtree(to_location)
 
683
                msg = "The branch %s cannot be used as a --basis" % (basis,)
 
684
                raise BzrCommandError(msg)
 
685
            if name:
 
686
                branch.control_files.put_utf8('branch-name', name)
 
687
            note('Branched %d revision(s).' % branch.revno())
 
688
        finally:
 
689
            br_from.unlock()
 
690
 
 
691
 
 
692
class cmd_checkout(Command):
 
693
    """Create a new checkout of an existing branch.
 
694
 
 
695
    If BRANCH_LOCATION is omitted, checkout will reconstitute a working tree for
 
696
    the branch found in '.'. This is useful if you have removed the working tree
 
697
    or if it was never created - i.e. if you pushed the branch to its current
 
698
    location using SFTP.
 
699
    
 
700
    If the TO_LOCATION is omitted, the last component of the BRANCH_LOCATION will
 
701
    be used.  In other words, "checkout ../foo/bar" will attempt to create ./bar.
 
702
 
 
703
    To retrieve the branch as of a particular revision, supply the --revision
 
704
    parameter, as in "checkout foo/bar -r 5". Note that this will be immediately
 
705
    out of date [so you cannot commit] but it may be useful (i.e. to examine old
 
706
    code.)
 
707
 
 
708
    --basis is to speed up checking out from remote branches.  When specified, it
 
709
    uses the inventory and file contents from the basis branch in preference to the
 
710
    branch being checked out.
 
711
    """
 
712
    takes_args = ['branch_location?', 'to_location?']
 
713
    takes_options = ['revision', # , 'basis']
 
714
                     Option('lightweight',
 
715
                            help="perform a lightweight checkout. Lightweight "
 
716
                                 "checkouts depend on access to the branch for "
 
717
                                 "every operation. Normal checkouts can perform "
 
718
                                 "common operations like diff and status without "
 
719
                                 "such access, and also support local commits."
 
720
                            ),
 
721
                     ]
 
722
    aliases = ['co']
 
723
 
 
724
    def run(self, branch_location=None, to_location=None, revision=None, basis=None,
 
725
            lightweight=False):
 
726
        if revision is None:
 
727
            revision = [None]
 
728
        elif len(revision) > 1:
 
729
            raise BzrCommandError(
 
730
                'bzr checkout --revision takes exactly 1 revision value')
 
731
        if branch_location is None:
 
732
            branch_location = osutils.getcwd()
 
733
            to_location = branch_location
 
734
        source = Branch.open(branch_location)
 
735
        if len(revision) == 1 and revision[0] is not None:
 
736
            revision_id = revision[0].in_history(source)[1]
 
737
        else:
 
738
            revision_id = None
 
739
        if to_location is None:
 
740
            to_location = os.path.basename(branch_location.rstrip("/\\"))
 
741
        # if the source and to_location are the same, 
 
742
        # and there is no working tree,
 
743
        # then reconstitute a branch
 
744
        if (osutils.abspath(to_location) == 
 
745
            osutils.abspath(branch_location)):
 
746
            try:
 
747
                source.bzrdir.open_workingtree()
 
748
            except errors.NoWorkingTree:
 
749
                source.bzrdir.create_workingtree()
 
750
                return
 
751
        try:
 
752
            os.mkdir(to_location)
 
753
        except OSError, e:
 
754
            if e.errno == errno.EEXIST:
 
755
                raise BzrCommandError('Target directory "%s" already'
 
756
                                      ' exists.' % to_location)
 
757
            if e.errno == errno.ENOENT:
 
758
                raise BzrCommandError('Parent of "%s" does not exist.' %
 
759
                                      to_location)
 
760
            else:
 
761
                raise
 
762
        old_format = bzrdir.BzrDirFormat.get_default_format()
 
763
        bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirMetaFormat1())
 
764
        try:
 
765
            source.create_checkout(to_location, revision_id, lightweight)
 
766
        finally:
 
767
            bzrdir.BzrDirFormat.set_default_format(old_format)
 
768
 
 
769
 
 
770
class cmd_renames(Command):
 
771
    """Show list of renamed files.
 
772
    """
 
773
    # TODO: Option to show renames between two historical versions.
 
774
 
 
775
    # TODO: Only show renames under dir, rather than in the whole branch.
 
776
    takes_args = ['dir?']
 
777
 
 
778
    @display_command
 
779
    def run(self, dir=u'.'):
 
780
        from bzrlib.tree import find_renames
 
781
        tree = WorkingTree.open_containing(dir)[0]
 
782
        old_inv = tree.basis_tree().inventory
 
783
        new_inv = tree.read_working_inventory()
 
784
        renames = list(find_renames(old_inv, new_inv))
 
785
        renames.sort()
 
786
        for old_name, new_name in renames:
 
787
            self.outf.write("%s => %s\n" % (old_name, new_name))
 
788
 
 
789
 
 
790
class cmd_update(Command):
 
791
    """Update a tree to have the latest code committed to its branch.
 
792
    
 
793
    This will perform a merge into the working tree, and may generate
 
794
    conflicts. If you have any local changes, you will still 
 
795
    need to commit them after the update for the update to be complete.
 
796
    
 
797
    If you want to discard your local changes, you can just do a 
 
798
    'bzr revert' instead of 'bzr commit' after the update.
 
799
    """
 
800
    takes_args = ['dir?']
 
801
    aliases = ['up']
 
802
 
 
803
    def run(self, dir='.'):
 
804
        tree = WorkingTree.open_containing(dir)[0]
 
805
        tree.lock_write()
 
806
        existing_pending_merges = tree.pending_merges()
 
807
        try:
 
808
            last_rev = tree.last_revision() 
 
809
            if last_rev == tree.branch.last_revision():
 
810
                # may be up to date, check master too.
 
811
                master = tree.branch.get_master_branch()
 
812
                if master is None or last_rev == master.last_revision():
 
813
                    revno = tree.branch.revision_id_to_revno(last_rev)
 
814
                    note("Tree is up to date at revision %d." % (revno,))
 
815
                    return 0
 
816
            conflicts = tree.update()
 
817
            revno = tree.branch.revision_id_to_revno(tree.last_revision())
 
818
            note('Updated to revision %d.' % (revno,))
 
819
            if tree.pending_merges() != existing_pending_merges:
 
820
                note('Your local commits will now show as pending merges with '
 
821
                     "'bzr status', and can be committed with 'bzr commit'.")
 
822
            if conflicts != 0:
 
823
                return 1
 
824
            else:
 
825
                return 0
 
826
        finally:
 
827
            tree.unlock()
 
828
 
 
829
 
 
830
class cmd_info(Command):
 
831
    """Show information about a working tree, branch or repository.
 
832
 
 
833
    This command will show all known locations and formats associated to the
 
834
    tree, branch or repository.  Statistical information is included with
 
835
    each report.
 
836
 
 
837
    Branches and working trees will also report any missing revisions.
 
838
    """
 
839
    takes_args = ['location?']
 
840
    takes_options = ['verbose']
 
841
 
 
842
    @display_command
 
843
    def run(self, location=None, verbose=False):
 
844
        from bzrlib.info import show_bzrdir_info
 
845
        show_bzrdir_info(bzrdir.BzrDir.open_containing(location)[0],
 
846
                         verbose=verbose)
 
847
 
 
848
 
 
849
class cmd_remove(Command):
 
850
    """Make a file unversioned.
 
851
 
 
852
    This makes bzr stop tracking changes to a versioned file.  It does
 
853
    not delete the working copy.
 
854
 
 
855
    You can specify one or more files, and/or --new.  If you specify --new,
 
856
    only 'added' files will be removed.  If you specify both, then new files
 
857
    in the specified directories will be removed.  If the directories are
 
858
    also new, they will also be removed.
 
859
    """
 
860
    takes_args = ['file*']
 
861
    takes_options = ['verbose', Option('new', help='remove newly-added files')]
 
862
    aliases = ['rm']
 
863
    encoding_type = 'replace'
 
864
    
 
865
    def run(self, file_list, verbose=False, new=False):
 
866
        tree, file_list = tree_files(file_list)
 
867
        if new is False:
 
868
            if file_list is None:
 
869
                raise BzrCommandError('Specify one or more files to remove, or'
 
870
                                      ' use --new.')
 
871
        else:
 
872
            added = tree.changes_from(tree.basis_tree(),
 
873
                specific_files=file_list).added
 
874
            file_list = sorted([f[0] for f in added], reverse=True)
 
875
            if len(file_list) == 0:
 
876
                raise BzrCommandError('No matching files.')
 
877
        tree.remove(file_list, verbose=verbose, to_file=self.outf)
 
878
 
 
879
 
 
880
class cmd_file_id(Command):
 
881
    """Print file_id of a particular file or directory.
 
882
 
 
883
    The file_id is assigned when the file is first added and remains the
 
884
    same through all revisions where the file exists, even when it is
 
885
    moved or renamed.
 
886
    """
 
887
 
 
888
    hidden = True
 
889
    takes_args = ['filename']
 
890
 
 
891
    @display_command
 
892
    def run(self, filename):
 
893
        tree, relpath = WorkingTree.open_containing(filename)
 
894
        i = tree.inventory.path2id(relpath)
 
895
        if i == None:
 
896
            raise BzrError("%r is not a versioned file" % filename)
 
897
        else:
 
898
            self.outf.write(i + '\n')
 
899
 
 
900
 
 
901
class cmd_file_path(Command):
 
902
    """Print path of file_ids to a file or directory.
 
903
 
 
904
    This prints one line for each directory down to the target,
 
905
    starting at the branch root.
 
906
    """
 
907
 
 
908
    hidden = True
 
909
    takes_args = ['filename']
 
910
 
 
911
    @display_command
 
912
    def run(self, filename):
 
913
        tree, relpath = WorkingTree.open_containing(filename)
 
914
        inv = tree.inventory
 
915
        fid = inv.path2id(relpath)
 
916
        if fid == None:
 
917
            raise BzrError("%r is not a versioned file" % filename)
 
918
        for fip in inv.get_idpath(fid):
 
919
            self.outf.write(fip + '\n')
 
920
 
 
921
 
 
922
class cmd_reconcile(Command):
 
923
    """Reconcile bzr metadata in a branch.
 
924
 
 
925
    This can correct data mismatches that may have been caused by
 
926
    previous ghost operations or bzr upgrades. You should only
 
927
    need to run this command if 'bzr check' or a bzr developer 
 
928
    advises you to run it.
 
929
 
 
930
    If a second branch is provided, cross-branch reconciliation is
 
931
    also attempted, which will check that data like the tree root
 
932
    id which was not present in very early bzr versions is represented
 
933
    correctly in both branches.
 
934
 
 
935
    At the same time it is run it may recompress data resulting in 
 
936
    a potential saving in disk space or performance gain.
 
937
 
 
938
    The branch *MUST* be on a listable system such as local disk or sftp.
 
939
    """
 
940
    takes_args = ['branch?']
 
941
 
 
942
    def run(self, branch="."):
 
943
        from bzrlib.reconcile import reconcile
 
944
        dir = bzrdir.BzrDir.open(branch)
 
945
        reconcile(dir)
 
946
 
 
947
 
 
948
class cmd_revision_history(Command):
 
949
    """Display the list of revision ids on a branch."""
 
950
    takes_args = ['location?']
 
951
 
 
952
    hidden = True
 
953
 
 
954
    @display_command
 
955
    def run(self, location="."):
 
956
        branch = Branch.open_containing(location)[0]
 
957
        for revid in branch.revision_history():
 
958
            self.outf.write(revid)
 
959
            self.outf.write('\n')
 
960
 
 
961
 
 
962
class cmd_ancestry(Command):
 
963
    """List all revisions merged into this branch."""
 
964
    takes_args = ['location?']
 
965
 
 
966
    hidden = True
 
967
 
 
968
    @display_command
 
969
    def run(self, location="."):
 
970
        try:
 
971
            wt = WorkingTree.open_containing(location)[0]
 
972
        except errors.NoWorkingTree:
 
973
            b = Branch.open(location)
 
974
            last_revision = b.last_revision()
 
975
        else:
 
976
            b = wt.branch
 
977
            last_revision = wt.last_revision()
 
978
 
 
979
        revision_ids = b.repository.get_ancestry(last_revision)
 
980
        assert revision_ids[0] == None
 
981
        revision_ids.pop(0)
 
982
        for revision_id in revision_ids:
 
983
            self.outf.write(revision_id + '\n')
 
984
 
 
985
branch_format_option = EnumOption('Branch Format', get_format_type, 
 
986
                                  [('default', 
 
987
                                    'The current best format (knit)'),
 
988
                                   ('knit', '0.8+ append-only format'),
 
989
                                   ('metaweave', '0.8 transitional format'),
 
990
                                   ('weave', '0.1+ format')
 
991
                                  ])
 
992
 
 
993
repo_format_option = EnumOption('Repository Format', get_format_type, 
 
994
                                [('default', 
 
995
                                  'The current best format (knit)'),
 
996
                                 ('knit', '0.8+ append-only format'),
 
997
                                 ('metaweave', '0.8 transitional format'),
 
998
                                ])
 
999
 
 
1000
class cmd_init(Command):
 
1001
    """Make a directory into a versioned branch.
 
1002
 
 
1003
    Use this to create an empty branch, or before importing an
 
1004
    existing project.
 
1005
 
 
1006
    If there is a repository in a parent directory of the location, then 
 
1007
    the history of the branch will be stored in the repository.  Otherwise
 
1008
    init creates a standalone branch which carries its own history in 
 
1009
    .bzr.
 
1010
 
 
1011
    If there is already a branch at the location but it has no working tree,
 
1012
    the tree can be populated with 'bzr checkout'.
 
1013
 
 
1014
    Recipe for importing a tree of files:
 
1015
        cd ~/project
 
1016
        bzr init
 
1017
        bzr add .
 
1018
        bzr status
 
1019
        bzr commit -m 'imported project'
 
1020
    """
 
1021
    takes_args = ['location?']
 
1022
    takes_options = [branch_format_option]
 
1023
    def run(self, location=None, branch_format=None):
 
1024
        if branch_format is None:
 
1025
            branch_format = get_format_type('default')
 
1026
        if location is None:
 
1027
            location = u'.'
 
1028
 
 
1029
        to_transport = transport.get_transport(location)
 
1030
 
 
1031
        # The path has to exist to initialize a
 
1032
        # branch inside of it.
 
1033
        # Just using os.mkdir, since I don't
 
1034
        # believe that we want to create a bunch of
 
1035
        # locations if the user supplies an extended path
 
1036
        # TODO: create-prefix
 
1037
        try:
 
1038
            to_transport.mkdir('.')
 
1039
        except errors.FileExists:
 
1040
            pass
 
1041
                    
 
1042
        try:
 
1043
            existing_bzrdir = bzrdir.BzrDir.open(location)
 
1044
        except NotBranchError:
 
1045
            # really a NotBzrDir error...
 
1046
            bzrdir.BzrDir.create_branch_convenience(location, 
 
1047
                                                    format=branch_format)
 
1048
        else:
 
1049
            if existing_bzrdir.has_branch():
 
1050
                if (isinstance(to_transport, LocalTransport)
 
1051
                    and not existing_bzrdir.has_workingtree()):
 
1052
                        raise errors.BranchExistsWithoutWorkingTree(location)
 
1053
                raise errors.AlreadyBranchError(location)
 
1054
            else:
 
1055
                existing_bzrdir.create_branch()
 
1056
                existing_bzrdir.create_workingtree()
 
1057
 
 
1058
 
 
1059
class cmd_init_repository(Command):
 
1060
    """Create a shared repository to hold branches.
 
1061
 
 
1062
    New branches created under the repository directory will store their revisions
 
1063
    in the repository, not in the branch directory, if the branch format supports
 
1064
    shared storage.
 
1065
 
 
1066
    example:
 
1067
        bzr init-repo repo
 
1068
        bzr init repo/trunk
 
1069
        bzr checkout --lightweight repo/trunk trunk-checkout
 
1070
        cd trunk-checkout
 
1071
        (add files here)
 
1072
    """
 
1073
    takes_args = ["location"] 
 
1074
    takes_options = [repo_format_option,
 
1075
                     Option('trees',
 
1076
                             help='Allows branches in repository to have'
 
1077
                             ' a working tree')]
 
1078
    aliases = ["init-repo"]
 
1079
    def run(self, location, repository_format=None, trees=False):
 
1080
        if repository_format is None:
 
1081
            repository_format = get_format_type('default')
 
1082
        to_transport = transport.get_transport(location)
 
1083
        try:
 
1084
            to_transport.mkdir('.')
 
1085
        except errors.FileExists:
 
1086
            pass
 
1087
 
 
1088
        newdir = repository_format.initialize_on_transport(to_transport)
 
1089
        repo = newdir.create_repository(shared=True)
 
1090
        repo.set_make_working_trees(trees)
 
1091
 
 
1092
 
 
1093
class cmd_diff(Command):
 
1094
    """Show differences in the working tree or between revisions.
 
1095
    
 
1096
    If files are listed, only the changes in those files are listed.
 
1097
    Otherwise, all changes for the tree are listed.
 
1098
 
 
1099
    "bzr diff -p1" is equivalent to "bzr diff --prefix old/:new/", and
 
1100
    produces patches suitable for "patch -p1".
 
1101
 
 
1102
    examples:
 
1103
        bzr diff
 
1104
            Shows the difference in the working tree versus the last commit
 
1105
        bzr diff -r1
 
1106
            Difference between the working tree and revision 1
 
1107
        bzr diff -r1..2
 
1108
            Difference between revision 2 and revision 1
 
1109
        bzr diff --diff-prefix old/:new/
 
1110
            Same as 'bzr diff' but prefix paths with old/ and new/
 
1111
        bzr diff bzr.mine bzr.dev
 
1112
            Show the differences between the two working trees
 
1113
        bzr diff foo.c
 
1114
            Show just the differences for 'foo.c'
 
1115
    """
 
1116
    # TODO: Option to use external diff command; could be GNU diff, wdiff,
 
1117
    #       or a graphical diff.
 
1118
 
 
1119
    # TODO: Python difflib is not exactly the same as unidiff; should
 
1120
    #       either fix it up or prefer to use an external diff.
 
1121
 
 
1122
    # TODO: Selected-file diff is inefficient and doesn't show you
 
1123
    #       deleted files.
 
1124
 
 
1125
    # TODO: This probably handles non-Unix newlines poorly.
 
1126
    
 
1127
    takes_args = ['file*']
 
1128
    takes_options = ['revision', 'diff-options', 'prefix']
 
1129
    aliases = ['di', 'dif']
 
1130
    encoding_type = 'exact'
 
1131
 
 
1132
    @display_command
 
1133
    def run(self, revision=None, file_list=None, diff_options=None,
 
1134
            prefix=None):
 
1135
        from bzrlib.diff import diff_cmd_helper, show_diff_trees
 
1136
 
 
1137
        if (prefix is None) or (prefix == '0'):
 
1138
            # diff -p0 format
 
1139
            old_label = ''
 
1140
            new_label = ''
 
1141
        elif prefix == '1':
 
1142
            old_label = 'old/'
 
1143
            new_label = 'new/'
 
1144
        else:
 
1145
            if not ':' in prefix:
 
1146
                 raise BzrError("--diff-prefix expects two values separated by a colon")
 
1147
            old_label, new_label = prefix.split(":")
 
1148
        
 
1149
        try:
 
1150
            tree1, file_list = internal_tree_files(file_list)
 
1151
            tree2 = None
 
1152
            b = None
 
1153
            b2 = None
 
1154
        except FileInWrongBranch:
 
1155
            if len(file_list) != 2:
 
1156
                raise BzrCommandError("Files are in different branches")
 
1157
 
 
1158
            tree1, file1 = WorkingTree.open_containing(file_list[0])
 
1159
            tree2, file2 = WorkingTree.open_containing(file_list[1])
 
1160
            if file1 != "" or file2 != "":
 
1161
                # FIXME diff those two files. rbc 20051123
 
1162
                raise BzrCommandError("Files are in different branches")
 
1163
            file_list = None
 
1164
        except NotBranchError:
 
1165
            if (revision is not None and len(revision) == 2
 
1166
                and not revision[0].needs_branch()
 
1167
                and not revision[1].needs_branch()):
 
1168
                # If both revision specs include a branch, we can
 
1169
                # diff them without needing a local working tree
 
1170
                tree1, tree2 = None, None
 
1171
            else:
 
1172
                raise
 
1173
        if revision is not None:
 
1174
            if tree2 is not None:
 
1175
                raise BzrCommandError("Can't specify -r with two branches")
 
1176
            if (len(revision) == 1) or (revision[1].spec is None):
 
1177
                return diff_cmd_helper(tree1, file_list, diff_options,
 
1178
                                       revision[0], 
 
1179
                                       old_label=old_label, new_label=new_label)
 
1180
            elif len(revision) == 2:
 
1181
                return diff_cmd_helper(tree1, file_list, diff_options,
 
1182
                                       revision[0], revision[1],
 
1183
                                       old_label=old_label, new_label=new_label)
 
1184
            else:
 
1185
                raise BzrCommandError('bzr diff --revision takes exactly one or two revision identifiers')
 
1186
        else:
 
1187
            if tree2 is not None:
 
1188
                return show_diff_trees(tree1, tree2, sys.stdout, 
 
1189
                                       specific_files=file_list,
 
1190
                                       external_diff_options=diff_options,
 
1191
                                       old_label=old_label, new_label=new_label)
 
1192
            else:
 
1193
                return diff_cmd_helper(tree1, file_list, diff_options,
 
1194
                                       old_label=old_label, new_label=new_label)
 
1195
 
 
1196
 
 
1197
class cmd_deleted(Command):
 
1198
    """List files deleted in the working tree.
 
1199
    """
 
1200
    # TODO: Show files deleted since a previous revision, or
 
1201
    # between two revisions.
 
1202
    # TODO: Much more efficient way to do this: read in new
 
1203
    # directories with readdir, rather than stating each one.  Same
 
1204
    # level of effort but possibly much less IO.  (Or possibly not,
 
1205
    # if the directories are very large...)
 
1206
    takes_options = ['show-ids']
 
1207
 
 
1208
    @display_command
 
1209
    def run(self, show_ids=False):
 
1210
        tree = WorkingTree.open_containing(u'.')[0]
 
1211
        old = tree.basis_tree()
 
1212
        for path, ie in old.inventory.iter_entries():
 
1213
            if not tree.has_id(ie.file_id):
 
1214
                self.outf.write(path)
 
1215
                if show_ids:
 
1216
                    self.outf.write(' ')
 
1217
                    self.outf.write(ie.file_id)
 
1218
                self.outf.write('\n')
 
1219
 
 
1220
 
 
1221
class cmd_modified(Command):
 
1222
    """List files modified in working tree."""
 
1223
    hidden = True
 
1224
    @display_command
 
1225
    def run(self):
 
1226
        tree = WorkingTree.open_containing(u'.')[0]
 
1227
        td = tree.changes_from(tree.basis_tree())
 
1228
        for path, id, kind, text_modified, meta_modified in td.modified:
 
1229
            self.outf.write(path + '\n')
 
1230
 
 
1231
 
 
1232
class cmd_added(Command):
 
1233
    """List files added in working tree."""
 
1234
    hidden = True
 
1235
    @display_command
 
1236
    def run(self):
 
1237
        wt = WorkingTree.open_containing(u'.')[0]
 
1238
        basis_inv = wt.basis_tree().inventory
 
1239
        inv = wt.inventory
 
1240
        for file_id in inv:
 
1241
            if file_id in basis_inv:
 
1242
                continue
 
1243
            path = inv.id2path(file_id)
 
1244
            if not os.access(osutils.abspath(path), os.F_OK):
 
1245
                continue
 
1246
            self.outf.write(path + '\n')
 
1247
 
 
1248
 
 
1249
class cmd_root(Command):
 
1250
    """Show the tree root directory.
 
1251
 
 
1252
    The root is the nearest enclosing directory with a .bzr control
 
1253
    directory."""
 
1254
    takes_args = ['filename?']
 
1255
    @display_command
 
1256
    def run(self, filename=None):
 
1257
        """Print the branch root."""
 
1258
        tree = WorkingTree.open_containing(filename)[0]
 
1259
        self.outf.write(tree.basedir + '\n')
 
1260
 
 
1261
 
 
1262
class cmd_log(Command):
 
1263
    """Show log of a branch, file, or directory.
 
1264
 
 
1265
    By default show the log of the branch containing the working directory.
 
1266
 
 
1267
    To request a range of logs, you can use the command -r begin..end
 
1268
    -r revision requests a specific revision, -r ..end or -r begin.. are
 
1269
    also valid.
 
1270
 
 
1271
    examples:
 
1272
        bzr log
 
1273
        bzr log foo.c
 
1274
        bzr log -r -10.. http://server/branch
 
1275
    """
 
1276
 
 
1277
    # TODO: Make --revision support uuid: and hash: [future tag:] notation.
 
1278
 
 
1279
    takes_args = ['location?']
 
1280
    takes_options = [Option('forward', 
 
1281
                            help='show from oldest to newest'),
 
1282
                     'timezone', 
 
1283
                     Option('verbose', 
 
1284
                             help='show files changed in each revision'),
 
1285
                     'show-ids', 'revision',
 
1286
                     'log-format',
 
1287
                     Option('message',
 
1288
                            help='show revisions whose message matches this regexp',
 
1289
                            type=str),
 
1290
                     ]
 
1291
    encoding_type = 'replace'
 
1292
 
 
1293
    @display_command
 
1294
    def run(self, location=None, timezone='original',
 
1295
            verbose=False,
 
1296
            show_ids=False,
 
1297
            forward=False,
 
1298
            revision=None,
 
1299
            log_format=None,
 
1300
            message=None):
 
1301
        from bzrlib.log import log_formatter, show_log
 
1302
        assert message is None or isinstance(message, basestring), \
 
1303
            "invalid message argument %r" % message
 
1304
        direction = (forward and 'forward') or 'reverse'
 
1305
        
 
1306
        # log everything
 
1307
        file_id = None
 
1308
        if location:
 
1309
            # find the file id to log:
 
1310
 
 
1311
            dir, fp = bzrdir.BzrDir.open_containing(location)
 
1312
            b = dir.open_branch()
 
1313
            if fp != '':
 
1314
                try:
 
1315
                    # might be a tree:
 
1316
                    inv = dir.open_workingtree().inventory
 
1317
                except (errors.NotBranchError, errors.NotLocalUrl):
 
1318
                    # either no tree, or is remote.
 
1319
                    inv = b.basis_tree().inventory
 
1320
                file_id = inv.path2id(fp)
 
1321
        else:
 
1322
            # local dir only
 
1323
            # FIXME ? log the current subdir only RBC 20060203 
 
1324
            dir, relpath = bzrdir.BzrDir.open_containing('.')
 
1325
            b = dir.open_branch()
 
1326
 
 
1327
        if revision is None:
 
1328
            rev1 = None
 
1329
            rev2 = None
 
1330
        elif len(revision) == 1:
 
1331
            rev1 = rev2 = revision[0].in_history(b).revno
 
1332
        elif len(revision) == 2:
 
1333
            if revision[0].spec is None:
 
1334
                # missing begin-range means first revision
 
1335
                rev1 = 1
 
1336
            else:
 
1337
                rev1 = revision[0].in_history(b).revno
 
1338
 
 
1339
            if revision[1].spec is None:
 
1340
                # missing end-range means last known revision
 
1341
                rev2 = b.revno()
 
1342
            else:
 
1343
                rev2 = revision[1].in_history(b).revno
 
1344
        else:
 
1345
            raise BzrCommandError('bzr log --revision takes one or two values.')
 
1346
 
 
1347
        # By this point, the revision numbers are converted to the +ve
 
1348
        # form if they were supplied in the -ve form, so we can do
 
1349
        # this comparison in relative safety
 
1350
        if rev1 > rev2:
 
1351
            (rev2, rev1) = (rev1, rev2)
 
1352
 
 
1353
        if (log_format == None):
 
1354
            log_format = b.get_config().log_format()
 
1355
        lf = log_formatter(log_format,
 
1356
                           show_ids=show_ids,
 
1357
                           to_file=self.outf,
 
1358
                           show_timezone=timezone)
 
1359
 
 
1360
        show_log(b,
 
1361
                 lf,
 
1362
                 file_id,
 
1363
                 verbose=verbose,
 
1364
                 direction=direction,
 
1365
                 start_revision=rev1,
 
1366
                 end_revision=rev2,
 
1367
                 search=message)
 
1368
 
 
1369
 
 
1370
class cmd_touching_revisions(Command):
 
1371
    """Return revision-ids which affected a particular file.
 
1372
 
 
1373
    A more user-friendly interface is "bzr log FILE".
 
1374
    """
 
1375
 
 
1376
    hidden = True
 
1377
    takes_args = ["filename"]
 
1378
 
 
1379
    @display_command
 
1380
    def run(self, filename):
 
1381
        tree, relpath = WorkingTree.open_containing(filename)
 
1382
        b = tree.branch
 
1383
        inv = tree.read_working_inventory()
 
1384
        file_id = inv.path2id(relpath)
 
1385
        for revno, revision_id, what in log.find_touching_revisions(b, file_id):
 
1386
            self.outf.write("%6d %s\n" % (revno, what))
 
1387
 
 
1388
 
 
1389
class cmd_ls(Command):
 
1390
    """List files in a tree.
 
1391
    """
 
1392
    # TODO: Take a revision or remote path and list that tree instead.
 
1393
    hidden = True
 
1394
    takes_options = ['verbose', 'revision',
 
1395
                     Option('non-recursive',
 
1396
                            help='don\'t recurse into sub-directories'),
 
1397
                     Option('from-root',
 
1398
                            help='Print all paths from the root of the branch.'),
 
1399
                     Option('unknown', help='Print unknown files'),
 
1400
                     Option('versioned', help='Print versioned files'),
 
1401
                     Option('ignored', help='Print ignored files'),
 
1402
 
 
1403
                     Option('null', help='Null separate the files'),
 
1404
                    ]
 
1405
    @display_command
 
1406
    def run(self, revision=None, verbose=False, 
 
1407
            non_recursive=False, from_root=False,
 
1408
            unknown=False, versioned=False, ignored=False,
 
1409
            null=False):
 
1410
 
 
1411
        if verbose and null:
 
1412
            raise BzrCommandError('Cannot set both --verbose and --null')
 
1413
        all = not (unknown or versioned or ignored)
 
1414
 
 
1415
        selection = {'I':ignored, '?':unknown, 'V':versioned}
 
1416
 
 
1417
        tree, relpath = WorkingTree.open_containing(u'.')
 
1418
        if from_root:
 
1419
            relpath = u''
 
1420
        elif relpath:
 
1421
            relpath += '/'
 
1422
        if revision is not None:
 
1423
            tree = tree.branch.repository.revision_tree(
 
1424
                revision[0].in_history(tree.branch).rev_id)
 
1425
 
 
1426
        for fp, fc, kind, fid, entry in tree.list_files():
 
1427
            if fp.startswith(relpath):
 
1428
                fp = fp[len(relpath):]
 
1429
                if non_recursive and '/' in fp:
 
1430
                    continue
 
1431
                if not all and not selection[fc]:
 
1432
                    continue
 
1433
                if verbose:
 
1434
                    kindch = entry.kind_character()
 
1435
                    self.outf.write('%-8s %s%s\n' % (fc, fp, kindch))
 
1436
                elif null:
 
1437
                    self.outf.write(fp + '\0')
 
1438
                    self.outf.flush()
 
1439
                else:
 
1440
                    self.outf.write(fp + '\n')
 
1441
 
 
1442
 
 
1443
class cmd_unknowns(Command):
 
1444
    """List unknown files."""
 
1445
    @display_command
 
1446
    def run(self):
 
1447
        for f in WorkingTree.open_containing(u'.')[0].unknowns():
 
1448
            self.outf.write(osutils.quotefn(f) + '\n')
 
1449
 
 
1450
 
 
1451
class cmd_ignore(Command):
 
1452
    """Ignore a command or pattern.
 
1453
 
 
1454
    To remove patterns from the ignore list, edit the .bzrignore file.
 
1455
 
 
1456
    If the pattern contains a slash, it is compared to the whole path
 
1457
    from the branch root.  Otherwise, it is compared to only the last
 
1458
    component of the path.  To match a file only in the root directory,
 
1459
    prepend './'.
 
1460
 
 
1461
    Ignore patterns are case-insensitive on case-insensitive systems.
 
1462
 
 
1463
    Note: wildcards must be quoted from the shell on Unix.
 
1464
 
 
1465
    examples:
 
1466
        bzr ignore ./Makefile
 
1467
        bzr ignore '*.class'
 
1468
    """
 
1469
    # TODO: Complain if the filename is absolute
 
1470
    takes_args = ['name_pattern?']
 
1471
    takes_options = [
 
1472
                     Option('old-default-rules',
 
1473
                            help='Out the ignore rules bzr < 0.9 always used.')
 
1474
                     ]
 
1475
    
 
1476
    def run(self, name_pattern=None, old_default_rules=None):
 
1477
        from bzrlib.atomicfile import AtomicFile
 
1478
        if old_default_rules is not None:
 
1479
            # dump the rules and exit
 
1480
            for pattern in ignores.OLD_DEFAULTS:
 
1481
                print pattern
 
1482
            return
 
1483
        if name_pattern is None:
 
1484
            raise BzrCommandError("ignore requires a NAME_PATTERN")
 
1485
        tree, relpath = WorkingTree.open_containing(u'.')
 
1486
        ifn = tree.abspath('.bzrignore')
 
1487
        if os.path.exists(ifn):
 
1488
            f = open(ifn, 'rt')
 
1489
            try:
 
1490
                igns = f.read().decode('utf-8')
 
1491
            finally:
 
1492
                f.close()
 
1493
        else:
 
1494
            igns = ''
 
1495
 
 
1496
        # TODO: If the file already uses crlf-style termination, maybe
 
1497
        # we should use that for the newly added lines?
 
1498
 
 
1499
        if igns and igns[-1] != '\n':
 
1500
            igns += '\n'
 
1501
        igns += name_pattern + '\n'
 
1502
 
 
1503
        f = AtomicFile(ifn, 'wt')
 
1504
        try:
 
1505
            f.write(igns.encode('utf-8'))
 
1506
            f.commit()
 
1507
        finally:
 
1508
            f.close()
 
1509
 
 
1510
        inv = tree.inventory
 
1511
        if inv.path2id('.bzrignore'):
 
1512
            mutter('.bzrignore is already versioned')
 
1513
        else:
 
1514
            mutter('need to make new .bzrignore file versioned')
 
1515
            tree.add(['.bzrignore'])
 
1516
 
 
1517
 
 
1518
class cmd_ignored(Command):
 
1519
    """List ignored files and the patterns that matched them.
 
1520
 
 
1521
    See also: bzr ignore"""
 
1522
    @display_command
 
1523
    def run(self):
 
1524
        tree = WorkingTree.open_containing(u'.')[0]
 
1525
        for path, file_class, kind, file_id, entry in tree.list_files():
 
1526
            if file_class != 'I':
 
1527
                continue
 
1528
            ## XXX: Slightly inefficient since this was already calculated
 
1529
            pat = tree.is_ignored(path)
 
1530
            print '%-50s %s' % (path, pat)
 
1531
 
 
1532
 
 
1533
class cmd_lookup_revision(Command):
 
1534
    """Lookup the revision-id from a revision-number
 
1535
 
 
1536
    example:
 
1537
        bzr lookup-revision 33
 
1538
    """
 
1539
    hidden = True
 
1540
    takes_args = ['revno']
 
1541
    
 
1542
    @display_command
 
1543
    def run(self, revno):
 
1544
        try:
 
1545
            revno = int(revno)
 
1546
        except ValueError:
 
1547
            raise BzrCommandError("not a valid revision-number: %r" % revno)
 
1548
 
 
1549
        print WorkingTree.open_containing(u'.')[0].branch.get_rev_id(revno)
 
1550
 
 
1551
 
 
1552
class cmd_export(Command):
 
1553
    """Export past revision to destination directory.
 
1554
 
 
1555
    If no revision is specified this exports the last committed revision.
 
1556
 
 
1557
    Format may be an "exporter" name, such as tar, tgz, tbz2.  If none is
 
1558
    given, try to find the format with the extension. If no extension
 
1559
    is found exports to a directory (equivalent to --format=dir).
 
1560
 
 
1561
    Root may be the top directory for tar, tgz and tbz2 formats. If none
 
1562
    is given, the top directory will be the root name of the file.
 
1563
 
 
1564
    Note: export of tree with non-ascii filenames to zip is not supported.
 
1565
 
 
1566
     Supported formats       Autodetected by extension
 
1567
     -----------------       -------------------------
 
1568
         dir                            -
 
1569
         tar                          .tar
 
1570
         tbz2                    .tar.bz2, .tbz2
 
1571
         tgz                      .tar.gz, .tgz
 
1572
         zip                          .zip
 
1573
    """
 
1574
    takes_args = ['dest']
 
1575
    takes_options = ['revision', 'format', 'root']
 
1576
    def run(self, dest, revision=None, format=None, root=None):
 
1577
        from bzrlib.export import export
 
1578
        tree = WorkingTree.open_containing(u'.')[0]
 
1579
        b = tree.branch
 
1580
        if revision is None:
 
1581
            # should be tree.last_revision  FIXME
 
1582
            rev_id = b.last_revision()
 
1583
        else:
 
1584
            if len(revision) != 1:
 
1585
                raise BzrError('bzr export --revision takes exactly 1 argument')
 
1586
            rev_id = revision[0].in_history(b).rev_id
 
1587
        t = b.repository.revision_tree(rev_id)
 
1588
        try:
 
1589
            export(t, dest, format, root)
 
1590
        except errors.NoSuchExportFormat, e:
 
1591
            raise BzrCommandError('Unsupported export format: %s' % e.format)
 
1592
 
 
1593
 
 
1594
class cmd_cat(Command):
 
1595
    """Write a file's text from a previous revision."""
 
1596
 
 
1597
    takes_options = ['revision']
 
1598
    takes_args = ['filename']
 
1599
 
 
1600
    @display_command
 
1601
    def run(self, filename, revision=None):
 
1602
        if revision is not None and len(revision) != 1:
 
1603
            raise BzrCommandError("bzr cat --revision takes exactly one number")
 
1604
        tree = None
 
1605
        try:
 
1606
            tree, relpath = WorkingTree.open_containing(filename)
 
1607
            b = tree.branch
 
1608
        except NotBranchError:
 
1609
            pass
 
1610
 
 
1611
        if tree is None:
 
1612
            b, relpath = Branch.open_containing(filename)
 
1613
        if revision is None:
 
1614
            revision_id = b.last_revision()
 
1615
        else:
 
1616
            revision_id = revision[0].in_history(b).rev_id
 
1617
        b.print_file(relpath, revision_id)
 
1618
 
 
1619
 
 
1620
class cmd_local_time_offset(Command):
 
1621
    """Show the offset in seconds from GMT to local time."""
 
1622
    hidden = True    
 
1623
    @display_command
 
1624
    def run(self):
 
1625
        print osutils.local_time_offset()
 
1626
 
 
1627
 
 
1628
 
 
1629
class cmd_commit(Command):
 
1630
    """Commit changes into a new revision.
 
1631
    
 
1632
    If no arguments are given, the entire tree is committed.
 
1633
 
 
1634
    If selected files are specified, only changes to those files are
 
1635
    committed.  If a directory is specified then the directory and everything 
 
1636
    within it is committed.
 
1637
 
 
1638
    A selected-file commit may fail in some cases where the committed
 
1639
    tree would be invalid, such as trying to commit a file in a
 
1640
    newly-added directory that is not itself committed.
 
1641
    """
 
1642
    # TODO: Run hooks on tree to-be-committed, and after commit.
 
1643
 
 
1644
    # TODO: Strict commit that fails if there are deleted files.
 
1645
    #       (what does "deleted files" mean ??)
 
1646
 
 
1647
    # TODO: Give better message for -s, --summary, used by tla people
 
1648
 
 
1649
    # XXX: verbose currently does nothing
 
1650
 
 
1651
    takes_args = ['selected*']
 
1652
    takes_options = ['message', 'verbose', 
 
1653
                     Option('unchanged',
 
1654
                            help='commit even if nothing has changed'),
 
1655
                     Option('file', type=str, 
 
1656
                            argname='msgfile',
 
1657
                            help='file containing commit message'),
 
1658
                     Option('strict',
 
1659
                            help="refuse to commit if there are unknown "
 
1660
                            "files in the working tree."),
 
1661
                     Option('local',
 
1662
                            help="perform a local only commit in a bound "
 
1663
                                 "branch. Such commits are not pushed to "
 
1664
                                 "the master branch until a normal commit "
 
1665
                                 "is performed."
 
1666
                            ),
 
1667
                     ]
 
1668
    aliases = ['ci', 'checkin']
 
1669
 
 
1670
    def run(self, message=None, file=None, verbose=True, selected_list=None,
 
1671
            unchanged=False, strict=False, local=False):
 
1672
        from bzrlib.commit import (NullCommitReporter, ReportCommitToLog)
 
1673
        from bzrlib.errors import (PointlessCommit, ConflictsInTree,
 
1674
                StrictCommitFailed)
 
1675
        from bzrlib.msgeditor import edit_commit_message, \
 
1676
                make_commit_message_template
 
1677
        from tempfile import TemporaryFile
 
1678
 
 
1679
        # TODO: Need a blackbox test for invoking the external editor; may be
 
1680
        # slightly problematic to run this cross-platform.
 
1681
 
 
1682
        # TODO: do more checks that the commit will succeed before 
 
1683
        # spending the user's valuable time typing a commit message.
 
1684
        #
 
1685
        # TODO: if the commit *does* happen to fail, then save the commit 
 
1686
        # message to a temporary file where it can be recovered
 
1687
        tree, selected_list = tree_files(selected_list)
 
1688
        if selected_list == ['']:
 
1689
            # workaround - commit of root of tree should be exactly the same
 
1690
            # as just default commit in that tree, and succeed even though
 
1691
            # selected-file merge commit is not done yet
 
1692
            selected_list = []
 
1693
 
 
1694
        if local and not tree.branch.get_bound_location():
 
1695
            raise errors.LocalRequiresBoundBranch()
 
1696
        if message is None and not file:
 
1697
            template = make_commit_message_template(tree, selected_list)
 
1698
            message = edit_commit_message(template)
 
1699
            if message is None:
 
1700
                raise BzrCommandError("please specify a commit message"
 
1701
                                      " with either --message or --file")
 
1702
        elif message and file:
 
1703
            raise BzrCommandError("please specify either --message or --file")
 
1704
        
 
1705
        if file:
 
1706
            message = codecs.open(file, 'rt', bzrlib.user_encoding).read()
 
1707
 
 
1708
        if message == "":
 
1709
            raise BzrCommandError("empty commit message specified")
 
1710
        
 
1711
        if verbose:
 
1712
            reporter = ReportCommitToLog()
 
1713
        else:
 
1714
            reporter = NullCommitReporter()
 
1715
        
 
1716
        try:
 
1717
            tree.commit(message, specific_files=selected_list,
 
1718
                        allow_pointless=unchanged, strict=strict, local=local,
 
1719
                        reporter=reporter)
 
1720
        except PointlessCommit:
 
1721
            # FIXME: This should really happen before the file is read in;
 
1722
            # perhaps prepare the commit; get the message; then actually commit
 
1723
            raise BzrCommandError("no changes to commit."
 
1724
                                  " use --unchanged to commit anyhow")
 
1725
        except ConflictsInTree:
 
1726
            raise BzrCommandError("Conflicts detected in working tree.  "
 
1727
                'Use "bzr conflicts" to list, "bzr resolve FILE" to resolve.')
 
1728
        except StrictCommitFailed:
 
1729
            raise BzrCommandError("Commit refused because there are unknown "
 
1730
                                  "files in the working tree.")
 
1731
        except errors.BoundBranchOutOfDate, e:
 
1732
            raise BzrCommandError(str(e) + "\n"
 
1733
                'To commit to master branch, run update and then commit.\n'
 
1734
                'You can also pass --local to commit to continue working '
 
1735
                'disconnected.')
 
1736
 
 
1737
class cmd_check(Command):
 
1738
    """Validate consistency of branch history.
 
1739
 
 
1740
    This command checks various invariants about the branch storage to
 
1741
    detect data corruption or bzr bugs.
 
1742
    """
 
1743
    takes_args = ['branch?']
 
1744
    takes_options = ['verbose']
 
1745
 
 
1746
    def run(self, branch=None, verbose=False):
 
1747
        from bzrlib.check import check
 
1748
        if branch is None:
 
1749
            tree = WorkingTree.open_containing()[0]
 
1750
            branch = tree.branch
 
1751
        else:
 
1752
            branch = Branch.open(branch)
 
1753
        check(branch, verbose)
 
1754
 
 
1755
 
 
1756
class cmd_scan_cache(Command):
 
1757
    hidden = True
 
1758
    def run(self):
 
1759
        from bzrlib.hashcache import HashCache
 
1760
 
 
1761
        c = HashCache(u'.')
 
1762
        c.read()
 
1763
        c.scan()
 
1764
            
 
1765
        print '%6d stats' % c.stat_count
 
1766
        print '%6d in hashcache' % len(c._cache)
 
1767
        print '%6d files removed from cache' % c.removed_count
 
1768
        print '%6d hashes updated' % c.update_count
 
1769
        print '%6d files changed too recently to cache' % c.danger_count
 
1770
 
 
1771
        if c.needs_write:
 
1772
            c.write()
 
1773
 
 
1774
 
 
1775
class cmd_upgrade(Command):
 
1776
    """Upgrade branch storage to current format.
 
1777
 
 
1778
    The check command or bzr developers may sometimes advise you to run
 
1779
    this command. When the default format has changed you may also be warned
 
1780
    during other operations to upgrade.
 
1781
    """
 
1782
    takes_args = ['url?']
 
1783
    takes_options = [ branch_format_option ]
 
1784
 
 
1785
    def run(self, url='.', branch_format=None):
 
1786
        from bzrlib.upgrade import upgrade
 
1787
        if branch_format is None:
 
1788
            branch_format = get_format_type('default')
 
1789
        upgrade(url, branch_format)
 
1790
 
 
1791
 
 
1792
class cmd_whoami(Command):
 
1793
    """Show or set bzr user id.
 
1794
    
 
1795
    examples:
 
1796
        bzr whoami --email
 
1797
        bzr whoami 'Frank Chu <fchu@example.com>'
 
1798
    """
 
1799
    takes_options = [ Option('email',
 
1800
                             help='display email address only'),
 
1801
                      Option('branch',
 
1802
                             help='set identity for the current branch instead of '
 
1803
                                  'globally'),
 
1804
                    ]
 
1805
    takes_args = ['name?']
 
1806
    encoding_type = 'replace'
 
1807
    
 
1808
    @display_command
 
1809
    def run(self, email=False, branch=False, name=None):
 
1810
        if name is None:
 
1811
            # use branch if we're inside one; otherwise global config
 
1812
            try:
 
1813
                c = Branch.open_containing('.')[0].get_config()
 
1814
            except NotBranchError:
 
1815
                c = config.GlobalConfig()
 
1816
            if email:
 
1817
                self.outf.write(c.user_email() + '\n')
 
1818
            else:
 
1819
                self.outf.write(c.username() + '\n')
 
1820
            return
 
1821
 
 
1822
        # display a warning if an email address isn't included in the given name.
 
1823
        try:
 
1824
            config.extract_email_address(name)
 
1825
        except BzrError, e:
 
1826
            warning('"%s" does not seem to contain an email address.  '
 
1827
                    'This is allowed, but not recommended.', name)
 
1828
        
 
1829
        # use global config unless --branch given
 
1830
        if branch:
 
1831
            c = Branch.open_containing('.')[0].get_config()
 
1832
        else:
 
1833
            c = config.GlobalConfig()
 
1834
        c.set_user_option('email', name)
 
1835
 
 
1836
 
 
1837
class cmd_nick(Command):
 
1838
    """Print or set the branch nickname.  
 
1839
 
 
1840
    If unset, the tree root directory name is used as the nickname
 
1841
    To print the current nickname, execute with no argument.  
 
1842
    """
 
1843
    takes_args = ['nickname?']
 
1844
    def run(self, nickname=None):
 
1845
        branch = Branch.open_containing(u'.')[0]
 
1846
        if nickname is None:
 
1847
            self.printme(branch)
 
1848
        else:
 
1849
            branch.nick = nickname
 
1850
 
 
1851
    @display_command
 
1852
    def printme(self, branch):
 
1853
        print branch.nick 
 
1854
 
 
1855
 
 
1856
class cmd_selftest(Command):
 
1857
    """Run internal test suite.
 
1858
    
 
1859
    This creates temporary test directories in the working directory,
 
1860
    but not existing data is affected.  These directories are deleted
 
1861
    if the tests pass, or left behind to help in debugging if they
 
1862
    fail and --keep-output is specified.
 
1863
    
 
1864
    If arguments are given, they are regular expressions that say
 
1865
    which tests should run.
 
1866
 
 
1867
    If the global option '--no-plugins' is given, plugins are not loaded
 
1868
    before running the selftests.  This has two effects: features provided or
 
1869
    modified by plugins will not be tested, and tests provided by plugins will
 
1870
    not be run.
 
1871
 
 
1872
    examples:
 
1873
        bzr selftest ignore
 
1874
        bzr --no-plugins selftest -v
 
1875
    """
 
1876
    # TODO: --list should give a list of all available tests
 
1877
 
 
1878
    # NB: this is used from the class without creating an instance, which is
 
1879
    # why it does not have a self parameter.
 
1880
    def get_transport_type(typestring):
 
1881
        """Parse and return a transport specifier."""
 
1882
        if typestring == "sftp":
 
1883
            from bzrlib.transport.sftp import SFTPAbsoluteServer
 
1884
            return SFTPAbsoluteServer
 
1885
        if typestring == "memory":
 
1886
            from bzrlib.transport.memory import MemoryServer
 
1887
            return MemoryServer
 
1888
        if typestring == "fakenfs":
 
1889
            from bzrlib.transport.fakenfs import FakeNFSServer
 
1890
            return FakeNFSServer
 
1891
        msg = "No known transport type %s. Supported types are: sftp\n" %\
 
1892
            (typestring)
 
1893
        raise BzrCommandError(msg)
 
1894
 
 
1895
    hidden = True
 
1896
    takes_args = ['testspecs*']
 
1897
    takes_options = ['verbose',
 
1898
                     Option('one', help='stop when one test fails'),
 
1899
                     Option('keep-output', 
 
1900
                            help='keep output directories when tests fail'),
 
1901
                     Option('transport', 
 
1902
                            help='Use a different transport by default '
 
1903
                                 'throughout the test suite.',
 
1904
                            type=get_transport_type),
 
1905
                     Option('benchmark', help='run the bzr bencharks.'),
 
1906
                     Option('lsprof-timed',
 
1907
                            help='generate lsprof output for benchmarked'
 
1908
                                 ' sections of code.'),
 
1909
                     ]
 
1910
 
 
1911
    def run(self, testspecs_list=None, verbose=None, one=False,
 
1912
            keep_output=False, transport=None, benchmark=None,
 
1913
            lsprof_timed=None):
 
1914
        import bzrlib.ui
 
1915
        from bzrlib.tests import selftest
 
1916
        import bzrlib.benchmarks as benchmarks
 
1917
        # we don't want progress meters from the tests to go to the
 
1918
        # real output; and we don't want log messages cluttering up
 
1919
        # the real logs.
 
1920
        save_ui = ui.ui_factory
 
1921
        print '%10s: %s' % ('bzr', osutils.realpath(sys.argv[0]))
 
1922
        print '%10s: %s' % ('bzrlib', bzrlib.__path__[0])
 
1923
        print
 
1924
        info('running tests...')
 
1925
        try:
 
1926
            ui.ui_factory = ui.SilentUIFactory()
 
1927
            if testspecs_list is not None:
 
1928
                pattern = '|'.join(testspecs_list)
 
1929
            else:
 
1930
                pattern = ".*"
 
1931
            if benchmark:
 
1932
                test_suite_factory = benchmarks.test_suite
 
1933
                if verbose is None:
 
1934
                    verbose = True
 
1935
            else:
 
1936
                test_suite_factory = None
 
1937
                if verbose is None:
 
1938
                    verbose = False
 
1939
            result = selftest(verbose=verbose, 
 
1940
                              pattern=pattern,
 
1941
                              stop_on_failure=one, 
 
1942
                              keep_output=keep_output,
 
1943
                              transport=transport,
 
1944
                              test_suite_factory=test_suite_factory,
 
1945
                              lsprof_timed=lsprof_timed)
 
1946
            if result:
 
1947
                info('tests passed')
 
1948
            else:
 
1949
                info('tests failed')
 
1950
            return int(not result)
 
1951
        finally:
 
1952
            ui.ui_factory = save_ui
 
1953
 
 
1954
 
 
1955
def _get_bzr_branch():
 
1956
    """If bzr is run from a branch, return Branch or None"""
 
1957
    from os.path import dirname
 
1958
    
 
1959
    try:
 
1960
        branch = Branch.open(dirname(osutils.abspath(dirname(__file__))))
 
1961
        return branch
 
1962
    except errors.BzrError:
 
1963
        return None
 
1964
    
 
1965
 
 
1966
def show_version():
 
1967
    import bzrlib
 
1968
    print "Bazaar (bzr) %s" % bzrlib.__version__
 
1969
    # is bzrlib itself in a branch?
 
1970
    branch = _get_bzr_branch()
 
1971
    if branch:
 
1972
        rh = branch.revision_history()
 
1973
        revno = len(rh)
 
1974
        print "  bzr checkout, revision %d" % (revno,)
 
1975
        print "  nick: %s" % (branch.nick,)
 
1976
        if rh:
 
1977
            print "  revid: %s" % (rh[-1],)
 
1978
    print "Using python interpreter:", sys.executable
 
1979
    import site
 
1980
    print "Using python standard library:", os.path.dirname(site.__file__)
 
1981
    print "Using bzrlib:",
 
1982
    if len(bzrlib.__path__) > 1:
 
1983
        # print repr, which is a good enough way of making it clear it's
 
1984
        # more than one element (eg ['/foo/bar', '/foo/bzr'])
 
1985
        print repr(bzrlib.__path__)
 
1986
    else:
 
1987
        print bzrlib.__path__[0]
 
1988
 
 
1989
    print
 
1990
    print bzrlib.__copyright__
 
1991
    print "http://bazaar-vcs.org/"
 
1992
    print
 
1993
    print "bzr comes with ABSOLUTELY NO WARRANTY.  bzr is free software, and"
 
1994
    print "you may use, modify and redistribute it under the terms of the GNU"
 
1995
    print "General Public License version 2 or later."
 
1996
 
 
1997
 
 
1998
class cmd_version(Command):
 
1999
    """Show version of bzr."""
 
2000
 
 
2001
    @display_command
 
2002
    def run(self):
 
2003
        show_version()
 
2004
 
 
2005
 
 
2006
class cmd_rocks(Command):
 
2007
    """Statement of optimism."""
 
2008
 
 
2009
    hidden = True
 
2010
 
 
2011
    @display_command
 
2012
    def run(self):
 
2013
        print "it sure does!"
 
2014
 
 
2015
 
 
2016
class cmd_find_merge_base(Command):
 
2017
    """Find and print a base revision for merging two branches."""
 
2018
    # TODO: Options to specify revisions on either side, as if
 
2019
    #       merging only part of the history.
 
2020
    takes_args = ['branch', 'other']
 
2021
    hidden = True
 
2022
    
 
2023
    @display_command
 
2024
    def run(self, branch, other):
 
2025
        from bzrlib.revision import common_ancestor, MultipleRevisionSources
 
2026
        
 
2027
        branch1 = Branch.open_containing(branch)[0]
 
2028
        branch2 = Branch.open_containing(other)[0]
 
2029
 
 
2030
        history_1 = branch1.revision_history()
 
2031
        history_2 = branch2.revision_history()
 
2032
 
 
2033
        last1 = branch1.last_revision()
 
2034
        last2 = branch2.last_revision()
 
2035
 
 
2036
        source = MultipleRevisionSources(branch1.repository, 
 
2037
                                         branch2.repository)
 
2038
        
 
2039
        base_rev_id = common_ancestor(last1, last2, source)
 
2040
 
 
2041
        print 'merge base is revision %s' % base_rev_id
 
2042
 
 
2043
 
 
2044
class cmd_merge(Command):
 
2045
    """Perform a three-way merge.
 
2046
    
 
2047
    The branch is the branch you will merge from.  By default, it will merge
 
2048
    the latest revision.  If you specify a revision, that revision will be
 
2049
    merged.  If you specify two revisions, the first will be used as a BASE,
 
2050
    and the second one as OTHER.  Revision numbers are always relative to the
 
2051
    specified branch.
 
2052
 
 
2053
    By default, bzr will try to merge in all new work from the other
 
2054
    branch, automatically determining an appropriate base.  If this
 
2055
    fails, you may need to give an explicit base.
 
2056
    
 
2057
    Merge will do its best to combine the changes in two branches, but there
 
2058
    are some kinds of problems only a human can fix.  When it encounters those,
 
2059
    it will mark a conflict.  A conflict means that you need to fix something,
 
2060
    before you should commit.
 
2061
 
 
2062
    Use bzr resolve when you have fixed a problem.  See also bzr conflicts.
 
2063
 
 
2064
    If there is no default branch set, the first merge will set it. After
 
2065
    that, you can omit the branch to use the default.  To change the
 
2066
    default, use --remember. The value will only be saved if the remote
 
2067
    location can be accessed.
 
2068
 
 
2069
    Examples:
 
2070
 
 
2071
    To merge the latest revision from bzr.dev
 
2072
    bzr merge ../bzr.dev
 
2073
 
 
2074
    To merge changes up to and including revision 82 from bzr.dev
 
2075
    bzr merge -r 82 ../bzr.dev
 
2076
 
 
2077
    To merge the changes introduced by 82, without previous changes:
 
2078
    bzr merge -r 81..82 ../bzr.dev
 
2079
    
 
2080
    merge refuses to run if there are any uncommitted changes, unless
 
2081
    --force is given.
 
2082
 
 
2083
    The following merge types are available:
 
2084
    """
 
2085
    takes_args = ['branch?']
 
2086
    takes_options = ['revision', 'force', 'merge-type', 'reprocess', 'remember',
 
2087
                     Option('show-base', help="Show base revision text in "
 
2088
                            "conflicts"), 
 
2089
                     Option('uncommitted', help='Apply uncommitted changes'
 
2090
                            ' from a working copy, instead of branch changes')]
 
2091
 
 
2092
    def help(self):
 
2093
        from merge import merge_type_help
 
2094
        from inspect import getdoc
 
2095
        return getdoc(self) + '\n' + merge_type_help() 
 
2096
 
 
2097
    def run(self, branch=None, revision=None, force=False, merge_type=None,
 
2098
            show_base=False, reprocess=False, remember=False, 
 
2099
            uncommitted=False):
 
2100
        if merge_type is None:
 
2101
            merge_type = Merge3Merger
 
2102
 
 
2103
        tree = WorkingTree.open_containing(u'.')[0]
 
2104
 
 
2105
        if branch is not None:
 
2106
            try:
 
2107
                reader = bundle.read_bundle_from_url(branch)
 
2108
            except NotABundle:
 
2109
                pass # Continue on considering this url a Branch
 
2110
            else:
 
2111
                conflicts = merge_bundle(reader, tree, not force, merge_type,
 
2112
                                            reprocess, show_base)
 
2113
                if conflicts == 0:
 
2114
                    return 0
 
2115
                else:
 
2116
                    return 1
 
2117
 
 
2118
        branch = self._get_remembered_parent(tree, branch, 'Merging from')
 
2119
 
 
2120
        if revision is None or len(revision) < 1:
 
2121
            if uncommitted:
 
2122
                base = [branch, -1]
 
2123
                other = [branch, None]
 
2124
            else:
 
2125
                base = [None, None]
 
2126
                other = [branch, -1]
 
2127
            other_branch, path = Branch.open_containing(branch)
 
2128
        else:
 
2129
            if uncommitted:
 
2130
                raise BzrCommandError('Cannot use --uncommitted and --revision'
 
2131
                                      ' at the same time.')
 
2132
            if len(revision) == 1:
 
2133
                base = [None, None]
 
2134
                other_branch, path = Branch.open_containing(branch)
 
2135
                revno = revision[0].in_history(other_branch).revno
 
2136
                other = [branch, revno]
 
2137
            else:
 
2138
                assert len(revision) == 2
 
2139
                if None in revision:
 
2140
                    raise BzrCommandError(
 
2141
                        "Merge doesn't permit that revision specifier.")
 
2142
                other_branch, path = Branch.open_containing(branch)
 
2143
 
 
2144
                base = [branch, revision[0].in_history(other_branch).revno]
 
2145
                other = [branch, revision[1].in_history(other_branch).revno]
 
2146
 
 
2147
        if tree.branch.get_parent() is None or remember:
 
2148
            tree.branch.set_parent(other_branch.base)
 
2149
 
 
2150
        if path != "":
 
2151
            interesting_files = [path]
 
2152
        else:
 
2153
            interesting_files = None
 
2154
        pb = ui.ui_factory.nested_progress_bar()
 
2155
        try:
 
2156
            try:
 
2157
                conflict_count = merge(other, base, check_clean=(not force),
 
2158
                                       merge_type=merge_type,
 
2159
                                       reprocess=reprocess,
 
2160
                                       show_base=show_base,
 
2161
                                       pb=pb, file_list=interesting_files)
 
2162
            finally:
 
2163
                pb.finished()
 
2164
            if conflict_count != 0:
 
2165
                return 1
 
2166
            else:
 
2167
                return 0
 
2168
        except errors.AmbiguousBase, e:
 
2169
            m = ("sorry, bzr can't determine the right merge base yet\n"
 
2170
                 "candidates are:\n  "
 
2171
                 + "\n  ".join(e.bases)
 
2172
                 + "\n"
 
2173
                 "please specify an explicit base with -r,\n"
 
2174
                 "and (if you want) report this to the bzr developers\n")
 
2175
            log_error(m)
 
2176
 
 
2177
    # TODO: move up to common parent; this isn't merge-specific anymore. 
 
2178
    def _get_remembered_parent(self, tree, supplied_location, verb_string):
 
2179
        """Use tree.branch's parent if none was supplied.
 
2180
 
 
2181
        Report if the remembered location was used.
 
2182
        """
 
2183
        if supplied_location is not None:
 
2184
            return supplied_location
 
2185
        stored_location = tree.branch.get_parent()
 
2186
        mutter("%s", stored_location)
 
2187
        if stored_location is None:
 
2188
            raise BzrCommandError("No location specified or remembered")
 
2189
        display_url = urlutils.unescape_for_display(stored_location, self.outf.encoding)
 
2190
        self.outf.write("%s remembered location %s\n" % (verb_string, display_url))
 
2191
        return stored_location
 
2192
 
 
2193
 
 
2194
class cmd_remerge(Command):
 
2195
    """Redo a merge.
 
2196
 
 
2197
    Use this if you want to try a different merge technique while resolving
 
2198
    conflicts.  Some merge techniques are better than others, and remerge 
 
2199
    lets you try different ones on different files.
 
2200
 
 
2201
    The options for remerge have the same meaning and defaults as the ones for
 
2202
    merge.  The difference is that remerge can (only) be run when there is a
 
2203
    pending merge, and it lets you specify particular files.
 
2204
 
 
2205
    Examples:
 
2206
    $ bzr remerge --show-base
 
2207
        Re-do the merge of all conflicted files, and show the base text in
 
2208
        conflict regions, in addition to the usual THIS and OTHER texts.
 
2209
 
 
2210
    $ bzr remerge --merge-type weave --reprocess foobar
 
2211
        Re-do the merge of "foobar", using the weave merge algorithm, with
 
2212
        additional processing to reduce the size of conflict regions.
 
2213
    
 
2214
    The following merge types are available:"""
 
2215
    takes_args = ['file*']
 
2216
    takes_options = ['merge-type', 'reprocess',
 
2217
                     Option('show-base', help="Show base revision text in "
 
2218
                            "conflicts")]
 
2219
 
 
2220
    def help(self):
 
2221
        from merge import merge_type_help
 
2222
        from inspect import getdoc
 
2223
        return getdoc(self) + '\n' + merge_type_help() 
 
2224
 
 
2225
    def run(self, file_list=None, merge_type=None, show_base=False,
 
2226
            reprocess=False):
 
2227
        from bzrlib.merge import merge_inner, transform_tree
 
2228
        if merge_type is None:
 
2229
            merge_type = Merge3Merger
 
2230
        tree, file_list = tree_files(file_list)
 
2231
        tree.lock_write()
 
2232
        try:
 
2233
            pending_merges = tree.pending_merges() 
 
2234
            if len(pending_merges) != 1:
 
2235
                raise BzrCommandError("Sorry, remerge only works after normal"
 
2236
                                      " merges.  Not cherrypicking or"
 
2237
                                      " multi-merges.")
 
2238
            repository = tree.branch.repository
 
2239
            base_revision = common_ancestor(tree.branch.last_revision(), 
 
2240
                                            pending_merges[0], repository)
 
2241
            base_tree = repository.revision_tree(base_revision)
 
2242
            other_tree = repository.revision_tree(pending_merges[0])
 
2243
            interesting_ids = None
 
2244
            new_conflicts = []
 
2245
            conflicts = tree.conflicts()
 
2246
            if file_list is not None:
 
2247
                interesting_ids = set()
 
2248
                for filename in file_list:
 
2249
                    file_id = tree.path2id(filename)
 
2250
                    if file_id is None:
 
2251
                        raise NotVersionedError(filename)
 
2252
                    interesting_ids.add(file_id)
 
2253
                    if tree.kind(file_id) != "directory":
 
2254
                        continue
 
2255
                    
 
2256
                    for name, ie in tree.inventory.iter_entries(file_id):
 
2257
                        interesting_ids.add(ie.file_id)
 
2258
                new_conflicts = conflicts.select_conflicts(tree, file_list)[0]
 
2259
            transform_tree(tree, tree.basis_tree(), interesting_ids)
 
2260
            tree.set_conflicts(ConflictList(new_conflicts))
 
2261
            if file_list is None:
 
2262
                restore_files = list(tree.iter_conflicts())
 
2263
            else:
 
2264
                restore_files = file_list
 
2265
            for filename in restore_files:
 
2266
                try:
 
2267
                    restore(tree.abspath(filename))
 
2268
                except NotConflicted:
 
2269
                    pass
 
2270
            conflicts = merge_inner(tree.branch, other_tree, base_tree,
 
2271
                                    this_tree=tree,
 
2272
                                    interesting_ids=interesting_ids, 
 
2273
                                    other_rev_id=pending_merges[0], 
 
2274
                                    merge_type=merge_type, 
 
2275
                                    show_base=show_base,
 
2276
                                    reprocess=reprocess)
 
2277
        finally:
 
2278
            tree.unlock()
 
2279
        if conflicts > 0:
 
2280
            return 1
 
2281
        else:
 
2282
            return 0
 
2283
 
 
2284
class cmd_revert(Command):
 
2285
    """Reverse all changes since the last commit.
 
2286
 
 
2287
    Only versioned files are affected.  Specify filenames to revert only 
 
2288
    those files.  By default, any files that are changed will be backed up
 
2289
    first.  Backup files have a '~' appended to their name.
 
2290
    """
 
2291
    takes_options = ['revision', 'no-backup']
 
2292
    takes_args = ['file*']
 
2293
    aliases = ['merge-revert']
 
2294
 
 
2295
    def run(self, revision=None, no_backup=False, file_list=None):
 
2296
        from bzrlib.commands import parse_spec
 
2297
        if file_list is not None:
 
2298
            if len(file_list) == 0:
 
2299
                raise BzrCommandError("No files specified")
 
2300
        else:
 
2301
            file_list = []
 
2302
        
 
2303
        tree, file_list = tree_files(file_list)
 
2304
        if revision is None:
 
2305
            # FIXME should be tree.last_revision
 
2306
            rev_id = tree.last_revision()
 
2307
        elif len(revision) != 1:
 
2308
            raise BzrCommandError('bzr revert --revision takes exactly 1 argument')
 
2309
        else:
 
2310
            rev_id = revision[0].in_history(tree.branch).rev_id
 
2311
        pb = ui.ui_factory.nested_progress_bar()
 
2312
        try:
 
2313
            tree.revert(file_list, 
 
2314
                        tree.branch.repository.revision_tree(rev_id),
 
2315
                        not no_backup, pb)
 
2316
        finally:
 
2317
            pb.finished()
 
2318
 
 
2319
 
 
2320
class cmd_assert_fail(Command):
 
2321
    """Test reporting of assertion failures"""
 
2322
    hidden = True
 
2323
    def run(self):
 
2324
        assert False, "always fails"
 
2325
 
 
2326
 
 
2327
class cmd_help(Command):
 
2328
    """Show help on a command or other topic.
 
2329
 
 
2330
    For a list of all available commands, say 'bzr help commands'."""
 
2331
    takes_options = [Option('long', 'show help on all commands')]
 
2332
    takes_args = ['topic?']
 
2333
    aliases = ['?', '--help', '-?', '-h']
 
2334
    
 
2335
    @display_command
 
2336
    def run(self, topic=None, long=False):
 
2337
        import help
 
2338
        if topic is None and long:
 
2339
            topic = "commands"
 
2340
        help.help(topic)
 
2341
 
 
2342
 
 
2343
class cmd_shell_complete(Command):
 
2344
    """Show appropriate completions for context.
 
2345
 
 
2346
    For a list of all available commands, say 'bzr shell-complete'."""
 
2347
    takes_args = ['context?']
 
2348
    aliases = ['s-c']
 
2349
    hidden = True
 
2350
    
 
2351
    @display_command
 
2352
    def run(self, context=None):
 
2353
        import shellcomplete
 
2354
        shellcomplete.shellcomplete(context)
 
2355
 
 
2356
 
 
2357
class cmd_fetch(Command):
 
2358
    """Copy in history from another branch but don't merge it.
 
2359
 
 
2360
    This is an internal method used for pull and merge."""
 
2361
    hidden = True
 
2362
    takes_args = ['from_branch', 'to_branch']
 
2363
    def run(self, from_branch, to_branch):
 
2364
        from bzrlib.fetch import Fetcher
 
2365
        from_b = Branch.open(from_branch)
 
2366
        to_b = Branch.open(to_branch)
 
2367
        Fetcher(to_b, from_b)
 
2368
 
 
2369
 
 
2370
class cmd_missing(Command):
 
2371
    """Show unmerged/unpulled revisions between two branches.
 
2372
 
 
2373
    OTHER_BRANCH may be local or remote."""
 
2374
    takes_args = ['other_branch?']
 
2375
    takes_options = [Option('reverse', 'Reverse the order of revisions'),
 
2376
                     Option('mine-only', 
 
2377
                            'Display changes in the local branch only'),
 
2378
                     Option('theirs-only', 
 
2379
                            'Display changes in the remote branch only'), 
 
2380
                     'log-format',
 
2381
                     'show-ids',
 
2382
                     'verbose'
 
2383
                     ]
 
2384
    encoding_type = 'replace'
 
2385
 
 
2386
    @display_command
 
2387
    def run(self, other_branch=None, reverse=False, mine_only=False,
 
2388
            theirs_only=False, log_format=None,
 
2389
            show_ids=False, verbose=False):
 
2390
        from bzrlib.missing import find_unmerged, iter_log_data
 
2391
        from bzrlib.log import log_formatter
 
2392
        local_branch = Branch.open_containing(u".")[0]
 
2393
        parent = local_branch.get_parent()
 
2394
        if other_branch is None:
 
2395
            other_branch = parent
 
2396
            if other_branch is None:
 
2397
                raise BzrCommandError("No peer location known or specified.")
 
2398
            print "Using last location: " + local_branch.get_parent()
 
2399
        remote_branch = Branch.open(other_branch)
 
2400
        if remote_branch.base == local_branch.base:
 
2401
            remote_branch = local_branch
 
2402
        local_branch.lock_read()
 
2403
        try:
 
2404
            remote_branch.lock_read()
 
2405
            try:
 
2406
                local_extra, remote_extra = find_unmerged(local_branch, remote_branch)
 
2407
                if (log_format is None):
 
2408
                    log_format = local_branch.get_config().log_format()
 
2409
                lf = log_formatter(log_format,
 
2410
                                   to_file=self.outf,
 
2411
                                   show_ids=show_ids,
 
2412
                                   show_timezone='original')
 
2413
                if reverse is False:
 
2414
                    local_extra.reverse()
 
2415
                    remote_extra.reverse()
 
2416
                if local_extra and not theirs_only:
 
2417
                    print "You have %d extra revision(s):" % len(local_extra)
 
2418
                    for data in iter_log_data(local_extra, local_branch.repository,
 
2419
                                              verbose):
 
2420
                        lf.show(*data)
 
2421
                    printed_local = True
 
2422
                else:
 
2423
                    printed_local = False
 
2424
                if remote_extra and not mine_only:
 
2425
                    if printed_local is True:
 
2426
                        print "\n\n"
 
2427
                    print "You are missing %d revision(s):" % len(remote_extra)
 
2428
                    for data in iter_log_data(remote_extra, remote_branch.repository, 
 
2429
                                              verbose):
 
2430
                        lf.show(*data)
 
2431
                if not remote_extra and not local_extra:
 
2432
                    status_code = 0
 
2433
                    print "Branches are up to date."
 
2434
                else:
 
2435
                    status_code = 1
 
2436
            finally:
 
2437
                remote_branch.unlock()
 
2438
        finally:
 
2439
            local_branch.unlock()
 
2440
        if not status_code and parent is None and other_branch is not None:
 
2441
            local_branch.lock_write()
 
2442
            try:
 
2443
                # handle race conditions - a parent might be set while we run.
 
2444
                if local_branch.get_parent() is None:
 
2445
                    local_branch.set_parent(remote_branch.base)
 
2446
            finally:
 
2447
                local_branch.unlock()
 
2448
        return status_code
 
2449
 
 
2450
 
 
2451
class cmd_plugins(Command):
 
2452
    """List plugins"""
 
2453
    hidden = True
 
2454
    @display_command
 
2455
    def run(self):
 
2456
        import bzrlib.plugin
 
2457
        from inspect import getdoc
 
2458
        for name, plugin in bzrlib.plugin.all_plugins().items():
 
2459
            if hasattr(plugin, '__path__'):
 
2460
                print plugin.__path__[0]
 
2461
            elif hasattr(plugin, '__file__'):
 
2462
                print plugin.__file__
 
2463
            else:
 
2464
                print `plugin`
 
2465
                
 
2466
            d = getdoc(plugin)
 
2467
            if d:
 
2468
                print '\t', d.split('\n')[0]
 
2469
 
 
2470
 
 
2471
class cmd_testament(Command):
 
2472
    """Show testament (signing-form) of a revision."""
 
2473
    takes_options = ['revision', 
 
2474
                     Option('long', help='Produce long-format testament'), 
 
2475
                     Option('strict', help='Produce a strict-format'
 
2476
                            ' testament')]
 
2477
    takes_args = ['branch?']
 
2478
    @display_command
 
2479
    def run(self, branch=u'.', revision=None, long=False, strict=False):
 
2480
        from bzrlib.testament import Testament, StrictTestament
 
2481
        if strict is True:
 
2482
            testament_class = StrictTestament
 
2483
        else:
 
2484
            testament_class = Testament
 
2485
        b = WorkingTree.open_containing(branch)[0].branch
 
2486
        b.lock_read()
 
2487
        try:
 
2488
            if revision is None:
 
2489
                rev_id = b.last_revision()
 
2490
            else:
 
2491
                rev_id = revision[0].in_history(b).rev_id
 
2492
            t = testament_class.from_revision(b.repository, rev_id)
 
2493
            if long:
 
2494
                sys.stdout.writelines(t.as_text_lines())
 
2495
            else:
 
2496
                sys.stdout.write(t.as_short_text())
 
2497
        finally:
 
2498
            b.unlock()
 
2499
 
 
2500
 
 
2501
class cmd_annotate(Command):
 
2502
    """Show the origin of each line in a file.
 
2503
 
 
2504
    This prints out the given file with an annotation on the left side
 
2505
    indicating which revision, author and date introduced the change.
 
2506
 
 
2507
    If the origin is the same for a run of consecutive lines, it is 
 
2508
    shown only at the top, unless the --all option is given.
 
2509
    """
 
2510
    # TODO: annotate directories; showing when each file was last changed
 
2511
    # TODO: if the working copy is modified, show annotations on that 
 
2512
    #       with new uncommitted lines marked
 
2513
    aliases = ['ann', 'blame', 'praise']
 
2514
    takes_args = ['filename']
 
2515
    takes_options = [Option('all', help='show annotations on all lines'),
 
2516
                     Option('long', help='show date in annotations'),
 
2517
                     'revision'
 
2518
                     ]
 
2519
 
 
2520
    @display_command
 
2521
    def run(self, filename, all=False, long=False, revision=None):
 
2522
        from bzrlib.annotate import annotate_file
 
2523
        tree, relpath = WorkingTree.open_containing(filename)
 
2524
        branch = tree.branch
 
2525
        branch.lock_read()
 
2526
        try:
 
2527
            if revision is None:
 
2528
                revision_id = branch.last_revision()
 
2529
            elif len(revision) != 1:
 
2530
                raise BzrCommandError('bzr annotate --revision takes exactly 1 argument')
 
2531
            else:
 
2532
                revision_id = revision[0].in_history(branch).rev_id
 
2533
            file_id = tree.inventory.path2id(relpath)
 
2534
            tree = branch.repository.revision_tree(revision_id)
 
2535
            file_version = tree.inventory[file_id].revision
 
2536
            annotate_file(branch, file_version, file_id, long, all, sys.stdout)
 
2537
        finally:
 
2538
            branch.unlock()
 
2539
 
 
2540
 
 
2541
class cmd_re_sign(Command):
 
2542
    """Create a digital signature for an existing revision."""
 
2543
    # TODO be able to replace existing ones.
 
2544
 
 
2545
    hidden = True # is this right ?
 
2546
    takes_args = ['revision_id*']
 
2547
    takes_options = ['revision']
 
2548
    
 
2549
    def run(self, revision_id_list=None, revision=None):
 
2550
        import bzrlib.gpg as gpg
 
2551
        if revision_id_list is not None and revision is not None:
 
2552
            raise BzrCommandError('You can only supply one of revision_id or --revision')
 
2553
        if revision_id_list is None and revision is None:
 
2554
            raise BzrCommandError('You must supply either --revision or a revision_id')
 
2555
        b = WorkingTree.open_containing(u'.')[0].branch
 
2556
        gpg_strategy = gpg.GPGStrategy(b.get_config())
 
2557
        if revision_id_list is not None:
 
2558
            for revision_id in revision_id_list:
 
2559
                b.repository.sign_revision(revision_id, gpg_strategy)
 
2560
        elif revision is not None:
 
2561
            if len(revision) == 1:
 
2562
                revno, rev_id = revision[0].in_history(b)
 
2563
                b.repository.sign_revision(rev_id, gpg_strategy)
 
2564
            elif len(revision) == 2:
 
2565
                # are they both on rh- if so we can walk between them
 
2566
                # might be nice to have a range helper for arbitrary
 
2567
                # revision paths. hmm.
 
2568
                from_revno, from_revid = revision[0].in_history(b)
 
2569
                to_revno, to_revid = revision[1].in_history(b)
 
2570
                if to_revid is None:
 
2571
                    to_revno = b.revno()
 
2572
                if from_revno is None or to_revno is None:
 
2573
                    raise BzrCommandError('Cannot sign a range of non-revision-history revisions')
 
2574
                for revno in range(from_revno, to_revno + 1):
 
2575
                    b.repository.sign_revision(b.get_rev_id(revno), 
 
2576
                                               gpg_strategy)
 
2577
            else:
 
2578
                raise BzrCommandError('Please supply either one revision, or a range.')
 
2579
 
 
2580
 
 
2581
class cmd_bind(Command):
 
2582
    """Bind the current branch to a master branch.
 
2583
 
 
2584
    After binding, commits must succeed on the master branch
 
2585
    before they are executed on the local one.
 
2586
    """
 
2587
 
 
2588
    takes_args = ['location']
 
2589
    takes_options = []
 
2590
 
 
2591
    def run(self, location=None):
 
2592
        b, relpath = Branch.open_containing(u'.')
 
2593
        b_other = Branch.open(location)
 
2594
        try:
 
2595
            b.bind(b_other)
 
2596
        except DivergedBranches:
 
2597
            raise BzrCommandError('These branches have diverged.'
 
2598
                                  ' Try merging, and then bind again.')
 
2599
 
 
2600
 
 
2601
class cmd_unbind(Command):
 
2602
    """Unbind the current branch from its master branch.
 
2603
 
 
2604
    After unbinding, the local branch is considered independent.
 
2605
    All subsequent commits will be local.
 
2606
    """
 
2607
 
 
2608
    takes_args = []
 
2609
    takes_options = []
 
2610
 
 
2611
    def run(self):
 
2612
        b, relpath = Branch.open_containing(u'.')
 
2613
        if not b.unbind():
 
2614
            raise BzrCommandError('Local branch is not bound')
 
2615
 
 
2616
 
 
2617
class cmd_uncommit(Command):
 
2618
    """Remove the last committed revision.
 
2619
 
 
2620
    --verbose will print out what is being removed.
 
2621
    --dry-run will go through all the motions, but not actually
 
2622
    remove anything.
 
2623
    
 
2624
    In the future, uncommit will create a revision bundle, which can then
 
2625
    be re-applied.
 
2626
    """
 
2627
 
 
2628
    # TODO: jam 20060108 Add an option to allow uncommit to remove
 
2629
    # unreferenced information in 'branch-as-repository' branches.
 
2630
    # TODO: jam 20060108 Add the ability for uncommit to remove unreferenced
 
2631
    # information in shared branches as well.
 
2632
    takes_options = ['verbose', 'revision',
 
2633
                    Option('dry-run', help='Don\'t actually make changes'),
 
2634
                    Option('force', help='Say yes to all questions.')]
 
2635
    takes_args = ['location?']
 
2636
    aliases = []
 
2637
 
 
2638
    def run(self, location=None,
 
2639
            dry_run=False, verbose=False,
 
2640
            revision=None, force=False):
 
2641
        from bzrlib.log import log_formatter, show_log
 
2642
        import sys
 
2643
        from bzrlib.uncommit import uncommit
 
2644
 
 
2645
        if location is None:
 
2646
            location = u'.'
 
2647
        control, relpath = bzrdir.BzrDir.open_containing(location)
 
2648
        try:
 
2649
            tree = control.open_workingtree()
 
2650
            b = tree.branch
 
2651
        except (errors.NoWorkingTree, errors.NotLocalUrl):
 
2652
            tree = None
 
2653
            b = control.open_branch()
 
2654
 
 
2655
        rev_id = None
 
2656
        if revision is None:
 
2657
            revno = b.revno()
 
2658
        else:
 
2659
            # 'bzr uncommit -r 10' actually means uncommit
 
2660
            # so that the final tree is at revno 10.
 
2661
            # but bzrlib.uncommit.uncommit() actually uncommits
 
2662
            # the revisions that are supplied.
 
2663
            # So we need to offset it by one
 
2664
            revno = revision[0].in_history(b).revno+1
 
2665
 
 
2666
        if revno <= b.revno():
 
2667
            rev_id = b.get_rev_id(revno)
 
2668
        if rev_id is None:
 
2669
            self.outf.write('No revisions to uncommit.\n')
 
2670
            return 1
 
2671
 
 
2672
        lf = log_formatter('short',
 
2673
                           to_file=self.outf,
 
2674
                           show_timezone='original')
 
2675
 
 
2676
        show_log(b,
 
2677
                 lf,
 
2678
                 verbose=False,
 
2679
                 direction='forward',
 
2680
                 start_revision=revno,
 
2681
                 end_revision=b.revno())
 
2682
 
 
2683
        if dry_run:
 
2684
            print 'Dry-run, pretending to remove the above revisions.'
 
2685
            if not force:
 
2686
                val = raw_input('Press <enter> to continue')
 
2687
        else:
 
2688
            print 'The above revision(s) will be removed.'
 
2689
            if not force:
 
2690
                val = raw_input('Are you sure [y/N]? ')
 
2691
                if val.lower() not in ('y', 'yes'):
 
2692
                    print 'Canceled'
 
2693
                    return 0
 
2694
 
 
2695
        uncommit(b, tree=tree, dry_run=dry_run, verbose=verbose,
 
2696
                revno=revno)
 
2697
 
 
2698
 
 
2699
class cmd_break_lock(Command):
 
2700
    """Break a dead lock on a repository, branch or working directory.
 
2701
 
 
2702
    CAUTION: Locks should only be broken when you are sure that the process
 
2703
    holding the lock has been stopped.
 
2704
 
 
2705
    You can get information on what locks are open via the 'bzr info' command.
 
2706
    
 
2707
    example:
 
2708
        bzr break-lock
 
2709
    """
 
2710
    takes_args = ['location?']
 
2711
 
 
2712
    def run(self, location=None, show=False):
 
2713
        if location is None:
 
2714
            location = u'.'
 
2715
        control, relpath = bzrdir.BzrDir.open_containing(location)
 
2716
        try:
 
2717
            control.break_lock()
 
2718
        except NotImplementedError:
 
2719
            pass
 
2720
        
 
2721
 
 
2722
 
 
2723
# command-line interpretation helper for merge-related commands
 
2724
def merge(other_revision, base_revision,
 
2725
          check_clean=True, ignore_zero=False,
 
2726
          this_dir=None, backup_files=False, merge_type=Merge3Merger,
 
2727
          file_list=None, show_base=False, reprocess=False,
 
2728
          pb=DummyProgress()):
 
2729
    """Merge changes into a tree.
 
2730
 
 
2731
    base_revision
 
2732
        list(path, revno) Base for three-way merge.  
 
2733
        If [None, None] then a base will be automatically determined.
 
2734
    other_revision
 
2735
        list(path, revno) Other revision for three-way merge.
 
2736
    this_dir
 
2737
        Directory to merge changes into; '.' by default.
 
2738
    check_clean
 
2739
        If true, this_dir must have no uncommitted changes before the
 
2740
        merge begins.
 
2741
    ignore_zero - If true, suppress the "zero conflicts" message when 
 
2742
        there are no conflicts; should be set when doing something we expect
 
2743
        to complete perfectly.
 
2744
    file_list - If supplied, merge only changes to selected files.
 
2745
 
 
2746
    All available ancestors of other_revision and base_revision are
 
2747
    automatically pulled into the branch.
 
2748
 
 
2749
    The revno may be -1 to indicate the last revision on the branch, which is
 
2750
    the typical case.
 
2751
 
 
2752
    This function is intended for use from the command line; programmatic
 
2753
    clients might prefer to call merge.merge_inner(), which has less magic 
 
2754
    behavior.
 
2755
    """
 
2756
    from bzrlib.merge import Merger
 
2757
    if this_dir is None:
 
2758
        this_dir = u'.'
 
2759
    this_tree = WorkingTree.open_containing(this_dir)[0]
 
2760
    if show_base and not merge_type is Merge3Merger:
 
2761
        raise BzrCommandError("Show-base is not supported for this merge"
 
2762
                              " type. %s" % merge_type)
 
2763
    if reprocess and not merge_type.supports_reprocess:
 
2764
        raise BzrCommandError("Conflict reduction is not supported for merge"
 
2765
                              " type %s." % merge_type)
 
2766
    if reprocess and show_base:
 
2767
        raise BzrCommandError("Cannot do conflict reduction and show base.")
 
2768
    try:
 
2769
        merger = Merger(this_tree.branch, this_tree=this_tree, pb=pb)
 
2770
        merger.pp = ProgressPhase("Merge phase", 5, pb)
 
2771
        merger.pp.next_phase()
 
2772
        merger.check_basis(check_clean)
 
2773
        merger.set_other(other_revision)
 
2774
        merger.pp.next_phase()
 
2775
        merger.set_base(base_revision)
 
2776
        if merger.base_rev_id == merger.other_rev_id:
 
2777
            note('Nothing to do.')
 
2778
            return 0
 
2779
        merger.backup_files = backup_files
 
2780
        merger.merge_type = merge_type 
 
2781
        merger.set_interesting_files(file_list)
 
2782
        merger.show_base = show_base 
 
2783
        merger.reprocess = reprocess
 
2784
        conflicts = merger.do_merge()
 
2785
        if file_list is None:
 
2786
            merger.set_pending()
 
2787
    finally:
 
2788
        pb.clear()
 
2789
    return conflicts
 
2790
 
 
2791
 
 
2792
# these get imported and then picked up by the scan for cmd_*
 
2793
# TODO: Some more consistent way to split command definitions across files;
 
2794
# we do need to load at least some information about them to know of 
 
2795
# aliases.  ideally we would avoid loading the implementation until the
 
2796
# details were needed.
 
2797
from bzrlib.conflicts import cmd_resolve, cmd_conflicts, restore
 
2798
from bzrlib.bundle.commands import cmd_bundle_revisions
 
2799
from bzrlib.sign_my_commits import cmd_sign_my_commits
 
2800
from bzrlib.weave_commands import cmd_weave_list, cmd_weave_join, \
 
2801
        cmd_weave_plan_merge, cmd_weave_merge_text