authentik/passbook/providers/app_gw/views.py

"""passbook app_gw views"""
import string
from random import SystemRandom

from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import HttpRequest, HttpResponse
from django.shortcuts import get_object_or_404, render
from django.views import View
from structlog import get_logger

from passbook import __version__
from passbook.providers.app_gw.models import ApplicationGatewayProvider

ORIGINAL_URL = "HTTP_X_ORIGINAL_URL"
LOGGER = get_logger()


def get_cookie_secret():
    """Generate random 50-character string for cookie-secret"""
    return "".join(
        SystemRandom().choice(string.ascii_uppercase + string.digits) for _ in range(32)
    )


class K8sManifestView(LoginRequiredMixin, View):
    """Generate K8s Deployment and SVC for gatekeeper"""

    def get(self, request: HttpRequest, provider: int) -> HttpResponse:
        """Render deployment template"""
        provider = get_object_or_404(ApplicationGatewayProvider, pk=provider)
        return render(
            request,
            "app_gw/k8s-manifest.yaml",
            {
                "provider": provider,
                "cookie_secret": get_cookie_secret(),
                "version": __version__,
            },
            content_type="text/yaml",
        )