From bdcb17e3c165024e36281fc8bfaa600f9fa614bf Mon Sep 17 00:00:00 2001
From: rimu <3310831+rimu@users.noreply.github.com>
Date: Wed, 10 Jan 2024 09:34:58 +1300
Subject: [PATCH] notification toggles

---
 app/post/routes.py                            | 22 ++++++++++++++++++-
 .../community/_notification_toggle.html       |  2 +-
 app/templates/post/_post_full.html            | 12 ++++++++--
 .../post/_post_notification_toggle.html       |  4 ++++
 .../post/_reply_notification_toggle.html      |  4 ++++
 app/templates/post/post.html                  | 15 ++++++++-----
 6 files changed, 50 insertions(+), 9 deletions(-)
 create mode 100644 app/templates/post/_post_notification_toggle.html
 create mode 100644 app/templates/post/_reply_notification_toggle.html

diff --git a/app/post/routes.py b/app/post/routes.py
index 1ac5c449..aaeb4f1d 100644
--- a/app/post/routes.py
+++ b/app/post/routes.py
@@ -830,4 +830,24 @@ def post_reply_delete(post_id: int, comment_id: int):
         post.flush_cache()
         flash(_('Comment deleted.'))
         # todo: federate delete
-    return redirect(url_for('activitypub.post_ap', post_id=post.id))
\ No newline at end of file
+    return redirect(url_for('activitypub.post_ap', post_id=post.id))
+
+
+@bp.route('/post/<int:post_id>/notification', methods=['GET', 'POST'])
+@login_required
+def post_notification(post_id: int):
+    post = Post.query.get_or_404(post_id)
+    if post.user_id == current_user.id:
+        post.notify_author = not post.notify_author
+        db.session.commit()
+    return render_template('post/_post_notification_toggle.html', post=post)
+
+
+@bp.route('/post_reply/<int:post_reply_id>/notification', methods=['GET', 'POST'])
+@login_required
+def post_reply_notification(post_reply_id: int):
+    post_reply = PostReply.query.get_or_404(post_reply_id)
+    if post_reply.user_id == current_user.id:
+        post_reply.notify_author = not post_reply.notify_author
+        db.session.commit()
+    return render_template('post/_reply_notification_toggle.html', comment={'comment': post_reply})
diff --git a/app/templates/community/_notification_toggle.html b/app/templates/community/_notification_toggle.html
index 42bfe40e..4cee1970 100644
--- a/app/templates/community/_notification_toggle.html
+++ b/app/templates/community/_notification_toggle.html
@@ -1,4 +1,4 @@
-<a href="/community/{{ community.id }}/notification"
+<a href="/community/{{ community.id }}/notification" rel="nofollow"
 class="fe {{ 'fe-bell' if community.notify_new_posts(current_user.id) else 'fe-no-bell' }} no-underline"
 hx-post="/community/{{ community.id }}/notification" hx-trigger="click throttle:1s" hx-swap="outerHTML"
 title="{{ _('Notify about every new post. Not advisable in high traffic communities!') }}"></a>
\ No newline at end of file
diff --git a/app/templates/post/_post_full.html b/app/templates/post/_post_full.html
index 39eabe87..db5d1f85 100644
--- a/app/templates/post/_post_full.html
+++ b/app/templates/post/_post_full.html
@@ -12,7 +12,11 @@
             <div class="voting_buttons">
                 {% include "post/_post_voting_buttons.html" %}
             </div>
-            <h1 class="mt-2 post_title">{{ post.title }}</h1>
+            <h1 class="mt-2 post_title">{{ post.title }}
+                {% if current_user.is_authenticated and post.user_id == current_user.id %}
+                    {% include 'post/_post_notification_toggle.html' %}
+                {% endif %}
+            </h1>
             {% if post.url %}
                 <p><small><a href="{{ post.url }}" rel="nofollow ugc" target="_blank">{{ post.url|shorten_url }}
                 <img src="/static/images/external_link_black.svg" class="external_link_icon" alt="External link"  />
@@ -53,7 +57,11 @@
             <div class="voting_buttons">
                 {% include "post/_post_voting_buttons.html" %}
             </div>
