share.py
125 lines
| 4.2 KiB
| text/x-python
|
PythonLexer
/ hgext / share.py
|
r8801 | # Copyright 2006, 2007 Matt Mackall <mpm@selenic.com> | ||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r8801 | |||
|
r8894 | '''share a common history between several working directories''' | ||
|
r8873 | |||
|
r8801 | from mercurial.i18n import _ | ||
|
r23548 | from mercurial import cmdutil, hg, util, extensions, bookmarks | ||
from mercurial.hg import repository, parseurl | ||||
import errno | ||||
|
r15079 | |||
|
r21253 | cmdtable = {} | ||
command = cmdutil.command(cmdtable) | ||||
|
r16743 | testedwith = 'internal' | ||
|
r21253 | @command('share', | ||
|
r24364 | [('U', 'noupdate', None, _('do not create a working directory')), | ||
|
r23614 | ('B', 'bookmarks', None, _('also share bookmarks'))], | ||
_('[-U] [-B] SOURCE [DEST]'), | ||||
|
r21772 | norepo=True) | ||
|
r23614 | def share(ui, source, dest=None, noupdate=False, bookmarks=False): | ||
|
r10798 | """create a new shared repository | ||
|
r8801 | |||
|
r9273 | Initialize a new repository and working directory that shares its | ||
|
r23614 | history (and optionally bookmarks) with another repository. | ||
|
r8801 | |||
|
r12389 | .. note:: | ||
|
r19997 | |||
|
r12389 | using rollback or extensions that destroy/modify history (mq, | ||
rebase, etc.) can cause considerable confusion with shared | ||||
clones. In particular, if two shared clones are both updated to | ||||
the same changeset, and one of them destroys that changeset | ||||
with rollback, the other clone will suddenly stop working: all | ||||
operations will fail with "abort: working directory has unknown | ||||
parent". The only known workaround is to use debugsetparents on | ||||
|
r19399 | the broken clone to reset it to a changeset that still exists. | ||
|
r8801 | """ | ||
|
r23614 | return hg.share(ui, source, dest, not noupdate, bookmarks) | ||
|
r8801 | |||
|
r21253 | @command('unshare', [], '') | ||
|
r15079 | def unshare(ui, repo): | ||
"""convert a shared repository to a normal one | ||||
Copy the store data to the repo and remove the sharedpath data. | ||||
""" | ||||
|
r23666 | if not repo.shared(): | ||
|
r15079 | raise util.Abort(_("this is not a shared repo")) | ||
destlock = lock = None | ||||
lock = repo.lock() | ||||
try: | ||||
# we use locks 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 | ||||
destlock = hg.copystore(ui, repo, repo.path) | ||||
sharefile = repo.join('sharedpath') | ||||
util.rename(sharefile, sharefile + '.old') | ||||
repo.requirements.discard('sharedpath') | ||||
repo._writerequirements() | ||||
finally: | ||||
destlock and destlock.release() | ||||
lock and lock.release() | ||||
# update store, spath, sopener and sjoin of repo | ||||
|
r20056 | repo.unfiltered().__init__(repo.baseui, repo.root) | ||
|
r23548 | |||
def extsetup(ui): | ||||
extensions.wrapfunction(bookmarks.bmstore, 'getbkfile', getbkfile) | ||||
extensions.wrapfunction(bookmarks.bmstore, 'recordchange', recordchange) | ||||
extensions.wrapfunction(bookmarks.bmstore, 'write', write) | ||||
def _hassharedbookmarks(repo): | ||||
"""Returns whether this repo has shared bookmarks""" | ||||
try: | ||||
|
r23883 | shared = repo.vfs.read('shared').splitlines() | ||
|
r23548 | except IOError, inst: | ||
if inst.errno != errno.ENOENT: | ||||
raise | ||||
return False | ||||
|
r23883 | return 'bookmarks' in shared | ||
|
r23548 | |||
def _getsrcrepo(repo): | ||||
""" | ||||
Returns the source repository object for a given shared repository. | ||||
If repo is not a shared repository, return None. | ||||
""" | ||||
|
r23626 | if repo.sharedpath == repo.path: | ||
return None | ||||
# the sharedpath always ends in the .hg; we want the path to the repo | ||||
source = repo.vfs.split(repo.sharedpath)[0] | ||||
srcurl, branches = parseurl(source) | ||||
return repository(repo.ui, srcurl) | ||||
|
r23548 | |||
def getbkfile(orig, self, repo): | ||||
if _hassharedbookmarks(repo): | ||||
srcrepo = _getsrcrepo(repo) | ||||
if srcrepo is not None: | ||||
repo = srcrepo | ||||
return orig(self, repo) | ||||
def recordchange(orig, self, tr): | ||||
# Continue with write to local bookmarks file as usual | ||||
orig(self, tr) | ||||
if _hassharedbookmarks(self._repo): | ||||
srcrepo = _getsrcrepo(self._repo) | ||||
if srcrepo is not None: | ||||
category = 'share-bookmarks' | ||||
tr.addpostclose(category, lambda tr: self._writerepo(srcrepo)) | ||||
def write(orig, self): | ||||
# First write local bookmarks file in case we ever unshare | ||||
orig(self) | ||||
if _hassharedbookmarks(self._repo): | ||||
srcrepo = _getsrcrepo(self._repo) | ||||
if srcrepo is not None: | ||||
self._writerepo(srcrepo) | ||||