# hgweb/hgweb_mod.py - Web interface for a repository.

#

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

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

#

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

# GNU General Public License version 2 or any later version.

import os

from mercurial import ui, hg, hook, error, encoding, templater, util, repoview

from mercurial.templatefilters import websub

from mercurial.i18n import _

from common import get_stat, ErrorResponse, permhooks, caching

from common import HTTP_OK, HTTP_NOT_MODIFIED, HTTP_BAD_REQUEST

from common import HTTP_NOT_FOUND, HTTP_SERVER_ERROR

from request import wsgirequest

import webcommands, protocol, webutil, re

perms = {

    'changegroup': 'pull',

    'changegroupsubset': 'pull',

    'getbundle': 'pull',

    'stream_out': 'pull',

    'listkeys': 'pull',

    'unbundle': 'push',

    'pushkey': 'push',

}

def makebreadcrumb(url, prefix=''):

    '''Return a 'URL breadcrumb' list

    A 'URL breadcrumb' is a list of URL-name pairs,

    corresponding to each of the path items on a URL.

    This can be used to create path navigation entries.

    '''

    if url.endswith('/'):

        url = url[:-1]

    if prefix:

        url = '/' + prefix + url

    relpath = url

    if relpath.startswith('/'):

        relpath = relpath[1:]

    breadcrumb = []

    urlel = url

    pathitems = [''] + relpath.split('/')

    for pathel in reversed(pathitems):

        if not pathel or not urlel:

            break

        breadcrumb.append({'url': urlel, 'name': pathel})

        urlel = os.path.dirname(urlel)

    return reversed(breadcrumb)

