41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
"""passbook Application Security Gateway Forms"""
|
|
from django import forms
|
|
from oauth2_provider.generators import generate_client_id, generate_client_secret
|
|
from oidc_provider.models import Client, ResponseType
|
|
|
|
from passbook.providers.app_gw.models import ApplicationGatewayProvider
|
|
|
|
|
|
class ApplicationGatewayProviderForm(forms.ModelForm):
|
|
"""Security Gateway Provider form"""
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.instance.pk:
|
|
# New instance, so we create a new OIDC client with random keys
|
|
self.instance.client = Client.objects.create(
|
|
client_id=generate_client_id(), client_secret=generate_client_secret()
|
|
)
|
|
self.instance.client.name = self.instance.name
|
|
self.instance.client.response_types.set(
|
|
[ResponseType.objects.get_by_natural_key("code")]
|
|
)
|
|
self.instance.client.redirect_uris = [
|
|
f"http://{self.instance.external_host}/oauth2/callback",
|
|
f"https://{self.instance.external_host}/oauth2/callback",
|
|
f"http://{self.instance.internal_host}/oauth2/callback",
|
|
f"https://{self.instance.internal_host}/oauth2/callback",
|
|
]
|
|
self.instance.client.scope = ["openid", "email"]
|
|
self.instance.client.save()
|
|
return super().save(*args, **kwargs)
|
|
|
|
class Meta:
|
|
|
|
model = ApplicationGatewayProvider
|
|
fields = ["name", "authorization_flow", "internal_host", "external_host"]
|
|
widgets = {
|
|
"name": forms.TextInput(),
|
|
"internal_host": forms.TextInput(),
|
|
"external_host": forms.TextInput(),
|
|
}
|