~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/log.py

  • Committer: Martin Pool
  • Date: 2005-05-17 07:51:51 UTC
  • Revision ID: mbp@sourcefrog.net-20050517075151-b64853a6a38a6225
- export bzrlib.find_branch

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007, 2009 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2005 Canonical Ltd
 
2
 
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
#
 
12
 
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
 
18
 
 
19
 
"""Code to show logs of changes.
20
 
 
21
 
Various flavors of log can be produced:
22
 
 
23
 
* for one file, or the whole tree, and (not done yet) for
24
 
  files in a given directory
25
 
 
26
 
* in "verbose" mode with a description of what changed from one
27
 
  version to the next
28
 
 
29
 
* with file-ids and revision-ids shown
30
 
 
31
 
Logs are actually written out through an abstract LogFormatter
32
 
interface, which allows for different preferred formats.  Plugins can
33
 
register formats too.
34
 
 
35
 
Logs can be produced in either forward (oldest->newest) or reverse
36
 
(newest->oldest) order.
37
 
 
38
 
Logs can be filtered to show only revisions matching a particular
39
 
search string, or within a particular range of revisions.  The range
40
 
can be given as date/times, which are reduced to revisions before
41
 
calling in here.
42
 
 
43
 
In verbose mode we show a summary of what changed in each particular
44
 
revision.  Note that this is the delta for changes in that revision
45
 
relative to its left-most parent, not the delta relative to the last
46
 
logged revision.  So for example if you ask for a verbose log of
47
 
changes touching hello.c you will get a list of those revisions also
48
 
listing other things that were changed in the same revision, but not
49
 
all the changes since the previous revision that touched hello.c.
50
 
"""
51
 
 
52
 
import codecs
53
 
from cStringIO import StringIO
54
 
from itertools import (
55
 
    chain,
56
 
    izip,
57
 
    )
58
 
import re
59
 
import sys
60
 
from warnings import (
61
 
    warn,
62
 
    )
63
 
 
64
 
from bzrlib.lazy_import import lazy_import
65
 
lazy_import(globals(), """
66
 
 
67
 
from bzrlib import (
68
 
    bzrdir,
69
 
    config,
70
 
    diff,
71
 
    errors,
72
 
    repository as _mod_repository,
73
 
    revision as _mod_revision,
74
 
    revisionspec,
75
 
    trace,
76
 
    tsort,
77
 
    )
78
 
""")
79
 
 
80
 
from bzrlib import (
81
 
    registry,
82
 
    )
83
 
from bzrlib.osutils import (
84
 
    format_date,
85
 
    get_terminal_encoding,
86
 
    re_compile_checked,
87
 
    terminal_width,
88
 
    )
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
89
18
 
90
19
 
91
20
def find_touching_revisions(branch, file_id):
103
32
    last_path = None
104
33
    revno = 1
105
34
    for revision_id in branch.revision_history():
106
 
        this_inv = branch.repository.get_revision_inventory(revision_id)
 
35
        this_inv = branch.get_revision_inventory(revision_id)
107
36
        if file_id in this_inv:
108
37
            this_ie = this_inv[file_id]
109
38
            this_path = this_inv.id2path(file_id)
132
61
        revno += 1
133
62
 
134
63
 
135
 
def _enumerate_history(branch):
136
 
    rh = []
137
 
    revno = 1
138
 
    for rev_id in branch.revision_history():
139
 
        rh.append((revno, rev_id))
140
 
        revno += 1
141
 
    return rh
142
 
 
143
 
 
144
64
def show_log(branch,
145
 
             lf,
146
 
             specific_fileid=None,
 
65
             filename=None,
 
66
             show_timezone='original',
147
67
             verbose=False,
148
 
             direction='reverse',
149
 
             start_revision=None,
150
 
             end_revision=None,
151
 
             search=None,
152
 
             limit=None,
153
 
             show_diff=False):
 
68
             show_ids=False,
 
69
             to_file=None):
154
70
    """Write out human-readable log of commits to this branch.
155
71
 
156
 
    This function is being retained for backwards compatibility but
157
 
    should not be extended with new parameters. Use the new Logger class
158
 
    instead, eg. Logger(branch, rqst).show(lf), adding parameters to the
159
 
    make_log_request_dict function.
160
 
 
161
 
    :param lf: The LogFormatter object showing the output.
162
 
 
163
 
    :param specific_fileid: If not None, list only the commits affecting the
164
 
        specified file, rather than all commits.
165
 
 
166
 
    :param verbose: If True show added/changed/deleted/renamed files.
167
 
 
168
 
    :param direction: 'reverse' (default) is latest to earliest; 'forward' is
169
 
        earliest to latest.
170
 
 
171
 
    :param start_revision: If not None, only show revisions >= start_revision
172
 
 
173
 
    :param end_revision: If not None, only show revisions <= end_revision
174
 
 
175
 
    :param search: If not None, only show revisions with matching commit
176
 
        messages
177
 
 
178
 
    :param limit: If set, shows only 'limit' revisions, all revisions are shown
179
 
        if None or 0.
180
 
 
181
 
    :param show_diff: If True, output a diff after each revision.
 
72
    filename
 
73
        If true, list only the commits affecting the specified
 
74
        file, rather than all commits.
 
75
 
 
76
    show_timezone
 
77
        'original' (committer's timezone),
 
78
        'utc' (universal time), or
 
79
        'local' (local user's timezone)
 
80
 
 
81
    verbose
 
82
        If true show added/changed/deleted/renamed files.
 
83
 
 
84
    show_ids
 
85
        If true, show revision and file ids.
 
86
 
 
87
    to_file
 
88
        File to send log to; by default stdout.
182
89
    """
183
 
    # Convert old-style parameters to new-style parameters
184
 
    if specific_fileid is not None:
185
 
        file_ids = [specific_fileid]
 
90
    from osutils import format_date
 
91
    from errors import BzrCheckError
 
92
    from diff import compare_trees
 
93
    from textui import show_status
 
94
 
 
95
    if to_file == None:
 
96
        import sys
 
97
        to_file = sys.stdout
 
98
 
 
99
    if filename:
 
100
        file_id = branch.read_working_inventory().path2id(filename)
 
101
        def which_revs():
 
102
            for revno, revid, why in find_touching_revisions(branch, file_id):
 
103
                yield revno, revid
186
104
    else:
187
 
        file_ids = None
 
105
        def which_revs():
 
106
            for i, revid in enumerate(branch.revision_history()):
 
107
                yield i+1, revid
 
108
        
 
109
    branch._need_readlock()
 
110
    precursor = None
188
111
    if verbose:
189
 
        if file_ids:
190
 
            delta_type = 'partial'
191
 
        else:
192
 
            delta_type = 'full'
193
 
    else:
194
 
        delta_type = None
195
 
    if show_diff:
196
 
        if file_ids:
197
 
            diff_type = 'partial'
198
 
        else:
199
 
            diff_type = 'full'
200
 
    else:
201
 
        diff_type = None
202
 
 
203
 
    # Build the request and execute it
204
 
    rqst = make_log_request_dict(direction=direction, specific_fileids=file_ids,
205
 
        start_revision=start_revision, end_revision=end_revision,
206
 
        limit=limit, message_search=search,
207
 
        delta_type=delta_type, diff_type=diff_type)
208
 
    Logger(branch, rqst).show(lf)
209
 
 
210
 
 
211
 
# Note: This needs to be kept this in sync with the defaults in
212
 
# make_log_request_dict() below
213
 
_DEFAULT_REQUEST_PARAMS = {
214
 
    'direction': 'reverse',
215
 
    'levels': 1,
216
 
    'generate_tags': True,
217
 
    '_match_using_deltas': True,
218
 
    }
219
 
 
220
 
 
221
 
def make_log_request_dict(direction='reverse', specific_fileids=None,
222
 
    start_revision=None, end_revision=None, limit=None,
223
 
    message_search=None, levels=1, generate_tags=True, delta_type=None,
224
 
    diff_type=None, _match_using_deltas=True):
225
 
    """Convenience function for making a logging request dictionary.
226
 
 
227
 
    Using this function may make code slightly safer by ensuring
228
 
    parameters have the correct names. It also provides a reference
229
 
    point for documenting the supported parameters.
230
 
 
231
 
    :param direction: 'reverse' (default) is latest to earliest;
232
 
      'forward' is earliest to latest.
233
 
 
234
 
    :param specific_fileids: If not None, only include revisions
235
 
      affecting the specified files, rather than all revisions.
236
 
 
237
 
    :param start_revision: If not None, only generate
238
 
      revisions >= start_revision
239
 
 
240
 
    :param end_revision: If not None, only generate
241
 
      revisions <= end_revision
242
 
 
243
 
    :param limit: If set, generate only 'limit' revisions, all revisions
244
 
      are shown if None or 0.
245
 
 
246
 
    :param message_search: If not None, only include revisions with
247
 
      matching commit messages
248
 
 
249
 
    :param levels: the number of levels of revisions to
250
 
      generate; 1 for just the mainline; 0 for all levels.
251
 
 
252
 
    :param generate_tags: If True, include tags for matched revisions.
253
 
 
254
 
    :param delta_type: Either 'full', 'partial' or None.
255
 
      'full' means generate the complete delta - adds/deletes/modifies/etc;
256
 
      'partial' means filter the delta using specific_fileids;
257
 
      None means do not generate any delta.
258
 
 
259
 
    :param diff_type: Either 'full', 'partial' or None.
260
 
      'full' means generate the complete diff - adds/deletes/modifies/etc;
261
 
      'partial' means filter the diff using specific_fileids;
262
 
      None means do not generate any diff.
263
 
 
264
 
    :param _match_using_deltas: a private parameter controlling the
265
 
      algorithm used for matching specific_fileids. This parameter
266
 
      may be removed in the future so bzrlib client code should NOT
267
 
      use it.
268
 
    """
269
 
    return {
270
 
        'direction': direction,
271
 
        'specific_fileids': specific_fileids,
272
 
        'start_revision': start_revision,
273
 
        'end_revision': end_revision,
274
 
        'limit': limit,
275
 
        'message_search': message_search,
276
 
        'levels': levels,
277
 
        'generate_tags': generate_tags,
278
 
        'delta_type': delta_type,
279
 
        'diff_type': diff_type,
280
 
        # Add 'private' attributes for features that may be deprecated
281
 
        '_match_using_deltas': _match_using_deltas,
282
 
    }
283
 
 
284
 
 
285
 
def _apply_log_request_defaults(rqst):
286
 
    """Apply default values to a request dictionary."""
287
 
    result = _DEFAULT_REQUEST_PARAMS
288
 
    if rqst:
289
 
        result.update(rqst)
290
 
    return result
291
 
 
292
 
 
293
 
class LogGenerator(object):
294
 
    """A generator of log revisions."""
295
 
 
296
 
    def iter_log_revisions(self):
297
 
        """Iterate over LogRevision objects.
298
 
 
299
 
        :return: An iterator yielding LogRevision objects.
300
 
        """
