Mercurial > hg > mercurial-crew-with-dirclash

comparison mercurial/hgweb/__init__.py @ 2311:b832b6eb65ab

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Moving hgweb.py into it's own module in preparation for breaking it up.
author Eric Hopper <hopper@omnifarious.org>
date Thu, 18 May 2006 11:48:33 -0700
parents mercurial/hgweb.py@714f4d25a7a9
children f789602ba840
comparison
equal deleted inserted replaced
2307:5b3a3e35f084 2311:b832b6eb65ab
1 # hgweb.py - web interface to a mercurial repository
2 #
3 # Copyright 21 May 2005 - (c) 2005 Jake Edge <jake@edge2.net>
4 # Copyright 2005 Matt Mackall <mpm@selenic.com>
5 #
6 # This software may be used and distributed according to the terms
7 # of the GNU General Public License, incorporated herein by reference.
8
9 import os, cgi, sys
10 import mimetypes
11 from mercurial.demandload import demandload
12 demandload(globals(), "time re socket zlib errno ConfigParser tempfile")
13 demandload(globals(), "StringIO BaseHTTPServer SocketServer urllib")
14 demandload(globals(), "mercurial:mdiff,ui,hg,util,archival,templater")
15 from mercurial.node import *
16 from mercurial.i18n import gettext as _
17
18 def splitURI(uri):
19 """ Return path and query splited from uri
20
21 Just like CGI environment, the path is unquoted, the query is
22 not.
23 """
24 if '?' in uri:
25 path, query = uri.split('?', 1)
26 else:
27 path, query = uri, ''
28 return urllib.unquote(path), query
29
30 def up(p):
31 if p[0] != "/":
32 p = "/" + p
33 if p[-1] == "/":
34 p = p[:-1]
35 up = os.path.dirname(p)
36 if up == "/":
37 return "/"
38 return up + "/"
39
40 def get_mtime(repo_path):
41 hg_path = os.path.join(repo_path, ".hg")
42 cl_path = os.path.join(hg_path, "00changelog.i")
43 if os.path.exists(os.path.join(cl_path)):
44 return os.stat(cl_path).st_mtime
45 else:
46 return os.stat(hg_path).st_mtime
47
48 def staticfile(directory, fname):
49 """return a file inside directory with guessed content-type header
50
51 fname always uses '/' as directory separator and isn't allowed to
52 contain unusual path components.
53 Content-type is guessed using the mimetypes module.
54 Return an empty string if fname is illegal or file not found.
55
56 """
57 parts = fname.split('/')
58 path = directory
59 for part in parts:
60 if (part in ('', os.curdir, os.pardir) or
61 os.sep in part or os.altsep is not None and os.altsep in part):
62 return ""
63 path = os.path.join(path, part)
64 try:
65 os.stat(path)
66 ct = mimetypes.guess_type(path)[0] or "text/plain"
67 return "Content-type: %s\n\n%s" % (ct, file(path).read())
68 except (TypeError, OSError):
69 # illegal fname or unreadable file
70 return ""
71
72 class hgrequest(object):
73 def __init__(self, inp=None, out=None, env=None):
74 self.inp = inp or sys.stdin
75 self.out = out or sys.stdout
76 self.env = env or os.environ
77 self.form = cgi.parse(self.inp, self.env, keep_blank_values=1)
78
79 def write(self, *things):
80 for thing in things:
81 if hasattr(thing, "__iter__"):
82 for part in thing:
83 self.write(part)
84 else:
85 try:
86 self.out.write(str(thing))
87 except socket.error, inst:
88 if inst[0] != errno.ECONNRESET:
89 raise
90
91 def header(self, headers=[('Content-type','text/html')]):
92 for header in headers:
93 self.out.write("%s: %s\r\n" % header)
94 self.out.write("\r\n")
95
96 def httphdr(self, type, file="", size=0):
97
98 headers = [('Content-type', type)]
99 if file:
100 headers.append(('Content-disposition', 'attachment; filename=%s' % file))
101 if size > 0:
102 headers.append(('Content-length', str(size)))
103 self.header(headers)
104
105 class hgweb(object):
106 def __init__(self, repo, name=None):
107 if type(repo) == type(""):
108 self.repo = hg.repository(ui.ui(), repo)
109 else:
110 self.repo = repo
111
112 self.mtime = -1
113 self.reponame = name
114 self.archives = 'zip', 'gz', 'bz2'
115
116 def refresh(self):
117 mtime = get_mtime(self.repo.root)
118 if mtime != self.mtime:
119 self.mtime = mtime
120 self.repo = hg.repository(self.repo.ui, self.repo.root)
121 self.maxchanges = int(self.repo.ui.config("web", "maxchanges", 10))
122 self.maxfiles = int(self.repo.ui.config("web", "maxfiles", 10))
123 self.allowpull = self.repo.ui.configbool("web", "allowpull", True)
124
125 def archivelist(self, nodeid):
126 for i in self.archives:
127 if self.repo.ui.configbool("web", "allow" + i, False):
128 yield {"type" : i, "node" : nodeid, "url": ""}
129
130 def listfiles(self, files, mf):
131 for f in files[:self.maxfiles]:
132 yield self.t("filenodelink", node=hex(mf[f]), file=f)
133 if len(files) > self.maxfiles:
134 yield self.t("fileellipses")
135
136 def listfilediffs(self, files, changeset):
137 for f in files[:self.maxfiles]:
138 yield self.t("filedifflink", node=hex(changeset), file=f)
139 if len(files) > self.maxfiles:
140 yield self.t("fileellipses")
141
142 def siblings(self, siblings=[], rev=None, hiderev=None, **args):
143 if not rev:
144 rev = lambda x: ""
145 siblings = [s for s in siblings if s != nullid]
146 if len(siblings) == 1 and rev(siblings[0]) == hiderev:
147 return
148 for s in siblings:
149 yield dict(node=hex(s), rev=rev(s), **args)
150
151 def renamelink(self, fl, node):
152 r = fl.renamed(node)
153 if r:
154 return [dict(file=r[0], node=hex(r[1]))]
155 return []
156
157 def showtag(self, t1, node=nullid, **args):
158 for t in self.repo.nodetags(node):
159 yield self.t(t1, tag=t, **args)
160
161 def diff(self, node1, node2, files):
162 def filterfiles(filters, files):
163 l = [x for x in files if x in filters]
164
165 for t in filters:
166 if t and t[-1] != os.sep:
167 t += os.sep
168 l += [x for x in files if x.startswith(t)]
169 return l
170
171 parity = [0]
172 def diffblock(diff, f, fn):
173 yield self.t("diffblock",
174 lines=prettyprintlines(diff),
175 parity=parity[0],
176 file=f,
177 filenode=hex(fn or nullid))
178 parity[0] = 1 - parity[0]
179
180 def prettyprintlines(diff):
181 for l in diff.splitlines(1):
182 if l.startswith('+'):
183 yield self.t("difflineplus", line=l)
184 elif l.startswith('-'):
185 yield self.t("difflineminus", line=l)
186 elif l.startswith('@'):
187 yield self.t("difflineat", line=l)
188 else:
189 yield self.t("diffline", line=l)
190
191 r = self.repo
192 cl = r.changelog
193 mf = r.manifest
194 change1 = cl.read(node1)
195 change2 = cl.read(node2)
196 mmap1 = mf.read(change1[0])
197 mmap2 = mf.read(change2[0])
198 date1 = util.datestr(change1[2])
199 date2 = util.datestr(change2[2])
200
201 modified, added, removed, deleted, unknown = r.changes(node1, node2)
202 if files:
203 modified, added, removed = map(lambda x: filterfiles(files, x),
204 (modified, added, removed))
205
206 diffopts = self.repo.ui.diffopts()
207 showfunc = diffopts['showfunc']
208 ignorews = diffopts['ignorews']
209 for f in modified:
210 to = r.file(f).read(mmap1[f])
211 tn = r.file(f).read(mmap2[f])
212 yield diffblock(mdiff.unidiff(to, date1, tn, date2, f,
213 showfunc=showfunc, ignorews=ignorews), f, tn)
214 for f in added:
215 to = None
216 tn = r.file(f).read(mmap2[f])
217 yield diffblock(mdiff.unidiff(to, date1, tn, date2, f,
218 showfunc=showfunc, ignorews=ignorews), f, tn)
219 for f in removed:
220 to = r.file(f).read(mmap1[f])
221 tn = None
222 yield diffblock(mdiff.unidiff(to, date1, tn, date2, f,
223 showfunc=showfunc, ignorews=ignorews), f, tn)
224
225 def changelog(self, pos):
226 def changenav(**map):
227 def seq(factor, maxchanges=None):
228 if maxchanges:
229 yield maxchanges
230 if maxchanges >= 20 and maxchanges <= 40:
231 yield 50
232 else:
233 yield 1 * factor
234 yield 3 * factor
235 for f in seq(factor * 10):
236 yield f
237
238 l = []
239 last = 0
240 for f in seq(1, self.maxchanges):
241 if f < self.maxchanges or f <= last:
242 continue
243 if f > count:
244 break
245 last = f
246 r = "%d" % f
247 if pos + f < count:
248 l.append(("+" + r, pos + f))
249 if pos - f >= 0:
250 l.insert(0, ("-" + r, pos - f))
251
252 yield {"rev": 0, "label": "(0)"}
253
254 for label, rev in l:
255 yield {"label": label, "rev": rev}
256
257 yield {"label": "tip", "rev": "tip"}
258
259 def changelist(**map):
260 parity = (start - end) & 1
261 cl = self.repo.changelog
262 l = [] # build a list in forward order for efficiency
263 for i in range(start, end):
264 n = cl.node(i)
265 changes = cl.read(n)
266 hn = hex(n)
267
268 l.insert(0, {"parity": parity,
269 "author": changes[1],
270 "parent": self.siblings(cl.parents(n), cl.rev,
271 cl.rev(n) - 1),
272 "child": self.siblings(cl.children(n), cl.rev,
273 cl.rev(n) + 1),
274 "changelogtag": self.showtag("changelogtag",n),
275 "manifest": hex(changes[0]),
276 "desc": changes[4],
277 "date": changes[2],
278 "files": self.listfilediffs(changes[3], n),
279 "rev": i,
280 "node": hn})
281 parity = 1 - parity
282
283 for e in l:
284 yield e
285
286 cl = self.repo.changelog
287 mf = cl.read(cl.tip())[0]
288 count = cl.count()
289 start = max(0, pos - self.maxchanges + 1)
290 end = min(count, start + self.maxchanges)
291 pos = end - 1
292
293 yield self.t('changelog',
294 changenav=changenav,
295 manifest=hex(mf),
296 rev=pos, changesets=count, entries=changelist,
297 archives=self.archivelist("tip"))
298
299 def search(self, query):
300
301 def changelist(**map):
302 cl = self.repo.changelog
303 count = 0
304 qw = query.lower().split()
305
306 def revgen():
307 for i in range(cl.count() - 1, 0, -100):
308 l = []
309 for j in range(max(0, i - 100), i):
310 n = cl.node(j)
311 changes = cl.read(n)
312 l.append((n, j, changes))
313 l.reverse()
314 for e in l:
315 yield e
316
317 for n, i, changes in revgen():
318 miss = 0
319 for q in qw:
320 if not (q in changes[1].lower() or
321 q in changes[4].lower() or
322 q in " ".join(changes[3][:20]).lower()):
323 miss = 1
324 break
325 if miss:
326 continue
327
328 count += 1
329 hn = hex(n)
330
331 yield self.t('searchentry',
332 parity=count & 1,
333 author=changes[1],
334 parent=self.siblings(cl.parents(n), cl.rev),
335 child=self.siblings(cl.children(n), cl.rev),
336 changelogtag=self.showtag("changelogtag",n),
337 manifest=hex(changes[0]),
338 desc=changes[4],
339 date=changes[2],
340 files=self.listfilediffs(changes[3], n),
341 rev=i,
342 node=hn)
343
344 if count >= self.maxchanges:
345 break
346
347 cl = self.repo.changelog
348 mf = cl.read(cl.tip())[0]
349
350 yield self.t('search',
351 query=query,
352 manifest=hex(mf),
353 entries=changelist)
354
355 def changeset(self, nodeid):
356 cl = self.repo.changelog
357 n = self.repo.lookup(nodeid)
358 nodeid = hex(n)
359 changes = cl.read(n)
360 p1 = cl.parents(n)[0]
361
362 files = []
363 mf = self.repo.manifest.read(changes[0])
364 for f in changes[3]:
365 files.append(self.t("filenodelink",
366 filenode=hex(mf.get(f, nullid)), file=f))
367
368 def diff(**map):
369 yield self.diff(p1, n, None)
370
371 yield self.t('changeset',
372 diff=diff,
373 rev=cl.rev(n),
374 node=nodeid,
375 parent=self.siblings(cl.parents(n), cl.rev),
376 child=self.siblings(cl.children(n), cl.rev),
377 changesettag=self.showtag("changesettag",n),
378 manifest=hex(changes[0]),
379 author=changes[1],
380 desc=changes[4],
381 date=changes[2],
382 files=files,
383 archives=self.archivelist(nodeid))
384
385 def filelog(self, f, filenode):
386 cl = self.repo.changelog
387 fl = self.repo.file(f)
388 filenode = hex(fl.lookup(filenode))
389 count = fl.count()
390
391 def entries(**map):
392 l = []
393 parity = (count - 1) & 1
394
395 for i in range(count):
396 n = fl.node(i)
397 lr = fl.linkrev(n)
398 cn = cl.node(lr)
399 cs = cl.read(cl.node(lr))
400
401 l.insert(0, {"parity": parity,
402 "filenode": hex(n),
403 "filerev": i,
404 "file": f,
405 "node": hex(cn),
406 "author": cs[1],
407 "date": cs[2],
408 "rename": self.renamelink(fl, n),
409 "parent": self.siblings(fl.parents(n),
410 fl.rev, file=f),
411 "child": self.siblings(fl.children(n),
412 fl.rev, file=f),
413 "desc": cs[4]})
414 parity = 1 - parity
415
416 for e in l:
417 yield e
418
419 yield self.t("filelog", file=f, filenode=filenode, entries=entries)
420
421 def filerevision(self, f, node):
422 fl = self.repo.file(f)
423 n = fl.lookup(node)
424 node = hex(n)
425 text = fl.read(n)
426 changerev = fl.linkrev(n)
427 cl = self.repo.changelog
428 cn = cl.node(changerev)
429 cs = cl.read(cn)
430 mfn = cs[0]
431
432 mt = mimetypes.guess_type(f)[0]
433 rawtext = text
434 if util.binary(text):
435 mt = mt or 'application/octet-stream'
436 text = "(binary:%s)" % mt
437 mt = mt or 'text/plain'
438
439 def lines():
440 for l, t in enumerate(text.splitlines(1)):
441 yield {"line": t,
442 "linenumber": "% 6d" % (l + 1),
443 "parity": l & 1}
444
445 yield self.t("filerevision",
446 file=f,
447 filenode=node,
448 path=up(f),
449 text=lines(),
450 raw=rawtext,
451 mimetype=mt,
452 rev=changerev,
453 node=hex(cn),
454 manifest=hex(mfn),
455 author=cs[1],
456 date=cs[2],
457 parent=self.siblings(fl.parents(n), fl.rev, file=f),
458 child=self.siblings(fl.children(n), fl.rev, file=f),
459 rename=self.renamelink(fl, n),
460 permissions=self.repo.manifest.readflags(mfn)[f])
461
462 def fileannotate(self, f, node):
463 bcache = {}
464 ncache = {}
465 fl = self.repo.file(f)
466 n = fl.lookup(node)
467 node = hex(n)
468 changerev = fl.linkrev(n)
469
470 cl = self.repo.changelog
471 cn = cl.node(changerev)
472 cs = cl.read(cn)
473 mfn = cs[0]
474
475 def annotate(**map):
476 parity = 1
477 last = None
478 for r, l in fl.annotate(n):
479 try:
480 cnode = ncache[r]
481 except KeyError:
482 cnode = ncache[r] = self.repo.changelog.node(r)
483
484 try:
485 name = bcache[r]
486 except KeyError:
487 cl = self.repo.changelog.read(cnode)
488 bcache[r] = name = self.repo.ui.shortuser(cl[1])
489
490 if last != cnode:
491 parity = 1 - parity
492 last = cnode
493
494 yield {"parity": parity,
495 "node": hex(cnode),
496 "rev": r,
497 "author": name,
498 "file": f,
499 "line": l}
500
501 yield self.t("fileannotate",
502 file=f,
503 filenode=node,
504 annotate=annotate,
505 path=up(f),
506 rev=changerev,
507 node=hex(cn),
508 manifest=hex(mfn),
509 author=cs[1],
510 date=cs[2],
511 rename=self.renamelink(fl, n),
512 parent=self.siblings(fl.parents(n), fl.rev, file=f),
513 child=self.siblings(fl.children(n), fl.rev, file=f),
514 permissions=self.repo.manifest.readflags(mfn)[f])
515
516 def manifest(self, mnode, path):
517 man = self.repo.manifest
518 mn = man.lookup(mnode)
519 mnode = hex(mn)
520 mf = man.read(mn)
521 rev = man.rev(mn)
522 node = self.repo.changelog.node(rev)
523 mff = man.readflags(mn)
524
525 files = {}
526
527 p = path[1:]
528 if p and p[-1] != "/":
529 p += "/"
530 l = len(p)
531
532 for f,n in mf.items():
533 if f[:l] != p:
534 continue
535 remain = f[l:]
536 if "/" in remain:
537 short = remain[:remain.find("/") + 1] # bleah
538 files[short] = (f, None)
539 else:
540 short = os.path.basename(remain)
541 files[short] = (f, n)
542
543 def filelist(**map):
544 parity = 0
545 fl = files.keys()
546 fl.sort()
547 for f in fl:
548 full, fnode = files[f]
549 if not fnode:
550 continue
551
552 yield {"file": full,
553 "manifest": mnode,
554 "filenode": hex(fnode),
555 "parity": parity,
556 "basename": f,
557 "permissions": mff[full]}
558 parity = 1 - parity
559
560 def dirlist(**map):
561 parity = 0
562 fl = files.keys()
563 fl.sort()
564 for f in fl:
565 full, fnode = files[f]
566 if fnode:
567 continue
568
569 yield {"parity": parity,
570 "path": os.path.join(path, f),
571 "manifest": mnode,
572 "basename": f[:-1]}
573 parity = 1 - parity
574
575 yield self.t("manifest",
576 manifest=mnode,
577 rev=rev,
578 node=hex(node),
579 path=path,
580 up=up(path),
581 fentries=filelist,
582 dentries=dirlist,
583 archives=self.archivelist(hex(node)))
584
585 def tags(self):
586 cl = self.repo.changelog
587 mf = cl.read(cl.tip())[0]
588
589 i = self.repo.tagslist()
590 i.reverse()
591
592 def entries(notip=False, **map):
593 parity = 0
594 for k,n in i:
595 if notip and k == "tip": continue
596 yield {"parity": parity,
597 "tag": k,
598 "tagmanifest": hex(cl.read(n)[0]),
599 "date": cl.read(n)[2],
600 "node": hex(n)}
601 parity = 1 - parity
602
603 yield self.t("tags",
604 manifest=hex(mf),
605 entries=lambda **x: entries(False, **x),
606 entriesnotip=lambda **x: entries(True, **x))
607
608 def summary(self):
609 cl = self.repo.changelog
610 mf = cl.read(cl.tip())[0]
611
612 i = self.repo.tagslist()
613 i.reverse()
614
615 def tagentries(**map):
616 parity = 0
617 count = 0
618 for k,n in i:
619 if k == "tip": # skip tip
620 continue;
621
622 count += 1
623 if count > 10: # limit to 10 tags
624 break;
625
626 c = cl.read(n)
627 m = c[0]
628 t = c[2]
629
630 yield self.t("tagentry",
631 parity = parity,
632 tag = k,
633 node = hex(n),
634 date = t,
635 tagmanifest = hex(m))
636 parity = 1 - parity
637
638 def changelist(**map):
639 parity = 0
640 cl = self.repo.changelog
641 l = [] # build a list in forward order for efficiency
642 for i in range(start, end):
643 n = cl.node(i)
644 changes = cl.read(n)
645 hn = hex(n)
646 t = changes[2]
647
648 l.insert(0, self.t(
649 'shortlogentry',
650 parity = parity,
651 author = changes[1],
652 manifest = hex(changes[0]),
653 desc = changes[4],
654 date = t,
655 rev = i,
656 node = hn))
657 parity = 1 - parity
658
659 yield l
660
661 cl = self.repo.changelog
662 mf = cl.read(cl.tip())[0]
663 count = cl.count()
664 start = max(0, count - self.maxchanges)
665 end = min(count, start + self.maxchanges)
666 pos = end - 1
667
668 yield self.t("summary",
669 desc = self.repo.ui.config("web", "description", "unknown"),
670 owner = (self.repo.ui.config("ui", "username") or # preferred
671 self.repo.ui.config("web", "contact") or # deprecated
672 self.repo.ui.config("web", "author", "unknown")), # also
673 lastchange = (0, 0), # FIXME
674 manifest = hex(mf),
675 tags = tagentries,
676 shortlog = changelist)
677
678 def filediff(self, file, changeset):
679 cl = self.repo.changelog
680 n = self.repo.lookup(changeset)
681 changeset = hex(n)
682 p1 = cl.parents(n)[0]
683 cs = cl.read(n)
684 mf = self.repo.manifest.read(cs[0])
685
686 def diff(**map):
687 yield self.diff(p1, n, [file])
688
689 yield self.t("filediff",
690 file=file,
691 filenode=hex(mf.get(file, nullid)),
692 node=changeset,
693 rev=self.repo.changelog.rev(n),
694 parent=self.siblings(cl.parents(n), cl.rev),
695 child=self.siblings(cl.children(n), cl.rev),
696 diff=diff)
697
698 archive_specs = {
699 'bz2': ('application/x-tar', 'tbz2', '.tar.bz2', 'x-bzip2'),
700 'gz': ('application/x-tar', 'tgz', '.tar.gz', 'x-gzip'),
701 'zip': ('application/zip', 'zip', '.zip', None),
702 }
703
704 def archive(self, req, cnode, type):
705 reponame = re.sub(r"\W+", "-", os.path.basename(self.reponame))
706 name = "%s-%s" % (reponame, short(cnode))
707 mimetype, artype, extension, encoding = self.archive_specs[type]
708 headers = [('Content-type', mimetype),
709 ('Content-disposition', 'attachment; filename=%s%s' %
710 (name, extension))]
711 if encoding:
712 headers.append(('Content-encoding', encoding))
713 req.header(headers)
714 archival.archive(self.repo, req.out, cnode, artype, prefix=name)
715
716 # add tags to things
717 # tags -> list of changesets corresponding to tags
718 # find tag, changeset, file
719
720 def run(self, req=hgrequest()):
721 def clean(path):
722 p = util.normpath(path)
723 if p[:2] == "..":
724 raise "suspicious path"
725 return p
726
727 def header(**map):
728 yield self.t("header", **map)
729
730 def footer(**map):
731 yield self.t("footer",
732 motd=self.repo.ui.config("web", "motd", ""),
733 **map)
734
735 def expand_form(form):
736 shortcuts = {
737 'cl': [('cmd', ['changelog']), ('rev', None)],
738 'cs': [('cmd', ['changeset']), ('node', None)],
739 'f': [('cmd', ['file']), ('filenode', None)],
740 'fl': [('cmd', ['filelog']), ('filenode', None)],
741 'fd': [('cmd', ['filediff']), ('node', None)],
742 'fa': [('cmd', ['annotate']), ('filenode', None)],
743 'mf': [('cmd', ['manifest']), ('manifest', None)],
744 'ca': [('cmd', ['archive']), ('node', None)],
745 'tags': [('cmd', ['tags'])],
746 'tip': [('cmd', ['changeset']), ('node', ['tip'])],
747 'static': [('cmd', ['static']), ('file', None)]
748 }
749
750 for k in shortcuts.iterkeys():
751 if form.has_key(k):
752 for name, value in shortcuts[k]:
753 if value is None:
754 value = form[k]
755 form[name] = value
756 del form[k]
757
758 self.refresh()
759
760 expand_form(req.form)
761
762 t = self.repo.ui.config("web", "templates", templater.templatepath())
763 static = self.repo.ui.config("web", "static", os.path.join(t,"static"))
764 m = os.path.join(t, "map")
765 style = self.repo.ui.config("web", "style", "")
766 if req.form.has_key('style'):
767 style = req.form['style'][0]
768 if style:
769 b = os.path.basename("map-" + style)
770 p = os.path.join(t, b)
771 if os.path.isfile(p):
772 m = p
773
774 port = req.env["SERVER_PORT"]
775 port = port != "80" and (":" + port) or ""
776 uri = req.env["REQUEST_URI"]
777 if "?" in uri:
778 uri = uri.split("?")[0]
779 url = "http://%s%s%s" % (req.env["SERVER_NAME"], port, uri)
780 if not self.reponame:
781 self.reponame = (self.repo.ui.config("web", "name")
782 or uri.strip('/') or self.repo.root)
783
784 self.t = templater.templater(m, templater.common_filters,
785 defaults={"url": url,
786 "repo": self.reponame,
787 "header": header,
788 "footer": footer,
789 })
790
791 if not req.form.has_key('cmd'):
792 req.form['cmd'] = [self.t.cache['default'],]
793
794 cmd = req.form['cmd'][0]
795 if cmd == 'changelog':
796 hi = self.repo.changelog.count() - 1
797 if req.form.has_key('rev'):
798 hi = req.form['rev'][0]
799 try:
800 hi = self.repo.changelog.rev(self.repo.lookup(hi))
801 except hg.RepoError:
802 req.write(self.search(hi)) # XXX redirect to 404 page?
803 return
804
805 req.write(self.changelog(hi))
806
807 elif cmd == 'changeset':
808 req.write(self.changeset(req.form['node'][0]))
809
810 elif cmd == 'manifest':
811 req.write(self.manifest(req.form['manifest'][0],
812 clean(req.form['path'][0])))
813
814 elif cmd == 'tags':
815 req.write(self.tags())
816
817 elif cmd == 'summary':
818 req.write(self.summary())
819
820 elif cmd == 'filediff':
821 req.write(self.filediff(clean(req.form['file'][0]),
822 req.form['node'][0]))
823
824 elif cmd == 'file':
825 req.write(self.filerevision(clean(req.form['file'][0]),
826 req.form['filenode'][0]))
827
828 elif cmd == 'annotate':
829 req.write(self.fileannotate(clean(req.form['file'][0]),
830 req.form['filenode'][0]))
831
832 elif cmd == 'filelog':
833 req.write(self.filelog(clean(req.form['file'][0]),
834 req.form['filenode'][0]))
835
836 elif cmd == 'heads':
837 req.httphdr("application/mercurial-0.1")
838 h = self.repo.heads()
839 req.write(" ".join(map(hex, h)) + "\n")
840
841 elif cmd == 'branches':
842 req.httphdr("application/mercurial-0.1")
843 nodes = []
844 if req.form.has_key('nodes'):
845 nodes = map(bin, req.form['nodes'][0].split(" "))
846 for b in self.repo.branches(nodes):
847 req.write(" ".join(map(hex, b)) + "\n")
848
849 elif cmd == 'between':
850 req.httphdr("application/mercurial-0.1")
851 nodes = []
852 if req.form.has_key('pairs'):
853 pairs = [map(bin, p.split("-"))
854 for p in req.form['pairs'][0].split(" ")]
855 for b in self.repo.between(pairs):
856 req.write(" ".join(map(hex, b)) + "\n")
857
858 elif cmd == 'changegroup':
859 req.httphdr("application/mercurial-0.1")
860 nodes = []
861 if not self.allowpull:
862 return
863
864 if req.form.has_key('roots'):
865 nodes = map(bin, req.form['roots'][0].split(" "))
866
867 z = zlib.compressobj()
868 f = self.repo.changegroup(nodes, 'serve')
869 while 1:
870 chunk = f.read(4096)
871 if not chunk:
872 break
873 req.write(z.compress(chunk))
874
875 req.write(z.flush())
876
877 elif cmd == 'archive':
878 changeset = self.repo.lookup(req.form['node'][0])
879 type = req.form['type'][0]
880 if (type in self.archives and
881 self.repo.ui.configbool("web", "allow" + type, False)):
882 self.archive(req, changeset, type)
883 return
884
885 req.write(self.t("error"))
886
887 elif cmd == 'static':
888 fname = req.form['file'][0]
889 req.write(staticfile(static, fname)
890 or self.t("error", error="%r not found" % fname))
891
892 else:
893 req.write(self.t("error"))
894
895 def create_server(ui, repo):
896 use_threads = True
897
898 def openlog(opt, default):
899 if opt and opt != '-':
900 return open(opt, 'w')
901 return default
902
903 address = ui.config("web", "address", "")
904 port = int(ui.config("web", "port", 8000))
905 use_ipv6 = ui.configbool("web", "ipv6")
906 webdir_conf = ui.config("web", "webdir_conf")
907 accesslog = openlog(ui.config("web", "accesslog", "-"), sys.stdout)
908 errorlog = openlog(ui.config("web", "errorlog", "-"), sys.stderr)
909
910 if use_threads:
911 try:
912 from threading import activeCount
913 except ImportError:
914 use_threads = False
915
916 if use_threads:
917 _mixin = SocketServer.ThreadingMixIn
918 else:
919 if hasattr(os, "fork"):
920 _mixin = SocketServer.ForkingMixIn
921 else:
922 class _mixin: pass
923
924 class MercurialHTTPServer(_mixin, BaseHTTPServer.HTTPServer):
925 pass
926
927 class IPv6HTTPServer(MercurialHTTPServer):
928 address_family = getattr(socket, 'AF_INET6', None)
929
930 def __init__(self, *args, **kwargs):
931 if self.address_family is None:
932 raise hg.RepoError(_('IPv6 not available on this system'))
933 BaseHTTPServer.HTTPServer.__init__(self, *args, **kwargs)
934
935 class hgwebhandler(BaseHTTPServer.BaseHTTPRequestHandler):
936
937 def log_error(self, format, *args):
938 errorlog.write("%s - - [%s] %s\n" % (self.address_string(),
939 self.log_date_time_string(),
940 format % args))
941
942 def log_message(self, format, *args):
943 accesslog.write("%s - - [%s] %s\n" % (self.address_string(),
944 self.log_date_time_string(),
945 format % args))
946
947 def do_POST(self):
948 try:
949 self.do_hgweb()
950 except socket.error, inst:
951 if inst[0] != errno.EPIPE:
952 raise
953
954 def do_GET(self):
955 self.do_POST()
956
957 def do_hgweb(self):
958 path_info, query = splitURI(self.path)
959
960 env = {}
961 env['GATEWAY_INTERFACE'] = 'CGI/1.1'
962 env['REQUEST_METHOD'] = self.command
963 env['SERVER_NAME'] = self.server.server_name
964 env['SERVER_PORT'] = str(self.server.server_port)
965 env['REQUEST_URI'] = "/"
966 env['PATH_INFO'] = path_info
967 if query:
968 env['QUERY_STRING'] = query
969 host = self.address_string()
970 if host != self.client_address[0]:
971 env['REMOTE_HOST'] = host
972 env['REMOTE_ADDR'] = self.client_address[0]
973
974 if self.headers.typeheader is None:
975 env['CONTENT_TYPE'] = self.headers.type
976 else:
977 env['CONTENT_TYPE'] = self.headers.typeheader
978 length = self.headers.getheader('content-length')
979 if length:
980 env['CONTENT_LENGTH'] = length
981 accept = []
982 for line in self.headers.getallmatchingheaders('accept'):
983 if line[:1] in "\t\n\r ":
984 accept.append(line.strip())
985 else:
986 accept = accept + line[7:].split(',')
987 env['HTTP_ACCEPT'] = ','.join(accept)
988
989 req = hgrequest(self.rfile, self.wfile, env)
990 self.send_response(200, "Script output follows")
991
992 if webdir_conf:
993 hgwebobj = hgwebdir(webdir_conf)
994 elif repo is not None:
995 hgwebobj = hgweb(repo.__class__(repo.ui, repo.origroot))
996 else:
997 raise hg.RepoError(_('no repo found'))
998 hgwebobj.run(req)
999
1000
1001 if use_ipv6:
1002 return IPv6HTTPServer((address, port), hgwebhandler)
1003 else:
1004 return MercurialHTTPServer((address, port), hgwebhandler)
1005
1006 # This is a stopgap
1007 class hgwebdir(object):
1008 def __init__(self, config):
1009 def cleannames(items):
1010 return [(name.strip(os.sep), path) for name, path in items]
1011
1012 self.motd = ""
1013 self.repos_sorted = ('name', False)
1014 if isinstance(config, (list, tuple)):
1015 self.repos = cleannames(config)
1016 self.repos_sorted = ('', False)
1017 elif isinstance(config, dict):
1018 self.repos = cleannames(config.items())
1019 self.repos.sort()
1020 else:
1021 cp = ConfigParser.SafeConfigParser()
1022 cp.read(config)
1023 self.repos = []
1024 if cp.has_section('web') and cp.has_option('web', 'motd'):
1025 self.motd = cp.get('web', 'motd')
1026 if cp.has_section('paths'):
1027 self.repos.extend(cleannames(cp.items('paths')))
1028 if cp.has_section('collections'):
1029 for prefix, root in cp.items('collections'):
1030 for path in util.walkrepos(root):
1031 repo = os.path.normpath(path)
1032 name = repo
1033 if name.startswith(prefix):
1034 name = name[len(prefix):]
1035 self.repos.append((name.lstrip(os.sep), repo))
1036 self.repos.sort()
1037
1038 def run(self, req=hgrequest()):
1039 def header(**map):
1040 yield tmpl("header", **map)
1041
1042 def footer(**map):
1043 yield tmpl("footer", motd=self.motd, **map)
1044
1045 m = os.path.join(templater.templatepath(), "map")
1046 tmpl = templater.templater(m, templater.common_filters,
1047 defaults={"header": header,
1048 "footer": footer})
1049
1050 def archivelist(ui, nodeid, url):
1051 for i in ['zip', 'gz', 'bz2']:
1052 if ui.configbool("web", "allow" + i, False):
1053 yield {"type" : i, "node": nodeid, "url": url}
1054
1055 def entries(sortcolumn="", descending=False, **map):
1056 rows = []
1057 parity = 0
1058 for name, path in self.repos:
1059 u = ui.ui()
1060 try:
1061 u.readconfig(os.path.join(path, '.hg', 'hgrc'))
1062 except IOError:
1063 pass
1064 get = u.config
1065
1066 url = ('/'.join([req.env["REQUEST_URI"].split('?')[0], name])
1067 .replace("//", "/"))
1068
1069 # update time with local timezone
1070 try:
1071 d = (get_mtime(path), util.makedate()[1])
1072 except OSError:
1073 continue
1074
1075 contact = (get("ui", "username") or # preferred
1076 get("web", "contact") or # deprecated
1077 get("web", "author", "")) # also
1078 description = get("web", "description", "")
1079 name = get("web", "name", name)
1080 row = dict(contact=contact or "unknown",
1081 contact_sort=contact.upper() or "unknown",
1082 name=name,
1083 name_sort=name,
1084 url=url,
1085 description=description or "unknown",
1086 description_sort=description.upper() or "unknown",
1087 lastchange=d,
1088 lastchange_sort=d[1]-d[0],
1089 archives=archivelist(u, "tip", url))
1090 if (not sortcolumn
1091 or (sortcolumn, descending) == self.repos_sorted):
1092 # fast path for unsorted output
1093 row['parity'] = parity
1094 parity = 1 - parity
1095 yield row
1096 else:
1097 rows.append((row["%s_sort" % sortcolumn], row))
1098 if rows:
1099 rows.sort()
1100 if descending:
1101 rows.reverse()
1102 for key, row in rows:
1103 row['parity'] = parity
1104 parity = 1 - parity
1105 yield row
1106
1107 virtual = req.env.get("PATH_INFO", "").strip('/')
1108 if virtual:
1109 real = dict(self.repos).get(virtual)
1110 if real:
1111 try:
1112 hgweb(real).run(req)
1113 except IOError, inst:
1114 req.write(tmpl("error", error=inst.strerror))
1115 except hg.RepoError, inst:
1116 req.write(tmpl("error", error=str(inst)))
1117 else:
1118 req.write(tmpl("notfound", repo=virtual))
1119 else:
1120 if req.form.has_key('static'):
1121 static = os.path.join(templater.templatepath(), "static")
1122 fname = req.form['static'][0]
1123 req.write(staticfile(static, fname)
1124 or tmpl("error", error="%r not found" % fname))
1125 else:
1126 sortable = ["name", "description", "contact", "lastchange"]
1127 sortcolumn, descending = self.repos_sorted
1128 if req.form.has_key('sort'):
1129 sortcolumn = req.form['sort'][0]
1130 descending = sortcolumn.startswith('-')
1131 if descending:
1132 sortcolumn = sortcolumn[1:]
1133 if sortcolumn not in sortable:
1134 sortcolumn = ""
1135
1136 sort = [("sort_%s" % column,
1137 "%s%s" % ((not descending and column == sortcolumn)
1138 and "-" or "", column))
1139 for column in sortable]
1140 req.write(tmpl("index", entries=entries,
1141 sortcolumn=sortcolumn, descending=descending,
1142 **dict(sort)))