import base64
import json
from flask import Blueprint,request
from rail_proto.lib import GateServer

query_gateway_blueprint = Blueprint('query_gateway', __name__)

@query_gateway_blueprint.route("/query_gateway", methods=["GET"])
def query_gateway():
    version = request.args.get('version')
    with open('version.json', 'r') as f:
        reader = json.load(f)
    config = reader.get(version)
    if config:
        rsp = GateServer(
            retcode=0,
            ip="127.0.0.1",
            port=23301,
            asset_bundle_url="",
            lua_url="",
            ex_resource_url="",
            mdk_res_version="",
            enable_version_update=True,
            enable_design_data_version_update=True,
            enable_save_replay_file=True,
            enable_upload_battle_log=True,
            enable_watermark=True,
            event_tracking_open=True,
        )
    else:
        rsp = GateServer(
            retcode=0,
            ip="127.0.0.1",
            port=23301,
            asset_bundle_url="",
            lua_url="",
            ex_resource_url="",
            enable_version_update=True,
            enable_design_data_version_update=True,
            enable_save_replay_file=True,
            enable_upload_battle_log=True,
            enable_watermark=True,
            event_tracking_open=True,
        )
    return base64.b64encode(rsp.SerializeToString()).decode()