hg.py
916 lines
| 31.5 KiB
| text/x-python
|
PythonLexer
/ mercurial / hg.py
|
r0 | # hg.py - repository classes for mercurial | ||
# | ||||
|
r4635 | # Copyright 2005-2007 Matt Mackall <mpm@selenic.com> | ||
|
r2859 | # Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com> | ||
|
r0 | # | ||
|
r8225 | # This software may be used and distributed according to the terms of the | ||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r0 | |||
|
r25939 | from __future__ import absolute_import | ||
import errno | ||||
import os | ||||
import shutil | ||||
from .i18n import _ | ||||
from .node import nullid | ||||
|
r22837 | |||
|
r25939 | from . import ( | ||
bookmarks, | ||||
bundlerepo, | ||||
cmdutil, | ||||
discovery, | ||||
error, | ||||
exchange, | ||||
extensions, | ||||
httppeer, | ||||
localrepo, | ||||
lock, | ||||
merge as mergemod, | ||||
node, | ||||
phases, | ||||
repoview, | ||||
scmutil, | ||||
sshpeer, | ||||
statichttprepo, | ||||
ui as uimod, | ||||
unionrepo, | ||||
url, | ||||
util, | ||||
verify as verifymod, | ||||
) | ||||
release = lock.release | ||||
|
r0 | |||
|
r2740 | def _local(path): | ||
|
r14825 | path = util.expandpath(util.urllocalpath(path)) | ||
|
r11154 | return (os.path.isfile(path) and bundlerepo or localrepo) | ||
|
r2469 | |||
|
r17191 | def addbranchrevs(lrepo, other, branches, revs): | ||
peer = other.peer() # a courtesy to callers using a localrepo for other | ||||
|
r11322 | hashbranch, branches = branches | ||
if not hashbranch and not branches: | ||||
|
r22818 | x = revs or None | ||
if util.safehasattr(revs, 'first'): | ||||
y = revs.first() | ||||
elif revs: | ||||
y = revs[0] | ||||
else: | ||||
y = None | ||||
return x, y | ||||
|
r24306 | if revs: | ||
revs = list(revs) | ||||
else: | ||||
revs = [] | ||||
|
r17191 | if not peer.capable('branchmap'): | ||
|
r11322 | if branches: | ||
|
r26587 | raise error.Abort(_("remote branch lookup not supported")) | ||
|
r11322 | revs.append(hashbranch) | ||
|
r10380 | return revs, revs[0] | ||
|
r17191 | branchmap = peer.branchmap() | ||
|
r11322 | |||
|
r13047 | def primary(branch): | ||
if branch == '.': | ||||
|
r17191 | if not lrepo: | ||
|
r26587 | raise error.Abort(_("dirstate branch not accessible")) | ||
|
r13047 | branch = lrepo.dirstate.branch() | ||
if branch in branchmap: | ||||
revs.extend(node.hex(r) for r in reversed(branchmap[branch])) | ||||
|
r11322 | return True | ||
|
r10365 | else: | ||
|
r11322 | return False | ||
for branch in branches: | ||||
|
r13047 | if not primary(branch): | ||
|
r11322 | raise error.RepoLookupError(_("unknown branch '%s'") % branch) | ||
if hashbranch: | ||||
|
r13047 | if not primary(hashbranch): | ||
|
r11322 | revs.append(hashbranch) | ||
|
r10365 | return revs, revs[0] | ||
|
r13824 | def parseurl(path, branches=None): | ||
|
r11322 | '''parse url#branch, returning (url, (branch, branches))''' | ||
|
r5177 | |||
|
r14076 | u = util.url(path) | ||
|
r13897 | branch = None | ||
if u.fragment: | ||||
branch = u.fragment | ||||
u.fragment = None | ||||
|
r13824 | return str(u), (branch, branches or []) | ||
|
r5177 | |||
|
r14606 | schemes = { | ||
|
r14568 | 'bundle': bundlerepo, | ||
|
r18944 | 'union': unionrepo, | ||
|
r14568 | 'file': _local, | ||
|
r17192 | 'http': httppeer, | ||
'https': httppeer, | ||||
'ssh': sshpeer, | ||||
|
r14568 | 'static-http': statichttprepo, | ||
} | ||||
def _peerlookup(path): | ||||
u = util.url(path) | ||||
scheme = u.scheme or 'file' | ||||
|
r14606 | thing = schemes.get(scheme) or schemes['file'] | ||
|
r14568 | try: | ||
return thing(path) | ||||
except TypeError: | ||||
|
r25365 | # we can't test callable(thing) because 'thing' can be an unloaded | ||
# module that implements __call__ | ||||
if not util.safehasattr(thing, 'instance'): | ||||
raise | ||||
|
r14568 | return thing | ||
|
r14605 | def islocal(repo): | ||
|
r20355 | '''return true if repo (or path pointing to repo) is local''' | ||
|
r14605 | if isinstance(repo, str): | ||
try: | ||||
return _peerlookup(repo).islocal(repo) | ||||
except AttributeError: | ||||
return False | ||||
return repo.local() | ||||
|
r17887 | def openpath(ui, path): | ||
'''open path with open if local, url.open if remote''' | ||||
|
r20354 | pathurl = util.url(path, parsequery=False, parsefragment=False) | ||
if pathurl.islocal(): | ||||
return util.posixfile(pathurl.localpath(), 'rb') | ||||
|
r17887 | else: | ||
return url.open(ui, path) | ||||
|
r20858 | # a list of (ui, repo) functions called for wire peer initialization | ||
wirepeersetupfuncs = [] | ||||
|
r17191 | def _peerorrepo(ui, path, create=False): | ||
|
r14605 | """return a repository object for the specified path""" | ||
|
r17191 | obj = _peerlookup(path).instance(ui, path, create) | ||
ui = getattr(obj, "ui", ui) | ||||
|
r19777 | for name, module in extensions.extensions(ui): | ||
|
r14605 | hook = getattr(module, 'reposetup', None) | ||
if hook: | ||||
|
r17191 | hook(ui, obj) | ||
|
r20858 | if not obj.local(): | ||
for f in wirepeersetupfuncs: | ||||
f(ui, obj) | ||||
|
r17191 | return obj | ||
def repository(ui, path='', create=False): | ||||
"""return a repository object for the specified path""" | ||||
peer = _peerorrepo(ui, path, create) | ||||
repo = peer.local() | ||||
if not repo: | ||||
|
r26587 | raise error.Abort(_("repository '%s' is not local") % | ||
|
r17191 | (path or peer.url())) | ||
|
r18382 | return repo.filtered('visible') | ||
|
r14605 | |||
|
r14839 | def peer(uiorrepo, opts, path, create=False): | ||
|
r14554 | '''return a repository peer for the specified path''' | ||
|
r14839 | rui = remoteui(uiorrepo, opts) | ||
|
r17191 | return _peerorrepo(rui, path, create).peer() | ||
|
r14554 | |||
|
r2719 | def defaultdest(source): | ||
|
r20799 | '''return default destination of clone if none is given | ||
>>> defaultdest('foo') | ||||
'foo' | ||||
>>> defaultdest('/foo/bar') | ||||
'bar' | ||||
>>> defaultdest('/') | ||||
'' | ||||
>>> defaultdest('') | ||||
|
r20800 | '' | ||
|
r20799 | >>> defaultdest('http://example.org/') | ||
|
r20800 | '' | ||
|
r20799 | >>> defaultdest('http://example.org/foo/') | ||
'foo' | ||||
''' | ||||
|
r20800 | path = util.url(source).path | ||
if not path: | ||||
return '' | ||||
return os.path.basename(os.path.normpath(path)) | ||||
|
r2774 | |||
|
r23614 | def share(ui, source, dest=None, update=True, bookmarks=True): | ||
|
r8800 | '''create a shared repository''' | ||
if not islocal(source): | ||||
|
r26587 | raise error.Abort(_('can only share local repositories')) | ||
|
r8800 | |||
|
r8807 | if not dest: | ||
|
r10099 | dest = defaultdest(source) | ||
|
r9344 | else: | ||
dest = ui.expandpath(dest) | ||||
|
r8807 | |||
|
r8800 | if isinstance(source, str): | ||
origsource = ui.expandpath(source) | ||||
|
r10365 | source, branches = parseurl(origsource) | ||
|
r8800 | srcrepo = repository(ui, source) | ||
|
r10365 | rev, checkout = addbranchrevs(srcrepo, srcrepo, branches, None) | ||
|
r8800 | else: | ||
|
r17191 | srcrepo = source.local() | ||
|
r8800 | origsource = source = srcrepo.url() | ||
checkout = None | ||||
sharedpath = srcrepo.sharedpath # if our source is already sharing | ||||
|
r21800 | destwvfs = scmutil.vfs(dest, realpath=True) | ||
|
r21801 | destvfs = scmutil.vfs(os.path.join(destwvfs.base, '.hg'), realpath=True) | ||
|
r8800 | |||
|
r21801 | if destvfs.lexists(): | ||
|
r26587 | raise error.Abort(_('destination already exists')) | ||
|
r8800 | |||
|
r21800 | if not destwvfs.isdir(): | ||
destwvfs.mkdir() | ||||
|
r21801 | destvfs.makedir() | ||
|
r8800 | |||
requirements = '' | ||||
try: | ||||
|
r23877 | requirements = srcrepo.vfs.read('requires') | ||
|
r25660 | except IOError as inst: | ||
|
r8800 | if inst.errno != errno.ENOENT: | ||
raise | ||||
requirements += 'shared\n' | ||||
|
r21802 | destvfs.write('requires', requirements) | ||
destvfs.write('sharedpath', sharedpath) | ||||
|
r8800 | |||
|
r21800 | r = repository(ui, destwvfs.base) | ||
|
r27354 | postshare(srcrepo, r, bookmarks=bookmarks) | ||
|
r8800 | |||
if update: | ||||
r.ui.status(_("updating working directory\n")) | ||||
if update is not True: | ||||
checkout = update | ||||
for test in (checkout, 'default', 'tip'): | ||||
|
r9423 | if test is None: | ||
continue | ||||
|
r8800 | try: | ||
uprev = r.lookup(test) | ||||
break | ||||
|
r9423 | except error.RepoLookupError: | ||
|
r8800 | continue | ||
_update(r, uprev) | ||||
|
r27354 | def postshare(sourcerepo, destrepo, bookmarks=True): | ||
"""Called after a new shared repo is created. | ||||
The new repo only has a requirements file and pointer to the source. | ||||
This function configures additional shared data. | ||||
Extensions can wrap this function and write additional entries to | ||||
destrepo/.hg/shared to indicate additional pieces of data to be shared. | ||||
""" | ||||
default = sourcerepo.ui.config('paths', 'default') | ||||
if default: | ||||
fp = destrepo.vfs("hgrc", "w", text=True) | ||||
fp.write("[paths]\n") | ||||
fp.write("default = %s\n" % default) | ||||
fp.close() | ||||
|
r23614 | if bookmarks: | ||
|
r27354 | fp = destrepo.vfs('shared', 'w') | ||
|
r23883 | fp.write('bookmarks\n') | ||
fp.close() | ||||
|
r23614 | |||
|
r15078 | def copystore(ui, srcrepo, destpath): | ||
'''copy files from store of srcrepo in destpath | ||||
returns destlock | ||||
''' | ||||
destlock = None | ||||
try: | ||||
hardlink = None | ||||
num = 0 | ||||
|
r24440 | closetopic = [None] | ||
def prog(topic, pos): | ||||
if pos is None: | ||||
closetopic[0] = topic | ||||
else: | ||||
ui.progress(topic, pos + num) | ||||
|
r25624 | srcpublishing = srcrepo.publishing() | ||
|
r20089 | srcvfs = scmutil.vfs(srcrepo.sharedpath) | ||
dstvfs = scmutil.vfs(destpath) | ||||
|
r15078 | for f in srcrepo.store.copylist(): | ||
|
r15741 | if srcpublishing and f.endswith('phaseroots'): | ||
continue | ||||
|
r20089 | dstbase = os.path.dirname(f) | ||
if dstbase and not dstvfs.exists(dstbase): | ||||
dstvfs.mkdir(dstbase) | ||||
if srcvfs.exists(f): | ||||
if f.endswith('data'): | ||||
|
r20825 | # 'dstbase' may be empty (e.g. revlog format 0) | ||
lockfile = os.path.join(dstbase, "lock") | ||||
|
r15078 | # lock to avoid premature writing to the target | ||
|
r20825 | destlock = lock.lock(dstvfs, lockfile) | ||
|
r20089 | hardlink, n = util.copyfiles(srcvfs.join(f), dstvfs.join(f), | ||
|
r24440 | hardlink, progress=prog) | ||
|
r15078 | num += n | ||
if hardlink: | ||||
ui.debug("linked %d files\n" % num) | ||||
|
r24440 | if closetopic[0]: | ||
ui.progress(closetopic[0], None) | ||||
|
r15078 | else: | ||
ui.debug("copied %d files\n" % num) | ||||
|
r24440 | if closetopic[0]: | ||
ui.progress(closetopic[0], None) | ||||
|
r15078 | return destlock | ||
|
r16705 | except: # re-raises | ||
|
r15078 | release(destlock) | ||
raise | ||||
|
r25761 | def clonewithshare(ui, peeropts, sharepath, source, srcpeer, dest, pull=False, | ||
rev=None, update=True, stream=False): | ||||
"""Perform a clone using a shared repo. | ||||
The store for the repository will be located at <sharepath>/.hg. The | ||||
specified revisions will be cloned or pulled from "source". A shared repo | ||||
will be created at "dest" and a working copy will be created if "update" is | ||||
True. | ||||
""" | ||||
revs = None | ||||
if rev: | ||||
if not srcpeer.capable('lookup'): | ||||
|
r26587 | raise error.Abort(_("src repository does not support " | ||
|
r25761 | "revision lookup and so doesn't " | ||
"support clone by revision")) | ||||
revs = [srcpeer.lookup(r) for r in rev] | ||||
basename = os.path.basename(sharepath) | ||||
if os.path.exists(sharepath): | ||||
ui.status(_('(sharing from existing pooled repository %s)\n') % | ||||
basename) | ||||
else: | ||||
ui.status(_('(sharing from new pooled repository %s)\n') % basename) | ||||
# Always use pull mode because hardlinks in share mode don't work well. | ||||
# Never update because working copies aren't necessary in share mode. | ||||
clone(ui, peeropts, source, dest=sharepath, pull=True, | ||||
rev=rev, update=False, stream=stream) | ||||
sharerepo = repository(ui, path=sharepath) | ||||
share(ui, sharerepo, dest=dest, update=update, bookmarks=False) | ||||
# We need to perform a pull against the dest repo to fetch bookmarks | ||||
# and other non-store data that isn't shared by default. In the case of | ||||
# non-existing shared repo, this means we pull from the remote twice. This | ||||
# is a bit weird. But at the time it was implemented, there wasn't an easy | ||||
# way to pull just non-changegroup data. | ||||
destrepo = repository(ui, path=dest) | ||||
exchange.pull(destrepo, srcpeer, heads=revs) | ||||
return srcpeer, peer(ui, peeropts, dest) | ||||
|
r14607 | def clone(ui, peeropts, source, dest=None, pull=False, rev=None, | ||
|
r25761 | update=True, stream=False, branch=None, shareopts=None): | ||
|
r2597 | """Make a copy of an existing repository. | ||
Create a copy of an existing repository in a new directory. The | ||||
source and destination are URLs, as passed to the repository | ||||
|
r17191 | function. Returns a pair of repository peers, the source and | ||
|
r2597 | newly created destination. | ||
The location of the source is added to the new repository's | ||||
.hg/hgrc file, as the default to be used for future pulls and | ||||
pushes. | ||||
If an exception is raised, the partly cloned/updated destination | ||||
repository will be deleted. | ||||
|
r2600 | |||
|
r2719 | Arguments: | ||
source: repository object or URL | ||||
|
r2597 | |||
dest: URL of destination repository to create (defaults to base | ||||
name of source repository) | ||||
|
r23545 | pull: always pull from source repository, even in local case or if the | ||
server prefers streaming | ||||
|
r2597 | |||
|
r2621 | stream: stream raw data uncompressed from repository (fast over | ||
LAN, slow over WAN) | ||||
|
r2613 | |||
|
r2597 | rev: revision to clone up to (implies pull=True) | ||
update: update working directory after clone completes, if | ||||
|
r6526 | destination is local repository (True means update to default rev, | ||
anything else is treated as a revision) | ||||
|
r10379 | |||
branch: branches to clone | ||||
|
r25761 | |||
shareopts: dict of options to control auto sharing behavior. The "pool" key | ||||
activates auto sharing mode and defines the directory for stores. The | ||||
"mode" key determines how to construct the directory name of the shared | ||||
repository. "identity" means the name is derived from the node of the first | ||||
changeset in the repository. "remote" means the name is derived from the | ||||
remote's path/URL. Defaults to "identity." | ||||
|
r2597 | """ | ||
|
r4478 | |||
|
r2719 | if isinstance(source, str): | ||
|
r6089 | origsource = ui.expandpath(source) | ||
|
r10379 | source, branch = parseurl(origsource, branch) | ||
|
r17191 | srcpeer = peer(ui, peeropts, source) | ||
|
r2719 | else: | ||
|
r17191 | srcpeer = source.peer() # in case we were called with a localrepo | ||
|
r11818 | branch = (None, branch or []) | ||
|
r17191 | origsource = source = srcpeer.url() | ||
rev, checkout = addbranchrevs(srcpeer, srcpeer, branch, rev) | ||||
|
r2719 | |||
|
r2597 | if dest is None: | ||
|
r2719 | dest = defaultdest(source) | ||
|
r20800 | if dest: | ||
ui.status(_("destination directory: %s\n") % dest) | ||||
|
r9344 | else: | ||
dest = ui.expandpath(dest) | ||||
|
r2719 | |||
|
r14825 | dest = util.urllocalpath(dest) | ||
source = util.urllocalpath(source) | ||||
|
r2597 | |||
|
r17159 | if not dest: | ||
|
r26587 | raise error.Abort(_("empty destination path is not valid")) | ||
|
r21803 | |||
destvfs = scmutil.vfs(dest, expandpath=True) | ||||
if destvfs.lexists(): | ||||
if not destvfs.isdir(): | ||||
|
r26587 | raise error.Abort(_("destination '%s' already exists") % dest) | ||
|
r21804 | elif destvfs.listdir(): | ||
|
r26587 | raise error.Abort(_("destination '%s' is not empty") % dest) | ||
|
r2597 | |||
|
r25761 | shareopts = shareopts or {} | ||
sharepool = shareopts.get('pool') | ||||
sharenamemode = shareopts.get('mode') | ||||
|
r26026 | if sharepool and islocal(dest): | ||
|
r25761 | sharepath = None | ||
if sharenamemode == 'identity': | ||||
# Resolve the name from the initial changeset in the remote | ||||
# repository. This returns nullid when the remote is empty. It | ||||
# raises RepoLookupError if revision 0 is filtered or otherwise | ||||
# not available. If we fail to resolve, sharing is not enabled. | ||||
try: | ||||
rootnode = srcpeer.lookup('0') | ||||
if rootnode != node.nullid: | ||||
sharepath = os.path.join(sharepool, node.hex(rootnode)) | ||||
else: | ||||
ui.status(_('(not using pooled storage: ' | ||||
'remote appears to be empty)\n')) | ||||
except error.RepoLookupError: | ||||
ui.status(_('(not using pooled storage: ' | ||||
'unable to resolve identity of remote)\n')) | ||||
elif sharenamemode == 'remote': | ||||
sharepath = os.path.join(sharepool, util.sha1(source).hexdigest()) | ||||
else: | ||||
|
r26587 | raise error.Abort('unknown share naming mode: %s' % sharenamemode) | ||
|
r25761 | |||
if sharepath: | ||||
return clonewithshare(ui, peeropts, sharepath, source, srcpeer, | ||||
dest, pull=pull, rev=rev, update=update, | ||||
stream=stream) | ||||
|
r18441 | srclock = destlock = cleandir = None | ||
|
r17191 | srcrepo = srcpeer.local() | ||
|
r4915 | try: | ||
|
r14377 | abspath = origsource | ||
if islocal(origsource): | ||||
|
r14825 | abspath = os.path.abspath(util.urllocalpath(origsource)) | ||
|
r14377 | |||
|
r4915 | if islocal(dest): | ||
|
r18441 | cleandir = dest | ||
|
r2597 | |||
|
r4915 | copy = False | ||
|
r17194 | if (srcrepo and srcrepo.cancopy() and islocal(dest) | ||
|
r17671 | and not phases.hassecret(srcrepo)): | ||
|
r4915 | copy = not pull and not rev | ||
|
r2597 | |||
|
r4915 | if copy: | ||
try: | ||||
# we use a lock here because if we race with commit, we | ||||
# can end up with extra data in the cloned revlogs that's | ||||
# not pointed to by changesets, thus causing verify to | ||||
# fail | ||||
|
r14463 | srclock = srcrepo.lock(wait=False) | ||
|
r7640 | except error.LockError: | ||
|
r4915 | copy = False | ||
|
r2597 | |||
|
r4915 | if copy: | ||
|
r14463 | srcrepo.hook('preoutgoing', throw=True, source='clone') | ||
|
r15381 | hgdir = os.path.realpath(os.path.join(dest, ".hg")) | ||
|
r4915 | if not os.path.exists(dest): | ||
os.mkdir(dest) | ||||
|
r7935 | else: | ||
# only clean up directories we create ourselves | ||||
|
r18441 | cleandir = hgdir | ||
|
r5569 | try: | ||
|
r14463 | destpath = hgdir | ||
util.makedir(destpath, notindexed=True) | ||||
|
r25660 | except OSError as inst: | ||
|
r5569 | if inst.errno == errno.EEXIST: | ||
|
r18441 | cleandir = None | ||
|
r26587 | raise error.Abort(_("destination '%s' already exists") | ||
|
r5569 | % dest) | ||
raise | ||||
|
r2597 | |||
|
r15078 | destlock = copystore(ui, srcrepo, destpath) | ||
|
r22646 | # copy bookmarks over | ||
srcbookmarks = srcrepo.join('bookmarks') | ||||
dstbookmarks = os.path.join(destpath, 'bookmarks') | ||||
if os.path.exists(srcbookmarks): | ||||
util.copyfile(srcbookmarks, dstbookmarks) | ||||
|
r2597 | |||
|
r17740 | # Recomputing branch cache might be slow on big repos, | ||
# so just copy it | ||||
|
r22264 | def copybranchcache(fname): | ||
srcbranchcache = srcrepo.join('cache/%s' % fname) | ||||
dstbranchcache = os.path.join(dstcachedir, fname) | ||||
if os.path.exists(srcbranchcache): | ||||
if not os.path.exists(dstcachedir): | ||||
os.mkdir(dstcachedir) | ||||
util.copyfile(srcbranchcache, dstbranchcache) | ||||
|
r17740 | dstcachedir = os.path.join(destpath, 'cache') | ||
|
r22264 | # In local clones we're copying all nodes, not just served | ||
|
r23139 | # ones. Therefore copy all branch caches over. | ||
|
r22264 | copybranchcache('branch2') | ||
for cachename in repoview.filtertable: | ||||
copybranchcache('branch2-%s' % cachename) | ||||
|
r17740 | |||
|
r4915 | # we need to re-init the repo after manually copying the data | ||
# into it | ||||
|
r17874 | destpeer = peer(srcrepo, peeropts, dest) | ||
|
r14463 | srcrepo.hook('outgoing', source='clone', | ||
|
r12144 | node=node.hex(node.nullid)) | ||
|
r4915 | else: | ||
|
r5569 | try: | ||
|
r17875 | destpeer = peer(srcrepo or ui, peeropts, dest, create=True) | ||
# only pass ui when no srcrepo | ||||
|
r25660 | except OSError as inst: | ||
|
r5569 | if inst.errno == errno.EEXIST: | ||
|
r18441 | cleandir = None | ||
|
r26587 | raise error.Abort(_("destination '%s' already exists") | ||
|
r5569 | % dest) | ||
raise | ||||
|
r2597 | |||
|
r4915 | revs = None | ||
if rev: | ||||
|
r17191 | if not srcpeer.capable('lookup'): | ||
|
r26587 | raise error.Abort(_("src repository does not support " | ||
|
r9171 | "revision lookup and so doesn't " | ||
"support clone by revision")) | ||||
|
r17191 | revs = [srcpeer.lookup(r) for r in rev] | ||
|
r8417 | checkout = revs[0] | ||
|
r27165 | local = destpeer.local() | ||
if local: | ||||
|
r23545 | if not stream: | ||
if pull: | ||||
stream = False | ||||
else: | ||||
stream = None | ||||
|
r27165 | # internal config: ui.quietbookmarkmove | ||
quiet = local.ui.backupconfig('ui', 'quietbookmarkmove') | ||||
try: | ||||
local.ui.setconfig( | ||||
'ui', 'quietbookmarkmove', True, 'clone') | ||||
exchange.pull(local, srcpeer, revs, | ||||
streamclonerequested=stream) | ||||
finally: | ||||
local.ui.restoreconfig(quiet) | ||||
|
r17191 | elif srcrepo: | ||
|
r22647 | exchange.push(srcrepo, destpeer, revs=revs, | ||
bookmarks=srcrepo._bookmarks.keys()) | ||||
|
r4915 | else: | ||
|
r26587 | raise error.Abort(_("clone from remote to remote not supported") | ||
) | ||||
|
r2597 | |||
|
r18441 | cleandir = None | ||
|
r2597 | |||
|
r17191 | destrepo = destpeer.local() | ||
if destrepo: | ||||
|
r22837 | template = uimod.samplehgrcs['cloned'] | ||
|
r23877 | fp = destrepo.vfs("hgrc", "w", text=True) | ||
|
r15552 | u = util.url(abspath) | ||
u.passwd = None | ||||
defaulturl = str(u) | ||||
|
r22380 | fp.write(template % defaulturl) | ||
|
r4915 | fp.close() | ||
|
r5185 | |||
|
r20790 | destrepo.ui.setconfig('paths', 'default', defaulturl, 'clone') | ||
|
r8814 | |||
|
r4915 | if update: | ||
|
r6526 | if update is not True: | ||
|
r17342 | checkout = srcpeer.lookup(update) | ||
|
r17867 | uprev = None | ||
|
r17882 | status = None | ||
|
r17867 | if checkout is not None: | ||
|
r5248 | try: | ||
|
r17867 | uprev = destrepo.lookup(checkout) | ||
|
r9423 | except error.RepoLookupError: | ||
|
r26354 | if update is not True: | ||
try: | ||||
uprev = destrepo.lookup(update) | ||||
except error.RepoLookupError: | ||||
pass | ||||
|
r17867 | if uprev is None: | ||
try: | ||||
uprev = destrepo._bookmarks['@'] | ||||
|
r17870 | update = '@' | ||
|
r17882 | bn = destrepo[uprev].branch() | ||
if bn == 'default': | ||||
status = _("updating to bookmark @\n") | ||||
else: | ||||
|
r20868 | status = (_("updating to bookmark @ on branch %s\n") | ||
|
r26353 | % bn) | ||
|
r17867 | except KeyError: | ||
try: | ||||
uprev = destrepo.branchtip('default') | ||||
except error.RepoLookupError: | ||||
uprev = destrepo.lookup('tip') | ||||
|
r17882 | if not status: | ||
bn = destrepo[uprev].branch() | ||||
status = _("updating to branch %s\n") % bn | ||||
destrepo.ui.status(status) | ||||
|
r14463 | _update(destrepo, uprev) | ||
|
r17703 | if update in destrepo._bookmarks: | ||
|
r24945 | bookmarks.activate(destrepo, update) | ||
|
r4915 | finally: | ||
|
r15908 | release(srclock, destlock) | ||
|
r18441 | if cleandir is not None: | ||
shutil.rmtree(cleandir, True) | ||||
|
r17191 | if srcpeer is not None: | ||
srcpeer.close() | ||||
|
r19313 | return srcpeer, destpeer | ||
|
r2775 | |||
|
r3316 | def _showstats(repo, stats): | ||
|
r9454 | repo.ui.status(_("%d files updated, %d files merged, " | ||
"%d files removed, %d files unresolved\n") % stats) | ||||
|
r3316 | |||
|
r17895 | def updaterepo(repo, node, overwrite): | ||
"""Update the working directory to node. | ||||
When overwrite is set, changes are clobbered, merged else | ||||
returns stats (see pydoc mercurial.merge.applyupdates)""" | ||||
|
r27344 | return mergemod.update(repo, node, False, overwrite, | ||
|
r21537 | labels=['working copy', 'destination']) | ||
|
r17895 | |||
|
r2808 | def update(repo, node): | ||
"""update the working directory to node, merging linear changes""" | ||||
|
r17895 | stats = updaterepo(repo, node, False) | ||
|
r3316 | _showstats(repo, stats) | ||
if stats[3]: | ||||
|
r6518 | repo.ui.status(_("use 'hg resolve' to retry unresolved file merges\n")) | ||
|
r5635 | return stats[3] > 0 | ||
|
r2775 | |||
|
r7546 | # naming conflict in clone() | ||
_update = update | ||||
|
r4917 | def clean(repo, node, show_stats=True): | ||
|
r2808 | """forcibly switch the working directory to node, clobbering changes""" | ||
|
r17895 | stats = updaterepo(repo, node, True) | ||
|
r19332 | util.unlinkpath(repo.join('graftstate'), ignoremissing=True) | ||
|
r10282 | if show_stats: | ||
_showstats(repo, stats) | ||||
|
r5635 | return stats[3] > 0 | ||
|
r2775 | |||
|
r4917 | def merge(repo, node, force=None, remind=True): | ||
|
r13162 | """Branch merge with node, resolving changes. Return true if any | ||
unresolved conflicts.""" | ||||
|
r27344 | stats = mergemod.update(repo, node, True, force) | ||
|
r3316 | _showstats(repo, stats) | ||
if stats[3]: | ||||
|
r7821 | repo.ui.status(_("use 'hg resolve' to retry unresolved file merges " | ||
|
r12314 | "or 'hg update -C .' to abandon\n")) | ||
|
r3316 | elif remind: | ||
repo.ui.status(_("(branch merge, don't forget to commit)\n")) | ||||
|
r5635 | return stats[3] > 0 | ||
|
r2808 | |||
|
r12730 | def _incoming(displaychlist, subreporecurse, ui, repo, source, | ||
opts, buffered=False): | ||||
""" | ||||
Helper for incoming / gincoming. | ||||
displaychlist gets called with | ||||
(remoterepo, incomingchangesetlist, displayer) parameters, | ||||
and is supposed to contain only code that can't be unified. | ||||
""" | ||||
source, branches = parseurl(ui.expandpath(source), opts.get('branch')) | ||||
|
r14556 | other = peer(repo, opts, source) | ||
|
r14076 | ui.status(_('comparing with %s\n') % util.hidepassword(source)) | ||
|
r12730 | revs, checkout = addbranchrevs(repo, other, branches, opts.get('rev')) | ||
if revs: | ||||
revs = [other.lookup(rev) for rev in revs] | ||||
|
r14161 | other, chlist, cleanupfn = bundlerepo.getremotechanges(ui, repo, other, | ||
revs, opts["bundle"], opts["force"]) | ||||
try: | ||||
if not chlist: | ||||
ui.status(_("no changes found\n")) | ||||
return subreporecurse() | ||||
|
r12730 | |||
displayer = cmdutil.show_changeset(ui, other, opts, buffered) | ||||
displaychlist(other, chlist, displayer) | ||||
displayer.close() | ||||
finally: | ||||
|
r14161 | cleanupfn() | ||
|
r12730 | subreporecurse() | ||
return 0 # exit code is zero since we found incoming changes | ||||
|
r12273 | def incoming(ui, repo, source, opts): | ||
|
r12730 | def subreporecurse(): | ||
|
r12400 | ret = 1 | ||
if opts.get('subrepos'): | ||||
ctx = repo[None] | ||||
for subpath in sorted(ctx.substate): | ||||
sub = ctx.sub(subpath) | ||||
ret = min(ret, sub.incoming(ui, source, opts)) | ||||
return ret | ||||
|
r12730 | def display(other, chlist, displayer): | ||
limit = cmdutil.loglimit(opts) | ||||
|
r12273 | if opts.get('newest_first'): | ||
|
r12729 | chlist.reverse() | ||
|
r12273 | count = 0 | ||
|
r12729 | for n in chlist: | ||
|
r12273 | if limit is not None and count >= limit: | ||
break | ||||
parents = [p for p in other.changelog.parents(n) if p != nullid] | ||||
if opts.get('no_merges') and len(parents) == 2: | ||||
continue | ||||
count += 1 | ||||
displayer.show(other[n]) | ||||
|
r12730 | return _incoming(display, subreporecurse, ui, repo, source, opts) | ||
|
r12273 | |||
|
r12735 | def _outgoing(ui, repo, dest, opts): | ||
dest = ui.expandpath(dest or 'default-push', dest or 'default') | ||||
dest, branches = parseurl(dest, opts.get('branch')) | ||||
|
r14076 | ui.status(_('comparing with %s\n') % util.hidepassword(dest)) | ||
|
r12735 | revs, checkout = addbranchrevs(repo, repo, branches, opts.get('rev')) | ||
if revs: | ||||
|
r17198 | revs = [repo.lookup(rev) for rev in scmutil.revrange(repo, revs)] | ||
|
r12735 | |||
|
r14556 | other = peer(repo, opts, dest) | ||
|
r18493 | outgoing = discovery.findcommonoutgoing(repo.unfiltered(), other, revs, | ||
|
r15837 | force=opts.get('force')) | ||
o = outgoing.missing | ||||
|
r12735 | if not o: | ||
|
r17248 | scmutil.nochangesfound(repo.ui, repo, outgoing.excluded) | ||
|
r21050 | return o, other | ||
|
r12735 | |||
|
r12271 | def outgoing(ui, repo, dest, opts): | ||
|
r12400 | def recurse(): | ||
ret = 1 | ||||
if opts.get('subrepos'): | ||||
ctx = repo[None] | ||||
for subpath in sorted(ctx.substate): | ||||
sub = ctx.sub(subpath) | ||||
ret = min(ret, sub.outgoing(ui, dest, opts)) | ||||
return ret | ||||
|
r12271 | limit = cmdutil.loglimit(opts) | ||
|
r21050 | o, other = _outgoing(ui, repo, dest, opts) | ||
|
r21049 | if not o: | ||
|
r21051 | cmdutil.outgoinghooks(ui, repo, other, opts, o) | ||
|
r12400 | return recurse() | ||
|
r12271 | if opts.get('newest_first'): | ||
o.reverse() | ||||
displayer = cmdutil.show_changeset(ui, repo, opts) | ||||
count = 0 | ||||
for n in o: | ||||
if limit is not None and count >= limit: | ||||
break | ||||
parents = [p for p in repo.changelog.parents(n) if p != nullid] | ||||
if opts.get('no_merges') and len(parents) == 2: | ||||
continue | ||||
count += 1 | ||||
displayer.show(repo[n]) | ||||
displayer.close() | ||||
|
r21051 | cmdutil.outgoinghooks(ui, repo, other, opts, o) | ||
|
r12400 | recurse() | ||
return 0 # exit code is zero since we found outgoing changes | ||||
|
r12271 | |||
|
r2778 | def verify(repo): | ||
"""verify the consistency of a repository""" | ||||
|
r25591 | ret = verifymod.verify(repo) | ||
# Broken subrepo references in hidden csets don't seem worth worrying about, | ||||
# since they can't be pushed/pulled, and --hidden can be used if they are a | ||||
# concern. | ||||
# pathto() is needed for -R case | ||||
revs = repo.revs("filelog(%s)", | ||||
util.pathto(repo.root, repo.getcwd(), '.hgsubstate')) | ||||
if revs: | ||||
repo.ui.status(_('checking subrepo links\n')) | ||||
for rev in revs: | ||||
ctx = repo[rev] | ||||
try: | ||||
for subpath in ctx.substate: | ||||
ret = ctx.sub(subpath).verify() or ret | ||||
except Exception: | ||||
repo.ui.warn(_('.hgsubstate is corrupt in revision %s\n') % | ||||
node.short(ctx.node())) | ||||
return ret | ||||
|
r11273 | |||
def remoteui(src, opts): | ||||
'build a remote ui from ui or repo and opts' | ||||
|
r14952 | if util.safehasattr(src, 'baseui'): # looks like a repository | ||
|
r11273 | dst = src.baseui.copy() # drop repo-specific config | ||
src = src.ui # copy target options from repo | ||||
else: # assume it's a global ui object | ||||
dst = src.copy() # keep all global options | ||||
# copy ssh-specific options | ||||
for o in 'ssh', 'remotecmd': | ||||
v = opts.get(o) or src.config('ui', o) | ||||
if v: | ||||
|
r20790 | dst.setconfig("ui", o, v, 'copied') | ||
|
r11273 | |||
# copy bundle-specific options | ||||
r = src.config('bundle', 'mainreporoot') | ||||
if r: | ||||
|
r20790 | dst.setconfig('bundle', 'mainreporoot', r, 'copied') | ||
|
r11273 | |||
|
r13192 | # copy selected local settings to the remote ui | ||
|
r13314 | for sect in ('auth', 'hostfingerprints', 'http_proxy'): | ||
|
r11273 | for key, val in src.configitems(sect): | ||
|
r20790 | dst.setconfig(sect, key, val, 'copied') | ||
|
r13192 | v = src.config('web', 'cacerts') | ||
|
r24290 | if v == '!': | ||
dst.setconfig('web', 'cacerts', v, 'copied') | ||||
elif v: | ||||
|
r20790 | dst.setconfig('web', 'cacerts', util.expandpath(v), 'copied') | ||
|
r11273 | |||
return dst | ||||
|
r26219 | |||
# Files of interest | ||||
# Used to check if the repository has changed looking at mtime and size of | ||||
|
r26781 | # these files. | ||
|
r26219 | foi = [('spath', '00changelog.i'), | ||
('spath', 'phaseroots'), # ! phase can change content at the same size | ||||
('spath', 'obsstore'), | ||||
('path', 'bookmarks'), # ! bookmark can change content at the same size | ||||
] | ||||
class cachedlocalrepo(object): | ||||
"""Holds a localrepository that can be cached and reused.""" | ||||
def __init__(self, repo): | ||||
"""Create a new cached repo from an existing repo. | ||||
We assume the passed in repo was recently created. If the | ||||
repo has changed between when it was created and when it was | ||||
turned into a cache, it may not refresh properly. | ||||
""" | ||||
assert isinstance(repo, localrepo.localrepository) | ||||
self._repo = repo | ||||
self._state, self.mtime = self._repostate() | ||||
def fetch(self): | ||||
"""Refresh (if necessary) and return a repository. | ||||
If the cached instance is out of date, it will be recreated | ||||
automatically and returned. | ||||
Returns a tuple of the repo and a boolean indicating whether a new | ||||
repo instance was created. | ||||
""" | ||||
# We compare the mtimes and sizes of some well-known files to | ||||
# determine if the repo changed. This is not precise, as mtimes | ||||
# are susceptible to clock skew and imprecise filesystems and | ||||
# file content can change while maintaining the same size. | ||||
state, mtime = self._repostate() | ||||
if state == self._state: | ||||
return self._repo, False | ||||
self._repo = repository(self._repo.baseui, self._repo.url()) | ||||
self._state = state | ||||
self.mtime = mtime | ||||
return self._repo, True | ||||
def _repostate(self): | ||||
state = [] | ||||
maxmtime = -1 | ||||
for attr, fname in foi: | ||||
prefix = getattr(self._repo, attr) | ||||
p = os.path.join(prefix, fname) | ||||
try: | ||||
st = os.stat(p) | ||||
except OSError: | ||||
st = os.stat(prefix) | ||||
state.append((st.st_mtime, st.st_size)) | ||||
maxmtime = max(maxmtime, st.st_mtime) | ||||
return tuple(state), maxmtime | ||||
def copy(self): | ||||
|
r26240 | """Obtain a copy of this class instance. | ||
A new localrepository instance is obtained. The new instance should be | ||||
completely independent of the original. | ||||
""" | ||||
repo = repository(self._repo.baseui, self._repo.origroot) | ||||
c = cachedlocalrepo(repo) | ||||
|
r26219 | c._state = self._state | ||
c.mtime = self.mtime | ||||
return c | ||||