/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
1
# Copyright (C) 2007 Canonical Ltd
0.200.910 by Jelmer Vernooij
update copyright years
2
# Copyright (C) 2009-2010 Jelmer Vernooij <jelmer@samba.org>
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
3
#
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; either version 2 of the License, or
7
# (at your option) any later version.
8
#
9
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
# GNU General Public License for more details.
13
#
14
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
18
"""An adapter between a Git Branch and a Bazaar Branch"""
19
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
20
from collections import defaultdict
21
0.200.261 by Jelmer Vernooij
More formatting fixes.
22
from dulwich.objects import (
23
    Commit,
24
    Tag,
25
    )
0.200.1102 by Jelmer Vernooij
Fix creating of colocated branches in git repositories.
26
from dulwich.protocol import ZERO_SHA
0.200.261 by Jelmer Vernooij
More formatting fixes.
27
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
28
from bzrlib import (
29
    branch,
0.200.513 by Jelmer Vernooij
Fix imports.
30
    bzrdir,
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
31
    config,
0.200.446 by Jelmer Vernooij
Support new 'local' argument.
32
    errors,
0.200.1097 by Jelmer Vernooij
Implement GitBranchFormat.initialize.
33
    repository as _mod_repository,
0.200.19 by John Arbash Meinel
More refactoring. Add some direct tests for GitModel.
34
    revision,
0.200.82 by Jelmer Vernooij
Support listing tags.
35
    tag,
0.230.1 by Jelmer Vernooij
Support lightweight checkouts.
36
    transport,
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
37
    )
0.200.261 by Jelmer Vernooij
More formatting fixes.
38
from bzrlib.decorators import (
39
    needs_read_lock,
40
    )
0.200.969 by Jelmer Vernooij
Use tuples with bzr revid and git sha to avoid lookups.
41
from bzrlib.revision import (
42
    NULL_REVISION,
43
    )
0.200.261 by Jelmer Vernooij
More formatting fixes.
44
from bzrlib.trace import (
0.200.342 by Jelmer Vernooij
Report git sha during pull.
45
    is_quiet,
0.200.261 by Jelmer Vernooij
More formatting fixes.
46
    mutter,
47
    )
48
0.200.386 by Jelmer Vernooij
Move config to a separate file, support BranchConfig.username().
49
from bzrlib.plugins.git.config import (
50
    GitBranchConfig,
51
    )
0.200.278 by Jelmer Vernooij
Update branch head appropriately during dpull.
52
from bzrlib.plugins.git.errors import (
0.200.472 by Jelmer Vernooij
Fix printing error when user attempts to push into git.
53
    NoPushSupport,
0.200.278 by Jelmer Vernooij
Update branch head appropriately during dpull.
54
    NoSuchRef,
55
    )
0.200.872 by Jelmer Vernooij
Move refs code to separate module.
56
from bzrlib.plugins.git.refs import (
0.200.1097 by Jelmer Vernooij
Implement GitBranchFormat.initialize.
57
    branch_name_to_ref,
0.200.1061 by Jelmer Vernooij
Add support for using unpeel map.
58
    extract_tags,
59
    is_tag,
0.200.872 by Jelmer Vernooij
Move refs code to separate module.
60
    ref_to_branch_name,
0.200.1061 by Jelmer Vernooij
Add support for using unpeel map.
61
    ref_to_tag_name,
0.200.875 by Jelmer Vernooij
Use new tag_name_to_ref function.
62
    tag_name_to_ref,
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
63
    UnpeelMap,
0.200.872 by Jelmer Vernooij
Move refs code to separate module.
64
    )
0.200.261 by Jelmer Vernooij
More formatting fixes.
65
0.238.5 by Jelmer Vernooij
Remove old backwards compatibility code.
66
from bzrlib.foreign import ForeignBranch
0.200.388 by Jelmer Vernooij
Support bzr 1.14 as well.
67
0.200.261 by Jelmer Vernooij
More formatting fixes.
68
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
69
class GitPullResult(branch.PullResult):
0.200.956 by Jelmer Vernooij
Add some more format tests.
70
    """Result of a pull from a Git branch."""
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
71
72
    def _lookup_revno(self, revid):
73
        assert isinstance(revid, str), "was %r" % revid
74
        # Try in source branch first, it'll be faster
75
        return self.target_branch.revision_id_to_revno(revid)
76
77
    @property
78
    def old_revno(self):
79
        return self._lookup_revno(self.old_revid)
80
81
    @property
82
    def new_revno(self):
83
        return self._lookup_revno(self.new_revid)
84
85
0.200.1064 by Jelmer Vernooij
Use common base class for tags.
86
class GitTags(tag.BasicTags):
87
    """Ref-based tag dictionary."""
0.200.82 by Jelmer Vernooij
Support listing tags.
88
0.200.89 by Jelmer Vernooij
Support sprouting branches.
89
    def __init__(self, branch):
90
        self.branch = branch
91
        self.repository = branch.repository
0.200.82 by Jelmer Vernooij
Support listing tags.
92
0.200.1066 by Jelmer Vernooij
Add GitTags.get_refs.
93
    def get_refs(self):
94
        raise NotImplementedError(self.get_refs)
95
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
96
    def _iter_tag_refs(self, refs):
97
        raise NotImplementedError(self._iter_tag_refs)
98
99
    def _merge_to_git(self, to_tags, refs, overwrite=False):
100
        target_repo = to_tags.repository
101
        conflicts = []
102
        for k, v in refs.iteritems():
103
            if not is_tag(k):
104
                continue
0.200.1128 by Jelmer Vernooij
remove duplicate definition of _matchingbzrdir.
105
            if overwrite or not k in target_repo._git.refs:
106
                target_repo._git.refs[k] = v
107
            elif target_repo._git.refs[k] == v:
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
108
                pass
109
            else:
110
                conflicts.append((ref_to_tag_name(k), v, target_repo.refs[k]))
111
        return conflicts
112
113
    def _merge_to_non_git(self, to_tags, refs, overwrite=False):
114
        unpeeled_map = defaultdict(set)
115
        conflicts = []
116
        result = dict(to_tags.get_tag_dict())
117
        for n, peeled, unpeeled, bzr_revid in self._iter_tag_refs(refs):
118
            if unpeeled is not None:
119
                unpeeled_map[peeled].add(unpeeled)
120
            if n not in result or overwrite:
121
                result[n] = bzr_revid
122
            elif result[n] == bzr_revid:
123
                pass
124
            else:
125
                conflicts.append((n, result[n], bzr_revid))
126
        to_tags._set_tag_dict(result)
127
        if len(unpeeled_map) > 0:
128
            map_file = UnpeelMap.from_repository(to_tags.branch.repository)
129
            map_file.update(unpeeled_map)
130
            map_file.save_in_repository(to_tags.branch.repository)
131
        return conflicts
132
133
    def merge_to(self, to_tags, overwrite=False, ignore_master=False,
134
                 source_refs=None):
0.200.1113 by Jelmer Vernooij
Fix Tags.merge_to.
135
        """See Tags.merge_to."""
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
136
        if source_refs is None:
0.200.1066 by Jelmer Vernooij
Add GitTags.get_refs.
137
            source_refs = self.get_refs()
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
138
        if self == to_tags:
139
            return
140
        if isinstance(to_tags, GitTags):
141
            return self._merge_to_git(to_tags, source_refs,
142
                                      overwrite=overwrite)
143
        else:
144
            if ignore_master:
