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
|
from mercurial import exthelper
from mercurial import repoview
from mercurial import statichttprepo
from mercurial import util as hgutil
from mercurial.node import bin
from .git_handler import GitHandler
from .gitrepo import gitrepo
from . import util
eh = exthelper.exthelper()
@eh.reposetup
def reposetup(ui, repo):
if isinstance(repo, (statichttprepo.statichttprepository, gitrepo)):
return
if hasattr(repo, '_wlockfreeprefix'):
repo._wlockfreeprefix |= {
GitHandler.map_file,
GitHandler.tags_file,
}
class hgrepo(repo.__class__):
@util.transform_notgit
def findoutgoing(self, remote, base=None, heads=None, force=False):
if isinstance(remote, gitrepo):
base, heads = self.githandler.get_refs(remote.path)
out, h = super().findoutgoing(remote, base, heads, force)
return out
else: # pragma: no cover
return super().findoutgoing(remote, base, heads, force)
def _findtags(self):
(tags, tagtypes) = super()._findtags()
for tag, rev in self.githandler.tags.items():
if tag not in tags:
assert isinstance(tag, bytes)
tags[tag] = bin(rev)
tagtypes[tag] = b'git'
for tag, rev in self.githandler.remote_refs.items():
assert isinstance(tag, bytes)
tags[tag] = rev
tagtypes[tag] = b'git-remote'
tags.update(self.githandler.remote_refs)
return (tags, tagtypes)
@hgutil.propertycache
def githandler(self):
'''get the GitHandler for an hg repo
This only makes sense if the repo talks to at least one git remote.
'''
return GitHandler(self, self.ui)
def tags(self):
# TODO consider using self._tagscache
tagscache = super().tags()
tagscache.update(self.githandler.remote_refs)
for tag, rev in self.githandler.tags.items():
if tag in tagscache:
continue
tagscache[tag] = bin(rev)
return tagscache
repo.__class__ = hgrepo
@eh.wrapfunction(repoview, 'pinnedrevs')
def pinnedrevs(orig, repo):
pinned = orig(repo)
# Mercurial pins bookmarks, even if obsoleted, so that they always
# appear in e.g. log; do the same with git tags and remotes.
if repo.local() and hasattr(repo, 'githandler'):
rev = repo.changelog.rev
pinned.update(rev(bin(r)) for r in repo.githandler.tags.values())
pinned.update(rev(r) for r in repo.githandler.remote_refs.values())
return pinned
|
