~abentley/bzrtools/bzrtools.dev

# Copyright (C) 2004, 2005 Aaron Bentley

# <aaron.bentley@utoronto.ca>

#

#    This program is free software; you can redistribute it and/or modify

#    it under the terms of the GNU General Public License as published by

#    the Free Software Foundation; either version 2 of the License, or

#    (at your option) any later version.

#

#    This program is distributed in the hope that it will be useful,

#    but WITHOUT ANY WARRANTY; without even the implied warranty of

#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

#    GNU General Public License for more details.

#

#    You should have received a copy of the GNU General Public License

#    along with this program; if not, write to the Free Software

#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

from bzrlib import Branch

from bzrlib.commands import Command

import os

import progress

import patches

import difflib

import sys

def iter_anno_data(branch, file_id):

    later_revision = branch.revno()

    q = range(branch.revno())

    q.reverse()

    later_text_id = branch.basis_tree().inventory[file_id].text_id

    i = 0

    for revno in q:

        i += 1

        cur_tree = branch.revision_tree(branch.lookup_revision(revno))

        if file_id not in cur_tree.inventory:

            text_id = None

        else:

            text_id = cur_tree.inventory[file_id].text_id

        if text_id != later_text_id:

            patch = get_patch(branch, revno, later_revision, file_id)

            yield revno, patch.iter_inserted(), patch

            later_revision = revno

            later_text_id = text_id

        yield progress.Progress("revisions", i)

def get_patch(branch, old_revno, new_revno, file_id):

    old_tree = branch.revision_tree(branch.lookup_revision(old_revno))

    new_tree = branch.revision_tree(branch.lookup_revision(new_revno))

    if file_id in old_tree.inventory:

        old_file = old_tree.get_file(file_id).readlines()

    else:

        old_file = []

    ud = difflib.unified_diff(old_file, new_tree.get_file(file_id).readlines())

    return patches.parse_patch(ud)

class cmd_annotate(Command):

    """Show which revision added each line in a file"""

    takes_args = ['filename']

    def run(self, filename):

        if not os.path.exists(filename):

            raise BzrCommandError("The file %s does not exist." % filename)

        branch = (Branch(filename))

        file_id = branch.working_tree().path2id(filename)

        if file_id is None:

            raise BzrCommandError("The file %s is not versioned." % filename)

        lines = branch.basis_tree().get_file(file_id)

        total = branch.revno()

        anno_d_iter = iter_anno_data(branch, file_id)

        progress_bar = progress.ProgressBar()

        try:

            for result in iter_annotate_file(lines, anno_d_iter):

                if isinstance(result, progress.Progress):

                    result.total = total

                    progress_bar(result)

                else:

                    anno_lines = result

        finally:

            progress_bar.clear()

        for line in anno_lines:

            sys.stdout.write("%4s:%s" % (str(line.log), line.text))

class AnnotateLine:

    """A line associated with the log that produced it"""

    def __init__(self, text, log=None):

        self.text = text

        self.log = log

class CantGetRevisionData(Exception):

    def __init__(self, revision):

        Exception.__init__(self, "Can't get data for revision %s" % revision)

def annotate_file2(file_lines, anno_iter):

    for result in iter_annotate_file(file_lines, anno_iter):

        pass

    return result

def iter_annotate_file(file_lines, anno_iter):

    lines = [AnnotateLine(f) for f in file_lines]

    patches = []

    try:

        for result in anno_iter:

            if isinstance(result, progress.Progress):

                yield result

                continue

            log, iter_inserted, patch = result

            for (num, line) in iter_inserted:

                old_num = num

                for cur_patch in patches:

                    num = cur_patch.pos_in_mod(num)

                    if num == None: 

                        break

                if num >= len(lines):

                    continue

                if num is not None and lines[num].log is None:

                    lines[num].log = log

            patches=[patch]+patches

    except CantGetRevisionData:

        pass

    yield lines