301
 
        raise NotImplementedError(self.iter_log_revisions)
302
 
 
303
 
 
304
 
class Logger(object):
305
 
    """An object the generates, formats and displays a log."""
306
 
 
307
 
    def __init__(self, branch, rqst):
308
 
        """Create a Logger.
309
 
 
310
 
        :param branch: the branch to log
311
 
        :param rqst: A dictionary specifying the query parameters.
312
 
          See make_log_request_dict() for supported values.
313
 
        """
314
 
        self.branch = branch
315
 
        self.rqst = _apply_log_request_defaults(rqst)
316
 
 
317
 
    def show(self, lf):
318
 
        """Display the log.
319
 
 
320
 
        :param lf: The LogFormatter object to send the output to.
321
 
        """
322
 
        if not isinstance(lf, LogFormatter):
323
 
            warn("not a LogFormatter instance: %r" % lf)
324
 
 
325
 
        self.branch.lock_read()
326
 
        try:
327
 
            if getattr(lf, 'begin_log', None):
328
 
                lf.begin_log()
329
 
            self._show_body(lf)
330
 
            if getattr(lf, 'end_log', None):
331
 
                lf.end_log()
332
 
        finally:
333
 
            self.branch.unlock()
334
 
 
335
 
    def _show_body(self, lf):
336
 
        """Show the main log output.
337
 
 
338
 
        Subclasses may wish to override this.
339
 
        """
340
 
        # Tweak the LogRequest based on what the LogFormatter can handle.
341
 
        # (There's no point generating stuff if the formatter can't display it.)
342
 
        rqst = self.rqst
343
 
        rqst['levels'] = lf.get_levels()
344
 
        if not getattr(lf, 'supports_tags', False):
345
 
            rqst['generate_tags'] = False
346
 
        if not getattr(lf, 'supports_delta', False):
347
 
            rqst['delta_type'] = None
348
 
        if not getattr(lf, 'supports_diff', False):
349
 
            rqst['diff_type'] = None
350
 
 
351
 
        # Find and print the interesting revisions
352
 
        generator = self._generator_factory(self.branch, rqst)
353
 
        for lr in generator.iter_log_revisions():
354
 
            lf.log_revision(lr)
355
 
        lf.show_advice()
356
 
 
357
 
    def _generator_factory(self, branch, rqst):
358
 
        """Make the LogGenerator object to use.
359
 
        
360
 
        Subclasses may wish to override this.
361
 
        """
362
 
        return _DefaultLogGenerator(branch, rqst)
363
 
 
364
 
 
365
 
class _StartNotLinearAncestor(Exception):
366
 
    """Raised when a start revision is not found walking left-hand history."""
367
 
 
368
 
 
369
 
class _DefaultLogGenerator(LogGenerator):
370
 
    """The default generator of log revisions."""
371
 
 
372
 
    def __init__(self, branch, rqst):
373
 
        self.branch = branch
374
 
        self.rqst = rqst
375
 
        if rqst.get('generate_tags') and branch.supports_tags():
376
 
            self.rev_tag_dict = branch.tags.get_reverse_tag_dict()
377
 
        else:
378
 
            self.rev_tag_dict = {}
379
 
 
380
 
    def iter_log_revisions(self):
381
 
        """Iterate over LogRevision objects.
382
 
 
383
 
        :return: An iterator yielding LogRevision objects.
384
 
        """
385
 
        rqst = self.rqst
386
 
        log_count = 0
387
 
        revision_iterator = self._create_log_revision_iterator()
388
 
        for revs in revision_iterator:
389
 
            for (rev_id, revno, merge_depth), rev, delta in revs:
390
 
                # 0 levels means show everything; merge_depth counts from 0
391
 
                levels = rqst.get('levels')
392
 
                if levels != 0 and merge_depth >= levels:
393
 
                    continue
394
 
                diff = self._format_diff(rev, rev_id)
395
 
                yield LogRevision(rev, revno, merge_depth, delta,
396
 
                    self.rev_tag_dict.get(rev_id), diff)
397
 
                limit = rqst.get('limit')
398
 
                if limit:
399
 
                    log_count += 1
400
 
                    if log_count >= limit:
401
 
                        return
402
 
 
403
 
    def _format_diff(self, rev, rev_id):
404
 
        diff_type = self.rqst.get('diff_type')
405
 
        if diff_type is None:
406
 
            return None
407
 
        repo = self.branch.repository
408
 
        if len(rev.parent_ids) == 0:
409
 
            ancestor_id = _mod_revision.NULL_REVISION
410
 
        else:
411
 
            ancestor_id = rev.parent_ids[0]
412
 
        tree_1 = repo.revision_tree(ancestor_id)
413
 
        tree_2 = repo.revision_tree(rev_id)
414
 
        file_ids = self.rqst.get('specific_fileids')
415
 
        if diff_type == 'partial' and file_ids is not None:
416
 
            specific_files = [tree_2.id2path(id) for id in file_ids]
417
 
        else:
418
 
            specific_files = None
419
 
        s = StringIO()
420
 
        diff.show_diff_trees(tree_1, tree_2, s, specific_files, old_label='',
421
 
            new_label='')
422
 
        return s.getvalue()
423
 
 
424
 
    def _create_log_revision_iterator(self):
425
 
        """Create a revision iterator for log.
426
 
 
427
 
        :return: An iterator over lists of ((rev_id, revno, merge_depth), rev,
428
 
            delta).
429
 
        """
430
 
        self.start_rev_id, self.end_rev_id = _get_revision_limits(
431
 
            self.branch, self.rqst.get('start_revision'),
432
 
            self.rqst.get('end_revision'))
433
 
        if self.rqst.get('_match_using_deltas'):
434
 
            return self._log_revision_iterator_using_delta_matching()
435
 
        else:
436
 
            # We're using the per-file-graph algorithm. This scales really
437
 
            # well but only makes sense if there is a single file and it's
438
 
            # not a directory
439
 
            file_count = len(self.rqst.get('specific_fileids'))
440
 
            if file_count != 1:
441
 
                raise BzrError("illegal LogRequest: must match-using-deltas "
442
 
                    "when logging %d files" % file_count)
443
 
            return self._log_revision_iterator_using_per_file_graph()
444
 
 
445
 
    def _log_revision_iterator_using_delta_matching(self):
446
 
        # Get the base revisions, filtering by the revision range
447
 
        rqst = self.rqst
448
 
        generate_merge_revisions = rqst.get('levels') != 1
449
 
        delayed_graph_generation = not rqst.get('specific_fileids') and (
450
 
                rqst.get('limit') or self.start_rev_id or self.end_rev_id)
451
 
        view_revisions = _calc_view_revisions(self.branch, self.start_rev_id,
452
 
            self.end_rev_id, rqst.get('direction'), generate_merge_revisions,
453
 
            delayed_graph_generation=delayed_graph_generation)
454
 
 
455
 
        # Apply the other filters
456
 
        return make_log_rev_iterator(self.branch, view_revisions,
457
 
            rqst.get('delta_type'), rqst.get('message_search'),
458
 
            file_ids=rqst.get('specific_fileids'),
459
 
            direction=rqst.get('direction'))
460
 
 
461
 
    def _log_revision_iterator_using_per_file_graph(self):
462
 
        # Get the base revisions, filtering by the revision range.
463
 
        # Note that we always generate the merge revisions because
464
 
        # filter_revisions_touching_file_id() requires them ...
465
 
        rqst = self.rqst
466
 
        view_revisions = _calc_view_revisions(self.branch, self.start_rev_id,
467
 
            self.end_rev_id, rqst.get('direction'), True)
468
 
        if not isinstance(view_revisions, list):
469
 
            view_revisions = list(view_revisions)
470
 
        view_revisions = _filter_revisions_touching_file_id(self.branch,
471
 
            rqst.get('specific_fileids')[0], view_revisions,
472
 
            include_merges=rqst.get('levels') != 1)
473
 
        return make_log_rev_iterator(self.branch, view_revisions,
474
 
            rqst.get('delta_type'), rqst.get('message_search'))
475
 
 
476
 
 
477
 
def _calc_view_revisions(branch, start_rev_id, end_rev_id, direction,
478
 
    generate_merge_revisions, delayed_graph_generation=False):
479
 
    """Calculate the revisions to view.
480
 
 
481
 
    :return: An iterator of (revision_id, dotted_revno, merge_depth) tuples OR
482
 
             a list of the same tuples.
483
 
    """
484
 
    br_revno, br_rev_id = branch.last_revision_info()
485
 
    if br_revno == 0:
486
 
        return []
487
 
 
488
 
    # If a single revision is requested, check we can handle it
489
 
    generate_single_revision = (end_rev_id and start_rev_id == end_rev_id and
490
 
        (not generate_merge_revisions or not _has_merges(branch, end_rev_id)))
491
 
    if generate_single_revision:
492
 
        return _generate_one_revision(branch, end_rev_id, br_rev_id, br_revno)
493
 
 
494
 
    # If we only want to see linear revisions, we can iterate ...
495
 
    if not generate_merge_revisions:
496
 
        return _generate_flat_revisions(branch, start_rev_id, end_rev_id,
497
 
            direction)
498
 
    else:
499
 
        return _generate_all_revisions(branch, start_rev_id, end_rev_id,
500
 
            direction, delayed_graph_generation)
501
 
 
502
 
 
503
 
def _generate_one_revision(branch, rev_id, br_rev_id, br_revno):
504
 
    if rev_id == br_rev_id:
505
 
        # It's the tip
506
 
        return [(br_rev_id, br_revno, 0)]
507
 
    else:
508
 
        revno = branch.revision_id_to_dotted_revno(rev_id)
509
 
        revno_str = '.'.join(str(n) for n in revno)
510
 
        return [(rev_id, revno_str, 0)]
511
 
 
512
 
 
513
 
def _generate_flat_revisions(branch, start_rev_id, end_rev_id, direction):
514
 
    result = _linear_view_revisions(branch, start_rev_id, end_rev_id)
515
 
    # If a start limit was given and it's not obviously an
516
 
    # ancestor of the end limit, check it before outputting anything
517
 
    if direction == 'forward' or (start_rev_id
518
 
        and not _is_obvious_ancestor(branch, start_rev_id, end_rev_id)):
519
 
        try:
520
 
            result = list(result)
521
 
        except _StartNotLinearAncestor:
522
 
            raise errors.BzrCommandError('Start revision not found in'
523
 
                ' left-hand history of end revision.')
524
 
    if direction == 'forward':
525
 
        result = reversed(result)
526
 
    return result
527
 
 
528
 
 
529
 
def _generate_all_revisions(branch, start_rev_id, end_rev_id, direction,
530
 
    delayed_graph_generation):
531
 
    # On large trees, generating the merge graph can take 30-60 seconds
532
 
    # so we delay doing it until a merge is detected, incrementally
533
 
    # returning initial (non-merge) revisions while we can.
534
 
    initial_revisions = []
535
 
    if delayed_graph_generation:
536
 
        try:
537
 
            for rev_id, revno, depth in \
