1cfe1aff13
* root: initial rename * web: rename custom element prefix * root: rename external functions with pb_ prefix * root: fix formatting * root: replace domain with goauthentik.io * proxy: update path * root: rename remaining prefixes * flows: rename file extension * root: pbadmin -> akadmin * docs: fix image filenames * lifecycle: ignore migration files * ci: copy default config from current source before loading last tagged * *: new sentry dsn * tests: fix missing python3.9-dev package * root: add additional migrations for service accounts created by outposts * core: mark system-created service accounts with attribute * policies/expression: fix pb_ replacement not working * web: fix last linting errors, add lit-analyse * policies/expressions: fix lint errors * web: fix sidebar display on screens where not all items fit * proxy: attempt to fix proxy pipeline * proxy: use go env GOPATH to get gopath * lib: fix user_default naming inconsistency * docs: add upgrade docs * docs: update screenshots to use authentik * admin: fix create button on empty-state of outpost * web: fix modal submit not refreshing SiteShell and Table * web: fix height of app-card and height of generic icon * web: fix rendering of subtext * admin: fix version check error not being caught * web: fix worker count not being shown * docs: update screenshots * root: new icon * web: fix lint error * admin: fix linting error * root: migrate coverage config to pyproject
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
"""authentik admin Middleware to impersonate users"""
|
|
from logging import Logger
|
|
from threading import local
|
|
from typing import Callable
|
|
from uuid import uuid4
|
|
|
|
from django.http import HttpRequest, HttpResponse
|
|
|
|
SESSION_IMPERSONATE_USER = "authentik_impersonate_user"
|
|
SESSION_IMPERSONATE_ORIGINAL_USER = "authentik_impersonate_original_user"
|
|
LOCAL = local()
|
|
|
|
|
|
class ImpersonateMiddleware:
|
|
"""Middleware to impersonate users"""
|
|
|
|
get_response: Callable[[HttpRequest], HttpResponse]
|
|
|
|
def __init__(self, get_response: Callable[[HttpRequest], HttpResponse]):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request: HttpRequest) -> HttpResponse:
|
|
# No permission checks are done here, they need to be checked before
|
|
# SESSION_IMPERSONATE_USER is set.
|
|
|
|
if SESSION_IMPERSONATE_USER in request.session:
|
|
request.user = request.session[SESSION_IMPERSONATE_USER]
|
|
|
|
return self.get_response(request)
|
|
|
|
|
|
class RequestIDMiddleware:
|
|
"""Add a unique ID to every request"""
|
|
|
|
get_response: Callable[[HttpRequest], HttpResponse]
|
|
|
|
def __init__(self, get_response: Callable[[HttpRequest], HttpResponse]):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request: HttpRequest) -> HttpResponse:
|
|
if not hasattr(request, "request_id"):
|
|
request_id = uuid4().hex
|
|
setattr(request, "request_id", request_id)
|
|
LOCAL.authentik = {"request_id": request_id}
|
|
response = self.get_response(request)
|
|
response["X-authentik-id"] = request.request_id
|
|
del LOCAL.authentik["request_id"]
|
|
return response
|
|
|
|
|
|
# pylint: disable=unused-argument
|
|
def structlog_add_request_id(logger: Logger, method_name: str, event_dict):
|
|
"""If threadlocal has authentik defined, add request_id to log"""
|
|
if hasattr(LOCAL, "authentik"):
|
|
event_dict["request_id"] = LOCAL.authentik.get("request_id", "")
|
|
return event_dict
|