[PATCH] schemes extension

[Alexander Solovyov]

# HG changeset patch
# User Alexander Solovyov <piranha at piranha.org.ua>
# Date 1259077703 -7200
# Node ID 814401c8e5da8aad9c149a3e9050b83b1ca8df89
# Parent  3d718761157b9dba2b7a9e1a45858bcc4d493766
schemes extension

diff --git a/hgext/schemes.py b/hgext/schemes.py
new file mode 100644
--- /dev/null
+++ b/hgext/schemes.py
@@ -0,0 +1,71 @@
+# Copyright 2009, Alexander Solovyov under terms of GPLv2
+"""extend schemes with shortcuts to repository swarms
+
+This extension allows you to specify shortcuts for parent urls with a lot of
+repositories to act like a scheme, for example::
+
+  [schemes]
+  py = http://code.python.org/hg/
+
+After that you can use it like::
+
+  hg clone py://trunk/
+
+Of course, it is possible to supply custom urls for schemes with suffixes::
+
+  [schemes]
+  ex = ssh://example.com/hg
+  ex+http = http://hg.example.com/
+
+Additionally there is support for some more complex schemas, for example used
+by Google Code::
+
+  [schemes]
+  gcode = http://{project}.googlecode.com/hg/
+
+The syntax is taken from Mercurial templates, there are two variables present,
+namely {project} and {repo}, which are determined from the url you're supplying:
+gcode://project/repo.
+"""
+
+from mercurial import hg, ui, util, templater
+
+
+class ShortRepository(object):
+    def __init__(self, url, scheme, templater):
+        self.scheme = scheme
+        self.templater = templater
+        if not '{project}' in url:
+            url += '{project}/'
+        if not '{repo}' in url:
+            url += '{repo}'
+        self.url = url
+
+    def __repr__(self):
+        return '<ShortRepository: %s>' % self.scheme
+
+    def instance(self, ui, url, create):
+        url = url.split('://', 1)[1]
+        try:
+            project, repo = url.split('/', 1)
+        except ValueError:
+            project, repo = url, None
+        context = {'project': project, 'repo': repo}
+        url = ''.join(self.templater.process(self.url, context))
+        return hg._lookup(url).instance(ui, url, create)
+
+schemes = {
+    'py': 'http://hg.python.org/',
+    'bb': 'http://bitbucket.org/',
+    'bb+ssh': 'ssh://hg@bitbucket.org/',
+    'gcode': 'http://{project}.googlecode.com/hg/'
+    }
+
+def getschemes():
+    u = ui.ui()
+    t = templater.engine(lambda x: x)
+    schemes.update(dict(u.configitems('schemes')))
+    for scheme, url in schemes.items():
+        hg.schemes[scheme] = ShortRepository(url, scheme, t)
+
+getschemes()
diff --git a/tests/test-schemes b/tests/test-schemes
new file mode 100755
--- /dev/null
+++ b/tests/test-schemes
@@ -0,0 +1,22 @@
+#!/bin/sh
+
+cat <<EOF >> $HGRCPATH
+[extensions]
+schemes=
+
+[schemes]
+l = http://localhost:$HGPORT/
+EOF
+
+hg init test
+cd test
+echo a > a
+hg ci -Am initial
+
+hg serve -n test -p $HGPORT -d --pid-file=hg.pid -A access.log -E errors.log
+cat hg.pid >> $DAEMON_PIDS
+
+hg incoming l://
+
+echo % errors
+cat errors.log
diff --git a/tests/test-schemes.out b/tests/test-schemes.out
new file mode 100644
--- /dev/null
+++ b/tests/test-schemes.out
@@ -0,0 +1,5 @@
+adding a
+comparing with l://
+searching for changes
+no changes found
+% errors