from apiflask import APIBlueprint

from app import rpc
from app.api.v1.schema.auth import WechatLoginIn, WechatLoginOut, RefreshTokenIn, RefreshTokenOut

auth = APIBlueprint('auth', __name__)


@auth.post('/wechat_login')
@auth.input(WechatLoginIn)
@auth.output(WechatLoginOut)
def wechat_login(data):
    try:
        result = rpc.admin.wechat_login(data['code'])
    except Exception as e:
        raise e
    return result


@auth.get('/refresh_token')
@auth.input(RefreshTokenIn, location='query')
@auth.output(RefreshTokenOut)
def refresh_token(data):
    result = rpc.admin.refresh_token(data['refresh_token'])
    return result