MonkeyBusiness/modules/core/cardmng.py

from fastapi import APIRouter, Request, Response
from tinydb import Query, where

from core_common import core_process_request, core_prepare_response, E
from core_database import get_db

router = APIRouter(prefix="/core", tags=["cardmng"])


def get_target_table(game_id):
    target_table = {
        "LDJ": "iidx_profile",
        "MDX": "ddr_profile",
        "KFC": "sdvx_profile",
        "JDZ": "iidx_profile",
        "KDZ": "iidx_profile",
    }

    return target_table[game_id]


def get_profile(game_id, cid):
    target_table = get_target_table(game_id)
    profile = get_db().table(target_table).get(where("card") == cid)

    if profile is None:
        profile = {
            "card": cid,
            "version": {},
        }

    return profile


def get_game_profile(game_id, game_version, cid):
    profile = get_profile(game_id, cid)

    if str(game_version) not in profile["version"]:
        profile["version"][str(game_version)] = {}

    return profile["version"][str(game_version)]


def create_profile(game_id, game_version, cid, pin):
    target_table = get_target_table(game_id)
    profile = get_profile(game_id, cid)

    profile["pin"] = pin

    get_db().table(target_table).upsert(profile, where("card") == cid)


@router.post("/{gameinfo}/cardmng/authpass")
async def cardmng_authpass(request: Request):
    request_info = await core_process_request(request)

    cid = request_info["root"][0].attrib["refid"]
    passwd = request_info["root"][0].attrib["pass"]

    profile = get_profile(request_info["model"], cid)
    if profile is None or passwd != profile.get("pin", None):
        status = 116
    else:
        status = 0

    response = E.response(E.authpass(status=status))

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/cardmng/bindmodel")
async def cardmng_bindmodel(request: Request):
    request_info = await core_process_request(request)

    response = E.response(
        E.bindmodel(
            dataid=1,
        )
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/cardmng/getrefid")
async def cardmng_getrefid(request: Request):
    request_info = await core_process_request(request)

    cid = request_info["root"][0].attrib["cardid"]
    passwd = request_info["root"][0].attrib["passwd"]

    create_profile(request_info["model"], request_info["game_version"], cid, passwd)

    response = E.response(
        E.getrefid(
            dataid=cid,
            refid=cid,
        )
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/cardmng/inquire")
async def cardmng_inquire(request: Request):
    request_info = await core_process_request(request)

    cid = request_info["root"][0].attrib["cardid"]

    profile = get_game_profile(request_info["model"], request_info["game_version"], cid)
    if profile:
        binded = 1
        newflag = 0
        status = 0
    else:
        binded = 0
        newflag = 1
        status = 112

    response = E.response(
        E.inquire(
            dataid=cid,
            ecflag=1,
            expired=0,
            binded=binded,
            newflag=newflag,
            refid=cid,
            status=status,
        )
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)