Handle post edits, follow undos

1 files changed, 20 insertions, 2 deletions

diff --git a/activities/models/post.py b/activities/models/post.py
index 3847b63..473755b 100644
--- a/activities/models/post.py
+++ b/activities/models/post.py
@@ -28,7 +28,7 @@ class PostStates(StateGraph):
         post = await instance.afetch_full()
         # Non-local posts should not be here
         if not post.local:
-            raise ValueError("Trying to run handle_new on a non-local post!")
+            raise ValueError(f"Trying to run handle_new on a non-local post {post.pk}!")
         # Build list of targets - mentions always included
         targets = set()
         async for mention in post.mentions.all():
@@ -122,6 +122,9 @@ class Post(StatorModel):
     # When the post was originally created (as opposed to when we received it)
     published = models.DateTimeField(default=timezone.now)
 
+    # If the post has been edited after initial publication
+    edited = models.DateTimeField(blank=True, null=True)
+
     created = models.DateTimeField(auto_now_add=True)
     updated = models.DateTimeField(auto_now=True)
 
@@ -245,7 +248,7 @@ class Post(StatorModel):
             post.sensitive = data.get("as:sensitive", False)
             post.url = data.get("url", None)
             post.published = parse_ld_date(data.get("published", None))
-            # TODO: to
+            post.edited = parse_ld_date(data.get("updated", None))
             # Mentions and hashtags
             post.hashtags = []
             for tag in get_list(data, "tag"):
@@ -254,6 +257,9 @@ class Post(StatorModel):
                     post.mentions.add(mention_identity)
                 elif tag["type"].lower() == "as:hashtag":
                     post.hashtags.append(tag["name"].lstrip("#"))
+                elif tag["type"].lower() == "http://joinmastodon.org/ns#emoji":
+                    # TODO: Handle incoming emoji
+                    pass
                 else:
                     raise ValueError(f"Unknown tag type {tag['type']}")
             # Visibility and to
@@ -313,6 +319,18 @@ class Post(StatorModel):
             post.transition_perform(PostStates.fanned_out)
 
     @classmethod
+    def handle_update_ap(cls, data):
+        """
+        Handles an incoming update request
+        """
+        with transaction.atomic():
+            # Ensure the Create actor is the Post's attributedTo
+            if data["actor"] != data["object"]["attributedTo"]:
+                raise ValueError("Create actor does not match its Post object", data)
+            # Find it and update it
+            cls.by_ap(data["object"], create=False, update=True)
+
+    @classmethod
     def handle_delete_ap(cls, data):
         """
         Handles an incoming create request