Implement some more server features

1 files changed, 50 insertions, 22 deletions

diff --git a/server.py b/server.py
index 2c43a46..27f36b0 100644
--- a/server.py
+++ b/server.py
@@ -15,26 +15,43 @@ domain = 'https://0.exozy.me'
 def collection_append(file, item):
     with open(file) as f:
         collection = load(f)
-    collection['totalItems'] += 1
     collection['orderedItems'].append(item)
+    collection['totalItems'] += 1
+    with open(file, 'w') as f:
+         dump(collection, f)
+
+
+def collection_pop(file, item):
+    with open(file) as f:
+        collection = load(f)
+    collection['orderedItems'].pop(item)
+    collection['totalItems'] -= 1
     with open(file, 'w') as f:
          dump(collection, f)
 
 
 def iri_to_actor(iri):
     if domain in iri:
-        name = search(f'^{domain}/users/(.*?)#main-key$', iri).group(1)
+        name = search(f'^{domain}/users/(.*?)$', iri.removesuffix('#main-key')).group(1)
         actorfile = f'users/{name}'
     else:
-        actorfile = f'users/{quote_plus(iri)}'
+        actorfile = f'users/{quote_plus(iri.removesuffix("#main-key"))}'
     if not isfile(actorfile):
         with open(actorfile, 'w') as f:
-            resp = get(iri.removesuffix('#main-key'), headers={'Accept': 'application/activity+json'})
+            resp = get(iri, headers={'Accept': 'application/activity+json'})
             f.write(resp.text)
     with open(actorfile) as f:
         return load(f)
 
 
+def send(to, headers, body):
+    actor = iri_to_actor(to)
+    headers['Host'] = to.split('/')[2]
+    resp = post(actor['inbox'], headers=headers, data=body)
+    print(resp)
+    print(resp.text)
+
+
 class fuwuqi(SimpleHTTPRequestHandler):
     def do_POST(self):
         body = self.rfile.read(int(self.headers['Content-Length']))
@@ -72,10 +89,10 @@ class fuwuqi(SimpleHTTPRequestHandler):
 
         # Make sure activity doer matches HTTP signature
         actor = keyid.removesuffix('#main-key')
-        if 'actor' in activity and activity['actor'] != actor:
-            self.send_response(401)
-            return
-        if 'attributedTo' in activity and activity['attributedTo'] != actor:
+        if ('actor' in activity and activity['actor'] != actor) or
+           ('attributedTo' in activity and activity['attributedTo'] != actor) or
+           ('actor' in activity['object'] and activity['object']['actor'] != actor) or
+           ('attributedTo' in activity['object'] and activity['object']['attributedTo'] != actor)
             self.send_response(401)
             return
 
@@ -85,25 +102,36 @@ class fuwuqi(SimpleHTTPRequestHandler):
         elif self.path.endswith('outbox'):
             # C2S
             collection_append(f'users/{username}.outbox', activity)
+            # Clients responsible for addressing activity
+            for to in activity['to']:
+                if 'followers' in to or to 'https://www.w3.org/ns/activitystreams#Public':
+                    with open(f'users/{username}.followers') as f:
+                        for follower in load(f)['orderedItems']:
+                            send(follower, self.headers, body)
+                else:
+                    send(to, self.headers, body)
+            # Process activity
             if activity['type'] == 'Create':
-                id = activity['id'].split('/')[-1]
+                # Post
+                id = activity['object']['id'].split('/')[-1]
                 with open(f'users/{username}.statuses/{id}', 'w') as f:
                     dump(activity['object'], f)
-                # Send to followers
-                with open(f'users/{username}.following') as f:
-                    for followed in load(f)['orderedItems']:
-                        actor = iri_to_actor(followed)
-                        self.headers['Host'] = followed.split('/')[2]
-                        resp = post(actor['inbox'], headers=self.headers, data=body)
-                        print(resp)
-                        print(resp.text)
+            elif activity['type'] == 'Accept':
+                # Accept follow request
+                collection_append(f'users/{username}.followers', activity['object']['actor'])
             elif activity['type'] == 'Follow':
-                object = iri_to_actor(activity['object'])
-                self.headers['Host'] = activity['object'].split('/')[2]
-                resp = post(object['inbox'], headers=self.headers, data=body)
-                print(resp)
-                print(resp.text)
+                # Follow request
                 collection_append(f'users/{username}.following', activity['object'])
+            elif activity['type'] == 'Like':
+                # Like post
+                collection_append(f'users/{username}.liked', activity['object'])
+            elif activity['type'] == 'Undo':
+                if activity['object']['type'] == 'Follow':
+                    # Unfollow request
+                    collection_remove(f'users/{username}.following', activity['object']['object'])
+                elif activity['object']['type'] == 'Like':
+                    # Unlike post
+                    collection_remove(f'users/{username}.liked', activity['object']['object'])
         
         self.send_response(200)
         self.end_headers()