/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: Jelmer Vernooij
  • Date: 2020-01-30 17:54:42 UTC
  • mto: This revision was merged to the branch mainline in revision 7473.
  • Revision ID: jelmer@jelmer.uk-20200130175442-82mohr4ji3o3mh48
Add VersionedFiles.add_content.

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