mirror of
https://github.com/GSA/notifications-admin.git
synced 2025-12-08 14:14:27 -05:00
137 lines
3.8 KiB
Python
137 lines
3.8 KiB
Python
from flask import request
|
|
|
|
from app.notify_client.events_api_client import events_api_client
|
|
|
|
EVENT_SCHEMAS = {
|
|
"sucessful_login": {"user_id"},
|
|
"update_user_email": {
|
|
"user_id",
|
|
"updated_by_id",
|
|
"original_email_address",
|
|
"new_email_address",
|
|
},
|
|
"update_user_mobile_number": {
|
|
"user_id",
|
|
"updated_by_id",
|
|
"original_mobile_number",
|
|
"new_mobile_number",
|
|
},
|
|
"remove_user_from_service": {"user_id", "removed_by_id", "service_id"},
|
|
"add_user_to_service": {"user_id", "invited_by_id", "service_id", "ui_permissions"},
|
|
"invite_user_to_service": {
|
|
"email_address",
|
|
"invited_by_id",
|
|
"service_id",
|
|
"ui_permissions",
|
|
},
|
|
"resend_user_invite_to_service": {"email_address", "resent_by_id", "service_id"},
|
|
"cancel_user_invite_to_service": {"email_address", "canceled_by_id", "service_id"},
|
|
"set_user_permissions": {
|
|
"user_id",
|
|
"service_id",
|
|
"original_ui_permissions",
|
|
"new_ui_permissions",
|
|
"set_by_id",
|
|
},
|
|
"archive_user": {"user_id", "archived_by_id"},
|
|
"archive_service": {"service_id", "archived_by_id"},
|
|
"suspend_service": {"service_id", "suspended_by_id"},
|
|
"resume_service": {"service_id", "resumed_by_id"},
|
|
}
|
|
|
|
|
|
def on_user_logged_in(_sender, user):
|
|
_send_event("sucessful_login", user_id=user.id)
|
|
|
|
|
|
def create_email_change_event(**kwargs):
|
|
_send_event("update_user_email", **kwargs)
|
|
|
|
|
|
def create_mobile_number_change_event(**kwargs):
|
|
_send_event("update_user_mobile_number", **kwargs)
|
|
|
|
|
|
def create_remove_user_from_service_event(**kwargs):
|
|
_send_event("remove_user_from_service", **kwargs)
|
|
|
|
|
|
def create_invite_user_to_service_event(**kwargs):
|
|
_send_event("invite_user_to_service", **kwargs)
|
|
|
|
|
|
def create_cancel_user_invite_to_service_event(**kwargs):
|
|
_send_event("cancel_user_invite_to_service", **kwargs)
|
|
|
|
|
|
def create_resend_user_invite_to_service_event(**kwargs):
|
|
_send_event("resend_user_invite_to_service", **kwargs)
|
|
|
|
|
|
def create_add_user_to_service_event(**kwargs):
|
|
_send_event("add_user_to_service", **kwargs)
|
|
|
|
|
|
def create_set_user_permissions_event(**kwargs):
|
|
_send_event("set_user_permissions", **kwargs)
|
|
|
|
|
|
def create_archive_user_event(**kwargs):
|
|
_send_event("archive_user", **kwargs)
|
|
|
|
|
|
def create_suspend_service_event(**kwargs):
|
|
_send_event("suspend_service", **kwargs)
|
|
|
|
|
|
def create_archive_service_event(**kwargs):
|
|
_send_event("archive_service", **kwargs)
|
|
|
|
|
|
def create_resume_service_event(**kwargs):
|
|
_send_event("resume_service", **kwargs)
|
|
|
|
|
|
def _send_event(event_type, **kwargs):
|
|
expected_keys = EVENT_SCHEMAS[event_type]
|
|
actual_keys = set(kwargs.keys())
|
|
|
|
if expected_keys != actual_keys:
|
|
raise ValueError(f"Expected {expected_keys}, but got {actual_keys}")
|
|
|
|
event_data = _construct_event_data(request)
|
|
event_data.update(kwargs)
|
|
|
|
events_api_client.create_event(event_type, event_data)
|
|
|
|
|
|
def _construct_event_data(request):
|
|
return {
|
|
"ip_address": _get_remote_addr(request),
|
|
"browser_fingerprint": _get_browser_fingerprint(request),
|
|
}
|
|
|
|
|
|
# This might not be totally correct depending on proxy setup
|
|
def _get_remote_addr(request):
|
|
if request.headers.getlist("X-Forwarded-For"):
|
|
return request.headers.getlist("X-Forwarded-For")[0]
|
|
else:
|
|
return request.remote_addr
|
|
|
|
|
|
def _get_browser_fingerprint(request):
|
|
browser = request.user_agent.browser
|
|
version = request.user_agent.version
|
|
platform = request.user_agent.platform
|
|
user_agent_string = request.user_agent.string
|
|
# at some point this may be hashed?
|
|
finger_print = {
|
|
"browser": browser,
|
|
"platform": platform,
|
|
"version": version,
|
|
"user_agent_string": user_agent_string,
|
|
}
|
|
|
|
return finger_print
|