From a39910fa984e5f4896aa860143af07c1044e0271 Mon Sep 17 00:00:00 2001
From: Alex Gleason <alex@alexgleason.me>
Date: Fri, 17 May 2024 13:12:40 -0500
Subject: [PATCH] Add a function to recalculate author stats

---
 scripts/stats-recompute.ts | 25 ++-----------------------
 src/stats.ts               | 38 +++++++++++++++++++++++++++++++++++---
 2 files changed, 37 insertions(+), 26 deletions(-)

diff --git a/scripts/stats-recompute.ts b/scripts/stats-recompute.ts
index dcb0bc0..4037a85 100644
--- a/scripts/stats-recompute.ts
+++ b/scripts/stats-recompute.ts
@@ -1,8 +1,6 @@
 import { nip19 } from 'nostr-tools';
 
-import { DittoDB } from '@/db/DittoDB.ts';
-import { DittoTables } from '@/db/DittoTables.ts';
-import { Storages } from '@/storages.ts';
+import { refreshAuthorStats } from '@/stats.ts';
 
 let pubkey: string;
 try {
@@ -17,23 +15,4 @@ try {
   Deno.exit(1);
 }
 
-const store = await Storages.db();
-const kysely = await DittoDB.getInstance();
-
-const [followList] = await store.query([{ kinds: [3], authors: [pubkey], limit: 1 }]);
-
-const authorStats: DittoTables['author_stats'] = {
-  pubkey,
-  followers_count: (await store.count([{ kinds: [3], '#p': [pubkey] }])).count,
-  following_count: followList?.tags.filter(([name]) => name === 'p')?.length ?? 0,
-  notes_count: (await store.count([{ kinds: [1], authors: [pubkey] }])).count,
-};
-
-await kysely.insertInto('author_stats')
-  .values(authorStats)
-  .onConflict((oc) =>
-    oc
-      .column('pubkey')
-      .doUpdateSet(authorStats)
-  )
-  .execute();
+await refreshAuthorStats(pubkey);
diff --git a/src/stats.ts b/src/stats.ts
index 9204071..74242f7 100644
--- a/src/stats.ts
+++ b/src/stats.ts
@@ -1,11 +1,12 @@
-import { NKinds, NostrEvent } from '@nostrify/nostrify';
+import { NKinds, NostrEvent, NStore } from '@nostrify/nostrify';
 import Debug from '@soapbox/stickynotes/debug';
 import { InsertQueryBuilder, Kysely } from 'kysely';
+import { SetRequired } from 'type-fest';
 
 import { DittoDB } from '@/db/DittoDB.ts';
 import { DittoTables } from '@/db/DittoTables.ts';
 import { Storages } from '@/storages.ts';
-import { findReplyTag } from '@/tags.ts';
+import { findReplyTag, getTagSet } from '@/tags.ts';
 
 type AuthorStat = keyof Omit<DittoTables['author_stats'], 'pubkey'>;
 type EventStat = keyof Omit<DittoTables['event_stats'], 'event_id'>;
@@ -216,4 +217,35 @@ function getFollowDiff(event: NostrEvent, prev?: NostrEvent): AuthorStatDiff[] {
   ];
 }
 
-export { updateStats };
+/** Refresh the author's stats in the database. */
+async function refreshAuthorStats(pubkey: string): Promise<void> {
+  const store = await Storages.db();
+  const stats = await countAuthorStats(store, pubkey);
+
+  const kysely = await DittoDB.getInstance();
+  await kysely.insertInto('author_stats')
+    .values(stats)
+    .onConflict((oc) => oc.column('pubkey').doUpdateSet(stats))
+    .execute();
+}
+
+/** Calculate author stats from the database. */
+async function countAuthorStats(
+  store: SetRequired<NStore, 'count'>,
+  pubkey: string,
+): Promise<DittoTables['author_stats']> {
+  const [{ count: followers_count }, { count: notes_count }, [followList]] = await Promise.all([
+    store.count([{ kinds: [3], '#p': [pubkey] }]),
+    store.count([{ kinds: [1], authors: [pubkey] }]),
+    store.query([{ kinds: [3], authors: [pubkey], limit: 1 }]),
+  ]);
+
+  return {
+    pubkey,
+    followers_count,
+    following_count: getTagSet(followList?.tags ?? [], 'p').size,
+    notes_count,
+  };
+}
+
+export { refreshAuthorStats, updateStats };