Compare commits
	
		
			3 Commits
		
	
	
		
			version/20
			...
			20240219-m
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| f8714aab11 | |||
| 6e707964b4 | |||
| ed30ae434d | 
| @ -1,5 +1,5 @@ | |||||||
| [bumpversion] | [bumpversion] | ||||||
| current_version = 2024.6.0-rc1 | current_version = 2023.10.7 | ||||||
| tag = True | tag = True | ||||||
| commit = True | commit = True | ||||||
| parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(?:-(?P<rc_t>[a-zA-Z-]+)(?P<rc_n>[1-9]\\d*))? | parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(?:-(?P<rc_t>[a-zA-Z-]+)(?P<rc_n>[1-9]\\d*))? | ||||||
| @ -21,8 +21,6 @@ optional_value = final | |||||||
|  |  | ||||||
| [bumpversion:file:schema.yml] | [bumpversion:file:schema.yml] | ||||||
|  |  | ||||||
| [bumpversion:file:blueprints/schema.json] |  | ||||||
|  |  | ||||||
| [bumpversion:file:authentik/__init__.py] | [bumpversion:file:authentik/__init__.py] | ||||||
|  |  | ||||||
| [bumpversion:file:internal/constants/constants.go] | [bumpversion:file:internal/constants/constants.go] | ||||||
|  | |||||||
| @ -6,7 +6,6 @@ build/** | |||||||
| build_docs/** | build_docs/** | ||||||
| *Dockerfile | *Dockerfile | ||||||
| blueprints/local | blueprints/local | ||||||
| .git |  | ||||||
| !gen-ts-api/node_modules | !gen-ts-api/node_modules | ||||||
| !gen-ts-api/dist/** | !gen-ts-api/dist/** | ||||||
| !gen-go-api/ | !gen-go-api/ | ||||||
|  | |||||||
							
								
								
									
										2
									
								
								.github/FUNDING.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/FUNDING.yml
									
									
									
									
										vendored
									
									
								
							| @ -1 +1 @@ | |||||||
| custom: https://goauthentik.io/pricing/ | github: [BeryJu] | ||||||
|  | |||||||
							
								
								
									
										2
									
								
								.github/ISSUE_TEMPLATE/question.md
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/ISSUE_TEMPLATE/question.md
									
									
									
									
										vendored
									
									
								
							| @ -9,7 +9,7 @@ assignees: "" | |||||||
| **Describe your question/** | **Describe your question/** | ||||||
| A clear and concise description of what you're trying to do. | A clear and concise description of what you're trying to do. | ||||||
|  |  | ||||||
| **Relevant info** | **Relevant infos** | ||||||
| i.e. Version of other software you're using, specifics of your setup | i.e. Version of other software you're using, specifics of your setup | ||||||
|  |  | ||||||
| **Screenshots** | **Screenshots** | ||||||
|  | |||||||
							
								
								
									
										69
									
								
								.github/actions/docker-push-variables/action.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										69
									
								
								.github/actions/docker-push-variables/action.yml
									
									
									
									
										vendored
									
									
								
							| @ -11,10 +11,6 @@ inputs: | |||||||
|     description: "Docker image arch" |     description: "Docker image arch" | ||||||
|  |  | ||||||
| outputs: | outputs: | ||||||
|   shouldBuild: |  | ||||||
|     description: "Whether to build image or not" |  | ||||||
|     value: ${{ steps.ev.outputs.shouldBuild }} |  | ||||||
|  |  | ||||||
|   sha: |   sha: | ||||||
|     description: "sha" |     description: "sha" | ||||||
|     value: ${{ steps.ev.outputs.sha }} |     value: ${{ steps.ev.outputs.sha }} | ||||||
| @ -38,10 +34,63 @@ runs: | |||||||
|   steps: |   steps: | ||||||
|     - name: Generate config |     - name: Generate config | ||||||
|       id: ev |       id: ev | ||||||
|       shell: bash |       shell: python | ||||||
|       env: |  | ||||||
|         IMAGE_NAME: ${{ inputs.image-name }} |  | ||||||
|         IMAGE_ARCH: ${{ inputs.image-arch }} |  | ||||||
|         PR_HEAD_SHA: ${{ github.event.pull_request.head.sha }} |  | ||||||
|       run: | |       run: | | ||||||
|         python3 ${{ github.action_path }}/push_vars.py |         """Helper script to get the actual branch name, docker safe""" | ||||||
|  |         import configparser | ||||||
|  |         import os | ||||||
|  |         from time import time | ||||||
|  |  | ||||||
|  |         parser = configparser.ConfigParser() | ||||||
|  |         parser.read(".bumpversion.cfg") | ||||||
|  |  | ||||||
|  |         branch_name = os.environ["GITHUB_REF"] | ||||||
|  |         if os.environ.get("GITHUB_HEAD_REF", "") != "": | ||||||
|  |             branch_name = os.environ["GITHUB_HEAD_REF"] | ||||||
|  |         safe_branch_name = branch_name.replace("refs/heads/", "").replace("/", "-") | ||||||
|  |  | ||||||
|  |         image_names = "${{ inputs.image-name }}".split(",") | ||||||
|  |         image_arch = "${{ inputs.image-arch }}" or None | ||||||
|  |  | ||||||
|  |         is_pull_request = bool("${{ github.event.pull_request.head.sha }}") | ||||||
|  |         is_release = "dev" not in image_names[0] | ||||||
|  |  | ||||||
|  |         sha = os.environ["GITHUB_SHA"] if not is_pull_request else "${{ github.event.pull_request.head.sha }}" | ||||||
|  |  | ||||||
|  |         # 2042.1.0 or 2042.1.0-rc1 | ||||||
|  |         version = parser.get("bumpversion", "current_version") | ||||||
|  |         # 2042.1 | ||||||
|  |         version_family = ".".join(version.split("-", 1)[0].split(".")[:-1]) | ||||||
|  |         prerelease = "-" in version | ||||||
|  |  | ||||||
|  |         image_tags = [] | ||||||
|  |         if is_release: | ||||||
|  |             for name in image_names: | ||||||
|  |                 image_tags += [ | ||||||
|  |                     f"{name}:{version}", | ||||||
|  |                 ] | ||||||
|  |             if not prerelease: | ||||||
|  |                 image_tags += [ | ||||||
|  |                     f"{name}:latest", | ||||||
|  |                     f"{name}:{version_family}", | ||||||
|  |                 ] | ||||||
|  |         else: | ||||||
|  |             suffix = "" | ||||||
|  |             if image_arch and image_arch != "amd64": | ||||||
|  |                 suffix = f"-{image_arch}" | ||||||
|  |             for name in image_names: | ||||||
|  |                 image_tags += [ | ||||||
|  |                     f"{name}:gh-{sha}{suffix}",  # Used for ArgoCD and PR comments | ||||||
|  |                     f"{name}:gh-{safe_branch_name}{suffix}",  # For convenience | ||||||
|  |                     f"{name}:gh-{safe_branch_name}-{int(time())}-{sha[:7]}{suffix}",  # Use by FluxCD | ||||||
|  |                 ] | ||||||
|  |  | ||||||
|  |         image_main_tag = image_tags[0] | ||||||
|  |         image_tags_rendered = ",".join(image_tags) | ||||||
|  |  | ||||||
|  |         with open(os.environ["GITHUB_OUTPUT"], "a+", encoding="utf-8") as _output: | ||||||
|  |             print("sha=%s" % sha, file=_output) | ||||||
|  |             print("version=%s" % version, file=_output) | ||||||
|  |             print("prerelease=%s" % prerelease, file=_output) | ||||||
|  |             print("imageTags=%s" % image_tags_rendered, file=_output) | ||||||
|  |             print("imageMainTag=%s" % image_main_tag, file=_output) | ||||||
|  | |||||||
| @ -1,62 +0,0 @@ | |||||||
| """Helper script to get the actual branch name, docker safe""" |  | ||||||
|  |  | ||||||
| import configparser |  | ||||||
| import os |  | ||||||
| from time import time |  | ||||||
|  |  | ||||||
| parser = configparser.ConfigParser() |  | ||||||
| parser.read(".bumpversion.cfg") |  | ||||||
|  |  | ||||||
| should_build = str(os.environ.get("DOCKER_USERNAME", None) is not None).lower() |  | ||||||
|  |  | ||||||
| branch_name = os.environ["GITHUB_REF"] |  | ||||||
| if os.environ.get("GITHUB_HEAD_REF", "") != "": |  | ||||||
|     branch_name = os.environ["GITHUB_HEAD_REF"] |  | ||||||
| safe_branch_name = branch_name.replace("refs/heads/", "").replace("/", "-").replace("'", "-") |  | ||||||
|  |  | ||||||
| image_names = os.getenv("IMAGE_NAME").split(",") |  | ||||||
| image_arch = os.getenv("IMAGE_ARCH") or None |  | ||||||
|  |  | ||||||
| is_pull_request = bool(os.getenv("PR_HEAD_SHA")) |  | ||||||
| is_release = "dev" not in image_names[0] |  | ||||||
|  |  | ||||||
| sha = os.environ["GITHUB_SHA"] if not is_pull_request else os.getenv("PR_HEAD_SHA") |  | ||||||
|  |  | ||||||
| # 2042.1.0 or 2042.1.0-rc1 |  | ||||||
| version = parser.get("bumpversion", "current_version") |  | ||||||
| # 2042.1 |  | ||||||
| version_family = ".".join(version.split("-", 1)[0].split(".")[:-1]) |  | ||||||
| prerelease = "-" in version |  | ||||||
|  |  | ||||||
| image_tags = [] |  | ||||||
| if is_release: |  | ||||||
|     for name in image_names: |  | ||||||
|         image_tags += [ |  | ||||||
|             f"{name}:{version}", |  | ||||||
|         ] |  | ||||||
|         if not prerelease: |  | ||||||
|             image_tags += [ |  | ||||||
|                 f"{name}:latest", |  | ||||||
|                 f"{name}:{version_family}", |  | ||||||
|             ] |  | ||||||
| else: |  | ||||||
|     suffix = "" |  | ||||||
|     if image_arch and image_arch != "amd64": |  | ||||||
|         suffix = f"-{image_arch}" |  | ||||||
|     for name in image_names: |  | ||||||
|         image_tags += [ |  | ||||||
|             f"{name}:gh-{sha}{suffix}",  # Used for ArgoCD and PR comments |  | ||||||
|             f"{name}:gh-{safe_branch_name}{suffix}",  # For convenience |  | ||||||
|             f"{name}:gh-{safe_branch_name}-{int(time())}-{sha[:7]}{suffix}",  # Use by FluxCD |  | ||||||
|         ] |  | ||||||
|  |  | ||||||
| image_main_tag = image_tags[0] |  | ||||||
| image_tags_rendered = ",".join(image_tags) |  | ||||||
|  |  | ||||||
| with open(os.environ["GITHUB_OUTPUT"], "a+", encoding="utf-8") as _output: |  | ||||||
|     print(f"shouldBuild={should_build}", file=_output) |  | ||||||
|     print(f"sha={sha}", file=_output) |  | ||||||
|     print(f"version={version}", file=_output) |  | ||||||
|     print(f"prerelease={prerelease}", file=_output) |  | ||||||
|     print(f"imageTags={image_tags_rendered}", file=_output) |  | ||||||
|     print(f"imageMainTag={image_main_tag}", file=_output) |  | ||||||
| @ -1,7 +0,0 @@ | |||||||
| #!/bin/bash -x |  | ||||||
| SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd ) |  | ||||||
| GITHUB_OUTPUT=/dev/stdout \ |  | ||||||
|     GITHUB_REF=ref \ |  | ||||||
|     GITHUB_SHA=sha \ |  | ||||||
|     IMAGE_NAME=ghcr.io/goauthentik/server,beryju/authentik \ |  | ||||||
|     python $SCRIPT_DIR/push_vars.py |  | ||||||
							
								
								
									
										8
									
								
								.github/actions/setup/action.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										8
									
								
								.github/actions/setup/action.yml
									
									
									
									
										vendored
									
									
								
							| @ -16,25 +16,25 @@ runs: | |||||||
|         sudo apt-get update |         sudo apt-get update | ||||||
|         sudo apt-get install --no-install-recommends -y libpq-dev openssl libxmlsec1-dev pkg-config gettext |         sudo apt-get install --no-install-recommends -y libpq-dev openssl libxmlsec1-dev pkg-config gettext | ||||||
|     - name: Setup python and restore poetry |     - name: Setup python and restore poetry | ||||||
|       uses: actions/setup-python@v5 |       uses: actions/setup-python@v4 | ||||||
|       with: |       with: | ||||||
|         python-version-file: "pyproject.toml" |         python-version-file: "pyproject.toml" | ||||||
|         cache: "poetry" |         cache: "poetry" | ||||||
|     - name: Setup node |     - name: Setup node | ||||||
|       uses: actions/setup-node@v4 |       uses: actions/setup-node@v3 | ||||||
|       with: |       with: | ||||||
|         node-version-file: web/package.json |         node-version-file: web/package.json | ||||||
|         cache: "npm" |         cache: "npm" | ||||||
|         cache-dependency-path: web/package-lock.json |         cache-dependency-path: web/package-lock.json | ||||||
|     - name: Setup go |     - name: Setup go | ||||||
|       uses: actions/setup-go@v5 |       uses: actions/setup-go@v4 | ||||||
|       with: |       with: | ||||||
|         go-version-file: "go.mod" |         go-version-file: "go.mod" | ||||||
|     - name: Setup dependencies |     - name: Setup dependencies | ||||||
|       shell: bash |       shell: bash | ||||||
|       run: | |       run: | | ||||||
|         export PSQL_TAG=${{ inputs.postgresql_version }} |         export PSQL_TAG=${{ inputs.postgresql_version }} | ||||||
|         docker compose -f .github/actions/setup/docker-compose.yml up -d |         docker-compose -f .github/actions/setup/docker-compose.yml up -d | ||||||
|         poetry install |         poetry install | ||||||
|         cd web && npm ci |         cd web && npm ci | ||||||
|     - name: Generate config |     - name: Generate config | ||||||
|  | |||||||
							
								
								
									
										2
									
								
								.github/actions/setup/docker-compose.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/actions/setup/docker-compose.yml
									
									
									
									
										vendored
									
									
								
							| @ -1,3 +1,5 @@ | |||||||
|  | version: "3.7" | ||||||
|  |  | ||||||
| services: | services: | ||||||
|   postgresql: |   postgresql: | ||||||
|     image: docker.io/library/postgres:${PSQL_TAG:-16} |     image: docker.io/library/postgres:${PSQL_TAG:-16} | ||||||
|  | |||||||
							
								
								
									
										1
									
								
								.github/codespell-words.txt
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.github/codespell-words.txt
									
									
									
									
										vendored
									
									
								
							| @ -4,4 +4,3 @@ hass | |||||||
| warmup | warmup | ||||||
| ontext | ontext | ||||||
| singed | singed | ||||||
| assertIn |  | ||||||
|  | |||||||
							
								
								
									
										4
									
								
								.github/dependabot.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										4
									
								
								.github/dependabot.yml
									
									
									
									
										vendored
									
									
								
							| @ -52,10 +52,6 @@ updates: | |||||||
|       esbuild: |       esbuild: | ||||||
|         patterns: |         patterns: | ||||||
|           - "@esbuild/*" |           - "@esbuild/*" | ||||||
|       rollup: |  | ||||||
|         patterns: |  | ||||||
|           - "@rollup/*" |  | ||||||
|           - "rollup-*" |  | ||||||
|   - package-ecosystem: npm |   - package-ecosystem: npm | ||||||
|     directory: "/tests/wdio" |     directory: "/tests/wdio" | ||||||
|     schedule: |     schedule: | ||||||
|  | |||||||
							
								
								
									
										65
									
								
								.github/workflows/api-py-publish.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										65
									
								
								.github/workflows/api-py-publish.yml
									
									
									
									
										vendored
									
									
								
							| @ -1,65 +0,0 @@ | |||||||
| name: authentik-api-py-publish |  | ||||||
| on: |  | ||||||
|   push: |  | ||||||
|     branches: [main] |  | ||||||
|     paths: |  | ||||||
|       - "schema.yml" |  | ||||||
|   workflow_dispatch: |  | ||||||
| jobs: |  | ||||||
|   build: |  | ||||||
|     runs-on: ubuntu-latest |  | ||||||
|     permissions: |  | ||||||
|       id-token: write |  | ||||||
|     steps: |  | ||||||
|       - id: generate_token |  | ||||||
|         uses: tibdex/github-app-token@v2 |  | ||||||
|         with: |  | ||||||
|           app_id: ${{ secrets.GH_APP_ID }} |  | ||||||
|           private_key: ${{ secrets.GH_APP_PRIVATE_KEY }} |  | ||||||
|       - uses: actions/checkout@v4 |  | ||||||
|         with: |  | ||||||
|           token: ${{ steps.generate_token.outputs.token }} |  | ||||||
|       - name: Install poetry & deps |  | ||||||
|         shell: bash |  | ||||||
|         run: | |  | ||||||
|           pipx install poetry || true |  | ||||||
|           sudo apt-get update |  | ||||||
|           sudo apt-get install --no-install-recommends -y libpq-dev openssl libxmlsec1-dev pkg-config gettext |  | ||||||
|       - name: Setup python and restore poetry |  | ||||||
|         uses: actions/setup-python@v5 |  | ||||||
|         with: |  | ||||||
|           python-version-file: "pyproject.toml" |  | ||||||
|           cache: "poetry" |  | ||||||
|       - name: Generate API Client |  | ||||||
|         run: make gen-client-py |  | ||||||
|       - name: Publish package |  | ||||||
|         working-directory: gen-py-api/ |  | ||||||
|         run: | |  | ||||||
|           poetry build |  | ||||||
|       - name: Publish package to PyPI |  | ||||||
|         uses: pypa/gh-action-pypi-publish@release/v1 |  | ||||||
|         with: |  | ||||||
|           packages-dir: gen-py-api/dist/ |  | ||||||
|       # We can't easily upgrade the API client being used due to poetry being poetry |  | ||||||
|       # so we'll have to rely on dependabot |  | ||||||
|       # - name: Upgrade / |  | ||||||
|       #   run: | |  | ||||||
|       #     export VERSION=$(cd gen-py-api && poetry version -s) |  | ||||||
|       #     poetry add "authentik_client=$VERSION" --allow-prereleases --lock |  | ||||||
|       # - uses: peter-evans/create-pull-request@v6 |  | ||||||
|       #   id: cpr |  | ||||||
|       #   with: |  | ||||||
|       #     token: ${{ steps.generate_token.outputs.token }} |  | ||||||
|       #     branch: update-root-api-client |  | ||||||
|       #     commit-message: "root: bump API Client version" |  | ||||||
|       #     title: "root: bump API Client version" |  | ||||||
|       #     body: "root: bump API Client version" |  | ||||||
|       #     delete-branch: true |  | ||||||
|       #     signoff: true |  | ||||||
|       #     # ID from https://api.github.com/users/authentik-automation[bot] |  | ||||||
|       #     author: authentik-automation[bot] <135050075+authentik-automation[bot]@users.noreply.github.com> |  | ||||||
|       # - uses: peter-evans/enable-pull-request-automerge@v3 |  | ||||||
|       #   with: |  | ||||||
|       #     token: ${{ steps.generate_token.outputs.token }} |  | ||||||
|       #     pull-request-number: ${{ steps.cpr.outputs.pull-request-number }} |  | ||||||
|       #     merge-method: squash |  | ||||||
							
								
								
									
										25
									
								
								.github/workflows/ci-main.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										25
									
								
								.github/workflows/ci-main.yml
									
									
									
									
										vendored
									
									
								
							| @ -7,6 +7,8 @@ on: | |||||||
|       - main |       - main | ||||||
|       - next |       - next | ||||||
|       - version-* |       - version-* | ||||||
|  |     paths-ignore: | ||||||
|  |       - website/** | ||||||
|   pull_request: |   pull_request: | ||||||
|     branches: |     branches: | ||||||
|       - main |       - main | ||||||
| @ -26,7 +28,10 @@ jobs: | |||||||
|           - bandit |           - bandit | ||||||
|           - black |           - black | ||||||
|           - codespell |           - codespell | ||||||
|  |           - isort | ||||||
|           - pending-migrations |           - pending-migrations | ||||||
|  |           # - pylint | ||||||
|  |           - pyright | ||||||
|           - ruff |           - ruff | ||||||
|     runs-on: ubuntu-latest |     runs-on: ubuntu-latest | ||||||
|     steps: |     steps: | ||||||
| @ -50,6 +55,7 @@ jobs: | |||||||
|       fail-fast: false |       fail-fast: false | ||||||
|       matrix: |       matrix: | ||||||
|         psql: |         psql: | ||||||
|  |           - 12-alpine | ||||||
|           - 15-alpine |           - 15-alpine | ||||||
|           - 16-alpine |           - 16-alpine | ||||||
|     steps: |     steps: | ||||||
| @ -103,6 +109,7 @@ jobs: | |||||||
|       fail-fast: false |       fail-fast: false | ||||||
|       matrix: |       matrix: | ||||||
|         psql: |         psql: | ||||||
|  |           - 12-alpine | ||||||
|           - 15-alpine |           - 15-alpine | ||||||
|           - 16-alpine |           - 16-alpine | ||||||
|     steps: |     steps: | ||||||
| @ -128,7 +135,7 @@ jobs: | |||||||
|       - name: Setup authentik env |       - name: Setup authentik env | ||||||
|         uses: ./.github/actions/setup |         uses: ./.github/actions/setup | ||||||
|       - name: Create k8s Kind Cluster |       - name: Create k8s Kind Cluster | ||||||
|         uses: helm/kind-action@v1.10.0 |         uses: helm/kind-action@v1.9.0 | ||||||
|       - name: run integration |       - name: run integration | ||||||
|         run: | |         run: | | ||||||
|           poetry run coverage run manage.py test tests/integration |           poetry run coverage run manage.py test tests/integration | ||||||
| @ -158,8 +165,6 @@ jobs: | |||||||
|             glob: tests/e2e/test_provider_ldap* tests/e2e/test_source_ldap* |             glob: tests/e2e/test_provider_ldap* tests/e2e/test_source_ldap* | ||||||
|           - name: radius |           - name: radius | ||||||
|             glob: tests/e2e/test_provider_radius* |             glob: tests/e2e/test_provider_radius* | ||||||
|           - name: scim |  | ||||||
|             glob: tests/e2e/test_source_scim* |  | ||||||
|           - name: flows |           - name: flows | ||||||
|             glob: tests/e2e/test_flows* |             glob: tests/e2e/test_flows* | ||||||
|     steps: |     steps: | ||||||
| @ -168,7 +173,7 @@ jobs: | |||||||
|         uses: ./.github/actions/setup |         uses: ./.github/actions/setup | ||||||
|       - name: Setup e2e env (chrome, etc) |       - name: Setup e2e env (chrome, etc) | ||||||
|         run: | |         run: | | ||||||
|           docker compose -f tests/e2e/docker-compose.yml up -d |           docker-compose -f tests/e2e/docker-compose.yml up -d | ||||||
|       - id: cache-web |       - id: cache-web | ||||||
|         uses: actions/cache@v4 |         uses: actions/cache@v4 | ||||||
|         with: |         with: | ||||||
| @ -214,6 +219,7 @@ jobs: | |||||||
|       # Needed to upload contianer images to ghcr.io |       # Needed to upload contianer images to ghcr.io | ||||||
|       packages: write |       packages: write | ||||||
|     timeout-minutes: 120 |     timeout-minutes: 120 | ||||||
|  |     if: "github.repository == 'goauthentik/authentik'" | ||||||
|     steps: |     steps: | ||||||
|       - uses: actions/checkout@v4 |       - uses: actions/checkout@v4 | ||||||
|         with: |         with: | ||||||
| @ -225,13 +231,10 @@ jobs: | |||||||
|       - name: prepare variables |       - name: prepare variables | ||||||
|         uses: ./.github/actions/docker-push-variables |         uses: ./.github/actions/docker-push-variables | ||||||
|         id: ev |         id: ev | ||||||
|         env: |  | ||||||
|           DOCKER_USERNAME: ${{ secrets.DOCKER_USERNAME }} |  | ||||||
|         with: |         with: | ||||||
|           image-name: ghcr.io/goauthentik/dev-server |           image-name: ghcr.io/goauthentik/dev-server | ||||||
|           image-arch: ${{ matrix.arch }} |           image-arch: ${{ matrix.arch }} | ||||||
|       - name: Login to Container Registry |       - name: Login to Container Registry | ||||||
|         if: ${{ steps.ev.outputs.shouldBuild == 'true' }} |  | ||||||
|         uses: docker/login-action@v3 |         uses: docker/login-action@v3 | ||||||
|         with: |         with: | ||||||
|           registry: ghcr.io |           registry: ghcr.io | ||||||
| @ -247,11 +250,11 @@ jobs: | |||||||
|             GEOIPUPDATE_ACCOUNT_ID=${{ secrets.GEOIPUPDATE_ACCOUNT_ID }} |             GEOIPUPDATE_ACCOUNT_ID=${{ secrets.GEOIPUPDATE_ACCOUNT_ID }} | ||||||
|             GEOIPUPDATE_LICENSE_KEY=${{ secrets.GEOIPUPDATE_LICENSE_KEY }} |             GEOIPUPDATE_LICENSE_KEY=${{ secrets.GEOIPUPDATE_LICENSE_KEY }} | ||||||
|           tags: ${{ steps.ev.outputs.imageTags }} |           tags: ${{ steps.ev.outputs.imageTags }} | ||||||
|           push: ${{ steps.ev.outputs.shouldBuild == 'true' }} |           push: true | ||||||
|           build-args: | |           build-args: | | ||||||
|             GIT_BUILD_HASH=${{ steps.ev.outputs.sha }} |             GIT_BUILD_HASH=${{ steps.ev.outputs.sha }} | ||||||
|           cache-from: type=registry,ref=ghcr.io/goauthentik/dev-server:buildcache |           cache-from: type=gha | ||||||
|           cache-to: type=registry,ref=ghcr.io/goauthentik/dev-server:buildcache,mode=max |           cache-to: type=gha,mode=max | ||||||
|           platforms: linux/${{ matrix.arch }} |           platforms: linux/${{ matrix.arch }} | ||||||
|   pr-comment: |   pr-comment: | ||||||
|     needs: |     needs: | ||||||
| @ -269,8 +272,6 @@ jobs: | |||||||
|       - name: prepare variables |       - name: prepare variables | ||||||
|         uses: ./.github/actions/docker-push-variables |         uses: ./.github/actions/docker-push-variables | ||||||
|         id: ev |         id: ev | ||||||
|         env: |  | ||||||
|           DOCKER_USERNAME: ${{ secrets.DOCKER_USERNAME }} |  | ||||||
|         with: |         with: | ||||||
|           image-name: ghcr.io/goauthentik/dev-server |           image-name: ghcr.io/goauthentik/dev-server | ||||||
|       - name: Comment on PR |       - name: Comment on PR | ||||||
|  | |||||||
							
								
								
									
										12
									
								
								.github/workflows/ci-outpost.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										12
									
								
								.github/workflows/ci-outpost.yml
									
									
									
									
										vendored
									
									
								
							| @ -29,7 +29,7 @@ jobs: | |||||||
|       - name: Generate API |       - name: Generate API | ||||||
|         run: make gen-client-go |         run: make gen-client-go | ||||||
|       - name: golangci-lint |       - name: golangci-lint | ||||||
|         uses: golangci/golangci-lint-action@v6 |         uses: golangci/golangci-lint-action@v4 | ||||||
|         with: |         with: | ||||||
|           version: v1.54.2 |           version: v1.54.2 | ||||||
|           args: --timeout 5000s --verbose |           args: --timeout 5000s --verbose | ||||||
| @ -71,6 +71,7 @@ jobs: | |||||||
|     permissions: |     permissions: | ||||||
|       # Needed to upload contianer images to ghcr.io |       # Needed to upload contianer images to ghcr.io | ||||||
|       packages: write |       packages: write | ||||||
|  |     if: "github.repository == 'goauthentik/authentik'" | ||||||
|     steps: |     steps: | ||||||
|       - uses: actions/checkout@v4 |       - uses: actions/checkout@v4 | ||||||
|         with: |         with: | ||||||
| @ -82,12 +83,9 @@ jobs: | |||||||
|       - name: prepare variables |       - name: prepare variables | ||||||
|         uses: ./.github/actions/docker-push-variables |         uses: ./.github/actions/docker-push-variables | ||||||
|         id: ev |         id: ev | ||||||
|         env: |  | ||||||
|           DOCKER_USERNAME: ${{ secrets.DOCKER_USERNAME }} |  | ||||||
|         with: |         with: | ||||||
|           image-name: ghcr.io/goauthentik/dev-${{ matrix.type }} |           image-name: ghcr.io/goauthentik/dev-${{ matrix.type }} | ||||||
|       - name: Login to Container Registry |       - name: Login to Container Registry | ||||||
|         if: ${{ steps.ev.outputs.shouldBuild == 'true' }} |  | ||||||
|         uses: docker/login-action@v3 |         uses: docker/login-action@v3 | ||||||
|         with: |         with: | ||||||
|           registry: ghcr.io |           registry: ghcr.io | ||||||
| @ -100,13 +98,13 @@ jobs: | |||||||
|         with: |         with: | ||||||
|           tags: ${{ steps.ev.outputs.imageTags }} |           tags: ${{ steps.ev.outputs.imageTags }} | ||||||
|           file: ${{ matrix.type }}.Dockerfile |           file: ${{ matrix.type }}.Dockerfile | ||||||
|           push: ${{ steps.ev.outputs.shouldBuild == 'true' }} |           push: true | ||||||
|           build-args: | |           build-args: | | ||||||
|             GIT_BUILD_HASH=${{ steps.ev.outputs.sha }} |             GIT_BUILD_HASH=${{ steps.ev.outputs.sha }} | ||||||
|           platforms: linux/amd64,linux/arm64 |           platforms: linux/amd64,linux/arm64 | ||||||
|           context: . |           context: . | ||||||
|           cache-from: type=registry,ref=ghcr.io/goauthentik/dev-${{ matrix.type }}:buildcache |           cache-from: type=gha | ||||||
|           cache-to: type=registry,ref=ghcr.io/goauthentik/dev-${{ matrix.type }}:buildcache,mode=max |           cache-to: type=gha,mode=max | ||||||
|   build-binary: |   build-binary: | ||||||
|     timeout-minutes: 120 |     timeout-minutes: 120 | ||||||
|     needs: |     needs: | ||||||
|  | |||||||
							
								
								
									
										8
									
								
								.github/workflows/ci-web.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										8
									
								
								.github/workflows/ci-web.yml
									
									
									
									
										vendored
									
									
								
							| @ -34,13 +34,6 @@ jobs: | |||||||
|       - name: Eslint |       - name: Eslint | ||||||
|         working-directory: ${{ matrix.project }}/ |         working-directory: ${{ matrix.project }}/ | ||||||
|         run: npm run lint |         run: npm run lint | ||||||
|   lint-lockfile: |  | ||||||
|     runs-on: ubuntu-latest |  | ||||||
|     steps: |  | ||||||
|       - uses: actions/checkout@v4 |  | ||||||
|       - working-directory: web/ |  | ||||||
|         run: | |  | ||||||
|           [ -z "$(jq -r '.packages | to_entries[] | select((.key | startswith("node_modules")) and (.value | has("resolved") | not)) | .key' < package-lock.json)" ] |  | ||||||
|   lint-build: |   lint-build: | ||||||
|     runs-on: ubuntu-latest |     runs-on: ubuntu-latest | ||||||
|     steps: |     steps: | ||||||
| @ -102,7 +95,6 @@ jobs: | |||||||
|         run: npm run lit-analyse |         run: npm run lit-analyse | ||||||
|   ci-web-mark: |   ci-web-mark: | ||||||
|     needs: |     needs: | ||||||
|       - lint-lockfile |  | ||||||
|       - lint-eslint |       - lint-eslint | ||||||
|       - lint-prettier |       - lint-prettier | ||||||
|       - lint-lit-analyse |       - lint-lit-analyse | ||||||
|  | |||||||
							
								
								
									
										9
									
								
								.github/workflows/ci-website.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										9
									
								
								.github/workflows/ci-website.yml
									
									
									
									
										vendored
									
									
								
							| @ -12,13 +12,6 @@ on: | |||||||
|       - version-* |       - version-* | ||||||
|  |  | ||||||
| jobs: | jobs: | ||||||
|   lint-lockfile: |  | ||||||
|     runs-on: ubuntu-latest |  | ||||||
|     steps: |  | ||||||
|       - uses: actions/checkout@v4 |  | ||||||
|       - working-directory: website/ |  | ||||||
|         run: | |  | ||||||
|           [ -z "$(jq -r '.packages | to_entries[] | select((.key | startswith("node_modules")) and (.value | has("resolved") | not)) | .key' < package-lock.json)" ] |  | ||||||
|   lint-prettier: |   lint-prettier: | ||||||
|     runs-on: ubuntu-latest |     runs-on: ubuntu-latest | ||||||
|     steps: |     steps: | ||||||
| @ -55,6 +48,7 @@ jobs: | |||||||
|       matrix: |       matrix: | ||||||
|         job: |         job: | ||||||
|           - build |           - build | ||||||
|  |           - build-docs-only | ||||||
|     steps: |     steps: | ||||||
|       - uses: actions/checkout@v4 |       - uses: actions/checkout@v4 | ||||||
|       - uses: actions/setup-node@v4 |       - uses: actions/setup-node@v4 | ||||||
| @ -69,7 +63,6 @@ jobs: | |||||||
|         run: npm run ${{ matrix.job }} |         run: npm run ${{ matrix.job }} | ||||||
|   ci-website-mark: |   ci-website-mark: | ||||||
|     needs: |     needs: | ||||||
|       - lint-lockfile |  | ||||||
|       - lint-prettier |       - lint-prettier | ||||||
|       - test |       - test | ||||||
|       - build |       - build | ||||||
|  | |||||||
							
								
								
									
										43
									
								
								.github/workflows/gen-update-webauthn-mds.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										43
									
								
								.github/workflows/gen-update-webauthn-mds.yml
									
									
									
									
										vendored
									
									
								
							| @ -1,43 +0,0 @@ | |||||||
| name: authentik-gen-update-webauthn-mds |  | ||||||
| on: |  | ||||||
|   workflow_dispatch: |  | ||||||
|   schedule: |  | ||||||
|     - cron: '30 1 1,15 * *' |  | ||||||
|  |  | ||||||
| env: |  | ||||||
|   POSTGRES_DB: authentik |  | ||||||
|   POSTGRES_USER: authentik |  | ||||||
|   POSTGRES_PASSWORD: "EK-5jnKfjrGRm<77" |  | ||||||
|  |  | ||||||
| jobs: |  | ||||||
|   build: |  | ||||||
|     runs-on: ubuntu-latest |  | ||||||
|     steps: |  | ||||||
|       - id: generate_token |  | ||||||
|         uses: tibdex/github-app-token@v2 |  | ||||||
|         with: |  | ||||||
|           app_id: ${{ secrets.GH_APP_ID }} |  | ||||||
|           private_key: ${{ secrets.GH_APP_PRIVATE_KEY }} |  | ||||||
|       - uses: actions/checkout@v4 |  | ||||||
|         with: |  | ||||||
|           token: ${{ steps.generate_token.outputs.token }} |  | ||||||
|       - name: Setup authentik env |  | ||||||
|         uses: ./.github/actions/setup |  | ||||||
|       - run: poetry run ak update_webauthn_mds |  | ||||||
|       - uses: peter-evans/create-pull-request@v6 |  | ||||||
|         id: cpr |  | ||||||
|         with: |  | ||||||
|           token: ${{ steps.generate_token.outputs.token }} |  | ||||||
|           branch: update-fido-mds-client |  | ||||||
|           commit-message: "stages/authenticator_webauthn: Update FIDO MDS3 & Passkey aaguid blobs" |  | ||||||
|           title: "stages/authenticator_webauthn: Update FIDO MDS3 & Passkey aaguid blobs" |  | ||||||
|           body: "stages/authenticator_webauthn: Update FIDO MDS3 & Passkey aaguid blobs" |  | ||||||
|           delete-branch: true |  | ||||||
|           signoff: true |  | ||||||
|           # ID from https://api.github.com/users/authentik-automation[bot] |  | ||||||
|           author: authentik-automation[bot] <135050075+authentik-automation[bot]@users.noreply.github.com> |  | ||||||
|       - uses: peter-evans/enable-pull-request-automerge@v3 |  | ||||||
|         with: |  | ||||||
|           token: ${{ steps.generate_token.outputs.token }} |  | ||||||
|           pull-request-number: ${{ steps.cpr.outputs.pull-request-number }} |  | ||||||
|           merge-method: squash |  | ||||||
							
								
								
									
										18
									
								
								.github/workflows/release-publish.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										18
									
								
								.github/workflows/release-publish.yml
									
									
									
									
										vendored
									
									
								
							| @ -20,8 +20,6 @@ jobs: | |||||||
|       - name: prepare variables |       - name: prepare variables | ||||||
|         uses: ./.github/actions/docker-push-variables |         uses: ./.github/actions/docker-push-variables | ||||||
|         id: ev |         id: ev | ||||||
|         env: |  | ||||||
|           DOCKER_USERNAME: ${{ secrets.DOCKER_USERNAME }} |  | ||||||
|         with: |         with: | ||||||
|           image-name: ghcr.io/goauthentik/server,beryju/authentik |           image-name: ghcr.io/goauthentik/server,beryju/authentik | ||||||
|       - name: Docker Login Registry |       - name: Docker Login Registry | ||||||
| @ -74,8 +72,6 @@ jobs: | |||||||
|       - name: prepare variables |       - name: prepare variables | ||||||
|         uses: ./.github/actions/docker-push-variables |         uses: ./.github/actions/docker-push-variables | ||||||
|         id: ev |         id: ev | ||||||
|         env: |  | ||||||
|           DOCKER_USERNAME: ${{ secrets.DOCKER_USERNAME }} |  | ||||||
|         with: |         with: | ||||||
|           image-name: ghcr.io/goauthentik/${{ matrix.type }},beryju/authentik-${{ matrix.type }} |           image-name: ghcr.io/goauthentik/${{ matrix.type }},beryju/authentik-${{ matrix.type }} | ||||||
|       - name: make empty clients |       - name: make empty clients | ||||||
| @ -155,12 +151,12 @@ jobs: | |||||||
|       - uses: actions/checkout@v4 |       - uses: actions/checkout@v4 | ||||||
|       - name: Run test suite in final docker images |       - name: Run test suite in final docker images | ||||||
|         run: | |         run: | | ||||||
|           echo "PG_PASS=$(openssl rand 32 | base64)" >> .env |           echo "PG_PASS=$(openssl rand -base64 32)" >> .env | ||||||
|           echo "AUTHENTIK_SECRET_KEY=$(openssl rand 32 | base64)" >> .env |           echo "AUTHENTIK_SECRET_KEY=$(openssl rand -base64 32)" >> .env | ||||||
|           docker compose pull -q |           docker-compose pull -q | ||||||
|           docker compose up --no-start |           docker-compose up --no-start | ||||||
|           docker compose start postgresql redis |           docker-compose start postgresql redis | ||||||
|           docker compose run -u root server test-all |           docker-compose run -u root server test-all | ||||||
|   sentry-release: |   sentry-release: | ||||||
|     needs: |     needs: | ||||||
|       - build-server |       - build-server | ||||||
| @ -172,8 +168,6 @@ jobs: | |||||||
|       - name: prepare variables |       - name: prepare variables | ||||||
|         uses: ./.github/actions/docker-push-variables |         uses: ./.github/actions/docker-push-variables | ||||||
|         id: ev |         id: ev | ||||||
|         env: |  | ||||||
|           DOCKER_USERNAME: ${{ secrets.DOCKER_USERNAME }} |  | ||||||
|         with: |         with: | ||||||
|           image-name: ghcr.io/goauthentik/server |           image-name: ghcr.io/goauthentik/server | ||||||
|       - name: Get static files from docker image |       - name: Get static files from docker image | ||||||
|  | |||||||
							
								
								
									
										12
									
								
								.github/workflows/release-tag.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										12
									
								
								.github/workflows/release-tag.yml
									
									
									
									
										vendored
									
									
								
							| @ -14,16 +14,16 @@ jobs: | |||||||
|       - uses: actions/checkout@v4 |       - uses: actions/checkout@v4 | ||||||
|       - name: Pre-release test |       - name: Pre-release test | ||||||
|         run: | |         run: | | ||||||
|           echo "PG_PASS=$(openssl rand 32 | base64)" >> .env |           echo "PG_PASS=$(openssl rand -base64 32)" >> .env | ||||||
|           echo "AUTHENTIK_SECRET_KEY=$(openssl rand 32 | base64)" >> .env |           echo "AUTHENTIK_SECRET_KEY=$(openssl rand -base64 32)" >> .env | ||||||
|           docker buildx install |           docker buildx install | ||||||
|           mkdir -p ./gen-ts-api |           mkdir -p ./gen-ts-api | ||||||
|           docker build -t testing:latest . |           docker build -t testing:latest . | ||||||
|           echo "AUTHENTIK_IMAGE=testing" >> .env |           echo "AUTHENTIK_IMAGE=testing" >> .env | ||||||
|           echo "AUTHENTIK_TAG=latest" >> .env |           echo "AUTHENTIK_TAG=latest" >> .env | ||||||
|           docker compose up --no-start |           docker-compose up --no-start | ||||||
|           docker compose start postgresql redis |           docker-compose start postgresql redis | ||||||
|           docker compose run -u root server test-all |           docker-compose run -u root server test-all | ||||||
|       - id: generate_token |       - id: generate_token | ||||||
|         uses: tibdex/github-app-token@v2 |         uses: tibdex/github-app-token@v2 | ||||||
|         with: |         with: | ||||||
| @ -32,8 +32,6 @@ jobs: | |||||||
|       - name: prepare variables |       - name: prepare variables | ||||||
|         uses: ./.github/actions/docker-push-variables |         uses: ./.github/actions/docker-push-variables | ||||||
|         id: ev |         id: ev | ||||||
|         env: |  | ||||||
|           DOCKER_USERNAME: ${{ secrets.DOCKER_USERNAME }} |  | ||||||
|         with: |         with: | ||||||
|           image-name: ghcr.io/goauthentik/server |           image-name: ghcr.io/goauthentik/server | ||||||
|       - name: Create Release |       - name: Create Release | ||||||
|  | |||||||
							
								
								
									
										2
									
								
								.github/workflows/repo-stale.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/workflows/repo-stale.yml
									
									
									
									
										vendored
									
									
								
							| @ -23,7 +23,7 @@ jobs: | |||||||
|           repo-token: ${{ steps.generate_token.outputs.token }} |           repo-token: ${{ steps.generate_token.outputs.token }} | ||||||
|           days-before-stale: 60 |           days-before-stale: 60 | ||||||
|           days-before-close: 7 |           days-before-close: 7 | ||||||
|           exempt-issue-labels: pinned,security,pr_wanted,enhancement,bug/confirmed,enhancement/confirmed,question,status/reviewing |           exempt-issue-labels: pinned,security,pr_wanted,enhancement,bug/confirmed,enhancement/confirmed,question | ||||||
|           stale-issue-label: wontfix |           stale-issue-label: wontfix | ||||||
|           stale-issue-message: > |           stale-issue-message: > | ||||||
|             This issue has been automatically marked as stale because it has not had |             This issue has been automatically marked as stale because it has not had | ||||||
|  | |||||||
| @ -1,4 +1,4 @@ | |||||||
| name: authentik-api-ts-publish | name: authentik-web-api-publish | ||||||
| on: | on: | ||||||
|   push: |   push: | ||||||
|     branches: [main] |     branches: [main] | ||||||
							
								
								
									
										3
									
								
								.vscode/extensions.json
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										3
									
								
								.vscode/extensions.json
									
									
									
									
										vendored
									
									
								
							| @ -10,7 +10,8 @@ | |||||||
|         "Gruntfuggly.todo-tree", |         "Gruntfuggly.todo-tree", | ||||||
|         "mechatroner.rainbow-csv", |         "mechatroner.rainbow-csv", | ||||||
|         "ms-python.black-formatter", |         "ms-python.black-formatter", | ||||||
|         "charliermarsh.ruff", |         "ms-python.isort", | ||||||
|  |         "ms-python.pylint", | ||||||
|         "ms-python.python", |         "ms-python.python", | ||||||
|         "ms-python.vscode-pylance", |         "ms-python.vscode-pylance", | ||||||
|         "ms-python.black-formatter", |         "ms-python.black-formatter", | ||||||
|  | |||||||
							
								
								
									
										13
									
								
								.vscode/settings.json
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										13
									
								
								.vscode/settings.json
									
									
									
									
										vendored
									
									
								
							| @ -4,21 +4,20 @@ | |||||||
|         "asgi", |         "asgi", | ||||||
|         "authentik", |         "authentik", | ||||||
|         "authn", |         "authn", | ||||||
|         "entra", |  | ||||||
|         "goauthentik", |         "goauthentik", | ||||||
|         "jwks", |         "jwks", | ||||||
|         "kubernetes", |  | ||||||
|         "oidc", |         "oidc", | ||||||
|         "openid", |         "openid", | ||||||
|         "passwordless", |  | ||||||
|         "plex", |         "plex", | ||||||
|         "saml", |         "saml", | ||||||
|         "scim", |  | ||||||
|         "slo", |  | ||||||
|         "sso", |  | ||||||
|         "totp", |         "totp", | ||||||
|         "traefik", |  | ||||||
|         "webauthn", |         "webauthn", | ||||||
|  |         "traefik", | ||||||
|  |         "passwordless", | ||||||
|  |         "kubernetes", | ||||||
|  |         "sso", | ||||||
|  |         "slo", | ||||||
|  |         "scim", | ||||||
|     ], |     ], | ||||||
|     "todo-tree.tree.showCountsInTree": true, |     "todo-tree.tree.showCountsInTree": true, | ||||||
|     "todo-tree.tree.showBadges": true, |     "todo-tree.tree.showBadges": true, | ||||||
|  | |||||||
							
								
								
									
										41
									
								
								Dockerfile
									
									
									
									
									
								
							
							
						
						
									
										41
									
								
								Dockerfile
									
									
									
									
									
								
							| @ -1,7 +1,7 @@ | |||||||
| # syntax=docker/dockerfile:1 | # syntax=docker/dockerfile:1 | ||||||
|  |  | ||||||
| # Stage 1: Build website | # Stage 1: Build website | ||||||
| FROM --platform=${BUILDPLATFORM} docker.io/node:22 as website-builder | FROM --platform=${BUILDPLATFORM} docker.io/node:21 as website-builder | ||||||
|  |  | ||||||
| ENV NODE_ENV=production | ENV NODE_ENV=production | ||||||
|  |  | ||||||
| @ -14,13 +14,12 @@ RUN --mount=type=bind,target=/work/website/package.json,src=./website/package.js | |||||||
|  |  | ||||||
| COPY ./website /work/website/ | COPY ./website /work/website/ | ||||||
| COPY ./blueprints /work/blueprints/ | COPY ./blueprints /work/blueprints/ | ||||||
| COPY ./schema.yml /work/ |  | ||||||
| COPY ./SECURITY.md /work/ | COPY ./SECURITY.md /work/ | ||||||
|  |  | ||||||
| RUN npm run build-bundled | RUN npm run build-docs-only | ||||||
|  |  | ||||||
| # Stage 2: Build webui | # Stage 2: Build webui | ||||||
| FROM --platform=${BUILDPLATFORM} docker.io/node:22 as web-builder | FROM --platform=${BUILDPLATFORM} docker.io/node:21 as web-builder | ||||||
|  |  | ||||||
| ENV NODE_ENV=production | ENV NODE_ENV=production | ||||||
|  |  | ||||||
| @ -28,6 +27,7 @@ WORKDIR /work/web | |||||||
|  |  | ||||||
| RUN --mount=type=bind,target=/work/web/package.json,src=./web/package.json \ | RUN --mount=type=bind,target=/work/web/package.json,src=./web/package.json \ | ||||||
|     --mount=type=bind,target=/work/web/package-lock.json,src=./web/package-lock.json \ |     --mount=type=bind,target=/work/web/package-lock.json,src=./web/package-lock.json \ | ||||||
|  |     --mount=type=bind,target=/work/.git,src=./.git,readonly \ | ||||||
|     --mount=type=cache,id=npm-web,sharing=shared,target=/root/.npm \ |     --mount=type=cache,id=npm-web,sharing=shared,target=/root/.npm \ | ||||||
|     npm ci --include=dev |     npm ci --include=dev | ||||||
|  |  | ||||||
| @ -38,7 +38,7 @@ COPY ./gen-ts-api /work/web/node_modules/@goauthentik/api | |||||||
| RUN npm run build | RUN npm run build | ||||||
|  |  | ||||||
| # Stage 3: Build go proxy | # Stage 3: Build go proxy | ||||||
| FROM --platform=${BUILDPLATFORM} mcr.microsoft.com/oss/go/microsoft/golang:1.22-fips-bookworm AS go-builder | FROM --platform=${BUILDPLATFORM} docker.io/golang:1.22.0-bookworm AS go-builder | ||||||
|  |  | ||||||
| ARG TARGETOS | ARG TARGETOS | ||||||
| ARG TARGETARCH | ARG TARGETARCH | ||||||
| @ -49,11 +49,6 @@ ARG GOARCH=$TARGETARCH | |||||||
|  |  | ||||||
| WORKDIR /go/src/goauthentik.io | WORKDIR /go/src/goauthentik.io | ||||||
|  |  | ||||||
| RUN --mount=type=cache,id=apt-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/var/cache/apt \ |  | ||||||
|     dpkg --add-architecture arm64 && \ |  | ||||||
|     apt-get update && \ |  | ||||||
|     apt-get install -y --no-install-recommends crossbuild-essential-arm64 gcc-aarch64-linux-gnu |  | ||||||
|  |  | ||||||
| RUN --mount=type=bind,target=/go/src/goauthentik.io/go.mod,src=./go.mod \ | RUN --mount=type=bind,target=/go/src/goauthentik.io/go.mod,src=./go.mod \ | ||||||
|     --mount=type=bind,target=/go/src/goauthentik.io/go.sum,src=./go.sum \ |     --mount=type=bind,target=/go/src/goauthentik.io/go.sum,src=./go.sum \ | ||||||
|     --mount=type=cache,target=/go/pkg/mod \ |     --mount=type=cache,target=/go/pkg/mod \ | ||||||
| @ -68,17 +63,17 @@ COPY ./internal /go/src/goauthentik.io/internal | |||||||
| COPY ./go.mod /go/src/goauthentik.io/go.mod | COPY ./go.mod /go/src/goauthentik.io/go.mod | ||||||
| COPY ./go.sum /go/src/goauthentik.io/go.sum | COPY ./go.sum /go/src/goauthentik.io/go.sum | ||||||
|  |  | ||||||
|  | ENV CGO_ENABLED=0 | ||||||
|  |  | ||||||
| RUN --mount=type=cache,sharing=locked,target=/go/pkg/mod \ | RUN --mount=type=cache,sharing=locked,target=/go/pkg/mod \ | ||||||
|     --mount=type=cache,id=go-build-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/root/.cache/go-build \ |     --mount=type=cache,id=go-build-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/root/.cache/go-build \ | ||||||
|     if [ "$TARGETARCH" = "arm64" ]; then export CC=aarch64-linux-gnu-gcc && export CC_FOR_TARGET=gcc-aarch64-linux-gnu; fi && \ |     GOARM="${TARGETVARIANT#v}" go build -o /go/authentik ./cmd/server | ||||||
|     CGO_ENABLED=1 GOEXPERIMENT="systemcrypto" GOFLAGS="-tags=requirefips" GOARM="${TARGETVARIANT#v}" \ |  | ||||||
|     go build -o /go/authentik ./cmd/server |  | ||||||
|  |  | ||||||
| # Stage 4: MaxMind GeoIP | # Stage 4: MaxMind GeoIP | ||||||
| FROM --platform=${BUILDPLATFORM} ghcr.io/maxmind/geoipupdate:v7.0.1 as geoip | FROM --platform=${BUILDPLATFORM} ghcr.io/maxmind/geoipupdate:v6.1 as geoip | ||||||
|  |  | ||||||
| ENV GEOIPUPDATE_EDITION_IDS="GeoLite2-City GeoLite2-ASN" | ENV GEOIPUPDATE_EDITION_IDS="GeoLite2-City GeoLite2-ASN" | ||||||
| ENV GEOIPUPDATE_VERBOSE="1" | ENV GEOIPUPDATE_VERBOSE="true" | ||||||
| ENV GEOIPUPDATE_ACCOUNT_ID_FILE="/run/secrets/GEOIPUPDATE_ACCOUNT_ID" | ENV GEOIPUPDATE_ACCOUNT_ID_FILE="/run/secrets/GEOIPUPDATE_ACCOUNT_ID" | ||||||
| ENV GEOIPUPDATE_LICENSE_KEY_FILE="/run/secrets/GEOIPUPDATE_LICENSE_KEY" | ENV GEOIPUPDATE_LICENSE_KEY_FILE="/run/secrets/GEOIPUPDATE_LICENSE_KEY" | ||||||
|  |  | ||||||
| @ -89,7 +84,7 @@ RUN --mount=type=secret,id=GEOIPUPDATE_ACCOUNT_ID \ | |||||||
|     /bin/sh -c "/usr/bin/entry.sh || echo 'Failed to get GeoIP database, disabling'; exit 0" |     /bin/sh -c "/usr/bin/entry.sh || echo 'Failed to get GeoIP database, disabling'; exit 0" | ||||||
|  |  | ||||||
| # Stage 5: Python dependencies | # Stage 5: Python dependencies | ||||||
| FROM ghcr.io/goauthentik/fips-python:3.12.3-slim-bookworm-fips-full AS python-deps | FROM docker.io/python:3.12.2-slim-bookworm AS python-deps | ||||||
|  |  | ||||||
| WORKDIR /ak-root/poetry | WORKDIR /ak-root/poetry | ||||||
|  |  | ||||||
| @ -102,21 +97,19 @@ RUN rm -f /etc/apt/apt.conf.d/docker-clean; echo 'Binary::apt::APT::Keep-Downloa | |||||||
| RUN --mount=type=cache,id=apt-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/var/cache/apt \ | RUN --mount=type=cache,id=apt-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/var/cache/apt \ | ||||||
|     apt-get update && \ |     apt-get update && \ | ||||||
|     # Required for installing pip packages |     # Required for installing pip packages | ||||||
|     apt-get install -y --no-install-recommends build-essential pkg-config libpq-dev |     apt-get install -y --no-install-recommends build-essential pkg-config libxmlsec1-dev zlib1g-dev libpq-dev | ||||||
|  |  | ||||||
| RUN --mount=type=bind,target=./pyproject.toml,src=./pyproject.toml \ | RUN --mount=type=bind,target=./pyproject.toml,src=./pyproject.toml \ | ||||||
|     --mount=type=bind,target=./poetry.lock,src=./poetry.lock \ |     --mount=type=bind,target=./poetry.lock,src=./poetry.lock \ | ||||||
|     --mount=type=cache,target=/root/.cache/pip \ |     --mount=type=cache,target=/root/.cache/pip \ | ||||||
|     --mount=type=cache,target=/root/.cache/pypoetry \ |     --mount=type=cache,target=/root/.cache/pypoetry \ | ||||||
|     python -m venv /ak-root/venv/ && \ |     python -m venv /ak-root/venv/ && \ | ||||||
|     bash -c "source ${VENV_PATH}/bin/activate && \ |  | ||||||
|     pip3 install --upgrade pip && \ |     pip3 install --upgrade pip && \ | ||||||
|     pip3 install poetry && \ |     pip3 install poetry && \ | ||||||
|     poetry install --only=main --no-ansi --no-interaction --no-root && \ |     poetry install --only=main --no-ansi --no-interaction | ||||||
|     pip install --force-reinstall /wheels/*" |  | ||||||
|  |  | ||||||
| # Stage 6: Run | # Stage 6: Run | ||||||
| FROM ghcr.io/goauthentik/fips-python:3.12.3-slim-bookworm-fips-full AS final-image | FROM docker.io/python:3.12.2-slim-bookworm AS final-image | ||||||
|  |  | ||||||
| ARG GIT_BUILD_HASH | ARG GIT_BUILD_HASH | ||||||
| ARG VERSION | ARG VERSION | ||||||
| @ -133,7 +126,7 @@ WORKDIR / | |||||||
| # We cannot cache this layer otherwise we'll end up with a bigger image | # We cannot cache this layer otherwise we'll end up with a bigger image | ||||||
| RUN apt-get update && \ | RUN apt-get update && \ | ||||||
|     # Required for runtime |     # Required for runtime | ||||||
|     apt-get install -y --no-install-recommends libpq5 libmaxminddb0 ca-certificates && \ |     apt-get install -y --no-install-recommends libpq5 openssl libxmlsec1-openssl libmaxminddb0 ca-certificates && \ | ||||||
|     # Required for bootstrap & healtcheck |     # Required for bootstrap & healtcheck | ||||||
|     apt-get install -y --no-install-recommends runit && \ |     apt-get install -y --no-install-recommends runit && \ | ||||||
|     apt-get clean && \ |     apt-get clean && \ | ||||||
| @ -157,7 +150,7 @@ COPY --from=go-builder /go/authentik /bin/authentik | |||||||
| COPY --from=python-deps /ak-root/venv /ak-root/venv | COPY --from=python-deps /ak-root/venv /ak-root/venv | ||||||
| COPY --from=web-builder /work/web/dist/ /web/dist/ | COPY --from=web-builder /work/web/dist/ /web/dist/ | ||||||
| COPY --from=web-builder /work/web/authentik/ /web/authentik/ | COPY --from=web-builder /work/web/authentik/ /web/authentik/ | ||||||
| COPY --from=website-builder /work/website/build/ /website/help/ | COPY --from=website-builder /work/website/help/ /website/help/ | ||||||
| COPY --from=geoip /usr/share/GeoIP /geoip | COPY --from=geoip /usr/share/GeoIP /geoip | ||||||
|  |  | ||||||
| USER 1000 | USER 1000 | ||||||
| @ -169,8 +162,6 @@ ENV TMPDIR=/dev/shm/ \ | |||||||
|     VENV_PATH="/ak-root/venv" \ |     VENV_PATH="/ak-root/venv" \ | ||||||
|     POETRY_VIRTUALENVS_CREATE=false |     POETRY_VIRTUALENVS_CREATE=false | ||||||
|  |  | ||||||
| ENV GOFIPS=1 |  | ||||||
|  |  | ||||||
| HEALTHCHECK --interval=30s --timeout=30s --start-period=60s --retries=3 CMD [ "ak", "healthcheck" ] | HEALTHCHECK --interval=30s --timeout=30s --start-period=60s --retries=3 CMD [ "ak", "healthcheck" ] | ||||||
|  |  | ||||||
| ENTRYPOINT [ "dumb-init", "--", "ak" ] | ENTRYPOINT [ "dumb-init", "--", "ak" ] | ||||||
|  | |||||||
							
								
								
									
										52
									
								
								Makefile
									
									
									
									
									
								
							
							
						
						
									
										52
									
								
								Makefile
									
									
									
									
									
								
							| @ -5,11 +5,10 @@ PWD = $(shell pwd) | |||||||
| UID = $(shell id -u) | UID = $(shell id -u) | ||||||
| GID = $(shell id -g) | GID = $(shell id -g) | ||||||
| NPM_VERSION = $(shell python -m scripts.npm_version) | NPM_VERSION = $(shell python -m scripts.npm_version) | ||||||
| PY_SOURCES = authentik tests scripts lifecycle .github | PY_SOURCES = authentik tests scripts lifecycle | ||||||
| DOCKER_IMAGE ?= "authentik:test" | DOCKER_IMAGE ?= "authentik:test" | ||||||
|  |  | ||||||
| GEN_API_TS = "gen-ts-api" | GEN_API_TS = "gen-ts-api" | ||||||
| GEN_API_PY = "gen-py-api" |  | ||||||
| GEN_API_GO = "gen-go-api" | GEN_API_GO = "gen-go-api" | ||||||
|  |  | ||||||
| pg_user := $(shell python -m authentik.lib.config postgresql.user 2>/dev/null) | pg_user := $(shell python -m authentik.lib.config postgresql.user 2>/dev/null) | ||||||
| @ -19,7 +18,6 @@ pg_name := $(shell python -m authentik.lib.config postgresql.name 2>/dev/null) | |||||||
| CODESPELL_ARGS = -D - -D .github/codespell-dictionary.txt \ | CODESPELL_ARGS = -D - -D .github/codespell-dictionary.txt \ | ||||||
| 		-I .github/codespell-words.txt \ | 		-I .github/codespell-words.txt \ | ||||||
| 		-S 'web/src/locales/**' \ | 		-S 'web/src/locales/**' \ | ||||||
| 		-S 'website/developer-docs/api/reference/**' \ |  | ||||||
| 		authentik \ | 		authentik \ | ||||||
| 		internal \ | 		internal \ | ||||||
| 		cmd \ | 		cmd \ | ||||||
| @ -47,12 +45,12 @@ test-go: | |||||||
| 	go test -timeout 0 -v -race -cover ./... | 	go test -timeout 0 -v -race -cover ./... | ||||||
|  |  | ||||||
| test-docker:  ## Run all tests in a docker-compose | test-docker:  ## Run all tests in a docker-compose | ||||||
| 	echo "PG_PASS=$(shell openssl rand 32 | base64)" >> .env | 	echo "PG_PASS=$(openssl rand -base64 32)" >> .env | ||||||
| 	echo "AUTHENTIK_SECRET_KEY=$(shell openssl rand 32 | base64)" >> .env | 	echo "AUTHENTIK_SECRET_KEY=$(openssl rand -base64 32)" >> .env | ||||||
| 	docker compose pull -q | 	docker-compose pull -q | ||||||
| 	docker compose up --no-start | 	docker-compose up --no-start | ||||||
| 	docker compose start postgresql redis | 	docker-compose start postgresql redis | ||||||
| 	docker compose run -u root server test-all | 	docker-compose run -u root server test-all | ||||||
| 	rm -f .env | 	rm -f .env | ||||||
|  |  | ||||||
| test: ## Run the server tests and produce a coverage report (locally) | test: ## Run the server tests and produce a coverage report (locally) | ||||||
| @ -61,12 +59,15 @@ test: ## Run the server tests and produce a coverage report (locally) | |||||||
| 	coverage report | 	coverage report | ||||||
|  |  | ||||||
| lint-fix:  ## Lint and automatically fix errors in the python source code. Reports spelling errors. | lint-fix:  ## Lint and automatically fix errors in the python source code. Reports spelling errors. | ||||||
|  | 	isort $(PY_SOURCES) | ||||||
| 	black $(PY_SOURCES) | 	black $(PY_SOURCES) | ||||||
| 	ruff check --fix $(PY_SOURCES) | 	ruff --fix $(PY_SOURCES) | ||||||
| 	codespell -w $(CODESPELL_ARGS) | 	codespell -w $(CODESPELL_ARGS) | ||||||
|  |  | ||||||
| lint: ## Lint the python and golang sources | lint: ## Lint the python and golang sources | ||||||
| 	bandit -r $(PY_SOURCES) -x web/node_modules -x tests/wdio/node_modules -x website/node_modules | 	bandit -r $(PY_SOURCES) -x node_modules | ||||||
|  | 	./web/node_modules/.bin/pyright $(PY_SOURCES) | ||||||
|  | 	pylint $(PY_SOURCES) | ||||||
| 	golangci-lint run -v | 	golangci-lint run -v | ||||||
|  |  | ||||||
| core-install: | core-install: | ||||||
| @ -139,10 +140,7 @@ gen-clean-ts:  ## Remove generated API client for Typescript | |||||||
| gen-clean-go:  ## Remove generated API client for Go | gen-clean-go:  ## Remove generated API client for Go | ||||||
| 	rm -rf ./${GEN_API_GO}/ | 	rm -rf ./${GEN_API_GO}/ | ||||||
|  |  | ||||||
| gen-clean-py:  ## Remove generated API client for Python | gen-clean: gen-clean-ts gen-clean-go  ## Remove generated API clients | ||||||
| 	rm -rf ./${GEN_API_PY}/ |  | ||||||
|  |  | ||||||
| gen-clean: gen-clean-ts gen-clean-go gen-clean-py  ## Remove generated API clients |  | ||||||
|  |  | ||||||
| gen-client-ts: gen-clean-ts  ## Build and install the authentik API for Typescript into the authentik UI Application | gen-client-ts: gen-clean-ts  ## Build and install the authentik API for Typescript into the authentik UI Application | ||||||
| 	docker run \ | 	docker run \ | ||||||
| @ -160,20 +158,6 @@ gen-client-ts: gen-clean-ts  ## Build and install the authentik API for Typescri | |||||||
| 	cd ./${GEN_API_TS} && npm i | 	cd ./${GEN_API_TS} && npm i | ||||||
| 	\cp -rf ./${GEN_API_TS}/* web/node_modules/@goauthentik/api | 	\cp -rf ./${GEN_API_TS}/* web/node_modules/@goauthentik/api | ||||||
|  |  | ||||||
| gen-client-py: gen-clean-py ## Build and install the authentik API for Python |  | ||||||
| 	docker run \ |  | ||||||
| 		--rm -v ${PWD}:/local \ |  | ||||||
| 		--user ${UID}:${GID} \ |  | ||||||
| 		docker.io/openapitools/openapi-generator-cli:v7.4.0 generate \ |  | ||||||
| 		-i /local/schema.yml \ |  | ||||||
| 		-g python \ |  | ||||||
| 		-o /local/${GEN_API_PY} \ |  | ||||||
| 		-c /local/scripts/api-py-config.yaml \ |  | ||||||
| 		--additional-properties=packageVersion=${NPM_VERSION} \ |  | ||||||
| 		--git-repo-id authentik \ |  | ||||||
| 		--git-user-id goauthentik |  | ||||||
| 	pip install ./${GEN_API_PY} |  | ||||||
|  |  | ||||||
| gen-client-go: gen-clean-go  ## Build and install the authentik API for Golang | gen-client-go: gen-clean-go  ## Build and install the authentik API for Golang | ||||||
| 	mkdir -p ./${GEN_API_GO} ./${GEN_API_GO}/templates | 	mkdir -p ./${GEN_API_GO} ./${GEN_API_GO}/templates | ||||||
| 	wget https://raw.githubusercontent.com/goauthentik/client-go/main/config.yaml -O ./${GEN_API_GO}/config.yaml | 	wget https://raw.githubusercontent.com/goauthentik/client-go/main/config.yaml -O ./${GEN_API_GO}/config.yaml | ||||||
| @ -253,7 +237,6 @@ website-watch:  ## Build and watch the documentation website, updating automatic | |||||||
| ######################### | ######################### | ||||||
|  |  | ||||||
| docker:  ## Build a docker image of the current source tree | docker:  ## Build a docker image of the current source tree | ||||||
| 	mkdir -p ${GEN_API_TS} |  | ||||||
| 	DOCKER_BUILDKIT=1 docker build . --progress plain --tag ${DOCKER_IMAGE} | 	DOCKER_BUILDKIT=1 docker build . --progress plain --tag ${DOCKER_IMAGE} | ||||||
|  |  | ||||||
| ######################### | ######################### | ||||||
| @ -266,6 +249,9 @@ ci--meta-debug: | |||||||
| 	python -V | 	python -V | ||||||
| 	node --version | 	node --version | ||||||
|  |  | ||||||
|  | ci-pylint: ci--meta-debug | ||||||
|  | 	pylint $(PY_SOURCES) | ||||||
|  |  | ||||||
| ci-black: ci--meta-debug | ci-black: ci--meta-debug | ||||||
| 	black --check $(PY_SOURCES) | 	black --check $(PY_SOURCES) | ||||||
|  |  | ||||||
| @ -275,8 +261,14 @@ ci-ruff: ci--meta-debug | |||||||
| ci-codespell: ci--meta-debug | ci-codespell: ci--meta-debug | ||||||
| 	codespell $(CODESPELL_ARGS) -s | 	codespell $(CODESPELL_ARGS) -s | ||||||
|  |  | ||||||
|  | ci-isort: ci--meta-debug | ||||||
|  | 	isort --check $(PY_SOURCES) | ||||||
|  |  | ||||||
| ci-bandit: ci--meta-debug | ci-bandit: ci--meta-debug | ||||||
| 	bandit -r $(PY_SOURCES) | 	bandit -r $(PY_SOURCES) | ||||||
|  |  | ||||||
|  | ci-pyright: ci--meta-debug | ||||||
|  | 	./web/node_modules/.bin/pyright $(PY_SOURCES) | ||||||
|  |  | ||||||
| ci-pending-migrations: ci--meta-debug | ci-pending-migrations: ci--meta-debug | ||||||
| 	ak makemigrations --check | 	ak makemigrations --check | ||||||
|  | |||||||
| @ -26,9 +26,9 @@ For bigger setups, there is a Helm Chart [here](https://github.com/goauthentik/h | |||||||
| ## Screenshots | ## Screenshots | ||||||
|  |  | ||||||
| | Light                                                  | Dark                                                  | | | Light                                                  | Dark                                                  | | ||||||
| | ----------------------------------------------------------- | ---------------------------------------------------------- | | | ------------------------------------------------------ | ----------------------------------------------------- | | ||||||
| |   |   | | |   |   | | ||||||
| |  |  | | |  |  | | ||||||
|  |  | ||||||
| ## Development | ## Development | ||||||
|  |  | ||||||
|  | |||||||
| @ -19,9 +19,9 @@ Even if the issue is not a CVE, we still greatly appreciate your help in hardeni | |||||||
| (.x being the latest patch release for each version) | (.x being the latest patch release for each version) | ||||||
|  |  | ||||||
| | Version | Supported | | | Version | Supported | | ||||||
| | --------- | --------- | | | --- | --- | | ||||||
| | 2023.10.x | ✅        | | | 2023.6.x | ✅ | | ||||||
| | 2024.2.x  | ✅        | | | 2023.8.x | ✅ | | ||||||
|  |  | ||||||
| ## Reporting a Vulnerability | ## Reporting a Vulnerability | ||||||
|  |  | ||||||
| @ -32,7 +32,7 @@ To report a vulnerability, send an email to [security@goauthentik.io](mailto:se | |||||||
| authentik reserves the right to reclassify CVSS as necessary. To determine severity, we will use the CVSS calculator from NVD (https://nvd.nist.gov/vuln-metrics/cvss/v3-calculator). The calculated CVSS score will then be translated into one of the following categories: | authentik reserves the right to reclassify CVSS as necessary. To determine severity, we will use the CVSS calculator from NVD (https://nvd.nist.gov/vuln-metrics/cvss/v3-calculator). The calculated CVSS score will then be translated into one of the following categories: | ||||||
|  |  | ||||||
| | Score | Severity | | | Score | Severity | | ||||||
| | ---------- | -------- | | | --- | --- | | ||||||
| | 0.0 | None | | | 0.0 | None | | ||||||
| | 0.1 – 3.9 | Low | | | 0.1 – 3.9 | Low | | ||||||
| | 4.0 – 6.9 | Medium | | | 4.0 – 6.9 | Medium | | ||||||
|  | |||||||
| @ -1,12 +1,13 @@ | |||||||
| """authentik root module""" | """authentik root module""" | ||||||
|  |  | ||||||
| from os import environ | from os import environ | ||||||
|  | from typing import Optional | ||||||
|  |  | ||||||
| __version__ = "2024.6.0" | __version__ = "2023.10.7" | ||||||
| ENV_GIT_HASH_KEY = "GIT_BUILD_HASH" | ENV_GIT_HASH_KEY = "GIT_BUILD_HASH" | ||||||
|  |  | ||||||
|  |  | ||||||
| def get_build_hash(fallback: str | None = None) -> str: | def get_build_hash(fallback: Optional[str] = None) -> str: | ||||||
|     """Get build hash""" |     """Get build hash""" | ||||||
|     build_hash = environ.get(ENV_GIT_HASH_KEY, fallback if fallback else "") |     build_hash = environ.get(ENV_GIT_HASH_KEY, fallback if fallback else "") | ||||||
|     return fallback if build_hash == "" and fallback else build_hash |     return fallback if build_hash == "" and fallback else build_hash | ||||||
|  | |||||||
| @ -2,21 +2,18 @@ | |||||||
|  |  | ||||||
| import platform | import platform | ||||||
| from datetime import datetime | from datetime import datetime | ||||||
| from ssl import OPENSSL_VERSION |  | ||||||
| from sys import version as python_version | from sys import version as python_version | ||||||
| from typing import TypedDict | from typing import TypedDict | ||||||
|  |  | ||||||
| from cryptography.hazmat.backends.openssl.backend import backend |  | ||||||
| from django.utils.timezone import now | from django.utils.timezone import now | ||||||
| from drf_spectacular.utils import extend_schema | from drf_spectacular.utils import extend_schema | ||||||
|  | from gunicorn import version_info as gunicorn_version | ||||||
| from rest_framework.fields import SerializerMethodField | from rest_framework.fields import SerializerMethodField | ||||||
| from rest_framework.request import Request | from rest_framework.request import Request | ||||||
| from rest_framework.response import Response | from rest_framework.response import Response | ||||||
| from rest_framework.views import APIView | from rest_framework.views import APIView | ||||||
|  |  | ||||||
| from authentik import get_full_version |  | ||||||
| from authentik.core.api.utils import PassiveSerializer | from authentik.core.api.utils import PassiveSerializer | ||||||
| from authentik.enterprise.license import LicenseKey |  | ||||||
| from authentik.lib.config import CONFIG | from authentik.lib.config import CONFIG | ||||||
| from authentik.lib.utils.reflection import get_env | from authentik.lib.utils.reflection import get_env | ||||||
| from authentik.outposts.apps import MANAGED_OUTPOST | from authentik.outposts.apps import MANAGED_OUTPOST | ||||||
| @ -28,13 +25,11 @@ class RuntimeDict(TypedDict): | |||||||
|     """Runtime information""" |     """Runtime information""" | ||||||
|  |  | ||||||
|     python_version: str |     python_version: str | ||||||
|  |     gunicorn_version: str | ||||||
|     environment: str |     environment: str | ||||||
|     architecture: str |     architecture: str | ||||||
|     platform: str |     platform: str | ||||||
|     uname: str |     uname: str | ||||||
|     openssl_version: str |  | ||||||
|     openssl_fips_enabled: bool | None |  | ||||||
|     authentik_version: str |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class SystemInfoSerializer(PassiveSerializer): | class SystemInfoSerializer(PassiveSerializer): | ||||||
| @ -69,15 +64,11 @@ class SystemInfoSerializer(PassiveSerializer): | |||||||
|     def get_runtime(self, request: Request) -> RuntimeDict: |     def get_runtime(self, request: Request) -> RuntimeDict: | ||||||
|         """Get versions""" |         """Get versions""" | ||||||
|         return { |         return { | ||||||
|             "architecture": platform.machine(), |  | ||||||
|             "authentik_version": get_full_version(), |  | ||||||
|             "environment": get_env(), |  | ||||||
|             "openssl_fips_enabled": ( |  | ||||||
|                 backend._fips_enabled if LicenseKey.get_total().is_valid() else None |  | ||||||
|             ), |  | ||||||
|             "openssl_version": OPENSSL_VERSION, |  | ||||||
|             "platform": platform.platform(), |  | ||||||
|             "python_version": python_version, |             "python_version": python_version, | ||||||
|  |             "gunicorn_version": ".".join(str(x) for x in gunicorn_version), | ||||||
|  |             "environment": get_env(), | ||||||
|  |             "architecture": platform.machine(), | ||||||
|  |             "platform": platform.platform(), | ||||||
|             "uname": " ".join(platform.uname()), |             "uname": " ".join(platform.uname()), | ||||||
|         } |         } | ||||||
|  |  | ||||||
|  | |||||||
| @ -10,7 +10,7 @@ from rest_framework.response import Response | |||||||
| from rest_framework.views import APIView | from rest_framework.views import APIView | ||||||
|  |  | ||||||
| from authentik import __version__, get_build_hash | from authentik import __version__, get_build_hash | ||||||
| from authentik.admin.tasks import VERSION_CACHE_KEY, VERSION_NULL, update_latest_version | from authentik.admin.tasks import VERSION_CACHE_KEY, update_latest_version | ||||||
| from authentik.core.api.utils import PassiveSerializer | from authentik.core.api.utils import PassiveSerializer | ||||||
|  |  | ||||||
|  |  | ||||||
| @ -19,7 +19,6 @@ class VersionSerializer(PassiveSerializer): | |||||||
|  |  | ||||||
|     version_current = SerializerMethodField() |     version_current = SerializerMethodField() | ||||||
|     version_latest = SerializerMethodField() |     version_latest = SerializerMethodField() | ||||||
|     version_latest_valid = SerializerMethodField() |  | ||||||
|     build_hash = SerializerMethodField() |     build_hash = SerializerMethodField() | ||||||
|     outdated = SerializerMethodField() |     outdated = SerializerMethodField() | ||||||
|  |  | ||||||
| @ -39,10 +38,6 @@ class VersionSerializer(PassiveSerializer): | |||||||
|             return __version__ |             return __version__ | ||||||
|         return version_in_cache |         return version_in_cache | ||||||
|  |  | ||||||
|     def get_version_latest_valid(self, _) -> bool: |  | ||||||
|         """Check if latest version is valid""" |  | ||||||
|         return cache.get(VERSION_CACHE_KEY) != VERSION_NULL |  | ||||||
|  |  | ||||||
|     def get_outdated(self, instance) -> bool: |     def get_outdated(self, instance) -> bool: | ||||||
|         """Check if we're running the latest version""" |         """Check if we're running the latest version""" | ||||||
|         return parse(self.get_version_current(instance)) < parse(self.get_version_latest(instance)) |         return parse(self.get_version_current(instance)) < parse(self.get_version_latest(instance)) | ||||||
|  | |||||||
| @ -18,7 +18,6 @@ from authentik.lib.utils.http import get_http_session | |||||||
| from authentik.root.celery import CELERY_APP | from authentik.root.celery import CELERY_APP | ||||||
|  |  | ||||||
| LOGGER = get_logger() | LOGGER = get_logger() | ||||||
| VERSION_NULL = "0.0.0" |  | ||||||
| VERSION_CACHE_KEY = "authentik_latest_version" | VERSION_CACHE_KEY = "authentik_latest_version" | ||||||
| VERSION_CACHE_TIMEOUT = 8 * 60 * 60  # 8 hours | VERSION_CACHE_TIMEOUT = 8 * 60 * 60  # 8 hours | ||||||
| # Chop of the first ^ because we want to search the entire string | # Chop of the first ^ because we want to search the entire string | ||||||
| @ -56,7 +55,7 @@ def clear_update_notifications(): | |||||||
| def update_latest_version(self: SystemTask): | def update_latest_version(self: SystemTask): | ||||||
|     """Update latest version info""" |     """Update latest version info""" | ||||||
|     if CONFIG.get_bool("disable_update_check"): |     if CONFIG.get_bool("disable_update_check"): | ||||||
|         cache.set(VERSION_CACHE_KEY, VERSION_NULL, VERSION_CACHE_TIMEOUT) |         cache.set(VERSION_CACHE_KEY, "0.0.0", VERSION_CACHE_TIMEOUT) | ||||||
|         self.set_status(TaskStatus.WARNING, "Version check disabled.") |         self.set_status(TaskStatus.WARNING, "Version check disabled.") | ||||||
|         return |         return | ||||||
|     try: |     try: | ||||||
| @ -83,7 +82,7 @@ def update_latest_version(self: SystemTask): | |||||||
|                 event_dict["message"] = f"Changelog: {match.group()}" |                 event_dict["message"] = f"Changelog: {match.group()}" | ||||||
|             Event.new(EventAction.UPDATE_AVAILABLE, **event_dict).save() |             Event.new(EventAction.UPDATE_AVAILABLE, **event_dict).save() | ||||||
|     except (RequestException, IndexError) as exc: |     except (RequestException, IndexError) as exc: | ||||||
|         cache.set(VERSION_CACHE_KEY, VERSION_NULL, VERSION_CACHE_TIMEOUT) |         cache.set(VERSION_CACHE_KEY, "0.0.0", VERSION_CACHE_TIMEOUT) | ||||||
|         self.set_error(exc) |         self.set_error(exc) | ||||||
|  |  | ||||||
|  |  | ||||||
|  | |||||||
| @ -10,3 +10,26 @@ class AuthentikAPIConfig(AppConfig): | |||||||
|     label = "authentik_api" |     label = "authentik_api" | ||||||
|     mountpoint = "api/" |     mountpoint = "api/" | ||||||
|     verbose_name = "authentik API" |     verbose_name = "authentik API" | ||||||
|  |  | ||||||
|  |     def ready(self) -> None: | ||||||
|  |         from drf_spectacular.extensions import OpenApiAuthenticationExtension | ||||||
|  |  | ||||||
|  |         from authentik.api.authentication import TokenAuthentication | ||||||
|  |  | ||||||
|  |         # Class is defined here as it needs to be created early enough that drf-spectacular will | ||||||
|  |         # find it, but also won't cause any import issues | ||||||
|  |         # pylint: disable=unused-variable | ||||||
|  |         class TokenSchema(OpenApiAuthenticationExtension): | ||||||
|  |             """Auth schema""" | ||||||
|  |  | ||||||
|  |             target_class = TokenAuthentication | ||||||
|  |             name = "authentik" | ||||||
|  |  | ||||||
|  |             def get_security_definition(self, auto_schema): | ||||||
|  |                 """Auth schema""" | ||||||
|  |                 return { | ||||||
|  |                     "type": "apiKey", | ||||||
|  |                     "in": "header", | ||||||
|  |                     "name": "Authorization", | ||||||
|  |                     "scheme": "bearer", | ||||||
|  |                 } | ||||||
|  | |||||||
| @ -1,10 +1,9 @@ | |||||||
| """API Authentication""" | """API Authentication""" | ||||||
|  |  | ||||||
| from hmac import compare_digest | from hmac import compare_digest | ||||||
| from typing import Any | from typing import Any, Optional | ||||||
|  |  | ||||||
| from django.conf import settings | from django.conf import settings | ||||||
| from drf_spectacular.extensions import OpenApiAuthenticationExtension |  | ||||||
| from rest_framework.authentication import BaseAuthentication, get_authorization_header | from rest_framework.authentication import BaseAuthentication, get_authorization_header | ||||||
| from rest_framework.exceptions import AuthenticationFailed | from rest_framework.exceptions import AuthenticationFailed | ||||||
| from rest_framework.request import Request | from rest_framework.request import Request | ||||||
| @ -18,7 +17,7 @@ from authentik.providers.oauth2.constants import SCOPE_AUTHENTIK_API | |||||||
| LOGGER = get_logger() | LOGGER = get_logger() | ||||||
|  |  | ||||||
|  |  | ||||||
| def validate_auth(header: bytes) -> str | None: | def validate_auth(header: bytes) -> Optional[str]: | ||||||
|     """Validate that the header is in a correct format, |     """Validate that the header is in a correct format, | ||||||
|     returns type and credentials""" |     returns type and credentials""" | ||||||
|     auth_credentials = header.decode().strip() |     auth_credentials = header.decode().strip() | ||||||
| @ -33,7 +32,7 @@ def validate_auth(header: bytes) -> str | None: | |||||||
|     return auth_credentials |     return auth_credentials | ||||||
|  |  | ||||||
|  |  | ||||||
| def bearer_auth(raw_header: bytes) -> User | None: | def bearer_auth(raw_header: bytes) -> Optional[User]: | ||||||
|     """raw_header in the Format of `Bearer ....`""" |     """raw_header in the Format of `Bearer ....`""" | ||||||
|     user = auth_user_lookup(raw_header) |     user = auth_user_lookup(raw_header) | ||||||
|     if not user: |     if not user: | ||||||
| @ -43,7 +42,7 @@ def bearer_auth(raw_header: bytes) -> User | None: | |||||||
|     return user |     return user | ||||||
|  |  | ||||||
|  |  | ||||||
| def auth_user_lookup(raw_header: bytes) -> User | None: | def auth_user_lookup(raw_header: bytes) -> Optional[User]: | ||||||
|     """raw_header in the Format of `Bearer ....`""" |     """raw_header in the Format of `Bearer ....`""" | ||||||
|     from authentik.providers.oauth2.models import AccessToken |     from authentik.providers.oauth2.models import AccessToken | ||||||
|  |  | ||||||
| @ -76,7 +75,7 @@ def auth_user_lookup(raw_header: bytes) -> User | None: | |||||||
|     raise AuthenticationFailed("Token invalid/expired") |     raise AuthenticationFailed("Token invalid/expired") | ||||||
|  |  | ||||||
|  |  | ||||||
| def token_secret_key(value: str) -> User | None: | def token_secret_key(value: str) -> Optional[User]: | ||||||
|     """Check if the token is the secret key |     """Check if the token is the secret key | ||||||
|     and return the service account for the managed outpost""" |     and return the service account for the managed outpost""" | ||||||
|     from authentik.outposts.apps import MANAGED_OUTPOST |     from authentik.outposts.apps import MANAGED_OUTPOST | ||||||
| @ -103,14 +102,3 @@ class TokenAuthentication(BaseAuthentication): | |||||||
|             return None |             return None | ||||||
|  |  | ||||||
|         return (user, None)  # pragma: no cover |         return (user, None)  # pragma: no cover | ||||||
|  |  | ||||||
|  |  | ||||||
| class TokenSchema(OpenApiAuthenticationExtension): |  | ||||||
|     """Auth schema""" |  | ||||||
|  |  | ||||||
|     target_class = TokenAuthentication |  | ||||||
|     name = "authentik" |  | ||||||
|  |  | ||||||
|     def get_security_definition(self, auto_schema): |  | ||||||
|         """Auth schema""" |  | ||||||
|         return {"type": "http", "scheme": "bearer"} |  | ||||||
|  | |||||||
| @ -1,7 +1,7 @@ | |||||||
| """API Decorators""" | """API Decorators""" | ||||||
| 
 | 
 | ||||||
| from collections.abc import Callable |  | ||||||
| from functools import wraps | from functools import wraps | ||||||
|  | from typing import Callable, Optional | ||||||
| 
 | 
 | ||||||
| from rest_framework.request import Request | from rest_framework.request import Request | ||||||
| from rest_framework.response import Response | from rest_framework.response import Response | ||||||
| @ -11,26 +11,21 @@ from structlog.stdlib import get_logger | |||||||
| LOGGER = get_logger() | LOGGER = get_logger() | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
| def permission_required(obj_perm: str | None = None, global_perms: list[str] | None = None): | def permission_required(obj_perm: Optional[str] = None, global_perms: Optional[list[str]] = None): | ||||||
|     """Check permissions for a single custom action""" |     """Check permissions for a single custom action""" | ||||||
| 
 | 
 | ||||||
|     def _check_obj_perm(self: ModelViewSet, request: Request): |     def wrapper_outter(func: Callable): | ||||||
|         # Check obj_perm both globally and on the specific object |  | ||||||
|         # Having the global permission has higher priority |  | ||||||
|         if request.user.has_perm(obj_perm): |  | ||||||
|             return |  | ||||||
|         obj = self.get_object() |  | ||||||
|         if not request.user.has_perm(obj_perm, obj): |  | ||||||
|             LOGGER.debug("denying access for object", user=request.user, perm=obj_perm, obj=obj) |  | ||||||
|             self.permission_denied(request) |  | ||||||
| 
 |  | ||||||
|     def wrapper_outer(func: Callable): |  | ||||||
|         """Check permissions for a single custom action""" |         """Check permissions for a single custom action""" | ||||||
| 
 | 
 | ||||||
|         @wraps(func) |         @wraps(func) | ||||||
|         def wrapper(self: ModelViewSet, request: Request, *args, **kwargs) -> Response: |         def wrapper(self: ModelViewSet, request: Request, *args, **kwargs) -> Response: | ||||||
|             if obj_perm: |             if obj_perm: | ||||||
|                 _check_obj_perm(self, request) |                 obj = self.get_object() | ||||||
|  |                 if not request.user.has_perm(obj_perm, obj): | ||||||
|  |                     LOGGER.debug( | ||||||
|  |                         "denying access for object", user=request.user, perm=obj_perm, obj=obj | ||||||
|  |                     ) | ||||||
|  |                     return self.permission_denied(request) | ||||||
|             if global_perms: |             if global_perms: | ||||||
|                 for other_perm in global_perms: |                 for other_perm in global_perms: | ||||||
|                     if not request.user.has_perm(other_perm): |                     if not request.user.has_perm(other_perm): | ||||||
| @ -40,4 +35,4 @@ def permission_required(obj_perm: str | None = None, global_perms: list[str] | N | |||||||
| 
 | 
 | ||||||
|         return wrapper |         return wrapper | ||||||
| 
 | 
 | ||||||
|     return wrapper_outer |     return wrapper_outter | ||||||
| @ -12,7 +12,6 @@ from drf_spectacular.settings import spectacular_settings | |||||||
| from drf_spectacular.types import OpenApiTypes | from drf_spectacular.types import OpenApiTypes | ||||||
| from rest_framework.settings import api_settings | from rest_framework.settings import api_settings | ||||||
|  |  | ||||||
| from authentik.api.apps import AuthentikAPIConfig |  | ||||||
| from authentik.api.pagination import PAGINATION_COMPONENT_NAME, PAGINATION_SCHEMA | from authentik.api.pagination import PAGINATION_COMPONENT_NAME, PAGINATION_SCHEMA | ||||||
|  |  | ||||||
|  |  | ||||||
| @ -102,12 +101,3 @@ def postprocess_schema_responses(result, generator: SchemaGenerator, **kwargs): | |||||||
|             comp = result["components"]["schemas"][component] |             comp = result["components"]["schemas"][component] | ||||||
|             comp["additionalProperties"] = {} |             comp["additionalProperties"] = {} | ||||||
|     return result |     return result | ||||||
|  |  | ||||||
|  |  | ||||||
| def preprocess_schema_exclude_non_api(endpoints, **kwargs): |  | ||||||
|     """Filter out all API Views which are not mounted under /api""" |  | ||||||
|     return [ |  | ||||||
|         (path, path_regex, method, callback) |  | ||||||
|         for path, path_regex, method, callback in endpoints |  | ||||||
|         if path.startswith("/" + AuthentikAPIConfig.mountpoint) |  | ||||||
|     ] |  | ||||||
|  | |||||||
| @ -25,17 +25,17 @@ class TestAPIAuth(TestCase): | |||||||
|     def test_invalid_type(self): |     def test_invalid_type(self): | ||||||
|         """Test invalid type""" |         """Test invalid type""" | ||||||
|         with self.assertRaises(AuthenticationFailed): |         with self.assertRaises(AuthenticationFailed): | ||||||
|             bearer_auth(b"foo bar") |             bearer_auth("foo bar".encode()) | ||||||
|  |  | ||||||
|     def test_invalid_empty(self): |     def test_invalid_empty(self): | ||||||
|         """Test invalid type""" |         """Test invalid type""" | ||||||
|         self.assertIsNone(bearer_auth(b"Bearer ")) |         self.assertIsNone(bearer_auth("Bearer ".encode())) | ||||||
|         self.assertIsNone(bearer_auth(b"")) |         self.assertIsNone(bearer_auth("".encode())) | ||||||
|  |  | ||||||
|     def test_invalid_no_token(self): |     def test_invalid_no_token(self): | ||||||
|         """Test invalid with no token""" |         """Test invalid with no token""" | ||||||
|         with self.assertRaises(AuthenticationFailed): |         with self.assertRaises(AuthenticationFailed): | ||||||
|             auth = b64encode(b":abc").decode() |             auth = b64encode(":abc".encode()).decode() | ||||||
|             self.assertIsNone(bearer_auth(f"Basic :{auth}".encode())) |             self.assertIsNone(bearer_auth(f"Basic :{auth}".encode())) | ||||||
|  |  | ||||||
|     def test_bearer_valid(self): |     def test_bearer_valid(self): | ||||||
|  | |||||||
							
								
								
									
										35
									
								
								authentik/api/tests/test_decorators.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								authentik/api/tests/test_decorators.py
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,35 @@ | |||||||
|  | """test decorators api""" | ||||||
|  |  | ||||||
|  | from django.urls import reverse | ||||||
|  | from guardian.shortcuts import assign_perm | ||||||
|  | from rest_framework.test import APITestCase | ||||||
|  |  | ||||||
|  | from authentik.core.models import Application, User | ||||||
|  | from authentik.lib.generators import generate_id | ||||||
|  |  | ||||||
|  |  | ||||||
|  | class TestAPIDecorators(APITestCase): | ||||||
|  |     """test decorators api""" | ||||||
|  |  | ||||||
|  |     def setUp(self) -> None: | ||||||
|  |         super().setUp() | ||||||
|  |         self.user = User.objects.create(username="test-user") | ||||||
|  |  | ||||||
|  |     def test_obj_perm_denied(self): | ||||||
|  |         """Test object perm denied""" | ||||||
|  |         self.client.force_login(self.user) | ||||||
|  |         app = Application.objects.create(name=generate_id(), slug=generate_id()) | ||||||
|  |         response = self.client.get( | ||||||
|  |             reverse("authentik_api:application-metrics", kwargs={"slug": app.slug}) | ||||||
|  |         ) | ||||||
|  |         self.assertEqual(response.status_code, 403) | ||||||
|  |  | ||||||
|  |     def test_other_perm_denied(self): | ||||||
|  |         """Test other perm denied""" | ||||||
|  |         self.client.force_login(self.user) | ||||||
|  |         app = Application.objects.create(name=generate_id(), slug=generate_id()) | ||||||
|  |         assign_perm("authentik_core.view_application", self.user, app) | ||||||
|  |         response = self.client.get( | ||||||
|  |             reverse("authentik_api:application-metrics", kwargs={"slug": app.slug}) | ||||||
|  |         ) | ||||||
|  |         self.assertEqual(response.status_code, 403) | ||||||
| @ -1,6 +1,6 @@ | |||||||
| """authentik API Modelviewset tests""" | """authentik API Modelviewset tests""" | ||||||
|  |  | ||||||
| from collections.abc import Callable | from typing import Callable | ||||||
|  |  | ||||||
| from django.test import TestCase | from django.test import TestCase | ||||||
| from rest_framework.viewsets import ModelViewSet, ReadOnlyModelViewSet | from rest_framework.viewsets import ModelViewSet, ReadOnlyModelViewSet | ||||||
| @ -26,6 +26,6 @@ def viewset_tester_factory(test_viewset: type[ModelViewSet]) -> Callable: | |||||||
|  |  | ||||||
|  |  | ||||||
| for _, viewset, _ in router.registry: | for _, viewset, _ in router.registry: | ||||||
|     if not issubclass(viewset, ModelViewSet | ReadOnlyModelViewSet): |     if not issubclass(viewset, (ModelViewSet, ReadOnlyModelViewSet)): | ||||||
|         continue |         continue | ||||||
|     setattr(TestModelViewSets, f"test_viewset_{viewset.__name__}", viewset_tester_factory(viewset)) |     setattr(TestModelViewSets, f"test_viewset_{viewset.__name__}", viewset_tester_factory(viewset)) | ||||||
|  | |||||||
| @ -68,11 +68,7 @@ class ConfigView(APIView): | |||||||
|         """Get all capabilities this server instance supports""" |         """Get all capabilities this server instance supports""" | ||||||
|         caps = [] |         caps = [] | ||||||
|         deb_test = settings.DEBUG or settings.TEST |         deb_test = settings.DEBUG or settings.TEST | ||||||
|         if ( |         if Path(settings.MEDIA_ROOT).is_mount() or deb_test: | ||||||
|             CONFIG.get("storage.media.backend", "file") == "s3" |  | ||||||
|             or Path(settings.STORAGES["default"]["OPTIONS"]["location"]).is_mount() |  | ||||||
|             or deb_test |  | ||||||
|         ): |  | ||||||
|             caps.append(Capabilities.CAN_SAVE_MEDIA) |             caps.append(Capabilities.CAN_SAVE_MEDIA) | ||||||
|         for processor in get_context_processors(): |         for processor in get_context_processors(): | ||||||
|             if cap := processor.capability(): |             if cap := processor.capability(): | ||||||
|  | |||||||
| @ -33,7 +33,7 @@ for _authentik_app in get_apps(): | |||||||
|             app_name=_authentik_app.name, |             app_name=_authentik_app.name, | ||||||
|         ) |         ) | ||||||
|         continue |         continue | ||||||
|     urls: list = api_urls.api_urlpatterns |     urls: list = getattr(api_urls, "api_urlpatterns") | ||||||
|     for url in urls: |     for url in urls: | ||||||
|         if isinstance(url, URLPattern): |         if isinstance(url, URLPattern): | ||||||
|             _other_urls.append(url) |             _other_urls.append(url) | ||||||
|  | |||||||
| @ -10,13 +10,13 @@ from rest_framework.response import Response | |||||||
| from rest_framework.serializers import ListSerializer, ModelSerializer | from rest_framework.serializers import ListSerializer, ModelSerializer | ||||||
| from rest_framework.viewsets import ModelViewSet | from rest_framework.viewsets import ModelViewSet | ||||||
|  |  | ||||||
|  | from authentik.api.decorators import permission_required | ||||||
| from authentik.blueprints.models import BlueprintInstance | from authentik.blueprints.models import BlueprintInstance | ||||||
| from authentik.blueprints.v1.importer import Importer | from authentik.blueprints.v1.importer import Importer | ||||||
| from authentik.blueprints.v1.oci import OCI_PREFIX | from authentik.blueprints.v1.oci import OCI_PREFIX | ||||||
| from authentik.blueprints.v1.tasks import apply_blueprint, blueprints_find_dict | from authentik.blueprints.v1.tasks import apply_blueprint, blueprints_find_dict | ||||||
| from authentik.core.api.used_by import UsedByMixin | from authentik.core.api.used_by import UsedByMixin | ||||||
| from authentik.core.api.utils import JSONDictField, PassiveSerializer | from authentik.core.api.utils import JSONDictField, PassiveSerializer | ||||||
| from authentik.rbac.decorators import permission_required |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class ManagedSerializer: | class ManagedSerializer: | ||||||
| @ -52,9 +52,7 @@ class BlueprintInstanceSerializer(ModelSerializer): | |||||||
|         valid, logs = Importer.from_string(content, context).validate() |         valid, logs = Importer.from_string(content, context).validate() | ||||||
|         if not valid: |         if not valid: | ||||||
|             text_logs = "\n".join([x["event"] for x in logs]) |             text_logs = "\n".join([x["event"] for x in logs]) | ||||||
|             raise ValidationError( |             raise ValidationError(_("Failed to validate blueprint: %(logs)s" % {"logs": text_logs})) | ||||||
|                 _("Failed to validate blueprint: {logs}".format_map({"logs": text_logs})) |  | ||||||
|             ) |  | ||||||
|         return content |         return content | ||||||
|  |  | ||||||
|     def validate(self, attrs: dict) -> dict: |     def validate(self, attrs: dict) -> dict: | ||||||
|  | |||||||
| @ -1,6 +1,5 @@ | |||||||
| """authentik Blueprints app""" | """authentik Blueprints app""" | ||||||
|  |  | ||||||
| from collections.abc import Callable |  | ||||||
| from importlib import import_module | from importlib import import_module | ||||||
| from inspect import ismethod | from inspect import ismethod | ||||||
|  |  | ||||||
| @ -8,16 +7,14 @@ from django.apps import AppConfig | |||||||
| from django.db import DatabaseError, InternalError, ProgrammingError | from django.db import DatabaseError, InternalError, ProgrammingError | ||||||
| from structlog.stdlib import BoundLogger, get_logger | from structlog.stdlib import BoundLogger, get_logger | ||||||
|  |  | ||||||
| from authentik.root.signals import startup |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class ManagedAppConfig(AppConfig): | class ManagedAppConfig(AppConfig): | ||||||
|     """Basic reconciliation logic for apps""" |     """Basic reconciliation logic for apps""" | ||||||
|  |  | ||||||
|     logger: BoundLogger |     logger: BoundLogger | ||||||
|  |  | ||||||
|     RECONCILE_GLOBAL_CATEGORY: str = "global" |     RECONCILE_GLOBAL_PREFIX: str = "reconcile_global_" | ||||||
|     RECONCILE_TENANT_CATEGORY: str = "tenant" |     RECONCILE_TENANT_PREFIX: str = "reconcile_tenant_" | ||||||
|  |  | ||||||
|     def __init__(self, app_name: str, *args, **kwargs) -> None: |     def __init__(self, app_name: str, *args, **kwargs) -> None: | ||||||
|         super().__init__(app_name, *args, **kwargs) |         super().__init__(app_name, *args, **kwargs) | ||||||
| @ -25,13 +22,10 @@ class ManagedAppConfig(AppConfig): | |||||||
|  |  | ||||||
|     def ready(self) -> None: |     def ready(self) -> None: | ||||||
|         self.import_related() |         self.import_related() | ||||||
|         startup.connect(self._on_startup_callback, dispatch_uid=self.label) |         self.reconcile_global() | ||||||
|  |         self.reconcile_tenant() | ||||||
|         return super().ready() |         return super().ready() | ||||||
|  |  | ||||||
|     def _on_startup_callback(self, sender, **_): |  | ||||||
|         self._reconcile_global() |  | ||||||
|         self._reconcile_tenant() |  | ||||||
|  |  | ||||||
|     def import_related(self): |     def import_related(self): | ||||||
|         """Automatically import related modules which rely on just being imported |         """Automatically import related modules which rely on just being imported | ||||||
|         to register themselves (mainly django signals and celery tasks)""" |         to register themselves (mainly django signals and celery tasks)""" | ||||||
| @ -57,8 +51,7 @@ class ManagedAppConfig(AppConfig): | |||||||
|             meth = getattr(self, meth_name) |             meth = getattr(self, meth_name) | ||||||
|             if not ismethod(meth): |             if not ismethod(meth): | ||||||
|                 continue |                 continue | ||||||
|             category = getattr(meth, "_authentik_managed_reconcile", None) |             if not meth_name.startswith(prefix): | ||||||
|             if category != prefix: |  | ||||||
|                 continue |                 continue | ||||||
|             name = meth_name.replace(prefix, "") |             name = meth_name.replace(prefix, "") | ||||||
|             try: |             try: | ||||||
| @ -68,19 +61,7 @@ class ManagedAppConfig(AppConfig): | |||||||
|             except (DatabaseError, ProgrammingError, InternalError) as exc: |             except (DatabaseError, ProgrammingError, InternalError) as exc: | ||||||
|                 self.logger.warning("Failed to run reconcile", name=name, exc=exc) |                 self.logger.warning("Failed to run reconcile", name=name, exc=exc) | ||||||
|  |  | ||||||
|     @staticmethod |     def reconcile_tenant(self) -> None: | ||||||
|     def reconcile_tenant(func: Callable): |  | ||||||
|         """Mark a function to be called on startup (for each tenant)""" |  | ||||||
|         func._authentik_managed_reconcile = ManagedAppConfig.RECONCILE_TENANT_CATEGORY |  | ||||||
|         return func |  | ||||||
|  |  | ||||||
|     @staticmethod |  | ||||||
|     def reconcile_global(func: Callable): |  | ||||||
|         """Mark a function to be called on startup (globally)""" |  | ||||||
|         func._authentik_managed_reconcile = ManagedAppConfig.RECONCILE_GLOBAL_CATEGORY |  | ||||||
|         return func |  | ||||||
|  |  | ||||||
|     def _reconcile_tenant(self) -> None: |  | ||||||
|         """reconcile ourselves for tenanted methods""" |         """reconcile ourselves for tenanted methods""" | ||||||
|         from authentik.tenants.models import Tenant |         from authentik.tenants.models import Tenant | ||||||
|  |  | ||||||
| @ -91,9 +72,9 @@ class ManagedAppConfig(AppConfig): | |||||||
|             return |             return | ||||||
|         for tenant in tenants: |         for tenant in tenants: | ||||||
|             with tenant: |             with tenant: | ||||||
|                 self._reconcile(self.RECONCILE_TENANT_CATEGORY) |                 self._reconcile(self.RECONCILE_TENANT_PREFIX) | ||||||
|  |  | ||||||
|     def _reconcile_global(self) -> None: |     def reconcile_global(self) -> None: | ||||||
|         """ |         """ | ||||||
|         reconcile ourselves for global methods. |         reconcile ourselves for global methods. | ||||||
|         Used for signals, tasks, etc. Database queries should not be made in here. |         Used for signals, tasks, etc. Database queries should not be made in here. | ||||||
| @ -101,7 +82,7 @@ class ManagedAppConfig(AppConfig): | |||||||
|         from django_tenants.utils import get_public_schema_name, schema_context |         from django_tenants.utils import get_public_schema_name, schema_context | ||||||
|  |  | ||||||
|         with schema_context(get_public_schema_name()): |         with schema_context(get_public_schema_name()): | ||||||
|             self._reconcile(self.RECONCILE_GLOBAL_CATEGORY) |             self._reconcile(self.RECONCILE_GLOBAL_PREFIX) | ||||||
|  |  | ||||||
|  |  | ||||||
| class AuthentikBlueprintsConfig(ManagedAppConfig): | class AuthentikBlueprintsConfig(ManagedAppConfig): | ||||||
| @ -112,13 +93,11 @@ class AuthentikBlueprintsConfig(ManagedAppConfig): | |||||||
|     verbose_name = "authentik Blueprints" |     verbose_name = "authentik Blueprints" | ||||||
|     default = True |     default = True | ||||||
|  |  | ||||||
|     @ManagedAppConfig.reconcile_global |     def reconcile_global_load_blueprints_v1_tasks(self): | ||||||
|     def load_blueprints_v1_tasks(self): |  | ||||||
|         """Load v1 tasks""" |         """Load v1 tasks""" | ||||||
|         self.import_module("authentik.blueprints.v1.tasks") |         self.import_module("authentik.blueprints.v1.tasks") | ||||||
|  |  | ||||||
|     @ManagedAppConfig.reconcile_tenant |     def reconcile_tenant_blueprints_discovery(self): | ||||||
|     def blueprints_discovery(self): |  | ||||||
|         """Run blueprint discovery""" |         """Run blueprint discovery""" | ||||||
|         from authentik.blueprints.v1.tasks import blueprints_discovery, clear_failed_blueprints |         from authentik.blueprints.v1.tasks import blueprints_discovery, clear_failed_blueprints | ||||||
|  |  | ||||||
|  | |||||||
| @ -4,14 +4,12 @@ from json import dumps | |||||||
| from typing import Any | from typing import Any | ||||||
|  |  | ||||||
| from django.core.management.base import BaseCommand, no_translations | from django.core.management.base import BaseCommand, no_translations | ||||||
| from django.db.models import Model, fields | from django.db.models import Model | ||||||
| from drf_jsonschema_serializer.convert import converter, field_to_converter | from drf_jsonschema_serializer.convert import field_to_converter | ||||||
| from rest_framework.fields import Field, JSONField, UUIDField | from rest_framework.fields import Field, JSONField, UUIDField | ||||||
| from rest_framework.relations import PrimaryKeyRelatedField |  | ||||||
| from rest_framework.serializers import Serializer | from rest_framework.serializers import Serializer | ||||||
| from structlog.stdlib import get_logger | from structlog.stdlib import get_logger | ||||||
|  |  | ||||||
| from authentik import __version__ |  | ||||||
| from authentik.blueprints.v1.common import BlueprintEntryDesiredState | from authentik.blueprints.v1.common import BlueprintEntryDesiredState | ||||||
| from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT, is_model_allowed | from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT, is_model_allowed | ||||||
| from authentik.blueprints.v1.meta.registry import BaseMetaModel, registry | from authentik.blueprints.v1.meta.registry import BaseMetaModel, registry | ||||||
| @ -20,23 +18,6 @@ from authentik.lib.models import SerializerModel | |||||||
| LOGGER = get_logger() | LOGGER = get_logger() | ||||||
|  |  | ||||||
|  |  | ||||||
| @converter |  | ||||||
| class PrimaryKeyRelatedFieldConverter: |  | ||||||
|     """Custom primary key field converter which is aware of non-integer based PKs |  | ||||||
|  |  | ||||||
|     This is not an exhaustive fix for other non-int PKs, however in authentik we either |  | ||||||
|     use UUIDs or ints""" |  | ||||||
|  |  | ||||||
|     field_class = PrimaryKeyRelatedField |  | ||||||
|  |  | ||||||
|     def convert(self, field: PrimaryKeyRelatedField): |  | ||||||
|         model: Model = field.queryset.model |  | ||||||
|         pk_field = model._meta.pk |  | ||||||
|         if isinstance(pk_field, fields.UUIDField): |  | ||||||
|             return {"type": "string", "format": "uuid"} |  | ||||||
|         return {"type": "integer"} |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class Command(BaseCommand): | class Command(BaseCommand): | ||||||
|     """Generate JSON Schema for blueprints""" |     """Generate JSON Schema for blueprints""" | ||||||
|  |  | ||||||
| @ -48,7 +29,7 @@ class Command(BaseCommand): | |||||||
|             "$schema": "http://json-schema.org/draft-07/schema", |             "$schema": "http://json-schema.org/draft-07/schema", | ||||||
|             "$id": "https://goauthentik.io/blueprints/schema.json", |             "$id": "https://goauthentik.io/blueprints/schema.json", | ||||||
|             "type": "object", |             "type": "object", | ||||||
|             "title": f"authentik {__version__} Blueprint schema", |             "title": "authentik Blueprint schema", | ||||||
|             "required": ["version", "entries"], |             "required": ["version", "entries"], | ||||||
|             "properties": { |             "properties": { | ||||||
|                 "version": { |                 "version": { | ||||||
|  | |||||||
| @ -71,19 +71,6 @@ class BlueprintInstance(SerializerModel, ManagedModel, CreatedUpdatedModel): | |||||||
|     enabled = models.BooleanField(default=True) |     enabled = models.BooleanField(default=True) | ||||||
|     managed_models = ArrayField(models.TextField(), default=list) |     managed_models = ArrayField(models.TextField(), default=list) | ||||||
|  |  | ||||||
|     class Meta: |  | ||||||
|         verbose_name = _("Blueprint Instance") |  | ||||||
|         verbose_name_plural = _("Blueprint Instances") |  | ||||||
|         unique_together = ( |  | ||||||
|             ( |  | ||||||
|                 "name", |  | ||||||
|                 "path", |  | ||||||
|             ), |  | ||||||
|         ) |  | ||||||
|  |  | ||||||
|     def __str__(self) -> str: |  | ||||||
|         return f"Blueprint Instance {self.name}" |  | ||||||
|  |  | ||||||
|     def retrieve_oci(self) -> str: |     def retrieve_oci(self) -> str: | ||||||
|         """Get blueprint from an OCI registry""" |         """Get blueprint from an OCI registry""" | ||||||
|         client = BlueprintOCIClient(self.path.replace(OCI_PREFIX, "https://")) |         client = BlueprintOCIClient(self.path.replace(OCI_PREFIX, "https://")) | ||||||
| @ -102,7 +89,7 @@ class BlueprintInstance(SerializerModel, ManagedModel, CreatedUpdatedModel): | |||||||
|                 raise BlueprintRetrievalFailed("Invalid blueprint path") |                 raise BlueprintRetrievalFailed("Invalid blueprint path") | ||||||
|             with full_path.open("r", encoding="utf-8") as _file: |             with full_path.open("r", encoding="utf-8") as _file: | ||||||
|                 return _file.read() |                 return _file.read() | ||||||
|         except OSError as exc: |         except (IOError, OSError) as exc: | ||||||
|             raise BlueprintRetrievalFailed(exc) from exc |             raise BlueprintRetrievalFailed(exc) from exc | ||||||
|  |  | ||||||
|     def retrieve(self) -> str: |     def retrieve(self) -> str: | ||||||
| @ -118,3 +105,16 @@ class BlueprintInstance(SerializerModel, ManagedModel, CreatedUpdatedModel): | |||||||
|         from authentik.blueprints.api import BlueprintInstanceSerializer |         from authentik.blueprints.api import BlueprintInstanceSerializer | ||||||
|  |  | ||||||
|         return BlueprintInstanceSerializer |         return BlueprintInstanceSerializer | ||||||
|  |  | ||||||
|  |     def __str__(self) -> str: | ||||||
|  |         return f"Blueprint Instance {self.name}" | ||||||
|  |  | ||||||
|  |     class Meta: | ||||||
|  |         verbose_name = _("Blueprint Instance") | ||||||
|  |         verbose_name_plural = _("Blueprint Instances") | ||||||
|  |         unique_together = ( | ||||||
|  |             ( | ||||||
|  |                 "name", | ||||||
|  |                 "path", | ||||||
|  |             ), | ||||||
|  |         ) | ||||||
|  | |||||||
| @ -1,7 +1,7 @@ | |||||||
| """Blueprint helpers""" | """Blueprint helpers""" | ||||||
|  |  | ||||||
| from collections.abc import Callable |  | ||||||
| from functools import wraps | from functools import wraps | ||||||
|  | from typing import Callable | ||||||
|  |  | ||||||
| from django.apps import apps | from django.apps import apps | ||||||
|  |  | ||||||
| @ -39,7 +39,7 @@ def reconcile_app(app_name: str): | |||||||
|         def wrapper(*args, **kwargs): |         def wrapper(*args, **kwargs): | ||||||
|             config = apps.get_app_config(app_name) |             config = apps.get_app_config(app_name) | ||||||
|             if isinstance(config, ManagedAppConfig): |             if isinstance(config, ManagedAppConfig): | ||||||
|                 config._on_startup_callback(None) |                 config.ready() | ||||||
|             return func(*args, **kwargs) |             return func(*args, **kwargs) | ||||||
|  |  | ||||||
|         return wrapper |         return wrapper | ||||||
|  | |||||||
| @ -1,7 +1,7 @@ | |||||||
| """test packaged blueprints""" | """test packaged blueprints""" | ||||||
|  |  | ||||||
| from collections.abc import Callable |  | ||||||
| from pathlib import Path | from pathlib import Path | ||||||
|  | from typing import Callable | ||||||
|  |  | ||||||
| from django.test import TransactionTestCase | from django.test import TransactionTestCase | ||||||
|  |  | ||||||
|  | |||||||
| @ -1,6 +1,6 @@ | |||||||
| """authentik managed models tests""" | """authentik managed models tests""" | ||||||
|  |  | ||||||
| from collections.abc import Callable | from typing import Callable, Type | ||||||
|  |  | ||||||
| from django.apps import apps | from django.apps import apps | ||||||
| from django.test import TestCase | from django.test import TestCase | ||||||
| @ -14,7 +14,7 @@ class TestModels(TestCase): | |||||||
|     """Test Models""" |     """Test Models""" | ||||||
|  |  | ||||||
|  |  | ||||||
| def serializer_tester_factory(test_model: type[SerializerModel]) -> Callable: | def serializer_tester_factory(test_model: Type[SerializerModel]) -> Callable: | ||||||
|     """Test serializer""" |     """Test serializer""" | ||||||
|  |  | ||||||
|     def tester(self: TestModels): |     def tester(self: TestModels): | ||||||
|  | |||||||
| @ -54,7 +54,7 @@ class TestBlueprintsV1Tasks(TransactionTestCase): | |||||||
|             file.seek(0) |             file.seek(0) | ||||||
|             file_hash = sha512(file.read().encode()).hexdigest() |             file_hash = sha512(file.read().encode()).hexdigest() | ||||||
|             file.flush() |             file.flush() | ||||||
|             blueprints_discovery() |             blueprints_discovery()  # pylint: disable=no-value-for-parameter | ||||||
|             instance = BlueprintInstance.objects.filter(name=blueprint_id).first() |             instance = BlueprintInstance.objects.filter(name=blueprint_id).first() | ||||||
|             self.assertEqual(instance.last_applied_hash, file_hash) |             self.assertEqual(instance.last_applied_hash, file_hash) | ||||||
|             self.assertEqual( |             self.assertEqual( | ||||||
| @ -82,7 +82,7 @@ class TestBlueprintsV1Tasks(TransactionTestCase): | |||||||
|                 ) |                 ) | ||||||
|             ) |             ) | ||||||
|             file.flush() |             file.flush() | ||||||
|             blueprints_discovery() |             blueprints_discovery()  # pylint: disable=no-value-for-parameter | ||||||
|             blueprint = BlueprintInstance.objects.filter(name="foo").first() |             blueprint = BlueprintInstance.objects.filter(name="foo").first() | ||||||
|             self.assertEqual( |             self.assertEqual( | ||||||
|                 blueprint.last_applied_hash, |                 blueprint.last_applied_hash, | ||||||
| @ -107,7 +107,7 @@ class TestBlueprintsV1Tasks(TransactionTestCase): | |||||||
|                 ) |                 ) | ||||||
|             ) |             ) | ||||||
|             file.flush() |             file.flush() | ||||||
|             blueprints_discovery() |             blueprints_discovery()  # pylint: disable=no-value-for-parameter | ||||||
|             blueprint.refresh_from_db() |             blueprint.refresh_from_db() | ||||||
|             self.assertEqual( |             self.assertEqual( | ||||||
|                 blueprint.last_applied_hash, |                 blueprint.last_applied_hash, | ||||||
| @ -149,7 +149,7 @@ class TestBlueprintsV1Tasks(TransactionTestCase): | |||||||
|                 instance.status, |                 instance.status, | ||||||
|                 BlueprintInstanceStatus.UNKNOWN, |                 BlueprintInstanceStatus.UNKNOWN, | ||||||
|             ) |             ) | ||||||
|             apply_blueprint(instance.pk) |             apply_blueprint(instance.pk)  # pylint: disable=no-value-for-parameter | ||||||
|             instance.refresh_from_db() |             instance.refresh_from_db() | ||||||
|             self.assertEqual(instance.last_applied_hash, "") |             self.assertEqual(instance.last_applied_hash, "") | ||||||
|             self.assertEqual( |             self.assertEqual( | ||||||
|  | |||||||
| @ -1,14 +1,13 @@ | |||||||
| """transfer common classes""" | """transfer common classes""" | ||||||
|  |  | ||||||
| from collections import OrderedDict | from collections import OrderedDict | ||||||
| from collections.abc import Iterable, Mapping |  | ||||||
| from copy import copy | from copy import copy | ||||||
| from dataclasses import asdict, dataclass, field, is_dataclass | from dataclasses import asdict, dataclass, field, is_dataclass | ||||||
| from enum import Enum | from enum import Enum | ||||||
| from functools import reduce | from functools import reduce | ||||||
| from operator import ixor | from operator import ixor | ||||||
| from os import getenv | from os import getenv | ||||||
| from typing import Any, Literal, Union | from typing import Any, Iterable, Literal, Mapping, Optional, Union | ||||||
| from uuid import UUID | from uuid import UUID | ||||||
|  |  | ||||||
| from deepmerge import always_merger | from deepmerge import always_merger | ||||||
| @ -46,7 +45,7 @@ def get_attrs(obj: SerializerModel) -> dict[str, Any]: | |||||||
| class BlueprintEntryState: | class BlueprintEntryState: | ||||||
|     """State of a single instance""" |     """State of a single instance""" | ||||||
|  |  | ||||||
|     instance: Model | None = None |     instance: Optional[Model] = None | ||||||
|  |  | ||||||
|  |  | ||||||
| class BlueprintEntryDesiredState(Enum): | class BlueprintEntryDesiredState(Enum): | ||||||
| @ -68,14 +67,14 @@ class BlueprintEntry: | |||||||
|     ) |     ) | ||||||
|     conditions: list[Any] = field(default_factory=list) |     conditions: list[Any] = field(default_factory=list) | ||||||
|     identifiers: dict[str, Any] = field(default_factory=dict) |     identifiers: dict[str, Any] = field(default_factory=dict) | ||||||
|     attrs: dict[str, Any] | None = field(default_factory=dict) |     attrs: Optional[dict[str, Any]] = field(default_factory=dict) | ||||||
|  |  | ||||||
|     id: str | None = None |     id: Optional[str] = None | ||||||
|  |  | ||||||
|     _state: BlueprintEntryState = field(default_factory=BlueprintEntryState) |     _state: BlueprintEntryState = field(default_factory=BlueprintEntryState) | ||||||
|  |  | ||||||
|     def __post_init__(self, *args, **kwargs) -> None: |     def __post_init__(self, *args, **kwargs) -> None: | ||||||
|         self.__tag_contexts: list[YAMLTagContext] = [] |         self.__tag_contexts: list["YAMLTagContext"] = [] | ||||||
|  |  | ||||||
|     @staticmethod |     @staticmethod | ||||||
|     def from_model(model: SerializerModel, *extra_identifier_names: str) -> "BlueprintEntry": |     def from_model(model: SerializerModel, *extra_identifier_names: str) -> "BlueprintEntry": | ||||||
| @ -93,10 +92,10 @@ class BlueprintEntry: | |||||||
|             attrs=all_attrs, |             attrs=all_attrs, | ||||||
|         ) |         ) | ||||||
|  |  | ||||||
|     def get_tag_context( |     def _get_tag_context( | ||||||
|         self, |         self, | ||||||
|         depth: int = 0, |         depth: int = 0, | ||||||
|         context_tag_type: type["YAMLTagContext"] | tuple["YAMLTagContext", ...] | None = None, |         context_tag_type: Optional[type["YAMLTagContext"] | tuple["YAMLTagContext", ...]] = None, | ||||||
|     ) -> "YAMLTagContext": |     ) -> "YAMLTagContext": | ||||||
|         """Get a YAMLTagContext object located at a certain depth in the tag tree""" |         """Get a YAMLTagContext object located at a certain depth in the tag tree""" | ||||||
|         if depth < 0: |         if depth < 0: | ||||||
| @ -109,8 +108,8 @@ class BlueprintEntry: | |||||||
|  |  | ||||||
|         try: |         try: | ||||||
|             return contexts[-(depth + 1)] |             return contexts[-(depth + 1)] | ||||||
|         except IndexError as exc: |         except IndexError: | ||||||
|             raise ValueError(f"invalid depth: {depth}. Max depth: {len(contexts) - 1}") from exc |             raise ValueError(f"invalid depth: {depth}. Max depth: {len(contexts) - 1}") | ||||||
|  |  | ||||||
|     def tag_resolver(self, value: Any, blueprint: "Blueprint") -> Any: |     def tag_resolver(self, value: Any, blueprint: "Blueprint") -> Any: | ||||||
|         """Check if we have any special tags that need handling""" |         """Check if we have any special tags that need handling""" | ||||||
| @ -171,7 +170,7 @@ class Blueprint: | |||||||
|     entries: list[BlueprintEntry] = field(default_factory=list) |     entries: list[BlueprintEntry] = field(default_factory=list) | ||||||
|     context: dict = field(default_factory=dict) |     context: dict = field(default_factory=dict) | ||||||
|  |  | ||||||
|     metadata: BlueprintMetadata | None = field(default=None) |     metadata: Optional[BlueprintMetadata] = field(default=None) | ||||||
|  |  | ||||||
|  |  | ||||||
| class YAMLTag: | class YAMLTag: | ||||||
| @ -219,7 +218,7 @@ class Env(YAMLTag): | |||||||
|     """Lookup environment variable with optional default""" |     """Lookup environment variable with optional default""" | ||||||
|  |  | ||||||
|     key: str |     key: str | ||||||
|     default: Any | None |     default: Optional[Any] | ||||||
|  |  | ||||||
|     def __init__(self, loader: "BlueprintLoader", node: ScalarNode | SequenceNode) -> None: |     def __init__(self, loader: "BlueprintLoader", node: ScalarNode | SequenceNode) -> None: | ||||||
|         super().__init__() |         super().__init__() | ||||||
| @ -238,7 +237,7 @@ class Context(YAMLTag): | |||||||
|     """Lookup key from instance context""" |     """Lookup key from instance context""" | ||||||
|  |  | ||||||
|     key: str |     key: str | ||||||
|     default: Any | None |     default: Optional[Any] | ||||||
|  |  | ||||||
|     def __init__(self, loader: "BlueprintLoader", node: ScalarNode | SequenceNode) -> None: |     def __init__(self, loader: "BlueprintLoader", node: ScalarNode | SequenceNode) -> None: | ||||||
|         super().__init__() |         super().__init__() | ||||||
| @ -282,7 +281,7 @@ class Format(YAMLTag): | |||||||
|         try: |         try: | ||||||
|             return self.format_string % tuple(args) |             return self.format_string % tuple(args) | ||||||
|         except TypeError as exc: |         except TypeError as exc: | ||||||
|             raise EntryInvalidError.from_entry(exc, entry) from exc |             raise EntryInvalidError.from_entry(exc, entry) | ||||||
|  |  | ||||||
|  |  | ||||||
| class Find(YAMLTag): | class Find(YAMLTag): | ||||||
| @ -367,7 +366,7 @@ class Condition(YAMLTag): | |||||||
|             comparator = self._COMPARATORS[self.mode.upper()] |             comparator = self._COMPARATORS[self.mode.upper()] | ||||||
|             return comparator(tuple(bool(x) for x in args)) |             return comparator(tuple(bool(x) for x in args)) | ||||||
|         except (TypeError, KeyError) as exc: |         except (TypeError, KeyError) as exc: | ||||||
|             raise EntryInvalidError.from_entry(exc, entry) from exc |             raise EntryInvalidError.from_entry(exc, entry) | ||||||
|  |  | ||||||
|  |  | ||||||
| class If(YAMLTag): | class If(YAMLTag): | ||||||
| @ -399,7 +398,7 @@ class If(YAMLTag): | |||||||
|                 blueprint, |                 blueprint, | ||||||
|             ) |             ) | ||||||
|         except TypeError as exc: |         except TypeError as exc: | ||||||
|             raise EntryInvalidError.from_entry(exc, entry) from exc |             raise EntryInvalidError.from_entry(exc, entry) | ||||||
|  |  | ||||||
|  |  | ||||||
| class Enumerate(YAMLTag, YAMLTagContext): | class Enumerate(YAMLTag, YAMLTagContext): | ||||||
| @ -413,7 +412,9 @@ class Enumerate(YAMLTag, YAMLTagContext): | |||||||
|         "SEQ": (list, lambda a, b: [*a, b]), |         "SEQ": (list, lambda a, b: [*a, b]), | ||||||
|         "MAP": ( |         "MAP": ( | ||||||
|             dict, |             dict, | ||||||
|             lambda a, b: always_merger.merge(a, {b[0]: b[1]} if isinstance(b, tuple | list) else b), |             lambda a, b: always_merger.merge( | ||||||
|  |                 a, {b[0]: b[1]} if isinstance(b, (tuple, list)) else b | ||||||
|  |             ), | ||||||
|         ), |         ), | ||||||
|     } |     } | ||||||
|  |  | ||||||
| @ -455,7 +456,7 @@ class Enumerate(YAMLTag, YAMLTagContext): | |||||||
|         try: |         try: | ||||||
|             output_class, add_fn = self._OUTPUT_BODIES[self.output_body.upper()] |             output_class, add_fn = self._OUTPUT_BODIES[self.output_body.upper()] | ||||||
|         except KeyError as exc: |         except KeyError as exc: | ||||||
|             raise EntryInvalidError.from_entry(exc, entry) from exc |             raise EntryInvalidError.from_entry(exc, entry) | ||||||
|  |  | ||||||
|         result = output_class() |         result = output_class() | ||||||
|  |  | ||||||
| @ -483,13 +484,13 @@ class EnumeratedItem(YAMLTag): | |||||||
|  |  | ||||||
|     _SUPPORTED_CONTEXT_TAGS = (Enumerate,) |     _SUPPORTED_CONTEXT_TAGS = (Enumerate,) | ||||||
|  |  | ||||||
|     def __init__(self, _loader: "BlueprintLoader", node: ScalarNode) -> None: |     def __init__(self, loader: "BlueprintLoader", node: ScalarNode) -> None: | ||||||
|         super().__init__() |         super().__init__() | ||||||
|         self.depth = int(node.value) |         self.depth = int(node.value) | ||||||
|  |  | ||||||
|     def resolve(self, entry: BlueprintEntry, blueprint: Blueprint) -> Any: |     def resolve(self, entry: BlueprintEntry, blueprint: Blueprint) -> Any: | ||||||
|         try: |         try: | ||||||
|             context_tag: Enumerate = entry.get_tag_context( |             context_tag: Enumerate = entry._get_tag_context( | ||||||
|                 depth=self.depth, |                 depth=self.depth, | ||||||
|                 context_tag_type=EnumeratedItem._SUPPORTED_CONTEXT_TAGS, |                 context_tag_type=EnumeratedItem._SUPPORTED_CONTEXT_TAGS, | ||||||
|             ) |             ) | ||||||
| @ -499,11 +500,9 @@ class EnumeratedItem(YAMLTag): | |||||||
|                     f"{self.__class__.__name__} tags are only usable " |                     f"{self.__class__.__name__} tags are only usable " | ||||||
|                     f"inside an {Enumerate.__name__} tag", |                     f"inside an {Enumerate.__name__} tag", | ||||||
|                     entry, |                     entry, | ||||||
|                 ) from exc |                 ) | ||||||
|  |  | ||||||
|             raise EntryInvalidError.from_entry( |             raise EntryInvalidError.from_entry(f"{self.__class__.__name__} tag: {exc}", entry) | ||||||
|                 f"{self.__class__.__name__} tag: {exc}", entry |  | ||||||
|             ) from exc |  | ||||||
|  |  | ||||||
|         return context_tag.get_context(entry, blueprint) |         return context_tag.get_context(entry, blueprint) | ||||||
|  |  | ||||||
| @ -516,8 +515,8 @@ class Index(EnumeratedItem): | |||||||
|  |  | ||||||
|         try: |         try: | ||||||
|             return context[0] |             return context[0] | ||||||
|         except IndexError as exc:  # pragma: no cover |         except IndexError:  # pragma: no cover | ||||||
|             raise EntryInvalidError.from_entry(f"Empty/invalid context: {context}", entry) from exc |             raise EntryInvalidError.from_entry(f"Empty/invalid context: {context}", entry) | ||||||
|  |  | ||||||
|  |  | ||||||
| class Value(EnumeratedItem): | class Value(EnumeratedItem): | ||||||
| @ -528,8 +527,8 @@ class Value(EnumeratedItem): | |||||||
|  |  | ||||||
|         try: |         try: | ||||||
|             return context[1] |             return context[1] | ||||||
|         except IndexError as exc:  # pragma: no cover |         except IndexError:  # pragma: no cover | ||||||
|             raise EntryInvalidError.from_entry(f"Empty/invalid context: {context}", entry) from exc |             raise EntryInvalidError.from_entry(f"Empty/invalid context: {context}", entry) | ||||||
|  |  | ||||||
|  |  | ||||||
| class BlueprintDumper(SafeDumper): | class BlueprintDumper(SafeDumper): | ||||||
| @ -556,11 +555,7 @@ class BlueprintDumper(SafeDumper): | |||||||
|  |  | ||||||
|             def factory(items): |             def factory(items): | ||||||
|                 final_dict = dict(items) |                 final_dict = dict(items) | ||||||
|                 # Remove internal state variables |  | ||||||
|                 final_dict.pop("_state", None) |                 final_dict.pop("_state", None) | ||||||
|                 # Future-proof to only remove the ID if we don't set a value |  | ||||||
|                 if "id" in final_dict and final_dict.get("id") is None: |  | ||||||
|                     final_dict.pop("id") |  | ||||||
|                 return final_dict |                 return final_dict | ||||||
|  |  | ||||||
|             data = asdict(data, dict_factory=factory) |             data = asdict(data, dict_factory=factory) | ||||||
| @ -587,13 +582,13 @@ class BlueprintLoader(SafeLoader): | |||||||
| class EntryInvalidError(SentryIgnoredException): | class EntryInvalidError(SentryIgnoredException): | ||||||
|     """Error raised when an entry is invalid""" |     """Error raised when an entry is invalid""" | ||||||
|  |  | ||||||
|     entry_model: str | None |     entry_model: Optional[str] | ||||||
|     entry_id: str | None |     entry_id: Optional[str] | ||||||
|     validation_error: ValidationError | None |     validation_error: Optional[ValidationError] | ||||||
|     serializer: Serializer | None = None |     serializer: Optional[Serializer] = None | ||||||
|  |  | ||||||
|     def __init__( |     def __init__( | ||||||
|         self, *args: object, validation_error: ValidationError | None = None, **kwargs |         self, *args: object, validation_error: Optional[ValidationError] = None, **kwargs | ||||||
|     ) -> None: |     ) -> None: | ||||||
|         super().__init__(*args) |         super().__init__(*args) | ||||||
|         self.entry_model = None |         self.entry_model = None | ||||||
|  | |||||||
| @ -1,6 +1,6 @@ | |||||||
| """Blueprint exporter""" | """Blueprint exporter""" | ||||||
|  |  | ||||||
| from collections.abc import Iterable | from typing import Iterable | ||||||
| from uuid import UUID | from uuid import UUID | ||||||
|  |  | ||||||
| from django.apps import apps | from django.apps import apps | ||||||
| @ -59,7 +59,7 @@ class Exporter: | |||||||
|         blueprint = Blueprint() |         blueprint = Blueprint() | ||||||
|         self._pre_export(blueprint) |         self._pre_export(blueprint) | ||||||
|         blueprint.metadata = BlueprintMetadata( |         blueprint.metadata = BlueprintMetadata( | ||||||
|             name=_("authentik Export - {date}".format_map({"date": str(now())})), |             name=_("authentik Export - %(date)s" % {"date": str(now())}), | ||||||
|             labels={ |             labels={ | ||||||
|                 LABEL_AUTHENTIK_GENERATED: "true", |                 LABEL_AUTHENTIK_GENERATED: "true", | ||||||
|             }, |             }, | ||||||
| @ -74,7 +74,7 @@ class Exporter: | |||||||
|  |  | ||||||
|  |  | ||||||
| class FlowExporter(Exporter): | class FlowExporter(Exporter): | ||||||
|     """Exporter customized to only return objects related to `flow`""" |     """Exporter customised to only return objects related to `flow`""" | ||||||
|  |  | ||||||
|     flow: Flow |     flow: Flow | ||||||
|     with_policies: bool |     with_policies: bool | ||||||
|  | |||||||
| @ -2,7 +2,7 @@ | |||||||
|  |  | ||||||
| from contextlib import contextmanager | from contextlib import contextmanager | ||||||
| from copy import deepcopy | from copy import deepcopy | ||||||
| from typing import Any | from typing import Any, Optional | ||||||
|  |  | ||||||
| from dacite.config import Config | from dacite.config import Config | ||||||
| from dacite.core import from_dict | from dacite.core import from_dict | ||||||
| @ -19,6 +19,8 @@ from guardian.models import UserObjectPermission | |||||||
| from rest_framework.exceptions import ValidationError | from rest_framework.exceptions import ValidationError | ||||||
| from rest_framework.serializers import BaseSerializer, Serializer | from rest_framework.serializers import BaseSerializer, Serializer | ||||||
| from structlog.stdlib import BoundLogger, get_logger | from structlog.stdlib import BoundLogger, get_logger | ||||||
|  | from structlog.testing import capture_logs | ||||||
|  | from structlog.types import EventDict | ||||||
| from yaml import load | from yaml import load | ||||||
|  |  | ||||||
| from authentik.blueprints.v1.common import ( | from authentik.blueprints.v1.common import ( | ||||||
| @ -39,16 +41,7 @@ from authentik.core.models import ( | |||||||
| ) | ) | ||||||
| from authentik.enterprise.license import LicenseKey | from authentik.enterprise.license import LicenseKey | ||||||
| from authentik.enterprise.models import LicenseUsage | from authentik.enterprise.models import LicenseUsage | ||||||
| from authentik.enterprise.providers.google_workspace.models import ( |  | ||||||
|     GoogleWorkspaceProviderGroup, |  | ||||||
|     GoogleWorkspaceProviderUser, |  | ||||||
| ) |  | ||||||
| from authentik.enterprise.providers.microsoft_entra.models import ( |  | ||||||
|     MicrosoftEntraProviderGroup, |  | ||||||
|     MicrosoftEntraProviderUser, |  | ||||||
| ) |  | ||||||
| from authentik.enterprise.providers.rac.models import ConnectionToken | from authentik.enterprise.providers.rac.models import ConnectionToken | ||||||
| from authentik.events.logs import LogEvent, capture_logs |  | ||||||
| from authentik.events.models import SystemTask | from authentik.events.models import SystemTask | ||||||
| from authentik.events.utils import cleanse_dict | from authentik.events.utils import cleanse_dict | ||||||
| from authentik.flows.models import FlowToken, Stage | from authentik.flows.models import FlowToken, Stage | ||||||
| @ -58,9 +51,7 @@ from authentik.outposts.models import OutpostServiceConnection | |||||||
| from authentik.policies.models import Policy, PolicyBindingModel | from authentik.policies.models import Policy, PolicyBindingModel | ||||||
| from authentik.policies.reputation.models import Reputation | from authentik.policies.reputation.models import Reputation | ||||||
| from authentik.providers.oauth2.models import AccessToken, AuthorizationCode, RefreshToken | from authentik.providers.oauth2.models import AccessToken, AuthorizationCode, RefreshToken | ||||||
| from authentik.providers.scim.models import SCIMProviderGroup, SCIMProviderUser | from authentik.providers.scim.models import SCIMGroup, SCIMUser | ||||||
| from authentik.sources.scim.models import SCIMSourceGroup, SCIMSourceUser |  | ||||||
| from authentik.stages.authenticator_webauthn.models import WebAuthnDeviceType |  | ||||||
| from authentik.tenants.models import Tenant | from authentik.tenants.models import Tenant | ||||||
|  |  | ||||||
| # Context set when the serializer is created in a blueprint context | # Context set when the serializer is created in a blueprint context | ||||||
| @ -71,7 +62,7 @@ SERIALIZER_CONTEXT_BLUEPRINT = "blueprint_entry" | |||||||
| def excluded_models() -> list[type[Model]]: | def excluded_models() -> list[type[Model]]: | ||||||
|     """Return a list of all excluded models that shouldn't be exposed via API |     """Return a list of all excluded models that shouldn't be exposed via API | ||||||
|     or other means (internal only, base classes, non-used objects, etc)""" |     or other means (internal only, base classes, non-used objects, etc)""" | ||||||
|  |     # pylint: disable=imported-auth-user | ||||||
|     from django.contrib.auth.models import Group as DjangoGroup |     from django.contrib.auth.models import Group as DjangoGroup | ||||||
|     from django.contrib.auth.models import User as DjangoUser |     from django.contrib.auth.models import User as DjangoUser | ||||||
|  |  | ||||||
| @ -94,11 +85,10 @@ def excluded_models() -> list[type[Model]]: | |||||||
|         # Classes that have other dependencies |         # Classes that have other dependencies | ||||||
|         AuthenticatedSession, |         AuthenticatedSession, | ||||||
|         # Classes which are only internally managed |         # Classes which are only internally managed | ||||||
|         # FIXME: these shouldn't need to be explicitly listed, but rather based off of a mixin |  | ||||||
|         FlowToken, |         FlowToken, | ||||||
|         LicenseUsage, |         LicenseUsage, | ||||||
|         SCIMProviderGroup, |         SCIMGroup, | ||||||
|         SCIMProviderUser, |         SCIMUser, | ||||||
|         Tenant, |         Tenant, | ||||||
|         SystemTask, |         SystemTask, | ||||||
|         ConnectionToken, |         ConnectionToken, | ||||||
| @ -106,19 +96,12 @@ def excluded_models() -> list[type[Model]]: | |||||||
|         AccessToken, |         AccessToken, | ||||||
|         RefreshToken, |         RefreshToken, | ||||||
|         Reputation, |         Reputation, | ||||||
|         WebAuthnDeviceType, |  | ||||||
|         SCIMSourceUser, |  | ||||||
|         SCIMSourceGroup, |  | ||||||
|         GoogleWorkspaceProviderUser, |  | ||||||
|         GoogleWorkspaceProviderGroup, |  | ||||||
|         MicrosoftEntraProviderUser, |  | ||||||
|         MicrosoftEntraProviderGroup, |  | ||||||
|     ) |     ) | ||||||
|  |  | ||||||
|  |  | ||||||
| def is_model_allowed(model: type[Model]) -> bool: | def is_model_allowed(model: type[Model]) -> bool: | ||||||
|     """Check if model is allowed""" |     """Check if model is allowed""" | ||||||
|     return model not in excluded_models() and issubclass(model, SerializerModel | BaseMetaModel) |     return model not in excluded_models() and issubclass(model, (SerializerModel, BaseMetaModel)) | ||||||
|  |  | ||||||
|  |  | ||||||
| class DoRollback(SentryIgnoredException): | class DoRollback(SentryIgnoredException): | ||||||
| @ -142,7 +125,7 @@ class Importer: | |||||||
|     logger: BoundLogger |     logger: BoundLogger | ||||||
|     _import: Blueprint |     _import: Blueprint | ||||||
|  |  | ||||||
|     def __init__(self, blueprint: Blueprint, context: dict | None = None): |     def __init__(self, blueprint: Blueprint, context: Optional[dict] = None): | ||||||
|         self.__pk_map: dict[Any, Model] = {} |         self.__pk_map: dict[Any, Model] = {} | ||||||
|         self._import = blueprint |         self._import = blueprint | ||||||
|         self.logger = get_logger() |         self.logger = get_logger() | ||||||
| @ -178,14 +161,14 @@ class Importer: | |||||||
|  |  | ||||||
|         def updater(value) -> Any: |         def updater(value) -> Any: | ||||||
|             if value in self.__pk_map: |             if value in self.__pk_map: | ||||||
|                 self.logger.debug("Updating reference in entry", value=value) |                 self.logger.debug("updating reference in entry", value=value) | ||||||
|                 return self.__pk_map[value] |                 return self.__pk_map[value] | ||||||
|             return value |             return value | ||||||
|  |  | ||||||
|         for key, value in attrs.items(): |         for key, value in attrs.items(): | ||||||
|             try: |             try: | ||||||
|                 if isinstance(value, dict): |                 if isinstance(value, dict): | ||||||
|                     for _, _inner_key in enumerate(value): |                     for idx, _inner_key in enumerate(value): | ||||||
|                         value[_inner_key] = updater(value[_inner_key]) |                         value[_inner_key] = updater(value[_inner_key]) | ||||||
|                 elif isinstance(value, list): |                 elif isinstance(value, list): | ||||||
|                     for idx, _inner_value in enumerate(value): |                     for idx, _inner_value in enumerate(value): | ||||||
| @ -214,7 +197,8 @@ class Importer: | |||||||
|  |  | ||||||
|         return main_query | sub_query |         return main_query | sub_query | ||||||
|  |  | ||||||
|     def _validate_single(self, entry: BlueprintEntry) -> BaseSerializer | None: |     # pylint: disable-msg=too-many-locals | ||||||
|  |     def _validate_single(self, entry: BlueprintEntry) -> Optional[BaseSerializer]: | ||||||
|         """Validate a single entry""" |         """Validate a single entry""" | ||||||
|         if not entry.check_all_conditions_match(self._import): |         if not entry.check_all_conditions_match(self._import): | ||||||
|             self.logger.debug("One or more conditions of this entry are not fulfilled, skipping") |             self.logger.debug("One or more conditions of this entry are not fulfilled, skipping") | ||||||
| @ -267,7 +251,7 @@ class Importer: | |||||||
|         model_instance = existing_models.first() |         model_instance = existing_models.first() | ||||||
|         if not isinstance(model(), BaseMetaModel) and model_instance: |         if not isinstance(model(), BaseMetaModel) and model_instance: | ||||||
|             self.logger.debug( |             self.logger.debug( | ||||||
|                 "Initialise serializer with instance", |                 "initialise serializer with instance", | ||||||
|                 model=model, |                 model=model, | ||||||
|                 instance=model_instance, |                 instance=model_instance, | ||||||
|                 pk=model_instance.pk, |                 pk=model_instance.pk, | ||||||
| @ -277,14 +261,14 @@ class Importer: | |||||||
|         elif model_instance and entry.state == BlueprintEntryDesiredState.MUST_CREATED: |         elif model_instance and entry.state == BlueprintEntryDesiredState.MUST_CREATED: | ||||||
|             raise EntryInvalidError.from_entry( |             raise EntryInvalidError.from_entry( | ||||||
|                 ( |                 ( | ||||||
|                     f"State is set to {BlueprintEntryDesiredState.MUST_CREATED} " |                     f"state is set to {BlueprintEntryDesiredState.MUST_CREATED} " | ||||||
|                     "and object exists already", |                     "and object exists already", | ||||||
|                 ), |                 ), | ||||||
|                 entry, |                 entry, | ||||||
|             ) |             ) | ||||||
|         else: |         else: | ||||||
|             self.logger.debug( |             self.logger.debug( | ||||||
|                 "Initialised new serializer instance", |                 "initialised new serializer instance", | ||||||
|                 model=model, |                 model=model, | ||||||
|                 **cleanse_dict(updated_identifiers), |                 **cleanse_dict(updated_identifiers), | ||||||
|             ) |             ) | ||||||
| @ -341,7 +325,7 @@ class Importer: | |||||||
|                 model: type[SerializerModel] = registry.get_model(model_app_label, model_name) |                 model: type[SerializerModel] = registry.get_model(model_app_label, model_name) | ||||||
|             except LookupError: |             except LookupError: | ||||||
|                 self.logger.warning( |                 self.logger.warning( | ||||||
|                     "App or Model does not exist", app=model_app_label, model=model_name |                     "app or model does not exist", app=model_app_label, model=model_name | ||||||
|                 ) |                 ) | ||||||
|                 return False |                 return False | ||||||
|             # Validate each single entry |             # Validate each single entry | ||||||
| @ -353,7 +337,7 @@ class Importer: | |||||||
|                 if entry.get_state(self._import) == BlueprintEntryDesiredState.ABSENT: |                 if entry.get_state(self._import) == BlueprintEntryDesiredState.ABSENT: | ||||||
|                     serializer = exc.serializer |                     serializer = exc.serializer | ||||||
|                 else: |                 else: | ||||||
|                     self.logger.warning(f"Entry invalid: {exc}", entry=entry, error=exc) |                     self.logger.warning(f"entry invalid: {exc}", entry=entry, error=exc) | ||||||
|                     if raise_errors: |                     if raise_errors: | ||||||
|                         raise exc |                         raise exc | ||||||
|                     return False |                     return False | ||||||
| @ -373,27 +357,27 @@ class Importer: | |||||||
|                     and state == BlueprintEntryDesiredState.CREATED |                     and state == BlueprintEntryDesiredState.CREATED | ||||||
|                 ): |                 ): | ||||||
|                     self.logger.debug( |                     self.logger.debug( | ||||||
|                         "Instance exists, skipping", |                         "instance exists, skipping", | ||||||
|                         model=model, |                         model=model, | ||||||
|                         instance=instance, |                         instance=instance, | ||||||
|                         pk=instance.pk, |                         pk=instance.pk, | ||||||
|                     ) |                     ) | ||||||
|                 else: |                 else: | ||||||
|                     instance = serializer.save() |                     instance = serializer.save() | ||||||
|                     self.logger.debug("Updated model", model=instance) |                     self.logger.debug("updated model", model=instance) | ||||||
|                 if "pk" in entry.identifiers: |                 if "pk" in entry.identifiers: | ||||||
|                     self.__pk_map[entry.identifiers["pk"]] = instance.pk |                     self.__pk_map[entry.identifiers["pk"]] = instance.pk | ||||||
|                 entry._state = BlueprintEntryState(instance) |                 entry._state = BlueprintEntryState(instance) | ||||||
|             elif state == BlueprintEntryDesiredState.ABSENT: |             elif state == BlueprintEntryDesiredState.ABSENT: | ||||||
|                 instance: Model | None = serializer.instance |                 instance: Optional[Model] = serializer.instance | ||||||
|                 if instance.pk: |                 if instance.pk: | ||||||
|                     instance.delete() |                     instance.delete() | ||||||
|                     self.logger.debug("Deleted model", mode=instance) |                     self.logger.debug("deleted model", mode=instance) | ||||||
|                     continue |                     continue | ||||||
|                 self.logger.debug("Entry to delete with no instance, skipping") |                 self.logger.debug("entry to delete with no instance, skipping") | ||||||
|         return True |         return True | ||||||
|  |  | ||||||
|     def validate(self, raise_validation_errors=False) -> tuple[bool, list[LogEvent]]: |     def validate(self, raise_validation_errors=False) -> tuple[bool, list[EventDict]]: | ||||||
|         """Validate loaded blueprint export, ensure all models are allowed |         """Validate loaded blueprint export, ensure all models are allowed | ||||||
|         and serializers have no errors""" |         and serializers have no errors""" | ||||||
|         self.logger.debug("Starting blueprint import validation") |         self.logger.debug("Starting blueprint import validation") | ||||||
| @ -407,7 +391,9 @@ class Importer: | |||||||
|         ): |         ): | ||||||
|             successful = self._apply_models(raise_errors=raise_validation_errors) |             successful = self._apply_models(raise_errors=raise_validation_errors) | ||||||
|             if not successful: |             if not successful: | ||||||
|                 self.logger.warning("Blueprint validation failed") |                 self.logger.debug("Blueprint validation failed") | ||||||
|  |         for log in logs: | ||||||
|  |             getattr(self.logger, log.get("log_level"))(**log) | ||||||
|         self.logger.debug("Finished blueprint import validation") |         self.logger.debug("Finished blueprint import validation") | ||||||
|         self._import = orig_import |         self._import = orig_import | ||||||
|         return successful, logs |         return successful, logs | ||||||
|  | |||||||
| @ -43,7 +43,7 @@ class ApplyBlueprintMetaSerializer(PassiveSerializer): | |||||||
|             LOGGER.info("Blueprint does not exist, but not required") |             LOGGER.info("Blueprint does not exist, but not required") | ||||||
|             return MetaResult() |             return MetaResult() | ||||||
|         LOGGER.debug("Applying blueprint from meta model", blueprint=self.blueprint_instance) |         LOGGER.debug("Applying blueprint from meta model", blueprint=self.blueprint_instance) | ||||||
|  |         # pylint: disable=no-value-for-parameter | ||||||
|         apply_blueprint(str(self.blueprint_instance.pk)) |         apply_blueprint(str(self.blueprint_instance.pk)) | ||||||
|         return MetaResult() |         return MetaResult() | ||||||
|  |  | ||||||
|  | |||||||
| @ -8,15 +8,15 @@ from rest_framework.serializers import Serializer | |||||||
| class BaseMetaModel(Model): | class BaseMetaModel(Model): | ||||||
|     """Base models""" |     """Base models""" | ||||||
|  |  | ||||||
|     class Meta: |  | ||||||
|         abstract = True |  | ||||||
|  |  | ||||||
|     @staticmethod |     @staticmethod | ||||||
|     def serializer() -> Serializer: |     def serializer() -> Serializer: | ||||||
|         """Serializer similar to SerializerModel, but as a static method since |         """Serializer similar to SerializerModel, but as a static method since | ||||||
|         this is an abstract model""" |         this is an abstract model""" | ||||||
|         raise NotImplementedError |         raise NotImplementedError | ||||||
|  |  | ||||||
|  |     class Meta: | ||||||
|  |         abstract = True | ||||||
|  |  | ||||||
|  |  | ||||||
| class MetaResult: | class MetaResult: | ||||||
|     """Result returned by Meta Models' serializers. Empty class but we can't return none as |     """Result returned by Meta Models' serializers. Empty class but we can't return none as | ||||||
|  | |||||||
| @ -4,6 +4,7 @@ from dataclasses import asdict, dataclass, field | |||||||
| from hashlib import sha512 | from hashlib import sha512 | ||||||
| from pathlib import Path | from pathlib import Path | ||||||
| from sys import platform | from sys import platform | ||||||
|  | from typing import Optional | ||||||
|  |  | ||||||
| from dacite.core import from_dict | from dacite.core import from_dict | ||||||
| from django.db import DatabaseError, InternalError, ProgrammingError | from django.db import DatabaseError, InternalError, ProgrammingError | ||||||
| @ -30,7 +31,6 @@ from authentik.blueprints.v1.common import BlueprintLoader, BlueprintMetadata, E | |||||||
| from authentik.blueprints.v1.importer import Importer | from authentik.blueprints.v1.importer import Importer | ||||||
| from authentik.blueprints.v1.labels import LABEL_AUTHENTIK_INSTANTIATE | from authentik.blueprints.v1.labels import LABEL_AUTHENTIK_INSTANTIATE | ||||||
| from authentik.blueprints.v1.oci import OCI_PREFIX | from authentik.blueprints.v1.oci import OCI_PREFIX | ||||||
| from authentik.events.logs import capture_logs |  | ||||||
| from authentik.events.models import TaskStatus | from authentik.events.models import TaskStatus | ||||||
| from authentik.events.system_tasks import SystemTask, prefill_task | from authentik.events.system_tasks import SystemTask, prefill_task | ||||||
| from authentik.events.utils import sanitize_dict | from authentik.events.utils import sanitize_dict | ||||||
| @ -50,14 +50,14 @@ class BlueprintFile: | |||||||
|     version: int |     version: int | ||||||
|     hash: str |     hash: str | ||||||
|     last_m: int |     last_m: int | ||||||
|     meta: BlueprintMetadata | None = field(default=None) |     meta: Optional[BlueprintMetadata] = field(default=None) | ||||||
|  |  | ||||||
|  |  | ||||||
| def start_blueprint_watcher(): | def start_blueprint_watcher(): | ||||||
|     """Start blueprint watcher, if it's not running already.""" |     """Start blueprint watcher, if it's not running already.""" | ||||||
|     # This function might be called twice since it's called on celery startup |     # This function might be called twice since it's called on celery startup | ||||||
|  |     # pylint: disable=global-statement | ||||||
|     global _file_watcher_started  # noqa: PLW0603 |     global _file_watcher_started | ||||||
|     if _file_watcher_started: |     if _file_watcher_started: | ||||||
|         return |         return | ||||||
|     observer = Observer() |     observer = Observer() | ||||||
| @ -126,7 +126,7 @@ def blueprints_find() -> list[BlueprintFile]: | |||||||
|         # Check if any part in the path starts with a dot and assume a hidden file |         # Check if any part in the path starts with a dot and assume a hidden file | ||||||
|         if any(part for part in path.parts if part.startswith(".")): |         if any(part for part in path.parts if part.startswith(".")): | ||||||
|             continue |             continue | ||||||
|         with open(path, encoding="utf-8") as blueprint_file: |         with open(path, "r", encoding="utf-8") as blueprint_file: | ||||||
|             try: |             try: | ||||||
|                 raw_blueprint = load(blueprint_file.read(), BlueprintLoader) |                 raw_blueprint = load(blueprint_file.read(), BlueprintLoader) | ||||||
|             except YAMLError as exc: |             except YAMLError as exc: | ||||||
| @ -150,7 +150,7 @@ def blueprints_find() -> list[BlueprintFile]: | |||||||
|     throws=(DatabaseError, ProgrammingError, InternalError), base=SystemTask, bind=True |     throws=(DatabaseError, ProgrammingError, InternalError), base=SystemTask, bind=True | ||||||
| ) | ) | ||||||
| @prefill_task | @prefill_task | ||||||
| def blueprints_discovery(self: SystemTask, path: str | None = None): | def blueprints_discovery(self: SystemTask, path: Optional[str] = None): | ||||||
|     """Find blueprints and check if they need to be created in the database""" |     """Find blueprints and check if they need to be created in the database""" | ||||||
|     count = 0 |     count = 0 | ||||||
|     for blueprint in blueprints_find(): |     for blueprint in blueprints_find(): | ||||||
| @ -197,7 +197,7 @@ def check_blueprint_v1_file(blueprint: BlueprintFile): | |||||||
| def apply_blueprint(self: SystemTask, instance_pk: str): | def apply_blueprint(self: SystemTask, instance_pk: str): | ||||||
|     """Apply single blueprint""" |     """Apply single blueprint""" | ||||||
|     self.save_on_success = False |     self.save_on_success = False | ||||||
|     instance: BlueprintInstance | None = None |     instance: Optional[BlueprintInstance] = None | ||||||
|     try: |     try: | ||||||
|         instance: BlueprintInstance = BlueprintInstance.objects.filter(pk=instance_pk).first() |         instance: BlueprintInstance = BlueprintInstance.objects.filter(pk=instance_pk).first() | ||||||
|         if not instance or not instance.enabled: |         if not instance or not instance.enabled: | ||||||
| @ -212,24 +212,23 @@ def apply_blueprint(self: SystemTask, instance_pk: str): | |||||||
|         if not valid: |         if not valid: | ||||||
|             instance.status = BlueprintInstanceStatus.ERROR |             instance.status = BlueprintInstanceStatus.ERROR | ||||||
|             instance.save() |             instance.save() | ||||||
|             self.set_status(TaskStatus.ERROR, *logs) |             self.set_status(TaskStatus.ERROR, *[x["event"] for x in logs]) | ||||||
|             return |             return | ||||||
|         with capture_logs() as logs: |  | ||||||
|         applied = importer.apply() |         applied = importer.apply() | ||||||
|         if not applied: |         if not applied: | ||||||
|             instance.status = BlueprintInstanceStatus.ERROR |             instance.status = BlueprintInstanceStatus.ERROR | ||||||
|             instance.save() |             instance.save() | ||||||
|                 self.set_status(TaskStatus.ERROR, *logs) |             self.set_status(TaskStatus.ERROR, "Failed to apply") | ||||||
|             return |             return | ||||||
|         instance.status = BlueprintInstanceStatus.SUCCESSFUL |         instance.status = BlueprintInstanceStatus.SUCCESSFUL | ||||||
|         instance.last_applied_hash = file_hash |         instance.last_applied_hash = file_hash | ||||||
|         instance.last_applied = now() |         instance.last_applied = now() | ||||||
|         self.set_status(TaskStatus.SUCCESSFUL) |         self.set_status(TaskStatus.SUCCESSFUL) | ||||||
|     except ( |     except ( | ||||||
|         OSError, |  | ||||||
|         DatabaseError, |         DatabaseError, | ||||||
|         ProgrammingError, |         ProgrammingError, | ||||||
|         InternalError, |         InternalError, | ||||||
|  |         IOError, | ||||||
|         BlueprintRetrievalFailed, |         BlueprintRetrievalFailed, | ||||||
|         EntryInvalidError, |         EntryInvalidError, | ||||||
|     ) as exc: |     ) as exc: | ||||||
|  | |||||||
| @ -1,6 +1,6 @@ | |||||||
| """Inject brand into current request""" | """Inject brand into current request""" | ||||||
|  |  | ||||||
| from collections.abc import Callable | from typing import Callable | ||||||
|  |  | ||||||
| from django.http.request import HttpRequest | from django.http.request import HttpRequest | ||||||
| from django.http.response import HttpResponse | from django.http.response import HttpResponse | ||||||
| @ -20,7 +20,7 @@ class BrandMiddleware: | |||||||
|     def __call__(self, request: HttpRequest) -> HttpResponse: |     def __call__(self, request: HttpRequest) -> HttpResponse: | ||||||
|         if not hasattr(request, "brand"): |         if not hasattr(request, "brand"): | ||||||
|             brand = get_brand_for_request(request) |             brand = get_brand_for_request(request) | ||||||
|             request.brand = brand |             setattr(request, "brand", brand) | ||||||
|             locale = brand.default_locale |             locale = brand.default_locale | ||||||
|             if locale != "": |             if locale != "": | ||||||
|                 activate(locale) |                 activate(locale) | ||||||
|  | |||||||
| @ -1,21 +0,0 @@ | |||||||
| # Generated by Django 5.0.4 on 2024-04-18 18:56 |  | ||||||
|  |  | ||||||
| from django.db import migrations, models |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class Migration(migrations.Migration): |  | ||||||
|  |  | ||||||
|     dependencies = [ |  | ||||||
|         ("authentik_brands", "0005_tenantuuid_to_branduuid"), |  | ||||||
|     ] |  | ||||||
|  |  | ||||||
|     operations = [ |  | ||||||
|         migrations.AddIndex( |  | ||||||
|             model_name="brand", |  | ||||||
|             index=models.Index(fields=["domain"], name="authentik_b_domain_b9b24a_idx"), |  | ||||||
|         ), |  | ||||||
|         migrations.AddIndex( |  | ||||||
|             model_name="brand", |  | ||||||
|             index=models.Index(fields=["default"], name="authentik_b_default_3ccf12_idx"), |  | ||||||
|         ), |  | ||||||
|     ] |  | ||||||
| @ -71,7 +71,7 @@ class Brand(SerializerModel): | |||||||
|         """Get default locale""" |         """Get default locale""" | ||||||
|         try: |         try: | ||||||
|             return self.attributes.get("settings", {}).get("locale", "") |             return self.attributes.get("settings", {}).get("locale", "") | ||||||
|  |         # pylint: disable=broad-except | ||||||
|         except Exception as exc: |         except Exception as exc: | ||||||
|             LOGGER.warning("Failed to get default locale", exc=exc) |             LOGGER.warning("Failed to get default locale", exc=exc) | ||||||
|             return "" |             return "" | ||||||
| @ -84,7 +84,3 @@ class Brand(SerializerModel): | |||||||
|     class Meta: |     class Meta: | ||||||
|         verbose_name = _("Brand") |         verbose_name = _("Brand") | ||||||
|         verbose_name_plural = _("Brands") |         verbose_name_plural = _("Brands") | ||||||
|         indexes = [ |  | ||||||
|             models.Index(fields=["domain"]), |  | ||||||
|             models.Index(fields=["default"]), |  | ||||||
|         ] |  | ||||||
|  | |||||||
| @ -9,7 +9,6 @@ from sentry_sdk.hub import Hub | |||||||
|  |  | ||||||
| from authentik import get_full_version | from authentik import get_full_version | ||||||
| from authentik.brands.models import Brand | from authentik.brands.models import Brand | ||||||
| from authentik.tenants.models import Tenant |  | ||||||
|  |  | ||||||
| _q_default = Q(default=True) | _q_default = Q(default=True) | ||||||
| DEFAULT_BRAND = Brand(domain="fallback") | DEFAULT_BRAND = Brand(domain="fallback") | ||||||
| @ -31,14 +30,13 @@ def get_brand_for_request(request: HttpRequest) -> Brand: | |||||||
| def context_processor(request: HttpRequest) -> dict[str, Any]: | def context_processor(request: HttpRequest) -> dict[str, Any]: | ||||||
|     """Context Processor that injects brand object into every template""" |     """Context Processor that injects brand object into every template""" | ||||||
|     brand = getattr(request, "brand", DEFAULT_BRAND) |     brand = getattr(request, "brand", DEFAULT_BRAND) | ||||||
|     tenant = getattr(request, "tenant", Tenant()) |  | ||||||
|     trace = "" |     trace = "" | ||||||
|     span = Hub.current.scope.span |     span = Hub.current.scope.span | ||||||
|     if span: |     if span: | ||||||
|         trace = span.to_traceparent() |         trace = span.to_traceparent() | ||||||
|     return { |     return { | ||||||
|         "brand": brand, |         "brand": brand, | ||||||
|         "footer_links": tenant.footer_links, |         "footer_links": request.tenant.footer_links, | ||||||
|         "sentry_trace": trace, |         "sentry_trace": trace, | ||||||
|         "version": get_full_version(), |         "version": get_full_version(), | ||||||
|     } |     } | ||||||
|  | |||||||
| @ -1,8 +1,8 @@ | |||||||
| """Application API Views""" | """Application API Views""" | ||||||
|  |  | ||||||
| from collections.abc import Iterator |  | ||||||
| from copy import copy | from copy import copy | ||||||
| from datetime import timedelta | from datetime import timedelta | ||||||
|  | from typing import Iterator, Optional | ||||||
|  |  | ||||||
| from django.core.cache import cache | from django.core.cache import cache | ||||||
| from django.db.models import QuerySet | from django.db.models import QuerySet | ||||||
| @ -20,15 +20,16 @@ from rest_framework.response import Response | |||||||
| from rest_framework.serializers import ModelSerializer | from rest_framework.serializers import ModelSerializer | ||||||
| from rest_framework.viewsets import ModelViewSet | from rest_framework.viewsets import ModelViewSet | ||||||
| from structlog.stdlib import get_logger | from structlog.stdlib import get_logger | ||||||
|  | from structlog.testing import capture_logs | ||||||
|  |  | ||||||
| from authentik.admin.api.metrics import CoordinateSerializer | from authentik.admin.api.metrics import CoordinateSerializer | ||||||
| from authentik.api.pagination import Pagination | from authentik.api.decorators import permission_required | ||||||
| from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT | from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT | ||||||
| from authentik.core.api.providers import ProviderSerializer | from authentik.core.api.providers import ProviderSerializer | ||||||
| from authentik.core.api.used_by import UsedByMixin | from authentik.core.api.used_by import UsedByMixin | ||||||
| from authentik.core.models import Application, User | from authentik.core.models import Application, User | ||||||
| from authentik.events.logs import LogEventSerializer, capture_logs |  | ||||||
| from authentik.events.models import EventAction | from authentik.events.models import EventAction | ||||||
|  | from authentik.events.utils import sanitize_dict | ||||||
| from authentik.lib.utils.file import ( | from authentik.lib.utils.file import ( | ||||||
|     FilePathSerializer, |     FilePathSerializer, | ||||||
|     FileUploadSerializer, |     FileUploadSerializer, | ||||||
| @ -37,19 +38,15 @@ from authentik.lib.utils.file import ( | |||||||
| ) | ) | ||||||
| from authentik.policies.api.exec import PolicyTestResultSerializer | from authentik.policies.api.exec import PolicyTestResultSerializer | ||||||
| from authentik.policies.engine import PolicyEngine | from authentik.policies.engine import PolicyEngine | ||||||
| from authentik.policies.types import CACHE_PREFIX, PolicyResult | from authentik.policies.types import PolicyResult | ||||||
| from authentik.rbac.decorators import permission_required |  | ||||||
| from authentik.rbac.filters import ObjectFilter | from authentik.rbac.filters import ObjectFilter | ||||||
|  |  | ||||||
| LOGGER = get_logger() | LOGGER = get_logger() | ||||||
|  |  | ||||||
|  |  | ||||||
| def user_app_cache_key(user_pk: str, page_number: int | None = None) -> str: | def user_app_cache_key(user_pk: str) -> str: | ||||||
|     """Cache key where application list for user is saved""" |     """Cache key where application list for user is saved""" | ||||||
|     key = f"{CACHE_PREFIX}/app_access/{user_pk}" |     return f"goauthentik.io/core/app_access/{user_pk}" | ||||||
|     if page_number: |  | ||||||
|         key += f"/{page_number}" |  | ||||||
|     return key |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class ApplicationSerializer(ModelSerializer): | class ApplicationSerializer(ModelSerializer): | ||||||
| @ -63,7 +60,7 @@ class ApplicationSerializer(ModelSerializer): | |||||||
|  |  | ||||||
|     meta_icon = ReadOnlyField(source="get_meta_icon") |     meta_icon = ReadOnlyField(source="get_meta_icon") | ||||||
|  |  | ||||||
|     def get_launch_url(self, app: Application) -> str | None: |     def get_launch_url(self, app: Application) -> Optional[str]: | ||||||
|         """Allow formatting of launch URL""" |         """Allow formatting of launch URL""" | ||||||
|         user = None |         user = None | ||||||
|         if "request" in self.context: |         if "request" in self.context: | ||||||
| @ -103,6 +100,7 @@ class ApplicationSerializer(ModelSerializer): | |||||||
| class ApplicationViewSet(UsedByMixin, ModelViewSet): | class ApplicationViewSet(UsedByMixin, ModelViewSet): | ||||||
|     """Application Viewset""" |     """Application Viewset""" | ||||||
|  |  | ||||||
|  |     # pylint: disable=no-member | ||||||
|     queryset = Application.objects.all().prefetch_related("provider") |     queryset = Application.objects.all().prefetch_related("provider") | ||||||
|     serializer_class = ApplicationSerializer |     serializer_class = ApplicationSerializer | ||||||
|     search_fields = [ |     search_fields = [ | ||||||
| @ -133,7 +131,7 @@ class ApplicationViewSet(UsedByMixin, ModelViewSet): | |||||||
|         return queryset |         return queryset | ||||||
|  |  | ||||||
|     def _get_allowed_applications( |     def _get_allowed_applications( | ||||||
|         self, pagined_apps: Iterator[Application], user: User | None = None |         self, pagined_apps: Iterator[Application], user: Optional[User] = None | ||||||
|     ) -> list[Application]: |     ) -> list[Application]: | ||||||
|         applications = [] |         applications = [] | ||||||
|         request = self.request._request |         request = self.request._request | ||||||
| @ -171,7 +169,7 @@ class ApplicationViewSet(UsedByMixin, ModelViewSet): | |||||||
|             try: |             try: | ||||||
|                 for_user = User.objects.filter(pk=request.query_params.get("for_user")).first() |                 for_user = User.objects.filter(pk=request.query_params.get("for_user")).first() | ||||||
|             except ValueError: |             except ValueError: | ||||||
|                 raise ValidationError({"for_user": "for_user must be numerical"}) from None |                 raise ValidationError({"for_user": "for_user must be numerical"}) | ||||||
|             if not for_user: |             if not for_user: | ||||||
|                 raise ValidationError({"for_user": "User not found"}) |                 raise ValidationError({"for_user": "User not found"}) | ||||||
|         engine = PolicyEngine(application, for_user, request) |         engine = PolicyEngine(application, for_user, request) | ||||||
| @ -185,9 +183,9 @@ class ApplicationViewSet(UsedByMixin, ModelViewSet): | |||||||
|         if request.user.is_superuser: |         if request.user.is_superuser: | ||||||
|             log_messages = [] |             log_messages = [] | ||||||
|             for log in logs: |             for log in logs: | ||||||
|                 if log.attributes.get("process", "") == "PolicyProcess": |                 if log.get("process", "") == "PolicyProcess": | ||||||
|                     continue |                     continue | ||||||
|                 log_messages.append(LogEventSerializer(log).data) |                 log_messages.append(sanitize_dict(log)) | ||||||
|             result.log_messages = log_messages |             result.log_messages = log_messages | ||||||
|             response = PolicyTestResultSerializer(result) |             response = PolicyTestResultSerializer(result) | ||||||
|         return Response(response.data) |         return Response(response.data) | ||||||
| @ -217,8 +215,7 @@ class ApplicationViewSet(UsedByMixin, ModelViewSet): | |||||||
|             return super().list(request) |             return super().list(request) | ||||||
|  |  | ||||||
|         queryset = self._filter_queryset_for_list(self.get_queryset()) |         queryset = self._filter_queryset_for_list(self.get_queryset()) | ||||||
|         paginator: Pagination = self.paginator |         pagined_apps = self.paginate_queryset(queryset) | ||||||
|         paginated_apps = paginator.paginate_queryset(queryset, request) |  | ||||||
|  |  | ||||||
|         if "for_user" in request.query_params: |         if "for_user" in request.query_params: | ||||||
|             try: |             try: | ||||||
| @ -232,22 +229,20 @@ class ApplicationViewSet(UsedByMixin, ModelViewSet): | |||||||
|                     raise ValidationError({"for_user": "User not found"}) |                     raise ValidationError({"for_user": "User not found"}) | ||||||
|             except ValueError as exc: |             except ValueError as exc: | ||||||
|                 raise ValidationError from exc |                 raise ValidationError from exc | ||||||
|             allowed_applications = self._get_allowed_applications(paginated_apps, user=for_user) |             allowed_applications = self._get_allowed_applications(pagined_apps, user=for_user) | ||||||
|             serializer = self.get_serializer(allowed_applications, many=True) |             serializer = self.get_serializer(allowed_applications, many=True) | ||||||
|             return self.get_paginated_response(serializer.data) |             return self.get_paginated_response(serializer.data) | ||||||
|  |  | ||||||
|         allowed_applications = [] |         allowed_applications = [] | ||||||
|         if not should_cache: |         if not should_cache: | ||||||
|             allowed_applications = self._get_allowed_applications(paginated_apps) |             allowed_applications = self._get_allowed_applications(pagined_apps) | ||||||
|         if should_cache: |         if should_cache: | ||||||
|             allowed_applications = cache.get( |             allowed_applications = cache.get(user_app_cache_key(self.request.user.pk)) | ||||||
|                 user_app_cache_key(self.request.user.pk, paginator.page.number) |  | ||||||
|             ) |  | ||||||
|             if not allowed_applications: |             if not allowed_applications: | ||||||
|                 LOGGER.debug("Caching allowed application list", page=paginator.page.number) |                 LOGGER.debug("Caching allowed application list") | ||||||
|                 allowed_applications = self._get_allowed_applications(paginated_apps) |                 allowed_applications = self._get_allowed_applications(pagined_apps) | ||||||
|                 cache.set( |                 cache.set( | ||||||
|                     user_app_cache_key(self.request.user.pk, paginator.page.number), |                     user_app_cache_key(self.request.user.pk), | ||||||
|                     allowed_applications, |                     allowed_applications, | ||||||
|                     timeout=86400, |                     timeout=86400, | ||||||
|                 ) |                 ) | ||||||
|  | |||||||
| @ -1,6 +1,6 @@ | |||||||
| """AuthenticatedSessions API Viewset""" | """AuthenticatedSessions API Viewset""" | ||||||
|  |  | ||||||
| from typing import TypedDict | from typing import Optional, TypedDict | ||||||
|  |  | ||||||
| from django_filters.rest_framework import DjangoFilterBackend | from django_filters.rest_framework import DjangoFilterBackend | ||||||
| from guardian.utils import get_anonymous_user | from guardian.utils import get_anonymous_user | ||||||
| @ -72,11 +72,11 @@ class AuthenticatedSessionSerializer(ModelSerializer): | |||||||
|         """Get parsed user agent""" |         """Get parsed user agent""" | ||||||
|         return user_agent_parser.Parse(instance.last_user_agent) |         return user_agent_parser.Parse(instance.last_user_agent) | ||||||
|  |  | ||||||
|     def get_geo_ip(self, instance: AuthenticatedSession) -> GeoIPDict | None:  # pragma: no cover |     def get_geo_ip(self, instance: AuthenticatedSession) -> Optional[GeoIPDict]:  # pragma: no cover | ||||||
|         """Get GeoIP Data""" |         """Get GeoIP Data""" | ||||||
|         return GEOIP_CONTEXT_PROCESSOR.city_dict(instance.last_ip) |         return GEOIP_CONTEXT_PROCESSOR.city_dict(instance.last_ip) | ||||||
|  |  | ||||||
|     def get_asn(self, instance: AuthenticatedSession) -> ASNDict | None:  # pragma: no cover |     def get_asn(self, instance: AuthenticatedSession) -> Optional[ASNDict]:  # pragma: no cover | ||||||
|         """Get ASN Data""" |         """Get ASN Data""" | ||||||
|         return ASN_CONTEXT_PROCESSOR.asn_dict(instance.last_ip) |         return ASN_CONTEXT_PROCESSOR.asn_dict(instance.last_ip) | ||||||
|  |  | ||||||
|  | |||||||
| @ -1,31 +1,25 @@ | |||||||
| """Groups API Viewset""" | """Groups API Viewset""" | ||||||
|  |  | ||||||
| from json import loads | from json import loads | ||||||
|  | from typing import Optional | ||||||
|  |  | ||||||
| from django.db.models import Prefetch |  | ||||||
| from django.http import Http404 | from django.http import Http404 | ||||||
| from django_filters.filters import CharFilter, ModelMultipleChoiceFilter | from django_filters.filters import CharFilter, ModelMultipleChoiceFilter | ||||||
| from django_filters.filterset import FilterSet | from django_filters.filterset import FilterSet | ||||||
| from drf_spectacular.utils import ( | from drf_spectacular.utils import OpenApiResponse, extend_schema | ||||||
|     OpenApiParameter, |  | ||||||
|     OpenApiResponse, |  | ||||||
|     extend_schema, |  | ||||||
|     extend_schema_field, |  | ||||||
| ) |  | ||||||
| from guardian.shortcuts import get_objects_for_user | from guardian.shortcuts import get_objects_for_user | ||||||
| from rest_framework.decorators import action | from rest_framework.decorators import action | ||||||
| from rest_framework.fields import CharField, IntegerField, SerializerMethodField | from rest_framework.fields import CharField, IntegerField | ||||||
| from rest_framework.request import Request | from rest_framework.request import Request | ||||||
| from rest_framework.response import Response | from rest_framework.response import Response | ||||||
| from rest_framework.serializers import ListSerializer, ModelSerializer, ValidationError | from rest_framework.serializers import ListSerializer, ModelSerializer, ValidationError | ||||||
| from rest_framework.validators import UniqueValidator |  | ||||||
| from rest_framework.viewsets import ModelViewSet | from rest_framework.viewsets import ModelViewSet | ||||||
|  |  | ||||||
|  | from authentik.api.decorators import permission_required | ||||||
| from authentik.core.api.used_by import UsedByMixin | from authentik.core.api.used_by import UsedByMixin | ||||||
| from authentik.core.api.utils import JSONDictField, PassiveSerializer | from authentik.core.api.utils import JSONDictField, PassiveSerializer | ||||||
| from authentik.core.models import Group, User | from authentik.core.models import Group, User | ||||||
| from authentik.rbac.api.roles import RoleSerializer | from authentik.rbac.api.roles import RoleSerializer | ||||||
| from authentik.rbac.decorators import permission_required |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class GroupMemberSerializer(ModelSerializer): | class GroupMemberSerializer(ModelSerializer): | ||||||
| @ -52,7 +46,9 @@ class GroupSerializer(ModelSerializer): | |||||||
|     """Group Serializer""" |     """Group Serializer""" | ||||||
|  |  | ||||||
|     attributes = JSONDictField(required=False) |     attributes = JSONDictField(required=False) | ||||||
|     users_obj = SerializerMethodField(allow_null=True) |     users_obj = ListSerializer( | ||||||
|  |         child=GroupMemberSerializer(), read_only=True, source="users", required=False | ||||||
|  |     ) | ||||||
|     roles_obj = ListSerializer( |     roles_obj = ListSerializer( | ||||||
|         child=RoleSerializer(), |         child=RoleSerializer(), | ||||||
|         read_only=True, |         read_only=True, | ||||||
| @ -63,20 +59,7 @@ class GroupSerializer(ModelSerializer): | |||||||
|  |  | ||||||
|     num_pk = IntegerField(read_only=True) |     num_pk = IntegerField(read_only=True) | ||||||
|  |  | ||||||
|     @property |     def validate_parent(self, parent: Optional[Group]): | ||||||
|     def _should_include_users(self) -> bool: |  | ||||||
|         request: Request = self.context.get("request", None) |  | ||||||
|         if not request: |  | ||||||
|             return True |  | ||||||
|         return str(request.query_params.get("include_users", "true")).lower() == "true" |  | ||||||
|  |  | ||||||
|     @extend_schema_field(GroupMemberSerializer(many=True)) |  | ||||||
|     def get_users_obj(self, instance: Group) -> list[GroupMemberSerializer] | None: |  | ||||||
|         if not self._should_include_users: |  | ||||||
|             return None |  | ||||||
|         return GroupMemberSerializer(instance.users, many=True).data |  | ||||||
|  |  | ||||||
|     def validate_parent(self, parent: Group | None): |  | ||||||
|         """Validate group parent (if set), ensuring the parent isn't itself""" |         """Validate group parent (if set), ensuring the parent isn't itself""" | ||||||
|         if not self.instance or not parent: |         if not self.instance or not parent: | ||||||
|             return parent |             return parent | ||||||
| @ -102,10 +85,7 @@ class GroupSerializer(ModelSerializer): | |||||||
|         extra_kwargs = { |         extra_kwargs = { | ||||||
|             "users": { |             "users": { | ||||||
|                 "default": list, |                 "default": list, | ||||||
|             }, |             } | ||||||
|             # TODO: This field isn't unique on the database which is hard to backport |  | ||||||
|             # hence we just validate the uniqueness here |  | ||||||
|             "name": {"validators": [UniqueValidator(Group.objects.all())]}, |  | ||||||
|         } |         } | ||||||
|  |  | ||||||
|  |  | ||||||
| @ -134,7 +114,7 @@ class GroupFilter(FilterSet): | |||||||
|         try: |         try: | ||||||
|             value = loads(value) |             value = loads(value) | ||||||
|         except ValueError: |         except ValueError: | ||||||
|             raise ValidationError(detail="filter: failed to parse JSON") from None |             raise ValidationError(detail="filter: failed to parse JSON") | ||||||
|         if not isinstance(value, dict): |         if not isinstance(value, dict): | ||||||
|             raise ValidationError(detail="filter: value must be key:value mapping") |             raise ValidationError(detail="filter: value must be key:value mapping") | ||||||
|         qs = {} |         qs = {} | ||||||
| @ -151,49 +131,23 @@ class GroupFilter(FilterSet): | |||||||
|         fields = ["name", "is_superuser", "members_by_pk", "attributes", "members_by_username"] |         fields = ["name", "is_superuser", "members_by_pk", "attributes", "members_by_username"] | ||||||
|  |  | ||||||
|  |  | ||||||
| class GroupViewSet(UsedByMixin, ModelViewSet): |  | ||||||
|     """Group Viewset""" |  | ||||||
|  |  | ||||||
| class UserAccountSerializer(PassiveSerializer): | class UserAccountSerializer(PassiveSerializer): | ||||||
|     """Account adding/removing operations""" |     """Account adding/removing operations""" | ||||||
|  |  | ||||||
|     pk = IntegerField(required=True) |     pk = IntegerField(required=True) | ||||||
|  |  | ||||||
|     queryset = Group.objects.none() |  | ||||||
|  | class GroupViewSet(UsedByMixin, ModelViewSet): | ||||||
|  |     """Group Viewset""" | ||||||
|  |  | ||||||
|  |     # pylint: disable=no-member | ||||||
|  |     queryset = Group.objects.all().select_related("parent").prefetch_related("users") | ||||||
|     serializer_class = GroupSerializer |     serializer_class = GroupSerializer | ||||||
|     search_fields = ["name", "is_superuser"] |     search_fields = ["name", "is_superuser"] | ||||||
|     filterset_class = GroupFilter |     filterset_class = GroupFilter | ||||||
|     ordering = ["name"] |     ordering = ["name"] | ||||||
|  |  | ||||||
|     def get_queryset(self): |     @permission_required(None, ["authentik_core.add_user"]) | ||||||
|         base_qs = Group.objects.all().select_related("parent").prefetch_related("roles") |  | ||||||
|  |  | ||||||
|         if self.serializer_class(context={"request": self.request})._should_include_users: |  | ||||||
|             base_qs = base_qs.prefetch_related("users") |  | ||||||
|         else: |  | ||||||
|             base_qs = base_qs.prefetch_related( |  | ||||||
|                 Prefetch("users", queryset=User.objects.all().only("id")) |  | ||||||
|             ) |  | ||||||
|  |  | ||||||
|         return base_qs |  | ||||||
|  |  | ||||||
|     @extend_schema( |  | ||||||
|         parameters=[ |  | ||||||
|             OpenApiParameter("include_users", bool, default=True), |  | ||||||
|         ] |  | ||||||
|     ) |  | ||||||
|     def list(self, request, *args, **kwargs): |  | ||||||
|         return super().list(request, *args, **kwargs) |  | ||||||
|  |  | ||||||
|     @extend_schema( |  | ||||||
|         parameters=[ |  | ||||||
|             OpenApiParameter("include_users", bool, default=True), |  | ||||||
|         ] |  | ||||||
|     ) |  | ||||||
|     def retrieve(self, request, *args, **kwargs): |  | ||||||
|         return super().retrieve(request, *args, **kwargs) |  | ||||||
|  |  | ||||||
|     @permission_required("authentik_core.add_user_to_group") |  | ||||||
|     @extend_schema( |     @extend_schema( | ||||||
|         request=UserAccountSerializer, |         request=UserAccountSerializer, | ||||||
|         responses={ |         responses={ | ||||||
| @ -201,13 +155,7 @@ class GroupViewSet(UsedByMixin, ModelViewSet): | |||||||
|             404: OpenApiResponse(description="User not found"), |             404: OpenApiResponse(description="User not found"), | ||||||
|         }, |         }, | ||||||
|     ) |     ) | ||||||
|     @action( |     @action(detail=True, methods=["POST"], pagination_class=None, filter_backends=[]) | ||||||
|         detail=True, |  | ||||||
|         methods=["POST"], |  | ||||||
|         pagination_class=None, |  | ||||||
|         filter_backends=[], |  | ||||||
|         permission_classes=[], |  | ||||||
|     ) |  | ||||||
|     def add_user(self, request: Request, pk: str) -> Response: |     def add_user(self, request: Request, pk: str) -> Response: | ||||||
|         """Add user to group""" |         """Add user to group""" | ||||||
|         group: Group = self.get_object() |         group: Group = self.get_object() | ||||||
| @ -223,7 +171,7 @@ class GroupViewSet(UsedByMixin, ModelViewSet): | |||||||
|         group.users.add(user) |         group.users.add(user) | ||||||
|         return Response(status=204) |         return Response(status=204) | ||||||
|  |  | ||||||
|     @permission_required("authentik_core.remove_user_from_group") |     @permission_required(None, ["authentik_core.add_user"]) | ||||||
|     @extend_schema( |     @extend_schema( | ||||||
|         request=UserAccountSerializer, |         request=UserAccountSerializer, | ||||||
|         responses={ |         responses={ | ||||||
| @ -231,13 +179,7 @@ class GroupViewSet(UsedByMixin, ModelViewSet): | |||||||
|             404: OpenApiResponse(description="User not found"), |             404: OpenApiResponse(description="User not found"), | ||||||
|         }, |         }, | ||||||
|     ) |     ) | ||||||
|     @action( |     @action(detail=True, methods=["POST"], pagination_class=None, filter_backends=[]) | ||||||
|         detail=True, |  | ||||||
|         methods=["POST"], |  | ||||||
|         pagination_class=None, |  | ||||||
|         filter_backends=[], |  | ||||||
|         permission_classes=[], |  | ||||||
|     ) |  | ||||||
|     def remove_user(self, request: Request, pk: str) -> Response: |     def remove_user(self, request: Request, pk: str) -> Response: | ||||||
|         """Add user to group""" |         """Add user to group""" | ||||||
|         group: Group = self.get_object() |         group: Group = self.get_object() | ||||||
|  | |||||||
| @ -1,79 +0,0 @@ | |||||||
| """API Utilities""" |  | ||||||
|  |  | ||||||
| from drf_spectacular.utils import extend_schema |  | ||||||
| from rest_framework.decorators import action |  | ||||||
| from rest_framework.fields import ( |  | ||||||
|     BooleanField, |  | ||||||
|     CharField, |  | ||||||
| ) |  | ||||||
| from rest_framework.request import Request |  | ||||||
| from rest_framework.response import Response |  | ||||||
|  |  | ||||||
| from authentik.core.api.utils import PassiveSerializer |  | ||||||
| from authentik.enterprise.apps import EnterpriseConfig |  | ||||||
| from authentik.lib.utils.reflection import all_subclasses |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class TypeCreateSerializer(PassiveSerializer): |  | ||||||
|     """Types of an object that can be created""" |  | ||||||
|  |  | ||||||
|     name = CharField(required=True) |  | ||||||
|     description = CharField(required=True) |  | ||||||
|     component = CharField(required=True) |  | ||||||
|     model_name = CharField(required=True) |  | ||||||
|  |  | ||||||
|     icon_url = CharField(required=False) |  | ||||||
|     requires_enterprise = BooleanField(default=False) |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class CreatableType: |  | ||||||
|     """Class to inherit from to mark a model as creatable, even if the model itself is marked |  | ||||||
|     as abstract""" |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class NonCreatableType: |  | ||||||
|     """Class to inherit from to mark a model as non-creatable even if it is not abstract""" |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class TypesMixin: |  | ||||||
|     """Mixin which adds an API endpoint to list all possible types that can be created""" |  | ||||||
|  |  | ||||||
|     @extend_schema(responses={200: TypeCreateSerializer(many=True)}) |  | ||||||
|     @action(detail=False, pagination_class=None, filter_backends=[]) |  | ||||||
|     def types(self, request: Request, additional: list[dict] | None = None) -> Response: |  | ||||||
|         """Get all creatable types""" |  | ||||||
|         data = [] |  | ||||||
|         for subclass in all_subclasses(self.queryset.model): |  | ||||||
|             instance = None |  | ||||||
|             if subclass._meta.abstract: |  | ||||||
|                 if not issubclass(subclass, CreatableType): |  | ||||||
|                     continue |  | ||||||
|                 # Circumvent the django protection for not being able to instantiate |  | ||||||
|                 # abstract models. We need a model instance to access .component |  | ||||||
|                 # and further down .icon_url |  | ||||||
|                 instance = subclass.__new__(subclass) |  | ||||||
|                 # Django re-sets abstract = False so we need to override that |  | ||||||
|                 instance.Meta.abstract = True |  | ||||||
|             else: |  | ||||||
|                 if issubclass(subclass, NonCreatableType): |  | ||||||
|                     continue |  | ||||||
|                 instance = subclass() |  | ||||||
|             try: |  | ||||||
|                 data.append( |  | ||||||
|                     { |  | ||||||
|                         "name": subclass._meta.verbose_name, |  | ||||||
|                         "description": subclass.__doc__, |  | ||||||
|                         "component": instance.component, |  | ||||||
|                         "model_name": subclass._meta.model_name, |  | ||||||
|                         "icon_url": getattr(instance, "icon_url", None), |  | ||||||
|                         "requires_enterprise": isinstance( |  | ||||||
|                             subclass._meta.app_config, EnterpriseConfig |  | ||||||
|                         ), |  | ||||||
|                     } |  | ||||||
|                 ) |  | ||||||
|             except NotImplementedError: |  | ||||||
|                 continue |  | ||||||
|         if additional: |  | ||||||
|             data.extend(additional) |  | ||||||
|         data = sorted(data, key=lambda x: x["name"]) |  | ||||||
|         return Response(TypeCreateSerializer(data, many=True).data) |  | ||||||
| @ -9,24 +9,20 @@ from rest_framework import mixins | |||||||
| from rest_framework.decorators import action | from rest_framework.decorators import action | ||||||
| from rest_framework.exceptions import PermissionDenied | from rest_framework.exceptions import PermissionDenied | ||||||
| from rest_framework.fields import BooleanField, CharField | from rest_framework.fields import BooleanField, CharField | ||||||
| from rest_framework.relations import PrimaryKeyRelatedField |  | ||||||
| from rest_framework.request import Request | from rest_framework.request import Request | ||||||
| from rest_framework.response import Response | from rest_framework.response import Response | ||||||
| from rest_framework.serializers import ModelSerializer, SerializerMethodField | from rest_framework.serializers import ModelSerializer, SerializerMethodField | ||||||
| from rest_framework.viewsets import GenericViewSet | from rest_framework.viewsets import GenericViewSet | ||||||
| 
 | 
 | ||||||
|  | from authentik.api.decorators import permission_required | ||||||
| from authentik.blueprints.api import ManagedSerializer | from authentik.blueprints.api import ManagedSerializer | ||||||
| from authentik.core.api.object_types import TypesMixin |  | ||||||
| from authentik.core.api.used_by import UsedByMixin | from authentik.core.api.used_by import UsedByMixin | ||||||
| from authentik.core.api.utils import ( | from authentik.core.api.utils import MetaNameSerializer, PassiveSerializer, TypeCreateSerializer | ||||||
|     MetaNameSerializer, |  | ||||||
|     PassiveSerializer, |  | ||||||
| ) |  | ||||||
| from authentik.core.expression.evaluator import PropertyMappingEvaluator | from authentik.core.expression.evaluator import PropertyMappingEvaluator | ||||||
| from authentik.core.models import Group, PropertyMapping, User | from authentik.core.models import PropertyMapping | ||||||
| from authentik.events.utils import sanitize_item | from authentik.events.utils import sanitize_item | ||||||
|  | from authentik.lib.utils.reflection import all_subclasses | ||||||
| from authentik.policies.api.exec import PolicyTestSerializer | from authentik.policies.api.exec import PolicyTestSerializer | ||||||
| from authentik.rbac.decorators import permission_required |  | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
| class PropertyMappingTestResultSerializer(PassiveSerializer): | class PropertyMappingTestResultSerializer(PassiveSerializer): | ||||||
| @ -68,7 +64,6 @@ class PropertyMappingSerializer(ManagedSerializer, ModelSerializer, MetaNameSeri | |||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
| class PropertyMappingViewSet( | class PropertyMappingViewSet( | ||||||
|     TypesMixin, |  | ||||||
|     mixins.RetrieveModelMixin, |     mixins.RetrieveModelMixin, | ||||||
|     mixins.DestroyModelMixin, |     mixins.DestroyModelMixin, | ||||||
|     UsedByMixin, |     UsedByMixin, | ||||||
| @ -77,15 +72,7 @@ class PropertyMappingViewSet( | |||||||
| ): | ): | ||||||
|     """PropertyMapping Viewset""" |     """PropertyMapping Viewset""" | ||||||
| 
 | 
 | ||||||
|     class PropertyMappingTestSerializer(PolicyTestSerializer): |     queryset = PropertyMapping.objects.none() | ||||||
|         """Test property mapping execution for a user/group with context""" |  | ||||||
| 
 |  | ||||||
|         user = PrimaryKeyRelatedField(queryset=User.objects.all(), required=False, allow_null=True) |  | ||||||
|         group = PrimaryKeyRelatedField( |  | ||||||
|             queryset=Group.objects.all(), required=False, allow_null=True |  | ||||||
|         ) |  | ||||||
| 
 |  | ||||||
|     queryset = PropertyMapping.objects.select_subclasses() |  | ||||||
|     serializer_class = PropertyMappingSerializer |     serializer_class = PropertyMappingSerializer | ||||||
|     search_fields = [ |     search_fields = [ | ||||||
|         "name", |         "name", | ||||||
| @ -93,9 +80,29 @@ class PropertyMappingViewSet( | |||||||
|     filterset_fields = {"managed": ["isnull"]} |     filterset_fields = {"managed": ["isnull"]} | ||||||
|     ordering = ["name"] |     ordering = ["name"] | ||||||
| 
 | 
 | ||||||
|  |     def get_queryset(self):  # pragma: no cover | ||||||
|  |         return PropertyMapping.objects.select_subclasses() | ||||||
|  | 
 | ||||||
|  |     @extend_schema(responses={200: TypeCreateSerializer(many=True)}) | ||||||
|  |     @action(detail=False, pagination_class=None, filter_backends=[]) | ||||||
|  |     def types(self, request: Request) -> Response: | ||||||
|  |         """Get all creatable property-mapping types""" | ||||||
|  |         data = [] | ||||||
|  |         for subclass in all_subclasses(self.queryset.model): | ||||||
|  |             subclass: PropertyMapping | ||||||
|  |             data.append( | ||||||
|  |                 { | ||||||
|  |                     "name": subclass._meta.verbose_name, | ||||||
|  |                     "description": subclass.__doc__, | ||||||
|  |                     "component": subclass().component, | ||||||
|  |                     "model_name": subclass._meta.model_name, | ||||||
|  |                 } | ||||||
|  |             ) | ||||||
|  |         return Response(TypeCreateSerializer(data, many=True).data) | ||||||
|  | 
 | ||||||
|     @permission_required("authentik_core.view_propertymapping") |     @permission_required("authentik_core.view_propertymapping") | ||||||
|     @extend_schema( |     @extend_schema( | ||||||
|         request=PropertyMappingTestSerializer(), |         request=PolicyTestSerializer(), | ||||||
|         responses={ |         responses={ | ||||||
|             200: PropertyMappingTestResultSerializer, |             200: PropertyMappingTestResultSerializer, | ||||||
|             400: OpenApiResponse(description="Invalid parameters"), |             400: OpenApiResponse(description="Invalid parameters"), | ||||||
| @ -113,43 +120,33 @@ class PropertyMappingViewSet( | |||||||
|         """Test Property Mapping""" |         """Test Property Mapping""" | ||||||
|         _mapping: PropertyMapping = self.get_object() |         _mapping: PropertyMapping = self.get_object() | ||||||
|         # Use `get_subclass` to get correct class and correct `.evaluate` implementation |         # Use `get_subclass` to get correct class and correct `.evaluate` implementation | ||||||
|         mapping: PropertyMapping = PropertyMapping.objects.get_subclass(pk=_mapping.pk) |         mapping = PropertyMapping.objects.get_subclass(pk=_mapping.pk) | ||||||
|         # FIXME: when we separate policy mappings between ones for sources |         # FIXME: when we separate policy mappings between ones for sources | ||||||
|         # and ones for providers, we need to make the user field optional for the source mapping |         # and ones for providers, we need to make the user field optional for the source mapping | ||||||
|         test_params = self.PropertyMappingTestSerializer(data=request.data) |         test_params = PolicyTestSerializer(data=request.data) | ||||||
|         if not test_params.is_valid(): |         if not test_params.is_valid(): | ||||||
|             return Response(test_params.errors, status=400) |             return Response(test_params.errors, status=400) | ||||||
| 
 | 
 | ||||||
|         format_result = str(request.GET.get("format_result", "false")).lower() == "true" |         format_result = str(request.GET.get("format_result", "false")).lower() == "true" | ||||||
| 
 | 
 | ||||||
|         context: dict = test_params.validated_data.get("context", {}) |  | ||||||
|         context.setdefault("user", None) |  | ||||||
| 
 |  | ||||||
|         if user := test_params.validated_data.get("user"): |  | ||||||
|         # User permission check, only allow mapping testing for users that are readable |         # User permission check, only allow mapping testing for users that are readable | ||||||
|         users = get_objects_for_user(request.user, "authentik_core.view_user").filter( |         users = get_objects_for_user(request.user, "authentik_core.view_user").filter( | ||||||
|                 pk=user.pk |             pk=test_params.validated_data["user"].pk | ||||||
|         ) |         ) | ||||||
|         if not users.exists(): |         if not users.exists(): | ||||||
|             raise PermissionDenied() |             raise PermissionDenied() | ||||||
|             context["user"] = user |  | ||||||
|         if group := test_params.validated_data.get("group"): |  | ||||||
|             # Group permission check, only allow mapping testing for groups that are readable |  | ||||||
|             groups = get_objects_for_user(request.user, "authentik_core.view_group").filter( |  | ||||||
|                 pk=group.pk |  | ||||||
|             ) |  | ||||||
|             if not groups.exists(): |  | ||||||
|                 raise PermissionDenied() |  | ||||||
|             context["group"] = group |  | ||||||
|         context["request"] = self.request |  | ||||||
| 
 | 
 | ||||||
|         response_data = {"successful": True, "result": ""} |         response_data = {"successful": True, "result": ""} | ||||||
|         try: |         try: | ||||||
|             result = mapping.evaluate(**context) |             result = mapping.evaluate( | ||||||
|  |                 users.first(), | ||||||
|  |                 self.request, | ||||||
|  |                 **test_params.validated_data.get("context", {}), | ||||||
|  |             ) | ||||||
|             response_data["result"] = dumps( |             response_data["result"] = dumps( | ||||||
|                 sanitize_item(result), indent=(4 if format_result else None) |                 sanitize_item(result), indent=(4 if format_result else None) | ||||||
|             ) |             ) | ||||||
|         except Exception as exc: |         except Exception as exc:  # pylint: disable=broad-except | ||||||
|             response_data["result"] = str(exc) |             response_data["result"] = str(exc) | ||||||
|             response_data["successful"] = False |             response_data["successful"] = False | ||||||
|         response = PropertyMappingTestResultSerializer(response_data) |         response = PropertyMappingTestResultSerializer(response_data) | ||||||
| @ -5,15 +5,20 @@ from django.db.models.query import Q | |||||||
| from django.utils.translation import gettext_lazy as _ | from django.utils.translation import gettext_lazy as _ | ||||||
| from django_filters.filters import BooleanFilter | from django_filters.filters import BooleanFilter | ||||||
| from django_filters.filterset import FilterSet | from django_filters.filterset import FilterSet | ||||||
|  | from drf_spectacular.utils import extend_schema | ||||||
| from rest_framework import mixins | from rest_framework import mixins | ||||||
|  | from rest_framework.decorators import action | ||||||
| from rest_framework.fields import ReadOnlyField | from rest_framework.fields import ReadOnlyField | ||||||
|  | from rest_framework.request import Request | ||||||
|  | from rest_framework.response import Response | ||||||
| from rest_framework.serializers import ModelSerializer, SerializerMethodField | from rest_framework.serializers import ModelSerializer, SerializerMethodField | ||||||
| from rest_framework.viewsets import GenericViewSet | from rest_framework.viewsets import GenericViewSet | ||||||
|  |  | ||||||
| from authentik.core.api.object_types import TypesMixin |  | ||||||
| from authentik.core.api.used_by import UsedByMixin | from authentik.core.api.used_by import UsedByMixin | ||||||
| from authentik.core.api.utils import MetaNameSerializer | from authentik.core.api.utils import MetaNameSerializer, TypeCreateSerializer | ||||||
| from authentik.core.models import Provider | from authentik.core.models import Provider | ||||||
|  | from authentik.enterprise.apps import EnterpriseConfig | ||||||
|  | from authentik.lib.utils.reflection import all_subclasses | ||||||
|  |  | ||||||
|  |  | ||||||
| class ProviderSerializer(ModelSerializer, MetaNameSerializer): | class ProviderSerializer(ModelSerializer, MetaNameSerializer): | ||||||
| @ -58,12 +63,8 @@ class ProviderFilter(FilterSet): | |||||||
|     """Filter for providers""" |     """Filter for providers""" | ||||||
|  |  | ||||||
|     application__isnull = BooleanFilter(method="filter_application__isnull") |     application__isnull = BooleanFilter(method="filter_application__isnull") | ||||||
|     backchannel = BooleanFilter( |     backchannel_only = BooleanFilter( | ||||||
|         method="filter_backchannel", |         method="filter_backchannel_only", | ||||||
|         label=_( |  | ||||||
|             "When not set all providers are returned. When set to true, only backchannel " |  | ||||||
|             "providers are returned. When set to false, backchannel providers are excluded" |  | ||||||
|         ), |  | ||||||
|     ) |     ) | ||||||
|  |  | ||||||
|     def filter_application__isnull(self, queryset: QuerySet, name, value): |     def filter_application__isnull(self, queryset: QuerySet, name, value): | ||||||
| @ -74,14 +75,12 @@ class ProviderFilter(FilterSet): | |||||||
|             | Q(application__isnull=value) |             | Q(application__isnull=value) | ||||||
|         ) |         ) | ||||||
|  |  | ||||||
|     def filter_backchannel(self, queryset: QuerySet, name, value): |     def filter_backchannel_only(self, queryset: QuerySet, name, value): | ||||||
|         """By default all providers are returned. When set to true, only backchannel providers are |         """Only return backchannel providers""" | ||||||
|         returned. When set to false, backchannel providers are excluded""" |  | ||||||
|         return queryset.filter(is_backchannel=value) |         return queryset.filter(is_backchannel=value) | ||||||
|  |  | ||||||
|  |  | ||||||
| class ProviderViewSet( | class ProviderViewSet( | ||||||
|     TypesMixin, |  | ||||||
|     mixins.RetrieveModelMixin, |     mixins.RetrieveModelMixin, | ||||||
|     mixins.DestroyModelMixin, |     mixins.DestroyModelMixin, | ||||||
|     UsedByMixin, |     UsedByMixin, | ||||||
| @ -100,3 +99,31 @@ class ProviderViewSet( | |||||||
|  |  | ||||||
|     def get_queryset(self):  # pragma: no cover |     def get_queryset(self):  # pragma: no cover | ||||||
|         return Provider.objects.select_subclasses() |         return Provider.objects.select_subclasses() | ||||||
|  |  | ||||||
|  |     @extend_schema(responses={200: TypeCreateSerializer(many=True)}) | ||||||
|  |     @action(detail=False, pagination_class=None, filter_backends=[]) | ||||||
|  |     def types(self, request: Request) -> Response: | ||||||
|  |         """Get all creatable provider types""" | ||||||
|  |         data = [] | ||||||
|  |         for subclass in all_subclasses(self.queryset.model): | ||||||
|  |             subclass: Provider | ||||||
|  |             if subclass._meta.abstract: | ||||||
|  |                 continue | ||||||
|  |             data.append( | ||||||
|  |                 { | ||||||
|  |                     "name": subclass._meta.verbose_name, | ||||||
|  |                     "description": subclass.__doc__, | ||||||
|  |                     "component": subclass().component, | ||||||
|  |                     "model_name": subclass._meta.model_name, | ||||||
|  |                     "requires_enterprise": isinstance(subclass._meta.app_config, EnterpriseConfig), | ||||||
|  |                 } | ||||||
|  |             ) | ||||||
|  |         data.append( | ||||||
|  |             { | ||||||
|  |                 "name": _("SAML Provider from Metadata"), | ||||||
|  |                 "description": _("Create a SAML Provider by importing its Metadata."), | ||||||
|  |                 "component": "ak-provider-saml-import-form", | ||||||
|  |                 "model_name": "", | ||||||
|  |             } | ||||||
|  |         ) | ||||||
|  |         return Response(TypeCreateSerializer(data, many=True).data) | ||||||
|  | |||||||
| @ -1,6 +1,6 @@ | |||||||
| """Source API Views""" | """Source API Views""" | ||||||
|  |  | ||||||
| from collections.abc import Iterable | from typing import Iterable | ||||||
|  |  | ||||||
| from django_filters.rest_framework import DjangoFilterBackend | from django_filters.rest_framework import DjangoFilterBackend | ||||||
| from drf_spectacular.utils import OpenApiResponse, extend_schema | from drf_spectacular.utils import OpenApiResponse, extend_schema | ||||||
| @ -16,10 +16,10 @@ from rest_framework.viewsets import GenericViewSet | |||||||
| from structlog.stdlib import get_logger | from structlog.stdlib import get_logger | ||||||
|  |  | ||||||
| from authentik.api.authorization import OwnerFilter, OwnerSuperuserPermissions | from authentik.api.authorization import OwnerFilter, OwnerSuperuserPermissions | ||||||
|  | from authentik.api.decorators import permission_required | ||||||
| from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT | from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT | ||||||
| from authentik.core.api.object_types import TypesMixin |  | ||||||
| from authentik.core.api.used_by import UsedByMixin | from authentik.core.api.used_by import UsedByMixin | ||||||
| from authentik.core.api.utils import MetaNameSerializer | from authentik.core.api.utils import MetaNameSerializer, TypeCreateSerializer | ||||||
| from authentik.core.models import Source, UserSourceConnection | from authentik.core.models import Source, UserSourceConnection | ||||||
| from authentik.core.types import UserSettingSerializer | from authentik.core.types import UserSettingSerializer | ||||||
| from authentik.lib.utils.file import ( | from authentik.lib.utils.file import ( | ||||||
| @ -28,8 +28,8 @@ from authentik.lib.utils.file import ( | |||||||
|     set_file, |     set_file, | ||||||
|     set_file_url, |     set_file_url, | ||||||
| ) | ) | ||||||
|  | from authentik.lib.utils.reflection import all_subclasses | ||||||
| from authentik.policies.engine import PolicyEngine | from authentik.policies.engine import PolicyEngine | ||||||
| from authentik.rbac.decorators import permission_required |  | ||||||
|  |  | ||||||
| LOGGER = get_logger() | LOGGER = get_logger() | ||||||
|  |  | ||||||
| @ -74,7 +74,6 @@ class SourceSerializer(ModelSerializer, MetaNameSerializer): | |||||||
|  |  | ||||||
|  |  | ||||||
| class SourceViewSet( | class SourceViewSet( | ||||||
|     TypesMixin, |  | ||||||
|     mixins.RetrieveModelMixin, |     mixins.RetrieveModelMixin, | ||||||
|     mixins.DestroyModelMixin, |     mixins.DestroyModelMixin, | ||||||
|     UsedByMixin, |     UsedByMixin, | ||||||
| @ -133,6 +132,30 @@ class SourceViewSet( | |||||||
|         source: Source = self.get_object() |         source: Source = self.get_object() | ||||||
|         return set_file_url(request, source, "icon") |         return set_file_url(request, source, "icon") | ||||||
|  |  | ||||||
|  |     @extend_schema(responses={200: TypeCreateSerializer(many=True)}) | ||||||
|  |     @action(detail=False, pagination_class=None, filter_backends=[]) | ||||||
|  |     def types(self, request: Request) -> Response: | ||||||
|  |         """Get all creatable source types""" | ||||||
|  |         data = [] | ||||||
|  |         for subclass in all_subclasses(self.queryset.model): | ||||||
|  |             subclass: Source | ||||||
|  |             component = "" | ||||||
|  |             if len(subclass.__subclasses__()) > 0: | ||||||
|  |                 continue | ||||||
|  |             if subclass._meta.abstract: | ||||||
|  |                 component = subclass.__bases__[0]().component | ||||||
|  |             else: | ||||||
|  |                 component = subclass().component | ||||||
|  |             data.append( | ||||||
|  |                 { | ||||||
|  |                     "name": subclass._meta.verbose_name, | ||||||
|  |                     "description": subclass.__doc__, | ||||||
|  |                     "component": component, | ||||||
|  |                     "model_name": subclass._meta.model_name, | ||||||
|  |                 } | ||||||
|  |             ) | ||||||
|  |         return Response(TypeCreateSerializer(data, many=True).data) | ||||||
|  |  | ||||||
|     @extend_schema(responses={200: UserSettingSerializer(many=True)}) |     @extend_schema(responses={200: UserSettingSerializer(many=True)}) | ||||||
|     @action(detail=False, pagination_class=None, filter_backends=[]) |     @action(detail=False, pagination_class=None, filter_backends=[]) | ||||||
|     def user_settings(self, request: Request) -> Response: |     def user_settings(self, request: Request) -> Response: | ||||||
|  | |||||||
| @ -2,7 +2,6 @@ | |||||||
|  |  | ||||||
| from typing import Any | from typing import Any | ||||||
|  |  | ||||||
| from django.utils.timezone import now |  | ||||||
| from django_filters.rest_framework import DjangoFilterBackend | from django_filters.rest_framework import DjangoFilterBackend | ||||||
| from drf_spectacular.utils import OpenApiResponse, extend_schema, inline_serializer | from drf_spectacular.utils import OpenApiResponse, extend_schema, inline_serializer | ||||||
| from guardian.shortcuts import assign_perm, get_anonymous_user | from guardian.shortcuts import assign_perm, get_anonymous_user | ||||||
| @ -16,23 +15,15 @@ from rest_framework.serializers import ModelSerializer | |||||||
| from rest_framework.viewsets import ModelViewSet | from rest_framework.viewsets import ModelViewSet | ||||||
|  |  | ||||||
| from authentik.api.authorization import OwnerSuperuserPermissions | from authentik.api.authorization import OwnerSuperuserPermissions | ||||||
|  | from authentik.api.decorators import permission_required | ||||||
| from authentik.blueprints.api import ManagedSerializer | from authentik.blueprints.api import ManagedSerializer | ||||||
| from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT | from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT | ||||||
| from authentik.core.api.used_by import UsedByMixin | from authentik.core.api.used_by import UsedByMixin | ||||||
| from authentik.core.api.users import UserSerializer | from authentik.core.api.users import UserSerializer | ||||||
| from authentik.core.api.utils import PassiveSerializer | from authentik.core.api.utils import PassiveSerializer | ||||||
| from authentik.core.models import ( | from authentik.core.models import USER_ATTRIBUTE_TOKEN_EXPIRING, Token, TokenIntents | ||||||
|     USER_ATTRIBUTE_TOKEN_EXPIRING, |  | ||||||
|     USER_ATTRIBUTE_TOKEN_MAXIMUM_LIFETIME, |  | ||||||
|     Token, |  | ||||||
|     TokenIntents, |  | ||||||
|     User, |  | ||||||
|     default_token_duration, |  | ||||||
| ) |  | ||||||
| from authentik.events.models import Event, EventAction | from authentik.events.models import Event, EventAction | ||||||
| from authentik.events.utils import model_to_dict | from authentik.events.utils import model_to_dict | ||||||
| from authentik.lib.utils.time import timedelta_from_string |  | ||||||
| from authentik.rbac.decorators import permission_required |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class TokenSerializer(ManagedSerializer, ModelSerializer): | class TokenSerializer(ManagedSerializer, ModelSerializer): | ||||||
| @ -58,32 +49,6 @@ class TokenSerializer(ManagedSerializer, ModelSerializer): | |||||||
|         attrs.setdefault("intent", TokenIntents.INTENT_API) |         attrs.setdefault("intent", TokenIntents.INTENT_API) | ||||||
|         if attrs.get("intent") not in [TokenIntents.INTENT_API, TokenIntents.INTENT_APP_PASSWORD]: |         if attrs.get("intent") not in [TokenIntents.INTENT_API, TokenIntents.INTENT_APP_PASSWORD]: | ||||||
|             raise ValidationError({"intent": f"Invalid intent {attrs.get('intent')}"}) |             raise ValidationError({"intent": f"Invalid intent {attrs.get('intent')}"}) | ||||||
|  |  | ||||||
|         if attrs.get("intent") == TokenIntents.INTENT_APP_PASSWORD: |  | ||||||
|             # user IS in attrs |  | ||||||
|             user: User = attrs.get("user") |  | ||||||
|             max_token_lifetime = user.group_attributes(request).get( |  | ||||||
|                 USER_ATTRIBUTE_TOKEN_MAXIMUM_LIFETIME, |  | ||||||
|             ) |  | ||||||
|             max_token_lifetime_dt = default_token_duration() |  | ||||||
|             if max_token_lifetime is not None: |  | ||||||
|                 try: |  | ||||||
|                     max_token_lifetime_dt = now() + timedelta_from_string(max_token_lifetime) |  | ||||||
|                 except ValueError: |  | ||||||
|                     pass |  | ||||||
|  |  | ||||||
|             if "expires" in attrs and attrs.get("expires") > max_token_lifetime_dt: |  | ||||||
|                 raise ValidationError( |  | ||||||
|                     { |  | ||||||
|                         "expires": ( |  | ||||||
|                             f"Token expires exceeds maximum lifetime ({max_token_lifetime_dt} UTC)." |  | ||||||
|                         ) |  | ||||||
|                     } |  | ||||||
|                 ) |  | ||||||
|         elif attrs.get("intent") == TokenIntents.INTENT_API: |  | ||||||
|             # For API tokens, expires cannot be overridden |  | ||||||
|             attrs["expires"] = default_token_duration() |  | ||||||
|  |  | ||||||
|         return attrs |         return attrs | ||||||
|  |  | ||||||
|     class Meta: |     class Meta: | ||||||
|  | |||||||
| @ -65,7 +65,7 @@ class TransactionApplicationSerializer(PassiveSerializer): | |||||||
|                 raise ValidationError("Invalid provider model") |                 raise ValidationError("Invalid provider model") | ||||||
|             self._provider_model = model |             self._provider_model = model | ||||||
|         except LookupError: |         except LookupError: | ||||||
|             raise ValidationError("Invalid provider model") from None |             raise ValidationError("Invalid provider model") | ||||||
|         return fq_model_name |         return fq_model_name | ||||||
|  |  | ||||||
|     def validate(self, attrs: dict) -> dict: |     def validate(self, attrs: dict) -> dict: | ||||||
| @ -106,7 +106,7 @@ class TransactionApplicationSerializer(PassiveSerializer): | |||||||
|                 { |                 { | ||||||
|                     exc.entry_id: exc.validation_error.detail, |                     exc.entry_id: exc.validation_error.detail, | ||||||
|                 } |                 } | ||||||
|             ) from None |             ) | ||||||
|         return blueprint |         return blueprint | ||||||
|  |  | ||||||
|  |  | ||||||
|  | |||||||
| @ -39,12 +39,12 @@ def get_delete_action(manager: Manager) -> str: | |||||||
|     """Get the delete action from the Foreign key, falls back to cascade""" |     """Get the delete action from the Foreign key, falls back to cascade""" | ||||||
|     if hasattr(manager, "field"): |     if hasattr(manager, "field"): | ||||||
|         if manager.field.remote_field.on_delete.__name__ == SET_NULL.__name__: |         if manager.field.remote_field.on_delete.__name__ == SET_NULL.__name__: | ||||||
|             return DeleteAction.SET_NULL.value |             return DeleteAction.SET_NULL.name | ||||||
|         if manager.field.remote_field.on_delete.__name__ == SET_DEFAULT.__name__: |         if manager.field.remote_field.on_delete.__name__ == SET_DEFAULT.__name__: | ||||||
|             return DeleteAction.SET_DEFAULT.value |             return DeleteAction.SET_DEFAULT.name | ||||||
|     if hasattr(manager, "source_field"): |     if hasattr(manager, "source_field"): | ||||||
|         return DeleteAction.CASCADE_MANY.value |         return DeleteAction.CASCADE_MANY.name | ||||||
|     return DeleteAction.CASCADE.value |     return DeleteAction.CASCADE.name | ||||||
|  |  | ||||||
|  |  | ||||||
| class UsedByMixin: | class UsedByMixin: | ||||||
| @ -54,6 +54,7 @@ class UsedByMixin: | |||||||
|         responses={200: UsedBySerializer(many=True)}, |         responses={200: UsedBySerializer(many=True)}, | ||||||
|     ) |     ) | ||||||
|     @action(detail=True, pagination_class=None, filter_backends=[]) |     @action(detail=True, pagination_class=None, filter_backends=[]) | ||||||
|  |     # pylint: disable=too-many-locals | ||||||
|     def used_by(self, request: Request, *args, **kwargs) -> Response: |     def used_by(self, request: Request, *args, **kwargs) -> Response: | ||||||
|         """Get a list of all objects that use this object""" |         """Get a list of all objects that use this object""" | ||||||
|         model: Model = self.get_object() |         model: Model = self.get_object() | ||||||
|  | |||||||
| @ -2,7 +2,7 @@ | |||||||
|  |  | ||||||
| from datetime import timedelta | from datetime import timedelta | ||||||
| from json import loads | from json import loads | ||||||
| from typing import Any | from typing import Any, Optional | ||||||
|  |  | ||||||
| from django.contrib.auth import update_session_auth_hash | from django.contrib.auth import update_session_auth_hash | ||||||
| from django.contrib.sessions.backends.cache import KEY_PREFIX | from django.contrib.sessions.backends.cache import KEY_PREFIX | ||||||
| @ -49,6 +49,7 @@ from rest_framework.viewsets import ModelViewSet | |||||||
| from structlog.stdlib import get_logger | from structlog.stdlib import get_logger | ||||||
|  |  | ||||||
| from authentik.admin.api.metrics import CoordinateSerializer | from authentik.admin.api.metrics import CoordinateSerializer | ||||||
|  | from authentik.api.decorators import permission_required | ||||||
| from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT | from authentik.blueprints.v1.importer import SERIALIZER_CONTEXT_BLUEPRINT | ||||||
| from authentik.brands.models import Brand | from authentik.brands.models import Brand | ||||||
| from authentik.core.api.used_by import UsedByMixin | from authentik.core.api.used_by import UsedByMixin | ||||||
| @ -73,7 +74,6 @@ from authentik.flows.models import FlowToken | |||||||
| from authentik.flows.planner import PLAN_CONTEXT_PENDING_USER, FlowPlanner | from authentik.flows.planner import PLAN_CONTEXT_PENDING_USER, FlowPlanner | ||||||
| from authentik.flows.views.executor import QS_KEY_TOKEN | from authentik.flows.views.executor import QS_KEY_TOKEN | ||||||
| from authentik.lib.avatars import get_avatar | from authentik.lib.avatars import get_avatar | ||||||
| from authentik.rbac.decorators import permission_required |  | ||||||
| from authentik.stages.email.models import EmailStage | from authentik.stages.email.models import EmailStage | ||||||
| from authentik.stages.email.tasks import send_mails | from authentik.stages.email.tasks import send_mails | ||||||
| from authentik.stages.email.utils import TemplateEmailMessage | from authentik.stages.email.utils import TemplateEmailMessage | ||||||
| @ -85,7 +85,7 @@ class UserGroupSerializer(ModelSerializer): | |||||||
|     """Simplified Group Serializer for user's groups""" |     """Simplified Group Serializer for user's groups""" | ||||||
|  |  | ||||||
|     attributes = JSONDictField(required=False) |     attributes = JSONDictField(required=False) | ||||||
|     parent_name = CharField(source="parent.name", read_only=True, allow_null=True) |     parent_name = CharField(source="parent.name", read_only=True) | ||||||
|  |  | ||||||
|     class Meta: |     class Meta: | ||||||
|         model = Group |         model = Group | ||||||
| @ -113,26 +113,13 @@ class UserSerializer(ModelSerializer): | |||||||
|         queryset=Group.objects.all().order_by("name"), |         queryset=Group.objects.all().order_by("name"), | ||||||
|         default=list, |         default=list, | ||||||
|     ) |     ) | ||||||
|     groups_obj = SerializerMethodField(allow_null=True) |     groups_obj = ListSerializer(child=UserGroupSerializer(), read_only=True, source="ak_groups") | ||||||
|     uid = CharField(read_only=True) |     uid = CharField(read_only=True) | ||||||
|     username = CharField( |     username = CharField( | ||||||
|         max_length=150, |         max_length=150, | ||||||
|         validators=[UniqueValidator(queryset=User.objects.all().order_by("username"))], |         validators=[UniqueValidator(queryset=User.objects.all().order_by("username"))], | ||||||
|     ) |     ) | ||||||
|  |  | ||||||
|     @property |  | ||||||
|     def _should_include_groups(self) -> bool: |  | ||||||
|         request: Request = self.context.get("request", None) |  | ||||||
|         if not request: |  | ||||||
|             return True |  | ||||||
|         return str(request.query_params.get("include_groups", "true")).lower() == "true" |  | ||||||
|  |  | ||||||
|     @extend_schema_field(UserGroupSerializer(many=True)) |  | ||||||
|     def get_groups_obj(self, instance: User) -> list[UserGroupSerializer] | None: |  | ||||||
|         if not self._should_include_groups: |  | ||||||
|             return None |  | ||||||
|         return UserGroupSerializer(instance.ak_groups, many=True).data |  | ||||||
|  |  | ||||||
|     def __init__(self, *args, **kwargs): |     def __init__(self, *args, **kwargs): | ||||||
|         super().__init__(*args, **kwargs) |         super().__init__(*args, **kwargs) | ||||||
|         if SERIALIZER_CONTEXT_BLUEPRINT in self.context: |         if SERIALIZER_CONTEXT_BLUEPRINT in self.context: | ||||||
| @ -155,7 +142,7 @@ class UserSerializer(ModelSerializer): | |||||||
|         self._set_password(instance, password) |         self._set_password(instance, password) | ||||||
|         return instance |         return instance | ||||||
|  |  | ||||||
|     def _set_password(self, instance: User, password: str | None): |     def _set_password(self, instance: User, password: Optional[str]): | ||||||
|         """Set password of user if we're in a blueprint context, and if it's an empty |         """Set password of user if we're in a blueprint context, and if it's an empty | ||||||
|         string then use an unusable password""" |         string then use an unusable password""" | ||||||
|         if SERIALIZER_CONTEXT_BLUEPRINT in self.context and password: |         if SERIALIZER_CONTEXT_BLUEPRINT in self.context and password: | ||||||
| @ -167,7 +154,7 @@ class UserSerializer(ModelSerializer): | |||||||
|  |  | ||||||
|     def get_avatar(self, user: User) -> str: |     def get_avatar(self, user: User) -> str: | ||||||
|         """User's avatar, either a http/https URL or a data URI""" |         """User's avatar, either a http/https URL or a data URI""" | ||||||
|         return get_avatar(user, self.context.get("request")) |         return get_avatar(user, self.context["request"]) | ||||||
|  |  | ||||||
|     def validate_path(self, path: str) -> str: |     def validate_path(self, path: str) -> str: | ||||||
|         """Validate path""" |         """Validate path""" | ||||||
| @ -231,7 +218,7 @@ class UserSelfSerializer(ModelSerializer): | |||||||
|  |  | ||||||
|     def get_avatar(self, user: User) -> str: |     def get_avatar(self, user: User) -> str: | ||||||
|         """User's avatar, either a http/https URL or a data URI""" |         """User's avatar, either a http/https URL or a data URI""" | ||||||
|         return get_avatar(user, self.context.get("request")) |         return get_avatar(user, self.context["request"]) | ||||||
|  |  | ||||||
|     @extend_schema_field( |     @extend_schema_field( | ||||||
|         ListSerializer( |         ListSerializer( | ||||||
| @ -371,7 +358,7 @@ class UsersFilter(FilterSet): | |||||||
|         try: |         try: | ||||||
|             value = loads(value) |             value = loads(value) | ||||||
|         except ValueError: |         except ValueError: | ||||||
|             raise ValidationError(detail="filter: failed to parse JSON") from None |             raise ValidationError(detail="filter: failed to parse JSON") | ||||||
|         if not isinstance(value, dict): |         if not isinstance(value, dict): | ||||||
|             raise ValidationError(detail="filter: value must be key:value mapping") |             raise ValidationError(detail="filter: value must be key:value mapping") | ||||||
|         qs = {} |         qs = {} | ||||||
| @ -407,28 +394,18 @@ class UserViewSet(UsedByMixin, ModelViewSet): | |||||||
|     search_fields = ["username", "name", "is_active", "email", "uuid"] |     search_fields = ["username", "name", "is_active", "email", "uuid"] | ||||||
|     filterset_class = UsersFilter |     filterset_class = UsersFilter | ||||||
|  |  | ||||||
|     def get_queryset(self): |     def get_queryset(self):  # pragma: no cover | ||||||
|         base_qs = User.objects.all().exclude_anonymous() |         return User.objects.all().exclude_anonymous().prefetch_related("ak_groups") | ||||||
|         if self.serializer_class(context={"request": self.request})._should_include_groups: |  | ||||||
|             base_qs = base_qs.prefetch_related("ak_groups") |  | ||||||
|         return base_qs |  | ||||||
|  |  | ||||||
|     @extend_schema( |     def _create_recovery_link(self) -> tuple[Optional[str], Optional[Token]]: | ||||||
|         parameters=[ |  | ||||||
|             OpenApiParameter("include_groups", bool, default=True), |  | ||||||
|         ] |  | ||||||
|     ) |  | ||||||
|     def list(self, request, *args, **kwargs): |  | ||||||
|         return super().list(request, *args, **kwargs) |  | ||||||
|  |  | ||||||
|     def _create_recovery_link(self) -> tuple[str, Token]: |  | ||||||
|         """Create a recovery link (when the current brand has a recovery flow set), |         """Create a recovery link (when the current brand has a recovery flow set), | ||||||
|         that can either be shown to an admin or sent to the user directly""" |         that can either be shown to an admin or sent to the user directly""" | ||||||
|         brand: Brand = self.request._request.brand |         brand: Brand = self.request._request.brand | ||||||
|         # Check that there is a recovery flow, if not return an error |         # Check that there is a recovery flow, if not return an error | ||||||
|         flow = brand.flow_recovery |         flow = brand.flow_recovery | ||||||
|         if not flow: |         if not flow: | ||||||
|             raise ValidationError({"non_field_errors": "No recovery flow set."}) |             LOGGER.debug("No recovery flow set") | ||||||
|  |             return None, None | ||||||
|         user: User = self.get_object() |         user: User = self.get_object() | ||||||
|         planner = FlowPlanner(flow) |         planner = FlowPlanner(flow) | ||||||
|         planner.allow_empty_flows = True |         planner.allow_empty_flows = True | ||||||
| @ -440,9 +417,8 @@ class UserViewSet(UsedByMixin, ModelViewSet): | |||||||
|                 }, |                 }, | ||||||
|             ) |             ) | ||||||
|         except FlowNonApplicableException: |         except FlowNonApplicableException: | ||||||
|             raise ValidationError( |             LOGGER.warning("Recovery flow not applicable to user") | ||||||
|                 {"non_field_errors": "Recovery flow not applicable to user"} |             return None, None | ||||||
|             ) from None |  | ||||||
|         token, __ = FlowToken.objects.update_or_create( |         token, __ = FlowToken.objects.update_or_create( | ||||||
|             identifier=f"{user.uid}-password-reset", |             identifier=f"{user.uid}-password-reset", | ||||||
|             defaults={ |             defaults={ | ||||||
| @ -557,7 +533,7 @@ class UserViewSet(UsedByMixin, ModelViewSet): | |||||||
|             400: OpenApiResponse(description="Bad request"), |             400: OpenApiResponse(description="Bad request"), | ||||||
|         }, |         }, | ||||||
|     ) |     ) | ||||||
|     @action(detail=True, methods=["POST"], permission_classes=[]) |     @action(detail=True, methods=["POST"]) | ||||||
|     def set_password(self, request: Request, pk: int) -> Response: |     def set_password(self, request: Request, pk: int) -> Response: | ||||||
|         """Set password for user""" |         """Set password for user""" | ||||||
|         user: User = self.get_object() |         user: User = self.get_object() | ||||||
| @ -587,13 +563,16 @@ class UserViewSet(UsedByMixin, ModelViewSet): | |||||||
|     @extend_schema( |     @extend_schema( | ||||||
|         responses={ |         responses={ | ||||||
|             "200": LinkSerializer(many=False), |             "200": LinkSerializer(many=False), | ||||||
|  |             "404": LinkSerializer(many=False), | ||||||
|         }, |         }, | ||||||
|         request=None, |  | ||||||
|     ) |     ) | ||||||
|     @action(detail=True, pagination_class=None, filter_backends=[], methods=["POST"]) |     @action(detail=True, pagination_class=None, filter_backends=[]) | ||||||
|     def recovery(self, request: Request, pk: int) -> Response: |     def recovery(self, request: Request, pk: int) -> Response: | ||||||
|         """Create a temporary link that a user can use to recover their accounts""" |         """Create a temporary link that a user can use to recover their accounts""" | ||||||
|         link, _ = self._create_recovery_link() |         link, _ = self._create_recovery_link() | ||||||
|  |         if not link: | ||||||
|  |             LOGGER.debug("Couldn't create token") | ||||||
|  |             return Response({"link": ""}, status=404) | ||||||
|         return Response({"link": link}) |         return Response({"link": link}) | ||||||
|  |  | ||||||
|     @permission_required("authentik_core.reset_user_password") |     @permission_required("authentik_core.reset_user_password") | ||||||
| @ -608,28 +587,31 @@ class UserViewSet(UsedByMixin, ModelViewSet): | |||||||
|         ], |         ], | ||||||
|         responses={ |         responses={ | ||||||
|             "204": OpenApiResponse(description="Successfully sent recover email"), |             "204": OpenApiResponse(description="Successfully sent recover email"), | ||||||
|  |             "404": OpenApiResponse(description="Bad request"), | ||||||
|         }, |         }, | ||||||
|         request=None, |  | ||||||
|     ) |     ) | ||||||
|     @action(detail=True, pagination_class=None, filter_backends=[], methods=["POST"]) |     @action(detail=True, pagination_class=None, filter_backends=[]) | ||||||
|     def recovery_email(self, request: Request, pk: int) -> Response: |     def recovery_email(self, request: Request, pk: int) -> Response: | ||||||
|         """Create a temporary link that a user can use to recover their accounts""" |         """Create a temporary link that a user can use to recover their accounts""" | ||||||
|         for_user: User = self.get_object() |         for_user: User = self.get_object() | ||||||
|         if for_user.email == "": |         if for_user.email == "": | ||||||
|             LOGGER.debug("User doesn't have an email address") |             LOGGER.debug("User doesn't have an email address") | ||||||
|             raise ValidationError({"non_field_errors": "User does not have an email address set."}) |             return Response(status=404) | ||||||
|         link, token = self._create_recovery_link() |         link, token = self._create_recovery_link() | ||||||
|  |         if not link: | ||||||
|  |             LOGGER.debug("Couldn't create token") | ||||||
|  |             return Response(status=404) | ||||||
|         # Lookup the email stage to assure the current user can access it |         # Lookup the email stage to assure the current user can access it | ||||||
|         stages = get_objects_for_user( |         stages = get_objects_for_user( | ||||||
|             request.user, "authentik_stages_email.view_emailstage" |             request.user, "authentik_stages_email.view_emailstage" | ||||||
|         ).filter(pk=request.query_params.get("email_stage")) |         ).filter(pk=request.query_params.get("email_stage")) | ||||||
|         if not stages.exists(): |         if not stages.exists(): | ||||||
|             LOGGER.debug("Email stage does not exist/user has no permissions") |             LOGGER.debug("Email stage does not exist/user has no permissions") | ||||||
|             raise ValidationError({"non_field_errors": "Email stage does not exist."}) |             return Response(status=404) | ||||||
|         email_stage: EmailStage = stages.first() |         email_stage: EmailStage = stages.first() | ||||||
|         message = TemplateEmailMessage( |         message = TemplateEmailMessage( | ||||||
|             subject=_(email_stage.subject), |             subject=_(email_stage.subject), | ||||||
|             to=[(for_user.name, for_user.email)], |             to=[for_user.email], | ||||||
|             template_name=email_stage.template, |             template_name=email_stage.template, | ||||||
|             language=for_user.locale(request), |             language=for_user.locale(request), | ||||||
|             template_context={ |             template_context={ | ||||||
| @ -649,7 +631,7 @@ class UserViewSet(UsedByMixin, ModelViewSet): | |||||||
|             "401": OpenApiResponse(description="Access denied"), |             "401": OpenApiResponse(description="Access denied"), | ||||||
|         }, |         }, | ||||||
|     ) |     ) | ||||||
|     @action(detail=True, methods=["POST"], permission_classes=[]) |     @action(detail=True, methods=["POST"]) | ||||||
|     def impersonate(self, request: Request, pk: int) -> Response: |     def impersonate(self, request: Request, pk: int) -> Response: | ||||||
|         """Impersonate a user""" |         """Impersonate a user""" | ||||||
|         if not request.tenant.impersonation: |         if not request.tenant.impersonation: | ||||||
|  | |||||||
| @ -6,16 +6,8 @@ from django.db.models import Model | |||||||
| from drf_spectacular.extensions import OpenApiSerializerFieldExtension | from drf_spectacular.extensions import OpenApiSerializerFieldExtension | ||||||
| from drf_spectacular.plumbing import build_basic_type | from drf_spectacular.plumbing import build_basic_type | ||||||
| from drf_spectacular.types import OpenApiTypes | from drf_spectacular.types import OpenApiTypes | ||||||
| from rest_framework.fields import ( | from rest_framework.fields import BooleanField, CharField, IntegerField, JSONField | ||||||
|     CharField, | from rest_framework.serializers import Serializer, SerializerMethodField, ValidationError | ||||||
|     IntegerField, |  | ||||||
|     JSONField, |  | ||||||
|     SerializerMethodField, |  | ||||||
| ) |  | ||||||
| from rest_framework.serializers import ( |  | ||||||
|     Serializer, |  | ||||||
|     ValidationError, |  | ||||||
| ) |  | ||||||
|  |  | ||||||
|  |  | ||||||
| def is_dict(value: Any): | def is_dict(value: Any): | ||||||
| @ -76,6 +68,16 @@ class MetaNameSerializer(PassiveSerializer): | |||||||
|         return f"{obj._meta.app_label}.{obj._meta.model_name}" |         return f"{obj._meta.app_label}.{obj._meta.model_name}" | ||||||
|  |  | ||||||
|  |  | ||||||
|  | class TypeCreateSerializer(PassiveSerializer): | ||||||
|  |     """Types of an object that can be created""" | ||||||
|  |  | ||||||
|  |     name = CharField(required=True) | ||||||
|  |     description = CharField(required=True) | ||||||
|  |     component = CharField(required=True) | ||||||
|  |     model_name = CharField(required=True) | ||||||
|  |     requires_enterprise = BooleanField(default=False) | ||||||
|  |  | ||||||
|  |  | ||||||
| class CacheSerializer(PassiveSerializer): | class CacheSerializer(PassiveSerializer): | ||||||
|     """Generic cache stats for an object""" |     """Generic cache stats for an object""" | ||||||
|  |  | ||||||
|  | |||||||
| @ -14,16 +14,14 @@ class AuthentikCoreConfig(ManagedAppConfig): | |||||||
|     mountpoint = "" |     mountpoint = "" | ||||||
|     default = True |     default = True | ||||||
|  |  | ||||||
|     @ManagedAppConfig.reconcile_global |     def reconcile_global_debug_worker_hook(self): | ||||||
|     def debug_worker_hook(self): |  | ||||||
|         """Dispatch startup tasks inline when debugging""" |         """Dispatch startup tasks inline when debugging""" | ||||||
|         if settings.DEBUG: |         if settings.DEBUG: | ||||||
|             from authentik.root.celery import worker_ready_hook |             from authentik.root.celery import worker_ready_hook | ||||||
|  |  | ||||||
|             worker_ready_hook() |             worker_ready_hook() | ||||||
|  |  | ||||||
|     @ManagedAppConfig.reconcile_tenant |     def reconcile_tenant_source_inbuilt(self): | ||||||
|     def source_inbuilt(self): |  | ||||||
|         """Reconcile inbuilt source""" |         """Reconcile inbuilt source""" | ||||||
|         from authentik.core.models import Source |         from authentik.core.models import Source | ||||||
|  |  | ||||||
|  | |||||||
| @ -1,6 +1,6 @@ | |||||||
| """Authenticate with tokens""" | """Authenticate with tokens""" | ||||||
|  |  | ||||||
| from typing import Any | from typing import Any, Optional | ||||||
|  |  | ||||||
| from django.contrib.auth.backends import ModelBackend | from django.contrib.auth.backends import ModelBackend | ||||||
| from django.http.request import HttpRequest | from django.http.request import HttpRequest | ||||||
| @ -16,24 +16,23 @@ class InbuiltBackend(ModelBackend): | |||||||
|     """Inbuilt backend""" |     """Inbuilt backend""" | ||||||
|  |  | ||||||
|     def authenticate( |     def authenticate( | ||||||
|         self, request: HttpRequest, username: str | None, password: str | None, **kwargs: Any |         self, request: HttpRequest, username: Optional[str], password: Optional[str], **kwargs: Any | ||||||
|     ) -> User | None: |     ) -> Optional[User]: | ||||||
|         user = super().authenticate(request, username=username, password=password, **kwargs) |         user = super().authenticate(request, username=username, password=password, **kwargs) | ||||||
|         if not user: |         if not user: | ||||||
|             return None |             return None | ||||||
|         self.set_method("password", request) |         self.set_method("password", request) | ||||||
|         return user |         return user | ||||||
|  |  | ||||||
|     def set_method(self, method: str, request: HttpRequest | None, **kwargs): |     def set_method(self, method: str, request: Optional[HttpRequest], **kwargs): | ||||||
|         """Set method data on current flow, if possbiel""" |         """Set method data on current flow, if possbiel""" | ||||||
|         if not request: |         if not request: | ||||||
|             return |             return | ||||||
|         # Since we can't directly pass other variables to signals, and we want to log the method |         # Since we can't directly pass other variables to signals, and we want to log the method | ||||||
|         # and the token used, we assume we're running in a flow and set a variable in the context |         # and the token used, we assume we're running in a flow and set a variable in the context | ||||||
|         flow_plan: FlowPlan = request.session.get(SESSION_KEY_PLAN, FlowPlan("")) |         flow_plan: FlowPlan = request.session.get(SESSION_KEY_PLAN, FlowPlan("")) | ||||||
|         flow_plan.context.setdefault(PLAN_CONTEXT_METHOD, method) |         flow_plan.context[PLAN_CONTEXT_METHOD] = method | ||||||
|         flow_plan.context.setdefault(PLAN_CONTEXT_METHOD_ARGS, {}) |         flow_plan.context[PLAN_CONTEXT_METHOD_ARGS] = cleanse_dict(sanitize_dict(kwargs)) | ||||||
|         flow_plan.context[PLAN_CONTEXT_METHOD_ARGS].update(cleanse_dict(sanitize_dict(kwargs))) |  | ||||||
|         request.session[SESSION_KEY_PLAN] = flow_plan |         request.session[SESSION_KEY_PLAN] = flow_plan | ||||||
|  |  | ||||||
|  |  | ||||||
| @ -41,18 +40,18 @@ class TokenBackend(InbuiltBackend): | |||||||
|     """Authenticate with token""" |     """Authenticate with token""" | ||||||
|  |  | ||||||
|     def authenticate( |     def authenticate( | ||||||
|         self, request: HttpRequest, username: str | None, password: str | None, **kwargs: Any |         self, request: HttpRequest, username: Optional[str], password: Optional[str], **kwargs: Any | ||||||
|     ) -> User | None: |     ) -> Optional[User]: | ||||||
|         try: |         try: | ||||||
|  |             # pylint: disable=no-member | ||||||
|             user = User._default_manager.get_by_natural_key(username) |             user = User._default_manager.get_by_natural_key(username) | ||||||
|  |         # pylint: disable=no-member | ||||||
|         except User.DoesNotExist: |         except User.DoesNotExist: | ||||||
|             # Run the default password hasher once to reduce the timing |             # Run the default password hasher once to reduce the timing | ||||||
|             # difference between an existing and a nonexistent user (#20760). |             # difference between an existing and a nonexistent user (#20760). | ||||||
|             User().set_password(password) |             User().set_password(password) | ||||||
|             return None |             return None | ||||||
|  |         # pylint: disable=no-member | ||||||
|         tokens = Token.filter_not_expired( |         tokens = Token.filter_not_expired( | ||||||
|             user=user, key=password, intent=TokenIntents.INTENT_APP_PASSWORD |             user=user, key=password, intent=TokenIntents.INTENT_APP_PASSWORD | ||||||
|         ) |         ) | ||||||
|  | |||||||
| @ -38,6 +38,6 @@ class TokenOutpostMiddleware: | |||||||
|                 raise DenyConnection() |                 raise DenyConnection() | ||||||
|         except AuthenticationFailed as exc: |         except AuthenticationFailed as exc: | ||||||
|             LOGGER.warning("Failed to authenticate", exc=exc) |             LOGGER.warning("Failed to authenticate", exc=exc) | ||||||
|             raise DenyConnection() from None |             raise DenyConnection() | ||||||
|  |  | ||||||
|         scope["user"] = user |         scope["user"] = user | ||||||
|  | |||||||
							
								
								
									
										7
									
								
								authentik/core/exceptions.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								authentik/core/exceptions.py
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,7 @@ | |||||||
|  | """authentik core exceptions""" | ||||||
|  |  | ||||||
|  | from authentik.lib.sentry import SentryIgnoredException | ||||||
|  |  | ||||||
|  |  | ||||||
|  | class PropertyMappingExpressionException(SentryIgnoredException): | ||||||
|  |     """Error when a PropertyMapping Exception expression could not be parsed or evaluated.""" | ||||||
| @ -1,13 +1,11 @@ | |||||||
| """Property Mapping Evaluator""" | """Property Mapping Evaluator""" | ||||||
|  |  | ||||||
| from types import CodeType | from typing import Any, Optional | ||||||
| from typing import Any |  | ||||||
|  |  | ||||||
| from django.db.models import Model | from django.db.models import Model | ||||||
| from django.http import HttpRequest | from django.http import HttpRequest | ||||||
| from prometheus_client import Histogram | from prometheus_client import Histogram | ||||||
|  |  | ||||||
| from authentik.core.expression.exceptions import SkipObjectException |  | ||||||
| from authentik.core.models import User | from authentik.core.models import User | ||||||
| from authentik.events.models import Event, EventAction | from authentik.events.models import Event, EventAction | ||||||
| from authentik.lib.expression.evaluator import BaseEvaluator | from authentik.lib.expression.evaluator import BaseEvaluator | ||||||
| @ -25,43 +23,31 @@ class PropertyMappingEvaluator(BaseEvaluator): | |||||||
|     """Custom Evaluator that adds some different context variables.""" |     """Custom Evaluator that adds some different context variables.""" | ||||||
|  |  | ||||||
|     dry_run: bool |     dry_run: bool | ||||||
|     model: Model |  | ||||||
|     _compiled: CodeType | None = None |  | ||||||
|  |  | ||||||
|     def __init__( |     def __init__( | ||||||
|         self, |         self, | ||||||
|         model: Model, |         model: Model, | ||||||
|         user: User | None = None, |         user: Optional[User] = None, | ||||||
|         request: HttpRequest | None = None, |         request: Optional[HttpRequest] = None, | ||||||
|         dry_run: bool | None = False, |         dry_run: Optional[bool] = False, | ||||||
|         **kwargs, |         **kwargs, | ||||||
|     ): |     ): | ||||||
|         self.model = model |  | ||||||
|         if hasattr(model, "name"): |         if hasattr(model, "name"): | ||||||
|             _filename = model.name |             _filename = model.name | ||||||
|         else: |         else: | ||||||
|             _filename = str(model) |             _filename = str(model) | ||||||
|         super().__init__(filename=_filename) |         super().__init__(filename=_filename) | ||||||
|         self.dry_run = dry_run |  | ||||||
|         self.set_context(user, request, **kwargs) |  | ||||||
|  |  | ||||||
|     def set_context( |  | ||||||
|         self, |  | ||||||
|         user: User | None = None, |  | ||||||
|         request: HttpRequest | None = None, |  | ||||||
|         **kwargs, |  | ||||||
|     ): |  | ||||||
|         req = PolicyRequest(user=User()) |         req = PolicyRequest(user=User()) | ||||||
|         req.obj = self.model |         req.obj = model | ||||||
|         if user: |         if user: | ||||||
|             req.user = user |             req.user = user | ||||||
|             self._context["user"] = user |             self._context["user"] = user | ||||||
|         if request: |         if request: | ||||||
|             req.http_request = request |             req.http_request = request | ||||||
|         req.context.update(**kwargs) |  | ||||||
|         self._context["request"] = req |         self._context["request"] = req | ||||||
|  |         req.context.update(**kwargs) | ||||||
|         self._context.update(**kwargs) |         self._context.update(**kwargs) | ||||||
|         self._globals["SkipObject"] = SkipObjectException |         self.dry_run = dry_run | ||||||
|  |  | ||||||
|     def handle_error(self, exc: Exception, expression_source: str): |     def handle_error(self, exc: Exception, expression_source: str): | ||||||
|         """Exception Handler""" |         """Exception Handler""" | ||||||
| @ -83,9 +69,3 @@ class PropertyMappingEvaluator(BaseEvaluator): | |||||||
|     def evaluate(self, *args, **kwargs) -> Any: |     def evaluate(self, *args, **kwargs) -> Any: | ||||||
|         with PROPERTY_MAPPING_TIME.labels(mapping_name=self._filename).time(): |         with PROPERTY_MAPPING_TIME.labels(mapping_name=self._filename).time(): | ||||||
|             return super().evaluate(*args, **kwargs) |             return super().evaluate(*args, **kwargs) | ||||||
|  |  | ||||||
|     def compile(self, expression: str | None = None) -> Any: |  | ||||||
|         if not self._compiled: |  | ||||||
|             compiled = super().compile(expression or self.model.expression) |  | ||||||
|             self._compiled = compiled |  | ||||||
|         return self._compiled |  | ||||||
|  | |||||||
| @ -1,18 +0,0 @@ | |||||||
| """authentik core exceptions""" |  | ||||||
|  |  | ||||||
| from authentik.lib.sentry import SentryIgnoredException |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class PropertyMappingExpressionException(SentryIgnoredException): |  | ||||||
|     """Error when a PropertyMapping Exception expression could not be parsed or evaluated.""" |  | ||||||
|  |  | ||||||
|     def __init__(self, exc: Exception, mapping) -> None: |  | ||||||
|         super().__init__() |  | ||||||
|         self.exc = exc |  | ||||||
|         self.mapping = mapping |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class SkipObjectException(PropertyMappingExpressionException): |  | ||||||
|     """Exception which can be raised in a property mapping to skip syncing an object. |  | ||||||
|     Only applies to Property mappings which sync objects, and not on mappings which transitively |  | ||||||
|     apply to a single user""" |  | ||||||
| @ -1,34 +1,10 @@ | |||||||
| """custom runserver command""" | """custom runserver command""" | ||||||
|  |  | ||||||
| from typing import TextIO |  | ||||||
|  |  | ||||||
| from daphne.management.commands.runserver import Command as RunServer | from daphne.management.commands.runserver import Command as RunServer | ||||||
| from daphne.server import Server |  | ||||||
|  |  | ||||||
| from authentik.root.signals import post_startup, pre_startup, startup |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class SignalServer(Server): |  | ||||||
|     """Server which signals back to authentik when it finished starting up""" |  | ||||||
|  |  | ||||||
|     def __init__(self, *args, **kwargs): |  | ||||||
|         super().__init__(*args, **kwargs) |  | ||||||
|  |  | ||||||
|         def ready_callable(): |  | ||||||
|             pre_startup.send(sender=self) |  | ||||||
|             startup.send(sender=self) |  | ||||||
|             post_startup.send(sender=self) |  | ||||||
|  |  | ||||||
|         self.ready_callable = ready_callable |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class Command(RunServer): | class Command(RunServer): | ||||||
|     """custom runserver command, which doesn't show the misleading django startup message""" |     """custom runserver command, which doesn't show the misleading django startup message""" | ||||||
|  |  | ||||||
|     server_cls = SignalServer |     def on_bind(self, server_port): | ||||||
|  |         pass | ||||||
|     def __init__(self, *args, **kwargs): |  | ||||||
|         super().__init__(*args, **kwargs) |  | ||||||
|         # Redirect standard stdout banner from Daphne into the void |  | ||||||
|         # as there are a couple more steps that happen before startup is fully done |  | ||||||
|         self.stdout = TextIO() |  | ||||||
|  | |||||||
| @ -16,8 +16,13 @@ from authentik.events.middleware import should_log_model | |||||||
| from authentik.events.models import Event, EventAction | from authentik.events.models import Event, EventAction | ||||||
| from authentik.events.utils import model_to_dict | from authentik.events.utils import model_to_dict | ||||||
|  |  | ||||||
| BANNER_TEXT = f"""### authentik shell ({get_full_version()}) | BANNER_TEXT = """### authentik shell ({authentik}) | ||||||
| ### Node {platform.node()} | Arch {platform.machine()} | Python {platform.python_version()} """ | ### Node {node} | Arch {arch} | Python {python} """.format( | ||||||
|  |     node=platform.node(), | ||||||
|  |     python=platform.python_version(), | ||||||
|  |     arch=platform.machine(), | ||||||
|  |     authentik=get_full_version(), | ||||||
|  | ) | ||||||
|  |  | ||||||
|  |  | ||||||
| class Command(BaseCommand): | class Command(BaseCommand): | ||||||
| @ -81,7 +86,7 @@ class Command(BaseCommand): | |||||||
|  |  | ||||||
|         # If Python code has been passed, execute it and exit. |         # If Python code has been passed, execute it and exit. | ||||||
|         if options["command"]: |         if options["command"]: | ||||||
|  |             # pylint: disable=exec-used | ||||||
|             exec(options["command"], namespace)  # nosec # noqa |             exec(options["command"], namespace)  # nosec # noqa | ||||||
|             return |             return | ||||||
|  |  | ||||||
| @ -94,7 +99,7 @@ class Command(BaseCommand): | |||||||
|         else: |         else: | ||||||
|             try: |             try: | ||||||
|                 hook() |                 hook() | ||||||
|             except Exception: |             except Exception:  # pylint: disable=broad-except | ||||||
|                 # Match the behavior of the cpython shell where an error in |                 # Match the behavior of the cpython shell where an error in | ||||||
|                 # sys.__interactivehook__ prints a warning and the exception |                 # sys.__interactivehook__ prints a warning and the exception | ||||||
|                 # and continues. |                 # and continues. | ||||||
|  | |||||||
| @ -1,7 +1,7 @@ | |||||||
| """authentik admin Middleware to impersonate users""" | """authentik admin Middleware to impersonate users""" | ||||||
|  |  | ||||||
| from collections.abc import Callable |  | ||||||
| from contextvars import ContextVar | from contextvars import ContextVar | ||||||
|  | from typing import Callable, Optional | ||||||
| from uuid import uuid4 | from uuid import uuid4 | ||||||
|  |  | ||||||
| from django.http import HttpRequest, HttpResponse | from django.http import HttpRequest, HttpResponse | ||||||
| @ -15,9 +15,9 @@ RESPONSE_HEADER_ID = "X-authentik-id" | |||||||
| KEY_AUTH_VIA = "auth_via" | KEY_AUTH_VIA = "auth_via" | ||||||
| KEY_USER = "user" | KEY_USER = "user" | ||||||
|  |  | ||||||
| CTX_REQUEST_ID = ContextVar[str | None](STRUCTLOG_KEY_PREFIX + "request_id", default=None) | CTX_REQUEST_ID = ContextVar[Optional[str]](STRUCTLOG_KEY_PREFIX + "request_id", default=None) | ||||||
| CTX_HOST = ContextVar[str | None](STRUCTLOG_KEY_PREFIX + "host", default=None) | CTX_HOST = ContextVar[Optional[str]](STRUCTLOG_KEY_PREFIX + "host", default=None) | ||||||
| CTX_AUTH_VIA = ContextVar[str | None](STRUCTLOG_KEY_PREFIX + KEY_AUTH_VIA, default=None) | CTX_AUTH_VIA = ContextVar[Optional[str]](STRUCTLOG_KEY_PREFIX + KEY_AUTH_VIA, default=None) | ||||||
|  |  | ||||||
|  |  | ||||||
| class ImpersonateMiddleware: | class ImpersonateMiddleware: | ||||||
| @ -55,7 +55,7 @@ class RequestIDMiddleware: | |||||||
|     def __call__(self, request: HttpRequest) -> HttpResponse: |     def __call__(self, request: HttpRequest) -> HttpResponse: | ||||||
|         if not hasattr(request, "request_id"): |         if not hasattr(request, "request_id"): | ||||||
|             request_id = uuid4().hex |             request_id = uuid4().hex | ||||||
|             request.request_id = request_id |             setattr(request, "request_id", request_id) | ||||||
|             CTX_REQUEST_ID.set(request_id) |             CTX_REQUEST_ID.set(request_id) | ||||||
|             CTX_HOST.set(request.get_host()) |             CTX_HOST.set(request.get_host()) | ||||||
|             set_tag("authentik.request_id", request_id) |             set_tag("authentik.request_id", request_id) | ||||||
| @ -67,7 +67,7 @@ class RequestIDMiddleware: | |||||||
|         response = self.get_response(request) |         response = self.get_response(request) | ||||||
|  |  | ||||||
|         response[RESPONSE_HEADER_ID] = request.request_id |         response[RESPONSE_HEADER_ID] = request.request_id | ||||||
|         response.ak_context = {} |         setattr(response, "ak_context", {}) | ||||||
|         response.ak_context["request_id"] = CTX_REQUEST_ID.get() |         response.ak_context["request_id"] = CTX_REQUEST_ID.get() | ||||||
|         response.ak_context["host"] = CTX_HOST.get() |         response.ak_context["host"] = CTX_HOST.get() | ||||||
|         response.ak_context[KEY_AUTH_VIA] = CTX_AUTH_VIA.get() |         response.ak_context[KEY_AUTH_VIA] = CTX_AUTH_VIA.get() | ||||||
|  | |||||||
| @ -5,7 +5,6 @@ from django.db import migrations, models | |||||||
| from django.db.backends.base.schema import BaseDatabaseSchemaEditor | from django.db.backends.base.schema import BaseDatabaseSchemaEditor | ||||||
|  |  | ||||||
| import authentik.core.models | import authentik.core.models | ||||||
| from authentik.lib.generators import generate_id |  | ||||||
|  |  | ||||||
|  |  | ||||||
| def set_default_token_key(apps: Apps, schema_editor: BaseDatabaseSchemaEditor): | def set_default_token_key(apps: Apps, schema_editor: BaseDatabaseSchemaEditor): | ||||||
| @ -17,10 +16,6 @@ def set_default_token_key(apps: Apps, schema_editor: BaseDatabaseSchemaEditor): | |||||||
|         token.save() |         token.save() | ||||||
|  |  | ||||||
|  |  | ||||||
| def default_token_key(): |  | ||||||
|     return generate_id(60) |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class Migration(migrations.Migration): | class Migration(migrations.Migration): | ||||||
|     replaces = [ |     replaces = [ | ||||||
|         ("authentik_core", "0012_auto_20201003_1737"), |         ("authentik_core", "0012_auto_20201003_1737"), | ||||||
| @ -67,7 +62,7 @@ class Migration(migrations.Migration): | |||||||
|         migrations.AddField( |         migrations.AddField( | ||||||
|             model_name="token", |             model_name="token", | ||||||
|             name="key", |             name="key", | ||||||
|             field=models.TextField(default=default_token_key), |             field=models.TextField(default=authentik.core.models.default_token_key), | ||||||
|         ), |         ), | ||||||
|         migrations.AlterUniqueTogether( |         migrations.AlterUniqueTogether( | ||||||
|             name="token", |             name="token", | ||||||
|  | |||||||
| @ -7,10 +7,9 @@ from django.db.backends.base.schema import BaseDatabaseSchemaEditor | |||||||
|  |  | ||||||
|  |  | ||||||
| def backport_is_backchannel(apps: Apps, schema_editor: BaseDatabaseSchemaEditor): | def backport_is_backchannel(apps: Apps, schema_editor: BaseDatabaseSchemaEditor): | ||||||
|     from authentik.providers.ldap.models import LDAPProvider |     from authentik.core.models import BackchannelProvider | ||||||
|     from authentik.providers.scim.models import SCIMProvider |  | ||||||
|  |  | ||||||
|     for model in [LDAPProvider, SCIMProvider]: |     for model in BackchannelProvider.__subclasses__(): | ||||||
|         try: |         try: | ||||||
|             for obj in model.objects.only("is_backchannel"): |             for obj in model.objects.only("is_backchannel"): | ||||||
|                 obj.is_backchannel = True |                 obj.is_backchannel = True | ||||||
|  | |||||||
| @ -1,31 +0,0 @@ | |||||||
| # Generated by Django 5.0.2 on 2024-02-29 10:15 |  | ||||||
|  |  | ||||||
| from django.db import migrations, models |  | ||||||
|  |  | ||||||
| import authentik.core.models |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class Migration(migrations.Migration): |  | ||||||
|  |  | ||||||
|     dependencies = [ |  | ||||||
|         ("authentik_core", "0033_alter_user_options"), |  | ||||||
|         ("authentik_tenants", "0002_tenant_default_token_duration_and_more"), |  | ||||||
|     ] |  | ||||||
|  |  | ||||||
|     operations = [ |  | ||||||
|         migrations.AlterField( |  | ||||||
|             model_name="authenticatedsession", |  | ||||||
|             name="expires", |  | ||||||
|             field=models.DateTimeField(default=None, null=True), |  | ||||||
|         ), |  | ||||||
|         migrations.AlterField( |  | ||||||
|             model_name="token", |  | ||||||
|             name="expires", |  | ||||||
|             field=models.DateTimeField(default=None, null=True), |  | ||||||
|         ), |  | ||||||
|         migrations.AlterField( |  | ||||||
|             model_name="token", |  | ||||||
|             name="key", |  | ||||||
|             field=models.TextField(default=authentik.core.models.default_token_key), |  | ||||||
|         ), |  | ||||||
|     ] |  | ||||||
| @ -1,52 +0,0 @@ | |||||||
| # Generated by Django 5.0.4 on 2024-04-15 11:28 |  | ||||||
|  |  | ||||||
| from django.db import migrations, models |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class Migration(migrations.Migration): |  | ||||||
|  |  | ||||||
|     dependencies = [ |  | ||||||
|         ("auth", "0012_alter_user_first_name_max_length"), |  | ||||||
|         ("authentik_core", "0034_alter_authenticatedsession_expires_and_more"), |  | ||||||
|         ("authentik_rbac", "0003_alter_systempermission_options"), |  | ||||||
|     ] |  | ||||||
|  |  | ||||||
|     operations = [ |  | ||||||
|         migrations.AlterModelOptions( |  | ||||||
|             name="group", |  | ||||||
|             options={ |  | ||||||
|                 "permissions": [ |  | ||||||
|                     ("add_user_to_group", "Add user to group"), |  | ||||||
|                     ("remove_user_from_group", "Remove user from group"), |  | ||||||
|                 ], |  | ||||||
|                 "verbose_name": "Group", |  | ||||||
|                 "verbose_name_plural": "Groups", |  | ||||||
|             }, |  | ||||||
|         ), |  | ||||||
|         migrations.AddIndex( |  | ||||||
|             model_name="group", |  | ||||||
|             index=models.Index(fields=["name"], name="authentik_c_name_9ba8e4_idx"), |  | ||||||
|         ), |  | ||||||
|         migrations.AddIndex( |  | ||||||
|             model_name="user", |  | ||||||
|             index=models.Index(fields=["last_login"], name="authentik_c_last_lo_f0179a_idx"), |  | ||||||
|         ), |  | ||||||
|         migrations.AddIndex( |  | ||||||
|             model_name="user", |  | ||||||
|             index=models.Index( |  | ||||||
|                 fields=["password_change_date"], name="authentik_c_passwor_eec915_idx" |  | ||||||
|             ), |  | ||||||
|         ), |  | ||||||
|         migrations.AddIndex( |  | ||||||
|             model_name="user", |  | ||||||
|             index=models.Index(fields=["uuid"], name="authentik_c_uuid_3dae2f_idx"), |  | ||||||
|         ), |  | ||||||
|         migrations.AddIndex( |  | ||||||
|             model_name="user", |  | ||||||
|             index=models.Index(fields=["path"], name="authentik_c_path_b1f502_idx"), |  | ||||||
|         ), |  | ||||||
|         migrations.AddIndex( |  | ||||||
|             model_name="user", |  | ||||||
|             index=models.Index(fields=["type"], name="authentik_c_type_ecf60d_idx"), |  | ||||||
|         ), |  | ||||||
|     ] |  | ||||||
| @ -1,6 +1,6 @@ | |||||||
| """authentik core models""" | """authentik core models""" | ||||||
|  |  | ||||||
| from datetime import datetime | from datetime import timedelta | ||||||
| from hashlib import sha256 | from hashlib import sha256 | ||||||
| from typing import Any, Optional, Self | from typing import Any, Optional, Self | ||||||
| from uuid import uuid4 | from uuid import uuid4 | ||||||
| @ -15,7 +15,6 @@ from django.http import HttpRequest | |||||||
| from django.utils.functional import SimpleLazyObject, cached_property | from django.utils.functional import SimpleLazyObject, cached_property | ||||||
| from django.utils.timezone import now | from django.utils.timezone import now | ||||||
| from django.utils.translation import gettext_lazy as _ | from django.utils.translation import gettext_lazy as _ | ||||||
| from django_cte import CTEQuerySet, With |  | ||||||
| from guardian.conf import settings | from guardian.conf import settings | ||||||
| from guardian.mixins import GuardianUserMixin | from guardian.mixins import GuardianUserMixin | ||||||
| from model_utils.managers import InheritanceManager | from model_utils.managers import InheritanceManager | ||||||
| @ -23,19 +22,18 @@ from rest_framework.serializers import Serializer | |||||||
| from structlog.stdlib import get_logger | from structlog.stdlib import get_logger | ||||||
|  |  | ||||||
| from authentik.blueprints.models import ManagedModel | from authentik.blueprints.models import ManagedModel | ||||||
| from authentik.core.expression.exceptions import PropertyMappingExpressionException | from authentik.core.exceptions import PropertyMappingExpressionException | ||||||
| from authentik.core.types import UILoginButton, UserSettingSerializer | from authentik.core.types import UILoginButton, UserSettingSerializer | ||||||
| from authentik.lib.avatars import get_avatar | from authentik.lib.avatars import get_avatar | ||||||
|  | from authentik.lib.config import CONFIG | ||||||
| from authentik.lib.generators import generate_id | from authentik.lib.generators import generate_id | ||||||
| from authentik.lib.models import ( | from authentik.lib.models import ( | ||||||
|     CreatedUpdatedModel, |     CreatedUpdatedModel, | ||||||
|     DomainlessFormattedURLValidator, |     DomainlessFormattedURLValidator, | ||||||
|     SerializerModel, |     SerializerModel, | ||||||
| ) | ) | ||||||
| from authentik.lib.utils.time import timedelta_from_string |  | ||||||
| from authentik.policies.models import PolicyBindingModel | from authentik.policies.models import PolicyBindingModel | ||||||
| from authentik.tenants.models import DEFAULT_TOKEN_DURATION, DEFAULT_TOKEN_LENGTH | from authentik.root.install_id import get_install_id | ||||||
| from authentik.tenants.utils import get_current_tenant, get_unique_identifier |  | ||||||
|  |  | ||||||
| LOGGER = get_logger() | LOGGER = get_logger() | ||||||
| USER_ATTRIBUTE_DEBUG = "goauthentik.io/user/debug" | USER_ATTRIBUTE_DEBUG = "goauthentik.io/user/debug" | ||||||
| @ -44,44 +42,33 @@ USER_ATTRIBUTE_EXPIRES = "goauthentik.io/user/expires" | |||||||
| USER_ATTRIBUTE_DELETE_ON_LOGOUT = "goauthentik.io/user/delete-on-logout" | USER_ATTRIBUTE_DELETE_ON_LOGOUT = "goauthentik.io/user/delete-on-logout" | ||||||
| USER_ATTRIBUTE_SOURCES = "goauthentik.io/user/sources" | USER_ATTRIBUTE_SOURCES = "goauthentik.io/user/sources" | ||||||
| USER_ATTRIBUTE_TOKEN_EXPIRING = "goauthentik.io/user/token-expires"  # nosec | USER_ATTRIBUTE_TOKEN_EXPIRING = "goauthentik.io/user/token-expires"  # nosec | ||||||
| USER_ATTRIBUTE_TOKEN_MAXIMUM_LIFETIME = "goauthentik.io/user/token-maximum-lifetime"  # nosec |  | ||||||
| USER_ATTRIBUTE_CHANGE_USERNAME = "goauthentik.io/user/can-change-username" | USER_ATTRIBUTE_CHANGE_USERNAME = "goauthentik.io/user/can-change-username" | ||||||
| USER_ATTRIBUTE_CHANGE_NAME = "goauthentik.io/user/can-change-name" | USER_ATTRIBUTE_CHANGE_NAME = "goauthentik.io/user/can-change-name" | ||||||
| USER_ATTRIBUTE_CHANGE_EMAIL = "goauthentik.io/user/can-change-email" | USER_ATTRIBUTE_CHANGE_EMAIL = "goauthentik.io/user/can-change-email" | ||||||
| USER_PATH_SYSTEM_PREFIX = "goauthentik.io" | USER_PATH_SYSTEM_PREFIX = "goauthentik.io" | ||||||
| USER_PATH_SERVICE_ACCOUNT = USER_PATH_SYSTEM_PREFIX + "/service-accounts" | USER_PATH_SERVICE_ACCOUNT = USER_PATH_SYSTEM_PREFIX + "/service-accounts" | ||||||
|  |  | ||||||
|  |  | ||||||
| options.DEFAULT_NAMES = options.DEFAULT_NAMES + ( | options.DEFAULT_NAMES = options.DEFAULT_NAMES + ( | ||||||
|     # used_by API that allows models to specify if they shadow an object |     # used_by API that allows models to specify if they shadow an object | ||||||
|     # for example the proxy provider which is built on top of an oauth provider |     # for example the proxy provider which is built on top of an oauth provider | ||||||
|     "authentik_used_by_shadows", |     "authentik_used_by_shadows", | ||||||
|  |     # List fields for which changes are not logged (due to them having dedicated objects) | ||||||
|  |     # for example user's password and last_login | ||||||
|  |     "authentik_signals_ignored_fields", | ||||||
| ) | ) | ||||||
|  |  | ||||||
| GROUP_RECURSION_LIMIT = 20 |  | ||||||
|  |  | ||||||
|  | def default_token_duration(): | ||||||
| def default_token_duration() -> datetime: |  | ||||||
|     """Default duration a Token is valid""" |     """Default duration a Token is valid""" | ||||||
|     current_tenant = get_current_tenant() |     return now() + timedelta(minutes=30) | ||||||
|     token_duration = ( |  | ||||||
|         current_tenant.default_token_duration |  | ||||||
|         if hasattr(current_tenant, "default_token_duration") |  | ||||||
|         else DEFAULT_TOKEN_DURATION |  | ||||||
|     ) |  | ||||||
|     return now() + timedelta_from_string(token_duration) |  | ||||||
|  |  | ||||||
|  |  | ||||||
| def default_token_key() -> str: | def default_token_key(): | ||||||
|     """Default token key""" |     """Default token key""" | ||||||
|     current_tenant = get_current_tenant() |  | ||||||
|     token_length = ( |  | ||||||
|         current_tenant.default_token_length |  | ||||||
|         if hasattr(current_tenant, "default_token_length") |  | ||||||
|         else DEFAULT_TOKEN_LENGTH |  | ||||||
|     ) |  | ||||||
|     # We use generate_id since the chars in the key should be easy |     # We use generate_id since the chars in the key should be easy | ||||||
|     # to use in Emails (for verification) and URLs (for recovery) |     # to use in Emails (for verification) and URLs (for recovery) | ||||||
|     return generate_id(token_length) |     return generate_id(CONFIG.get_int("default_token_length")) | ||||||
|  |  | ||||||
|  |  | ||||||
| class UserTypes(models.TextChoices): | class UserTypes(models.TextChoices): | ||||||
| @ -99,40 +86,6 @@ class UserTypes(models.TextChoices): | |||||||
|     INTERNAL_SERVICE_ACCOUNT = "internal_service_account" |     INTERNAL_SERVICE_ACCOUNT = "internal_service_account" | ||||||
|  |  | ||||||
|  |  | ||||||
| class GroupQuerySet(CTEQuerySet): |  | ||||||
|     def with_children_recursive(self): |  | ||||||
|         """Recursively get all groups that have the current queryset as parents |  | ||||||
|         or are indirectly related.""" |  | ||||||
|  |  | ||||||
|         def make_cte(cte): |  | ||||||
|             """Build the query that ends up in WITH RECURSIVE""" |  | ||||||
|             # Start from self, aka the current query |  | ||||||
|             # Add a depth attribute to limit the recursion |  | ||||||
|             return self.annotate( |  | ||||||
|                 relative_depth=models.Value(0, output_field=models.IntegerField()) |  | ||||||
|             ).union( |  | ||||||
|                 # Here is the recursive part of the query. cte refers to the previous iteration |  | ||||||
|                 # Only select groups for which the parent is part of the previous iteration |  | ||||||
|                 # and increase the depth |  | ||||||
|                 # Finally, limit the depth |  | ||||||
|                 cte.join(Group, group_uuid=cte.col.parent_id) |  | ||||||
|                 .annotate( |  | ||||||
|                     relative_depth=models.ExpressionWrapper( |  | ||||||
|                         cte.col.relative_depth |  | ||||||
|                         + models.Value(1, output_field=models.IntegerField()), |  | ||||||
|                         output_field=models.IntegerField(), |  | ||||||
|                     ) |  | ||||||
|                 ) |  | ||||||
|                 .filter(relative_depth__lt=GROUP_RECURSION_LIMIT), |  | ||||||
|                 all=True, |  | ||||||
|             ) |  | ||||||
|  |  | ||||||
|         # Build the recursive query, see above |  | ||||||
|         cte = With.recursive(make_cte) |  | ||||||
|         # Return the result, as a usable queryset for Group. |  | ||||||
|         return cte.join(Group, group_uuid=cte.col.group_uuid).with_cte(cte) |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class Group(SerializerModel): | class Group(SerializerModel): | ||||||
|     """Group model which supports a basic hierarchy and has attributes""" |     """Group model which supports a basic hierarchy and has attributes""" | ||||||
|  |  | ||||||
| @ -155,8 +108,6 @@ class Group(SerializerModel): | |||||||
|     ) |     ) | ||||||
|     attributes = models.JSONField(default=dict, blank=True) |     attributes = models.JSONField(default=dict, blank=True) | ||||||
|  |  | ||||||
|     objects = GroupQuerySet.as_manager() |  | ||||||
|  |  | ||||||
|     @property |     @property | ||||||
|     def serializer(self) -> Serializer: |     def serializer(self) -> Serializer: | ||||||
|         from authentik.core.api.groups import GroupSerializer |         from authentik.core.api.groups import GroupSerializer | ||||||
| @ -175,11 +126,36 @@ class Group(SerializerModel): | |||||||
|         return user.all_groups().filter(group_uuid=self.group_uuid).exists() |         return user.all_groups().filter(group_uuid=self.group_uuid).exists() | ||||||
|  |  | ||||||
|     def children_recursive(self: Self | QuerySet["Group"]) -> QuerySet["Group"]: |     def children_recursive(self: Self | QuerySet["Group"]) -> QuerySet["Group"]: | ||||||
|         """Compatibility layer for Group.objects.with_children_recursive()""" |         """Recursively get all groups that have this as parent or are indirectly related""" | ||||||
|         qs = self |         direct_groups = [] | ||||||
|         if not isinstance(self, QuerySet): |         if isinstance(self, QuerySet): | ||||||
|             qs = Group.objects.filter(group_uuid=self.group_uuid) |             direct_groups = list(x for x in self.all().values_list("pk", flat=True).iterator()) | ||||||
|         return qs.with_children_recursive() |         else: | ||||||
|  |             direct_groups = [self.pk] | ||||||
|  |         if len(direct_groups) < 1: | ||||||
|  |             return Group.objects.none() | ||||||
|  |         query = """ | ||||||
|  |         WITH RECURSIVE parents AS ( | ||||||
|  |             SELECT authentik_core_group.*, 0 AS relative_depth | ||||||
|  |             FROM authentik_core_group | ||||||
|  |             WHERE authentik_core_group.group_uuid = ANY(%s) | ||||||
|  |  | ||||||
|  |             UNION ALL | ||||||
|  |  | ||||||
|  |             SELECT authentik_core_group.*, parents.relative_depth + 1 | ||||||
|  |             FROM authentik_core_group, parents | ||||||
|  |             WHERE ( | ||||||
|  |                 authentik_core_group.group_uuid = parents.parent_id and | ||||||
|  |                 parents.relative_depth < 20 | ||||||
|  |             ) | ||||||
|  |         ) | ||||||
|  |         SELECT group_uuid | ||||||
|  |         FROM parents | ||||||
|  |         GROUP BY group_uuid, name | ||||||
|  |         ORDER BY name; | ||||||
|  |         """ | ||||||
|  |         group_pks = [group.pk for group in Group.objects.raw(query, [direct_groups]).iterator()] | ||||||
|  |         return Group.objects.filter(pk__in=group_pks) | ||||||
|  |  | ||||||
|     def __str__(self): |     def __str__(self): | ||||||
|         return f"Group {self.name}" |         return f"Group {self.name}" | ||||||
| @ -191,13 +167,8 @@ class Group(SerializerModel): | |||||||
|                 "parent", |                 "parent", | ||||||
|             ), |             ), | ||||||
|         ) |         ) | ||||||
|         indexes = [models.Index(fields=["name"])] |  | ||||||
|         verbose_name = _("Group") |         verbose_name = _("Group") | ||||||
|         verbose_name_plural = _("Groups") |         verbose_name_plural = _("Groups") | ||||||
|         permissions = [ |  | ||||||
|             ("add_user_to_group", _("Add user to group")), |  | ||||||
|             ("remove_user_from_group", _("Remove user from group")), |  | ||||||
|         ] |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class UserQuerySet(models.QuerySet): | class UserQuerySet(models.QuerySet): | ||||||
| @ -246,10 +217,12 @@ class User(SerializerModel, GuardianUserMixin, AbstractUser): | |||||||
|         return User._meta.get_field("path").default |         return User._meta.get_field("path").default | ||||||
|  |  | ||||||
|     def all_groups(self) -> QuerySet[Group]: |     def all_groups(self) -> QuerySet[Group]: | ||||||
|         """Recursively get all groups this user is a member of.""" |         """Recursively get all groups this user is a member of. | ||||||
|         return self.ak_groups.all().with_children_recursive() |         At least one query is done to get the direct groups of the user, with groups | ||||||
|  |         there are at most 3 queries done""" | ||||||
|  |         return Group.children_recursive(self.ak_groups.all()) | ||||||
|  |  | ||||||
|     def group_attributes(self, request: HttpRequest | None = None) -> dict[str, Any]: |     def group_attributes(self, request: Optional[HttpRequest] = None) -> dict[str, Any]: | ||||||
|         """Get a dictionary containing the attributes from all groups the user belongs to, |         """Get a dictionary containing the attributes from all groups the user belongs to, | ||||||
|         including the users attributes""" |         including the users attributes""" | ||||||
|         final_attributes = {} |         final_attributes = {} | ||||||
| @ -303,13 +276,13 @@ class User(SerializerModel, GuardianUserMixin, AbstractUser): | |||||||
|     @property |     @property | ||||||
|     def uid(self) -> str: |     def uid(self) -> str: | ||||||
|         """Generate a globally unique UID, based on the user ID and the hashed secret key""" |         """Generate a globally unique UID, based on the user ID and the hashed secret key""" | ||||||
|         return sha256(f"{self.id}-{get_unique_identifier()}".encode("ascii")).hexdigest() |         return sha256(f"{self.id}-{get_install_id()}".encode("ascii")).hexdigest() | ||||||
|  |  | ||||||
|     def locale(self, request: HttpRequest | None = None) -> str: |     def locale(self, request: Optional[HttpRequest] = None) -> str: | ||||||
|         """Get the locale the user has configured""" |         """Get the locale the user has configured""" | ||||||
|         try: |         try: | ||||||
|             return self.attributes.get("settings", {}).get("locale", "") |             return self.attributes.get("settings", {}).get("locale", "") | ||||||
|  |         # pylint: disable=broad-except | ||||||
|         except Exception as exc: |         except Exception as exc: | ||||||
|             LOGGER.warning("Failed to get default locale", exc=exc) |             LOGGER.warning("Failed to get default locale", exc=exc) | ||||||
|         if request: |         if request: | ||||||
| @ -332,12 +305,13 @@ class User(SerializerModel, GuardianUserMixin, AbstractUser): | |||||||
|             ("preview_user", _("Can preview user data sent to providers")), |             ("preview_user", _("Can preview user data sent to providers")), | ||||||
|             ("view_user_applications", _("View applications the user has access to")), |             ("view_user_applications", _("View applications the user has access to")), | ||||||
|         ] |         ] | ||||||
|         indexes = [ |         authentik_signals_ignored_fields = [ | ||||||
|             models.Index(fields=["last_login"]), |             # Logged by the events `password_set` | ||||||
|             models.Index(fields=["password_change_date"]), |             # the `password_set` action/signal doesn't currently convey which user | ||||||
|             models.Index(fields=["uuid"]), |             # initiated the password change, so for now we'll log two actions | ||||||
|             models.Index(fields=["path"]), |             # ("password", "password_change_date"), | ||||||
|             models.Index(fields=["type"]), |             # Logged by `login` | ||||||
|  |             ("last_login",), | ||||||
|         ] |         ] | ||||||
|  |  | ||||||
|  |  | ||||||
| @ -384,15 +358,11 @@ class Provider(SerializerModel): | |||||||
|     objects = InheritanceManager() |     objects = InheritanceManager() | ||||||
|  |  | ||||||
|     @property |     @property | ||||||
|     def launch_url(self) -> str | None: |     def launch_url(self) -> Optional[str]: | ||||||
|         """URL to this provider and initiate authorization for the user. |         """URL to this provider and initiate authorization for the user. | ||||||
|         Can return None for providers that are not URL-based""" |         Can return None for providers that are not URL-based""" | ||||||
|         return None |         return None | ||||||
|  |  | ||||||
|     @property |  | ||||||
|     def icon_url(self) -> str | None: |  | ||||||
|         return None |  | ||||||
|  |  | ||||||
|     @property |     @property | ||||||
|     def component(self) -> str: |     def component(self) -> str: | ||||||
|         """Return component used to edit this object""" |         """Return component used to edit this object""" | ||||||
| @ -465,7 +435,7 @@ class Application(SerializerModel, PolicyBindingModel): | |||||||
|         return ApplicationSerializer |         return ApplicationSerializer | ||||||
|  |  | ||||||
|     @property |     @property | ||||||
|     def get_meta_icon(self) -> str | None: |     def get_meta_icon(self) -> Optional[str]: | ||||||
|         """Get the URL to the App Icon image. If the name is /static or starts with http |         """Get the URL to the App Icon image. If the name is /static or starts with http | ||||||
|         it is returned as-is""" |         it is returned as-is""" | ||||||
|         if not self.meta_icon: |         if not self.meta_icon: | ||||||
| @ -474,7 +444,7 @@ class Application(SerializerModel, PolicyBindingModel): | |||||||
|             return self.meta_icon.name |             return self.meta_icon.name | ||||||
|         return self.meta_icon.url |         return self.meta_icon.url | ||||||
|  |  | ||||||
|     def get_launch_url(self, user: Optional["User"] = None) -> str | None: |     def get_launch_url(self, user: Optional["User"] = None) -> Optional[str]: | ||||||
|         """Get launch URL if set, otherwise attempt to get launch URL based on provider.""" |         """Get launch URL if set, otherwise attempt to get launch URL based on provider.""" | ||||||
|         url = None |         url = None | ||||||
|         if self.meta_launch_url: |         if self.meta_launch_url: | ||||||
| @ -487,13 +457,13 @@ class Application(SerializerModel, PolicyBindingModel): | |||||||
|                 user = user._wrapped |                 user = user._wrapped | ||||||
|             try: |             try: | ||||||
|                 return url % user.__dict__ |                 return url % user.__dict__ | ||||||
|  |             # pylint: disable=broad-except | ||||||
|             except Exception as exc: |             except Exception as exc: | ||||||
|                 LOGGER.warning("Failed to format launch url", exc=exc) |                 LOGGER.warning("Failed to format launch url", exc=exc) | ||||||
|                 return url |                 return url | ||||||
|         return url |         return url | ||||||
|  |  | ||||||
|     def get_provider(self) -> Provider | None: |     def get_provider(self) -> Optional[Provider]: | ||||||
|         """Get casted provider instance""" |         """Get casted provider instance""" | ||||||
|         if not self.provider: |         if not self.provider: | ||||||
|             return None |             return None | ||||||
| @ -581,7 +551,7 @@ class Source(ManagedModel, SerializerModel, PolicyBindingModel): | |||||||
|     objects = InheritanceManager() |     objects = InheritanceManager() | ||||||
|  |  | ||||||
|     @property |     @property | ||||||
|     def icon_url(self) -> str | None: |     def icon_url(self) -> Optional[str]: | ||||||
|         """Get the URL to the Icon. If the name is /static or |         """Get the URL to the Icon. If the name is /static or | ||||||
|         starts with http it is returned as-is""" |         starts with http it is returned as-is""" | ||||||
|         if not self.icon: |         if not self.icon: | ||||||
| @ -596,7 +566,7 @@ class Source(ManagedModel, SerializerModel, PolicyBindingModel): | |||||||
|             return self.user_path_template % { |             return self.user_path_template % { | ||||||
|                 "slug": self.slug, |                 "slug": self.slug, | ||||||
|             } |             } | ||||||
|  |         # pylint: disable=broad-except | ||||||
|         except Exception as exc: |         except Exception as exc: | ||||||
|             LOGGER.warning("Failed to template user path", exc=exc, source=self) |             LOGGER.warning("Failed to template user path", exc=exc, source=self) | ||||||
|             return User.default_path() |             return User.default_path() | ||||||
| @ -606,12 +576,12 @@ class Source(ManagedModel, SerializerModel, PolicyBindingModel): | |||||||
|         """Return component used to edit this object""" |         """Return component used to edit this object""" | ||||||
|         raise NotImplementedError |         raise NotImplementedError | ||||||
|  |  | ||||||
|     def ui_login_button(self, request: HttpRequest) -> UILoginButton | None: |     def ui_login_button(self, request: HttpRequest) -> Optional[UILoginButton]: | ||||||
|         """If source uses a http-based flow, return UI Information about the login |         """If source uses a http-based flow, return UI Information about the login | ||||||
|         button. If source doesn't use http-based flow, return None.""" |         button. If source doesn't use http-based flow, return None.""" | ||||||
|         return None |         return None | ||||||
|  |  | ||||||
|     def ui_user_settings(self) -> UserSettingSerializer | None: |     def ui_user_settings(self) -> Optional[UserSettingSerializer]: | ||||||
|         """Entrypoint to integrate with User settings. Can either return None if no |         """Entrypoint to integrate with User settings. Can either return None if no | ||||||
|         user settings are available, or UserSettingSerializer.""" |         user settings are available, or UserSettingSerializer.""" | ||||||
|         return None |         return None | ||||||
| @ -647,9 +617,6 @@ class UserSourceConnection(SerializerModel, CreatedUpdatedModel): | |||||||
|         """Get serializer for this model""" |         """Get serializer for this model""" | ||||||
|         raise NotImplementedError |         raise NotImplementedError | ||||||
|  |  | ||||||
|     def __str__(self) -> str: |  | ||||||
|         return f"User-source connection (user={self.user_id}, source={self.source_id})" |  | ||||||
|  |  | ||||||
|     class Meta: |     class Meta: | ||||||
|         unique_together = (("user", "source"),) |         unique_together = (("user", "source"),) | ||||||
|  |  | ||||||
| @ -657,12 +624,9 @@ class UserSourceConnection(SerializerModel, CreatedUpdatedModel): | |||||||
| class ExpiringModel(models.Model): | class ExpiringModel(models.Model): | ||||||
|     """Base Model which can expire, and is automatically cleaned up.""" |     """Base Model which can expire, and is automatically cleaned up.""" | ||||||
|  |  | ||||||
|     expires = models.DateTimeField(default=None, null=True) |     expires = models.DateTimeField(default=default_token_duration) | ||||||
|     expiring = models.BooleanField(default=True) |     expiring = models.BooleanField(default=True) | ||||||
|  |  | ||||||
|     class Meta: |  | ||||||
|         abstract = True |  | ||||||
|  |  | ||||||
|     def expire_action(self, *args, **kwargs): |     def expire_action(self, *args, **kwargs): | ||||||
|         """Handler which is called when this object is expired. By |         """Handler which is called when this object is expired. By | ||||||
|         default the object is deleted. This is less efficient compared |         default the object is deleted. This is less efficient compared | ||||||
| @ -671,7 +635,7 @@ class ExpiringModel(models.Model): | |||||||
|         return self.delete(*args, **kwargs) |         return self.delete(*args, **kwargs) | ||||||
|  |  | ||||||
|     @classmethod |     @classmethod | ||||||
|     def filter_not_expired(cls, **kwargs) -> QuerySet["Token"]: |     def filter_not_expired(cls, **kwargs) -> QuerySet: | ||||||
|         """Filer for tokens which are not expired yet or are not expiring, |         """Filer for tokens which are not expired yet or are not expiring, | ||||||
|         and match filters in `kwargs`""" |         and match filters in `kwargs`""" | ||||||
|         for obj in cls.objects.filter(**kwargs).filter(Q(expires__lt=now(), expiring=True)): |         for obj in cls.objects.filter(**kwargs).filter(Q(expires__lt=now(), expiring=True)): | ||||||
| @ -685,6 +649,9 @@ class ExpiringModel(models.Model): | |||||||
|             return False |             return False | ||||||
|         return now() > self.expires |         return now() > self.expires | ||||||
|  |  | ||||||
|  |     class Meta: | ||||||
|  |         abstract = True | ||||||
|  |  | ||||||
|  |  | ||||||
| class TokenIntents(models.TextChoices): | class TokenIntents(models.TextChoices): | ||||||
|     """Intents a Token can be created for.""" |     """Intents a Token can be created for.""" | ||||||
| @ -714,21 +681,6 @@ class Token(SerializerModel, ManagedModel, ExpiringModel): | |||||||
|     user = models.ForeignKey("User", on_delete=models.CASCADE, related_name="+") |     user = models.ForeignKey("User", on_delete=models.CASCADE, related_name="+") | ||||||
|     description = models.TextField(default="", blank=True) |     description = models.TextField(default="", blank=True) | ||||||
|  |  | ||||||
|     class Meta: |  | ||||||
|         verbose_name = _("Token") |  | ||||||
|         verbose_name_plural = _("Tokens") |  | ||||||
|         indexes = [ |  | ||||||
|             models.Index(fields=["identifier"]), |  | ||||||
|             models.Index(fields=["key"]), |  | ||||||
|         ] |  | ||||||
|         permissions = [("view_token_key", _("View token's key"))] |  | ||||||
|  |  | ||||||
|     def __str__(self): |  | ||||||
|         description = f"{self.identifier}" |  | ||||||
|         if self.expiring: |  | ||||||
|             description += f" (expires={self.expires})" |  | ||||||
|         return description |  | ||||||
|  |  | ||||||
|     @property |     @property | ||||||
|     def serializer(self) -> type[Serializer]: |     def serializer(self) -> type[Serializer]: | ||||||
|         from authentik.core.api.tokens import TokenSerializer |         from authentik.core.api.tokens import TokenSerializer | ||||||
| @ -756,6 +708,21 @@ class Token(SerializerModel, ManagedModel, ExpiringModel): | |||||||
|             message=f"Token {self.identifier}'s secret was rotated.", |             message=f"Token {self.identifier}'s secret was rotated.", | ||||||
|         ).save() |         ).save() | ||||||
|  |  | ||||||
|  |     def __str__(self): | ||||||
|  |         description = f"{self.identifier}" | ||||||
|  |         if self.expiring: | ||||||
|  |             description += f" (expires={self.expires})" | ||||||
|  |         return description | ||||||
|  |  | ||||||
|  |     class Meta: | ||||||
|  |         verbose_name = _("Token") | ||||||
|  |         verbose_name_plural = _("Tokens") | ||||||
|  |         indexes = [ | ||||||
|  |             models.Index(fields=["identifier"]), | ||||||
|  |             models.Index(fields=["key"]), | ||||||
|  |         ] | ||||||
|  |         permissions = [("view_token_key", _("View token's key"))] | ||||||
|  |  | ||||||
|  |  | ||||||
| class PropertyMapping(SerializerModel, ManagedModel): | class PropertyMapping(SerializerModel, ManagedModel): | ||||||
|     """User-defined key -> x mapping which can be used by providers to expose extra data.""" |     """User-defined key -> x mapping which can be used by providers to expose extra data.""" | ||||||
| @ -776,7 +743,7 @@ class PropertyMapping(SerializerModel, ManagedModel): | |||||||
|         """Get serializer for this model""" |         """Get serializer for this model""" | ||||||
|         raise NotImplementedError |         raise NotImplementedError | ||||||
|  |  | ||||||
|     def evaluate(self, user: User | None, request: HttpRequest | None, **kwargs) -> Any: |     def evaluate(self, user: Optional[User], request: Optional[HttpRequest], **kwargs) -> Any: | ||||||
|         """Evaluate `self.expression` using `**kwargs` as Context.""" |         """Evaluate `self.expression` using `**kwargs` as Context.""" | ||||||
|         from authentik.core.expression.evaluator import PropertyMappingEvaluator |         from authentik.core.expression.evaluator import PropertyMappingEvaluator | ||||||
|  |  | ||||||
| @ -784,7 +751,7 @@ class PropertyMapping(SerializerModel, ManagedModel): | |||||||
|         try: |         try: | ||||||
|             return evaluator.evaluate(self.expression) |             return evaluator.evaluate(self.expression) | ||||||
|         except Exception as exc: |         except Exception as exc: | ||||||
|             raise PropertyMappingExpressionException(self, exc) from exc |             raise PropertyMappingExpressionException(exc) from exc | ||||||
|  |  | ||||||
|     def __str__(self): |     def __str__(self): | ||||||
|         return f"Property Mapping {self.name}" |         return f"Property Mapping {self.name}" | ||||||
| @ -812,13 +779,6 @@ class AuthenticatedSession(ExpiringModel): | |||||||
|     last_user_agent = models.TextField(blank=True) |     last_user_agent = models.TextField(blank=True) | ||||||
|     last_used = models.DateTimeField(auto_now=True) |     last_used = models.DateTimeField(auto_now=True) | ||||||
|  |  | ||||||
|     class Meta: |  | ||||||
|         verbose_name = _("Authenticated Session") |  | ||||||
|         verbose_name_plural = _("Authenticated Sessions") |  | ||||||
|  |  | ||||||
|     def __str__(self) -> str: |  | ||||||
|         return f"Authenticated Session {self.session_key[:10]}" |  | ||||||
|  |  | ||||||
|     @staticmethod |     @staticmethod | ||||||
|     def from_request(request: HttpRequest, user: User) -> Optional["AuthenticatedSession"]: |     def from_request(request: HttpRequest, user: User) -> Optional["AuthenticatedSession"]: | ||||||
|         """Create a new session from a http request""" |         """Create a new session from a http request""" | ||||||
| @ -833,3 +793,7 @@ class AuthenticatedSession(ExpiringModel): | |||||||
|             last_user_agent=request.META.get("HTTP_USER_AGENT", ""), |             last_user_agent=request.META.get("HTTP_USER_AGENT", ""), | ||||||
|             expires=request.session.get_expiry_date(), |             expires=request.session.get_expiry_date(), | ||||||
|         ) |         ) | ||||||
|  |  | ||||||
|  |     class Meta: | ||||||
|  |         verbose_name = _("Authenticated Session") | ||||||
|  |         verbose_name_plural = _("Authenticated Sessions") | ||||||
|  | |||||||
| @ -10,14 +10,7 @@ from django.dispatch import receiver | |||||||
| from django.http.request import HttpRequest | from django.http.request import HttpRequest | ||||||
| from structlog.stdlib import get_logger | from structlog.stdlib import get_logger | ||||||
|  |  | ||||||
| from authentik.core.models import ( | from authentik.core.models import Application, AuthenticatedSession, BackchannelProvider, User | ||||||
|     Application, |  | ||||||
|     AuthenticatedSession, |  | ||||||
|     BackchannelProvider, |  | ||||||
|     ExpiringModel, |  | ||||||
|     User, |  | ||||||
|     default_token_duration, |  | ||||||
| ) |  | ||||||
|  |  | ||||||
| # Arguments: user: User, password: str | # Arguments: user: User, password: str | ||||||
| password_changed = Signal() | password_changed = Signal() | ||||||
| @ -68,12 +61,3 @@ def backchannel_provider_pre_save(sender: type[Model], instance: Model, **_): | |||||||
|     if not isinstance(instance, BackchannelProvider): |     if not isinstance(instance, BackchannelProvider): | ||||||
|         return |         return | ||||||
|     instance.is_backchannel = True |     instance.is_backchannel = True | ||||||
|  |  | ||||||
|  |  | ||||||
| @receiver(pre_save) |  | ||||||
| def expiring_model_pre_save(sender: type[Model], instance: Model, **_): |  | ||||||
|     """Ensure expires is set on ExpiringModels that are set to expire""" |  | ||||||
|     if not issubclass(sender, ExpiringModel): |  | ||||||
|         return |  | ||||||
|     if instance.expiring and instance.expires is None: |  | ||||||
|         instance.expires = default_token_duration() |  | ||||||
|  | |||||||
| @ -1,7 +1,7 @@ | |||||||
| """Source decision helper""" | """Source decision helper""" | ||||||
|  |  | ||||||
| from enum import Enum | from enum import Enum | ||||||
| from typing import Any | from typing import Any, Optional | ||||||
|  |  | ||||||
| from django.contrib import messages | from django.contrib import messages | ||||||
| from django.db import IntegrityError | from django.db import IntegrityError | ||||||
| @ -13,12 +13,11 @@ from django.utils.translation import gettext as _ | |||||||
| from structlog.stdlib import get_logger | from structlog.stdlib import get_logger | ||||||
|  |  | ||||||
| from authentik.core.models import Source, SourceUserMatchingModes, User, UserSourceConnection | from authentik.core.models import Source, SourceUserMatchingModes, User, UserSourceConnection | ||||||
| from authentik.core.sources.stage import PLAN_CONTEXT_SOURCES_CONNECTION, PostSourceStage | from authentik.core.sources.stage import PLAN_CONTEXT_SOURCES_CONNECTION, PostUserEnrollmentStage | ||||||
| from authentik.events.models import Event, EventAction | from authentik.events.models import Event, EventAction | ||||||
| from authentik.flows.exceptions import FlowNonApplicableException | from authentik.flows.exceptions import FlowNonApplicableException | ||||||
| from authentik.flows.models import Flow, FlowToken, Stage, in_memory_stage | from authentik.flows.models import Flow, Stage, in_memory_stage | ||||||
| from authentik.flows.planner import ( | from authentik.flows.planner import ( | ||||||
|     PLAN_CONTEXT_IS_RESTORED, |  | ||||||
|     PLAN_CONTEXT_PENDING_USER, |     PLAN_CONTEXT_PENDING_USER, | ||||||
|     PLAN_CONTEXT_REDIRECT, |     PLAN_CONTEXT_REDIRECT, | ||||||
|     PLAN_CONTEXT_SOURCE, |     PLAN_CONTEXT_SOURCE, | ||||||
| @ -36,8 +35,6 @@ from authentik.stages.password.stage import PLAN_CONTEXT_AUTHENTICATION_BACKEND | |||||||
| from authentik.stages.prompt.stage import PLAN_CONTEXT_PROMPT | from authentik.stages.prompt.stage import PLAN_CONTEXT_PROMPT | ||||||
| from authentik.stages.user_write.stage import PLAN_CONTEXT_USER_PATH | from authentik.stages.user_write.stage import PLAN_CONTEXT_USER_PATH | ||||||
|  |  | ||||||
| SESSION_KEY_OVERRIDE_FLOW_TOKEN = "authentik/flows/source_override_flow_token"  # nosec |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class Action(Enum): | class Action(Enum): | ||||||
|     """Actions that can be decided based on the request |     """Actions that can be decided based on the request | ||||||
| @ -93,13 +90,16 @@ class SourceFlowManager: | |||||||
|         self._logger = get_logger().bind(source=source, identifier=identifier) |         self._logger = get_logger().bind(source=source, identifier=identifier) | ||||||
|         self.policy_context = {} |         self.policy_context = {} | ||||||
|  |  | ||||||
|     def get_action(self, **kwargs) -> tuple[Action, UserSourceConnection | None]:  # noqa: PLR0911 |     # pylint: disable=too-many-return-statements | ||||||
|  |     def get_action(self, **kwargs) -> tuple[Action, Optional[UserSourceConnection]]: | ||||||
|         """decide which action should be taken""" |         """decide which action should be taken""" | ||||||
|         new_connection = self.connection_type(source=self.source, identifier=self.identifier) |         new_connection = self.connection_type(source=self.source, identifier=self.identifier) | ||||||
|         # When request is authenticated, always link |         # When request is authenticated, always link | ||||||
|         if self.request.user.is_authenticated: |         if self.request.user.is_authenticated: | ||||||
|             new_connection.user = self.request.user |             new_connection.user = self.request.user | ||||||
|             new_connection = self.update_connection(new_connection, **kwargs) |             new_connection = self.update_connection(new_connection, **kwargs) | ||||||
|  |             # pylint: disable=no-member | ||||||
|  |             new_connection.save() | ||||||
|             return Action.LINK, new_connection |             return Action.LINK, new_connection | ||||||
|  |  | ||||||
|         existing_connections = self.connection_type.objects.filter( |         existing_connections = self.connection_type.objects.filter( | ||||||
| @ -146,6 +146,7 @@ class SourceFlowManager: | |||||||
|         ]: |         ]: | ||||||
|             new_connection.user = user |             new_connection.user = user | ||||||
|             new_connection = self.update_connection(new_connection, **kwargs) |             new_connection = self.update_connection(new_connection, **kwargs) | ||||||
|  |             new_connection.save() | ||||||
|             return Action.LINK, new_connection |             return Action.LINK, new_connection | ||||||
|         if self.source.user_matching_mode in [ |         if self.source.user_matching_mode in [ | ||||||
|             SourceUserMatchingModes.EMAIL_DENY, |             SourceUserMatchingModes.EMAIL_DENY, | ||||||
| @ -187,10 +188,8 @@ class SourceFlowManager: | |||||||
|         # Default case, assume deny |         # Default case, assume deny | ||||||
|         error = Exception( |         error = Exception( | ||||||
|             _( |             _( | ||||||
|                 "Request to authenticate with {source} has been denied. Please authenticate " |                 "Request to authenticate with %(source)s has been denied. Please authenticate " | ||||||
|                 "with the source you've previously signed up with.".format_map( |                 "with the source you've previously signed up with." % {"source": self.source.name} | ||||||
|                     {"source": self.source.name} |  | ||||||
|                 ) |  | ||||||
|             ), |             ), | ||||||
|         ) |         ) | ||||||
|         return self.error_handler(error) |         return self.error_handler(error) | ||||||
| @ -206,55 +205,38 @@ class SourceFlowManager: | |||||||
|  |  | ||||||
|     def get_stages_to_append(self, flow: Flow) -> list[Stage]: |     def get_stages_to_append(self, flow: Flow) -> list[Stage]: | ||||||
|         """Hook to override stages which are appended to the flow""" |         """Hook to override stages which are appended to the flow""" | ||||||
|  |         if not self.source.enrollment_flow: | ||||||
|  |             return [] | ||||||
|  |         if flow.slug == self.source.enrollment_flow.slug: | ||||||
|             return [ |             return [ | ||||||
|             in_memory_stage(PostSourceStage), |                 in_memory_stage(PostUserEnrollmentStage), | ||||||
|             ] |             ] | ||||||
|  |         return [] | ||||||
|  |  | ||||||
|     def _prepare_flow( |     def _prepare_flow( | ||||||
|         self, |         self, | ||||||
|         flow: Flow, |         flow: Flow, | ||||||
|         connection: UserSourceConnection, |         connection: UserSourceConnection, | ||||||
|         stages: list[StageView] | None = None, |         stages: Optional[list[StageView]] = None, | ||||||
|         **kwargs, |         **kwargs, | ||||||
|     ) -> HttpResponse: |     ) -> HttpResponse: | ||||||
|         """Prepare Authentication Plan, redirect user FlowExecutor""" |         """Prepare Authentication Plan, redirect user FlowExecutor""" | ||||||
|  |         # Ensure redirect is carried through when user was trying to | ||||||
|  |         # authorize application | ||||||
|  |         final_redirect = self.request.session.get(SESSION_KEY_GET, {}).get( | ||||||
|  |             NEXT_ARG_NAME, "authentik_core:if-user" | ||||||
|  |         ) | ||||||
|         kwargs.update( |         kwargs.update( | ||||||
|             { |             { | ||||||
|                 # Since we authenticate the user by their token, they have no backend set |                 # Since we authenticate the user by their token, they have no backend set | ||||||
|                 PLAN_CONTEXT_AUTHENTICATION_BACKEND: BACKEND_INBUILT, |                 PLAN_CONTEXT_AUTHENTICATION_BACKEND: BACKEND_INBUILT, | ||||||
|                 PLAN_CONTEXT_SSO: True, |                 PLAN_CONTEXT_SSO: True, | ||||||
|                 PLAN_CONTEXT_SOURCE: self.source, |                 PLAN_CONTEXT_SOURCE: self.source, | ||||||
|  |                 PLAN_CONTEXT_REDIRECT: final_redirect, | ||||||
|                 PLAN_CONTEXT_SOURCES_CONNECTION: connection, |                 PLAN_CONTEXT_SOURCES_CONNECTION: connection, | ||||||
|             } |             } | ||||||
|         ) |         ) | ||||||
|         kwargs.update(self.policy_context) |         kwargs.update(self.policy_context) | ||||||
|         if SESSION_KEY_OVERRIDE_FLOW_TOKEN in self.request.session: |  | ||||||
|             token: FlowToken = self.request.session.get(SESSION_KEY_OVERRIDE_FLOW_TOKEN) |  | ||||||
|             self._logger.info("Replacing source flow with overridden flow", flow=token.flow.slug) |  | ||||||
|             plan = token.plan |  | ||||||
|             plan.context[PLAN_CONTEXT_IS_RESTORED] = token |  | ||||||
|             plan.context.update(kwargs) |  | ||||||
|             for stage in self.get_stages_to_append(flow): |  | ||||||
|                 plan.append_stage(stage) |  | ||||||
|             if stages: |  | ||||||
|                 for stage in stages: |  | ||||||
|                     plan.append_stage(stage) |  | ||||||
|             self.request.session[SESSION_KEY_PLAN] = plan |  | ||||||
|             flow_slug = token.flow.slug |  | ||||||
|             token.delete() |  | ||||||
|             return redirect_with_qs( |  | ||||||
|                 "authentik_core:if-flow", |  | ||||||
|                 self.request.GET, |  | ||||||
|                 flow_slug=flow_slug, |  | ||||||
|             ) |  | ||||||
|         # Ensure redirect is carried through when user was trying to |  | ||||||
|         # authorize application |  | ||||||
|         final_redirect = self.request.session.get(SESSION_KEY_GET, {}).get( |  | ||||||
|             NEXT_ARG_NAME, "authentik_core:if-user" |  | ||||||
|         ) |  | ||||||
|         if PLAN_CONTEXT_REDIRECT not in kwargs: |  | ||||||
|             kwargs[PLAN_CONTEXT_REDIRECT] = final_redirect |  | ||||||
|  |  | ||||||
|         if not flow: |         if not flow: | ||||||
|             return bad_request_message( |             return bad_request_message( | ||||||
|                 self.request, |                 self.request, | ||||||
| @ -262,9 +244,6 @@ class SourceFlowManager: | |||||||
|             ) |             ) | ||||||
|         # We run the Flow planner here so we can pass the Pending user in the context |         # We run the Flow planner here so we can pass the Pending user in the context | ||||||
|         planner = FlowPlanner(flow) |         planner = FlowPlanner(flow) | ||||||
|         # We append some stages so the initial flow we get might be empty |  | ||||||
|         planner.allow_empty_flows = True |  | ||||||
|         planner.use_cache = False |  | ||||||
|         plan = planner.plan(self.request, kwargs) |         plan = planner.plan(self.request, kwargs) | ||||||
|         for stage in self.get_stages_to_append(flow): |         for stage in self.get_stages_to_append(flow): | ||||||
|             plan.append_stage(stage) |             plan.append_stage(stage) | ||||||
| @ -291,9 +270,7 @@ class SourceFlowManager: | |||||||
|                 in_memory_stage( |                 in_memory_stage( | ||||||
|                     MessageStage, |                     MessageStage, | ||||||
|                     message=_( |                     message=_( | ||||||
|                         "Successfully authenticated with {source}!".format_map( |                         "Successfully authenticated with %(source)s!" % {"source": self.source.name} | ||||||
|                             {"source": self.source.name} |  | ||||||
|                         ) |  | ||||||
|                     ), |                     ), | ||||||
|                 ) |                 ) | ||||||
|             ], |             ], | ||||||
| @ -317,13 +294,13 @@ class SourceFlowManager: | |||||||
|         ).from_http(self.request) |         ).from_http(self.request) | ||||||
|         messages.success( |         messages.success( | ||||||
|             self.request, |             self.request, | ||||||
|             _("Successfully linked {source}!".format_map({"source": self.source.name})), |             _("Successfully linked %(source)s!" % {"source": self.source.name}), | ||||||
|         ) |         ) | ||||||
|         return redirect( |         return redirect( | ||||||
|             reverse( |             reverse( | ||||||
|                 "authentik_core:if-user", |                 "authentik_core:if-user", | ||||||
|             ) |             ) | ||||||
|             + "#/settings;page-sources" |             + f"#/settings;page-{self.source.slug}" | ||||||
|         ) |         ) | ||||||
|  |  | ||||||
|     def handle_enroll( |     def handle_enroll( | ||||||
| @ -345,9 +322,7 @@ class SourceFlowManager: | |||||||
|                 in_memory_stage( |                 in_memory_stage( | ||||||
|                     MessageStage, |                     MessageStage, | ||||||
|                     message=_( |                     message=_( | ||||||
|                         "Successfully authenticated with {source}!".format_map( |                         "Successfully authenticated with %(source)s!" % {"source": self.source.name} | ||||||
|                             {"source": self.source.name} |  | ||||||
|                         ) |  | ||||||
|                     ), |                     ), | ||||||
|                 ) |                 ) | ||||||
|             ], |             ], | ||||||
|  | |||||||
| @ -10,7 +10,7 @@ from authentik.flows.stage import StageView | |||||||
| PLAN_CONTEXT_SOURCES_CONNECTION = "goauthentik.io/sources/connection" | PLAN_CONTEXT_SOURCES_CONNECTION = "goauthentik.io/sources/connection" | ||||||
|  |  | ||||||
|  |  | ||||||
| class PostSourceStage(StageView): | class PostUserEnrollmentStage(StageView): | ||||||
|     """Dynamically injected stage which saves the Connection after |     """Dynamically injected stage which saves the Connection after | ||||||
|     the user has been enrolled.""" |     the user has been enrolled.""" | ||||||
|  |  | ||||||
| @ -21,9 +21,7 @@ class PostSourceStage(StageView): | |||||||
|         ] |         ] | ||||||
|         user: User = self.executor.plan.context[PLAN_CONTEXT_PENDING_USER] |         user: User = self.executor.plan.context[PLAN_CONTEXT_PENDING_USER] | ||||||
|         connection.user = user |         connection.user = user | ||||||
|         linked = connection.pk is None |  | ||||||
|         connection.save() |         connection.save() | ||||||
|         if linked: |  | ||||||
|         Event.new( |         Event.new( | ||||||
|             EventAction.SOURCE_LINKED, |             EventAction.SOURCE_LINKED, | ||||||
|             message="Linked Source", |             message="Linked Source", | ||||||
|  | |||||||
| @ -2,9 +2,7 @@ | |||||||
|  |  | ||||||
| from datetime import datetime, timedelta | from datetime import datetime, timedelta | ||||||
|  |  | ||||||
| from django.conf import ImproperlyConfigured |  | ||||||
| from django.contrib.sessions.backends.cache import KEY_PREFIX | from django.contrib.sessions.backends.cache import KEY_PREFIX | ||||||
| from django.contrib.sessions.backends.db import SessionStore as DBSessionStore |  | ||||||
| from django.core.cache import cache | from django.core.cache import cache | ||||||
| from django.utils.timezone import now | from django.utils.timezone import now | ||||||
| from structlog.stdlib import get_logger | from structlog.stdlib import get_logger | ||||||
| @ -17,7 +15,6 @@ from authentik.core.models import ( | |||||||
|     User, |     User, | ||||||
| ) | ) | ||||||
| from authentik.events.system_tasks import SystemTask, TaskStatus, prefill_task | from authentik.events.system_tasks import SystemTask, TaskStatus, prefill_task | ||||||
| from authentik.lib.config import CONFIG |  | ||||||
| from authentik.root.celery import CELERY_APP | from authentik.root.celery import CELERY_APP | ||||||
|  |  | ||||||
| LOGGER = get_logger() | LOGGER = get_logger() | ||||||
| @ -40,35 +37,20 @@ def clean_expired_models(self: SystemTask): | |||||||
|         messages.append(f"Expired {amount} {cls._meta.verbose_name_plural}") |         messages.append(f"Expired {amount} {cls._meta.verbose_name_plural}") | ||||||
|     # Special case |     # Special case | ||||||
|     amount = 0 |     amount = 0 | ||||||
|  |     # pylint: disable=no-member | ||||||
|     for session in AuthenticatedSession.objects.all(): |     for session in AuthenticatedSession.objects.all(): | ||||||
|         match CONFIG.get("session_storage", "cache"): |  | ||||||
|             case "cache": |  | ||||||
|         cache_key = f"{KEY_PREFIX}{session.session_key}" |         cache_key = f"{KEY_PREFIX}{session.session_key}" | ||||||
|         value = None |         value = None | ||||||
|         try: |         try: | ||||||
|             value = cache.get(cache_key) |             value = cache.get(cache_key) | ||||||
|  |         # pylint: disable=broad-except | ||||||
|         except Exception as exc: |         except Exception as exc: | ||||||
|             LOGGER.debug("Failed to get session from cache", exc=exc) |             LOGGER.debug("Failed to get session from cache", exc=exc) | ||||||
|         if not value: |         if not value: | ||||||
|             session.delete() |             session.delete() | ||||||
|             amount += 1 |             amount += 1 | ||||||
|             case "db": |  | ||||||
|                 if not ( |  | ||||||
|                     DBSessionStore.get_model_class() |  | ||||||
|                     .objects.filter(session_key=session.session_key, expire_date__gt=now()) |  | ||||||
|                     .exists() |  | ||||||
|                 ): |  | ||||||
|                     session.delete() |  | ||||||
|                     amount += 1 |  | ||||||
|             case _: |  | ||||||
|                 # Should never happen, as we check for other values in authentik/root/settings.py |  | ||||||
|                 raise ImproperlyConfigured( |  | ||||||
|                     "Invalid session_storage setting, allowed values are db and cache" |  | ||||||
|                 ) |  | ||||||
|     LOGGER.debug("Expired sessions", model=AuthenticatedSession, amount=amount) |     LOGGER.debug("Expired sessions", model=AuthenticatedSession, amount=amount) | ||||||
|  |     # pylint: disable=no-member | ||||||
|     messages.append(f"Expired {amount} {AuthenticatedSession._meta.verbose_name_plural}") |     messages.append(f"Expired {amount} {AuthenticatedSession._meta.verbose_name_plural}") | ||||||
|     self.set_status(TaskStatus.SUCCESSFUL, *messages) |     self.set_status(TaskStatus.SUCCESSFUL, *messages) | ||||||
|  |  | ||||||
|  | |||||||
| @ -1,11 +1,10 @@ | |||||||
| """Test Groups API""" | """Test Groups API""" | ||||||
|  |  | ||||||
| from django.urls.base import reverse | from django.urls.base import reverse | ||||||
| from guardian.shortcuts import assign_perm |  | ||||||
| from rest_framework.test import APITestCase | from rest_framework.test import APITestCase | ||||||
|  |  | ||||||
| from authentik.core.models import Group, User | from authentik.core.models import Group, User | ||||||
| from authentik.core.tests.utils import create_test_admin_user, create_test_user | from authentik.core.tests.utils import create_test_admin_user | ||||||
| from authentik.lib.generators import generate_id | from authentik.lib.generators import generate_id | ||||||
|  |  | ||||||
|  |  | ||||||
| @ -13,33 +12,13 @@ class TestGroupsAPI(APITestCase): | |||||||
|     """Test Groups API""" |     """Test Groups API""" | ||||||
|  |  | ||||||
|     def setUp(self) -> None: |     def setUp(self) -> None: | ||||||
|         self.login_user = create_test_user() |         self.admin = create_test_admin_user() | ||||||
|         self.user = User.objects.create(username="test-user") |         self.user = User.objects.create(username="test-user") | ||||||
|  |  | ||||||
|     def test_list_with_users(self): |  | ||||||
|         """Test listing with users""" |  | ||||||
|         admin = create_test_admin_user() |  | ||||||
|         self.client.force_login(admin) |  | ||||||
|         response = self.client.get(reverse("authentik_api:group-list"), {"include_users": "true"}) |  | ||||||
|         self.assertEqual(response.status_code, 200) |  | ||||||
|  |  | ||||||
|     def test_retrieve_with_users(self): |  | ||||||
|         """Test retrieve with users""" |  | ||||||
|         admin = create_test_admin_user() |  | ||||||
|         group = Group.objects.create(name=generate_id()) |  | ||||||
|         self.client.force_login(admin) |  | ||||||
|         response = self.client.get( |  | ||||||
|             reverse("authentik_api:group-detail", kwargs={"pk": group.pk}), |  | ||||||
|             {"include_users": "true"}, |  | ||||||
|         ) |  | ||||||
|         self.assertEqual(response.status_code, 200) |  | ||||||
|  |  | ||||||
|     def test_add_user(self): |     def test_add_user(self): | ||||||
|         """Test add_user""" |         """Test add_user""" | ||||||
|         group = Group.objects.create(name=generate_id()) |         group = Group.objects.create(name=generate_id()) | ||||||
|         assign_perm("authentik_core.add_user_to_group", self.login_user, group) |         self.client.force_login(self.admin) | ||||||
|         assign_perm("authentik_core.view_user", self.login_user) |  | ||||||
|         self.client.force_login(self.login_user) |  | ||||||
|         res = self.client.post( |         res = self.client.post( | ||||||
|             reverse("authentik_api:group-add-user", kwargs={"pk": group.pk}), |             reverse("authentik_api:group-add-user", kwargs={"pk": group.pk}), | ||||||
|             data={ |             data={ | ||||||
| @ -53,9 +32,7 @@ class TestGroupsAPI(APITestCase): | |||||||
|     def test_add_user_404(self): |     def test_add_user_404(self): | ||||||
|         """Test add_user""" |         """Test add_user""" | ||||||
|         group = Group.objects.create(name=generate_id()) |         group = Group.objects.create(name=generate_id()) | ||||||
|         assign_perm("authentik_core.add_user_to_group", self.login_user, group) |         self.client.force_login(self.admin) | ||||||
|         assign_perm("authentik_core.view_user", self.login_user) |  | ||||||
|         self.client.force_login(self.login_user) |  | ||||||
|         res = self.client.post( |         res = self.client.post( | ||||||
|             reverse("authentik_api:group-add-user", kwargs={"pk": group.pk}), |             reverse("authentik_api:group-add-user", kwargs={"pk": group.pk}), | ||||||
|             data={ |             data={ | ||||||
| @ -67,10 +44,8 @@ class TestGroupsAPI(APITestCase): | |||||||
|     def test_remove_user(self): |     def test_remove_user(self): | ||||||
|         """Test remove_user""" |         """Test remove_user""" | ||||||
|         group = Group.objects.create(name=generate_id()) |         group = Group.objects.create(name=generate_id()) | ||||||
|         assign_perm("authentik_core.remove_user_from_group", self.login_user, group) |  | ||||||
|         assign_perm("authentik_core.view_user", self.login_user) |  | ||||||
|         group.users.add(self.user) |         group.users.add(self.user) | ||||||
|         self.client.force_login(self.login_user) |         self.client.force_login(self.admin) | ||||||
|         res = self.client.post( |         res = self.client.post( | ||||||
|             reverse("authentik_api:group-remove-user", kwargs={"pk": group.pk}), |             reverse("authentik_api:group-remove-user", kwargs={"pk": group.pk}), | ||||||
|             data={ |             data={ | ||||||
| @ -84,10 +59,8 @@ class TestGroupsAPI(APITestCase): | |||||||
|     def test_remove_user_404(self): |     def test_remove_user_404(self): | ||||||
|         """Test remove_user""" |         """Test remove_user""" | ||||||
|         group = Group.objects.create(name=generate_id()) |         group = Group.objects.create(name=generate_id()) | ||||||
|         assign_perm("authentik_core.remove_user_from_group", self.login_user, group) |  | ||||||
|         assign_perm("authentik_core.view_user", self.login_user) |  | ||||||
|         group.users.add(self.user) |         group.users.add(self.user) | ||||||
|         self.client.force_login(self.login_user) |         self.client.force_login(self.admin) | ||||||
|         res = self.client.post( |         res = self.client.post( | ||||||
|             reverse("authentik_api:group-remove-user", kwargs={"pk": group.pk}), |             reverse("authentik_api:group-remove-user", kwargs={"pk": group.pk}), | ||||||
|             data={ |             data={ | ||||||
| @ -99,12 +72,11 @@ class TestGroupsAPI(APITestCase): | |||||||
|     def test_parent_self(self): |     def test_parent_self(self): | ||||||
|         """Test parent""" |         """Test parent""" | ||||||
|         group = Group.objects.create(name=generate_id()) |         group = Group.objects.create(name=generate_id()) | ||||||
|         assign_perm("view_group", self.login_user, group) |         self.client.force_login(self.admin) | ||||||
|         assign_perm("change_group", self.login_user, group) |  | ||||||
|         self.client.force_login(self.login_user) |  | ||||||
|         res = self.client.patch( |         res = self.client.patch( | ||||||
|             reverse("authentik_api:group-detail", kwargs={"pk": group.pk}), |             reverse("authentik_api:group-detail", kwargs={"pk": group.pk}), | ||||||
|             data={ |             data={ | ||||||
|  |                 "pk": self.user.pk + 3, | ||||||
|                 "parent": group.pk, |                 "parent": group.pk, | ||||||
|             }, |             }, | ||||||
|         ) |         ) | ||||||
|  | |||||||
| @ -1,14 +1,14 @@ | |||||||
| """authentik core models tests""" | """authentik core models tests""" | ||||||
|  |  | ||||||
| from collections.abc import Callable | from time import sleep | ||||||
| from datetime import timedelta | from typing import Callable | ||||||
|  |  | ||||||
| from django.test import RequestFactory, TestCase | from django.test import RequestFactory, TestCase | ||||||
| from django.utils.timezone import now | from django.utils.timezone import now | ||||||
| from freezegun import freeze_time |  | ||||||
| from guardian.shortcuts import get_anonymous_user | from guardian.shortcuts import get_anonymous_user | ||||||
|  |  | ||||||
| from authentik.core.models import Provider, Source, Token | from authentik.core.models import Provider, Source, Token | ||||||
|  | from authentik.flows.models import Stage | ||||||
| from authentik.lib.utils.reflection import all_subclasses | from authentik.lib.utils.reflection import all_subclasses | ||||||
|  |  | ||||||
|  |  | ||||||
| @ -17,20 +17,18 @@ class TestModels(TestCase): | |||||||
|  |  | ||||||
|     def test_token_expire(self): |     def test_token_expire(self): | ||||||
|         """Test token expiring""" |         """Test token expiring""" | ||||||
|         with freeze_time() as freeze: |  | ||||||
|         token = Token.objects.create(expires=now(), user=get_anonymous_user()) |         token = Token.objects.create(expires=now(), user=get_anonymous_user()) | ||||||
|             freeze.tick(timedelta(seconds=1)) |         sleep(0.5) | ||||||
|         self.assertTrue(token.is_expired) |         self.assertTrue(token.is_expired) | ||||||
|  |  | ||||||
|     def test_token_expire_no_expire(self): |     def test_token_expire_no_expire(self): | ||||||
|         """Test token expiring with "expiring" set""" |         """Test token expiring with "expiring" set""" | ||||||
|         with freeze_time() as freeze: |  | ||||||
|         token = Token.objects.create(expires=now(), user=get_anonymous_user(), expiring=False) |         token = Token.objects.create(expires=now(), user=get_anonymous_user(), expiring=False) | ||||||
|             freeze.tick(timedelta(seconds=1)) |         sleep(0.5) | ||||||
|         self.assertFalse(token.is_expired) |         self.assertFalse(token.is_expired) | ||||||
|  |  | ||||||
|  |  | ||||||
| def source_tester_factory(test_model: type[Source]) -> Callable: | def source_tester_factory(test_model: type[Stage]) -> Callable: | ||||||
|     """Test source""" |     """Test source""" | ||||||
|  |  | ||||||
|     factory = RequestFactory() |     factory = RequestFactory() | ||||||
| @ -38,19 +36,19 @@ def source_tester_factory(test_model: type[Source]) -> Callable: | |||||||
|  |  | ||||||
|     def tester(self: TestModels): |     def tester(self: TestModels): | ||||||
|         model_class = None |         model_class = None | ||||||
|         if test_model._meta.abstract: |         if test_model._meta.abstract:  # pragma: no cover | ||||||
|             model_class = [x for x in test_model.__bases__ if issubclass(x, Source)][0]() |             model_class = test_model.__bases__[0]() | ||||||
|         else: |         else: | ||||||
|             model_class = test_model() |             model_class = test_model() | ||||||
|         model_class.slug = "test" |         model_class.slug = "test" | ||||||
|         self.assertIsNotNone(model_class.component) |         self.assertIsNotNone(model_class.component) | ||||||
|         model_class.ui_login_button(request) |         _ = model_class.ui_login_button(request) | ||||||
|         model_class.ui_user_settings() |         _ = model_class.ui_user_settings() | ||||||
|  |  | ||||||
|     return tester |     return tester | ||||||
|  |  | ||||||
|  |  | ||||||
| def provider_tester_factory(test_model: type[Provider]) -> Callable: | def provider_tester_factory(test_model: type[Stage]) -> Callable: | ||||||
|     """Test provider""" |     """Test provider""" | ||||||
|  |  | ||||||
|     def tester(self: TestModels): |     def tester(self: TestModels): | ||||||
|  | |||||||
| @ -3,7 +3,7 @@ | |||||||
| from django.test import RequestFactory, TestCase | from django.test import RequestFactory, TestCase | ||||||
| from guardian.shortcuts import get_anonymous_user | from guardian.shortcuts import get_anonymous_user | ||||||
|  |  | ||||||
| from authentik.core.expression.exceptions import PropertyMappingExpressionException | from authentik.core.exceptions import PropertyMappingExpressionException | ||||||
| from authentik.core.models import PropertyMapping | from authentik.core.models import PropertyMapping | ||||||
| from authentik.core.tests.utils import create_test_admin_user | from authentik.core.tests.utils import create_test_admin_user | ||||||
| from authentik.events.models import Event, EventAction | from authentik.events.models import Event, EventAction | ||||||
| @ -66,11 +66,14 @@ class TestPropertyMappings(TestCase): | |||||||
|             expression="return request.http_request.path", |             expression="return request.http_request.path", | ||||||
|         ) |         ) | ||||||
|         http_request = self.factory.get("/") |         http_request = self.factory.get("/") | ||||||
|         tmpl = f""" |         tmpl = ( | ||||||
|         res = ak_call_policy('{expr.name}') |             """ | ||||||
|  |         res = ak_call_policy('%s') | ||||||
|         result = [request.http_request.path, res.raw_result] |         result = [request.http_request.path, res.raw_result] | ||||||
|         return result |         return result | ||||||
|         """ |         """ | ||||||
|  |             % expr.name | ||||||
|  |         ) | ||||||
|         evaluator = PropertyMapping(expression=tmpl, name=generate_id()) |         evaluator = PropertyMapping(expression=tmpl, name=generate_id()) | ||||||
|         res = evaluator.evaluate(self.user, http_request) |         res = evaluator.evaluate(self.user, http_request) | ||||||
|         self.assertEqual(res, ["/", "/"]) |         self.assertEqual(res, ["/", "/"]) | ||||||
|  | |||||||
| @ -6,10 +6,9 @@ from django.urls import reverse | |||||||
| from rest_framework.serializers import ValidationError | from rest_framework.serializers import ValidationError | ||||||
| from rest_framework.test import APITestCase | from rest_framework.test import APITestCase | ||||||
|  |  | ||||||
| from authentik.core.api.property_mappings import PropertyMappingSerializer | from authentik.core.api.propertymappings import PropertyMappingSerializer | ||||||
| from authentik.core.models import Group, PropertyMapping | from authentik.core.models import PropertyMapping | ||||||
| from authentik.core.tests.utils import create_test_admin_user | from authentik.core.tests.utils import create_test_admin_user | ||||||
| from authentik.lib.generators import generate_id |  | ||||||
|  |  | ||||||
|  |  | ||||||
| class TestPropertyMappingAPI(APITestCase): | class TestPropertyMappingAPI(APITestCase): | ||||||
| @ -17,40 +16,23 @@ class TestPropertyMappingAPI(APITestCase): | |||||||
|  |  | ||||||
|     def setUp(self) -> None: |     def setUp(self) -> None: | ||||||
|         super().setUp() |         super().setUp() | ||||||
|  |         self.mapping = PropertyMapping.objects.create( | ||||||
|  |             name="dummy", expression="""return {'foo': 'bar'}""" | ||||||
|  |         ) | ||||||
|         self.user = create_test_admin_user() |         self.user = create_test_admin_user() | ||||||
|         self.client.force_login(self.user) |         self.client.force_login(self.user) | ||||||
|  |  | ||||||
|     def test_test_call(self): |     def test_test_call(self): | ||||||
|         """Test PropertyMappings's test endpoint""" |         """Test PropertMappings's test endpoint""" | ||||||
|         mapping = PropertyMapping.objects.create( |  | ||||||
|             name="dummy", expression="""return {'foo': 'bar', 'baz': user.username}""" |  | ||||||
|         ) |  | ||||||
|         response = self.client.post( |         response = self.client.post( | ||||||
|             reverse("authentik_api:propertymapping-test", kwargs={"pk": mapping.pk}), |             reverse("authentik_api:propertymapping-test", kwargs={"pk": self.mapping.pk}), | ||||||
|             data={ |             data={ | ||||||
|                 "user": self.user.pk, |                 "user": self.user.pk, | ||||||
|             }, |             }, | ||||||
|         ) |         ) | ||||||
|         self.assertJSONEqual( |         self.assertJSONEqual( | ||||||
|             response.content.decode(), |             response.content.decode(), | ||||||
|             {"result": dumps({"foo": "bar", "baz": self.user.username}), "successful": True}, |             {"result": dumps({"foo": "bar"}), "successful": True}, | ||||||
|         ) |  | ||||||
|  |  | ||||||
|     def test_test_call_group(self): |  | ||||||
|         """Test PropertyMappings's test endpoint""" |  | ||||||
|         mapping = PropertyMapping.objects.create( |  | ||||||
|             name="dummy", expression="""return {'foo': 'bar', 'baz': group.name}""" |  | ||||||
|         ) |  | ||||||
|         group = Group.objects.create(name=generate_id()) |  | ||||||
|         response = self.client.post( |  | ||||||
|             reverse("authentik_api:propertymapping-test", kwargs={"pk": mapping.pk}), |  | ||||||
|             data={ |  | ||||||
|                 "group": group.pk, |  | ||||||
|             }, |  | ||||||
|         ) |  | ||||||
|         self.assertJSONEqual( |  | ||||||
|             response.content.decode(), |  | ||||||
|             {"result": dumps({"foo": "bar", "baz": group.name}), "successful": True}, |  | ||||||
|         ) |         ) | ||||||
|  |  | ||||||
|     def test_validate(self): |     def test_validate(self): | ||||||
|  | |||||||
| @ -2,15 +2,11 @@ | |||||||
|  |  | ||||||
| from django.contrib.auth.models import AnonymousUser | from django.contrib.auth.models import AnonymousUser | ||||||
| from django.test import TestCase | from django.test import TestCase | ||||||
| from django.urls import reverse |  | ||||||
| from guardian.utils import get_anonymous_user | from guardian.utils import get_anonymous_user | ||||||
|  |  | ||||||
| from authentik.core.models import SourceUserMatchingModes, User | from authentik.core.models import SourceUserMatchingModes, User | ||||||
| from authentik.core.sources.flow_manager import Action | from authentik.core.sources.flow_manager import Action | ||||||
| from authentik.core.sources.stage import PostSourceStage |  | ||||||
| from authentik.core.tests.utils import create_test_flow | from authentik.core.tests.utils import create_test_flow | ||||||
| from authentik.flows.planner import FlowPlan |  | ||||||
| from authentik.flows.views.executor import SESSION_KEY_PLAN |  | ||||||
| from authentik.lib.generators import generate_id | from authentik.lib.generators import generate_id | ||||||
| from authentik.lib.tests.utils import get_request | from authentik.lib.tests.utils import get_request | ||||||
| from authentik.policies.denied import AccessDeniedResponse | from authentik.policies.denied import AccessDeniedResponse | ||||||
| @ -25,62 +21,42 @@ class TestSourceFlowManager(TestCase): | |||||||
|  |  | ||||||
|     def setUp(self) -> None: |     def setUp(self) -> None: | ||||||
|         super().setUp() |         super().setUp() | ||||||
|         self.authentication_flow = create_test_flow() |         self.source: OAuthSource = OAuthSource.objects.create(name="test") | ||||||
|         self.enrollment_flow = create_test_flow() |  | ||||||
|         self.source: OAuthSource = OAuthSource.objects.create( |  | ||||||
|             name=generate_id(), |  | ||||||
|             slug=generate_id(), |  | ||||||
|             authentication_flow=self.authentication_flow, |  | ||||||
|             enrollment_flow=self.enrollment_flow, |  | ||||||
|         ) |  | ||||||
|         self.identifier = generate_id() |         self.identifier = generate_id() | ||||||
|  |  | ||||||
|     def test_unauthenticated_enroll(self): |     def test_unauthenticated_enroll(self): | ||||||
|         """Test un-authenticated user enrolling""" |         """Test un-authenticated user enrolling""" | ||||||
|         request = get_request("/", user=AnonymousUser()) |         flow_manager = OAuthSourceFlowManager( | ||||||
|         flow_manager = OAuthSourceFlowManager(self.source, request, self.identifier, {}) |             self.source, get_request("/", user=AnonymousUser()), self.identifier, {} | ||||||
|  |         ) | ||||||
|         action, _ = flow_manager.get_action() |         action, _ = flow_manager.get_action() | ||||||
|         self.assertEqual(action, Action.ENROLL) |         self.assertEqual(action, Action.ENROLL) | ||||||
|         response = flow_manager.get_flow() |         flow_manager.get_flow() | ||||||
|         self.assertEqual(response.status_code, 302) |  | ||||||
|         flow_plan: FlowPlan = request.session[SESSION_KEY_PLAN] |  | ||||||
|         self.assertEqual(flow_plan.bindings[0].stage.view, PostSourceStage) |  | ||||||
|  |  | ||||||
|     def test_unauthenticated_auth(self): |     def test_unauthenticated_auth(self): | ||||||
|         """Test un-authenticated user authenticating""" |         """Test un-authenticated user authenticating""" | ||||||
|         UserOAuthSourceConnection.objects.create( |         UserOAuthSourceConnection.objects.create( | ||||||
|             user=get_anonymous_user(), source=self.source, identifier=self.identifier |             user=get_anonymous_user(), source=self.source, identifier=self.identifier | ||||||
|         ) |         ) | ||||||
|         request = get_request("/", user=AnonymousUser()) |  | ||||||
|         flow_manager = OAuthSourceFlowManager(self.source, request, self.identifier, {}) |         flow_manager = OAuthSourceFlowManager( | ||||||
|  |             self.source, get_request("/", user=AnonymousUser()), self.identifier, {} | ||||||
|  |         ) | ||||||
|         action, _ = flow_manager.get_action() |         action, _ = flow_manager.get_action() | ||||||
|         self.assertEqual(action, Action.AUTH) |         self.assertEqual(action, Action.AUTH) | ||||||
|         response = flow_manager.get_flow() |         flow_manager.get_flow() | ||||||
|         self.assertEqual(response.status_code, 302) |  | ||||||
|         flow_plan: FlowPlan = request.session[SESSION_KEY_PLAN] |  | ||||||
|         self.assertEqual(flow_plan.bindings[0].stage.view, PostSourceStage) |  | ||||||
|  |  | ||||||
|     def test_authenticated_link(self): |     def test_authenticated_link(self): | ||||||
|         """Test authenticated user linking""" |         """Test authenticated user linking""" | ||||||
|         user = User.objects.create(username="foo", email="foo@bar.baz") |         UserOAuthSourceConnection.objects.create( | ||||||
|         request = get_request("/", user=user) |             user=get_anonymous_user(), source=self.source, identifier=self.identifier | ||||||
|         flow_manager = OAuthSourceFlowManager(self.source, request, self.identifier, {}) |  | ||||||
|         action, connection = flow_manager.get_action() |  | ||||||
|         self.assertEqual(action, Action.LINK) |  | ||||||
|         self.assertIsNone(connection.pk) |  | ||||||
|         response = flow_manager.get_flow() |  | ||||||
|         self.assertEqual(response.status_code, 302) |  | ||||||
|         self.assertEqual( |  | ||||||
|             response.url, |  | ||||||
|             reverse("authentik_core:if-user") + "#/settings;page-sources", |  | ||||||
|         ) |         ) | ||||||
|  |         user = User.objects.create(username="foo", email="foo@bar.baz") | ||||||
|     def test_unauthenticated_link(self): |         flow_manager = OAuthSourceFlowManager( | ||||||
|         """Test un-authenticated user linking""" |             self.source, get_request("/", user=user), self.identifier, {} | ||||||
|         flow_manager = OAuthSourceFlowManager(self.source, get_request("/"), self.identifier, {}) |         ) | ||||||
|         action, connection = flow_manager.get_action() |         action, _ = flow_manager.get_action() | ||||||
|         self.assertEqual(action, Action.LINK) |         self.assertEqual(action, Action.LINK) | ||||||
|         self.assertIsNone(connection.pk) |  | ||||||
|         flow_manager.get_flow() |         flow_manager.get_flow() | ||||||
|  |  | ||||||
|     def test_unauthenticated_enroll_email(self): |     def test_unauthenticated_enroll_email(self): | ||||||
| @ -197,5 +173,5 @@ class TestSourceFlowManager(TestCase): | |||||||
|         self.assertEqual(action, Action.ENROLL) |         self.assertEqual(action, Action.ENROLL) | ||||||
|         response = flow_manager.get_flow() |         response = flow_manager.get_flow() | ||||||
|         self.assertIsInstance(response, AccessDeniedResponse) |         self.assertIsInstance(response, AccessDeniedResponse) | ||||||
|  |         # pylint: disable=no-member | ||||||
|         self.assertEqual(response.error_message, "foo") |         self.assertEqual(response.error_message, "foo") | ||||||
|  | |||||||
| @ -1,6 +1,5 @@ | |||||||
| """Test token API""" | """Test token API""" | ||||||
|  |  | ||||||
| from datetime import datetime, timedelta |  | ||||||
| from json import loads | from json import loads | ||||||
|  |  | ||||||
| from django.urls.base import reverse | from django.urls.base import reverse | ||||||
| @ -8,13 +7,7 @@ from guardian.shortcuts import get_anonymous_user | |||||||
| from rest_framework.test import APITestCase | from rest_framework.test import APITestCase | ||||||
|  |  | ||||||
| from authentik.core.api.tokens import TokenSerializer | from authentik.core.api.tokens import TokenSerializer | ||||||
| from authentik.core.models import ( | from authentik.core.models import USER_ATTRIBUTE_TOKEN_EXPIRING, Token, TokenIntents, User | ||||||
|     USER_ATTRIBUTE_TOKEN_EXPIRING, |  | ||||||
|     USER_ATTRIBUTE_TOKEN_MAXIMUM_LIFETIME, |  | ||||||
|     Token, |  | ||||||
|     TokenIntents, |  | ||||||
|     User, |  | ||||||
| ) |  | ||||||
| from authentik.core.tests.utils import create_test_admin_user | from authentik.core.tests.utils import create_test_admin_user | ||||||
| from authentik.lib.generators import generate_id | from authentik.lib.generators import generate_id | ||||||
|  |  | ||||||
| @ -83,77 +76,6 @@ class TestTokenAPI(APITestCase): | |||||||
|         self.assertEqual(token.intent, TokenIntents.INTENT_API) |         self.assertEqual(token.intent, TokenIntents.INTENT_API) | ||||||
|         self.assertEqual(token.expiring, False) |         self.assertEqual(token.expiring, False) | ||||||
|  |  | ||||||
|     def test_token_create_expiring(self): |  | ||||||
|         """Test token creation endpoint""" |  | ||||||
|         self.user.attributes[USER_ATTRIBUTE_TOKEN_EXPIRING] = True |  | ||||||
|         self.user.save() |  | ||||||
|         response = self.client.post( |  | ||||||
|             reverse("authentik_api:token-list"), {"identifier": "test-token"} |  | ||||||
|         ) |  | ||||||
|         self.assertEqual(response.status_code, 201) |  | ||||||
|         token = Token.objects.get(identifier="test-token") |  | ||||||
|         self.assertEqual(token.user, self.user) |  | ||||||
|         self.assertEqual(token.intent, TokenIntents.INTENT_API) |  | ||||||
|         self.assertEqual(token.expiring, True) |  | ||||||
|  |  | ||||||
|     def test_token_create_expiring_custom_ok(self): |  | ||||||
|         """Test token creation endpoint""" |  | ||||||
|         self.user.attributes[USER_ATTRIBUTE_TOKEN_EXPIRING] = True |  | ||||||
|         self.user.attributes[USER_ATTRIBUTE_TOKEN_MAXIMUM_LIFETIME] = "hours=2" |  | ||||||
|         self.user.save() |  | ||||||
|         expires = datetime.now() + timedelta(hours=1) |  | ||||||
|         response = self.client.post( |  | ||||||
|             reverse("authentik_api:token-list"), |  | ||||||
|             { |  | ||||||
|                 "identifier": "test-token", |  | ||||||
|                 "expires": expires, |  | ||||||
|                 "intent": TokenIntents.INTENT_APP_PASSWORD, |  | ||||||
|             }, |  | ||||||
|         ) |  | ||||||
|         self.assertEqual(response.status_code, 201) |  | ||||||
|         token = Token.objects.get(identifier="test-token") |  | ||||||
|         self.assertEqual(token.user, self.user) |  | ||||||
|         self.assertEqual(token.intent, TokenIntents.INTENT_APP_PASSWORD) |  | ||||||
|         self.assertEqual(token.expiring, True) |  | ||||||
|         self.assertEqual(token.expires.timestamp(), expires.timestamp()) |  | ||||||
|  |  | ||||||
|     def test_token_create_expiring_custom_nok(self): |  | ||||||
|         """Test token creation endpoint""" |  | ||||||
|         self.user.attributes[USER_ATTRIBUTE_TOKEN_EXPIRING] = True |  | ||||||
|         self.user.attributes[USER_ATTRIBUTE_TOKEN_MAXIMUM_LIFETIME] = "hours=2" |  | ||||||
|         self.user.save() |  | ||||||
|         expires = datetime.now() + timedelta(hours=3) |  | ||||||
|         response = self.client.post( |  | ||||||
|             reverse("authentik_api:token-list"), |  | ||||||
|             { |  | ||||||
|                 "identifier": "test-token", |  | ||||||
|                 "expires": expires, |  | ||||||
|                 "intent": TokenIntents.INTENT_APP_PASSWORD, |  | ||||||
|             }, |  | ||||||
|         ) |  | ||||||
|         self.assertEqual(response.status_code, 400) |  | ||||||
|  |  | ||||||
|     def test_token_create_expiring_custom_api(self): |  | ||||||
|         """Test token creation endpoint""" |  | ||||||
|         self.user.attributes[USER_ATTRIBUTE_TOKEN_EXPIRING] = True |  | ||||||
|         self.user.attributes[USER_ATTRIBUTE_TOKEN_MAXIMUM_LIFETIME] = "hours=2" |  | ||||||
|         self.user.save() |  | ||||||
|         expires = datetime.now() + timedelta(seconds=3) |  | ||||||
|         response = self.client.post( |  | ||||||
|             reverse("authentik_api:token-list"), |  | ||||||
|             { |  | ||||||
|                 "identifier": "test-token", |  | ||||||
|                 "expires": expires, |  | ||||||
|                 "intent": TokenIntents.INTENT_API, |  | ||||||
|             }, |  | ||||||
|         ) |  | ||||||
|         self.assertEqual(response.status_code, 201) |  | ||||||
|         token = Token.objects.get(identifier="test-token") |  | ||||||
|         self.assertEqual(token.user, self.user) |  | ||||||
|         self.assertEqual(token.intent, TokenIntents.INTENT_API) |  | ||||||
|         self.assertEqual(token.expiring, True) |  | ||||||
|         self.assertNotEqual(token.expires.timestamp(), expires.timestamp()) |  | ||||||
|  |  | ||||||
|     def test_list(self): |     def test_list(self): | ||||||
|         """Test Token List (Test normal authentication)""" |         """Test Token List (Test normal authentication)""" | ||||||
|         Token.objects.all().delete() |         Token.objects.all().delete() | ||||||
|  | |||||||
| @ -41,12 +41,6 @@ class TestUsersAPI(APITestCase): | |||||||
|         ) |         ) | ||||||
|         self.assertEqual(response.status_code, 200) |         self.assertEqual(response.status_code, 200) | ||||||
|  |  | ||||||
|     def test_list_with_groups(self): |  | ||||||
|         """Test listing with groups""" |  | ||||||
|         self.client.force_login(self.admin) |  | ||||||
|         response = self.client.get(reverse("authentik_api:user-list"), {"include_groups": "true"}) |  | ||||||
|         self.assertEqual(response.status_code, 200) |  | ||||||
|  |  | ||||||
|     def test_metrics(self): |     def test_metrics(self): | ||||||
|         """Test user's metrics""" |         """Test user's metrics""" | ||||||
|         self.client.force_login(self.admin) |         self.client.force_login(self.admin) | ||||||
| @ -66,11 +60,10 @@ class TestUsersAPI(APITestCase): | |||||||
|     def test_recovery_no_flow(self): |     def test_recovery_no_flow(self): | ||||||
|         """Test user recovery link (no recovery flow set)""" |         """Test user recovery link (no recovery flow set)""" | ||||||
|         self.client.force_login(self.admin) |         self.client.force_login(self.admin) | ||||||
|         response = self.client.post( |         response = self.client.get( | ||||||
|             reverse("authentik_api:user-recovery", kwargs={"pk": self.user.pk}) |             reverse("authentik_api:user-recovery", kwargs={"pk": self.user.pk}) | ||||||
|         ) |         ) | ||||||
|         self.assertEqual(response.status_code, 400) |         self.assertEqual(response.status_code, 404) | ||||||
|         self.assertJSONEqual(response.content, {"non_field_errors": "No recovery flow set."}) |  | ||||||
|  |  | ||||||
|     def test_set_password(self): |     def test_set_password(self): | ||||||
|         """Test Direct password set""" |         """Test Direct password set""" | ||||||
| @ -91,7 +84,7 @@ class TestUsersAPI(APITestCase): | |||||||
|         brand.flow_recovery = flow |         brand.flow_recovery = flow | ||||||
|         brand.save() |         brand.save() | ||||||
|         self.client.force_login(self.admin) |         self.client.force_login(self.admin) | ||||||
|         response = self.client.post( |         response = self.client.get( | ||||||
|             reverse("authentik_api:user-recovery", kwargs={"pk": self.user.pk}) |             reverse("authentik_api:user-recovery", kwargs={"pk": self.user.pk}) | ||||||
|         ) |         ) | ||||||
|         self.assertEqual(response.status_code, 200) |         self.assertEqual(response.status_code, 200) | ||||||
| @ -99,20 +92,16 @@ class TestUsersAPI(APITestCase): | |||||||
|     def test_recovery_email_no_flow(self): |     def test_recovery_email_no_flow(self): | ||||||
|         """Test user recovery link (no recovery flow set)""" |         """Test user recovery link (no recovery flow set)""" | ||||||
|         self.client.force_login(self.admin) |         self.client.force_login(self.admin) | ||||||
|         response = self.client.post( |         response = self.client.get( | ||||||
|             reverse("authentik_api:user-recovery-email", kwargs={"pk": self.user.pk}) |             reverse("authentik_api:user-recovery-email", kwargs={"pk": self.user.pk}) | ||||||
|         ) |         ) | ||||||
|         self.assertEqual(response.status_code, 400) |         self.assertEqual(response.status_code, 404) | ||||||
|         self.assertJSONEqual( |  | ||||||
|             response.content, {"non_field_errors": "User does not have an email address set."} |  | ||||||
|         ) |  | ||||||
|         self.user.email = "foo@bar.baz" |         self.user.email = "foo@bar.baz" | ||||||
|         self.user.save() |         self.user.save() | ||||||
|         response = self.client.post( |         response = self.client.get( | ||||||
|             reverse("authentik_api:user-recovery-email", kwargs={"pk": self.user.pk}) |             reverse("authentik_api:user-recovery-email", kwargs={"pk": self.user.pk}) | ||||||
|         ) |         ) | ||||||
|         self.assertEqual(response.status_code, 400) |         self.assertEqual(response.status_code, 404) | ||||||
|         self.assertJSONEqual(response.content, {"non_field_errors": "No recovery flow set."}) |  | ||||||
|  |  | ||||||
|     def test_recovery_email_no_stage(self): |     def test_recovery_email_no_stage(self): | ||||||
|         """Test user recovery link (no email stage)""" |         """Test user recovery link (no email stage)""" | ||||||
| @ -123,11 +112,10 @@ class TestUsersAPI(APITestCase): | |||||||
|         brand.flow_recovery = flow |         brand.flow_recovery = flow | ||||||
|         brand.save() |         brand.save() | ||||||
|         self.client.force_login(self.admin) |         self.client.force_login(self.admin) | ||||||
|         response = self.client.post( |         response = self.client.get( | ||||||
|             reverse("authentik_api:user-recovery-email", kwargs={"pk": self.user.pk}) |             reverse("authentik_api:user-recovery-email", kwargs={"pk": self.user.pk}) | ||||||
|         ) |         ) | ||||||
|         self.assertEqual(response.status_code, 400) |         self.assertEqual(response.status_code, 404) | ||||||
|         self.assertJSONEqual(response.content, {"non_field_errors": "Email stage does not exist."}) |  | ||||||
|  |  | ||||||
|     def test_recovery_email(self): |     def test_recovery_email(self): | ||||||
|         """Test user recovery link""" |         """Test user recovery link""" | ||||||
| @ -141,7 +129,7 @@ class TestUsersAPI(APITestCase): | |||||||
|         stage = EmailStage.objects.create(name="email") |         stage = EmailStage.objects.create(name="email") | ||||||
|  |  | ||||||
|         self.client.force_login(self.admin) |         self.client.force_login(self.admin) | ||||||
|         response = self.client.post( |         response = self.client.get( | ||||||
|             reverse( |             reverse( | ||||||
|                 "authentik_api:user-recovery-email", |                 "authentik_api:user-recovery-email", | ||||||
|                 kwargs={"pk": self.user.pk}, |                 kwargs={"pk": self.user.pk}, | ||||||
|  | |||||||
| @ -8,6 +8,7 @@ from rest_framework.test import APITestCase | |||||||
|  |  | ||||||
| from authentik.core.models import User | from authentik.core.models import User | ||||||
| from authentik.core.tests.utils import create_test_admin_user | from authentik.core.tests.utils import create_test_admin_user | ||||||
|  | from authentik.lib.config import CONFIG | ||||||
| from authentik.tenants.utils import get_current_tenant | from authentik.tenants.utils import get_current_tenant | ||||||
|  |  | ||||||
|  |  | ||||||
| @ -24,6 +25,7 @@ class TestUsersAvatars(APITestCase): | |||||||
|         tenant.avatars = mode |         tenant.avatars = mode | ||||||
|         tenant.save() |         tenant.save() | ||||||
|  |  | ||||||
|  |     @CONFIG.patch("avatars", "none") | ||||||
|     def test_avatars_none(self): |     def test_avatars_none(self): | ||||||
|         """Test avatars none""" |         """Test avatars none""" | ||||||
|         self.set_avatar_mode("none") |         self.set_avatar_mode("none") | ||||||
| @ -42,8 +44,8 @@ class TestUsersAvatars(APITestCase): | |||||||
|         with Mocker() as mocker: |         with Mocker() as mocker: | ||||||
|             mocker.head( |             mocker.head( | ||||||
|                 ( |                 ( | ||||||
|                     "https://www.gravatar.com/avatar/76eb3c74c8beb6faa037f1b6e2ecb3e252bdac" |                     "https://secure.gravatar.com/avatar/84730f9c1851d1ea03f1a" | ||||||
|                     "6cf71fb567ae36025a9d4ea86b?size=158&rating=g&default=404" |                     "a9ed85bd1ea?size=158&rating=g&default=404" | ||||||
|                 ), |                 ), | ||||||
|                 text="foo", |                 text="foo", | ||||||
|             ) |             ) | ||||||
|  | |||||||
| @ -1,10 +1,12 @@ | |||||||
| """Test Utils""" | """Test Utils""" | ||||||
|  |  | ||||||
|  | from typing import Optional | ||||||
|  |  | ||||||
| from django.utils.text import slugify | from django.utils.text import slugify | ||||||
|  |  | ||||||
| from authentik.brands.models import Brand | from authentik.brands.models import Brand | ||||||
| from authentik.core.models import Group, User | from authentik.core.models import Group, User | ||||||
| from authentik.crypto.builder import CertificateBuilder, PrivateKeyAlg | from authentik.crypto.builder import CertificateBuilder | ||||||
| from authentik.crypto.models import CertificateKeyPair | from authentik.crypto.models import CertificateKeyPair | ||||||
| from authentik.flows.models import Flow, FlowDesignation | from authentik.flows.models import Flow, FlowDesignation | ||||||
| from authentik.lib.generators import generate_id | from authentik.lib.generators import generate_id | ||||||
| @ -20,7 +22,7 @@ def create_test_flow( | |||||||
|     ) |     ) | ||||||
|  |  | ||||||
|  |  | ||||||
| def create_test_user(name: str | None = None, **kwargs) -> User: | def create_test_user(name: Optional[str] = None, **kwargs) -> User: | ||||||
|     """Generate a test user""" |     """Generate a test user""" | ||||||
|     uid = generate_id(20) if not name else name |     uid = generate_id(20) if not name else name | ||||||
|     kwargs.setdefault("email", f"{uid}@goauthentik.io") |     kwargs.setdefault("email", f"{uid}@goauthentik.io") | ||||||
| @ -34,7 +36,7 @@ def create_test_user(name: str | None = None, **kwargs) -> User: | |||||||
|     return user |     return user | ||||||
|  |  | ||||||
|  |  | ||||||
| def create_test_admin_user(name: str | None = None, **kwargs) -> User: | def create_test_admin_user(name: Optional[str] = None, **kwargs) -> User: | ||||||
|     """Generate a test-admin user""" |     """Generate a test-admin user""" | ||||||
|     user = create_test_user(name, **kwargs) |     user = create_test_user(name, **kwargs) | ||||||
|     group = Group.objects.create(name=user.name or name, is_superuser=True) |     group = Group.objects.create(name=user.name or name, is_superuser=True) | ||||||
| @ -50,10 +52,12 @@ def create_test_brand(**kwargs) -> Brand: | |||||||
|     return Brand.objects.create(domain=uid, default=True, **kwargs) |     return Brand.objects.create(domain=uid, default=True, **kwargs) | ||||||
|  |  | ||||||
|  |  | ||||||
| def create_test_cert(alg=PrivateKeyAlg.RSA) -> CertificateKeyPair: | def create_test_cert(use_ec_private_key=False) -> CertificateKeyPair: | ||||||
|     """Generate a certificate for testing""" |     """Generate a certificate for testing""" | ||||||
|     builder = CertificateBuilder(f"{generate_id()}.self-signed.goauthentik.io") |     builder = CertificateBuilder( | ||||||
|     builder.alg = alg |         name=f"{generate_id()}.self-signed.goauthentik.io", | ||||||
|  |         use_ec_private_key=use_ec_private_key, | ||||||
|  |     ) | ||||||
|     builder.build( |     builder.build( | ||||||
|         subject_alt_names=[f"{generate_id()}.self-signed.goauthentik.io"], |         subject_alt_names=[f"{generate_id()}.self-signed.goauthentik.io"], | ||||||
|         validity_days=360, |         validity_days=360, | ||||||
|  | |||||||
| @ -1,6 +1,7 @@ | |||||||
| """authentik core dataclasses""" | """authentik core dataclasses""" | ||||||
|  |  | ||||||
| from dataclasses import dataclass | from dataclasses import dataclass | ||||||
|  | from typing import Optional | ||||||
|  |  | ||||||
| from rest_framework.fields import CharField | from rest_framework.fields import CharField | ||||||
|  |  | ||||||
| @ -19,7 +20,7 @@ class UILoginButton: | |||||||
|     challenge: Challenge |     challenge: Challenge | ||||||
|  |  | ||||||
|     # Icon URL, used as-is |     # Icon URL, used as-is | ||||||
|     icon_url: str | None = None |     icon_url: Optional[str] = None | ||||||
|  |  | ||||||
|  |  | ||||||
| class UserSettingSerializer(PassiveSerializer): | class UserSettingSerializer(PassiveSerializer): | ||||||
|  | |||||||
| @ -12,7 +12,7 @@ from authentik.core.api.applications import ApplicationViewSet | |||||||
| from authentik.core.api.authenticated_sessions import AuthenticatedSessionViewSet | from authentik.core.api.authenticated_sessions import AuthenticatedSessionViewSet | ||||||
| from authentik.core.api.devices import AdminDeviceViewSet, DeviceViewSet | from authentik.core.api.devices import AdminDeviceViewSet, DeviceViewSet | ||||||
| from authentik.core.api.groups import GroupViewSet | from authentik.core.api.groups import GroupViewSet | ||||||
| from authentik.core.api.property_mappings import PropertyMappingViewSet | from authentik.core.api.propertymappings import PropertyMappingViewSet | ||||||
| from authentik.core.api.providers import ProviderViewSet | from authentik.core.api.providers import ProviderViewSet | ||||||
| from authentik.core.api.sources import SourceViewSet, UserSourceConnectionViewSet | from authentik.core.api.sources import SourceViewSet, UserSourceConnectionViewSet | ||||||
| from authentik.core.api.tokens import TokenViewSet | from authentik.core.api.tokens import TokenViewSet | ||||||
|  | |||||||
| @ -57,7 +57,7 @@ class RedirectToAppLaunch(View): | |||||||
|                 }, |                 }, | ||||||
|             ) |             ) | ||||||
|         except FlowNonApplicableException: |         except FlowNonApplicableException: | ||||||
|             raise Http404 from None |             raise Http404 | ||||||
|         plan.insert_stage(in_memory_stage(RedirectToAppStage)) |         plan.insert_stage(in_memory_stage(RedirectToAppStage)) | ||||||
|         request.session[SESSION_KEY_PLAN] = plan |         request.session[SESSION_KEY_PLAN] = plan | ||||||
|         return redirect_with_qs("authentik_core:if-flow", request.GET, flow_slug=flow.slug) |         return redirect_with_qs("authentik_core:if-flow", request.GET, flow_slug=flow.slug) | ||||||
|  | |||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user
	