95 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			95 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| """saml sp models"""
 | |
| from django.db import models
 | |
| from django.urls import reverse_lazy
 | |
| from django.utils.translation import gettext_lazy as _
 | |
| 
 | |
| from passbook.core.models import Source
 | |
| from passbook.core.types import UILoginButton
 | |
| from passbook.crypto.models import CertificateKeyPair
 | |
| from passbook.providers.saml.utils.time import timedelta_string_validator
 | |
| 
 | |
| 
 | |
| class SAMLBindingTypes(models.TextChoices):
 | |
|     """SAML Binding types"""
 | |
| 
 | |
|     Redirect = "REDIRECT"
 | |
|     POST = "POST"
 | |
| 
 | |
| 
 | |
| class SAMLSource(Source):
 | |
|     """Authenticate using an external SAML Identity Provider."""
 | |
| 
 | |
|     issuer = models.TextField(
 | |
|         blank=True,
 | |
|         default=None,
 | |
|         verbose_name=_("Issuer"),
 | |
|         help_text=_("Also known as Entity ID. Defaults the Metadata URL."),
 | |
|     )
 | |
| 
 | |
|     sso_url = models.URLField(
 | |
|         verbose_name=_("SSO URL"),
 | |
|         help_text=_("URL that the initial Login request is sent to."),
 | |
|     )
 | |
|     binding_type = models.CharField(
 | |
|         max_length=100,
 | |
|         choices=SAMLBindingTypes.choices,
 | |
|         default=SAMLBindingTypes.Redirect,
 | |
|     )
 | |
| 
 | |
|     slo_url = models.URLField(
 | |
|         default=None,
 | |
|         blank=True,
 | |
|         null=True,
 | |
|         verbose_name=_("SLO URL"),
 | |
|         help_text=_("Optional URL if your IDP supports Single-Logout."),
 | |
|     )
 | |
| 
 | |
|     temporary_user_delete_after = models.TextField(
 | |
|         default="days=1",
 | |
|         verbose_name=_("Delete temporary users after"),
 | |
|         validators=[timedelta_string_validator],
 | |
|         help_text=_(
 | |
|             (
 | |
|                 "Time offset when temporary users should be deleted. This only applies if your IDP "
 | |
|                 "uses the NameID Format 'transient', and the user doesn't log out manually. "
 | |
|                 "(Format: hours=1;minutes=2;seconds=3)."
 | |
|             )
 | |
|         ),
 | |
|     )
 | |
| 
 | |
|     signing_kp = models.ForeignKey(
 | |
|         CertificateKeyPair,
 | |
|         verbose_name=_("Singing Keypair"),
 | |
|         help_text=_(
 | |
|             "Certificate Key Pair of the IdP which Assertion's Signature is validated against."
 | |
|         ),
 | |
|         on_delete=models.PROTECT,
 | |
|     )
 | |
| 
 | |
|     form = "passbook.sources.saml.forms.SAMLSourceForm"
 | |
| 
 | |
|     @property
 | |
|     def ui_login_button(self) -> UILoginButton:
 | |
|         return UILoginButton(
 | |
|             name=self.name,
 | |
|             url=reverse_lazy(
 | |
|                 "passbook_sources_saml:login", kwargs={"source_slug": self.slug}
 | |
|             ),
 | |
|             icon_path="",
 | |
|         )
 | |
| 
 | |
|     @property
 | |
|     def ui_additional_info(self) -> str:
 | |
|         metadata_url = reverse_lazy(
 | |
|             "passbook_sources_saml:metadata", kwargs={"source_slug": self.slug}
 | |
|         )
 | |
|         return f'<a href="{metadata_url}" class="btn btn-default btn-sm">Metadata Download</a>'
 | |
| 
 | |
|     def __str__(self):
 | |
|         return f"SAML Source {self.name}"
 | |
| 
 | |
|     class Meta:
 | |
| 
 | |
|         verbose_name = _("SAML Source")
 | |
|         verbose_name_plural = _("SAML Sources")
 | 
