From bbcd73d3f1be1253dba049c22cca8d6d40fe54c4 Mon Sep 17 00:00:00 2001
From: Eric Davis <eric@davislv.com>
Date: Wed, 20 Nov 2024 05:43:43 +0000
Subject: [PATCH] feat: track other atproto collections

---
 bsky-activity.py | 24 ++++++++++++++++--------
 1 file(s) changed, 16 insertion(s)(+), 8 deletion(s)(-)

diff --git a/bsky-activity.py b/bsky-activity.py
--- a/bsky-activity.py
+++ b/bsky-activity.py
@@ -10,7 +10,7 @@
 import redis
 import websockets
 
-app_bsky_allowlist = set([
+app_bsky_allowlist = (
     'app.bsky.actor.profile',
     'app.bsky.feed.generator',
     'app.bsky.feed.like',
@@ -26,13 +26,17 @@ 'app.bsky.graph.listitem',
     'app.bsky.graph.starterpack',
     'app.bsky.labeler.service',
     'chat.bsky.actor.declaration',
-])
+)
 
-other_allowlist = set([
-    'social.psky.feed.post',
-    'social.psky.chat.message',
-    'blue.zio.atfile.upload',
-])
+other_allowlist = (
+    'social.psky',
+    'blue.zio.atfile',
+    'com.shinolabs.pinksea',
+    'com.whtwnd',
+    'events.smokesignal',
+    'fyi.unravel',
+    'xyz.statusphere',
+)
 
 async def bsky_activity():
     relay_url = 'wss://jetstream1.us-west.bsky.network/subscribe'
@@ -64,8 +68,12 @@ if payload['operation'] != 'create':
             continue
 
         collection = payload['collection']
-        if collection not in app_bsky_allowlist | other_allowlist:
+        if not collection.startswith(app_bsky_allowlist + other_allowlist):
             continue
+
+        for prefix in other_allowlist:
+            if collection.startswith(prefix):
+                redis_pipe.incr('dev.edavis.atproto.collection.' + prefix.replace('.', '_'))
 
         repo_did = event['did']
         repo_update_time = datetime.now(timezone.utc)

--
tangled.sh