authentik/passbook/providers/app_gw/models.py

"""passbook app_gw models"""
import string
from random import SystemRandom
from typing import Optional, Type

from django.db import models
from django.forms import ModelForm
from django.http import HttpRequest
from django.utils.translation import gettext as _
from oidc_provider.models import Client

from passbook import __version__
from passbook.core.models import Provider
from passbook.lib.utils.template import render_to_string


class ApplicationGatewayProvider(Provider):
    """Protect applications that don't support any of the other
    Protocols by using a Reverse-Proxy."""

    name = models.TextField()
    internal_host = models.TextField()
    external_host = models.TextField()

    client = models.ForeignKey(Client, on_delete=models.CASCADE)

    def form(self) -> Type[ModelForm]:
        from passbook.providers.app_gw.forms import ApplicationGatewayProviderForm

        return ApplicationGatewayProviderForm

    def html_setup_urls(self, request: HttpRequest) -> Optional[str]:
        """return template and context modal with URLs for authorize, token, openid-config, etc"""
        cookie_secret = "".join(
            SystemRandom().choice(string.ascii_uppercase + string.digits)
            for _ in range(50)
        )
        return render_to_string(
            "app_gw/setup_modal.html",
            {"provider": self, "cookie_secret": cookie_secret, "version": __version__},
        )

    def __str__(self):
        return self.name

    class Meta:

        verbose_name = _("Application Gateway Provider")
        verbose_name_plural = _("Application Gateway Providers")