Files
notifications-api/app/user/rest.py
2025-07-16 11:33:15 -07:00

735 lines
25 KiB
Python

import json
import uuid
from urllib.parse import urlencode
from flask import Blueprint, abort, current_app, jsonify, request
from sqlalchemy.exc import IntegrityError
from sqlalchemy.orm.exc import NoResultFound
from app import db, redis_store
from app.config import QueueNames
from app.dao.permissions_dao import permission_dao
from app.dao.service_user_dao import dao_get_service_user, dao_update_service_user
from app.dao.services_dao import dao_fetch_service_by_id
from app.dao.template_folder_dao import dao_get_template_folder_by_id_and_service_id
from app.dao.templates_dao import dao_get_template_by_id
from app.dao.users_dao import (
count_user_verify_codes,
create_secret_code,
create_user_code,
dao_archive_user,
dao_report_users,
get_login_gov_user,
get_user_and_accounts,
get_user_by_email,
get_user_by_id,
get_user_code,
get_users,
get_users_by_partial_email,
increment_failed_login_count,
reset_failed_login_count,
save_model_user,
save_user_attribute,
use_user_code,
)
from app.enums import CodeType, KeyType, NotificationType, TemplateType
from app.errors import InvalidRequest, register_errors
from app.models import Permission, Service
from app.notifications.process_notifications import (
persist_notification,
send_notification_to_queue,
)
from app.schema_validation import validate
from app.schemas import (
create_user_schema,
email_data_request_schema,
partial_email_data_request_schema,
user_update_schema_load_json,
)
from app.user.users_schema import (
post_send_user_email_code_schema,
post_send_user_sms_code_schema,
post_set_permissions_schema,
post_verify_code_schema,
post_verify_webauthn_schema,
)
from app.utils import (
check_suspicious_id,
debug_not_production,
hilite,
url_with_token,
utc_now,
)
from notifications_utils.recipients import is_us_phone_number, use_numeric_sender
user_blueprint = Blueprint("user", __name__)
register_errors(user_blueprint)
@user_blueprint.errorhandler(IntegrityError)
def handle_integrity_error(exc):
"""
Handle integrity errors caused by the auth type/mobile number check constraint
"""
if "ck_user_has_mobile_or_other_auth" in str(exc):
# we don't expect this to trip, so still log error
current_app.logger.exception(
"Check constraint ck_user_has_mobile_or_other_auth triggered"
)
return (
jsonify(
result="error",
message="Mobile number must be set if auth_type is set to AuthType.SMS",
),
400,
)
raise exc
@user_blueprint.route("", methods=["POST"])
def create_user():
current_app.logger.info("#invites: enter create_user()")
req_json = request.get_json()
user_to_create = create_user_schema.load(req_json)
save_model_user(
user_to_create, password=req_json.get("password"), validated_email_access=True
)
result = user_to_create.serialize()
current_app.logger.info("#invites: returning the new user")
return jsonify(data=result), 201
@user_blueprint.route("/<uuid:user_id>", methods=["POST"])
def update_user_attribute(user_id):
check_suspicious_id(user_id)
user_to_update = get_user_by_id(user_id=user_id)
req_json = request.get_json()
if "updated_by" in req_json:
updated_by = get_user_by_id(user_id=req_json.pop("updated_by"))
else:
updated_by = None
update_dct = user_update_schema_load_json.load(req_json)
save_user_attribute(user_to_update, update_dict=update_dct)
if updated_by:
if "email_address" in update_dct:
template = dao_get_template_by_id(
current_app.config["TEAM_MEMBER_EDIT_EMAIL_TEMPLATE_ID"]
)
recipient = user_to_update.email_address
reply_to = template.service.get_default_reply_to_email_address()
elif "mobile_number" in update_dct:
template = dao_get_template_by_id(
current_app.config["TEAM_MEMBER_EDIT_MOBILE_TEMPLATE_ID"]
)
recipient = user_to_update.mobile_number
reply_to = get_sms_reply_to_for_notify_service(recipient, template)
else:
return jsonify(data=user_to_update.serialize()), 200
service = db.session.get(Service, current_app.config["NOTIFY_SERVICE_ID"])
personalisation = {
"name": user_to_update.name,
"servicemanagername": updated_by.name,
"email address": user_to_update.email_address,
}
saved_notification = persist_notification(
template_id=template.id,
template_version=template.version,
recipient=recipient,
service=service,
personalisation={},
notification_type=template.template_type,
api_key_id=None,
key_type=KeyType.NORMAL,
reply_to_text=reply_to,
)
saved_notification.personalisation = personalisation
redis_store.set(
f"email-personalisation-{saved_notification.id}",
json.dumps(personalisation),
ex=60 * 60,
)
send_notification_to_queue(saved_notification, queue=QueueNames.NOTIFY)
return jsonify(data=user_to_update.serialize()), 200
def get_sms_reply_to_for_notify_service(recipient, template):
if not is_us_phone_number(recipient) and use_numeric_sender(recipient):
reply_to = current_app.config["NOTIFY_INTERNATIONAL_SMS_SENDER"]
else:
reply_to = template.service.get_default_sms_sender()
return reply_to
@user_blueprint.route("/<uuid:user_id>/archive", methods=["POST"])
def archive_user(user_id):
check_suspicious_id(user_id)
user = get_user_by_id(user_id)
dao_archive_user(user)
return "", 204
@user_blueprint.route("/<uuid:user_id>/activate", methods=["POST"])
def activate_user(user_id):
check_suspicious_id(user_id)
user = get_user_by_id(user_id=user_id)
if user.state == "active":
raise InvalidRequest("User already active", status_code=400)
user.state = "active"
save_model_user(user)
return jsonify(data=user.serialize()), 200
@user_blueprint.route("/<uuid:user_id>/deactivate", methods=["POST"])
def deactivate_user(user_id):
check_suspicious_id(user_id)
user = get_user_by_id(user_id=user_id)
if user.state == "pending":
raise InvalidRequest("User already inactive", status_code=400)
user.state = "pending"
save_model_user(user)
return jsonify(data=user.serialize()), 200
@user_blueprint.route("/<uuid:user_id>/reset-failed-login-count", methods=["POST"])
def user_reset_failed_login_count(user_id):
check_suspicious_id(user_id)
user_to_update = get_user_by_id(user_id=user_id)
reset_failed_login_count(user_to_update)
return jsonify(data=user_to_update.serialize()), 200
@user_blueprint.route("/<uuid:user_id>/verify/password", methods=["POST"])
def verify_user_password(user_id):
check_suspicious_id(user_id)
user_to_verify = get_user_by_id(user_id=user_id)
try:
txt_pwd = request.get_json()["password"]
except KeyError:
message = "Required field missing data"
errors = {"password": [message]}
raise InvalidRequest(errors, status_code=400)
if user_to_verify.check_password(txt_pwd):
reset_failed_login_count(user_to_verify)
return jsonify({}), 204
else:
increment_failed_login_count(user_to_verify)
message = "Incorrect password"
errors = {"password": [message]}
raise InvalidRequest(errors, status_code=400)
@user_blueprint.route("/<uuid:user_id>/verify/code", methods=["POST"])
def verify_user_code(user_id):
check_suspicious_id(user_id)
data = request.get_json()
validate(data, post_verify_code_schema)
user_to_verify = get_user_by_id(user_id=user_id)
code = get_user_code(user_to_verify, data["code"], data["code_type"])
if user_to_verify.failed_login_count >= current_app.config.get(
"MAX_FAILED_LOGIN_COUNT"
):
raise InvalidRequest("Code not found", status_code=404)
if not code:
# only relevant from sms
increment_failed_login_count(user_to_verify)
raise InvalidRequest("Code not found", status_code=404)
if utc_now() > code.expiry_datetime or code.code_used:
# sms and email
increment_failed_login_count(user_to_verify)
raise InvalidRequest("Code has expired", status_code=400)
user_to_verify.current_session_id = str(uuid.uuid4())
user_to_verify.logged_in_at = utc_now()
if data["code_type"] == CodeType.EMAIL:
user_to_verify.email_access_validated_at = utc_now()
user_to_verify.failed_login_count = 0
save_model_user(user_to_verify)
use_user_code(code.id)
return jsonify({}), 204
# TODO: Remove the "verify" endpoint once admin no longer points at it
@user_blueprint.route("/<uuid:user_id>/complete/webauthn-login", methods=["POST"])
@user_blueprint.route("/<uuid:user_id>/verify/webauthn-login", methods=["POST"])
def complete_login_after_webauthn_authentication_attempt(user_id):
check_suspicious_id(user_id)
"""
complete login after a webauthn authentication. There's nothing webauthn specific in this code
but the sms/email flows do this as part of `verify_user_code` above and this is the equivalent spot in the
webauthn flow.
If the authentication was successful, we've already confirmed the user holds the right security key,
but we still need to check the max login count and set up a current_session_id and last_logged_in_at here.
If the authentication was unsuccessful then we just bump the failed_login_count in the db.
"""
data = request.get_json()
validate(data, post_verify_webauthn_schema)
user = get_user_by_id(user_id=user_id)
successful = data["successful"]
if user.failed_login_count >= current_app.config.get("MAX_VERIFY_CODE_COUNT"):
raise InvalidRequest("Maximum login count exceeded", status_code=403)
if successful:
user.current_session_id = str(uuid.uuid4())
user.logged_in_at = utc_now()
user.failed_login_count = 0
save_model_user(user)
else:
increment_failed_login_count(user)
return jsonify({}), 204
@user_blueprint.route("/<uuid:user_id>/<code_type>-code", methods=["POST"])
def send_user_2fa_code(user_id, code_type):
check_suspicious_id(user_id)
user_to_send_to = get_user_by_id(user_id=user_id)
if count_user_verify_codes(user_to_send_to) >= current_app.config.get(
"MAX_VERIFY_CODE_COUNT"
):
# Prevent more than `MAX_VERIFY_CODE_COUNT` active verify codes at a time
current_app.logger.warning(
"Too many verify codes created for user {}".format(user_to_send_to.id)
)
else:
data = request.get_json()
if NotificationType(code_type) == NotificationType.SMS:
validate(data, post_send_user_sms_code_schema)
send_user_sms_code(user_to_send_to, data)
elif NotificationType(code_type) == NotificationType.EMAIL:
validate(data, post_send_user_email_code_schema)
send_user_email_code(user_to_send_to, data)
else:
abort(404)
return "{}", 204
def send_user_sms_code(user_to_send_to, data):
recipient = data.get("to") or user_to_send_to.mobile_number
secret_code = create_secret_code()
personalisation = {"verify_code": secret_code}
create_2fa_code(
current_app.config["SMS_CODE_TEMPLATE_ID"],
user_to_send_to,
secret_code,
recipient,
personalisation,
)
def send_user_email_code(user_to_send_to, data):
recipient = user_to_send_to.email_address
secret_code = str(uuid.uuid4())
personalisation = {
"name": user_to_send_to.name,
"url": _create_2fa_url(
user_to_send_to,
secret_code,
data.get("next"),
data.get("email_auth_link_host"),
),
}
create_2fa_code(
current_app.config["EMAIL_2FA_TEMPLATE_ID"],
user_to_send_to,
secret_code,
recipient,
personalisation,
)
def create_2fa_code(
template_id, user_to_send_to, secret_code, recipient, personalisation
):
template = dao_get_template_by_id(template_id)
# save the code in the VerifyCode table
create_user_code(user_to_send_to, secret_code, template.template_type)
reply_to = None
if template.template_type == TemplateType.SMS:
reply_to = get_sms_reply_to_for_notify_service(recipient, template)
elif template.template_type == TemplateType.EMAIL:
reply_to = template.service.get_default_reply_to_email_address()
saved_notification = persist_notification(
template_id=template.id,
template_version=template.version,
recipient=recipient,
service=template.service,
personalisation=personalisation,
notification_type=template.template_type,
api_key_id=None,
key_type=KeyType.NORMAL,
reply_to_text=reply_to,
)
saved_notification.personalisation = personalisation
key = f"2facode-{saved_notification.id}".replace(" ", "")
recipient = str(recipient)
redis_store.set(key, recipient, ex=60 * 60)
# Assume that we never want to observe the Notify service's research mode
# setting for this notification - we still need to be able to log into the
# admin even if we're doing user research using this service:
redis_store.set(
f"email-personalisation-{saved_notification.id}",
json.dumps(personalisation),
ex=60 * 60,
)
send_notification_to_queue(saved_notification, queue=QueueNames.NOTIFY)
@user_blueprint.route("/<uuid:user_id>/change-email-verification", methods=["POST"])
def send_user_confirm_new_email(user_id):
check_suspicious_id(user_id)
user_to_send_to = get_user_by_id(user_id=user_id)
email = email_data_request_schema.load(request.get_json())
template = dao_get_template_by_id(
current_app.config["CHANGE_EMAIL_CONFIRMATION_TEMPLATE_ID"]
)
service = db.session.get(Service, current_app.config["NOTIFY_SERVICE_ID"])
personalisation = {
"name": user_to_send_to.name,
"url": _create_confirmation_url(
user=user_to_send_to, email_address=email["email"]
),
"feedback_url": current_app.config["ADMIN_BASE_URL"] + "/support",
}
saved_notification = persist_notification(
template_id=template.id,
template_version=template.version,
recipient=email["email"],
service=service,
personalisation={},
notification_type=template.template_type,
api_key_id=None,
key_type=KeyType.NORMAL,
reply_to_text=service.get_default_reply_to_email_address(),
)
saved_notification.personalisation = personalisation
redis_store.set(
f"email-personalisation-{saved_notification.id}",
json.dumps(personalisation),
ex=60 * 60,
)
send_notification_to_queue(saved_notification, queue=QueueNames.NOTIFY)
return jsonify({}), 204
@user_blueprint.route("/<uuid:user_id>/email-verification", methods=["POST"])
def send_new_user_email_verification(user_id):
check_suspicious_id(user_id)
current_app.logger.info("Sending email verification for user {}".format(user_id))
request_json = request.get_json()
# when registering, we verify all users' email addresses using this function
user_to_send_to = get_user_by_id(user_id=user_id)
template = dao_get_template_by_id(
current_app.config["NEW_USER_EMAIL_VERIFICATION_TEMPLATE_ID"]
)
service = db.session.get(Service, current_app.config["NOTIFY_SERVICE_ID"])
current_app.logger.info("template.id is {}".format(template.id))
current_app.logger.info("service.id is {}".format(service.id))
personalisation = {
"name": user_to_send_to.name,
"url": _create_verification_url(
user_to_send_to,
base_url=request_json.get("admin_base_url"),
),
}
saved_notification = persist_notification(
template_id=template.id,
template_version=template.version,
recipient=user_to_send_to.email_address,
service=service,
personalisation={},
notification_type=template.template_type,
api_key_id=None,
key_type=KeyType.NORMAL,
reply_to_text=service.get_default_reply_to_email_address(),
)
saved_notification.personalisation = personalisation
redis_store.set(
f"email-address-{saved_notification.id}",
str(user_to_send_to.email_address),
ex=60 * 60,
)
redis_store.set(
f"email-personalisation-{saved_notification.id}",
json.dumps(personalisation),
ex=60 * 60,
)
current_app.logger.info("Sending notification to queue")
send_notification_to_queue(saved_notification, queue=QueueNames.NOTIFY)
current_app.logger.info("Sent notification to queue")
return jsonify({}), 204
@user_blueprint.route("/<uuid:user_id>/email-already-registered", methods=["POST"])
def send_already_registered_email(user_id):
check_suspicious_id(user_id)
current_app.logger.info("Email already registered for user {}".format(user_id))
to = email_data_request_schema.load(request.get_json())
current_app.logger.info("To email is {}".format(to["email"]))
template = dao_get_template_by_id(
current_app.config["ALREADY_REGISTERED_EMAIL_TEMPLATE_ID"]
)
service = db.session.get(Service, current_app.config["NOTIFY_SERVICE_ID"])
current_app.logger.info("template.id is {}".format(template.id))
current_app.logger.info("service.id is {}".format(service.id))
personalisation = {
"signin_url": current_app.config["ADMIN_BASE_URL"] + "/sign-in",
"forgot_password_url": current_app.config["ADMIN_BASE_URL"]
+ "/forgot-password",
"feedback_url": current_app.config["ADMIN_BASE_URL"] + "/support",
}
saved_notification = persist_notification(
template_id=template.id,
template_version=template.version,
recipient=to["email"],
service=service,
personalisation={},
notification_type=template.template_type,
api_key_id=None,
key_type=KeyType.NORMAL,
reply_to_text=service.get_default_reply_to_email_address(),
)
saved_notification.personalisation = personalisation
current_app.logger.info("Sending notification to queue")
redis_store.set(
f"email-personalisation-{saved_notification.id}",
json.dumps(personalisation),
ex=60 * 60,
)
send_notification_to_queue(saved_notification, queue=QueueNames.NOTIFY)
current_app.logger.info("Sent notification to queue")
return jsonify({}), 204
@user_blueprint.route("/<uuid:user_id>", methods=["GET"])
@user_blueprint.route("", methods=["GET"])
def get_user(user_id=None):
check_suspicious_id(user_id)
users = get_user_by_id(user_id=user_id)
result = (
[x.serialize() for x in users] if isinstance(users, list) else users.serialize()
)
return jsonify(data=result)
@user_blueprint.route(
"/<uuid:user_id>/service/<uuid:service_id>/permission", methods=["POST"]
)
def set_permissions(user_id, service_id):
check_suspicious_id(user_id, service_id)
# TODO fix security hole, how do we verify that the user
# who is making this request has permission to make the request.
service_user = dao_get_service_user(user_id, service_id)
# TODO: Below exception is raised to account for the test case failure that got handled
# on its own in 1.4 when dao_get_service_user() returned an excpetion in case no result was found
if service_user is None:
raise NoResultFound(
"No ServiceUser found with the provided user_id and service_id"
)
user = get_user_by_id(user_id)
service = dao_fetch_service_by_id(service_id=service_id)
data = request.get_json()
validate(data, post_set_permissions_schema)
permission_list = [
Permission(service_id=service_id, user_id=user_id, permission=p["permission"])
for p in data["permissions"]
]
permission_dao.set_user_service_permission(
user, service, permission_list, _commit=True, replace=True
)
if "folder_permissions" in data:
folders = [
dao_get_template_folder_by_id_and_service_id(folder_id, service_id)
for folder_id in data["folder_permissions"]
]
service_user.folders = folders
dao_update_service_user(service_user)
return jsonify({}), 204
@user_blueprint.route("/get-login-gov-user", methods=["POST"])
def get_user_login_gov_user():
current_app.logger.info("#invites: enter get_user_login_gov_user")
request_args = request.get_json()
login_uuid = request_args["login_uuid"]
email = request_args["email"]
current_app.logger.info(
f"#invites: login_uuid {login_uuid} email is null? {email is None}"
)
user = get_login_gov_user(login_uuid, email)
if user is None:
current_app.logger.info("#invites: couldn't find user, returning empty data")
return jsonify({})
current_app.logger.info("#invites: found the user, serializing")
result = user.serialize()
current_app.logger.info("#invites: returning successfully")
return jsonify(data=result)
@user_blueprint.route("/email", methods=["POST"])
def fetch_user_by_email():
try:
debug_not_production(
hilite(f"enter fetch_user_by_email with {request.get_json()}")
)
email = email_data_request_schema.load(request.get_json())
fetched_user = get_user_by_email(email["email"])
result = fetched_user.serialize()
return jsonify(data=result)
except Exception as e:
debug_not_production(hilite(f"Failed with {e}!!"))
raise e
# TODO: Deprecate this GET endpoint
@user_blueprint.route("/email", methods=["GET"])
def get_by_email():
email = request.args.get("email")
if not email:
error = "Invalid request. Email query string param required"
raise InvalidRequest(error, status_code=400)
fetched_user = get_user_by_email(email)
result = fetched_user.serialize()
return jsonify(data=result)
@user_blueprint.route("/find-users-by-email", methods=["POST"])
def find_users_by_email():
email = partial_email_data_request_schema.load(request.get_json())
fetched_users = get_users_by_partial_email(email["email"])
result = [user.serialize_for_users_list() for user in fetched_users]
return jsonify(data=result), 200
@user_blueprint.route("/get-all-users", methods=["GET"])
def get_all_users():
users = get_users()
result = [user.serialize_for_users_list() for user in users]
return jsonify(data=result), 200
@user_blueprint.route("/report-all-users", methods=["GET"])
def report_all_users():
users = dao_report_users()
download_users = []
for user in users:
new_user = {
"name": user[0],
"email_address": user[1],
"mobile_number": user[2],
"service": user[3],
}
download_users.append(new_user)
return jsonify(data=download_users, status=200, mime_type="application/json"), 200
@user_blueprint.route("/<uuid:user_id>/organizations-and-services", methods=["GET"])
def get_organizations_and_services_for_user(user_id):
check_suspicious_id(user_id)
user = get_user_and_accounts(user_id)
data = get_orgs_and_services(user)
return jsonify(data)
def _create_verification_url(user, base_url):
data = json.dumps({"user_id": str(user.id), "email": user.email_address})
url = "/verify-email/"
return url_with_token(data, url, current_app.config, base_url=base_url)
def _create_confirmation_url(user, email_address):
data = json.dumps({"user_id": str(user.id), "email": email_address})
url = "/user-profile/email/confirm/"
return url_with_token(data, url, current_app.config)
def _create_2fa_url(user, secret_code, next_redirect, email_auth_link_host):
data = json.dumps({"user_id": str(user.id), "secret_code": secret_code})
url = "/email-auth/"
full_url = url_with_token(
data, url, current_app.config, base_url=email_auth_link_host
)
if next_redirect:
full_url += "?{}".format(urlencode({"next": next_redirect}))
return full_url
def get_orgs_and_services(user):
return {
"organizations": [
{
"name": org.name,
"id": org.id,
"count_of_live_services": len(org.live_services),
}
for org in user.organizations
if org.active
],
"services": [
{
"id": service.id,
"name": service.name,
"restricted": service.restricted,
"organization": (
service.organization.id if service.organization else None
),
}
for service in user.services
if service.active
],
}