538
 
                _linear_view_revisions(branch, start_rev_id, end_rev_id):
539
 
                if _has_merges(branch, rev_id):
540
 
                    end_rev_id = rev_id
541
 
                    break
542
 
                else:
543
 
                    initial_revisions.append((rev_id, revno, depth))
544
 
            else:
545
 
                # No merged revisions found
546
 
                if direction == 'reverse':
547
 
                    return initial_revisions
548
 
                elif direction == 'forward':
549
 
                    return reversed(initial_revisions)
550
 
                else:
551
 
                    raise ValueError('invalid direction %r' % direction)
552
 
        except _StartNotLinearAncestor:
553
 
            # A merge was never detected so the lower revision limit can't
554
 
            # be nested down somewhere
555
 
            raise errors.BzrCommandError('Start revision not found in'
556
 
                ' history of end revision.')
557
 
 
558
 
    # A log including nested merges is required. If the direction is reverse,
559
 
    # we rebase the initial merge depths so that the development line is
560
 
    # shown naturally, i.e. just like it is for linear logging. We can easily
561
 
    # make forward the exact opposite display, but showing the merge revisions
562
 
    # indented at the end seems slightly nicer in that case.
563
 
    view_revisions = chain(iter(initial_revisions),
564
 
        _graph_view_revisions(branch, start_rev_id, end_rev_id,
565
 
        rebase_initial_depths=direction == 'reverse'))
566
 
    if direction == 'reverse':
567
 
        return view_revisions
568
 
    elif direction == 'forward':
569
 
        # Forward means oldest first, adjusting for depth.
570
 
        view_revisions = reverse_by_depth(list(view_revisions))
571
 
        return _rebase_merge_depth(view_revisions)
572
 
    else:
573
 
        raise ValueError('invalid direction %r' % direction)
574
 
 
575
 
 
576
 
def _has_merges(branch, rev_id):
577
 
    """Does a revision have multiple parents or not?"""
578
 
    parents = branch.repository.get_parent_map([rev_id]).get(rev_id, [])
579
 
    return len(parents) > 1
580
 
 
581
 
 
582
 
def _is_obvious_ancestor(branch, start_rev_id, end_rev_id):
583
 
    """Is start_rev_id an obvious ancestor of end_rev_id?"""
584
 
    if start_rev_id and end_rev_id:
585
 
        start_dotted = branch.revision_id_to_dotted_revno(start_rev_id)
586
 
        end_dotted = branch.revision_id_to_dotted_revno(end_rev_id)
587
 
        if len(start_dotted) == 1 and len(end_dotted) == 1:
588
 
            # both on mainline
589
 
            return start_dotted[0] <= end_dotted[0]
590
 
        elif (len(start_dotted) == 3 and len(end_dotted) == 3 and
591
 
            start_dotted[0:1] == end_dotted[0:1]):
592
 
            # both on same development line
593
 
            return start_dotted[2] <= end_dotted[2]
594
 
        else:
595
 
            # not obvious
596
 
            return False
597
 
    return True
598
 
 
599
 
 
600
 
def _linear_view_revisions(branch, start_rev_id, end_rev_id):
601
 
    """Calculate a sequence of revisions to view, newest to oldest.
602
 
 
603
 
    :param start_rev_id: the lower revision-id
604
 
    :param end_rev_id: the upper revision-id
605
 
    :return: An iterator of (revision_id, dotted_revno, merge_depth) tuples.
606
 
    :raises _StartNotLinearAncestor: if a start_rev_id is specified but
607
 
      is not found walking the left-hand history
608
 
    """
609
 
    br_revno, br_rev_id = branch.last_revision_info()
610
 
    repo = branch.repository
611
 
    if start_rev_id is None and end_rev_id is None:
612
 
        cur_revno = br_revno
613
 
        for revision_id in repo.iter_reverse_revision_history(br_rev_id):
614
 
            yield revision_id, str(cur_revno), 0
615
 
            cur_revno -= 1
616
 
    else:
617
 
        if end_rev_id is None:
618
 
            end_rev_id = br_rev_id
619
 
        found_start = start_rev_id is None
620
 
        for revision_id in repo.iter_reverse_revision_history(end_rev_id):
621
 
            revno = branch.revision_id_to_dotted_revno(revision_id)
622
 
            revno_str = '.'.join(str(n) for n in revno)
623
 
            if not found_start and revision_id == start_rev_id:
624
 
                yield revision_id, revno_str, 0
625
 
                found_start = True
626
 
                break
627
 
            else:
628
 
                yield revision_id, revno_str, 0
629
 
        else:
630
 
            if not found_start:
631
 
                raise _StartNotLinearAncestor()
632
 
 
633
 
 
634
 
def _graph_view_revisions(branch, start_rev_id, end_rev_id,
635
 
    rebase_initial_depths=True):
636
 
    """Calculate revisions to view including merges, newest to oldest.
637
 
 
638
 
    :param branch: the branch
639
 
    :param start_rev_id: the lower revision-id
640
 
    :param end_rev_id: the upper revision-id
641
 
    :param rebase_initial_depth: should depths be rebased until a mainline
642
 
      revision is found?
643
 
    :return: An iterator of (revision_id, dotted_revno, merge_depth) tuples.
644
 
    """
645
 
    view_revisions = branch.iter_merge_sorted_revisions(
646
 
        start_revision_id=end_rev_id, stop_revision_id=start_rev_id,
647
 
        stop_rule="with-merges")
648
 
    if not rebase_initial_depths:
649
 
        for (rev_id, merge_depth, revno, end_of_merge
650
 
             ) in view_revisions:
651
 
            yield rev_id, '.'.join(map(str, revno)), merge_depth
652
 
    else:
653
 
        # We're following a development line starting at a merged revision.
654
 
        # We need to adjust depths down by the initial depth until we find
655
 
        # a depth less than it. Then we use that depth as the adjustment.
656
 
        # If and when we reach the mainline, depth adjustment ends.
657
 
        depth_adjustment = None
658
 
        for (rev_id, merge_depth, revno, end_of_merge
659
 
             ) in view_revisions:
660
 
            if depth_adjustment is None:
661
 
                depth_adjustment = merge_depth
662
 
            if depth_adjustment:
663
 
                if merge_depth < depth_adjustment:
664
 
                    depth_adjustment = merge_depth
665
 
                merge_depth -= depth_adjustment
666
 
            yield rev_id, '.'.join(map(str, revno)), merge_depth
667
 
 
668
 
 
669
 
def calculate_view_revisions(branch, start_revision, end_revision, direction,
670
 
        specific_fileid, generate_merge_revisions):
671
 
    """Calculate the revisions to view.
672
 
 
673
 
    :return: An iterator of (revision_id, dotted_revno, merge_depth) tuples OR
674
 
             a list of the same tuples.
675
 
    """
676
 
    # This method is no longer called by the main code path.
677
 
    # It is retained for API compatibility and may be deprecated
678
 
    # soon. IGC 20090116
679
 
    start_rev_id, end_rev_id = _get_revision_limits(branch, start_revision,
680
 
        end_revision)
681
 
    view_revisions = list(_calc_view_revisions(branch, start_rev_id, end_rev_id,
682
 
        direction, generate_merge_revisions or specific_fileid))
683
 
    if specific_fileid:
684
 
        view_revisions = _filter_revisions_touching_file_id(branch,
685
 
            specific_fileid, view_revisions,
686
 
            include_merges=generate_merge_revisions)
687
 
    return _rebase_merge_depth(view_revisions)
688
 
 
689
 
 
690
 
def _rebase_merge_depth(view_revisions):
691
 
    """Adjust depths upwards so the top level is 0."""
692
 
    # If either the first or last revision have a merge_depth of 0, we're done
693
 
    if view_revisions and view_revisions[0][2] and view_revisions[-1][2]:
694
 
        min_depth = min([d for r,n,d in view_revisions])
695
 
        if min_depth != 0:
696
 
            view_revisions = [(r,n,d-min_depth) for r,n,d in view_revisions]
697
 
    return view_revisions
698
 
 
699
 
 
700
 
def make_log_rev_iterator(branch, view_revisions, generate_delta, search,
701
 
        file_ids=None, direction='reverse'):
702
 
    """Create a revision iterator for log.
703
 
 
704
 
    :param branch: The branch being logged.
705
 
    :param view_revisions: The revisions being viewed.
706
 
    :param generate_delta: Whether to generate a delta for each revision.
707
 
      Permitted values are None, 'full' and 'partial'.
708
 
    :param search: A user text search string.
709
 
    :param file_ids: If non empty, only revisions matching one or more of
710
 
      the file-ids are to be kept.
711
 
    :param direction: the direction in which view_revisions is sorted
712
 
    :return: An iterator over lists of ((rev_id, revno, merge_depth), rev,
713
 
        delta).
714
 
    """
715
 
    # Convert view_revisions into (view, None, None) groups to fit with
716
 
    # the standard interface here.
717
 
    if type(view_revisions) == list:
718
 
        # A single batch conversion is faster than many incremental ones.
719
 
        # As we have all the data, do a batch conversion.
720
 
        nones = [None] * len(view_revisions)
721
 
        log_rev_iterator = iter([zip(view_revisions, nones, nones)])
722
 
    else:
723
 
        def _convert():
724
 
            for view in view_revisions:
725
 
                yield (view, None, None)
726
 
        log_rev_iterator = iter([_convert()])
727
 
    for adapter in log_adapters:
728
 
        # It would be nicer if log adapters were first class objects
729
 
        # with custom parameters. This will do for now. IGC 20090127
730
 
        if adapter == _make_delta_filter:
731
 
            log_rev_iterator = adapter(branch, generate_delta,
732
 
                search, log_rev_iterator, file_ids, direction)
733
 
        else:
734
 
            log_rev_iterator = adapter(branch, generate_delta,
735
 
                search, log_rev_iterator)
736
 
    return log_rev_iterator
737
 
 
738
 
 
739
 
def _make_search_filter(branch, generate_delta, search, log_rev_iterator):
740
 
    """Create a filtered iterator of log_rev_iterator matching on a regex.
741
 
 
742
 
    :param branch: The branch being logged.
743
 
    :param generate_delta: Whether to generate a delta for each revision.
744
 
    :param search: A user text search string.
745
 
    :param log_rev_iterator: An input iterator containing all revisions that
746
 
        could be displayed, in lists.
747
 
    :return: An iterator over lists of ((rev_id, revno, merge_depth), rev,
748
 
        delta).
749
 
    """
750
 
    if search is None:
751
 
        return log_rev_iterator
752
 
    searchRE = re_compile_checked(search, re.IGNORECASE,
753
 
            'log message filter')
754
 
    return _filter_message_re(searchRE, log_rev_iterator)
755
 
 
756
 
 
757
 
def _filter_message_re(searchRE, log_rev_iterator):
758
 
    for revs in log_rev_iterator:
759
 
        new_revs = []
760
 
        for (rev_id, revno, merge_depth), rev, delta in revs:
761
 
            if searchRE.search(rev.message):
