import cgi

import urllib.parse

import answer

from model.model import Model

class Request:

  def __init__(self, server, env):

    self.server = server

    self.env = env

    self.method = str(self.env["REQUEST_METHOD"])

    self.protocol = self.server.config['protocol']

    self.domain = self.server.config['domain']

    self.prefix = self.server.config['urlprefix']

    self.urlvars = dict()

    self.postvars = dict()

    self.action = ''

    self.path = None

    self.cookie = ''

    self.connection = None

    self.model = None

    self.user = None

    self.answer = None

    self.session = None

    # read url vars

    if 'QUERY_STRING' in self.env:

      urlvars = urllib.parse.parse_qs(self.env["QUERY_STRING"])

      for k, v in urlvars.items():

        assert k not in self.urlvars

        self.urlvars[k] = v

    # read cookies

    self.cookie = self.env.get('HTTP_COOKIE', '')

    assert(type(self.cookie) is str)

    # read path

    path = self.env["REQUEST_URI"].split('?')[0]

    if path.startswith(self.prefix):

      path = path[len(self.prefix):]

      path_list = [x for x in path.split('/') if x]

      path_list_filtered = list()

      for i in path_list:

        if i == '..':

          if path_list_filtered:

            path_list_filtered.pop();

        elif i != '.':

          path_list_filtered.append(i)

      self.path = path_list_filtered

    self.answer = answer.Answer(self)

    self.server.sessions.attach_session(self)

  def read_postvars(self):

    if self.method == 'POST':

      postvars = cgi.FieldStorage(

        fp = self.env["wsgi.input"],

        environ = self.env )

      duplicates = set()

      for k in postvars:

        kk = str(k)

        if kk in self.postvars:

          duplicates.add(kk)

          del self.postvars[kk]

        elif not kk in duplicates:

          self.postvars[kk] = postvars[kk].value

      # get action

      if 'action' in self.postvars:

        action = self.postvars['action']

        del self.postvars['action']

        if not action is None:

          self.action = str(action)

  def create_model(self, connection):

    self.connection = connection

    self.model = Model(self.connection, self.answer, self.session.user_id if self.session else 0)

    if self.session:

      self.user = self.model.users.get_by_id(self.session.user_id)

      if not self.user:

        self.server.sessions.close_session(self)

        self.model = Model(self.connection, self.answer, 0)

  def get_urlpath(self, path = None, domain = False):

    if path is None:

      path = self.path

    assert(type(path) is list)

    urlpath = self.prefix + '/' + '/'.join(path)

    if domain:

      urlpath = 'https://' + self.domain + urlpath

    return urlpath

  def get_urlpath_escaped(self, path = None, domain = False):

    return self.answer.e( self.get_urlpath(path, domain) )

  def translate(self, text):

    return self.answer.translate(text)

  def t(self, text):

    return self.translate(text)