145
                master = None
146
            else:
147
                master = to_tags.branch.get_master_branch()
148
            conflicts = self._merge_to_non_git(to_tags, source_refs,
149
                                              overwrite=overwrite)
150
            if master is not None:
0.200.1113 by Jelmer Vernooij
Fix Tags.merge_to.
151
                conflicts += self.merge_to(master.tags, overwrite=overwrite,
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
152
                                           source_refs=source_refs,
153
                                           ignore_master=ignore_master)
154
            return conflicts
155
156
    def get_tag_dict(self):
157
        ret = {}
0.200.1066 by Jelmer Vernooij
Add GitTags.get_refs.
158
        refs = self.get_refs()
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
159
        for (name, peeled, unpeeled, bzr_revid) in self._iter_tag_refs(refs):
160
            ret[name] = bzr_revid
161
        return ret
162
0.200.1064 by Jelmer Vernooij
Use common base class for tags.
163
164
class LocalGitTagDict(GitTags):
165
    """Dictionary with tags in a local repository."""
166
167
    def __init__(self, branch):
168
        super(LocalGitTagDict, self).__init__(branch)
169
        self.refs = self.repository._git.refs
170
0.200.1066 by Jelmer Vernooij
Add GitTags.get_refs.
171
    def get_refs(self):
172
        return self.repository._git.get_refs()
173
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
174
    def _iter_tag_refs(self, refs):
175
        """Iterate over the tag refs.
176
177
        :param refs: Refs dictionary (name -> git sha1)
178
        :return: iterator over (name, peeled_sha1, unpeeled_sha1, bzr_revid)
179
        """
0.200.1060 by Jelmer Vernooij
Return unpeeled tags in extract_tags.
180
        for k, (peeled, unpeeled) in extract_tags(refs).iteritems():
0.200.609 by Jelmer Vernooij
Cope with tags pointing at nonexisting objects.
181
            try:
0.200.1060 by Jelmer Vernooij
Return unpeeled tags in extract_tags.
182
                obj = self.repository._git[peeled]
0.200.609 by Jelmer Vernooij
Cope with tags pointing at nonexisting objects.
183
            except KeyError:
0.200.1060 by Jelmer Vernooij
Return unpeeled tags in extract_tags.
184
                mutter("Tag %s points at unknown object %s, ignoring", peeled,
185
                       obj)
0.200.609 by Jelmer Vernooij
Cope with tags pointing at nonexisting objects.
186
                continue
0.200.1060 by Jelmer Vernooij
Return unpeeled tags in extract_tags.
187
            # FIXME: this shouldn't really be necessary, the repository
188
            # already should have these unpeeled.
0.200.194 by Jelmer Vernooij
Look for commit object in heavyweight tags.
189
            while isinstance(obj, Tag):
0.200.1060 by Jelmer Vernooij
Return unpeeled tags in extract_tags.
190
                peeled = obj.object[1]
191
                obj = self.repository._git[peeled]
0.200.194 by Jelmer Vernooij
Look for commit object in heavyweight tags.
192
            if not isinstance(obj, Commit):
0.200.261 by Jelmer Vernooij
More formatting fixes.
193
                mutter("Tag %s points at object %r that is not a commit, "
194
                       "ignoring", k, obj)
0.200.194 by Jelmer Vernooij
Look for commit object in heavyweight tags.
195
                continue
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
196
            yield (k, peeled, unpeeled,
197
                   self.branch.lookup_foreign_revision_id(peeled))
198
0.200.711 by Jelmer Vernooij
Support merging tags to a local Git repository.
199
    def _set_tag_dict(self, to_dict):
0.200.1066 by Jelmer Vernooij
Add GitTags.get_refs.
200
        extra = set(self.get_refs().keys())
0.200.711 by Jelmer Vernooij
Support merging tags to a local Git repository.
201
        for k, revid in to_dict.iteritems():
0.200.875 by Jelmer Vernooij
Use new tag_name_to_ref function.
202
            name = tag_name_to_ref(k)
0.200.711 by Jelmer Vernooij
Support merging tags to a local Git repository.
203
            if name in extra:
204
                extra.remove(name)
205
            self.set_tag(k, revid)
206
        for name in extra:
0.200.1061 by Jelmer Vernooij
Add support for using unpeel map.
207
            if is_tag(name):
0.200.711 by Jelmer Vernooij
Support merging tags to a local Git repository.
208
                del self.repository._git[name]
0.200.956 by Jelmer Vernooij
Add some more format tests.
209
0.200.86 by Jelmer Vernooij
Clearer error when setting tags.
210
    def set_tag(self, name, revid):
0.200.1064 by Jelmer Vernooij
Use common base class for tags.
211
        self.refs[tag_name_to_ref(name)], _ = \
0.200.1030 by Jelmer Vernooij
More work on supporting roundtripping push.
212
            self.branch.lookup_bzr_revision_id(revid)
0.200.86 by Jelmer Vernooij
Clearer error when setting tags.
213
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
214
0.200.1078 by Jelmer Vernooij
Fix git-import from remote repositories.
215
class DictTagDict(tag.BasicTags):
0.239.1 by Jelmer Vernooij
Avoid re-connecting to fetch tags we already know.
216
217
    def __init__(self, branch, tags):
218
        super(DictTagDict, self).__init__(branch)
219
        self._tags = tags
220
221
    def get_tag_dict(self):
222
        return self._tags
223
224
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
225
class GitBranchFormat(branch.BranchFormat):
226
0.200.70 by Jelmer Vernooij
Implement GitBranchFormat.get_format_description.
227
    def get_format_description(self):
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
228
        return 'Git Branch'
229
0.243.1 by Jelmer Vernooij
Use foreign branch testing infrastructure.
230
    def network_name(self):
231
        return "git"
232
0.200.82 by Jelmer Vernooij
Support listing tags.
233
    def supports_tags(self):
234
        return True
235
0.200.1105 by Jelmer Vernooij
Don't claim to support leaving locks.
236
    def supports_leaving_lock(self):
237
        return False
238
0.200.1091 by Jelmer Vernooij
Provide _matchingbzrdir for testing.
239
    @property
240
    def _matchingbzrdir(self):
0.200.1140 by Jelmer Vernooij
Update now that the control dir formats are no longer in __init__.
241
        from bzrlib.plugins.git.dir import LocalGitControlDirFormat
0.200.1091 by Jelmer Vernooij
Provide _matchingbzrdir for testing.
242
        return LocalGitControlDirFormat()
243
0.243.1 by Jelmer Vernooij
Use foreign branch testing infrastructure.
244
    def get_foreign_tests_branch_factory(self):
245
        from bzrlib.plugins.git.tests.test_branch import ForeignTestsBranchFactory
246
        return ForeignTestsBranchFactory()
247
0.200.246 by Jelmer Vernooij
Cope with API changes in 1.13.
248
    def make_tags(self, branch):
0.228.3 by Jelmer Vernooij
Fix tags when fetching from remotes.
249
        if getattr(branch.repository, "get_refs", None) is not None:
250
            from bzrlib.plugins.git.remote import RemoteGitTagDict
251
            return RemoteGitTagDict(branch)
0.200.261 by Jelmer Vernooij
More formatting fixes.
252
        else:
253
            return LocalGitTagDict(branch)
0.200.246 by Jelmer Vernooij
Cope with API changes in 1.13.
254
0.200.1097 by Jelmer Vernooij
Implement GitBranchFormat.initialize.
255
    def initialize(self, a_bzrdir, name=None, repository=None):
