--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mercurial/hgweb.py	Sun May 22 08:13:38 2005 -0800
@@ -0,0 +1,397 @@
+#!/usr/bin/env python
+#
+# hgweb.py - 0.2 - 21 May 2005 - (c) 2005 Jake Edge <jake@edge2.net>
+#    - web interface to a mercurial repository
+#
+# This software may be used and distributed according to the terms
+# of the GNU General Public License, incorporated herein by reference.
+
+# useful for debugging
+import cgitb
+cgitb.enable()
+
+import os, cgi, time, re, difflib, sys, zlib
+from mercurial import hg, mdiff
+
+def nl2br(text):
+    return re.sub('\n', '<br />', text)
+
+def obfuscate(text):
+    l = []
+    for c in text:
+        l.append('&#%d;' % ord(c))
+    return ''.join(l)
+
+def httphdr(type):
+    print 'Content-type: %s\n' % type
+
+def write(*things):
+    for thing in things:
+        if hasattr(thing, "__iter__"):
+            for part in thing:
+                write(part)
+        else:
+            sys.stdout.write(str(thing))
+
+class template:
+    def __init__(self, tmpl_dir):
+        self.tmpl_dir = tmpl_dir
+    def do_page(self, tmpl_fn, **map):
+        txt = file(os.path.join(self.tmpl_dir, tmpl_fn)).read()
+        while txt:
+            m = re.search(r"#([a-zA-Z0-9]+)#", txt)
+            if m:
+                yield txt[:m.start(0)]
+                v = map.get(m.group(1), "")
+                if callable(v):
+                   for y in v(**map): yield y
+                else:
+                   yield v
+                txt = txt[m.end(0):]
+            else:
+                yield txt
+                txt = ''
+
+class page:
+    def __init__(self, tmpl_dir = "", type="text/html", title="Mercurial Web", 
+            charset="ISO-8859-1"):
+        self.tmpl = template(tmpl_dir)
+
+        print 'Content-type: %s; charset=%s\n' % (type, charset)
+        write(self.tmpl.do_page('htmlstart.tmpl', title = title))
+
+    def endpage(self):
+        print '</BODY>'
+        print '</HTML>'
+
+    def show_diff(self, a, b, fn):
+        a = a.splitlines(1)
+        b = b.splitlines(1)
+        l = difflib.unified_diff(a, b, fn, fn)
+        print '<pre>'
+        for line in l:
+            line = cgi.escape(line[:-1])
+            if line.startswith('+'):
+                print '<span class="plusline">%s</span>' % (line, )
+            elif line.startswith('-'):
+                print '<span class="minusline">%s</span>' % (line, )
+            elif line.startswith('@'):
+                print '<span class="atline">%s</span>' % (line, )
+            else:
+                print line
+        print '</pre>'
+
+class errpage(page):
+    def __init__(self, tmpl_dir):
+        page.__init__(self, tmpl_dir, title="Mercurial Web Error Page")
+
+class change_list(page):
+    def __init__(self, repo, tmpl_dir, reponame, numchanges = 50):
+        page.__init__(self, tmpl_dir)
+        self.repo = repo
+        self.numchanges = numchanges
+        write(self.tmpl.do_page('changestitle.tmpl', reponame=reponame))
+
+    def content(self, hi=None):
+        cl = []
+        count = self.repo.changelog.count()
+        if not hi:
+            hi = count
+        elif hi < self.numchanges:
+            hi = self.numchanges
+
+        start = 0
+        if hi - self.numchanges >= 0:
+            start = hi - self.numchanges
+
+        nav = "Displaying Revisions: %d-%d" % (start, hi-1)
+        if start != 0:
+            nav = ('<a href="?cmd=changes;hi=%d">Previous %d</a>&nbsp;&nbsp;' \
+                    % (start, self.numchanges)) + nav
+        if hi != count:
+            if hi + self.numchanges <= count:
+                nav += '&nbsp;&nbsp;<a href="?cmd=changes;hi=%d">Next %d</a>' \
+                        % (hi + self.numchanges, self.numchanges)
+            else:
+                nav += '&nbsp;&nbsp;<a href="?cmd=changes">Next %d</a>' % \
+                        self.numchanges
+
+        print '<center>%s</center>' % nav
+
+        for i in xrange(start, hi):
+            n = self.repo.changelog.node(i)
+            cl.append((n, self.repo.changelog.read(n)))
+        cl.reverse()
+
+        print '<table summary="" width="100%" align="center">'
+        for n, ch in cl:
+            print '<tr><td>'
+            self.change_table(n, ch)
+            print '</td></tr>'
+        print '</table>'
+
+        print '<center>%s</center>' % nav
+
+    def change_table(self, nodeid, changes):
+        hn = hg.hex(nodeid)
+        i = self.repo.changelog.rev(nodeid)
+        (h1, h2) = [ hg.hex(x) for x in self.repo.changelog.parents(nodeid) ]
+        datestr = time.asctime(time.gmtime(float(changes[2].split(' ')[0])))
+        files = []
+        for f in changes[3]:
+            files.append('<a href="?cmd=file;cs=%s;fn=%s">%s</a>&nbsp;&nbsp;' \
+                % (hn, f, cgi.escape(f)))
+        write(self.tmpl.do_page('change_table.tmpl',
+                author=obfuscate(changes[1]),
+                desc=nl2br(cgi.escape(changes[4])), date=datestr, 
+                files=' '.join(files), revnum=i, revnode=hn))
+
+class checkin(page):
+    def __init__(self, repo, tmpl_dir, nodestr):
+        page.__init__(self, tmpl_dir)
+        self.repo = repo
+        self.node = hg.bin(nodestr)
+        self.nodestr = nodestr
+        print '<h3>Checkin: %s</h3>' % nodestr
+
+    def content(self):
+        changes = self.repo.changelog.read(self.node)
+        i = self.repo.changelog.rev(self.node)
+        parents = self.repo.changelog.parents(self.node)
+        (h1, h2) = [ hg.hex(x) for x in parents ]
+        (i1, i2) = [ self.repo.changelog.rev(x) for x in parents ]
+        datestr = time.asctime(time.gmtime(float(changes[2].split(' ')[0])))
+        mf = self.repo.manifest.read(changes[0])
+        files = []
+        for f in changes[3]:
+            files.append('<a href="?cmd=file;nd=%s;fn=%s">%s</a>&nbsp;&nbsp;' \
+                % (hg.hex(mf[f]), f, cgi.escape(f)))
+        p2link = h2
+        if i2 != -1:
+            p2link = '<a href="?cmd=chkin;nd=%s">%s</a>' % (h2, h2)
+
+        write(self.tmpl.do_page('checkin.tmpl', revnum=i, revnode=self.nodestr,
+                p1num=i1, p1node=h1, p2num=i2, p2node=h2, p2link=p2link,
+                mfnum=self.repo.manifest.rev(changes[0]), 
+                mfnode=hg.hex(changes[0]), author=obfuscate(changes[1]),
+                desc=nl2br(cgi.escape(changes[4])), date=datestr,
+                files=' '.join(files)))
+
+        (c, a, d) = self.repo.diffrevs(parents[0], self.node)
+        change = self.repo.changelog.read(parents[0])
+        mf2 = self.repo.manifest.read(change[0])
+        for f in c:
+            self.show_diff(self.repo.file(f).read(mf2[f]), \
+                    self.repo.file(f).read(mf[f]), f)
+        for f in a:
+            self.show_diff('', self.repo.file(f).read(mf[f]), f)
+        for f in d:
+            self.show_diff(self.repo.file(f).read(mf2[f]), '', f)
+
+class filepage(page):
+    def __init__(self, repo, tmpl_dir, fn, node=None, cs=None):
+        page.__init__(self, tmpl_dir)
+        self.repo = repo
+        self.fn = fn
+        if cs: 
+            chng = self.repo.changelog.read(hg.bin(cs))
+            mf = self.repo.manifest.read(chng[0])
+            self.node = mf[self.fn]
+            self.nodestr = hg.hex(self.node)
+        else:
+            self.nodestr = node
+            self.node = hg.bin(node)
+        print '<div class="filename">%s (%s)</div>' % \
+                (cgi.escape(self.fn), self.nodestr, )
+        print '<a href="?cmd=hist;fn=%s">history</a><br />' % self.fn
+        print '<a href="?cmd=ann;fn=%s;nd=%s">annotate</a><br />' % \
+                (self.fn, self.nodestr)
+
+    def content(self):
+        print '<pre>'
+        print cgi.escape(self.repo.file(self.fn).read(self.node))
+        print '</pre>'
+
+class annpage(page):
+    def __init__(self, repo, tmpl_dir, fn, node):
+        page.__init__(self, tmpl_dir)
+        self.repo = repo
+        self.fn = fn
+        self.nodestr = node
+        self.node = hg.bin(node)
+        print '<div class="annotation">Annotated: %s (%s)</div>' % \
+                (cgi.escape(self.fn), self.nodestr, )
+
+    def content(self):
+        print '<pre>'
+        for n, l in self.repo.file(self.fn).annotate(self.node):
+            cnode = self.repo.changelog.lookup(n)
+            write(self.tmpl.do_page('annline.tmpl', cnode=hg.hex(cnode),
+                    cnum='% 6s' % n, fn=self.fn, line=cgi.escape(l[:-1])))
+        print '</pre>'
+
+class mfpage(page):
+    def __init__(self, repo, tmpl_dir, node):
+        page.__init__(self, tmpl_dir)
+        self.repo = repo
+        self.nodestr = node
+        self.node = hg.bin(node)
+
+    def content(self):
+        mf = self.repo.manifest.read(self.node)
+        fns = mf.keys()
+        fns.sort()
+        write(self.tmpl.do_page('mftitle.tmpl', node = self.nodestr))
+        for f in fns:
+            write(self.tmpl.do_page('mfentry.tmpl', fn=f, node=hg.hex(mf[f])))
+
+class histpage(page):
+    def __init__(self, repo, tmpl_dir, fn):
+        page.__init__(self, tmpl_dir)
+        self.repo = repo
+        self.fn = fn
+
+    def content(self):
+        print '<div class="filehist">File History: %s</div>' % self.fn
+        r = self.repo.file(self.fn)
+        print '<br />'
+        print '<table summary="" width="100%" align="center">'
+        for i in xrange(r.count()-1, -1, -1):
+            print '<tr><td>'
+            self.hist_ent(i, r)
+            print '</tr></td>'
+        print '</table>'
+
+    def hist_ent(self, i, r):
+        n = r.node(i)
+        (p1, p2) = r.parents(n)
+        (h, h1, h2) = map(hg.hex, (n, p1, p2))
+        (i1, i2) = map(r.rev, (p1, p2))
+        ci = r.linkrev(n)
+        cn = self.repo.changelog.node(ci)
+        cs = hg.hex(cn)
+        changes = self.repo.changelog.read(cn)
+        datestr = time.asctime(time.gmtime(float(changes[2].split(' ')[0])))
+        p2entry = ''
+        if i2 != -1:
+            p2entry = '&nbsp;&nbsp;%d:<a href="?cmd=file;nd=%s;fn=%s">%s</a>' \
+                    % (i2, h2, self.fn, h2 ),
+        write(self.tmpl.do_page('hist_ent.tmpl', author=obfuscate(changes[1]),
+                csnode=cs, desc=nl2br(cgi.escape(changes[4])), 
+                date = datestr, fn=self.fn, revnode=h, p1num = i1,
+                p1node=h1, p2entry=p2entry))
+                
+class hgweb:
+    repo_path = "."
+    numchanges = 50
+    tmpl_dir = "templates"
+
+    def __init__(self):
+        pass
+
+    def run(self):
+
+        args = cgi.parse()
+
+        ui = hg.ui()
+        repo = hg.repository(ui, self.repo_path)
+
+        if not args.has_key('cmd') or args['cmd'][0] == 'changes':
+            page = change_list(repo, self.tmpl_dir, 'Mercurial', 
+                    self.numchanges)
+            hi = args.get('hi', ( repo.changelog.count(), ))
+            page.content(hi = int(hi[0]))
+            page.endpage()
+            
+        elif args['cmd'][0] == 'chkin':
+            if not args.has_key('nd'):
+                page = errpage(self.tmpl_dir)
+                print '<div class="errmsg">No Node!</div>'
+            else:
+                page = checkin(repo, self.tmpl_dir, args['nd'][0])
+                page.content()
+            page.endpage()
+
+        elif args['cmd'][0] == 'file':
+            if not (args.has_key('nd') and args.has_key('fn')) and \
+                    not (args.has_key('cs') and args.has_key('fn')):
+                page = errpage(self.tmpl_dir)
+                print '<div class="errmsg">Invalid Args!</div>'
+            else:
+                if args.has_key('nd'):
+                    page = filepage(repo, self.tmpl_dir, 
+                            args['fn'][0], node=args['nd'][0])
+                else:
+                    page = filepage(repo, self.tmpl_dir,
+                            args['fn'][0], cs=args['cs'][0])
+                page.content()
+            page.endpage()
+
+        elif args['cmd'][0] == 'mf':
+            if not args.has_key('nd'):
+                page = errpage(self.tmpl_dir)
+                print '<div class="errmsg">No Node!</div>'
+            else:
+                page = mfpage(repo, self.tmpl_dir, args['nd'][0])
+                page.content()
+            page.endpage()
+
+        elif args['cmd'][0] == 'hist':
+            if not args.has_key('fn'):
+                page = errpage(self.tmpl_dir)
+                print '<div class="errmsg">No Filename!</div>'
+            else:
+                page = histpage(repo, self.tmpl_dir, args['fn'][0])
+                page.content()
+            page.endpage()
+
+        elif args['cmd'][0] == 'ann':
+            if not args.has_key('fn'):
+                page = errpage(self.tmpl_dir)
+                print '<div class="errmsg">No Filename!</div>'
+            elif not args.has_key('nd'):
+                page = errpage(self.tmpl_dir)
+                print '<div class="errmsg">No Node!</div>'
+            else:
+                page = annpage(repo, self.tmpl_dir, args['fn'][0], 
+                        args['nd'][0])
+                page.content()
+            page.endpage()
+
+        elif args['cmd'][0] == 'branches':
+            httphdr("text/plain")
+            nodes = []
+            if args.has_key('nodes'):
+                nodes = map(hg.bin, args['nodes'][0].split(" "))
+            for b in repo.branches(nodes):
+                print " ".join(map(hg.hex, b))
+
+        elif args['cmd'][0] == 'between':
+            httphdr("text/plain")
+            nodes = []
+            if args.has_key('pairs'):
+                pairs = [ map(hg.bin, p.split("-"))
+                          for p in args['pairs'][0].split(" ") ]
+            for b in repo.between(pairs):
+                print " ".join(map(hg.hex, b))
+
+        elif args['cmd'][0] == 'changegroup':
+            httphdr("application/hg-changegroup")
+            nodes = []
+            if args.has_key('roots'):
+                nodes = map(hg.bin, args['roots'][0].split(" "))
+
+            z = zlib.compressobj()
+            for chunk in repo.changegroup(nodes):
+                sys.stdout.write(z.compress(chunk))
+
+            sys.stdout.write(z.flush())
+
+        else:
+            page = errpage(self.tmpl_dir)
+            print '<div class="errmsg">unknown command: %s</div>' % \
+                    cgi.escape(args['cmd'][0])
+            page.endpage()
+
+if __name__ == "__main__":
+    hgweb().run()