Render https:// posts as at://

Author: jcsalterego

render-configs.py

@@ -3,17 +3,59 @@
 import json
 import os
 import re
+import sys
+
+import requests
 
 WORKER_SENTINEL = "\n\n// CONFIGS\n\n"
 LIST_ITEM_REGEX = re.compile(r"^- ")
+POST_REGEX = re.compile(r"^.*[\./]bsky\.app/profile/(.+?)/post/([a-z0-9]+)")
+
+
+def resolve_handles(handles):
+    dids = {}
+    for handle in handles:
+        url = f"https://bsky.social/xrpc/com.atproto.identity.resolveHandle?handle={handle}"
+        response = requests.get(url)
+        if response.status_code == 200:
+            did = response.json()["did"]
+            dids[handle] = did
+    return dids
 
 
 def render_search_terms(search_terms):
-    rv = []
-    for term in search_terms:
-        term = re.compile(LIST_ITEM_REGEX).sub("", term)
-        rv.append(term)
-    return rv
+    handles = set()
+    rendered_terms = []
+
+    # strip out list item markers
+    terms = [re.compile(LIST_ITEM_REGEX).sub("", term) for term in search_terms]
+
+    # collect handles and pins
+    for term in terms:
+        post_matches = POST_REGEX.match(term)
+        if post_matches:
+            handle = post_matches.group(1)
+            handles.add(handle)
+
+    # resolve handles
+    dids = resolve_handles(handles)
+
+    # replace handles with DIDs
+    for term in terms:
+        post_matches = POST_REGEX.match(term)
+        if post_matches:
+            handle = post_matches.group(1)
+            rkey = post_matches.group(2)
+            did = dids[handle]
+            if did:
+                at_url = f"at://{did}/app.bsky.feed.post/{rkey}"
+                rendered_terms.append(at_url)
+            else:
+                print(f"WARN: Failed to resolve handle {handle}", file=sys.stderr)
+        else:
+            rendered_terms.append(term)
+
+    return rendered_terms
 
 
 def parse_config(dirname, markdown_contents):