256
        from bzrlib.plugins.git.dir import LocalGitDir
257
        if not isinstance(a_bzrdir, LocalGitDir):
258
            raise errors.IncompatibleFormat(self, a_bzrdir._format)
259
        if repository is None:
260
            repository = a_bzrdir.open_repository()
0.200.1102 by Jelmer Vernooij
Fix creating of colocated branches in git repositories.
261
        ref = branch_name_to_ref(name, "HEAD")
262
        repository._git[ref] = ZERO_SHA
263
        return LocalGitBranch(a_bzrdir, repository, ref, a_bzrdir._lockfiles)
0.200.1097 by Jelmer Vernooij
Implement GitBranchFormat.initialize.
264
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
265
0.200.911 by Jelmer Vernooij
Cope with locking changes in bzr.dev.
266
class GitReadLock(object):
267
268
    def __init__(self, unlock):
269
        self.unlock = unlock
270
271
272
class GitWriteLock(object):
273
274
    def __init__(self, unlock):
275
        self.unlock = unlock
276
277
0.200.388 by Jelmer Vernooij
Support bzr 1.14 as well.
278
class GitBranch(ForeignBranch):
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
279
    """An adapter to git repositories for bzr Branch objects."""
280
0.200.1129 by Jelmer Vernooij
Implement GitBranch.control_transport.
281
    @property
282
    def control_transport(self):
283
        return self.bzrdir.control_transport
284
0.200.770 by Jelmer Vernooij
Proper branch names.
285
    def __init__(self, bzrdir, repository, ref, lockfiles, tagsdict=None):
0.200.82 by Jelmer Vernooij
Support listing tags.
286
        self.repository = repository
0.200.246 by Jelmer Vernooij
Cope with API changes in 1.13.
287
        self._format = GitBranchFormat()
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
288
        self.control_files = lockfiles
0.200.59 by Jelmer Vernooij
Add more tests, fix revision history.
289
        self.bzrdir = bzrdir
0.231.1 by Jelmer Vernooij
Check that regenerated objects have the expected sha1.
290
        super(GitBranch, self).__init__(repository.get_mapping())
0.239.1 by Jelmer Vernooij
Avoid re-connecting to fetch tags we already know.
291
        if tagsdict is not None:
292
            self.tags = DictTagDict(self, tagsdict)
0.200.770 by Jelmer Vernooij
Proper branch names.
293
        self.ref = ref
294
        self.name = ref_to_branch_name(ref)
0.200.461 by Jelmer Vernooij
Reduce number of round trips when fetching from Git.
295
        self._head = None
0.200.630 by Jelmer Vernooij
Fix base url of Git branches - use the working tree path rather than the control directory path.
296
        self.base = bzrdir.root_transport.base
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
297
0.239.8 by Jelmer Vernooij
Support checkouts.
298
    def _get_checkout_format(self):
299
        """Return the most suitable metadir for a checkout of this branch.
300
        Weaves are used if this branch's repository uses weaves.
301
        """
0.200.927 by Jelmer Vernooij
Remove explicit use of rich root formats.
302
        return bzrdir.format_registry.make_bzrdir("default")
0.239.8 by Jelmer Vernooij
Support checkouts.
303
0.238.3 by Jelmer Vernooij
Remove svn references, prefer git send format when submitting changes against a git branch.
304
    def get_child_submit_format(self):
305
        """Return the preferred format of submissions to this branch."""
306
        ret = self.get_config().get_user_option("child_submit_format")
307
        if ret is not None:
308
            return ret
309
        return "git"
310
0.200.293 by Jelmer Vernooij
Fix branch nicks.
311
    def _get_nick(self, local=False, possible_master_transports=None):
312
        """Find the nick name for this branch.
313
314
        :return: Branch nick
315
        """
0.200.920 by Jelmer Vernooij
Fix some more tests.
316
        return self.name or "HEAD"
0.200.293 by Jelmer Vernooij
Fix branch nicks.
317
0.200.331 by Jelmer Vernooij
Add stub for setting nick function.
318
    def _set_nick(self, nick):
319
        raise NotImplementedError
320
321
    nick = property(_get_nick, _set_nick)
0.200.293 by Jelmer Vernooij
Fix branch nicks.
322
0.200.412 by Jelmer Vernooij
Implement GitBranch.__repr__.
323
    def __repr__(self):
0.200.770 by Jelmer Vernooij
Proper branch names.
324
        return "<%s(%r, %r)>" % (self.__class__.__name__, self.repository.base,
0.200.920 by Jelmer Vernooij
Fix some more tests.
325
            self.ref or "HEAD")
0.200.412 by Jelmer Vernooij
Implement GitBranch.__repr__.
326
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
327
    def generate_revision_history(self, revid, old_revid=None):
0.200.1103 by Jelmer Vernooij
Support generate_revision_history(NULL_REVISION).
328
        if revid == NULL_REVISION:
329
            newhead = ZERO_SHA
330
        else:
331
            # FIXME: Check that old_revid is in the ancestry of revid
332
            newhead, self.mapping = self.mapping.revision_id_bzr_to_foreign(revid)
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
333
        self._set_head(newhead)
334
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
335
    def lock_write(self):
336
        self.control_files.lock_write()
0.200.911 by Jelmer Vernooij
Cope with locking changes in bzr.dev.
337
        return GitWriteLock(self.unlock)
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
338
0.200.139 by Jelmer Vernooij
Share more code between local and remote classes, support opening remote branches.
339
    def get_stacked_on_url(self):
340
        # Git doesn't do stacking (yet...)
0.200.631 by Jelmer Vernooij
Raise proper exception in Branch.get_stacked_on_url().
341
        raise errors.UnstackableBranchFormat(self._format, self.base)
0.200.139 by Jelmer Vernooij
Share more code between local and remote classes, support opening remote branches.
342
343
    def get_parent(self):
344
        """See Branch.get_parent()."""
0.200.312 by Jelmer Vernooij
Add notes about parent locations.
345
        # FIXME: Set "origin" url from .git/config ?
0.200.139 by Jelmer Vernooij
Share more code between local and remote classes, support opening remote branches.
346
        return None
347
0.200.175 by Jelmer Vernooij
Add optimized handling when fetching from git to git.
348
    def set_parent(self, url):
0.200.312 by Jelmer Vernooij
Add notes about parent locations.
349
        # FIXME: Set "origin" url in .git/config ?
0.200.175 by Jelmer Vernooij
Add optimized handling when fetching from git to git.
350
        pass
351
0.200.139 by Jelmer Vernooij
Share more code between local and remote classes, support opening remote branches.
352
    def lock_read(self):
353
        self.control_files.lock_read()
0.200.911 by Jelmer Vernooij
Cope with locking changes in bzr.dev.
354
        return GitReadLock(self.unlock)
0.200.139 by Jelmer Vernooij
Share more code between local and remote classes, support opening remote branches.
355
0.200.432 by Jelmer Vernooij
Support Branch.is_locked, required for loggerhead.
356
    def is_locked(self):
357
        return self.control_files.is_locked()
358
0.200.139 by Jelmer Vernooij
Share more code between local and remote classes, support opening remote branches.
359
    def unlock(self):
360
        self.control_files.unlock()
361
362
    def get_physical_lock_status(self):
363
        return False
364
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
365
    @needs_read_lock
366
    def last_revision(self):
367
        # perhaps should escape this ?