-            <h1 class="mt-2 post_title">{{ post.title }}</h1>
+            <h1 class="mt-2 post_title">{{ post.title }}
+                {% if current_user.is_authenticated and post.user_id == current_user.id %}
+                    {% include 'post/_post_notification_toggle.html' %}
+                {% endif %}
+            </h1>
             {% if post.type == POST_TYPE_LINK and post.image_id and not (post.url and 'youtube.com' in post.url) %}
                 <div class="url_thumbnail">
                     <a href="{{ post.url }}" target="_blank"><img src="{{ post.image.thumbnail_url() }}" alt="{{ post.image.alt_text }}"
diff --git a/app/templates/post/_post_notification_toggle.html b/app/templates/post/_post_notification_toggle.html
new file mode 100644
index 00000000..2b5eac0c
--- /dev/null
+++ b/app/templates/post/_post_notification_toggle.html
@@ -0,0 +1,4 @@
+<a href="{{ url_for('post.post_notification', post_id=post.id) }}" rel="nofollow"
+class="small fe {{ 'fe-bell' if post.notify_author else 'fe-no-bell' }} no-underline"
+hx-post="{{ url_for('post.post_notification', post_id=post.id) }}" hx-trigger="click throttle:1s" hx-swap="outerHTML"
+title="{{ _('Notify about replies') }}"></a>
\ No newline at end of file
diff --git a/app/templates/post/_reply_notification_toggle.html b/app/templates/post/_reply_notification_toggle.html
new file mode 100644
index 00000000..448e36d8
--- /dev/null
+++ b/app/templates/post/_reply_notification_toggle.html
@@ -0,0 +1,4 @@
+<a href="{{ url_for('post.post_reply_notification', post_reply_id=comment['comment'].id) }}" rel="nofollow"
+class="pl-2 small fe {{ 'fe-bell' if comment['comment'].notify_author else 'fe-no-bell' }} no-underline"
+hx-post="{{ url_for('post.post_reply_notification', post_reply_id=comment['comment'].id) }}" hx-trigger="click throttle:1s" hx-swap="outerHTML"
+title="{{ _('Notify about replies') }}"></a>
\ No newline at end of file
diff --git a/app/templates/post/post.html b/app/templates/post/post.html
index 8b2b10ed..386f43ba 100644
--- a/app/templates/post/post.html
+++ b/app/templates/post/post.html
@@ -87,11 +87,13 @@
                                 {% if comment['comment'].author.created_recently() %}
                                     <span class="fe fe-new-account small" title="New account"> </span>
                                 {% endif %}
-                                {% if comment['comment'].author.reputation < -10 %}
-                                    <span class="fe fe-warning red" title="Very low reputation. Beware."> </span>
-                                    <span class="fe fe-warning red" title="Very low reputation. Beware!"> </span>
-                                {% elif comment['comment'].author.reputation < 0 %}
-                                    <span class="fe fe-warning orangered" title="Low reputation."> </span>
+                                {% if comment['comment'].author.id != current_user.id %}
+                                    {% if comment['comment'].author.reputation < -10 %}
+                                        <span class="fe fe-warning red" title="Very low reputation. Beware."> </span>
+                                        <span class="fe fe-warning red" title="Very low reputation. Beware!"> </span>
+                                    {% elif comment['comment'].author.reputation < 0 %}
+                                        <span class="fe fe-warning orangered" title="Low reputation."> </span>
+                                    {% endif %}
                                 {% endif %}
                                 {% if comment['comment'].author.id == post.author.id %}<span title="Submitter of original post" aria-label="submitter" class="small">[OP]</span>{% endif %}
                                 <span class="text-muted small">{{ moment(comment['comment'].posted_at).fromNow(refresh=True) }}{% if comment['comment'].edited_at %}, edited {{ moment(comment['comment'].edited_at).fromNow(refresh=True) }} {% endif %}</span>
@@ -108,6 +110,9 @@
                                 {% if post.comments_enabled %}
                                     <a href="{{ url_for('post.add_reply', post_id=post.id, comment_id=comment['comment'].id) }}" rel="nofollow"><span class="fe fe-reply"></span> reply</a>
                                 {% endif %}
+                                {% if current_user.id == comment['comment'].author.id %}
+                                    {% include "post/_reply_notification_toggle.html" %}
+                                {% endif %}
                             {% endif %}
                             <a href="{{ url_for('post.post_reply_options', post_id=post.id, comment_id=comment['comment'].id) }}" class="comment_actions_link" rel="nofollow"><span class="fe fe-options" title="Options"> </span></a>
                         </div>