/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: Breezy landing bot
  • Author(s): Colin Watson
  • Date: 2020-11-16 21:47:08 UTC
  • mfrom: (7521.1.1 remove-lp-workaround)
  • Revision ID: breezy.the.bot@gmail.com-20201116214708-jos209mgxi41oy15
Remove breezy.git workaround for bazaar.launchpad.net.

Merged from https://code.launchpad.net/~cjwatson/brz/remove-lp-workaround/+merge/393710

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 dulwich.object_store import (
 
20
    tree_lookup_path,
 
21
    )
 
22
 
 
23
from .. import osutils
 
24
from ..bzr.versionedfile import UnavailableRepresentation
 
25
from ..errors import (
 
26
    NoSuchRevision,
 
27
    )
 
28
from ..graph import Graph
 
29
from ..revision import (
 
30
    NULL_REVISION,
 
31
    )
 
32
 
 
33
from .mapping import (
 
34
    decode_git_path,
 
35
    encode_git_path,
 
36
    )
 
37
 
 
38
 
 
39
class GitBlobContentFactory(object):
 
40
    """Static data content factory.
 
41
 
 
42
    This takes a fulltext when created and just returns that during
 
43
    get_bytes_as('fulltext').
 
44
 
 
45
    :ivar sha1: None, or the sha1 of the content fulltext.
 
46
    :ivar storage_kind: The native storage kind of this factory. Always
 
47
        'fulltext'.
 
48
    :ivar key: The key of this content. Each key is a tuple with a single
 
49
        string in it.
 
50
    :ivar parents: A tuple of parent keys for self.key. If the object has
 
51
        no parent information, None (as opposed to () for an empty list of
 
52
        parents).
 
53
     """
 
54
 
 
55
    def __init__(self, store, path, revision, blob_id):
 
56
        """Create a ContentFactory."""
 
57
        self.store = store
 
58
        self.key = (path, revision)
 
59
        self.storage_kind = 'git-blob'
 
60
        self.parents = None
 
61
        self.blob_id = blob_id
 
62
        self.size = None
 
63
 
 
64
    def get_bytes_as(self, storage_kind):
 
65
        if storage_kind == 'fulltext':
 
66
            return self.store[self.blob_id].as_raw_string()
 
67
        elif storage_kind == 'lines':
 
68
            return list(osutils.chunks_to_lines(self.store[self.blob_id].as_raw_chunks()))
 
69
        elif storage_kind == 'chunked':
 
70
            return self.store[self.blob_id].as_raw_chunks()
 
71
        raise UnavailableRepresentation(self.key, storage_kind,
 
72
                                        self.storage_kind)
 
73
 
 
74
    def iter_bytes_as(self, storage_kind):
 
75
        if storage_kind == 'lines':
 
76
            return iter(osutils.chunks_to_lines(self.store[self.blob_id].as_raw_chunks()))
 
77
        elif storage_kind == 'chunked':
 
78
            return iter(self.store[self.blob_id].as_raw_chunks())
 
79
        raise UnavailableRepresentation(self.key, storage_kind,
 
80
                                        self.storage_kind)
 
81
 
 
82
 
 
83
class GitAbsentContentFactory(object):
 
84
    """Absent data content factory.
 
85
 
 
86
    :ivar sha1: None, or the sha1 of the content fulltext.
 
87
    :ivar storage_kind: The native storage kind of this factory. Always
 
88
        'fulltext'.
 
89
    :ivar key: The key of this content. Each key is a tuple with a single
 
90
        string in it.
 
91
    :ivar parents: A tuple of parent keys for self.key. If the object has
 
92
        no parent information, None (as opposed to () for an empty list of
 
93
        parents).
 
94
     """
 
95
 
 
96
    def __init__(self, store, path, revision):
 
97
        """Create a ContentFactory."""
 
98
        self.store = store
 
99
        self.key = (path, revision)
 
100
        self.storage_kind = 'absent'
 
101
        self.parents = None
 
102
        self.size = None
 
103
 
 
104
    def get_bytes_as(self, storage_kind):
 
105
        raise ValueError
 
106
 
 
107
    def iter_bytes_as(self, storage_kind):
 
108
        raise ValueError
 
109
 
 
110
 
 
111
class AnnotateProvider(object):
 
112
 
 
113
    def __init__(self, change_scanner):
 
114
        self.change_scanner = change_scanner
 
115
        self.store = self.change_scanner.repository._git.object_store
 
116
 
 
117
    def _get_parents(self, path, text_revision):
 
118
        commit_id, mapping = (
 
119
            self.change_scanner.repository.lookup_bzr_revision_id(
 
120
                text_revision))
 
121
        text_parents = []
 
122
        path = encode_git_path(path)
 
123
        for commit_parent in self.store[commit_id].parents:
 
124
            try:
 
125
                (path, text_parent) = (
 
126
                    self.change_scanner.find_last_change_revision(
 
127
                        path, commit_parent))
 
128
            except KeyError:
 
129
                continue
 
130
            if text_parent not in text_parents:
 
131
                text_parents.append(text_parent)
 
132
        return tuple([
 
133
            (decode_git_path(path),
 
134
                self.change_scanner.repository.lookup_foreign_revision_id(p))
 
135
            for p in text_parents])
 
136
 
 
137
    def get_parent_map(self, keys):
 
138
        ret = {}
 
139
        for key in keys:
 
140
            (path, text_revision) = key
 
141
            if text_revision == NULL_REVISION:
 
142
                ret[key] = ()
 
143
                continue
 
144
            try:
 
145
                ret[key] = self._get_parents(path, text_revision)
 
146
            except KeyError:
 
147
                pass
 
148
        return ret
 
149
 
 
150
    def get_record_stream(self, keys, ordering, include_delta_closure):
 
151
        if ordering == 'topological':
 
152
            graph = Graph(self)
 
153
            keys = graph.iter_topo_order(keys)
 
154
        store = self.change_scanner.repository._git.object_store
 
155
        for (path, text_revision) in keys:
 
156
            try:
 
157
                commit_id, mapping = (
 
158
                    self.change_scanner.repository.lookup_bzr_revision_id(
 
159
                        text_revision))
 
160
            except NoSuchRevision:
 
161
                yield GitAbsentContentFactory(store, path, text_revision)
 
162
                continue
 
163
 
 
164
            try:
 
165
                tree_id = store[commit_id].tree
 
166
            except KeyError:
 
167
                yield GitAbsentContentFactory(store, path, text_revision)
 
168
                continue
 
169
            try:
 
170
                (mode, blob_sha) = tree_lookup_path(
 
171
                    store.__getitem__, tree_id, encode_git_path(path))
 
172
            except KeyError:
 
173
                yield GitAbsentContentFactory(store, path, text_revision)
 
174
            else:
 
175
                yield GitBlobContentFactory(
 
176
                    store, path, text_revision, blob_sha)