From edc4607c8db35bce3e8e7799f73e63597a7c39b3 Mon Sep 17 00:00:00 2001
From: Alex Gleason <alex@alexgleason.me>
Date: Mon, 29 Jul 2024 23:40:08 -0500
Subject: [PATCH 1/2] Improve performance of /context endpoint

---
 src/controllers/api/statuses.ts | 2 +-
 src/queries.ts                  | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/controllers/api/statuses.ts b/src/controllers/api/statuses.ts
index 6a9ed1f5..6d61c76c 100644
--- a/src/controllers/api/statuses.ts
+++ b/src/controllers/api/statuses.ts
@@ -247,7 +247,7 @@ const contextController: AppController = async (c) => {
   if (event) {
     const [ancestorEvents, descendantEvents] = await Promise.all([
       getAncestors(store, event),
-      getDescendants(store, event.id),
+      getDescendants(store, event),
     ]);
 
     await hydrateEvents({
diff --git a/src/queries.ts b/src/queries.ts
index 745429d6..9bce58ca 100644
--- a/src/queries.ts
+++ b/src/queries.ts
@@ -74,7 +74,7 @@ async function getAncestors(store: NStore, event: NostrEvent, result: NostrEvent
     const inReplyTo = replyTag ? replyTag[1] : undefined;
 
     if (inReplyTo) {
-      const [parentEvent] = await store.query([{ kinds: [1], ids: [inReplyTo], limit: 1 }]);
+      const [parentEvent] = await store.query([{ kinds: [1], ids: [inReplyTo], until: event.created_at, limit: 1 }]);
 
       if (parentEvent) {
         result.push(parentEvent);
@@ -88,12 +88,12 @@ async function getAncestors(store: NStore, event: NostrEvent, result: NostrEvent
 
 async function getDescendants(
   store: NStore,
-  eventId: string,
+  event: NostrEvent,
   signal = AbortSignal.timeout(2000),
 ): Promise<NostrEvent[]> {
   return await store
-    .query([{ kinds: [1], '#e': [eventId], limit: 200 }], { signal })
-    .then((events) => events.filter(({ tags }) => findReplyTag(tags)?.[1] === eventId));
+    .query([{ kinds: [1], '#e': [event.id], since: event.created_at, limit: 200 }], { signal })
+    .then((events) => events.filter(({ tags }) => findReplyTag(tags)?.[1] === event.id));
 }
 
 /** Returns whether the pubkey is followed by a local user. */

From a8e81e304e4ea13cfa355773b531845367dd31e4 Mon Sep 17 00:00:00 2001
From: Alex Gleason <alex@alexgleason.me>
Date: Mon, 29 Jul 2024 23:43:04 -0500
Subject: [PATCH 2/2] context: reverse descendants

---
 src/controllers/api/statuses.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/controllers/api/statuses.ts b/src/controllers/api/statuses.ts
index 6d61c76c..4e8d051f 100644
--- a/src/controllers/api/statuses.ts
+++ b/src/controllers/api/statuses.ts
@@ -258,7 +258,7 @@ const contextController: AppController = async (c) => {
 
     const [ancestors, descendants] = await Promise.all([
       renderStatuses(ancestorEvents),
-      renderStatuses(descendantEvents),
+      renderStatuses(descendantEvents.reverse()),
     ]);
 
     return c.json({ ancestors, descendants });