0.200.57 by Jelmer Vernooij
Fix more tests.
368
        if self.head is None:
0.200.19 by John Arbash Meinel
More refactoring. Add some direct tests for GitModel.
369
            return revision.NULL_REVISION
0.252.44 by Jelmer Vernooij
Properly look up Bazaar revision ids for revision parents in case they are round-tripped.
370
        return self.lookup_foreign_revision_id(self.head)
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
371
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
372
    def _basic_push(self, target, overwrite=False, stop_revision=None):
373
        return branch.InterBranch.get(self, target)._basic_push(
374
            overwrite, stop_revision)
375
0.252.49 by Jelmer Vernooij
Avoid trying to set HEAD for remote branches.
376
    def lookup_foreign_revision_id(self, foreign_revid):
0.200.956 by Jelmer Vernooij
Add some more format tests.
377
        return self.repository.lookup_foreign_revision_id(foreign_revid,
0.252.49 by Jelmer Vernooij
Avoid trying to set HEAD for remote branches.
378
            self.mapping)
379
0.200.1030 by Jelmer Vernooij
More work on supporting roundtripping push.
380
    def lookup_bzr_revision_id(self, revid):
381
        return self.repository.lookup_bzr_revision_id(
382
            revid, mapping=self.mapping)
383
0.200.692 by Jelmer Vernooij
Refuse pulling into non-rich-root branches rather than erroring out with an AttributeError.
384
0.200.465 by Jelmer Vernooij
Use dulwich standard functionality for finding missing revisions.
385
class LocalGitBranch(GitBranch):
386
    """A local Git branch."""
387
0.200.1102 by Jelmer Vernooij
Fix creating of colocated branches in git repositories.
388
    def __init__(self, bzrdir, repository, ref, lockfiles, tagsdict=None):
389
        super(LocalGitBranch, self).__init__(bzrdir, repository, ref,
0.200.763 by Jelmer Vernooij
Provide proper colocated branch support.
390
              lockfiles, tagsdict)
0.200.918 by Jelmer Vernooij
Cope with 'self.ref is None' in a couple more places.
391
        refs = repository._git.get_refs()
0.200.1102 by Jelmer Vernooij
Fix creating of colocated branches in git repositories.
392
        if not (ref in refs.keys() or "HEAD" in refs.keys()):
0.200.763 by Jelmer Vernooij
Provide proper colocated branch support.
393
            raise errors.NotBranchError(self.base)
394
0.200.261 by Jelmer Vernooij
More formatting fixes.
395
    def create_checkout(self, to_location, revision_id=None, lightweight=False,
396
        accelerator_tree=None, hardlink=False):
0.200.210 by Jelmer Vernooij
properly error out about not support lightweight checkouts.
397
        if lightweight:
0.230.1 by Jelmer Vernooij
Support lightweight checkouts.
398
            t = transport.get_transport(to_location)
399
            t.ensure_base()
400
            format = self._get_checkout_format()
401
            checkout = format.initialize_on_transport(t)
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
402
            from_branch = branch.BranchReferenceFormat().initialize(checkout,
0.230.1 by Jelmer Vernooij
Support lightweight checkouts.
403
                self)
404
            tree = checkout.create_workingtree(revision_id,
405
                from_branch=from_branch, hardlink=hardlink)
406
            return tree
407
        else:
408
            return self._create_heavyweight_checkout(to_location, revision_id,
0.257.1 by Jelmer Vernooij
use transport repo objects even for local access.
409
                hardlink)
0.200.210 by Jelmer Vernooij
properly error out about not support lightweight checkouts.
410
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
411
    def _create_heavyweight_checkout(self, to_location, revision_id=None,
0.200.210 by Jelmer Vernooij
properly error out about not support lightweight checkouts.
412
                                     hardlink=False):
413
        """Create a new heavyweight checkout of this branch.
414
415
        :param to_location: URL of location to create the new checkout in.
416
        :param revision_id: Revision that should be the tip of the checkout.
417
        :param hardlink: Whether to hardlink
418
        :return: WorkingTree object of checkout.
419
        """
0.200.513 by Jelmer Vernooij
Fix imports.
420
        checkout_branch = bzrdir.BzrDir.create_branch_convenience(
0.200.927 by Jelmer Vernooij
Remove explicit use of rich root formats.
421
            to_location, force_new_tree=False)
0.200.210 by Jelmer Vernooij
properly error out about not support lightweight checkouts.
422
        checkout = checkout_branch.bzrdir
423
        checkout_branch.bind(self)
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
424
        # pull up to the specified revision_id to set the initial
0.200.210 by Jelmer Vernooij
properly error out about not support lightweight checkouts.
425
        # branch tip correctly, and seed it with history.
426
        checkout_branch.pull(self, stop_revision=revision_id)
427
        return checkout.create_workingtree(revision_id, hardlink=hardlink)
428
0.200.57 by Jelmer Vernooij
Fix more tests.
429
    def _gen_revision_history(self):
0.200.58 by Jelmer Vernooij
Fix remaining tests.
430
        if self.head is None:
431
            return []
0.200.261 by Jelmer Vernooij
More formatting fixes.
432
        ret = list(self.repository.iter_reverse_revision_history(
433
            self.last_revision()))
0.200.59 by Jelmer Vernooij
Add more tests, fix revision history.
434
        ret.reverse()
0.200.57 by Jelmer Vernooij
Fix more tests.
435
        return ret
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
436
0.200.461 by Jelmer Vernooij
Reduce number of round trips when fetching from Git.
437
    def _get_head(self):
0.200.480 by Jelmer Vernooij
Cope with API changes in Dulwich.
438
        try:
0.200.918 by Jelmer Vernooij
Cope with 'self.ref is None' in a couple more places.
439
            return self.repository._git.ref(self.ref or "HEAD")
0.200.480 by Jelmer Vernooij
Cope with API changes in Dulwich.
440
        except KeyError:
441
            return None
0.200.461 by Jelmer Vernooij
Reduce number of round trips when fetching from Git.
442
0.200.507 by Jelmer Vernooij
Implement set_last_revision{_info,}.
443
    def set_last_revision_info(self, revno, revid):
444
        self.set_last_revision(revid)
445
446
    def set_last_revision(self, revid):
0.252.5 by Jelmer Vernooij
enable 'bzr push'.
447
        (newhead, self.mapping) = self.repository.lookup_bzr_revision_id(revid)
0.200.523 by Jelmer Vernooij
Fix undefined error.
448
        self.head = newhead
0.200.507 by Jelmer Vernooij
Implement set_last_revision{_info,}.
449
0.200.461 by Jelmer Vernooij
Reduce number of round trips when fetching from Git.
450
    def _set_head(self, value):
451
        self._head = value
0.200.918 by Jelmer Vernooij
Cope with 'self.ref is None' in a couple more places.
452
        self.repository._git.refs[self.ref or "HEAD"] = self._head
0.200.461 by Jelmer Vernooij
Reduce number of round trips when fetching from Git.
453
        self._clear_cached_state()
454
455
    head = property(_get_head, _set_head)
456
0.200.18 by John Arbash Meinel
Start splitting up the Git{Branch,Dir,Repository} into separate modules, etc.
457
    def get_config(self):
458
        return GitBranchConfig(self)
459
460
    def get_push_location(self):
461
        """See Branch.get_push_location."""
462
        push_loc = self.get_config().get_user_option('push_location')
463
        return push_loc
464
465
    def set_push_location(self, location):