762
 
                new_revs.append(((rev_id, revno, merge_depth), rev, delta))
763
 
        yield new_revs
764
 
 
765
 
 
766
 
def _make_delta_filter(branch, generate_delta, search, log_rev_iterator,
767
 
    fileids=None, direction='reverse'):
768
 
    """Add revision deltas to a log iterator if needed.
769
 
 
770
 
    :param branch: The branch being logged.
771
 
    :param generate_delta: Whether to generate a delta for each revision.
772
 
      Permitted values are None, 'full' and 'partial'.
773
 
    :param search: A user text search string.
774
 
    :param log_rev_iterator: An input iterator containing all revisions that
775
 
        could be displayed, in lists.
776
 
    :param fileids: If non empty, only revisions matching one or more of
777
 
      the file-ids are to be kept.
778
 
    :param direction: the direction in which view_revisions is sorted
779
 
    :return: An iterator over lists of ((rev_id, revno, merge_depth), rev,
780
 
        delta).
781
 
    """
782
 
    if not generate_delta and not fileids:
783
 
        return log_rev_iterator
784
 
    return _generate_deltas(branch.repository, log_rev_iterator,
785
 
        generate_delta, fileids, direction)
786
 
 
787
 
 
788
 
def _generate_deltas(repository, log_rev_iterator, delta_type, fileids,
789
 
    direction):
790
 
    """Create deltas for each batch of revisions in log_rev_iterator.
791
 
 
792
 
    If we're only generating deltas for the sake of filtering against
793
 
    file-ids, we stop generating deltas once all file-ids reach the
794
 
    appropriate life-cycle point. If we're receiving data newest to
795
 
    oldest, then that life-cycle point is 'add', otherwise it's 'remove'.
796
 
    """
797
 
    check_fileids = fileids is not None and len(fileids) > 0
798
 
    if check_fileids:
799
 
        fileid_set = set(fileids)
800
 
        if direction == 'reverse':
801
 
            stop_on = 'add'
802
 
        else:
803
 
            stop_on = 'remove'
804
 
    else:
805
 
        fileid_set = None
806
 
    for revs in log_rev_iterator:
807
 
        # If we were matching against fileids and we've run out,
808
 
        # there's nothing left to do
809
 
        if check_fileids and not fileid_set:
810
 
            return
811
 
        revisions = [rev[1] for rev in revs]
812
 
        new_revs = []
813
 
        if delta_type == 'full' and not check_fileids:
814
 
            deltas = repository.get_deltas_for_revisions(revisions)
815
 
            for rev, delta in izip(revs, deltas):
816
 
                new_revs.append((rev[0], rev[1], delta))
817
 
        else:
818
 
            deltas = repository.get_deltas_for_revisions(revisions, fileid_set)
819
 
            for rev, delta in izip(revs, deltas):
820
 
                if check_fileids:
821
 
                    if delta is None or not delta.has_changed():
822
 
                        continue
823
 
                    else:
824
 
                        _update_fileids(delta, fileid_set, stop_on)
825
 
                        if delta_type is None:
826
 
                            delta = None
827
 
                        elif delta_type == 'full':
828
 
                            # If the file matches all the time, rebuilding
829
 
                            # a full delta like this in addition to a partial
830
 
                            # one could be slow. However, it's likely that
831
 
                            # most revisions won't get this far, making it
832
 
                            # faster to filter on the partial deltas and
833
 
                            # build the occasional full delta than always
834
 
                            # building full deltas and filtering those.
835
 
                            rev_id = rev[0][0]
836
 
                            delta = repository.get_revision_delta(rev_id)
837
 
                new_revs.append((rev[0], rev[1], delta))
838
 
        yield new_revs
839
 
 
840
 
 
841
 
def _update_fileids(delta, fileids, stop_on):
842
 
    """Update the set of file-ids to search based on file lifecycle events.
843
 
    
844
 
    :param fileids: a set of fileids to update
845
 
    :param stop_on: either 'add' or 'remove' - take file-ids out of the
846
 
      fileids set once their add or remove entry is detected respectively
847
 
    """
848
 
    if stop_on == 'add':
849
 
        for item in delta.added:
850
 
            if item[1] in fileids:
851
 
                fileids.remove(item[1])
852
 
    elif stop_on == 'delete':
853
 
        for item in delta.removed:
854
 
            if item[1] in fileids:
855
 
                fileids.remove(item[1])
856
 
 
857
 
 
858
 
def _make_revision_objects(branch, generate_delta, search, log_rev_iterator):
859
 
    """Extract revision objects from the repository
860
 
 
861
 
    :param branch: The branch being logged.
862
 
    :param generate_delta: Whether to generate a delta for each revision.
863
 
    :param search: A user text search string.
864
 
    :param log_rev_iterator: An input iterator containing all revisions that
865
 
        could be displayed, in lists.
866
 
    :return: An iterator over lists of ((rev_id, revno, merge_depth), rev,
867
 
        delta).
868
 
    """
869
 
    repository = branch.repository
870
 
    for revs in log_rev_iterator:
871
 
        # r = revision_id, n = revno, d = merge depth
872
 
        revision_ids = [view[0] for view, _, _ in revs]
873
 
        revisions = repository.get_revisions(revision_ids)
874
 
        revs = [(rev[0], revision, rev[2]) for rev, revision in
875
 
            izip(revs, revisions)]
876
 
        yield revs
877
 
 
878
 
 
879
 
def _make_batch_filter(branch, generate_delta, search, log_rev_iterator):
880
 
    """Group up a single large batch into smaller ones.
881
 
 
882
 
    :param branch: The branch being logged.
883
 
    :param generate_delta: Whether to generate a delta for each revision.
884
 
    :param search: A user text search string.
885
 
    :param log_rev_iterator: An input iterator containing all revisions that
886
 
        could be displayed, in lists.
887
 
    :return: An iterator over lists of ((rev_id, revno, merge_depth), rev,
888
 
        delta).
889
 
    """
890
 
    repository = branch.repository
891
 
    num = 9
892
 
    for batch in log_rev_iterator:
893
 
        batch = iter(batch)
894
 
        while True:
895
 
            step = [detail for _, detail in zip(range(num), batch)]
896
 
            if len(step) == 0:
897
 
                break
898
 
            yield step
899
 
            num = min(int(num * 1.5), 200)
900
 
 
901
 
 
902
 
def _get_revision_limits(branch, start_revision, end_revision):
903
 
    """Get and check revision limits.
904
 
 
905
 
    :param  branch: The branch containing the revisions.
906
 
 
907
 
    :param  start_revision: The first revision to be logged.
908
 
            For backwards compatibility this may be a mainline integer revno,
909
 
            but for merge revision support a RevisionInfo is expected.
910
 
 
911
 
    :param  end_revision: The last revision to be logged.
912
 
            For backwards compatibility this may be a mainline integer revno,
913
 
            but for merge revision support a RevisionInfo is expected.
914
 
 
915
 
    :return: (start_rev_id, end_rev_id) tuple.
916
 
    """
917
 
    branch_revno, branch_rev_id = branch.last_revision_info()
918
 
    start_rev_id = None
919
 
    if start_revision is None:
920
 
        start_revno = 1
921
 
    else:
922
 
        if isinstance(start_revision, revisionspec.RevisionInfo):
923
 
            start_rev_id = start_revision.rev_id
924
 
            start_revno = start_revision.revno or 1
925
 
        else:
926
 
            branch.check_real_revno(start_revision)
927
 
            start_revno = start_revision
928
 
            start_rev_id = branch.get_rev_id(start_revno)
929
 
 
930
 
    end_rev_id = None
931
 
    if end_revision is None:
932
 
        end_revno = branch_revno
933
 
    else:
934
 
        if isinstance(end_revision, revisionspec.RevisionInfo):
935
 
            end_rev_id = end_revision.rev_id
936
 
            end_revno = end_revision.revno or branch_revno
937
 
        else:
938
 
            branch.check_real_revno(end_revision)
939
 
            end_revno = end_revision
940
 
            end_rev_id = branch.get_rev_id(end_revno)
941
 
 
942
 
    if branch_revno != 0:
943
 
        if (start_rev_id == _mod_revision.NULL_REVISION
944
 
            or end_rev_id == _mod_revision.NULL_REVISION):
945
 
            raise errors.BzrCommandError('Logging revision 0 is invalid.')
946
 
        if start_revno > end_revno:
947
 
            raise errors.BzrCommandError("Start revision must be older than "
948
 
                                         "the end revision.")
949
 
    return (start_rev_id, end_rev_id)
950
 
 
951
 
 
952
 
def _get_mainline_revs(branch, start_revision, end_revision):
953
 
    """Get the mainline revisions from the branch.
954
 
 
955
 
    Generates the list of mainline revisions for the branch.
956
 
 
957
 
    :param  branch: The branch containing the revisions.
958
 
 
959
 
    :param  start_revision: The first revision to be logged.
960
 
            For backwards compatibility this may be a mainline integer revno,
961
 
            but for merge revision support a RevisionInfo is expected.
962
 
 
963
 
    :param  end_revision: The last revision to be logged.
964
 
            For backwards compatibility this may be a mainline integer revno,
965
 
            but for merge revision support a RevisionInfo is expected.
966
 
 
967
 
    :return: A (mainline_revs, rev_nos, start_rev_id, end_rev_id) tuple.
968
 
    """
969
 
    branch_revno, branch_last_revision = branch.last_revision_info()
970
 
    if branch_revno == 0:
971
 
        return None, None, None, None
972
 
 
973
 
    # For mainline generation, map start_revision and end_revision to
974
 
    # mainline revnos. If the revision is not on the mainline choose the
975
 
    # appropriate extreme of the mainline instead - the extra will be
976
 
    # filtered later.
977
 
    # Also map the revisions to rev_ids, to be used in the later filtering
978
 
    # stage.
979
 
    start_rev_id = None
980
 
    if start_revision is None:
981
 
        start_revno = 1
982
 
    else:
983
 
        if isinstance(start_revision, revisionspec.RevisionInfo):
984
 
            start_rev_id = start_revision.rev_id
985
 
            start_revno = start_revision.revno or 1
986
 
        else:
987
 
            branch.check_real_revno(start_revision)
988
 
            start_revno = start_revision
989
 
 
990
 
    end_rev_id = None
991
 
    if end_revision is None:
992
 
        end_revno = branch_revno
993
 
    else:
994
 
        if isinstance(end_revision, revisionspec.RevisionInfo):
995
 
            end_rev_id = end_revision.rev_id
996
 
            end_revno = end_revision.revno or branch_revno
997
 
        else:
998
 
            branch.check_real_revno(end_revision)
999
 
            end_revno = end_revision
1000
 
 
1001
 
    if ((start_rev_id == _mod_revision.NULL_REVISION)
1002
 
        or (end_rev_id == _mod_revision.NULL_REVISION)):
1003
 
        raise errors.BzrCommandError('Logging revision 0 is invalid.')
1004
 
    if start_revno > end_revno:
1005
 
        raise errors.BzrCommandError("Start revision must be older than "
1006
 
                                     "the end revision.")
