From a7a292a84cff5103e96b4b68ab2761438040827d Mon Sep 17 00:00:00 2001
From: Andrew Godwin <andrew@aeracode.org>
Date: Tue, 3 Jan 2023 01:32:00 -0700
Subject: [PATCH] Pagination styling and "max users" signup limit

---
 core/models/config.py           |  1 +
 static/css/style.css            |  8 ++++----
 templates/activities/tag.html   |  4 ++--
 templates/admin/federation.html |  5 ++++-
 templates/admin/hashtags.html   |  5 ++++-
 templates/admin/identities.html |  5 ++++-
 templates/admin/invites.html    |  5 ++++-
 templates/admin/reports.html    |  5 ++++-
 templates/admin/users.html      |  5 ++++-
 templates/identity/follows.html |  2 +-
 templates/identity/view.html    |  4 +++-
 users/views/admin/settings.py   |  5 +++++
 users/views/auth.py             | 12 ++++++++++--
 13 files changed, 50 insertions(+), 16 deletions(-)

diff --git a/core/models/config.py b/core/models/config.py
index 7dda07b..eb07313 100644
--- a/core/models/config.py
+++ b/core/models/config.py
@@ -213,6 +213,7 @@ class Config(models.Model):
 
         signup_allowed: bool = True
         signup_text: str = ""
+        signup_max_users: int = 0
         content_warning_text: str = "Content Warning"
 
         post_length: int = 500