466
        """See Branch.set_push_location."""
0.200.19 by John Arbash Meinel
More refactoring. Add some direct tests for GitModel.
467
        self.get_config().set_user_option('push_location', location,
0.217.54 by John Carr
set_user_option breaks - doesnt have a local option in BranchConfig. Follow the bzr.dev syntax instead.
468
                                          store=config.STORE_LOCATION)
0.200.43 by David Allouche
Ultra-experimental support for "bzr pull". No test. No sanity.
469
470
    def supports_tags(self):
0.200.82 by Jelmer Vernooij
Support listing tags.
471
        return True
0.200.956 by Jelmer Vernooij
Add some more format tests.
472
0.200.225 by Jelmer Vernooij
Implement custom InterBranch to support fetching from remote git branches.
473
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
474
def _quick_lookup_revno(local_branch, remote_branch, revid):
475
    assert isinstance(revid, str), "was %r" % revid
476
    # Try in source branch first, it'll be faster
477
    try:
478
        return local_branch.revision_id_to_revno(revid)
479
    except errors.NoSuchRevision:
480
        graph = local_branch.repository.get_graph()
481
        try:
482
            return graph.find_distance_to_null(revid)
483
        except errors.GhostRevisionsHaveNoRevno:
484
            # FIXME: Check using graph.find_distance_to_null() ?
485
            return remote_branch.revision_id_to_revno(revid)
486
487
0.200.342 by Jelmer Vernooij
Report git sha during pull.
488
class GitBranchPullResult(branch.PullResult):
489
0.252.36 by Jelmer Vernooij
Fix pull.
490
    def __init__(self):
491
        super(GitBranchPullResult, self).__init__()
492
        self.new_git_head = None
493
        self._old_revno = None
494
        self._new_revno = None
495
0.200.342 by Jelmer Vernooij
Report git sha during pull.
496
    def report(self, to_file):
497
        if not is_quiet():
498
            if self.old_revid == self.new_revid:
499
                to_file.write('No revisions to pull.\n')
0.200.728 by Jelmer Vernooij
Fix pulling when all revisions are already in the repo.
500
            elif self.new_git_head is not None:
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
501
                to_file.write('Now on revision %d (git sha: %s).\n' %
0.200.342 by Jelmer Vernooij
Report git sha during pull.
502
                        (self.new_revno, self.new_git_head))
0.200.728 by Jelmer Vernooij
Fix pulling when all revisions are already in the repo.
503
            else:
504
                to_file.write('Now on revision %d.\n' % (self.new_revno,))
0.200.342 by Jelmer Vernooij
Report git sha during pull.
505
        self._show_tag_conficts(to_file)
506
0.252.36 by Jelmer Vernooij
Fix pull.
507
    def _lookup_revno(self, revid):
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
508
        return _quick_lookup_revno(self.target_branch, self.source_branch, revid)
0.252.36 by Jelmer Vernooij
Fix pull.
509
510
    def _get_old_revno(self):
511
        if self._old_revno is not None:
512
            return self._old_revno
513
        return self._lookup_revno(self.old_revid)
514
515
    def _set_old_revno(self, revno):
516
        self._old_revno = revno
517
518
    old_revno = property(_get_old_revno, _set_old_revno)
519
520
    def _get_new_revno(self):
521
        if self._new_revno is not None:
522
            return self._new_revno
523
        return self._lookup_revno(self.new_revid)
524
525
    def _set_new_revno(self, revno):
526
        self._new_revno = revno
0.200.956 by Jelmer Vernooij
Add some more format tests.
527
0.252.36 by Jelmer Vernooij
Fix pull.
528
    new_revno = property(_get_new_revno, _set_new_revno)
529
0.200.342 by Jelmer Vernooij
Report git sha during pull.
530
0.200.504 by Jelmer Vernooij
Lazily find revno's for git branches.
531
class GitBranchPushResult(branch.BranchPushResult):
532
533
    def _lookup_revno(self, revid):
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
534
        return _quick_lookup_revno(self.source_branch, self.target_branch, revid)
0.200.504 by Jelmer Vernooij
Lazily find revno's for git branches.
535
536
    @property
537
    def old_revno(self):
538
        return self._lookup_revno(self.old_revid)
539
540
    @property
541
    def new_revno(self):
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
542
        new_original_revno = getattr(self, "new_original_revno", None)
543
        if new_original_revno:
544
            return new_original_revno
545
        if getattr(self, "new_original_revid", None) is not None:
546
            return self._lookup_revno(self.new_original_revid)
0.200.504 by Jelmer Vernooij
Lazily find revno's for git branches.
547
        return self._lookup_revno(self.new_revid)
548
549
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
550
class InterFromGitBranch(branch.GenericInterBranch):
0.200.261 by Jelmer Vernooij
More formatting fixes.
551
    """InterBranch implementation that pulls from Git into bzr."""
0.200.225 by Jelmer Vernooij
Implement custom InterBranch to support fetching from remote git branches.
552
0.200.996 by Jelmer Vernooij
Fix test run of InterBranches.
553
    @staticmethod
554
    def _get_branch_formats_to_test():
0.200.1100 by Jelmer Vernooij
Provide test combinations for InterBranch implementations.
555
        try:
556
            default_format = branch.format_registry.get_default()
557
        except AttributeError:
558
            default_format = branch.BranchFormat._default_format
559
        return [
560
            (GitBranchFormat(), GitBranchFormat()),
561
            (GitBranchFormat(), default_format)]
0.200.996 by Jelmer Vernooij
Fix test run of InterBranches.
562
0.200.225 by Jelmer Vernooij
Implement custom InterBranch to support fetching from remote git branches.
563
    @classmethod
0.200.692 by Jelmer Vernooij
Refuse pulling into non-rich-root branches rather than erroring out with an AttributeError.
564
    def _get_interrepo(self, source, target):
0.200.1097 by Jelmer Vernooij
Implement GitBranchFormat.initialize.
565
        return _mod_repository.InterRepository.get(source.repository, target.repository)
0.200.692 by Jelmer Vernooij
Refuse pulling into non-rich-root branches rather than erroring out with an AttributeError.
566
567
    @classmethod
568
    def is_compatible(cls, source, target):
569
        return (isinstance(source, GitBranch) and
570
                not isinstance(target, GitBranch) and
571
                (getattr(cls._get_interrepo(source, target), "fetch_objects", None) is not None))
0.200.225 by Jelmer Vernooij
Implement custom InterBranch to support fetching from remote git branches.
572
0.247.7 by Michael Hudson
preserve the interface of update_revisions()
573
    def _update_revisions(self, stop_revision=None, overwrite=False,
0.247.2 by Michael Hudson
this works for my tests, but i'm pretty sure it's wrong in general
574
        graph=None, limit=None):
0.247.7 by Michael Hudson
preserve the interface of update_revisions()
575
        """Like InterBranch.update_revisions(), but with additions.
576
577
        Compared to the `update_revisions()` below, this function takes a
578
        `limit` argument that limits how many git commits will be converted
0.200.1062 by Jelmer Vernooij
Pass remote refs along in _update_revisions.
579
        and returns the new git head and remote refs.
0.247.7 by Michael Hudson
preserve the interface of update_revisions()
580
        """
0.200.692 by Jelmer Vernooij
Refuse pulling into non-rich-root branches rather than erroring out with an AttributeError.
581
        interrepo = self._get_interrepo(self.source, self.target)