1007
 
 
1008
 
    if end_revno < start_revno:
1009
 
        return None, None, None, None
1010
 
    cur_revno = branch_revno
1011
 
    rev_nos = {}
1012
 
    mainline_revs = []
1013
 
    for revision_id in branch.repository.iter_reverse_revision_history(
1014
 
                        branch_last_revision):
1015
 
        if cur_revno < start_revno:
1016
 
            # We have gone far enough, but we always add 1 more revision
1017
 
            rev_nos[revision_id] = cur_revno
1018
 
            mainline_revs.append(revision_id)
1019
 
            break
1020
 
        if cur_revno <= end_revno:
1021
 
            rev_nos[revision_id] = cur_revno
1022
 
            mainline_revs.append(revision_id)
1023
 
        cur_revno -= 1
1024
 
    else:
1025
 
        # We walked off the edge of all revisions, so we add a 'None' marker
1026
 
        mainline_revs.append(None)
1027
 
 
1028
 
    mainline_revs.reverse()
1029
 
 
1030
 
    # override the mainline to look like the revision history.
1031
 
    return mainline_revs, rev_nos, start_rev_id, end_rev_id
1032
 
 
1033
 
 
1034
 
def _filter_revision_range(view_revisions, start_rev_id, end_rev_id):
1035
 
    """Filter view_revisions based on revision ranges.
1036
 
 
1037
 
    :param view_revisions: A list of (revision_id, dotted_revno, merge_depth)
1038
 
            tuples to be filtered.
1039
 
 
1040
 
    :param start_rev_id: If not NONE specifies the first revision to be logged.
1041
 
            If NONE then all revisions up to the end_rev_id are logged.
1042
 
 
1043
 
    :param end_rev_id: If not NONE specifies the last revision to be logged.
1044
 
            If NONE then all revisions up to the end of the log are logged.
1045
 
 
1046
 
    :return: The filtered view_revisions.
1047
 
    """
1048
 
    # This method is no longer called by the main code path.
1049
 
    # It may be removed soon. IGC 20090127
1050
 
    if start_rev_id or end_rev_id:
1051
 
        revision_ids = [r for r, n, d in view_revisions]
1052
 
        if start_rev_id:
1053
 
            start_index = revision_ids.index(start_rev_id)
1054
 
        else:
1055
 
            start_index = 0
1056
 
        if start_rev_id == end_rev_id:
1057
 
            end_index = start_index
1058
 
        else:
1059
 
            if end_rev_id:
1060
 
                end_index = revision_ids.index(end_rev_id)
1061
 
            else:
1062
 
                end_index = len(view_revisions) - 1
1063
 
        # To include the revisions merged into the last revision,
1064
 
        # extend end_rev_id down to, but not including, the next rev
1065
 
        # with the same or lesser merge_depth
1066
 
        end_merge_depth = view_revisions[end_index][2]
1067
 
        try:
1068
 
            for index in xrange(end_index+1, len(view_revisions)+1):
1069
 
                if view_revisions[index][2] <= end_merge_depth:
1070
 
                    end_index = index - 1
1071
 
                    break
1072
 
        except IndexError:
1073
 
            # if the search falls off the end then log to the end as well
1074
 
            end_index = len(view_revisions) - 1
1075
 
        view_revisions = view_revisions[start_index:end_index+1]
1076
 
    return view_revisions
1077
 
 
1078
 
 
1079
 
def _filter_revisions_touching_file_id(branch, file_id, view_revisions,
1080
 
    include_merges=True):
1081
 
    r"""Return the list of revision ids which touch a given file id.
1082
 
 
1083
 
    The function filters view_revisions and returns a subset.
1084
 
    This includes the revisions which directly change the file id,
1085
 
    and the revisions which merge these changes. So if the
1086
 
    revision graph is::
1087
 
        A-.
1088
 
        |\ \
1089
 
        B C E
1090
 
        |/ /
1091
 
        D |
1092
 
        |\|
1093
 
        | F
1094
 
        |/
1095
 
        G
1096
 
 
1097
 
    And 'C' changes a file, then both C and D will be returned. F will not be
1098
 
    returned even though it brings the changes to C into the branch starting
1099
 
    with E. (Note that if we were using F as the tip instead of G, then we
1100
 
    would see C, D, F.)
1101
 
 
1102
 
    This will also be restricted based on a subset of the mainline.
1103
 
 
1104
 
    :param branch: The branch where we can get text revision information.
1105
 
 
1106
 
    :param file_id: Filter out revisions that do not touch file_id.
1107
 
 
1108
 
    :param view_revisions: A list of (revision_id, dotted_revno, merge_depth)
1109
 
        tuples. This is the list of revisions which will be filtered. It is
1110
 
        assumed that view_revisions is in merge_sort order (i.e. newest
1111
 
        revision first ).
1112
 
 
1113
 
    :param include_merges: include merge revisions in the result or not
1114
 
 
1115
 
    :return: A list of (revision_id, dotted_revno, merge_depth) tuples.
1116
 
    """
1117
 
    # Lookup all possible text keys to determine which ones actually modified
1118
 
    # the file.
1119
 
    text_keys = [(file_id, rev_id) for rev_id, revno, depth in view_revisions]
1120
 
    next_keys = None
1121
 
    # Looking up keys in batches of 1000 can cut the time in half, as well as
1122
 
    # memory consumption. GraphIndex *does* like to look for a few keys in
1123
 
    # parallel, it just doesn't like looking for *lots* of keys in parallel.
1124
 
    # TODO: This code needs to be re-evaluated periodically as we tune the
1125
 
    #       indexing layer. We might consider passing in hints as to the known
1126
 
    #       access pattern (sparse/clustered, high success rate/low success
1127
 
    #       rate). This particular access is clustered with a low success rate.
1128
 
    get_parent_map = branch.repository.texts.get_parent_map
1129
 
    modified_text_revisions = set()
1130
 
    chunk_size = 1000
1131
 
    for start in xrange(0, len(text_keys), chunk_size):
1132
 
        next_keys = text_keys[start:start + chunk_size]
1133
 
        # Only keep the revision_id portion of the key
1134
 
        modified_text_revisions.update(
1135
 
            [k[1] for k in get_parent_map(next_keys)])
1136
 
    del text_keys, next_keys
1137
 
 
1138
 
    result = []
1139
 
    # Track what revisions will merge the current revision, replace entries
1140
 
    # with 'None' when they have been added to result
1141
 
    current_merge_stack = [None]
1142
 
    for info in view_revisions:
1143
 
        rev_id, revno, depth = info
1144
 
        if depth == len(current_merge_stack):
1145
 
            current_merge_stack.append(info)
1146
 
        else:
1147
 
            del current_merge_stack[depth + 1:]
1148
 
            current_merge_stack[-1] = info
1149
 
 
1150
 
        if rev_id in modified_text_revisions:
1151
 
            # This needs to be logged, along with the extra revisions
1152
 
            for idx in xrange(len(current_merge_stack)):
1153
 
                node = current_merge_stack[idx]
1154
 
                if node is not None:
1155
 
                    if include_merges or node[2] == 0:
1156
 
                        result.append(node)
1157
 
                        current_merge_stack[idx] = None
1158
 
    return result
1159
 
 
1160
 
 
1161
 
def get_view_revisions(mainline_revs, rev_nos, branch, direction,
1162
 
                       include_merges=True):
1163
 
    """Produce an iterator of revisions to show
1164
 
    :return: an iterator of (revision_id, revno, merge_depth)
1165
 
    (if there is no revno for a revision, None is supplied)
1166
 
    """
1167
 
    # This method is no longer called by the main code path.
1168
 
    # It is retained for API compatibility and may be deprecated
1169
 
    # soon. IGC 20090127
1170
 
    if not include_merges:
1171
 
        revision_ids = mainline_revs[1:]
1172
 
        if direction == 'reverse':
1173
 
            revision_ids.reverse()
1174
 
        for revision_id in revision_ids:
1175
 
            yield revision_id, str(rev_nos[revision_id]), 0
1176
 
        return
1177
 
    graph = branch.repository.get_graph()
1178
 
    # This asks for all mainline revisions, which means we only have to spider
1179
 
    # sideways, rather than depth history. That said, its still size-of-history
1180
 
    # and should be addressed.
1181
 
    # mainline_revisions always includes an extra revision at the beginning, so
1182
 
    # don't request it.
1183
 
    parent_map = dict(((key, value) for key, value in
1184
 
        graph.iter_ancestry(mainline_revs[1:]) if value is not None))
1185
 
    # filter out ghosts; merge_sort errors on ghosts.
1186
 
    rev_graph = _mod_repository._strip_NULL_ghosts(parent_map)
1187
 
    merge_sorted_revisions = tsort.merge_sort(
1188
 
        rev_graph,
1189
 
        mainline_revs[-1],
1190
 
        mainline_revs,
1191
 
        generate_revno=True)
1192
 
 
1193
 
    if direction == 'forward':
1194
 
        # forward means oldest first.
1195
 
        merge_sorted_revisions = reverse_by_depth(merge_sorted_revisions)
1196
 
    elif direction != 'reverse':
1197
 
        raise ValueError('invalid direction %r' % direction)
1198
 
 
1199
 
    for (sequence, rev_id, merge_depth, revno, end_of_merge
1200
 
         ) in merge_sorted_revisions:
1201
 
        yield rev_id, '.'.join(map(str, revno)), merge_depth
1202
 
 
1203
 
 
1204
 
def reverse_by_depth(merge_sorted_revisions, _depth=0):
1205
 
    """Reverse revisions by depth.
1206
 
 
1207
 
    Revisions with a different depth are sorted as a group with the previous
1208
 
    revision of that depth.  There may be no topological justification for this,
1209
 
    but it looks much nicer.
1210
 
    """
1211
 
    # Add a fake revision at start so that we can always attach sub revisions
1212
 
    merge_sorted_revisions = [(None, None, _depth)] + merge_sorted_revisions
1213
 
    zd_revisions = []
1214
 
    for val in merge_sorted_revisions:
1215
 
        if val[2] == _depth:
1216
 
            # Each revision at the current depth becomes a chunk grouping all
1217
 
            # higher depth revisions.
1218
 
            zd_revisions.append([val])
1219
 
        else:
1220
 
            zd_revisions[-1].append(val)
1221
 
    for revisions in zd_revisions:
1222
 
        if len(revisions) > 1:
1223
 
            # We have higher depth revisions, let reverse them locally
1224
 
            revisions[1:] = reverse_by_depth(revisions[1:], _depth + 1)
1225
 
    zd_revisions.reverse()
1226
 
    result = []
1227
 
    for chunk in zd_revisions:
1228
 
        result.extend(chunk)
1229
 
    if _depth == 0:
1230
 
        # Top level call, get rid of the fake revisions that have been added
1231
 
        result = [r for r in result if r[0] is not None and r[1] is not None]
1232
 
    return result
1233
 
 
1234
 
 
1235
 
