Add stats:recompute script

scripts/stats-recompute.ts

@@ -0,0 +1,36 @@
+import { nip19 } from 'nostr-tools';
+
+import { db } from '@/db.ts';
+import { DittoTables } from '@/db/DittoTables.ts';
+import { Storages } from '@/storages.ts';
+
+let pubkey: string;
+try {
+  const result = nip19.decode(Deno.args[0]);
+  if (result.type === 'npub') {
+    pubkey = result.data;
+  } else {
+    throw new Error('Invalid npub');
+  }
+} catch {
+  console.error('Invalid npub');
+  Deno.exit(1);
+}
+
+const [followList] = await Storages.db.query([{ kinds: [3], authors: [pubkey], limit: 1 }]);
+
+const authorStats: DittoTables['author_stats'] = {
+  pubkey,
+  followers_count: (await Storages.db.count([{ kinds: [3], '#p': [pubkey] }])).count,
+  following_count: followList?.tags.filter(([name]) => name === 'p')?.length ?? 0,
+  notes_count: (await Storages.db.count([{ kinds: [1], authors: [pubkey] }])).count,
+};
+
+await db.insertInto('author_stats')
+  .values(authorStats)
+  .onConflict((oc) =>
+    oc
+      .column('pubkey')
+      .doUpdateSet(authorStats)
+  )
+  .execute();