244 lines
		
	
	
		
			8.4 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			244 lines
		
	
	
		
			8.4 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
import re
 | 
						|
from functools import partial
 | 
						|
from typing import ClassVar, Dict
 | 
						|
 | 
						|
from django import forms
 | 
						|
from django.db import models
 | 
						|
from django.shortcuts import get_object_or_404, redirect
 | 
						|
from django.utils.decorators import method_decorator
 | 
						|
from django.views.generic import FormView, RedirectView, TemplateView
 | 
						|
 | 
						|
from core.models import Config
 | 
						|
from users.decorators import admin_required
 | 
						|
from users.models import Domain
 | 
						|
 | 
						|
 | 
						|
@method_decorator(admin_required, name="dispatch")
 | 
						|
class SystemSettingsRoot(RedirectView):
 | 
						|
    url = "/settings/system/basic/"
 | 
						|
 | 
						|
 | 
						|
@method_decorator(admin_required, name="dispatch")
 | 
						|
class SystemSettingsPage(FormView):
 | 
						|
    """
 | 
						|
    Shows a settings page dynamically created from our settings layout
 | 
						|
    at the bottom of the page. Don't add this to a URL directly - subclass!
 | 
						|
    """
 | 
						|
 | 
						|
    template_name = "settings/settings_system.html"
 | 
						|
    options_class = Config.SystemOptions
 | 
						|
    section: ClassVar[str]
 | 
						|
    options: Dict[str, Dict[str, str]]
 | 
						|
 | 
						|
    extra_context = {"top_section": "settings_system"}
 | 
						|
 | 
						|
    def get_form_class(self):
 | 
						|
        # Create the fields dict from the config object
 | 
						|
        fields = {}
 | 
						|
        for key, details in self.options.items():
 | 
						|
            config_field = self.options_class.__fields__[key]
 | 
						|
            if config_field.type_ is bool:
 | 
						|
                form_field = partial(
 | 
						|
                    forms.BooleanField,
 | 
						|
                    widget=forms.Select(
 | 
						|
                        choices=[(True, "Enabled"), (False, "Disabled")]
 | 
						|
                    ),
 | 
						|
                )
 | 
						|
            elif config_field.type_ is str:
 | 
						|
                form_field = forms.CharField
 | 
						|
            else:
 | 
						|
                raise ValueError(f"Cannot render settings type {config_field.type_}")
 | 
						|
            fields[key] = form_field(
 | 
						|
                label=details["title"],
 | 
						|
                help_text=details.get("help_text", ""),
 | 
						|
                required=details.get("required", False),
 | 
						|
            )
 | 
						|
        # Create a form class dynamically (yeah, right?) and return that
 | 
						|
        return type("SettingsForm", (forms.Form,), fields)
 | 
						|
 | 
						|
    def load_config(self):
 | 
						|
        return Config.load_system()
 | 
						|
 | 
						|
    def save_config(self, key, value):
 | 
						|
        Config.set_system(key, value)
 | 
						|
 | 
						|
    def get_initial(self):
 | 
						|
        config = self.load_config()
 | 
						|
        initial = {}
 | 
						|
        for key in self.options.keys():
 | 
						|
            initial[key] = getattr(config, key)
 | 
						|
        return initial
 | 
						|
 | 
						|
    def get_context_data(self):
 | 
						|
        context = super().get_context_data()
 | 
						|
        context["section"] = self.section
 | 
						|
        return context
 | 
						|
 | 
						|
    def form_valid(self, form):
 | 
						|
        # Save each key
 | 
						|
        for field in form:
 | 
						|
            self.save_config(
 | 
						|
                field.name,
 | 
						|
                form.cleaned_data[field.name],
 | 
						|
            )
 | 
						|
        return redirect(".")
 | 
						|
 | 
						|
 | 
						|
class BasicPage(SystemSettingsPage):
 | 
						|
 | 
						|
    section = "basic"
 | 
						|
 | 
						|
    options = {
 | 
						|
        "site_name": {
 | 
						|
            "title": "Site Name",
 | 
						|
            "help_text": "Shown in the top-left of the page, and titles",
 | 
						|
        },
 | 
						|
        "highlight_color": {
 | 
						|
            "title": "Highlight Color",
 | 
						|
            "help_text": "Used for logo background and other highlights",
 | 
						|
        },
 | 
						|
    }
 | 
						|
 | 
						|
 | 
						|
@method_decorator(admin_required, name="dispatch")
 | 
						|
class DomainsPage(TemplateView):
 | 
						|
 | 
						|
    template_name = "settings/settings_system_domains.html"
 | 
						|
 | 
						|
    def get_context_data(self):
 | 
						|
        return {
 | 
						|
            "domains": Domain.objects.filter(local=True).order_by("domain"),
 | 
						|
            "section": "domains",
 | 
						|
        }
 | 
						|
 | 
						|
 | 
						|
@method_decorator(admin_required, name="dispatch")
 | 
						|
