mephisto.abstractions.architects.router.flask.app

#!/usr/bin/env python3

# Copyright (c) Meta Platforms and its affiliates.
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

from gevent import monkey  # type: ignore

monkey.patch_all()

try:
    from mephisto.abstractions.architects.router.flask.mephisto_flask_blueprint import (  # type: ignore
        MephistoRouter,
        mephisto_router,
    )
except:
    from mephisto_flask_blueprint import (  # type: ignore
        MephistoRouter,
        mephisto_router,
    )
from geventwebsocket import WebSocketServer, Resource  # type: ignore
from werkzeug.debug import DebuggedApplication  # type: ignore


from flask import Flask  # type: ignore
import os

port = int(os.environ.get("PORT", 3000))

flask_app = Flask(__name__)
flask_app.register_blueprint(mephisto_router, url_prefix=r"/")

if __name__ == "__main__":
    WebSocketServer(
        ("", port),
        Resource([("^/.*", MephistoRouter), ("^/.*", DebuggedApplication(flask_app))]),
        debug=False,
    ).serve_forever()