class LogRevision(object):
1236
 
    """A revision to be logged (by LogFormatter.log_revision).
1237
 
 
1238
 
    A simple wrapper for the attributes of a revision to be logged.
1239
 
    The attributes may or may not be populated, as determined by the
1240
 
    logging options and the log formatter capabilities.
1241
 
    """
1242
 
 
1243
 
    def __init__(self, rev=None, revno=None, merge_depth=0, delta=None,
1244
 
                 tags=None, diff=None):
1245
 
        self.rev = rev
1246
 
        self.revno = str(revno)
1247
 
        self.merge_depth = merge_depth
1248
 
        self.delta = delta
1249
 
        self.tags = tags
1250
 
        self.diff = diff
1251
 
 
1252
 
 
1253
 
class LogFormatter(object):
1254
 
    """Abstract class to display log messages.
1255
 
 
1256
 
    At a minimum, a derived class must implement the log_revision method.
1257
 
 
1258
 
    If the LogFormatter needs to be informed of the beginning or end of
1259
 
    a log it should implement the begin_log and/or end_log hook methods.
1260
 
 
1261
 
    A LogFormatter should define the following supports_XXX flags
1262
 
    to indicate which LogRevision attributes it supports:
1263
 
 
1264
 
    - supports_delta must be True if this log formatter supports delta.
1265
 
        Otherwise the delta attribute may not be populated.  The 'delta_format'
1266
 
        attribute describes whether the 'short_status' format (1) or the long
1267
 
        one (2) should be used.
1268
 
 
1269
 
    - supports_merge_revisions must be True if this log formatter supports
1270
 
        merge revisions.  If not, then only mainline revisions will be passed
1271
 
        to the formatter.
1272
 
 
1273
 
    - preferred_levels is the number of levels this formatter defaults to.
1274
 
        The default value is zero meaning display all levels.
1275
 
        This value is only relevant if supports_merge_revisions is True.
1276
 
 
1277
 
    - supports_tags must be True if this log formatter supports tags.
1278
 
        Otherwise the tags attribute may not be populated.
1279
 
 
1280
 
    - supports_diff must be True if this log formatter supports diffs.
1281
 
        Otherwise the diff attribute may not be populated.
1282
 
 
1283
 
    Plugins can register functions to show custom revision properties using
1284
 
    the properties_handler_registry. The registered function
1285
 
    must respect the following interface description:
1286
 
        def my_show_properties(properties_dict):
1287
 
            # code that returns a dict {'name':'value'} of the properties
1288
 
            # to be shown
1289
 
    """
1290
 
    preferred_levels = 0
1291
 
 
1292
 
    def __init__(self, to_file, show_ids=False, show_timezone='original',
1293
 
                 delta_format=None, levels=None):
1294
 
        """Create a LogFormatter.
1295
 
 
1296
 
        :param to_file: the file to output to
1297
 
        :param show_ids: if True, revision-ids are to be displayed
1298
 
        :param show_timezone: the timezone to use
1299
 
        :param delta_format: the level of delta information to display
1300
 
          or None to leave it u to the formatter to decide
1301
 
        :param levels: the number of levels to display; None or -1 to
1302
 
          let the log formatter decide.
1303
 
        """
1304
 
        self.to_file = to_file
1305
 
        # 'exact' stream used to show diff, it should print content 'as is'
1306
 
        # and should not try to decode/encode it to unicode to avoid bug #328007
1307
 
        self.to_exact_file = getattr(to_file, 'stream', to_file)
1308
 
        self.show_ids = show_ids
1309
 
        self.show_timezone = show_timezone
1310
 
        if delta_format is None:
1311
 
            # Ensures backward compatibility
1312
 
            delta_format = 2 # long format
1313
 
        self.delta_format = delta_format
1314
 
        self.levels = levels
1315
 
        self._merge_count = 0
1316
 
 
1317
 
    def get_levels(self):
1318
 
        """Get the number of levels to display or 0 for all."""
1319
 
        if getattr(self, 'supports_merge_revisions', False):
1320
 
            if self.levels is None or self.levels == -1:
1321
 
                self.levels = self.preferred_levels
1322
 
        else:
1323
 
            self.levels = 1
1324
 
        return self.levels
1325
 
 
1326
 
    def log_revision(self, revision):
1327
 
        """Log a revision.
1328
 
 
1329
 
        :param  revision:   The LogRevision to be logged.
1330
 
        """
1331
 
        raise NotImplementedError('not implemented in abstract base')
1332
 
 
1333
 
    def show_advice(self):
1334
 
        """Output user advice, if any, when the log is completed."""
1335
 
        if self.levels == 1 and self._merge_count > 0:
1336
 
            advice_sep = self.get_advice_separator()
1337
 
            if advice_sep:
1338
 
                self.to_file.write(advice_sep)
1339
 
            self.to_file.write(
1340
 
                "Use --levels 0 (or -n0) to see merged revisions.\n")
1341
 
 
1342
 
    def get_advice_separator(self):
1343
 
        """Get the text separating the log from the closing advice."""
1344
 
        return ''
1345
 
 
1346
 
    def short_committer(self, rev):
1347
 
        name, address = config.parse_username(rev.committer)
1348
 
        if name:
1349
 
            return name
1350
 
        return address
1351
 
 
1352
 
    def short_author(self, rev):
1353
 
        name, address = config.parse_username(rev.get_apparent_authors()[0])
1354
 
        if name:
1355
 
            return name
1356
 
        return address
1357
 
 
1358
 
    def merge_marker(self, revision):
1359
 
        """Get the merge marker to include in the output or '' if none."""
1360
 
        if len(revision.rev.parent_ids) > 1:
1361
 
            self._merge_count += 1
1362
 
            return ' [merge]'
1363
 
        else:
1364
 
            return ''
1365
 
 
1366
 
    def show_properties(self, revision, indent):
1367
 
        """Displays the custom properties returned by each registered handler.
1368
 
 
1369
 
        If a registered handler raises an error it is propagated.
1370
 
        """
1371
 
        for key, handler in properties_handler_registry.iteritems():
1372
 
            for key, value in handler(revision).items():
1373
 
                self.to_file.write(indent + key + ': ' + value + '\n')
1374
 
 
1375
 
    def show_diff(self, to_file, diff, indent):
1376
 
        for l in diff.rstrip().split('\n'):
1377
 
            to_file.write(indent + '%s\n' % (l,))
1378
 
 
1379
 
 
1380
 
class LongLogFormatter(LogFormatter):
1381
 
 
1382
 
    supports_merge_revisions = True
1383
 
    preferred_levels = 1
1384
 
    supports_delta = True
1385
 
    supports_tags = True
1386
 
    supports_diff = True
1387
 
 
1388
 
    def log_revision(self, revision):
1389
 
        """Log a revision, either merged or not."""
1390
 
        indent = '    ' * revision.merge_depth
1391
 
        to_file = self.to_file
1392
 
        to_file.write(indent + '-' * 60 + '\n')
1393
 
        if revision.revno is not None:
1394
 
            to_file.write(indent + 'revno: %s%s\n' % (revision.revno,
1395
 
                self.merge_marker(revision)))
1396
 
        if revision.tags:
1397
 
            to_file.write(indent + 'tags: %s\n' % (', '.join(revision.tags)))
1398
 
        if self.show_ids:
1399
 
            to_file.write(indent + 'revision-id: ' + revision.rev.revision_id)
1400
 
            to_file.write('\n')
1401
 
            for parent_id in revision.rev.parent_ids:
1402
 
                to_file.write(indent + 'parent: %s\n' % (parent_id,))
1403
 
        self.show_properties(revision.rev, indent)
1404
 
 
1405
 
        committer = revision.rev.committer
1406
 
        authors = revision.rev.get_apparent_authors()
1407
 
        if authors != [committer]:
1408
 
            to_file.write(indent + 'author: %s\n' % (", ".join(authors),))
1409
 
        to_file.write(indent + 'committer: %s\n' % (committer,))
1410
 
 
1411
 
        branch_nick = revision.rev.properties.get('branch-nick', None)
1412
 
        if branch_nick is not None:
1413
 
            to_file.write(indent + 'branch nick: %s\n' % (branch_nick,))
1414
 
 
1415
 
        date_str = format_date(revision.rev.timestamp,
1416
 
                               revision.rev.timezone or 0,
1417
 
                               self.show_timezone)
1418
 
        to_file.write(indent + 'timestamp: %s\n' % (date_str,))
1419
 
 
1420
 
        to_file.write(indent + 'message:\n')
1421
 
        if not revision.rev.message:
1422
 
            to_file.write(indent + '  (no message)\n')
1423
 
        else:
1424
 
            message = revision.rev.message.rstrip('\r\n')
1425
 
            for l in message.split('\n'):
1426
 
                to_file.write(indent + '  %s\n' % (l,))
1427
 
        if revision.delta is not None:
1428
 
            # We don't respect delta_format for compatibility
1429
 
            revision.delta.show(to_file, self.show_ids, indent=indent,
1430
 
                                short_status=False)
1431
 
        if revision.diff is not None:
1432
 
            to_file.write(indent + 'diff:\n')
1433
 
            # Note: we explicitly don't indent the diff (relative to the
1434
 
            # revision information) so that the output can be fed to patch -p0
1435
 
            self.show_diff(self.to_exact_file, revision.diff, indent)
1436
 
 
1437
 
    def get_advice_separator(self):
1438
 
        """Get the text separating the log from the closing advice."""
1439
 
        return '-' * 60 + '\n'
1440
 
 
1441
 
 
1442
 
class ShortLogFormatter(LogFormatter):
1443
 
 
1444
 
    supports_merge_revisions = True
1445
 
    preferred_levels = 1
1446
 
    supports_delta = True
1447
 
    supports_tags = True
1448
 
    supports_diff = True
1449
 
 
1450
 
    def __init__(self, *args, **kwargs):
1451
 
        super(ShortLogFormatter, self).__init__(*args, **kwargs)
1452
 
        self.revno_width_by_depth = {}
1453
 
 
1454
 
    def log_revision(self, revision):
1455
 
        # We need two indents: one per depth and one for the information
1456
 
        # relative to that indent. Most mainline revnos are 5 chars or
1457
 
        # less while dotted revnos are typically 11 chars or less. Once
1458
 
        # calculated, we need to remember the offset for a given depth
1459
 
        # as we might be starting from a dotted revno in the first column
1460
 
        # and we want subsequent mainline revisions to line up.
1461
 
        depth = revision.merge_depth
1462
 
        indent = '    ' * depth
1463
 
        revno_width = self.revno_width_by_depth.get(depth)
1464
 
        if revno_width is None:
1465
 
            if revision.revno.find('.') == -1:
1466
 
                # mainline revno, e.g. 12345
1467
 
                revno_width = 5
1468
 
            else:
1469
 
                # dotted revno, e.g. 12345.10.55
1470
 
                revno_width = 11
1471
 
            self.revno_width_by_depth[depth] = revno_width
1472
 
        offset = ' ' * (revno_width + 1)
1473
 
 
1474
 
        to_file = self.to_file
