mercurial/hgweb/webcommands.py
changeset 5598 d534ba1c4eb4
parent 5595 b95b2525c6e8
child 5600 9d900f7282e6
equal deleted inserted replaced
5597:e7f99a3ed008 5598:d534ba1c4eb4
     3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
     3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
     4 #
     4 #
     5 # This software may be used and distributed according to the terms
     5 # This software may be used and distributed according to the terms
     6 # of the GNU General Public License, incorporated herein by reference.
     6 # of the GNU General Public License, incorporated herein by reference.
     7 
     7 
     8 import cStringIO, zlib, bz2, tempfile, errno, os, sys
     8 import os
     9 from mercurial import revlog, util, streamclone
     9 from mercurial import revlog
    10 from mercurial.i18n import gettext as _
       
    11 from mercurial.node import *
       
    12 from common import staticfile
    10 from common import staticfile
    13 
    11 
    14 def log(web, req):
    12 def log(web, req):
    15     if req.form.has_key('file') and req.form['file'][0]:
    13     if req.form.has_key('file') and req.form['file'][0]:
    16         filelog(web, req)
    14         filelog(web, req)
    71     req.write(web.fileannotate(web.filectx(req)))
    69     req.write(web.fileannotate(web.filectx(req)))
    72 
    70 
    73 def filelog(web, req):
    71 def filelog(web, req):
    74     req.write(web.filelog(web.filectx(req)))
    72     req.write(web.filelog(web.filectx(req)))
    75 
    73 
    76 def lookup(web, req):
       
    77     try:
       
    78         r = hex(web.repo.lookup(req.form['key'][0]))
       
    79         success = 1
       
    80     except Exception,inst:
       
    81         r = str(inst)
       
    82         success = 0
       
    83     resp = "%s %s\n" % (success, r)
       
    84     req.httphdr("application/mercurial-0.1", length=len(resp))
       
    85     req.write(resp)
       
    86 
       
    87 def heads(web, req):
       
    88     resp = " ".join(map(hex, web.repo.heads())) + "\n"
       
    89     req.httphdr("application/mercurial-0.1", length=len(resp))
       
    90     req.write(resp)
       
    91 
       
    92 def branches(web, req):
       
    93     nodes = []
       
    94     if req.form.has_key('nodes'):
       
    95         nodes = map(bin, req.form['nodes'][0].split(" "))
       
    96     resp = cStringIO.StringIO()
       
    97     for b in web.repo.branches(nodes):
       
    98         resp.write(" ".join(map(hex, b)) + "\n")
       
    99     resp = resp.getvalue()
       
   100     req.httphdr("application/mercurial-0.1", length=len(resp))
       
   101     req.write(resp)
       
   102 
       
   103 def between(web, req):
       
   104     if req.form.has_key('pairs'):
       
   105         pairs = [map(bin, p.split("-"))
       
   106                  for p in req.form['pairs'][0].split(" ")]
       
   107     resp = cStringIO.StringIO()
       
   108     for b in web.repo.between(pairs):
       
   109         resp.write(" ".join(map(hex, b)) + "\n")
       
   110     resp = resp.getvalue()
       
   111     req.httphdr("application/mercurial-0.1", length=len(resp))
       
   112     req.write(resp)
       
   113 
       
   114 def changegroup(web, req):
       
   115     req.httphdr("application/mercurial-0.1")
       
   116     nodes = []
       
   117     if not web.allowpull:
       
   118         return
       
   119 
       
   120     if req.form.has_key('roots'):
       
   121         nodes = map(bin, req.form['roots'][0].split(" "))
       
   122 
       
   123     z = zlib.compressobj()
       
   124     f = web.repo.changegroup(nodes, 'serve')
       
   125     while 1:
       
   126         chunk = f.read(4096)
       
   127         if not chunk:
       
   128             break
       
   129         req.write(z.compress(chunk))
       
   130 
       
   131     req.write(z.flush())
       
   132 
       
   133 def changegroupsubset(web, req):
       
   134     req.httphdr("application/mercurial-0.1")
       
   135     bases = []
       
   136     heads = []
       
   137     if not web.allowpull:
       
   138         return
       
   139 
       
   140     if req.form.has_key('bases'):
       
   141         bases = [bin(x) for x in req.form['bases'][0].split(' ')]
       
   142     if req.form.has_key('heads'):
       
   143         heads = [bin(x) for x in req.form['heads'][0].split(' ')]
       
   144 
       
   145     z = zlib.compressobj()
       
   146     f = web.repo.changegroupsubset(bases, heads, 'serve')
       
   147     while 1:
       
   148         chunk = f.read(4096)
       
   149         if not chunk:
       
   150             break
       
   151         req.write(z.compress(chunk))
       
   152 
       
   153     req.write(z.flush())
       
   154 
       
   155 def archive(web, req):
    74 def archive(web, req):
   156     type_ = req.form['type'][0]
    75     type_ = req.form['type'][0]
   157     allowed = web.configlist("web", "allow_archive")
    76     allowed = web.configlist("web", "allow_archive")
   158     if (type_ in web.archives and (type_ in allowed or
    77     if (type_ in web.archives and (type_ in allowed or
   159         web.configbool("web", "allow" + type_, False))):
    78         web.configbool("web", "allow" + type_, False))):
   169     # readable by the user running the CGI script
    88     # readable by the user running the CGI script
   170     static = web.config("web", "static",
    89     static = web.config("web", "static",
   171                         os.path.join(web.templatepath, "static"),
    90                         os.path.join(web.templatepath, "static"),
   172                         untrusted=False)
    91                         untrusted=False)
   173     req.write(staticfile(static, fname, req))
    92     req.write(staticfile(static, fname, req))
   174 
       
   175 def capabilities(web, req):
       
   176     caps = ['lookup', 'changegroupsubset']
       
   177     if web.configbool('server', 'uncompressed'):
       
   178         caps.append('stream=%d' % web.repo.changelog.version)
       
   179     # XXX: make configurable and/or share code with do_unbundle:
       
   180     unbundleversions = ['HG10GZ', 'HG10BZ', 'HG10UN']
       
   181     if unbundleversions:
       
   182         caps.append('unbundle=%s' % ','.join(unbundleversions))
       
   183     resp = ' '.join(caps)
       
   184     req.httphdr("application/mercurial-0.1", length=len(resp))
       
   185     req.write(resp)
       
   186 
       
   187 def unbundle(web, req):
       
   188     def bail(response, headers={}):
       
   189         length = int(req.env['CONTENT_LENGTH'])
       
   190         for s in util.filechunkiter(req, limit=length):
       
   191             # drain incoming bundle, else client will not see
       
   192             # response when run outside cgi script
       
   193             pass
       
   194         req.httphdr("application/mercurial-0.1", headers=headers)
       
   195         req.write('0\n')
       
   196         req.write(response)
       
   197 
       
   198     # require ssl by default, auth info cannot be sniffed and
       
   199     # replayed
       
   200     ssl_req = web.configbool('web', 'push_ssl', True)
       
   201     if ssl_req:
       
   202         if req.env.get('wsgi.url_scheme') != 'https':
       
   203             bail(_('ssl required\n'))
       
   204             return
       
   205         proto = 'https'
       
   206     else:
       
   207         proto = 'http'
       
   208 
       
   209     # do not allow push unless explicitly allowed
       
   210     if not web.check_perm(req, 'push', False):
       
   211         bail(_('push not authorized\n'),
       
   212              headers={'status': '401 Unauthorized'})
       
   213         return
       
   214 
       
   215     their_heads = req.form['heads'][0].split(' ')
       
   216 
       
   217     def check_heads():
       
   218         heads = map(hex, web.repo.heads())
       
   219         return their_heads == [hex('force')] or their_heads == heads
       
   220 
       
   221     # fail early if possible
       
   222     if not check_heads():
       
   223         bail(_('unsynced changes\n'))
       
   224         return
       
   225 
       
   226     req.httphdr("application/mercurial-0.1")
       
   227 
       
   228     # do not lock repo until all changegroup data is
       
   229     # streamed. save to temporary file.
       
   230 
       
   231     fd, tempname = tempfile.mkstemp(prefix='hg-unbundle-')
       
   232     fp = os.fdopen(fd, 'wb+')
       
   233     try:
       
   234         length = int(req.env['CONTENT_LENGTH'])
       
   235         for s in util.filechunkiter(req, limit=length):
       
   236             fp.write(s)
       
   237 
       
   238         try:
       
   239             lock = web.repo.lock()
       
   240             try:
       
   241                 if not check_heads():
       
   242                     req.write('0\n')
       
   243                     req.write(_('unsynced changes\n'))
       
   244                     return
       
   245 
       
   246                 fp.seek(0)
       
   247                 header = fp.read(6)
       
   248                 if not header.startswith("HG"):
       
   249                     # old client with uncompressed bundle
       
   250                     def generator(f):
       
   251                         yield header
       
   252                         for chunk in f:
       
   253                             yield chunk
       
   254                 elif not header.startswith("HG10"):
       
   255                     req.write("0\n")
       
   256                     req.write(_("unknown bundle version\n"))
       
   257                     return
       
   258                 elif header == "HG10GZ":
       
   259                     def generator(f):
       
   260                         zd = zlib.decompressobj()
       
   261                         for chunk in f:
       
   262                             yield zd.decompress(chunk)
       
   263                 elif header == "HG10BZ":
       
   264                     def generator(f):
       
   265                         zd = bz2.BZ2Decompressor()
       
   266                         zd.decompress("BZ")
       
   267                         for chunk in f:
       
   268                             yield zd.decompress(chunk)
       
   269                 elif header == "HG10UN":
       
   270                     def generator(f):
       
   271                         for chunk in f:
       
   272                             yield chunk
       
   273                 else:
       
   274                     req.write("0\n")
       
   275                     req.write(_("unknown bundle compression type\n"))
       
   276                     return
       
   277                 gen = generator(util.filechunkiter(fp, 4096))
       
   278 
       
   279                 # send addchangegroup output to client
       
   280 
       
   281                 old_stdout = sys.stdout
       
   282                 sys.stdout = cStringIO.StringIO()
       
   283 
       
   284                 try:
       
   285                     url = 'remote:%s:%s' % (proto,
       
   286                                             req.env.get('REMOTE_HOST', ''))
       
   287                     try:
       
   288                         ret = web.repo.addchangegroup(
       
   289                                     util.chunkbuffer(gen), 'serve', url)
       
   290                     except util.Abort, inst:
       
   291                         sys.stdout.write("abort: %s\n" % inst)
       
   292                         ret = 0
       
   293                 finally:
       
   294                     val = sys.stdout.getvalue()
       
   295                     sys.stdout = old_stdout
       
   296                 req.write('%d\n' % ret)
       
   297                 req.write(val)
       
   298             finally:
       
   299                 del lock
       
   300         except (OSError, IOError), inst:
       
   301             req.write('0\n')
       
   302             filename = getattr(inst, 'filename', '')
       
   303             # Don't send our filesystem layout to the client
       
   304             if filename.startswith(web.repo.root):
       
   305                 filename = filename[len(web.repo.root)+1:]
       
   306             else:
       
   307                 filename = ''
       
   308             error = getattr(inst, 'strerror', 'Unknown error')
       
   309             if inst.errno == errno.ENOENT:
       
   310                 code = 404
       
   311             else:
       
   312                 code = 500
       
   313             req.respond(code, '%s: %s\n' % (error, filename))
       
   314     finally:
       
   315         fp.close()
       
   316         os.unlink(tempname)
       
   317 
       
   318 def stream_out(web, req):
       
   319     req.httphdr("application/mercurial-0.1")
       
   320     streamclone.stream_out(web.repo, req, untrusted=True)