class DomainCreatePage(FormView):
 | 
						|
 | 
						|
    template_name = "settings/settings_system_domain_create.html"
 | 
						|
    extra_context = {"section": "domains"}
 | 
						|
 | 
						|
    class form_class(forms.Form):
 | 
						|
        domain = forms.CharField(
 | 
						|
            help_text="The domain displayed as part of a user's identity.\nCannot be changed after the domain has been created.",
 | 
						|
        )
 | 
						|
        service_domain = forms.CharField(
 | 
						|
            help_text="Optional - a domain that serves Takahē if it is not running on the main domain.\nCannot be changed after the domain has been created.",
 | 
						|
            required=False,
 | 
						|
        )
 | 
						|
        public = forms.BooleanField(
 | 
						|
            help_text="If any user on this server can create identities here",
 | 
						|
            widget=forms.Select(choices=[(True, "Public"), (False, "Private")]),
 | 
						|
            required=False,
 | 
						|
        )
 | 
						|
 | 
						|
        domain_regex = re.compile(
 | 
						|
            r"^((?!-))(xn--)?[a-z0-9][a-z0-9-_]{0,61}[a-z0-9]{0,1}\.(xn--)?([a-z0-9\-]{1,61}|[a-z0-9-]{1,30}\.[a-z]{2,})$"
 | 
						|
        )
 | 
						|
 | 
						|
        def clean_domain(self):
 | 
						|
            if not self.domain_regex.match(self.cleaned_data["domain"]):
 | 
						|
                raise forms.ValidationError("This does not look like a domain name")
 | 
						|
            if Domain.objects.filter(
 | 
						|
                models.Q(domain=self.cleaned_data["domain"])
 | 
						|
                | models.Q(service_domain=self.cleaned_data["domain"])
 | 
						|
            ):
 | 
						|
                raise forms.ValidationError("This domain name is already in use")
 | 
						|
            return self.cleaned_data["domain"]
 | 
						|
 | 
						|
        def clean_service_domain(self):
 | 
						|
            if not self.cleaned_data["service_domain"]:
 | 
						|
                return None
 | 
						|
            if not self.domain_regex.match(self.cleaned_data["service_domain"]):
 | 
						|
                raise forms.ValidationError("This does not look like a domain name")
 | 
						|
            if Domain.objects.filter(
 | 
						|
                models.Q(domain=self.cleaned_data["service_domain"])
 | 
						|
                | models.Q(service_domain=self.cleaned_data["service_domain"])
 | 
						|
            ):
 | 
						|
                raise forms.ValidationError("This domain name is already in use")
 | 
						|
            if self.cleaned_data.get("domain") == self.cleaned_data["service_domain"]:
 | 
						|
                raise forms.ValidationError(
 | 
						|
                    "You cannot have the domain and service domain be the same (did you mean to leave service domain blank?)"
 | 
						|
                )
 | 
						|
            return self.cleaned_data["service_domain"]
 | 
						|
 | 
						|
    def form_valid(self, form):
 | 
						|
        Domain.objects.create(
 | 
						|
            domain=form.cleaned_data["domain"],
 | 
						|
            service_domain=form.cleaned_data["service_domain"] or None,
 | 
						|
            public=form.cleaned_data["public"],
 | 
						|
            local=True,
 | 
						|
        )
 | 
						|
        return redirect(Domain.urls.root)
 | 
						|
 | 
						|
 | 
						|
@method_decorator(admin_required, name="dispatch")
 | 
						|
class DomainEditPage(FormView):
 | 
						|
 | 
						|
    template_name = "settings/settings_system_domain_edit.html"
 | 
						|
    extra_context = {"section": "domains"}
 | 
						|
 | 
						|
    class form_class(forms.Form):
 | 
						|
        domain = forms.CharField(
 | 
						|
            help_text="The domain displayed as part of a user's identity.\nCannot be changed after the domain has been created.",
 | 
						|
            disabled=True,
 | 
						|
        )
 | 
						|
        service_domain = forms.CharField(
 | 
						|
            help_text="Optional - a domain that serves Takahē if it is not running on the main domain.\nCannot be changed after the domain has been created.",
 | 
						|
            disabled=True,
 | 
						|
            required=False,
 | 
						|
        )
 | 
						|
        public = forms.BooleanField(
 | 
						|
            help_text="If any user on this server can create identities here",
 | 
						|
            widget=forms.Select(choices=[(True, "Public"), (False, "Private")]),
 | 
						|
            required=False,
 | 
						|
        )
 | 
						|
 | 
						|
    def dispatch(self, request, domain):
 | 
						|
        self.domain = get_object_or_404(
 | 
						|
            Domain.objects.filter(local=True), domain=domain
 | 
						|
        )
 | 
						|
        return super().dispatch(request)
 | 
						|
 | 
						|
    def get_context_data(self):
 | 
						|
        context = super().get_context_data()
 | 
						|
        context["domain"] = self.domain
 | 
						|
        return context
 | 
						|
 | 
						|
    def form_valid(self, form):
 | 
						|
        self.domain.public = form.cleaned_data["public"]
 | 
						|
        self.domain.save()
 | 
						|
        return redirect(Domain.urls.root)
 | 
						|
 | 
						|
    def get_initial(self):
 | 
						|
        return {
 | 
						|
            "domain": self.domain.domain,
 | 
						|
            "service_domain": self.domain.service_domain,
 | 
						|
            "public": self.domain.public,
 | 
						|
        }
 | 
						|
 | 
						|
 | 
						|
@method_decorator(admin_required, name="dispatch")
 | 
						|
class DomainDeletePage(TemplateView):
 | 
						|
 | 
						|
    template_name = "settings/settings_system_domain_delete.html"
 | 
						|
 | 
						|
    def dispatch(self, request, domain):
 | 
						|
        self.domain = get_object_or_404(
 | 
						|
            Domain.objects.filter(public=True), domain=domain
 | 
						|
        )
 | 
						|
        return super().dispatch(request)
 | 
						|
 | 
						|
    def get_context_data(self):
 | 
						|
        return {
 | 
						|
            "domain": self.domain,
 | 
						|
            "num_identities": self.domain.identities.count(),
 | 
						|
            "section": "domains",
 | 
						|
        }
 | 
						|
 | 
						|
    def post(self, request):
 | 
						|
        if self.domain.identities.exists():
 | 
						|
            raise ValueError("Tried to delete domain with identities!")
 | 
						|
        self.domain.delete()
 | 
						|
        return redirect("/settings/system/domains/")
 |