1475
 
        tags = ''
1476
 
        if revision.tags:
1477
 
            tags = ' {%s}' % (', '.join(revision.tags))
1478
 
        to_file.write(indent + "%*s %s\t%s%s%s\n" % (revno_width,
1479
 
                revision.revno, self.short_author(revision.rev),
1480
 
                format_date(revision.rev.timestamp,
1481
 
                            revision.rev.timezone or 0,
1482
 
                            self.show_timezone, date_fmt="%Y-%m-%d",
1483
 
                            show_offset=False),
1484
 
                tags, self.merge_marker(revision)))
1485
 
        self.show_properties(revision.rev, indent+offset)
1486
 
        if self.show_ids:
1487
 
            to_file.write(indent + offset + 'revision-id:%s\n'
1488
 
                          % (revision.rev.revision_id,))
1489
 
        if not revision.rev.message:
1490
 
            to_file.write(indent + offset + '(no message)\n')
1491
 
        else:
1492
 
            message = revision.rev.message.rstrip('\r\n')
1493
 
            for l in message.split('\n'):
1494
 
                to_file.write(indent + offset + '%s\n' % (l,))
1495
 
 
1496
 
        if revision.delta is not None:
1497
 
            revision.delta.show(to_file, self.show_ids, indent=indent + offset,
1498
 
                                short_status=self.delta_format==1)
1499
 
        if revision.diff is not None:
1500
 
            self.show_diff(self.to_exact_file, revision.diff, '      ')
1501
 
        to_file.write('\n')
1502
 
 
1503
 
 
1504
 
class LineLogFormatter(LogFormatter):
1505
 
 
1506
 
    supports_merge_revisions = True
1507
 
    preferred_levels = 1
1508
 
    supports_tags = True
1509
 
 
1510
 
    def __init__(self, *args, **kwargs):
1511
 
        super(LineLogFormatter, self).__init__(*args, **kwargs)
1512
 
        self._max_chars = terminal_width() - 1
1513
 
 
1514
 
    def truncate(self, str, max_len):
1515
 
        if len(str) <= max_len:
1516
 
            return str
1517
 
        return str[:max_len-3]+'...'
1518
 
 
1519
 
    def date_string(self, rev):
1520
 
        return format_date(rev.timestamp, rev.timezone or 0,
1521
 
                           self.show_timezone, date_fmt="%Y-%m-%d",
1522
 
                           show_offset=False)
1523
 
 
1524
 
    def message(self, rev):
 
112
        from tree import EmptyTree
 
113
        prev_tree = EmptyTree()
 
114
    for revno, revision_id in which_revs():
 
115
        print >>to_file,  '-' * 60
 
116
        print >>to_file,  'revno:', revno
 
117
        rev = branch.get_revision(revision_id)
 
118
        if show_ids:
 
119
            print >>to_file,  'revision-id:', revision_id
 
120
        print >>to_file,  'committer:', rev.committer
 
121
        print >>to_file,  'timestamp: %s' % (format_date(rev.timestamp, rev.timezone or 0,
 
122
                                             show_timezone))
 
123
 
 
124
        if revision_id != rev.revision_id:
 
125
            raise BzrCheckError("retrieved wrong revision: %r"
 
126
                                % (revision_id, rev.revision_id))
 
127
 
 
128
        print >>to_file,  'message:'
1525
129
        if not rev.message:
1526
 
            return '(no message)'
1527
 
        else:
1528
 
            return rev.message
1529
 
 
1530
 
    def log_revision(self, revision):
1531
 
        indent = '  ' * revision.merge_depth
1532
 
        self.to_file.write(self.log_string(revision.revno, revision.rev,
1533
 
            self._max_chars, revision.tags, indent))
1534
 
        self.to_file.write('\n')
1535
 
 
1536
 
    def log_string(self, revno, rev, max_chars, tags=None, prefix=''):
1537
 
        """Format log info into one string. Truncate tail of string
1538
 
        :param  revno:      revision number or None.
1539
 
                            Revision numbers counts from 1.
1540
 
        :param  rev:        revision object
1541
 
        :param  max_chars:  maximum length of resulting string
1542
 
        :param  tags:       list of tags or None
1543
 
        :param  prefix:     string to prefix each line
1544
 
        :return:            formatted truncated string
1545
 
        """
1546
 
        out = []
1547
 
        if revno:
1548
 
            # show revno only when is not None
1549
 
            out.append("%s:" % revno)
1550
 
        out.append(self.truncate(self.short_author(rev), 20))
1551
 
        out.append(self.date_string(rev))
1552
 
        if len(rev.parent_ids) > 1:
1553
 
            out.append('[merge]')
1554
 
        if tags:
1555
 
            tag_str = '{%s}' % (', '.join(tags))
1556
 
            out.append(tag_str)
1557
 
        out.append(rev.get_summary())
1558
 
        return self.truncate(prefix + " ".join(out).rstrip('\n'), max_chars)
1559
 
 
1560
 
 
1561
 
class GnuChangelogLogFormatter(LogFormatter):
1562
 
 
1563
 
    supports_merge_revisions = True
1564
 
    supports_delta = True
1565
 
 
1566
 
    def log_revision(self, revision):
1567
 
        """Log a revision, either merged or not."""
1568
 
        to_file = self.to_file
1569
 
 
1570
 
        date_str = format_date(revision.rev.timestamp,
1571
 
                               revision.rev.timezone or 0,
1572
 
                               self.show_timezone,
1573
 
                               date_fmt='%Y-%m-%d',
1574
 
                               show_offset=False)
1575
 
        committer_str = revision.rev.committer.replace (' <', '  <')
1576
 
        to_file.write('%s  %s\n\n' % (date_str,committer_str))
1577
 
 
1578
 
        if revision.delta is not None and revision.delta.has_changed():
1579
 
            for c in revision.delta.added + revision.delta.removed + revision.delta.modified:
1580
 
                path, = c[:1]
1581
 
                to_file.write('\t* %s:\n' % (path,))
1582
 
            for c in revision.delta.renamed:
1583
 
                oldpath,newpath = c[:2]
1584
 
                # For renamed files, show both the old and the new path
1585
 
                to_file.write('\t* %s:\n\t* %s:\n' % (oldpath,newpath))
1586
 
            to_file.write('\n')
1587
 
 
1588
 
        if not revision.rev.message:
1589
 
            to_file.write('\tNo commit message\n')
1590
 
        else:
1591
 
            message = revision.rev.message.rstrip('\r\n')
1592
 
            for l in message.split('\n'):
1593
 
                to_file.write('\t%s\n' % (l.lstrip(),))
1594
 
            to_file.write('\n')
1595
 
 
1596
 
 
1597
 
def line_log(rev, max_chars):
1598
 
    lf = LineLogFormatter(None)
1599
 
    return lf.log_string(None, rev, max_chars)
1600
 
 
1601
 
 
1602
 
class LogFormatterRegistry(registry.Registry):
1603
 
    """Registry for log formatters"""
1604
 
 
1605
 
    def make_formatter(self, name, *args, **kwargs):
1606
 
        """Construct a formatter from arguments.
1607
 
 
1608
 
        :param name: Name of the formatter to construct.  'short', 'long' and
1609
 
            'line' are built-in.
1610
 
        """
1611
 
        return self.get(name)(*args, **kwargs)
1612
 
 
1613
 
    def get_default(self, branch):
1614
 
        return self.get(branch.get_config().log_format())
1615
 
 
1616
 
 
1617
 
log_formatter_registry = LogFormatterRegistry()
1618
 
 
1619
 
 
1620
 
log_formatter_registry.register('short', ShortLogFormatter,
1621
 
                                'Moderately short log format')
1622
 
log_formatter_registry.register('long', LongLogFormatter,
1623
 
                                'Detailed log format')
1624
 
log_formatter_registry.register('line', LineLogFormatter,
1625
 
                                'Log format with one line per revision')
1626
 
log_formatter_registry.register('gnu-changelog', GnuChangelogLogFormatter,
1627
 
                                'Format used by GNU ChangeLog files')
1628
 
 
1629
 
 
1630
 
def register_formatter(name, formatter):
1631
 
    log_formatter_registry.register(name, formatter)
1632
 
 
1633
 
 
1634
 
def log_formatter(name, *args, **kwargs):
1635
 
    """Construct a formatter from arguments.
1636
 
 
1637
 
    name -- Name of the formatter to construct; currently 'long', 'short' and
1638
 
        'line' are supported.
1639
 
    """
1640
 
    try:
1641
 
        return log_formatter_registry.make_formatter(name, *args, **kwargs)
1642
 
    except KeyError:
1643
 
        raise errors.BzrCommandError("unknown log formatter: %r" % name)
1644
 
 
1645
 
 
1646
 
def show_one_log(revno, rev, delta, verbose, to_file, show_timezone):
1647
 
    # deprecated; for compatibility
1648
 
    lf = LongLogFormatter(to_file=to_file, show_timezone=show_timezone)
1649
 
    lf.show(revno, rev, delta)
1650
 
 
1651
 
 
1652
 
def show_changed_revisions(branch, old_rh, new_rh, to_file=None,
1653
 
                           log_format='long'):
1654
 
    """Show the change in revision history comparing the old revision history to the new one.
1655
 
 
1656
 
    :param branch: The branch where the revisions exist
1657
 
    :param old_rh: The old revision history
1658
 
    :param new_rh: The new revision history
1659
 
    :param to_file: A file to write the results to. If None, stdout will be used
1660
 
    """
1661
 
    if to_file is None:
1662
 
        to_file = codecs.getwriter(get_terminal_encoding())(sys.stdout,
1663
 
            errors='replace')
1664
 
    lf = log_formatter(log_format,
1665
 
                       show_ids=False,
1666
 
                       to_file=to_file,
1667
 
                       show_timezone='original')
1668
 
 
1669
 
    # This is the first index which is different between
1670
 
    # old and new
1671
 
    base_idx = None
1672
 
    for i in xrange(max(len(new_rh),
1673
 
                        len(old_rh))):
1674
 
        if (len(new_rh) <= i
1675
 
            or len(old_rh) <= i
1676
 
            or new_rh[i] != old_rh[i]):
1677
 
            base_idx = i
1678
 
            break
1679
 
 
1680
 
    if base_idx is None:
1681
 
        to_file.write('Nothing seems to have changed\n')
1682
 
        return
1683
 
    ## TODO: It might be nice to do something like show_log
1684
 
    ##       and show the merged entries. But since this is the
1685
 
    ##       removed revisions, it shouldn't be as important
1686
 
    if base_idx < len(old_rh):
1687
 
        to_file.write('*'*60)
1688
 
        to_file.write('\nRemoved Revisions:\n')
1689
 
        for i in range(base_idx, len(old_rh)):
1690
 
            rev = branch.repository.get_revision(old_rh[i])
1691
 
            lr = LogRevision(rev, i+1, 0, None)
1692
 
            lf.log_revision(lr)
1693
 
        to_file.write('*'*60)
1694
 
        to_file.write('\n\n')
