#
# Copyright 2015-2016 IBM Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

import sys
import os
import json
import subprocess
import codecs
import traceback
import flask
from gevent.wsgi import WSGIServer

proxy = flask.Flask(__name__)
proxy.debug = False
with open("function.py") as f:
    flask.g = f.read()

@proxy.route("/run", methods=['POST'])
def run():
    message = flask.request.get_json(force=True,silent=True)

    if not message or not isinstance(message, dict):
        flask.abort(403)

    if not "value" in message:
        flask.abort(403)

    value = message["value"]

    if not isinstance(value, dict):
        flask.abort(403)

    # initialize the namespace for the execution
    namespace = {}
    result = None
    try:
        exec(flask.g, namespace)
        exec("param = " + json.dumps(value), namespace)
        exec("fun = main(param)", namespace)
        result = namespace['fun']
    except Exception:
        traceback.print_exc(file = sys.stderr)
    sys.stdout.flush()
    sys.stderr.flush()
    if result and isinstance(result, dict):
        response = flask.jsonify(result)
        response.status_code = 200
        return response
    else:
        response = flask.jsonify({ "error": "the action did not return a dictionary", "action_output": result })
        response.status_code = 502
        return response


# start server in a forever loop
if __name__ == "__main__":
    server = WSGIServer(('', int(os.getenv("PORT", 8080))), proxy, log=None)
    server.serve_forever()
