import html
import urllib.parse

from translator import Translator


class Result:
  def __init__(self, code, headers, data):
    assert(type(code) is str)
    assert(type(headers) is list)
    self.code = code
    self.headers = headers
    self.data = data
    
  def complete(self, start_response):
    start_response(self.code, self.headers)
    return self.data


class Answer(Translator):
  def __init__(self, request):
    super().__init__()
    
    self.request = request
    self.urlprefix = self.request.server.config['urlprefix']
    self.urldataprefix = self.request.server.config['urldataprefix']
    
    self.status = "200 OK"
    self.headers = list()
    self.datalist = None

    self.data = None
    self.text = ''
    self.html = ''

    self.uipath = list()
    self.content = ''
    self.errors = list()
    self.fields = dict()
    self.template = None
    
    self.objects = dict()

  def htmlescape(self, text):
    return html.escape(str('' if text is None else text))
  def urlescape(self, text):
    return urllib.parse.quote(str('' if text is None else text))

  def e(self, text):
    return self.htmlescape(text)
  def ue(self, text):
    return self.urlescape(text)
  def te(self, text):
    return self.e( self.t(text) )
  
  def encodedfield(self, name):
    return self.e( self.fields.get(str(name), '') )
  
  def add_uipath_entry(self, title, path):
    assert(type(title) is str)
    assert(type(path) is list)
    self.uipath.append((title, path))
    
  def complete_redirect(self, path = None):
    self.status = '303 See Other'
    self.headers.append( ('Location', self.request.get_urlpath_escaped(path)) )
    return Result(self.status, self.headers, list())

  def complete_error(self, status = None):
    if not status is None:
      self.status = status
    return Result(self.status, list(), list())

  def complete_data(self, data = None):
    if not data is None:
      self.data = data
    if self.data is None:
      self.data = bytes()
    if not type(self.data) is bytes:
      self.data = bytes(str(self.data), "utf8")
    if not self.data and self.status == "200 OK":
      self.status = "204 No Content"
    
    size = 65536
    self.datalist = []
    for i in range(0, len(self.data), size):
      self.datalist.append(self.data[i:i + size])
    return Result(self.status, self.headers, self.datalist)

  def complete_text(self, text = None):
    if not text is None:
      self.text = text
    if self.text is None:
      self.text = ""
    return self.complete_data(text)

  def complete_html(self, html = None):
    if not html is None:
      self.html = html
    self.headers.append( ('Content-Type','text/html') );
    return self.complete_text( self.html )

  def complete_content(self, content = None, template = None):
    if not content is None:
      self.content = content
    if not template is None:
      self.template = template
    return self.complete_html( self.template.wrap(self) if self.template else self.content )