rest.py

# Reece Williams | https://reece.sh | Jan 2023

import json

from flask import Flask, jsonify, request
from flask_cors import CORS, cross_origin

import CONFIG as CONFIG
from CONFIG import KV_STORE
from HELPERS import (
    Mode,
    download_openapi_locally,
    get_config_values,
    get_stats_html,
    get_swagger_code_from_source,
    increment_call_value,
    ttl_block_only,
)
from HELPERS_TYPES import CallType
from RequestsHandler import RestApiHandler

app = Flask(__name__)
cors = CORS(app, resources={r"/*": {"origins": "*"}})


REST_SWAGGER_HTML = ""
REST_HANDLER: RestApiHandler


@app.before_first_request
def before_first_request():
    global REST_HANDLER
    CONFIG.update_cache_times()
    download_openapi_locally()
    REST_HANDLER = RestApiHandler()

    # not used yet here
    # if len(CONFIG.RPC_WEBSOCKET) > 0:
    #     tmrpc = TendermintRPCWebSocket(enableSignal=False, logLevel=logging.DEBUG)
    #     t = threading.Thread(target=tmrpc.ws.run_forever)
    #     t.daemon = True
    #     t.start()


@app.route("/", methods=["GET"])
@cross_origin()
def root():
    global REST_SWAGGER_HTML

    if CONFIG.DISABLE_SWAGGER_UI:
        return jsonify({"code": 12, "message": "Not Implemented", "details": []})

    if len(REST_SWAGGER_HTML) > 0:
        return REST_SWAGGER_HTML

    REST_SWAGGER_HTML = get_swagger_code_from_source()
    return REST_SWAGGER_HTML


@app.route("/<path:path>", methods=["GET"])
@cross_origin()
def get_rest(path):
    if path == "stats":
        # https://url/stats?password=123
        if (
            len(CONFIG.STATS_PASSWORD) > 0
            and request.args.get("password") != CONFIG.STATS_PASSWORD
        ):
            return "Invalid password"

        return get_stats_html()

    if path == "config":
        # https://url/config?password=123
        if (
            len(CONFIG.STATS_PASSWORD) > 0
            and request.args.get("password") != CONFIG.STATS_PASSWORD
        ):
            return "Invalid password"

        return get_config_values()

    if path == "debug":
        return jsonify(KV_STORE.to_json())

    args = request.args

    cache_seconds = CONFIG.get_cache_time_seconds(path, is_rpc=False)
    if cache_seconds == Mode.DISABLED.value:
        return jsonify(
            {
                "error": f"cosmos endpoint cache: The path '{path}' is disabled on this node..."
            }
        )

    # print(cache_seconds)

    # Every rest requests is an hset because of diff arguments
    key = f"rest;{ttl_block_only(cache_seconds)};{path};"

    v = KV_STORE.hget(key, str(args))
    if CONFIG.DEBUGGING:
        print(f"get_rest. Key: {key} | value: {v}")
    if v:
        increment_call_value(CallType.REST_GET_CACHE.value)
        return jsonify(json.loads(v))

    return jsonify(
        REST_HANDLER.handle_single_rest_get_requests(
            path, key, cache_seconds, args, request.headers
        )
    )


@app.route("/<path:path>", methods=["POST"])
@cross_origin()
def post_rest(path):
    # REQ_DATA = json.loads(json.dumps(request.get_json(), separators=(",", ":")))
    # print(type(REQ_DATA))
    # return jsonify(REST_HANDLER.handle_single_rest_post_requests(path, REQ_DATA))
    return jsonify(
        {
            "error": f"cosmos endpoint cache: The path '{path}' does not yet have support on this REST API..."
        }
    )


if __name__ == "__main__":
    before_first_request()
    app.run(debug=True, host="0.0.0.0", port=CONFIG.REST_PORT)