comparison upmana/mercurial/hgweb/protocol.py @ 135:dcf4fbe09b70 beta

Traipse Beta 'OpenRPG' {091010-00} Traipse is a distribution of OpenRPG that is designed to be easy to setup and go. Traipse also makes it easy for developers to work on code without fear of sacrifice. 'Ornery-Orc' continues the trend of 'Grumpy' and adds fixes to the code. 'Ornery-Orc's main goal is to offer more advanced features and enhance the productivity of the user. Update Summary (Beta) Added Bookmarks Fix to Remote Admin Commands Minor fix to text based Server Fix to Pretty Print, from Core Fix to Splitter Nodes not being created Fix to massive amounts of images loading, from Core Added 'boot' command to remote admin Added confirmation window for sent nodes Minor changes to allow for portability to an OpenSUSE linux OS Miniatures Layer pop up box allows users to turn off Mini labels, from FlexiRPG Zoom Mouse plugin added Images added to Plugin UI Switching to Element Tree Map efficiency, from FlexiRPG Added Status Bar to Update Manager default_manifest.xml renamed to default_upmana.xml Cleaner clode for saved repositories New TrueDebug Class in orpg_log (See documentation for usage) Mercurial's hgweb folder is ported to upmana **Pretty important update that can help remove thousands of dead children from your gametree. **Children, <forms />, <group_atts />, <horizontal />, <cols />, <rows />, <height />, etc... are all tags now. Check your gametree and look for dead children!! **New Gamtree Recusion method, mapping, and context sensitivity. !!Alpha - Watch out for infinite loops!!
author sirebral
date Tue, 10 Nov 2009 14:11:28 -0600
parents
children
comparison
equal deleted inserted replaced
101:394ebb3b6a0f 135:dcf4fbe09b70
1 #
2 # Copyright 21 May 2005 - (c) 2005 Jake Edge <jake@edge2.net>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2, incorporated herein by reference.
7
8 import cStringIO, zlib, tempfile, errno, os, sys, urllib
9 from upmana.mercurial import util, streamclone
10 from upmana.mercurial.node import bin, hex
11 from upmana.mercurial import changegroup as changegroupmod
12 from common import ErrorResponse, HTTP_OK, HTTP_NOT_FOUND, HTTP_SERVER_ERROR
13
14 # __all__ is populated with the allowed commands. Be sure to add to it if
15 # you're adding a new command, or the new command won't work.
16
17 __all__ = [
18 'lookup', 'heads', 'branches', 'between', 'changegroup',
19 'changegroupsubset', 'capabilities', 'unbundle', 'stream_out',
20 'branchmap',
21 ]
22
23 HGTYPE = 'application/mercurial-0.1'
24
25 def lookup(repo, req):
26 try:
27 r = hex(repo.lookup(req.form['key'][0]))
28 success = 1
29 except Exception,inst:
30 r = str(inst)
31 success = 0
32 resp = "%s %s\n" % (success, r)
33 req.respond(HTTP_OK, HGTYPE, length=len(resp))
34 yield resp
35
36 def heads(repo, req):
37 resp = " ".join(map(hex, repo.heads())) + "\n"
38 req.respond(HTTP_OK, HGTYPE, length=len(resp))
39 yield resp
40
41 def branchmap(repo, req):
42 branches = repo.branchmap()
43 heads = []
44 for branch, nodes in branches.iteritems():
45 branchname = urllib.quote(branch)
46 branchnodes = [hex(node) for node in nodes]
47 heads.append('%s %s' % (branchname, ' '.join(branchnodes)))
48 resp = '\n'.join(heads)
49 req.respond(HTTP_OK, HGTYPE, length=len(resp))
50 yield resp
51
52 def branches(repo, req):
53 nodes = []
54 if 'nodes' in req.form:
55 nodes = map(bin, req.form['nodes'][0].split(" "))
56 resp = cStringIO.StringIO()
57 for b in repo.branches(nodes):
58 resp.write(" ".join(map(hex, b)) + "\n")
59 resp = resp.getvalue()
60 req.respond(HTTP_OK, HGTYPE, length=len(resp))
61 yield resp
62
63 def between(repo, req):
64 if 'pairs' in req.form:
65 pairs = [map(bin, p.split("-"))
66 for p in req.form['pairs'][0].split(" ")]
67 resp = cStringIO.StringIO()
68 for b in repo.between(pairs):
69 resp.write(" ".join(map(hex, b)) + "\n")
70 resp = resp.getvalue()
71 req.respond(HTTP_OK, HGTYPE, length=len(resp))
72 yield resp
73
74 def changegroup(repo, req):
75 req.respond(HTTP_OK, HGTYPE)
76 nodes = []
77
78 if 'roots' in req.form:
79 nodes = map(bin, req.form['roots'][0].split(" "))
80
81 z = zlib.compressobj()
82 f = repo.changegroup(nodes, 'serve')
83 while 1:
84 chunk = f.read(4096)
85 if not chunk:
86 break
87 yield z.compress(chunk)
88
89 yield z.flush()
90
91 def changegroupsubset(repo, req):
92 req.respond(HTTP_OK, HGTYPE)
93 bases = []
94 heads = []
95
96 if 'bases' in req.form:
97 bases = [bin(x) for x in req.form['bases'][0].split(' ')]
98 if 'heads' in req.form:
99 heads = [bin(x) for x in req.form['heads'][0].split(' ')]
100
101 z = zlib.compressobj()
102 f = repo.changegroupsubset(bases, heads, 'serve')
103 while 1:
104 chunk = f.read(4096)
105 if not chunk:
106 break
107 yield z.compress(chunk)
108
109 yield z.flush()
110
111 def capabilities(repo, req):
112 caps = ['lookup', 'changegroupsubset', 'branchmap']
113 if repo.ui.configbool('server', 'uncompressed', untrusted=True):
114 caps.append('stream=%d' % repo.changelog.version)
115 if changegroupmod.bundlepriority:
116 caps.append('unbundle=%s' % ','.join(changegroupmod.bundlepriority))
117 rsp = ' '.join(caps)
118 req.respond(HTTP_OK, HGTYPE, length=len(rsp))
119 yield rsp
120
121 def unbundle(repo, req):
122
123 proto = req.env.get('wsgi.url_scheme') or 'http'
124 their_heads = req.form['heads'][0].split(' ')
125
126 def check_heads():
127 heads = map(hex, repo.heads())
128 return their_heads == [hex('force')] or their_heads == heads
129
130 # fail early if possible
131 if not check_heads():
132 req.drain()
133 raise ErrorResponse(HTTP_OK, 'unsynced changes')
134
135 # do not lock repo until all changegroup data is
136 # streamed. save to temporary file.
137
138 fd, tempname = tempfile.mkstemp(prefix='hg-unbundle-')
139 fp = os.fdopen(fd, 'wb+')
140 try:
141 length = int(req.env['CONTENT_LENGTH'])
142 for s in util.filechunkiter(req, limit=length):
143 fp.write(s)
144
145 try:
146 lock = repo.lock()
147 try:
148 if not check_heads():
149 raise ErrorResponse(HTTP_OK, 'unsynced changes')
150
151 fp.seek(0)
152 header = fp.read(6)
153 if header.startswith('HG') and not header.startswith('HG10'):
154 raise ValueError('unknown bundle version')
155 elif header not in changegroupmod.bundletypes:
156 raise ValueError('unknown bundle compression type')
157 gen = changegroupmod.unbundle(header, fp)
158
159 # send addchangegroup output to client
160
161 oldio = sys.stdout, sys.stderr
162 sys.stderr = sys.stdout = cStringIO.StringIO()
163
164 try:
165 url = 'remote:%s:%s:%s' % (
166 proto,
167 urllib.quote(req.env.get('REMOTE_HOST', '')),
168 urllib.quote(req.env.get('REMOTE_USER', '')))
169 try:
170 ret = repo.addchangegroup(gen, 'serve', url)
171 except util.Abort, inst:
172 sys.stdout.write("abort: %s\n" % inst)
173 ret = 0
174 finally:
175 val = sys.stdout.getvalue()
176 sys.stdout, sys.stderr = oldio
177 req.respond(HTTP_OK, HGTYPE)
178 return '%d\n%s' % (ret, val),
179 finally:
180 lock.release()
181 except ValueError, inst:
182 raise ErrorResponse(HTTP_OK, inst)
183 except (OSError, IOError), inst:
184 filename = getattr(inst, 'filename', '')
185 # Don't send our filesystem layout to the client
186 if filename.startswith(repo.root):
187 filename = filename[len(repo.root)+1:]
188 else:
189 filename = ''
190 error = getattr(inst, 'strerror', 'Unknown error')
191 if inst.errno == errno.ENOENT:
192 code = HTTP_NOT_FOUND
193 else:
194 code = HTTP_SERVER_ERROR
195 raise ErrorResponse(code, '%s: %s' % (error, filename))
196 finally:
197 fp.close()
198 os.unlink(tempname)
199
200 def stream_out(repo, req):
201 req.respond(HTTP_OK, HGTYPE)
202 try:
203 for chunk in streamclone.stream_out(repo, untrusted=True):
204 yield chunk
205 except streamclone.StreamException, inst:
206 yield str(inst)