--- a/mercurial/hg.py Tue Jul 11 15:52:56 2006 -0700
+++ b/mercurial/hg.py Tue Jul 11 16:18:53 2006 -0700
@@ -10,7 +10,7 @@
from demandload import *
from i18n import gettext as _
demandload(globals(), "localrepo bundlerepo httprepo sshrepo statichttprepo")
-demandload(globals(), "os util")
+demandload(globals(), "errno lock os shutil util")
def bundle(ui, path):
if path.startswith('bundle://'):
@@ -73,3 +73,133 @@
raise util.Abort(_('cannot create new repository over "%s" protocol') %
scheme)
return ctor(ui, path)
+
+def clone(ui, source, dest=None, pull=False, rev=None, update=True):
+ """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
+ function. Returns a pair of repository objects, the source and
+ 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.
+
+ Keyword arguments:
+
+ dest: URL of destination repository to create (defaults to base
+ name of source repository)
+
+ pull: always pull from source repository, even in local case
+
+ rev: revision to clone up to (implies pull=True)
+
+ update: update working directory after clone completes, if
+ destination is local repository
+ """
+ if dest is None:
+ dest = os.path.basename(os.path.normpath(source))
+
+ if os.path.exists(dest):
+ raise util.Abort(_("destination '%s' already exists"), dest)
+
+ class DirCleanup(object):
+ def __init__(self, dir_):
+ self.rmtree = shutil.rmtree
+ self.dir_ = dir_
+ def close(self):
+ self.dir_ = None
+ def __del__(self):
+ if self.dir_:
+ self.rmtree(self.dir_, True)
+
+ src_repo = repository(ui, source)
+
+ dest_repo = None
+ try:
+ dest_repo = repository(ui, dest)
+ raise util.Abort(_("destination '%s' already exists." % dest))
+ except RepoError:
+ dest_repo = repository(ui, dest, create=True)
+
+ dest_path = None
+ dir_cleanup = None
+ if dest_repo.local():
+ dest_path = os.path.realpath(dest)
+ dir_cleanup = DirCleanup(dest_path)
+
+ abspath = source
+ copy = False
+ if src_repo.local() and dest_repo.local():
+ abspath = os.path.abspath(source)
+ copy = not pull and not rev
+
+ src_lock, dest_lock = None, None
+ 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
+ src_lock = src_repo.lock()
+ except lock.LockException:
+ copy = False
+
+ if copy:
+ # we lock here to avoid premature writing to the target
+ dest_lock = lock.lock(os.path.join(dest_path, ".hg", "lock"))
+
+ # we need to remove the (empty) data dir in dest so copyfiles
+ # can do its work
+ os.rmdir(os.path.join(dest_path, ".hg", "data"))
+ files = "data 00manifest.d 00manifest.i 00changelog.d 00changelog.i"
+ for f in files.split():
+ src = os.path.join(source, ".hg", f)
+ dst = os.path.join(dest_path, ".hg", f)
+ try:
+ util.copyfiles(src, dst)
+ except OSError, inst:
+ if inst.errno != errno.ENOENT:
+ raise
+
+ # we need to re-init the repo after manually copying the data
+ # into it
+ dest_repo = repository(ui, dest)
+
+ else:
+ revs = None
+ if rev:
+ if not src_repo.local():
+ raise util.Abort(_("clone by revision not supported yet "
+ "for remote repositories"))
+ revs = [src_repo.lookup(r) for r in rev]
+
+ if dest_repo.local():
+ dest_repo.pull(src_repo, heads=revs)
+ elif src_repo.local():
+ src_repo.push(dest_repo, revs=revs)
+ else:
+ raise util.Abort(_("clone from remote to remote not supported"))
+
+ if src_lock:
+ src_lock.release()
+
+ if dest_repo.local():
+ fp = dest_repo.opener("hgrc", "w", text=True)
+ fp.write("[paths]\n")
+ fp.write("default = %s\n" % abspath)
+ fp.close()
+
+ if dest_lock:
+ dest_lock.release()
+
+ if update:
+ dest_repo.update(dest_repo.changelog.tip())
+ if dir_cleanup:
+ dir_cleanup.close()
+
+ return src_repo, dest_repo