from flask import request
from flask_restful import Resource, abort, reqparse
from Gulag import GulagException

class GulagResource(Resource):
  gulag = None
  def __init__(self,gulag_object):
    self.gulag = gulag_object
#XXX    self.check_trusted_proxy()
#XXX    self.check_auth()

  def check_trusted_proxy(self):
    remote_ip = request.remote_addr
    if 'trusted_proxies' not in self.gulag.config:
      # Trusted-proxies not configured
      return True
    for proxy in self.gulag.config['trusted_proxies']:
      for trusted_proxy_ip in self.gulag.config['trusted_proxies'][proxy]:
        if(remote_ip == trusted_proxy_ip):
          return True
    abort(403, message="Untrusted client IP-address!")

  def check_auth(self):
    if not 'API-KEY' in request.headers:
      abort(400, message="API-KEY header missing!")
    api_key = request.headers['API-KEY']
    if api_key not in self.gulag.config['api_keys']:
      abort(401, message="NOT AUTHORIZED!")

class ResRoot(GulagResource):
  def get(self):
    return {"resource": "Root :)"}

class ResMailboxes(GulagResource):
  def get(self):
    try:
      return self.gulag.get_mailboxes()
    except GulagException as e:
      abort(500, message=e.message)

class ResMailbox(GulagResource):
  def get(self,id):
    return {"resource": "Mailbox by ID"}

class ResQuarMails(GulagResource):
  def get(self):
    try:
      return self.gulag.get_quarmails(request.args.to_dict())
    except GulagException as e:
      abort(400, message=e.message)

class ResQuarMail(GulagResource):
  def get(self,quarmail_id):
    args = {"quarmail_id": quarmail_id}
    try:
      if(request.args.get('rfc822_message')):
        args['rfc822_message'] = True
      return self.gulag.get_quarmail(args)
    except GulagException as e:
      abort(400, message=e.message)

class ResQuarMailAttachments(GulagResource):
  def get(self,quarmail_id):
    args = {"quarmail_id": quarmail_id}
    if(request.args.get('data')):
      args['data'] = True
    try:
      return self.gulag.get_quarmail_attachments(args)
    except GulagException as e:
      abort(400, message=e.message)

class ResQuarMailAttachment(GulagResource):
  def get(self,quarmail_id,attachment_id):
    args = {
      "quarmail_id": quarmail_id,
      "attachment_id": attachment_id
    }
    if(request.args.get('data')):
      args['data'] = True
    try:
      return self.gulag.get_quarmail_attachment(args)
    except GulagException as e:
      abort(400, message=e.message)

class ResQuarMailURIs(GulagResource):
  def get(self,quarmail_id): 
    args = {
      "quarmail_id": quarmail_id
    }
    if(request.args.get('from_rfc822_message')):
        args['from_rfc822_message'] = True
    try:
      return self.gulag.get_quarmail_uris(args)
    except GulagException as e:
      abort(400, message=e.message)

class ResAttachments(GulagResource):
  def get(self):
    return {"resource": "Attachments"}

class ResAttachment(GulagResource):
  def get(self,attachment_id):
    args = {"id": attachment_id}
    try:
      return self.gulag.get_attachment(args)
    except GulagException as e:
      abort(400, message=e.message)

class ResRSPAMDImporter(GulagResource):
  def post(self,mailbox_id):
    try:
      self.gulag.rspamd_http2imap({
        "mailbox_id": mailbox_id,
        "req_headers": request.headers,
        "rfc822_message": request.get_data(as_text=True)
      })
      # TODO: Response mit Location-Header?
      # https://stackoverflow.com/a/22707491
      return {"resource: ": "HTTP2IMAP for RSPAMD"}
    except GulagException as e:
      abort(400, message=e.message)