class hgweb(object):

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

        if isinstance(repo, str):

            if baseui:

                u = baseui.copy()

            else:

                u = ui.ui()

            self.repo = hg.repository(u, repo)

        else:

            self.repo = repo

        self.repo = self._getview(self.repo)

        self.repo.ui.setconfig('ui', 'report_untrusted', 'off')

        self.repo.baseui.setconfig('ui', 'report_untrusted', 'off')

        self.repo.ui.setconfig('ui', 'nontty', 'true')

        self.repo.baseui.setconfig('ui', 'nontty', 'true')

        hook.redirect(True)

        self.mtime = -1

        self.size = -1

        self.reponame = name

        self.archives = 'zip', 'gz', 'bz2'

        self.stripecount = 1

        # a repo owner may set web.templates in .hg/hgrc to get any file

        # readable by the user running the CGI script

        self.templatepath = self.config('web', 'templates')

        self.websubtable = self.loadwebsub()

    # The CGI scripts are often run by a user different from the repo owner.

    # Trust the settings from the .hg/hgrc files by default.

    def config(self, section, name, default=None, untrusted=True):

        return self.repo.ui.config(section, name, default,

                                   untrusted=untrusted)

    def configbool(self, section, name, default=False, untrusted=True):

        return self.repo.ui.configbool(section, name, default,

                                       untrusted=untrusted)

    def configlist(self, section, name, default=None, untrusted=True):

        return self.repo.ui.configlist(section, name, default,

                                       untrusted=untrusted)

    def _getview(self, repo):

        viewconfig = self.config('web', 'view', 'served')

        if viewconfig == 'all':

            return repo.unfiltered()

        elif viewconfig in repoview.filtertable:

            return repo.filtered(viewconfig)

        else:

            return repo.filtered('served')

    def refresh(self, request=None):

        st = get_stat(self.repo.spath)

        # compare changelog size in addition to mtime to catch

        # rollbacks made less than a second ago

        if st.st_mtime != self.mtime or st.st_size != self.size:

            self.mtime = st.st_mtime

            self.size = st.st_size

            r = hg.repository(self.repo.baseui, self.repo.root)

            self.repo = self._getview(r)

            self.maxchanges = int(self.config("web", "maxchanges", 10))

            self.stripecount = int(self.config("web", "stripes", 1))

            self.maxshortchanges = int(self.config("web", "maxshortchanges",

                                                   60))

            self.maxfiles = int(self.config("web", "maxfiles", 10))

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

            encoding.encoding = self.config("web", "encoding",

                                            encoding.encoding)

        if request:

            self.repo.ui.environ = request.env

    def run(self):

        if not os.environ.get('GATEWAY_INTERFACE', '').startswith("CGI/1."):

            raise RuntimeError("This function is only intended to be "

                               "called while running as a CGI script.")

        import mercurial.hgweb.wsgicgi as wsgicgi

        wsgicgi.launch(self)

    def __call__(self, env, respond):

        req = wsgirequest(env, respond)

        return self.run_wsgi(req)

    def run_wsgi(self, req):

        self.refresh(req)

        # work with CGI variables to create coherent structure

        # use SCRIPT_NAME, PATH_INFO and QUERY_STRING as well as our REPO_NAME

        req.url = req.env['SCRIPT_NAME']

        if not req.url.endswith('/'):

            req.url += '/'

        if 'REPO_NAME' in req.env:

            req.url += req.env['REPO_NAME'] + '/'

        if 'PATH_INFO' in req.env:

            parts = req.env['PATH_INFO'].strip('/').split('/')

            repo_parts = req.env.get('REPO_NAME', '').split('/')

            if parts[:len(repo_parts)] == repo_parts:

                parts = parts[len(repo_parts):]

            query = '/'.join(parts)

        else:

            query = req.env['QUERY_STRING'].split('&', 1)[0]

            query = query.split(';', 1)[0]

        # process this if it's a protocol request

        # protocol bits don't need to create any URLs

        # and the clients always use the old URL structure

        cmd = req.form.get('cmd', [''])[0]

        if protocol.iscmd(cmd):

            try:

                if query:

                    raise ErrorResponse(HTTP_NOT_FOUND)

                if cmd in perms:

                    self.check_perm(req, perms[cmd])

                return protocol.call(self.repo, req, cmd)

            except ErrorResponse, inst:

                # A client that sends unbundle without 100-continue will

                # break if we respond early.

                if (cmd == 'unbundle' and

                    (req.env.get('HTTP_EXPECT',

                                 '').lower() != '100-continue') or

                    req.env.get('X-HgHttp2', '')):

                    req.drain()

                req.respond(inst, protocol.HGTYPE,

                            body='0\n%s\n' % inst.message)

                return ''

        # translate user-visible url structure to internal structure

        args = query.split('/', 2)

        if 'cmd' not in req.form and args and args[0]:

            cmd = args.pop(0)

            style = cmd.rfind('-')

            if style != -1:

                req.form['style'] = [cmd[:style]]

                cmd = cmd[style + 1:]

            # avoid accepting e.g. style parameter as command

            if util.safehasattr(webcommands, cmd):

                req.form['cmd'] = [cmd]

            else:

                cmd = ''

            if cmd == 'static':

                req.form['file'] = ['/'.join(args)]

            else:

                if args and args[0]:

                    node = args.pop(0)

                    req.form['node'] = [node]

                if args:

                    req.form['file'] = args

            ua = req.env.get('HTTP_USER_AGENT', '')

            if cmd == 'rev' and 'mercurial' in ua:

                req.form['style'] = ['raw']

            if cmd == 'archive':

                fn = req.form['node'][0]

                for type_, spec in self.archive_specs.iteritems():

                    ext = spec[2]

                    if fn.endswith(ext):

                        req.form['node'] = [fn[:-len(ext)]]

                        req.form['type'] = [type_]

        # process the web interface request

        try:

            tmpl = self.templater(req)

            ctype = tmpl('mimetype', encoding=encoding.encoding)

            ctype = templater.stringify(ctype)

            # check read permissions non-static content

            if cmd != 'static':

                self.check_perm(req, None)

            if cmd == '':

                req.form['cmd'] = [tmpl.cache['default']]

                cmd = req.form['cmd'][0]

            if self.configbool('web', 'cache', True):

                caching(self, req) # sets ETag header or raises NOT_MODIFIED

            if cmd not in webcommands.__all__:

                msg = 'no such method: %s' % cmd

                raise ErrorResponse(HTTP_BAD_REQUEST, msg)

            elif cmd == 'file' and 'raw' in req.form.get('style', []):

                self.ctype = ctype

                content = webcommands.rawfile(self, req, tmpl)

            else:

                content = getattr(webcommands, cmd)(self, req, tmpl)

                req.respond(HTTP_OK, ctype)

            return content

        except (error.LookupError, error.RepoLookupError), err:

            req.respond(HTTP_NOT_FOUND, ctype)

            msg = str(err)

            if (util.safehasattr(err, 'name') and

                not isinstance(err,  error.ManifestLookupError)):

                msg = 'revision not found: %s' % err.name

            return tmpl('error', error=msg)

        except (error.RepoError, error.RevlogError), inst:

            req.respond(HTTP_SERVER_ERROR, ctype)

            return tmpl('error', error=str(inst))

        except ErrorResponse, inst:

            req.respond(inst, ctype)

            if inst.code == HTTP_NOT_MODIFIED:

                # Not allowed to return a body on a 304

                return ['']

            return tmpl('error', error=inst.message)

    def loadwebsub(self):

        websubtable = []

        websubdefs = self.repo.ui.configitems('websub')

        # we must maintain interhg backwards compatibility

        websubdefs += self.repo.ui.configitems('interhg')

        for key, pattern in websubdefs:

            # grab the delimiter from the character after the "s"

            unesc = pattern[1]

            delim = re.escape(unesc)

            # identify portions of the pattern, taking care to avoid escaped

            # delimiters. the replace format and flags are optional, but

            # delimiters are required.

            match = re.match(

                r'^s%s(.+)(?:(?<=\\\\)|(?<!\\))%s(.*)%s([ilmsux])*$'

                % (delim, delim, delim), pattern)

            if not match:

                self.repo.ui.warn(_("websub: invalid pattern for %s: %s\n")

                                  % (key, pattern))

                continue

            # we need to unescape the delimiter for regexp and format

            delim_re = re.compile(r'(?<!\\)\\%s' % delim)

            regexp = delim_re.sub(unesc, match.group(1))

            format = delim_re.sub(unesc, match.group(2))

            # the pattern allows for 6 regexp flags, so set them if necessary

            flagin = match.group(3)

            flags = 0

            if flagin:

                for flag in flagin.upper():

                    flags |= re.__dict__[flag]

            try:

                regexp = re.compile(regexp, flags)

                websubtable.append((regexp, format))

            except re.error:

                self.repo.ui.warn(_("websub: invalid regexp for %s: %s\n")

                                  % (key, regexp))

        return websubtable

    def templater(self, req):

        # determine scheme, port and server name

        # this is needed to create absolute urls

        proto = req.env.get('wsgi.url_scheme')

        if proto == 'https':

            proto = 'https'

            default_port = "443"

        else:

            proto = 'http'

            default_port = "80"

        port = req.env["SERVER_PORT"]

        port = port != default_port and (":" + port) or ""

        urlbase = '%s://%s%s' % (proto, req.env['SERVER_NAME'], port)

        logourl = self.config("web", "logourl", "http://mercurial.selenic.com/")

        logoimg = self.config("web", "logoimg", "hglogo.png")

        staticurl = self.config("web", "staticurl") or req.url + 'static/'

        if not staticurl.endswith('/'):

            staticurl += '/'

        # some functions for the templater

        def header(**map):

            yield tmpl('header', encoding=encoding.encoding, **map)

        def footer(**map):

            yield tmpl("footer", **map)

        def motd(**map):

            yield self.config("web", "motd", "")

        # figure out which style to use

        vars = {}

        styles = (

            req.form.get('style', [None])[0],

            self.config('web', 'style'),

            'paper',

        )

        style, mapfile = templater.stylemap(styles, self.templatepath)

        if style == styles[0]: