mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 16:57:49 +03:00
86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
import os
|
|
|
|
from mercurial.node import bin
|
|
from mercurial import util
|
|
|
|
from git_handler import GitHandler
|
|
from gitrepo import gitrepo
|
|
|
|
from dulwich import errors
|
|
|
|
def _transform_notgit(f):
|
|
def inner(*args, **kwargs):
|
|
try:
|
|
return f(*args, **kwargs)
|
|
except errors.NotGitRepository:
|
|
raise util.Abort('not a git repository')
|
|
return inner
|
|
|
|
|
|
def generate_repo_subclass(baseclass):
|
|
class hgrepo(baseclass):
|
|
@_transform_notgit
|
|
def pull(self, remote, heads=None, force=False):
|
|
if isinstance(remote, gitrepo):
|
|
return self.githandler.fetch(remote.path, heads)
|
|
else: #pragma: no cover
|
|
return super(hgrepo, self).pull(remote, heads, force)
|
|
|
|
# TODO figure out something useful to do with the newbranch param
|
|
@_transform_notgit
|
|
def push(self, remote, force=False, revs=None, newbranch=False):
|
|
if isinstance(remote, gitrepo):
|
|
return self.githandler.push(remote.path, revs, force)
|
|
else: #pragma: no cover
|
|
return super(hgrepo, self).push(remote, force, revs, newbranch)
|
|
|
|
@_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(hgrepo, self).findoutgoing(remote, base, heads, force)
|
|
return out
|
|
else: #pragma: no cover
|
|
return super(hgrepo, self).findoutgoing(remote, base, heads, force)
|
|
|
|
def _findtags(self):
|
|
(tags, tagtypes) = super(hgrepo, self)._findtags()
|
|
|
|
for tag, rev in self.githandler.tags.iteritems():
|
|
tags[tag] = bin(rev)
|
|
tagtypes[tag] = 'git'
|
|
|
|
tags.update(self.gitrefs())
|
|
return (tags, tagtypes)
|
|
|
|
@util.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 gitrefs(self):
|
|
tagfile = self.join(os.path.join('git-remote-refs'))
|
|
if os.path.exists(tagfile):
|
|
tf = open(tagfile, 'rb')
|
|
tagdata = tf.read().split('\n')
|
|
td = [line.split(' ', 1) for line in tagdata if line]
|
|
return dict([(name, bin(sha)) for sha, name in td])
|
|
return {}
|
|
|
|
def tags(self):
|
|
# TODO consider using self._tagscache
|
|
tagscache = super(hgrepo, self).tags()
|
|
tagscache.update(self.gitrefs())
|
|
for tag, rev in self.githandler.tags.iteritems():
|
|
if tag in tagscache:
|
|
continue
|
|
|
|
tagscache[tag] = bin(rev)
|
|
|
|
return tagscache
|
|
|
|
return hgrepo
|