# hgweb.py - web interface to a mercurial repository

#

# Copyright 21 May 2005 - (c) 2005 Jake Edge <jake@edge2.net>

# Copyright 2005 Matt Mackall <mpm@selenic.com>

#

# This software may be used and distributed according to the terms

# of the GNU General Public License, incorporated herein by reference.

import os, cgi, time, re, difflib, socket, sys, zlib

from mercurial.hg import *

from mercurial.ui import *

def templatepath():

    for f in "templates", "../templates":

        p = os.path.join(os.path.dirname(__file__), f)

        if os.path.isdir(p):

            return p

def age(t):

    def plural(t, c):

        if c == 1:

            return t

        return t + "s"

    def fmt(t, c):

        return "%d %s" % (c, plural(t, c))

    now = time.time()

    delta = max(1, int(now - t))

    scales = [["second", 1],

              ["minute", 60],

              ["hour", 3600],

              ["day", 3600 * 24],

              ["week", 3600 * 24 * 7],

              ["month", 3600 * 24 * 30],

              ["year", 3600 * 24 * 365]]

    scales.reverse()

    for t, s in scales:

        n = delta / s

        if n >= 2 or s == 1:

            return fmt(t, n)

def nl2br(text):

    return text.replace('\n', '<br/>\n')

def obfuscate(text):

    return ''.join(['&#%d;' % ord(c) for c in text])

def up(p):

    if p[0] != "/":

        p = "/" + p

    if p[-1] == "/":

        p = p[:-1]

    up = os.path.dirname(p)

    if up == "/":

        return "/"

    return up + "/"

def httphdr(type):

    sys.stdout.write('Content-type: %s\n\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 templater:

    def __init__(self, mapfile, filters={}, defaults={}):

        self.cache = {}

        self.map = {}

        self.base = os.path.dirname(mapfile)

        self.filters = filters

        self.defaults = defaults

        for l in file(mapfile):

            m = re.match(r'(\S+)\s*=\s*"(.*)"$', l)

            if m:

                self.cache[m.group(1)] = m.group(2)

            else:

                m = re.match(r'(\S+)\s*=\s*(\S+)', l)

                if m:

                    self.map[m.group(1)] = os.path.join(self.base, m.group(2))

                else:

                    raise LookupError("unknown map entry '%s'" % l)

    def __call__(self, t, **map):

        m = self.defaults.copy()

        m.update(map)

        try:

            tmpl = self.cache[t]

        except KeyError:

            tmpl = self.cache[t] = file(self.map[t]).read()

        return self.template(tmpl, self.filters, **m)

    def template(self, tmpl, filters={}, **map):

        while tmpl:

            m = re.search(r"#([a-zA-Z0-9]+)((%[a-zA-Z0-9]+)*)((\|[a-zA-Z0-9]+)*)#", tmpl)

            if m:

                yield tmpl[:m.start(0)]

                v = map.get(m.group(1), "")

                v = callable(v) and v(**map) or v

                format = m.group(2)

                fl = m.group(4)

                if format:

                    q = v.__iter__

                    for i in q():

                        lm = map.copy()

                        lm.update(i)

                        yield self(format[1:], **lm)

                    v = ""

                elif fl:

                    for f in fl.split("|")[1:]:

                        v = filters[f](v)

                yield v

                tmpl = tmpl[m.end(0):]

            else:

                yield tmpl

                return

def rfc822date(x):

    return time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.gmtime(x))

common_filters = {

    "escape": cgi.escape,

    "age": age,

    "date": (lambda x: time.asctime(time.gmtime(x))),

    "addbreaks": nl2br,

    "obfuscate": obfuscate,

    "short": (lambda x: x[:12]),

    "firstline": (lambda x: x.splitlines(1)[0]),

    "permissions": (lambda x: x and "-rwxr-xr-x" or "-rw-r--r--"),

    "rfc822date": rfc822date,

    }

