summaryrefslogtreecommitdiffstats
path: root/activities/models
diff options
context:
space:
mode:
authorAndrew Godwin2022-11-21 19:59:27 -0700
committerAndrew Godwin2022-11-21 19:59:27 -0700
commitf6132a8e21aef93f11359048cf544bcad1f9a2fc (patch)
tree1cd7cb7aa86a156bc5870ceb71c14d449718baf9 /activities/models
parent45c7e0a38c5e0dac2538bfe5fb3253fdaeece534 (diff)
downloadtakahe-f6132a8e21aef93f11359048cf544bcad1f9a2fc.tar.gz
takahe-f6132a8e21aef93f11359048cf544bcad1f9a2fc.tar.bz2
takahe-f6132a8e21aef93f11359048cf544bcad1f9a2fc.zip
Initial mentions-linking code
Diffstat (limited to 'activities/models')
-rw-r--r--activities/models/post.py39
1 files changed, 38 insertions, 1 deletions
diff --git a/activities/models/post.py b/activities/models/post.py
index c3de16f..b5cd4d6 100644
--- a/activities/models/post.py
+++ b/activities/models/post.py
@@ -1,9 +1,11 @@
+import re
from typing import Dict, Optional
import httpx
import urlman
from django.db import models, transaction
from django.utils import timezone
+from django.utils.safestring import mark_safe
from activities.models.fan_out import FanOut
from activities.models.timeline_event import TimelineEvent
@@ -151,9 +153,44 @@ class Post(StatorModel):
def get_absolute_url(self):
return self.urls.view
+ ### Content cleanup and extraction ###
+
+ mention_regex = re.compile(
+ r"([^\w\d\-_])(@[\w\d\-_]+(?:@[\w\d\-_]+\.[\w\d\-_\.]+)?)"
+ )
+
+ def linkify_mentions(self, content):
+ """
+ Links mentions _in the context of the post_ - meaning that if there's
+ a short @andrew mention, it will look at the mentions link to resolve
+ it rather than presuming it's local.
+ """
+
+ def replacer(match):
+ print(match)
+ precursor = match.group(1)
+ handle = match.group(2)
+ # If the handle has no domain, try to match it with a mention
+ if "@" not in handle.lstrip("@"):
+ identity = self.mentions.filter(username=handle.lstrip("@")).first()
+ if identity:
+ url = identity.urls.view
+ else:
+ url = None
+ else:
+ url = f"/{handle}/"
+ # If we have a URL, link to it, otherwise don't link
+ if url:
+ return f"{precursor}<a href='{url}'>{handle}</a>"
+ else:
+ return match.group()
+
+ print(f"replacing on {content!r}")
+ return mark_safe(self.mention_regex.sub(replacer, content))
+
@property
def safe_content(self):
- return sanitize_post(self.content)
+ return self.linkify_mentions(sanitize_post(self.content))
### Async helpers ###