- add __tablename__ for all models - use sa and orm instead of db - rollback all changes in tests - remove session in @app.teardown_appcontext
72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
import pyotp
|
|
from flask import jsonify, request
|
|
from flask_login import login_user
|
|
from itsdangerous import Signer
|
|
|
|
from app.api.base import api_bp
|
|
from app.config import FLASK_SECRET
|
|
from app.db import Session
|
|
from app.log import LOG
|
|
from app.models import User, ApiKey
|
|
|
|
|
|
@api_bp.route("/auth/mfa", methods=["POST"])
|
|
def auth_mfa():
|
|
"""
|
|
Validate the OTP Token
|
|
Input:
|
|
mfa_token: OTP token that user enters
|
|
mfa_key: MFA key obtained in previous auth request, e.g. /api/auth/login
|
|
device: the device name, used to create an ApiKey associated with this device
|
|
Output:
|
|
200 and user info containing:
|
|
{
|
|
name: "John Wick",
|
|
api_key: "a long string",
|
|
email: "user email"
|
|
}
|
|
|
|
"""
|
|
data = request.get_json()
|
|
if not data:
|
|
return jsonify(error="request body cannot be empty"), 400
|
|
|
|
mfa_token = data.get("mfa_token")
|
|
mfa_key = data.get("mfa_key")
|
|
device = data.get("device")
|
|
|
|
s = Signer(FLASK_SECRET)
|
|
try:
|
|
user_id = int(s.unsign(mfa_key))
|
|
except Exception:
|
|
return jsonify(error="Invalid mfa_key"), 400
|
|
|
|
user = User.get(user_id)
|
|
|
|
if not user:
|
|
return jsonify(error="Invalid mfa_key"), 400
|
|
elif not user.enable_otp:
|
|
return (
|
|
jsonify(error="This endpoint should only be used by user who enables MFA"),
|
|
400,
|
|
)
|
|
|
|
totp = pyotp.TOTP(user.otp_secret)
|
|
if not totp.verify(mfa_token):
|
|
return jsonify(error="Wrong TOTP Token"), 400
|
|
|
|
ret = {"name": user.name or "", "email": user.email}
|
|
|
|
api_key = ApiKey.get_by(user_id=user.id, name=device)
|
|
if not api_key:
|
|
LOG.d("create new api key for %s and %s", user, device)
|
|
api_key = ApiKey.create(user.id, device)
|
|
Session.commit()
|
|
|
|
ret["api_key"] = api_key.code
|
|
|
|
# so user is logged in automatically on the web
|
|
login_user(user)
|
|
|
|
return jsonify(**ret), 200
|