class hgweb:

    def __init__(self, repo, name=None):

        if type(repo) == type(""):

            self.repo = repository(ui(), repo)

        else:

            self.repo = repo

        self.mtime = -1

        self.reponame = name or self.repo.ui.config("web", "name",

                                                    self.repo.root)

    def refresh(self):

        s = os.stat(os.path.join(self.repo.root, ".hg", "00changelog.i"))

        if s.st_mtime != self.mtime:

            self.mtime = s.st_mtime

            self.repo = repository(self.repo.ui, self.repo.root)

            self.maxchanges = self.repo.ui.config("web", "maxchanges", 10)

            self.maxfiles = self.repo.ui.config("web", "maxchanges", 10)

            self.allowpull = self.repo.ui.configbool("web", "allowpull", True)

    def date(self, cs):

        return time.asctime(time.gmtime(float(cs[2].split(' ')[0])))

    def listfiles(self, files, mf):

        for f in files[:self.maxfiles]:

            yield self.t("filenodelink", node=hex(mf[f]), file=f)

        if len(files) > self.maxfiles:

            yield self.t("fileellipses")

    def listfilediffs(self, files, changeset):

        for f in files[:self.maxfiles]:

            yield self.t("filedifflink", node=hex(changeset), file=f)

        if len(files) > self.maxfiles:

            yield self.t("fileellipses")

    def parents(self, t1, nodes=[], rev=None,**args):

        if not rev:

            rev = lambda x: ""

        for node in nodes:

            if node != nullid:

                yield self.t(t1, node=hex(node), rev=rev(node), **args)

    def showtag(self, t1, node=nullid, **args):

        for t in self.repo.nodetags(node):

             yield self.t(t1, tag=t, **args)

    def diff(self, node1, node2, files):

        def filterfiles(list, files):

            l = [x for x in list if x in files]

            for f in files:

                if f[-1] != os.sep:

                    f += os.sep

                l += [x for x in list if x.startswith(f)]

            return l

        parity = [0]

        def diffblock(diff, f, fn):

            yield self.t("diffblock",

                         lines=prettyprintlines(diff),

                         parity=parity[0],

                         file=f,

                         filenode=hex(fn or nullid))

            parity[0] = 1 - parity[0]

        def prettyprintlines(diff):

            for l in diff.splitlines(1):

                if l.startswith('+'):

                    yield self.t("difflineplus", line=l)

                elif l.startswith('-'):

                    yield self.t("difflineminus", line=l)

                elif l.startswith('@'):

                    yield self.t("difflineat", line=l)

                else:

                    yield self.t("diffline", line=l)

        r = self.repo

        cl = r.changelog

        mf = r.manifest

        change1 = cl.read(node1)

        change2 = cl.read(node2)

        mmap1 = mf.read(change1[0])

        mmap2 = mf.read(change2[0])

        date1 = self.date(change1)

        date2 = self.date(change2)

        c, a, d, u = r.changes(node1, node2)

        if files:

            c, a, d = map(lambda x: filterfiles(x, files), (c, a, d))

        for f in c:

            to = r.file(f).read(mmap1[f])

            tn = r.file(f).read(mmap2[f])

            yield diffblock(mdiff.unidiff(to, date1, tn, date2, f), f, tn)

        for f in a:

            to = None

            tn = r.file(f).read(mmap2[f])

            yield diffblock(mdiff.unidiff(to, date1, tn, date2, f), f, tn)

        for f in d:

            to = r.file(f).read(mmap1[f])

            tn = None

            yield diffblock(mdiff.unidiff(to, date1, tn, date2, f), f, tn)

    def changelog(self, pos):

        def changenav(**map):

            def seq(factor=1):

                yield 1 * factor

                yield 3 * factor

                #yield 5 * factor

                for f in seq(factor * 10):

                    yield f

            l = []

            for f in seq():

                if f < self.maxchanges / 2:

                    continue

                if f > count:

                    break

                r = "%d" % f

                if pos + f < count:

                    l.append(("+" + r, pos + f))

                if pos - f >= 0:

                    l.insert(0, ("-" + r, pos - f))

            yield {"rev": 0, "label": "(0)"}

            for label, rev in l:

                yield {"label": label, "rev": rev}

            yield {"label": "tip", "rev": ""}

        def changelist(**map):

            parity = (start - end) & 1

            cl = self.repo.changelog

            l = [] # build a list in forward order for efficiency

            for i in range(start, end):

                n = cl.node(i)

                changes = cl.read(n)

                hn = hex(n)

                t = float(changes[2].split(' ')[0])

                l.insert(0, {"parity": parity,

                             "author": changes[1],

                             "parent": self.parents("changelogparent",

                                                    cl.parents(n), cl.rev),

                             "changelogtag": self.showtag("changelogtag",n),

                             "manifest": hex(changes[0]),

                             "desc": changes[4],

                             "date": t,

                             "files": self.listfilediffs(changes[3], n),

                             "rev": i,

                             "node": hn})

                parity = 1 - parity

            for e in l:

                yield e

        cl = self.repo.changelog

        mf = cl.read(cl.tip())[0]

        count = cl.count()

        start = max(0, pos - self.maxchanges + 1)

        end = min(count, start + self.maxchanges)

        pos = end - 1

        yield self.t('changelog',

                     changenav=changenav,

                     manifest=hex(mf),

                     rev=pos, changesets=count, entries=changelist)

    def search(self, query):

        def changelist(**map):

            cl = self.repo.changelog

            count = 0

            qw = query.lower().split()

            def revgen():

                for i in range(cl.count() - 1, 0, -100):

                    l = []

                    for j in range(max(0, i - 100), i):

                        n = cl.node(j)

                        changes = cl.read(n)

                        l.append((n, j, changes))

                    l.reverse()

                    for e in l:

                        yield e

            for n, i, changes in revgen():

                miss = 0

                for q in qw:

                    if not (q in changes[1].lower() or

                            q in changes[4].lower() or

                            q in " ".join(changes[3][:20]).lower()):

                        miss = 1

                        break

                if miss:

                    continue

                count += 1

                hn = hex(n)

                t = float(changes[2].split(' ')[0])

                yield self.t('searchentry',

                             parity=count & 1,

                             author=changes[1],

                             parent=self.parents("changelogparent",

                                                 cl.parents(n), cl.rev),

                             changelogtag=self.showtag("changelogtag",n),

                             manifest=hex(changes[0]),

                             desc=changes[4],

                             date=t,

                             files=self.listfilediffs(changes[3], n),

                             rev=i,

                             node=hn)

                if count >= self.maxchanges:

                    break

        cl = self.repo.changelog

        mf = cl.read(cl.tip())[0]

        yield self.t('search',

                     query=query,

                     manifest=hex(mf),

                     entries=changelist)

    def changeset(self, nodeid):

        n = bin(nodeid)

        cl = self.repo.changelog

        changes = cl.read(n)

        p1 = cl.parents(n)[0]

        t = float(changes[2].split(' ')[0])

        files = []

        mf = self.repo.manifest.read(changes[0])

        for f in changes[3]:

            files.append(self.t("filenodelink",

                                filenode=hex(mf.get(f, nullid)), file=f))

        def diff(**map):

            yield self.diff(p1, n, None)

        yield self.t('changeset',

                     diff=diff,

                     rev=cl.rev(n),

                     node=nodeid,

                     parent=self.parents("changesetparent",

                                         cl.parents(n), cl.rev),

                     changesettag=self.showtag("changesettag",n),

                     manifest=hex(changes[0]),

                     author=changes[1],

                     desc=changes[4],

                     date=t,

                     files=files)

    def filelog(self, f, filenode):

        cl = self.repo.changelog

        fl = self.repo.file(f)

        count = fl.count()

        def entries(**map):

            l = []

            parity = (count - 1) & 1

            for i in range(count):

                n = fl.node(i)

                lr = fl.linkrev(n)

                cn = cl.node(lr)

                cs = cl.read(cl.node(lr))

                t = float(cs[2].split(' ')[0])

                l.insert(0, {"parity": parity,

                             "filenode": hex(n),

                             "filerev": i,

                             "file": f,

                             "node": hex(cn),

                             "author": cs[1],

                             "date": t,

                             "parent": self.parents("filelogparent",

                                                    fl.parents(n),

                                                    fl.rev, file=f),

                             "desc": cs[4]})

                parity = 1 - parity

            for e in l:

                yield e

        yield self.t("filelog", file=f, filenode=filenode, entries=entries)

    def filerevision(self, f, node):

        fl = self.repo.file(f)

        n = bin(node)

        text = fl.read(n)

        changerev = fl.linkrev(n)

        cl = self.repo.changelog

        cn = cl.node(changerev)

        cs = cl.read(cn)

        t = float(cs[2].split(' ')[0])

        mfn = cs[0]

        def lines():

            for l, t in enumerate(text.splitlines(1)):

                yield {"line": t,

                       "linenumber": "% 6d" % (l + 1),

                       "parity": l & 1}

        yield self.t("filerevision",

                     file=f,

                     filenode=node,

                     path=up(f),

                     text=lines(),

                     rev=changerev,

                     node=hex(cn),

                     manifest=hex(mfn),

                     author=cs[1],

                     date=t,

                     parent=self.parents("filerevparent",

                                         fl.parents(n), fl.rev, file=f),

                     permissions=self.repo.manifest.readflags(mfn)[f])

    def fileannotate(self, f, node):

        bcache = {}

        ncache = {}

        fl = self.repo.file(f)

        n = bin(node)

        changerev = fl.linkrev(n)

        cl = self.repo.changelog

        cn = cl.node(changerev)

        cs = cl.read(cn)

        t = float(cs[2].split(' ')[0])

        mfn = cs[0]

        def annotate(**map):

            parity = 1

            last = None

            for r, l in fl.annotate(n):

                try:

                    cnode = ncache[r]

                except KeyError: