authentik/passbook/providers/oauth/models.py

"""Oauth2 provider product extension"""

from typing import Optional, Type

from django.forms import ModelForm
from django.http import HttpRequest
from django.shortcuts import reverse
from django.utils.translation import gettext as _
from oauth2_provider.models import AbstractApplication

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


class OAuth2Provider(Provider, AbstractApplication):
    """Generic OAuth2 Provider for applications not using OpenID-Connect.
    This Provider also supports the GitHub-pretend mode for Applications that don't support
    generic OAuth."""

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

        return OAuth2ProviderForm

    def __str__(self):
        return self.name

    def html_setup_urls(self, request: HttpRequest) -> Optional[str]:
        """return template and context modal with URLs for authorize, token, openid-config, etc"""
        return render_to_string(
            "providers/oauth/setup_url_modal.html",
            {
                "provider": self,
                "authorize_url": request.build_absolute_uri(
                    reverse("passbook_providers_oauth:oauth2-authorize")
                ),
                "token_url": request.build_absolute_uri(
                    reverse("passbook_providers_oauth:token")
                ),
                "userinfo_url": request.build_absolute_uri(
                    reverse("passbook_api:openid")
                ),
            },
        )

    class Meta:

        verbose_name = _("OAuth2 Provider")
        verbose_name_plural = _("OAuth2 Providers")