From 0851fbd1ec09b142608667bf90ee806e59cafb28 Mon Sep 17 00:00:00 2001
From: Andrew Godwin
Date: Thu, 17 Nov 2022 18:52:00 -0700
Subject: Add search and better notifications

---
 activities/models/fan_out.py        | 2 +-
 activities/models/timeline_event.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

(limited to 'activities/models')

diff --git a/activities/models/fan_out.py b/activities/models/fan_out.py
index 771be19..6ebbe0a 100644
--- a/activities/models/fan_out.py
+++ b/activities/models/fan_out.py
@@ -37,7 +37,6 @@ class FanOutStates(StateGraph):
                     private_key=post.author.private_key,
                     key_id=post.author.public_key_id,
                 )
-            return cls.sent
         # Handle boosts/likes
         elif fan_out.type == FanOut.Types.interaction:
             interaction = await fan_out.subject_post_interaction.afetch_full()
@@ -74,6 +73,7 @@ class FanOutStates(StateGraph):
                 )
         else:
             raise ValueError(f"Cannot fan out with type {fan_out.type}")
+        return cls.sent
 
 
 class FanOut(StatorModel):
diff --git a/activities/models/timeline_event.py b/activities/models/timeline_event.py
index 368fdad..cf93661 100644
--- a/activities/models/timeline_event.py
+++ b/activities/models/timeline_event.py
@@ -66,7 +66,7 @@ class TimelineEvent(models.Model):
         """
         return cls.objects.get_or_create(
             identity=identity,
-            type=cls.Types.follow,
+            type=cls.Types.followed,
             subject_identity=source_identity,
         )[0]
 
@@ -90,6 +90,7 @@ class TimelineEvent(models.Model):
             identity=identity,
             type=cls.Types.mentioned,
             subject_post=post,
+            subject_identity=post.author,
         )[0]
 
     @classmethod
-- 
cgit v1.2.3