diff --git a/static/css/style.css b/static/css/style.css
index 62b87c9..4077d3b 100644
--- a/static/css/style.css
+++ b/static/css/style.css
@@ -1538,15 +1538,15 @@ form .post {
 .pagination {
     display: flex;
     justify-content: center;
+    align-items: center;
     gap: 1em;
 }
 
-.load-more {
-    margin: 10px 0;
-    text-align: center;
+.pagination .count {
+    display: inline-block;
+    color: var(--color-text-dull);
 }
 
-
 @media (max-width: 920px),
 (display-mode: standalone) {
     div.columns {
diff --git a/templates/activities/tag.html b/templates/activities/tag.html
index 1a2b5d7..2bb7406 100644
--- a/templates/activities/tag.html
+++ b/templates/activities/tag.html
@@ -12,11 +12,11 @@
 
     <div class="pagination">
         {% if page_obj.has_previous %}
-            <div class="load-more"><a class="button" href=".?page={{ page_obj.previous_page_number }}">Previous Page</a></div>
+            <a class="button" href=".?page={{ page_obj.previous_page_number }}">Previous Page</a>
         {% endif %}
 
         {% if page_obj.has_next %}
-            <div class="load-more"><a class="button" href=".?page={{ page_obj.next_page_number }} {% if config_identity.infinite_scroll %}hx-trigger="revealed"{% endif %}">Next Page</a></div>
+            <a class="button" href=".?page={{ page_obj.next_page_number }} {% if config_identity.infinite_scroll %}hx-trigger="revealed"{% endif %}">Next Page</a>
         {% endif %}
     </div>
 {% endblock %}
diff --git a/templates/admin/federation.html b/templates/admin/federation.html
index 98ee09d..0f20ada 100644
--- a/templates/admin/federation.html
+++ b/templates/admin/federation.html
@@ -24,10 +24,13 @@
         {% empty %}
             <p class="option empty">There are no federation links yet.</p>
         {% endfor %}
-        <div class="load-more">
+        <div class="pagination">
             {% if page_obj.has_previous %}
                 <a class="button" href=".?page={{ page_obj.previous_page_number }}">Previous Page</a>
             {% endif %}
+            {% if page_obj.paginator.count %}
+                <span class="count">{{ page_obj.paginator.count }} domain{{page_obj.paginator.count|pluralize }}</span>
+            {% endif %}
             {% if page_obj.has_next %}
                 <a class="button" href=".?page={{ page_obj.next_page_number }}">Next Page</a>
             {% endif %}
diff --git a/templates/admin/hashtags.html b/templates/admin/hashtags.html
index eab3aa0..b2a63b1 100644
--- a/templates/admin/hashtags.html
+++ b/templates/admin/hashtags.html
@@ -37,10 +37,13 @@
         {% empty %}
             <p class="option empty">There are no hashtags yet.</p>
         {% endfor %}
-        <div class="load-more">
+        <div class="pagination">
             {% if page_obj.has_previous %}
                 <a class="button" href=".?page={{ page_obj.previous_page_number }}">Previous Page</a>
             {% endif %}
+            {% if page_obj.paginator.count %}
+                <span class="count">{{ page_obj.paginator.count }} hashtag{{page_obj.paginator.count|pluralize }}</span>
+            {% endif %}
             {% if page_obj.has_next %}
                 <a class="button" href=".?page={{ page_obj.next_page_number }}">Next Page</a>
             {% endif %}
diff --git a/templates/admin/identities.html b/templates/admin/identities.html
index 7be7fec..ae97a6d 100644
--- a/templates/admin/identities.html
+++ b/templates/admin/identities.html
@@ -38,10 +38,13 @@
                 {% endif %}
             </p>
         {% endfor %}
-        <div class="load-more">
+        <div class="pagination">
             {% if page_obj.has_previous %}
                 <a class="button" href=".?page={{ page_obj.previous_page_number }}{% if local_only %}&amp;local_only=true{% endif %}{% if query %}&amp;query={{ query }}{% endif %}">Previous Page</a>
             {% endif %}
+            {% if page_obj.paginator.count %}
+                <span class="count">{{ page_obj.paginator.count }} identit{{page_obj.paginator.count|pluralize:"y,ies" }}</span>
+            {% endif %}
             {% if page_obj.has_next %}
                 <a class="button" href=".?page={{ page_obj.next_page_number }}{% if local_only %}&amp;local_only=true{% endif %}{% if query %}&amp;query={{ query }}{% endif %}">Next Page</a>
             {% endif %}
diff --git a/templates/admin/invites.html b/templates/admin/invites.html
index 880fa3e..86d1281 100644
--- a/templates/admin/invites.html
+++ b/templates/admin/invites.html
@@ -39,10 +39,13 @@
                 There are no unused invites.
             </p>
         {% endfor %}
-        <div class="load-more">
+        <div class="pagination">
             {% if page_obj.has_previous %}
                 <a class="button" href=".?page={{ page_obj.previous_page_number }}">Previous Page</a>
             {% endif %}
+            {% if page_obj.paginator.count %}
+                <span class="count">{{ page_obj.paginator.count }} invite{{page_obj.paginator.count|pluralize }}</span>
+            {% endif %}
             {% if page_obj.has_next %}
                 <a class="button" href=".?page={{ page_obj.next_page_number }}">Next Page</a>
             {% endif %}
diff --git a/templates/admin/reports.html b/templates/admin/reports.html
index 1634443..0388123 100644
--- a/templates/admin/reports.html
+++ b/templates/admin/reports.html
@@ -31,10 +31,13 @@
                 There are no {% if all %}reports yet{% else %}unresolved reports{% endif %}.
             </p>
         {% endfor %}
-        <div class="load-more">
+        <div class="pagination">
             {% if page_obj.has_previous %}
                 <a class="button" href=".?page={{ page_obj.previous_page_number }}{% if all %}&amp;all=true{% endif %}">Previous Page</a>
             {% endif %}
+            {% if page_obj.paginator.count %}
+                <span class="count">{{ page_obj.paginator.count }} report{{page_obj.paginator.count|pluralize }}</span>
+            {% endif %}
             {% if page_obj.has_next %}
                 <a class="button" href=".?page={{ page_obj.next_page_number }}{% if all %}&amp;all=true{% endif %}">Next Page</a>
             {% endif %}
diff --git a/templates/admin/users.html b/templates/admin/users.html
index 1bf5b2e..c942a39 100644
--- a/templates/admin/users.html
+++ b/templates/admin/users.html
@@ -30,10 +30,13 @@
                 {% endif %}
             </p>
         {% endfor %}
-        <div class="load-more">
+        <div class="pagination">
             {% if page_obj.has_previous %}
                 <a class="button" href=".?page={{ page_obj.previous_page_number }}">Previous Page</a>
             {% endif %}
+            {% if page_obj.paginator.count %}
+                <span class="count">{{ page_obj.paginator.count }} user{{page_obj.paginator.count|pluralize }}</span>
+            {% endif %}
             {% if page_obj.has_next %}
                 <a class="button" href=".?page={{ page_obj.next_page_number }}">Next Page</a>
             {% endif %}
diff --git a/templates/identity/follows.html b/templates/identity/follows.html
index 36f1094..26744c6 100644
--- a/templates/identity/follows.html
+++ b/templates/identity/follows.html
@@ -12,7 +12,7 @@
         </span>
     {% endfor %}
 
-    <div class="load-more">
+    <div class="pagination">
         {% if page_obj.has_previous %}
             <a class="button" href=".?page={{ page_obj.previous_page_number }}">Previous Page</a>
         {% endif %}
diff --git a/templates/identity/view.html b/templates/identity/view.html
index d449437..13899c9 100644
--- a/templates/identity/view.html
+++ b/templates/identity/view.html
@@ -97,7 +97,9 @@
         {% endfor %}
 
         {% if page_obj.has_next %}
-            <div class="load-more"><a class="button" href=".?page={{ page_obj.next_page_number }}">Next Page</a></div>
+            <div class="pagination">
+                <a class="button" href=".?page={{ page_obj.next_page_number }}">Next Page</a>
+            </div>
         {% endif %}
 
     {% endblock %}
diff --git a/users/views/admin/settings.py b/users/views/admin/settings.py
index 6d6380b..0931e39 100644
--- a/users/views/admin/settings.py
+++ b/users/views/admin/settings.py
@@ -76,6 +76,10 @@ class BasicSettings(AdminSettingsPage):
             "help_text": "Shown above the signup form.\nUse Markdown for formatting.",
             "display": "textarea",
         },
+        "signup_max_users": {
+            "title": "Maximum User Limit",
+            "help_text": "Signups will be auto-disabled if your server grows to this many users.\nUse 0 for unlimited.",
+        },
         "restricted_usernames": {
             "title": "Restricted Usernames",
             "help_text": "Usernames that only admins can register for identities. One per line.",
@@ -109,6 +113,7 @@ class BasicSettings(AdminSettingsPage):
         ],
         "Signups": [
             "signup_allowed",
+            "signup_max_users",
             "signup_text",
         ],
         "Posts": [
diff --git a/users/views/auth.py b/users/views/auth.py
index 1bae751..c76ccf2 100644
--- a/users/views/auth.py
+++ b/users/views/auth.py
@@ -79,17 +79,25 @@ class Signup(FormView):
             return email
 
     def dispatch(self, request, token=None, *args, **kwargs):
+        # See if we have an invite token
         if token:
             self.invite = get_object_or_404(Invite, token=token)
             if not self.invite.valid:
                 raise Http404()
         else:
             self.invite = None
+        # Calculate if we're at or over the user limit
+        self.at_max_users = False
+        if (
+            Config.system.signup_max_users
+            and User.objects.count() >= Config.system.signup_max_users
+        ):
+            self.at_max_users = True
         return super().dispatch(request, *args, **kwargs)
 
     def form_valid(self, form):
         # Don't allow anything if there's no invite and no signup allowed
-        if not Config.system.signup_allowed and not self.invite:
+        if (not Config.system.signup_allowed or self.at_max_users) and not self.invite:
             return self.render_to_response(self.get_context_data())
         # Make the new user
         user = User.objects.create(email=form.cleaned_data["email"])
@@ -113,7 +121,7 @@ class Signup(FormView):
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
-        if not Config.system.signup_allowed and not self.invite:
+        if (not Config.system.signup_allowed or self.at_max_users) and not self.invite:
             del context["form"]
         if Config.system.signup_text:
             context["signup_text"] = mark_safe(