0.200.225 by Jelmer Vernooij
Implement custom InterBranch to support fetching from remote git branches.
582
        def determine_wants(heads):
0.200.917 by Jelmer Vernooij
Cope with implicit branches during pull.
583
            if self.source.ref is not None and not self.source.ref in heads:
0.200.777 by Jelmer Vernooij
Fix colocated remote branches.
584
                raise NoSuchRef(self.source.ref, heads.keys())
0.200.314 by Jelmer Vernooij
Support stop_revision.
585
            if stop_revision is not None:
0.200.728 by Jelmer Vernooij
Fix pulling when all revisions are already in the repo.
586
                self._last_revid = stop_revision
0.200.912 by Jelmer Vernooij
Merge roundtrip support.
587
                head, mapping = self.source.repository.lookup_bzr_revision_id(
0.200.316 by Jelmer Vernooij
Fix formatting.
588
                    stop_revision)
0.200.314 by Jelmer Vernooij
Support stop_revision.
589
            else:
0.200.917 by Jelmer Vernooij
Cope with implicit branches during pull.
590
                if self.source.ref is not None:
591
                    head = heads[self.source.ref]
592
                else:
593
                    head = heads["HEAD"]
0.252.44 by Jelmer Vernooij
Properly look up Bazaar revision ids for revision parents in case they are round-tripped.
594
                self._last_revid = self.source.lookup_foreign_revision_id(head)
0.200.728 by Jelmer Vernooij
Fix pulling when all revisions are already in the repo.
595
            if self.target.repository.has_revision(self._last_revid):
0.200.225 by Jelmer Vernooij
Implement custom InterBranch to support fetching from remote git branches.
596
                return []
0.247.6 by Michael Hudson
away with underscore prefixed local variables
597
            return [head]
0.200.1002 by Jelmer Vernooij
Fix regression in git-import.
598
        pack_hint, head, refs = interrepo.fetch_objects(
0.247.2 by Michael Hudson
this works for my tests, but i'm pretty sure it's wrong in general
599
            determine_wants, self.source.mapping, limit=limit)
0.252.45 by Jelmer Vernooij
Finish fetching roundtripped revisions back into bzr.
600
        if (pack_hint is not None and
601
            self.target.repository._format.pack_compresses):
0.248.5 by Jelmer Vernooij
Reformatting, fix dpush.
602
            self.target.repository.pack(hint=pack_hint)
0.200.728 by Jelmer Vernooij
Fix pulling when all revisions are already in the repo.
603
        if head is not None:
0.252.44 by Jelmer Vernooij
Properly look up Bazaar revision ids for revision parents in case they are round-tripped.
604
            self._last_revid = self.source.lookup_foreign_revision_id(head)
0.200.313 by Jelmer Vernooij
Support overwrite parameter.
605
        if overwrite:
0.200.314 by Jelmer Vernooij
Support stop_revision.
606
            prev_last_revid = None
0.200.313 by Jelmer Vernooij
Support overwrite parameter.
607
        else:
0.200.314 by Jelmer Vernooij
Support stop_revision.
608
            prev_last_revid = self.target.last_revision()
0.248.5 by Jelmer Vernooij
Reformatting, fix dpush.
609
        self.target.generate_revision_history(self._last_revid,
610
            prev_last_revid)
0.200.1062 by Jelmer Vernooij
Pass remote refs along in _update_revisions.
611
        return head, refs
0.200.225 by Jelmer Vernooij
Implement custom InterBranch to support fetching from remote git branches.
612
0.200.726 by Jelmer Vernooij
Factor out conversion of branch names to refs.
613
    def update_revisions(self, stop_revision=None, overwrite=False,
614
                         graph=None):
0.247.7 by Michael Hudson
preserve the interface of update_revisions()
615
        """See InterBranch.update_revisions()."""
616
        self._update_revisions(stop_revision, overwrite, graph)
617
0.200.338 by Jelmer Vernooij
Fix dpushing without changes necessary.
618
    def pull(self, overwrite=False, stop_revision=None,
619
             possible_transports=None, _hook_master=None, run_hooks=True,
0.247.2 by Michael Hudson
this works for my tests, but i'm pretty sure it's wrong in general
620
             _override_hook_target=None, local=False, limit=None):
0.200.338 by Jelmer Vernooij
Fix dpushing without changes necessary.
621
        """See Branch.pull.
622
623
        :param _hook_master: Private parameter - set the branch to
624
            be supplied as the master to pull hooks.
625
        :param run_hooks: Private parameter - if false, this branch
626
            is being called because it's the master of the primary branch,
627
            so it should not run its hooks.
628
        :param _override_hook_target: Private parameter - set the branch to be
629
            supplied as the target_branch to pull hooks.
0.247.2 by Michael Hudson
this works for my tests, but i'm pretty sure it's wrong in general
630
        :param limit: Only import this many revisons.  `None`, the default,
631
            means import all revisions.
0.200.338 by Jelmer Vernooij
Fix dpushing without changes necessary.
632
        """
0.200.446 by Jelmer Vernooij
Support new 'local' argument.
633
        # This type of branch can't be bound.
634
        if local:
635
            raise errors.LocalRequiresBoundBranch()
0.200.342 by Jelmer Vernooij
Report git sha during pull.
636
        result = GitBranchPullResult()
0.200.338 by Jelmer Vernooij
Fix dpushing without changes necessary.
637
        result.source_branch = self.source
638
        if _override_hook_target is None:
639
            result.target_branch = self.target
640
        else:
641
            result.target_branch = _override_hook_target
642
        self.source.lock_read()
643
        try:
644
            # We assume that during 'pull' the target repository is closer than
645
            # the source one.
646
            graph = self.target.repository.get_graph(self.source.repository)
0.200.726 by Jelmer Vernooij
Factor out conversion of branch names to refs.
647
            (result.old_revno, result.old_revid) = \
648
                self.target.last_revision_info()
0.200.1062 by Jelmer Vernooij
Pass remote refs along in _update_revisions.
649
            result.new_git_head, remote_refs = self._update_revisions(
0.247.6 by Michael Hudson
away with underscore prefixed local variables
650
                stop_revision, overwrite=overwrite, graph=graph, limit=limit)
0.200.338 by Jelmer Vernooij
Fix dpushing without changes necessary.
651
            result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
652
                overwrite)
0.200.726 by Jelmer Vernooij
Factor out conversion of branch names to refs.
653
            (result.new_revno, result.new_revid) = \
654
                self.target.last_revision_info()
0.200.338 by Jelmer Vernooij
Fix dpushing without changes necessary.
655
            if _hook_master:
656
                result.master_branch = _hook_master
657
                result.local_branch = result.target_branch
658
            else:
659
                result.master_branch = result.target_branch
660
                result.local_branch = None
661
            if run_hooks:
662
                for hook in branch.Branch.hooks['post_pull']:
663
                    hook(result)
664
        finally:
665
            self.source.unlock()
666
        return result
667
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
668
    def _basic_push(self, overwrite=False, stop_revision=None):
669
        result = branch.BranchPushResult()
670
        result.source_branch = self.source
671
        result.target_branch = self.target
0.200.505 by Jelmer Vernooij
Remove duplicate code.
672
        graph = self.target.repository.get_graph(self.source.repository)
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
673
        result.old_revno, result.old_revid = self.target.last_revision_info()
