#!/usr/bin/env python3

import argparse,sys
from flask import Flask
from flask_restful import Api
from Gulag import Gulag,GulagException
from Resources import ResRoot,ResMailboxes,ResQuarMails,ResAttachments,ResRSPAMDImporter

parser = argparse.ArgumentParser()
parser.add_argument('--config', required=True, help="Path to config file")
args = parser.parse_args()

try:
  try:
    gulag = Gulag(args.config)
  except GulagException as e:
    raise Exception(e.message) from e
  app = Flask(__name__)
  api = Api(app, catch_all_404s=True)
  api.add_resource(ResRoot,
    '/api/v1/',
    resource_class_kwargs={'gulag_object': gulag}
  )
  api.add_resource(ResMailboxes,
    '/api/v1/mailboxes/',
    resource_class_kwargs={'gulag_object': gulag}
  )
  api.add_resource(ResQuarMails,
    '/api/v1/quarmails/',
    resource_class_kwargs={'gulag_object': gulag}
  )
  api.add_resource(ResAttachments,
    '/api/v1/attachments/',
    resource_class_kwargs={'gulag_object': gulag}
  )
  api.add_resource(ResRSPAMDImporter,
    '/api/v1/mailboxes/<string:mailbox_id>/rspamdimporter/',
    resource_class_kwargs={'gulag_object': gulag}
  )
  if __name__ == '__main__':
    app.run(debug=False,
      # will be overriden by uwsgi.ini
      host=gulag.config['daemon']['listen_host'],
      port=gulag.config['daemon']['listen_port']
    )
    gulag.db.close()
    sys.exit(0)
except:
  print("MAIN-EXCEPTION: " + str(sys.exc_info()))