/brz/remove-bazaar

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

« back to all changes in this revision

Viewing changes to breezy/git/annotate.py

  • Committer: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/git/annotate.py
1
 
# Copyright (C) 2018 Jelmer Vernooij <jelmer@jelmer.uk>
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
"""Annotate."""
18
 
 
19
 
from __future__ import absolute_import
20
 
 
21
 
from dulwich.object_store import (
22
 
    tree_lookup_path,
23
 
    )
24
 
 
25
 
from ..errors import (
26
 
    NoSuchRevision,
27
 
    UnavailableRepresentation,
28
 
    )
29
 
from ..graph import Graph
30
 
from ..revision import (
31
 
    NULL_REVISION,
32
 
    )
33
 
 
34
 
 
35
 
class GitFulltextContentFactory(object):
36
 
    """Static data content factory.
37
 
 
38
 
    This takes a fulltext when created and just returns that during
39
 
    get_bytes_as('fulltext').
40
 
 
41
 
    :ivar sha1: None, or the sha1 of the content fulltext.
42
 
    :ivar storage_kind: The native storage kind of this factory. Always
43
 
        'fulltext'.
44
 
    :ivar key: The key of this content. Each key is a tuple with a single
45
 
        string in it.
46
 
    :ivar parents: A tuple of parent keys for self.key. If the object has
47
 
        no parent information, None (as opposed to () for an empty list of
48
 
        parents).
49
 
     """
50
 
 
51
 
    def __init__(self, store, path, revision, blob_id):
52
 
        """Create a ContentFactory."""
53
 
        self.store = store
54
 
        self.key = (path, revision)
55
 
        self.storage_kind = 'fulltext'
56
 
        self.parents = None
57
 
        self.blob_id = blob_id
58
 
 
59
 
    def get_bytes_as(self, storage_kind):
60
 
        if storage_kind == 'fulltext':
61
 
            return self.store[self.blob_id].as_raw_string()
62
 
        elif storage_kind == 'chunked':
63
 
            return self.store[self.blob_id].as_raw_chunks()
64
 
        raise UnavailableRepresentation(self.key, storage_kind,
65
 
                                        'fulltext')
66
 
 
67
 
 
68
 
class GitAbsentContentFactory(object):
69
 
    """Absent data content factory.
70
 
 
71
 
    :ivar sha1: None, or the sha1 of the content fulltext.
72
 
    :ivar storage_kind: The native storage kind of this factory. Always
73
 
        'fulltext'.
74
 
    :ivar key: The key of this content. Each key is a tuple with a single
75
 
        string in it.
76
 
    :ivar parents: A tuple of parent keys for self.key. If the object has
77
 
        no parent information, None (as opposed to () for an empty list of
78
 
        parents).
79
 
     """
80
 
 
81
 
    def __init__(self, store, path, revision):
82
 
        """Create a ContentFactory."""
83
 
        self.store = store
84
 
        self.key = (path, revision)
85
 
        self.storage_kind = 'absent'
86
 
        self.parents = None
87
 
 
88
 
    def get_bytes_as(self, storage_kind):
89
 
        raise ValueError
90
 
 
91
 
 
92
 
class AnnotateProvider(object):
93
 
 
94
 
    def __init__(self, change_scanner):
95
 
        self.change_scanner = change_scanner
96
 
        self.store = self.change_scanner.repository._git.object_store
97
 
 
98
 
    def _get_parents(self, path, text_revision):
99
 
        commit_id, mapping = (
100
 
            self.change_scanner.repository.lookup_bzr_revision_id(
101
 
                text_revision))
102
 
        text_parents = []
103
 
        for commit_parent in self.store[commit_id].parents:
104
 
            try:
105
 
                (path, text_parent) = (
106
 
                    self.change_scanner.find_last_change_revision(
107
 
                        path.encode('utf-8'), commit_parent))
108
 
            except KeyError:
109
 
                continue
110
 
            if text_parent not in text_parents:
111
 
                text_parents.append(text_parent)
112
 
        return tuple([
113
 
            (path.decode('utf-8'),
114
 
                self.change_scanner.repository.lookup_foreign_revision_id(p))
115
 
            for p in text_parents])
116
 
 
117
 
    def get_parent_map(self, keys):
118
 
        ret = {}
119
 
        for key in keys:
120
 
            (path, text_revision) = key
121
 
            if text_revision == NULL_REVISION:
122
 
                ret[key] = ()
123
 
                continue
124
 
            try:
125
 
                ret[key] = self._get_parents(path, text_revision)
126
 
            except KeyError:
127
 
                pass
128
 
        return ret
129
 
 
130
 
    def get_record_stream(self, keys, ordering, include_delta_closure):
131
 
        if ordering == 'topological':
132
 
            graph = Graph(self)
133
 
            keys = graph.iter_topo_order(keys)
134
 
        store = self.change_scanner.repository._git.object_store
135
 
        for (path, text_revision) in keys:
136
 
            try:
137
 
                commit_id, mapping = (
138
 
                    self.change_scanner.repository.lookup_bzr_revision_id(
139
 
                        text_revision))
140
 
            except NoSuchRevision:
141
 
                yield GitAbsentContentFactory(store, path, text_revision)
142
 
                continue
143
 
 
144
 
            try:
145
 
                tree_id = store[commit_id].tree
146
 
            except KeyError:
147
 
                yield GitAbsentContentFactory(store, path, text_revision)
148
 
                continue
149
 
            try:
150
 
                (mode, blob_sha) = tree_lookup_path(
151
 
                    store.__getitem__, tree_id, path.encode('utf-8'))
152
 
            except KeyError:
153
 
                yield GitAbsentContentFactory(store, path, text_revision)
154
 
            else:
155
 
                yield GitFulltextContentFactory(
156
 
                    store, path, text_revision, blob_sha)