0.200.1062 by Jelmer Vernooij
Pass remote refs along in _update_revisions.
674
        result.new_git_head, remote_refs = self._update_revisions(
0.247.3 by Michael Hudson
oh, so it wasn't (particularly) wrong, but it was a bit obscure
675
            stop_revision, overwrite=overwrite, graph=graph)
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
676
        result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
677
            overwrite)
678
        result.new_revno, result.new_revid = self.target.last_revision_info()
679
        return result
680
0.200.338 by Jelmer Vernooij
Fix dpushing without changes necessary.
681
0.200.512 by Jelmer Vernooij
Support pushing git->git.
682
class InterGitBranch(branch.GenericInterBranch):
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
683
    """InterBranch implementation that pulls between Git branches."""
684
0.200.512 by Jelmer Vernooij
Support pushing git->git.
685
686
class InterGitLocalRemoteBranch(InterGitBranch):
687
    """InterBranch that copies from a local to a remote git branch."""
688
0.200.996 by Jelmer Vernooij
Fix test run of InterBranches.
689
    @staticmethod
690
    def _get_branch_formats_to_test():
0.200.1100 by Jelmer Vernooij
Provide test combinations for InterBranch implementations.
691
        # FIXME
0.200.996 by Jelmer Vernooij
Fix test run of InterBranches.
692
        return []
693
0.200.512 by Jelmer Vernooij
Support pushing git->git.
694
    @classmethod
695
    def is_compatible(self, source, target):
696
        from bzrlib.plugins.git.remote import RemoteGitBranch
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
697
        return (isinstance(source, LocalGitBranch) and
0.200.512 by Jelmer Vernooij
Support pushing git->git.
698
                isinstance(target, RemoteGitBranch))
699
700
    def _basic_push(self, overwrite=False, stop_revision=None):
701
        result = GitBranchPushResult()
702
        result.source_branch = self.source
703
        result.target_branch = self.target
704
        if stop_revision is None:
705
            stop_revision = self.source.last_revision()
706
        # FIXME: Check for diverged branches
707
        def get_changed_refs(old_refs):
0.252.44 by Jelmer Vernooij
Properly look up Bazaar revision ids for revision parents in case they are round-tripped.
708
            result.old_revid = self.target.lookup_foreign_revision_id(old_refs.get(self.target.ref, ZERO_SHA))
0.200.822 by Jelmer Vernooij
Fix indication of number of revisions pushed in dpush.
709
            refs = { self.target.ref: self.source.repository.lookup_bzr_revision_id(stop_revision)[0] }
0.200.512 by Jelmer Vernooij
Support pushing git->git.
710
            result.new_revid = stop_revision
711
            for name, sha in self.source.repository._git.refs.as_dict("refs/tags").iteritems():
0.200.875 by Jelmer Vernooij
Use new tag_name_to_ref function.
712
                refs[tag_name_to_ref(name)] = sha
0.200.512 by Jelmer Vernooij
Support pushing git->git.
713
            return refs
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
714
        self.target.repository.send_pack(get_changed_refs,
0.200.726 by Jelmer Vernooij
Factor out conversion of branch names to refs.
715
            self.source.repository._git.object_store.generate_pack_contents)
0.200.512 by Jelmer Vernooij
Support pushing git->git.
716
        return result
717
718
719
class InterGitRemoteLocalBranch(InterGitBranch):
720
    """InterBranch that copies from a remote to a local git branch."""
721
0.200.996 by Jelmer Vernooij
Fix test run of InterBranches.
722
    @staticmethod
723
    def _get_branch_formats_to_test():
0.200.1100 by Jelmer Vernooij
Provide test combinations for InterBranch implementations.
724
        # FIXME
0.200.996 by Jelmer Vernooij
Fix test run of InterBranches.
725
        return []
726
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
727
    @classmethod
728
    def is_compatible(self, source, target):
729
        from bzrlib.plugins.git.remote import RemoteGitBranch
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
730
        return (isinstance(source, RemoteGitBranch) and
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
731
                isinstance(target, LocalGitBranch))
732
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
733
    def _basic_push(self, overwrite=False, stop_revision=None):
734
        result = branch.BranchPushResult()
735
        result.source_branch = self.source
736
        result.target_branch = self.target
737
        result.old_revid = self.target.last_revision()
738
        refs, stop_revision = self.update_refs(stop_revision)
739
        self.target.generate_revision_history(stop_revision, result.old_revid)
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
740
        result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
741
            source_refs=refs, overwrite=overwrite)
0.200.505 by Jelmer Vernooij
Remove duplicate code.
742
        result.new_revid = self.target.last_revision()
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
743
        return result
744
745
    def update_refs(self, stop_revision=None):
0.200.1097 by Jelmer Vernooij
Implement GitBranchFormat.initialize.
746
        interrepo = _mod_repository.InterRepository.get(self.source.repository,
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
747
            self.target.repository)
748
        if stop_revision is None:
0.200.940 by Jelmer Vernooij
Avoid confusion between different fetch functions with different semantics.
749
            refs = interrepo.fetch(branches=["HEAD"])
0.252.44 by Jelmer Vernooij
Properly look up Bazaar revision ids for revision parents in case they are round-tripped.
750
            stop_revision = self.target.lookup_foreign_revision_id(refs["HEAD"])
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
751
        else:
0.200.940 by Jelmer Vernooij
Avoid confusion between different fetch functions with different semantics.
752
            refs = interrepo.fetch(revision_id=stop_revision)
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
753
        return refs, stop_revision
754
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
755
    def pull(self, stop_revision=None, overwrite=False,
0.200.732 by Jelmer Vernooij
Support run_hooks argument to InterGitRemoteLocalBranch.pull().
756
        possible_transports=None, run_hooks=True,local=False):
0.200.446 by Jelmer Vernooij
Support new 'local' argument.
757
        # This type of branch can't be bound.
758
        if local:
759
            raise errors.LocalRequiresBoundBranch()
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
760
        result = GitPullResult()
761
        result.source_branch = self.source
762
        result.target_branch = self.target
763
        result.old_revid = self.target.last_revision()
0.200.501 by Jelmer Vernooij
Support push from git into bzr.
764
        refs, stop_revision = self.update_refs(stop_revision)
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
765
        self.target.generate_revision_history(stop_revision, result.old_revid)
0.200.1065 by Jelmer Vernooij
Don't peel tags automatically when pushing back.
766
        result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
767
            overwrite=overwrite, source_refs=refs)
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
768
        result.new_revid = self.target.last_revision()
769
        return result
770
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
771
0.200.960 by Jelmer Vernooij
Use GenericInterBranch.
772
class InterToGitBranch(branch.GenericInterBranch):
0.200.468 by Jelmer Vernooij
Move dpush logic onto InterBranch.
773
    """InterBranch implementation that pulls from Git into bzr."""
774
0.200.939 by Jelmer Vernooij
Use InterRepo directly.
775
    def __init__(self, source, target):
776
        super(InterToGitBranch, self).__init__(source, target)
0.200.1097 by Jelmer Vernooij
Implement GitBranchFormat.initialize.
777
        self.interrepo = _mod_repository.InterRepository.get(source.repository,
0.200.939 by Jelmer Vernooij
Use InterRepo directly.
778
                                           target.repository)
779
0.200.631 by Jelmer Vernooij
Raise proper exception in Branch.get_stacked_on_url().
780
    @staticmethod
781
    def _get_branch_formats_to_test():
0.200.1100 by Jelmer Vernooij
Provide test combinations for InterBranch implementations.
782
        try:
783
            default_format = branch.format_registry.get_default()
784
        except AttributeError:
785
            default_format = branch.BranchFormat._default_format
786
        return [(default_format, GitBranchFormat())]
0.200.631 by Jelmer Vernooij
Raise proper exception in Branch.get_stacked_on_url().
787
0.200.468 by Jelmer Vernooij
Move dpush logic onto InterBranch.
788
    @classmethod
789
    def is_compatible(self, source, target):
0.200.695 by Jelmer Vernooij
Clean up trailing whitespace.
790
        return (not isinstance(source, GitBranch) and
0.200.468 by Jelmer Vernooij
Move dpush logic onto InterBranch.
791
                isinstance(target, GitBranch))
792
0.200.542 by Jelmer Vernooij
Proper error for push in 1.14.
793
    def update_revisions(self, *args, **kwargs):
794
        raise NoPushSupport()
795
0.252.38 by Jelmer Vernooij
Minor cleanups.
796
    def _get_new_refs(self, stop_revision=None):
797
        if stop_revision is None:
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
798
            (stop_revno, stop_revision) = self.source.last_revision_info()
0.200.969 by Jelmer Vernooij
Use tuples with bzr revid and git sha to avoid lookups.
799
        assert type(stop_revision) is str
0.200.916 by Jelmer Vernooij
Set refs/heads/master if no ref is set yet.
800
        main_ref = self.target.ref or "refs/heads/master"
0.200.969 by Jelmer Vernooij
Use tuples with bzr revid and git sha to avoid lookups.
801
        refs = { main_ref: (None, stop_revision) }
0.252.37 by Jelmer Vernooij
Factor out some common code for finding refs to send.
802
        for name, revid in self.source.tags.get_tag_dict().iteritems():
803
            if self.source.repository.has_revision(revid):
0.200.969 by Jelmer Vernooij
Use tuples with bzr revid and git sha to avoid lookups.
804
                refs[tag_name_to_ref(name)] = (None, revid)
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
805
        return refs, main_ref, (stop_revno, stop_revision)
0.252.37 by Jelmer Vernooij
Factor out some common code for finding refs to send.
806
0.252.36 by Jelmer Vernooij
Fix pull.
807
    def pull(self, overwrite=False, stop_revision=None, local=False,
0.200.1131 by Jelmer Vernooij
Accept run_hooks argument to InterToGitBranch.pull().
808
             possible_transports=None, run_hooks=True):
0.252.36 by Jelmer Vernooij
Fix pull.
809
        result = GitBranchPullResult()
810
        result.source_branch = self.source
811
        result.target_branch = self.target
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
812
        new_refs, main_ref, stop_revinfo = self._get_new_refs(stop_revision)
0.200.941 by Jelmer Vernooij
Pass update_refs argument to fetch_refs.
813
        def update_refs(old_refs):
814
            refs = dict(old_refs)
0.200.945 by Jelmer Vernooij
Move fixmes
815
            # FIXME: Check for diverged branches
0.200.941 by Jelmer Vernooij
Pass update_refs argument to fetch_refs.
816
            refs.update(new_refs)
817
            return refs
818
        old_refs, new_refs = self.interrepo.fetch_refs(update_refs)
0.200.1042 by Jelmer Vernooij
Fix pull into git branches.
819
        (result.old_revid, old_sha1) = old_refs.get(main_ref, (ZERO_SHA, NULL_REVISION))
820
        if result.old_revid is None:
821
            result.old_revid = self.target.lookup_foreign_revision_id(old_sha1)
822
        result.new_revid = new_refs[main_ref][1]
0.252.36 by Jelmer Vernooij
Fix pull.
823
        return result
824
0.252.38 by Jelmer Vernooij
Minor cleanups.
825
    def push(self, overwrite=False, stop_revision=None,
0.200.472 by Jelmer Vernooij
Fix printing error when user attempts to push into git.
826
             _override_hook_source_branch=None):
0.252.5 by Jelmer Vernooij
enable 'bzr push'.
827
        result = GitBranchPushResult()
828
        result.source_branch = self.source
829
        result.target_branch = self.target
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
830
        new_refs, main_ref, stop_revinfo = self._get_new_refs(stop_revision)
0.200.941 by Jelmer Vernooij
Pass update_refs argument to fetch_refs.
831
        def update_refs(old_refs):
832
            refs = dict(old_refs)
0.200.945 by Jelmer Vernooij
Move fixmes
833
            # FIXME: Check for diverged branches
0.200.941 by Jelmer Vernooij
Pass update_refs argument to fetch_refs.
834
            refs.update(new_refs)
835
            return refs
836
        old_refs, new_refs = self.interrepo.fetch_refs(update_refs)
0.200.1035 by Jelmer Vernooij
Cope with tuples in refs dictionary.
837
        (result.old_revid, old_sha1) = old_refs.get(main_ref, (ZERO_SHA, NULL_REVISION))
838
        if result.old_revid is None:
839
            result.old_revid = self.target.lookup_foreign_revision_id(old_sha1)
0.200.1042 by Jelmer Vernooij
Fix pull into git branches.
840
        result.new_revid = new_refs[main_ref][1]
0.252.5 by Jelmer Vernooij
enable 'bzr push'.
841
        return result
0.200.472 by Jelmer Vernooij
Fix printing error when user attempts to push into git.
842
0.200.468 by Jelmer Vernooij
Move dpush logic onto InterBranch.
843
    def lossy_push(self, stop_revision=None):
0.200.504 by Jelmer Vernooij
Lazily find revno's for git branches.
844
        result = GitBranchPushResult()
0.200.503 by Jelmer Vernooij
Remove dpull, return BranchPushResult in lossy_push.
845
        result.source_branch = self.source
846
        result.target_branch = self.target
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
847
        new_refs, main_ref, stop_revinfo = self._get_new_refs(stop_revision)
0.200.942 by Jelmer Vernooij
pass update_refs to dfetch_refs.
848
        def update_refs(old_refs):
849
            refs = dict(old_refs)
0.200.945 by Jelmer Vernooij
Move fixmes
850
            # FIXME: Check for diverged branches
0.200.942 by Jelmer Vernooij
pass update_refs to dfetch_refs.
851
            refs.update(new_refs)
852
            return refs
853
        result.revidmap, old_refs, new_refs = self.interrepo.dfetch_refs(
854
            update_refs)
0.200.969 by Jelmer Vernooij
Use tuples with bzr revid and git sha to avoid lookups.
855
        result.old_revid = old_refs.get(self.target.ref, (None, NULL_REVISION))[1]
856
        result.new_revid = new_refs[main_ref][1]
0.200.1048 by Jelmer Vernooij
Make lookup of revno's after push/pull as efficient as possible.
857
        (result.new_original_revno, result.new_original_revid) = stop_revinfo
0.200.503 by Jelmer Vernooij
Remove dpull, return BranchPushResult in lossy_push.
858
        return result
0.200.468 by Jelmer Vernooij
Move dpush logic onto InterBranch.
859
0.200.334 by Jelmer Vernooij
Support pulling from git to git.
860
861
branch.InterBranch.register_optimiser(InterGitRemoteLocalBranch)
0.200.468 by Jelmer Vernooij
Move dpush logic onto InterBranch.
862
branch.InterBranch.register_optimiser(InterFromGitBranch)
863
branch.InterBranch.register_optimiser(InterToGitBranch)
0.200.512 by Jelmer Vernooij
Support pushing git->git.
864
branch.InterBranch.register_optimiser(InterGitLocalRemoteBranch)