From 0bced8fe174de24ebd3823c93997cb2d2347dfe9 Mon Sep 17 00:00:00 2001
From: Andrew Godwin
Date: Mon, 28 Nov 2022 22:14:52 -0700
Subject: A few hashtag fixups

---
 activities/models/hashtag.py |  3 ++-
 activities/models/post.py    | 14 ++++++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

(limited to 'activities/models')

diff --git a/activities/models/hashtag.py b/activities/models/hashtag.py
index a5754f7..7085b15 100644
--- a/activities/models/hashtag.py
+++ b/activities/models/hashtag.py
@@ -8,6 +8,7 @@ from django.db import models
 from django.utils import timezone
 from django.utils.safestring import mark_safe
 
+from core.html import strip_html
 from core.models import Config
 from stator.models import State, StateField, StateGraph, StatorModel
 
@@ -174,7 +175,7 @@ class Hashtag(StatorModel):
         Return a parsed and sanitized of hashtags found in content without
         leading '#'.
         """
-        hashtag_hits = cls.hashtag_regex.findall(content)
+        hashtag_hits = cls.hashtag_regex.findall(strip_html(content))
         hashtags = sorted({tag[1].lower() for tag in hashtag_hits})
         return list(hashtags)
 
diff --git a/activities/models/post.py b/activities/models/post.py
index f3b1879..eecce04 100644
--- a/activities/models/post.py
+++ b/activities/models/post.py
@@ -77,6 +77,17 @@ class PostStates(StateGraph):
 
 
 class PostQuerySet(models.QuerySet):
+    def public(self, include_replies: bool = False):
+        query = self.filter(
+            visibility__in=[
+                Post.Visibilities.public,
+                Post.Visibilities.local_only,
+            ],
+        )
+        if not include_replies:
+            return query.filter(in_reply_to__isnull=True)
+        return query
+
     def local_public(self, include_replies: bool = False):
         query = self.filter(
             visibility__in=[
@@ -104,6 +115,9 @@ class PostManager(models.Manager):
     def get_queryset(self):
         return PostQuerySet(self.model, using=self._db)
 
+    def public(self, include_replies: bool = False):
+        return self.get_queryset().public(include_replies=include_replies)
+
     def local_public(self, include_replies: bool = False):
         return self.get_queryset().local_public(include_replies=include_replies)
 
-- 
cgit v1.2.3