1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
# Copyright (C) 2008 Canonical Ltd
#
# 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.errors import InvalidRevisionId
from bzrlib.repository import InterRepository
from bzrlib.trace import info
from bzrlib.plugins.git.repository import LocalGitRepository, GitRepository, GitFormat
from bzrlib.plugins.git.remote import RemoteGitRepository
from cStringIO import StringIO
class BzrFetchGraphWalker(object):
def __init__(self, repository, mapping):
self.repository = repository
self.mapping = mapping
self.done = set()
self.heads = set(repository.all_revision_ids())
self.parents = {}
def ack(self, sha):
revid = self.mapping.revision_id_foreign_to_bzr(sha)
self.remove(revid)
def remove(self, revid):
self.done.add(revid)
if ref in self.heads:
self.heads.remove(revid)
if revid in self.parents:
for p in self.parents[revid]:
self.remove(p)
def next(self):
while self.heads:
ret = self.heads.pop()
ps = self.repository.get_parent_map([ret])[ret]
self.parents[ret] = ps
self.heads.update([p for p in ps if not p in self.done])
try:
self.done.add(ret)
return self.mapping.revision_id_bzr_to_foreign(ret)
except InvalidRevisionId:
pass
return None
def import_git_object(repo, object):
raise NotImplementedError(import_git_object)
class InterGitRepository(InterRepository):
_matching_repo_format = GitFormat()
@staticmethod
def _get_repo_format_to_test():
return None
def copy_content(self, revision_id=None, pb=None):
"""See InterRepository.copy_content."""
self.fetch(revision_id, pb, find_ghosts=False)
def fetch(self, revision_id=None, pb=None, find_ghosts=False,
mapping=None):
if mapping is None:
mapping = self.source.get_mapping()
def progress(text):
if pb is not None:
pb.note("git: %s" % text)
else:
info("git: %s" % text)
def determine_wants(heads):
if revision_id is None:
ret = heads.values()
else:
ret = [mapping.revision_id_bzr_to_foreign(revision_id)]
return [rev for rev in ret if not self.target.has_revision(mapping.revision_id_foreign_to_bzr(rev))]
graph_walker = BzrFetchGraphWalker(self.target, mapping)
self.target.lock_write()
try:
for o in self.source.fetch_objects(determine_wants, graph_walker, progress):
import_git_object(o)
finally:
self.target.unlock()
@staticmethod
def is_compatible(source, target):
"""Be compatible with GitRepository."""
# FIXME: Also check target uses VersionedFile
return isinstance(source, LocalGitRepository) and target.supports_rich_root()
|