1695
 
    if base_idx < len(new_rh):
1696
 
        to_file.write('Added Revisions:\n')
1697
 
        show_log(branch,
1698
 
                 lf,
1699
 
                 None,
1700
 
                 verbose=False,
1701
 
                 direction='forward',
1702
 
                 start_revision=base_idx+1,
1703
 
                 end_revision=len(new_rh),
1704
 
                 search=None)
1705
 
 
1706
 
 
1707
 
def get_history_change(old_revision_id, new_revision_id, repository):
1708
 
    """Calculate the uncommon lefthand history between two revisions.
1709
 
 
1710
 
    :param old_revision_id: The original revision id.
1711
 
    :param new_revision_id: The new revision id.
1712
 
    :param repository: The repository to use for the calculation.
1713
 
 
1714
 
    return old_history, new_history
1715
 
    """
1716
 
    old_history = []
1717
 
    old_revisions = set()
1718
 
    new_history = []
1719
 
    new_revisions = set()
1720
 
    new_iter = repository.iter_reverse_revision_history(new_revision_id)
1721
 
    old_iter = repository.iter_reverse_revision_history(old_revision_id)
1722
 
    stop_revision = None
1723
 
    do_old = True
1724
 
    do_new = True
1725
 
    while do_new or do_old:
1726
 
        if do_new:
1727
 
            try:
1728
 
                new_revision = new_iter.next()
1729
 
            except StopIteration:
1730
 
                do_new = False
1731
 
            else:
1732
 
                new_history.append(new_revision)
1733
 
                new_revisions.add(new_revision)
1734
 
                if new_revision in old_revisions:
1735
 
                    stop_revision = new_revision
1736
 
                    break
1737
 
        if do_old:
1738
 
            try:
1739
 
                old_revision = old_iter.next()
1740
 
            except StopIteration:
1741
 
                do_old = False
1742
 
            else:
1743
 
                old_history.append(old_revision)
1744
 
                old_revisions.add(old_revision)
1745
 
                if old_revision in new_revisions:
1746
 
                    stop_revision = old_revision
1747
 
                    break
1748
 
    new_history.reverse()
1749
 
    old_history.reverse()
1750
 
    if stop_revision is not None:
1751
 
        new_history = new_history[new_history.index(stop_revision) + 1:]
1752
 
        old_history = old_history[old_history.index(stop_revision) + 1:]
1753
 
    return old_history, new_history
1754
 
 
1755
 
 
1756
 
def show_branch_change(branch, output, old_revno, old_revision_id):
1757
 
    """Show the changes made to a branch.
1758
 
 
1759
 
    :param branch: The branch to show changes about.
1760
 
    :param output: A file-like object to write changes to.
1761
 
    :param old_revno: The revno of the old tip.
1762
 
    :param old_revision_id: The revision_id of the old tip.
1763
 
    """
1764
 
    new_revno, new_revision_id = branch.last_revision_info()
1765
 
    old_history, new_history = get_history_change(old_revision_id,
1766
 
                                                  new_revision_id,
1767
 
                                                  branch.repository)
1768
 
    if old_history == [] and new_history == []:
1769
 
        output.write('Nothing seems to have changed\n')
1770
 
        return
1771
 
 
1772
 
    log_format = log_formatter_registry.get_default(branch)
1773
 
    lf = log_format(show_ids=False, to_file=output, show_timezone='original')
1774
 
    if old_history != []:
1775
 
        output.write('*'*60)
1776
 
        output.write('\nRemoved Revisions:\n')
1777
 
        show_flat_log(branch.repository, old_history, old_revno, lf)
1778
 
        output.write('*'*60)
1779
 
        output.write('\n\n')
1780
 
    if new_history != []:
1781
 
        output.write('Added Revisions:\n')
1782
 
        start_revno = new_revno - len(new_history) + 1
1783
 
        show_log(branch, lf, None, verbose=False, direction='forward',
1784
 
                 start_revision=start_revno,)
1785
 
 
1786
 
 
1787
 
def show_flat_log(repository, history, last_revno, lf):
1788
 
    """Show a simple log of the specified history.
1789
 
 
1790
 
    :param repository: The repository to retrieve revisions from.
1791
 
    :param history: A list of revision_ids indicating the lefthand history.
1792
 
    :param last_revno: The revno of the last revision_id in the history.
1793
 
    :param lf: The log formatter to use.
1794
 
    """
1795
 
    start_revno = last_revno - len(history) + 1
1796
 
    revisions = repository.get_revisions(history)
1797
 
    for i, rev in enumerate(revisions):
1798
 
        lr = LogRevision(rev, i + last_revno, 0, None)
1799
 
        lf.log_revision(lr)
1800
 
 
1801
 
 
1802
 
def _get_info_for_log_files(revisionspec_list, file_list):
1803
 
    """Find file-ids and kinds given a list of files and a revision range.
1804
 
 
1805
 
    We search for files at the end of the range. If not found there,
1806
 
    we try the start of the range.
1807
 
 
1808
 
    :param revisionspec_list: revision range as parsed on the command line
1809
 
    :param file_list: the list of paths given on the command line;
1810
 
      the first of these can be a branch location or a file path,
1811
 
      the remainder must be file paths
1812
 
    :return: (branch, info_list, start_rev_info, end_rev_info) where
1813
 
      info_list is a list of (relative_path, file_id, kind) tuples where
1814
 
      kind is one of values 'directory', 'file', 'symlink', 'tree-reference'.
1815
 
    """
1816
 
    from builtins import _get_revision_range, safe_relpath_files
1817
 
    tree, b, path = bzrdir.BzrDir.open_containing_tree_or_branch(file_list[0])
1818
 
    # XXX: It's damn messy converting a list of paths to relative paths when
1819
 
    # those paths might be deleted ones, they might be on a case-insensitive
1820
 
    # filesystem and/or they might be in silly locations (like another branch).
1821
 
    # For example, what should "log bzr://branch/dir/file1 file2" do? (Is
1822
 
    # file2 implicitly in the same dir as file1 or should its directory be
1823
 
    # taken from the current tree somehow?) For now, this solves the common
1824
 
    # case of running log in a nested directory, assuming paths beyond the
1825
 
    # first one haven't been deleted ...
1826
 
    if tree:
1827
 
        relpaths = [path] + safe_relpath_files(tree, file_list[1:])
1828
 
    else:
1829
 
        relpaths = [path] + file_list[1:]
1830
 
    info_list = []
1831
 
    start_rev_info, end_rev_info = _get_revision_range(revisionspec_list, b,
1832
 
        "log")
1833
 
    if start_rev_info is None and end_rev_info is None:
1834
 
        if tree is None:
1835
 
            tree = b.basis_tree()
1836
 
        tree1 = None
1837
 
        for fp in relpaths:
1838
 
            file_id = tree.path2id(fp)
1839
 
            kind = _get_kind_for_file_id(tree, file_id)
1840
 
            if file_id is None:
1841
 
                # go back to when time began
1842
 
                if tree1 is None:
1843
 
                    try:
1844
 
                        rev1 = b.get_rev_id(1)
1845
 
                    except errors.NoSuchRevision:
1846
 
                        # No history at all
1847
 
                        file_id = None
1848
 
                        kind = None
1849
 
                    else:
1850
 
                        tree1 = b.repository.revision_tree(rev1)
1851
 
                if tree1:
1852
 
                    file_id = tree1.path2id(fp)
1853
 
                    kind = _get_kind_for_file_id(tree1, file_id)
1854
 
            info_list.append((fp, file_id, kind))
1855
 
 
1856
 
    elif start_rev_info == end_rev_info:
1857
 
        # One revision given - file must exist in it
1858
 
        tree = b.repository.revision_tree(end_rev_info.rev_id)
1859
 
        for fp in relpaths:
1860
 
            file_id = tree.path2id(fp)
1861
 
            kind = _get_kind_for_file_id(tree, file_id)
1862
 
            info_list.append((fp, file_id, kind))
1863
 
 
1864
 
    else:
1865
 
        # Revision range given. Get the file-id from the end tree.
1866
 
        # If that fails, try the start tree.
1867
 
        rev_id = end_rev_info.rev_id
1868
 
        if rev_id is None:
1869
 
            tree = b.basis_tree()
1870
 
        else:
1871
 
            tree = b.repository.revision_tree(rev_id)
1872
 
        tree1 = None
1873
 
        for fp in relpaths:
1874
 
            file_id = tree.path2id(fp)
1875
 
            kind = _get_kind_for_file_id(tree, file_id)
1876
 
            if file_id is None:
1877
 
                if tree1 is None:
1878
 
                    rev_id = start_rev_info.rev_id
1879
 
                    if rev_id is None:
1880
 
                        rev1 = b.get_rev_id(1)
1881
 
                        tree1 = b.repository.revision_tree(rev1)
1882
 
                    else:
1883
 
                        tree1 = b.repository.revision_tree(rev_id)
1884
 
                file_id = tree1.path2id(fp)
1885
 
                kind = _get_kind_for_file_id(tree1, file_id)
1886
 
            info_list.append((fp, file_id, kind))
1887
 
    return b, info_list, start_rev_info, end_rev_info
1888
 
 
1889
 
 
1890
 
def _get_kind_for_file_id(tree, file_id):
1891
 
    """Return the kind of a file-id or None if it doesn't exist."""
1892
 
    if file_id is not None:
1893
 
        return tree.kind(file_id)
1894
 
    else:
1895
 
        return None
1896
 
 
1897
 
 
1898
 
properties_handler_registry = registry.Registry()
1899
 
properties_handler_registry.register_lazy("foreign",
1900
 
                                          "bzrlib.foreign",
1901
 
                                          "show_foreign_properties")
1902
 
 
1903
 
 
1904
 
# adapters which revision ids to log are filtered. When log is called, the
1905
 
# log_rev_iterator is adapted through each of these factory methods.
1906
 
# Plugins are welcome to mutate this list in any way they like - as long
1907
 
# as the overall behaviour is preserved. At this point there is no extensible
1908
 
# mechanism for getting parameters to each factory method, and until there is
1909
 
# this won't be considered a stable api.
1910
 
log_adapters = [
1911
 
    # core log logic
1912
 
    _make_batch_filter,
1913
 
    # read revision objects
1914
 
    _make_revision_objects,
1915
 
    # filter on log messages
1916
 
    _make_search_filter,
1917
 
    # generate deltas for things we will show
1918
 
    _make_delta_filter
1919
 
    ]
 
130
            print >>to_file,  '  (no message)'
 
131
        else:
 
132
            for l in rev.message.split('\n'):
 
133
                print >>to_file,  '  ' + l
 
134
 
 
135
        # Don't show a list of changed files if we were asked about
 
136
        # one specific file.
 
137
 
 
138
        if verbose:
 
139
            this_tree = branch.revision_tree(revision_id)
 
140
            delta = compare_trees(prev_tree, this_tree, want_unchanged=False)
 
141
            delta.show(to_file, show_ids)
 
142
            prev_tree = this_tree
 
143
 
 
144
        precursor = revision_id
 
145