# hgweb/request.py - An http request from either CGI or the standalone server.

#

# 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.

from mercurial.demandload import demandload

demandload(globals(), "socket sys cgi os errno")

from mercurial.i18n import gettext as _

class hgrequest(object):

    def __init__(self, inp=None, out=None, env=None):

        self.inp = inp or sys.stdin

        self.out = out or sys.stdout

        self.env = env or os.environ

        self.form = cgi.parse(self.inp, self.env, keep_blank_values=1)

        self.will_close = True

    def write(self, *things):

        for thing in things:

            if hasattr(thing, "__iter__"):

                for part in thing:

                    self.write(part)

            else:

                try:

                    self.out.write(str(thing))

                except socket.error, inst:

                    if inst[0] != errno.ECONNRESET:

                        raise

    def done(self):

        if self.will_close:

            self.inp.close()

            self.out.close()

        else:

            self.out.flush()

    def header(self, headers=[('Content-type','text/html')]):

        for header in headers:

            self.out.write("%s: %s\r\n" % header)

        self.out.write("\r\n")

    def httphdr(self, type, filename=None, length=0):

        headers = [('Content-type', type)]

        if filename:

            headers.append(('Content-disposition', 'attachment; filename=%s' %

                            filename))

        # we do not yet support http 1.1 chunked transfer, so we have

        # to force connection to close if content-length not known

        if length:

            headers.append(('Content-length', str(length)))

            self.will_close = False

        else:

            headers.append(('Connection', 'close'))

            self.will_close = True

        self.header(headers)