Add quotes count

This commit is contained in:
Alex Gleason 2024-06-02 20:30:57 -05:00
parent a9dcebd7c2
commit 22c840092f
No known key found for this signature in database
GPG Key ID: 7211D1F99744FBB7
6 changed files with 31 additions and 4 deletions

View File

@ -21,6 +21,7 @@ interface EventStatsRow {
replies_count: number; replies_count: number;
reposts_count: number; reposts_count: number;
reactions_count: number; reactions_count: number;
quotes_count: number;
reactions: string; reactions: string;
} }

View File

@ -0,0 +1,12 @@
import { Kysely } from 'kysely';
export async function up(db: Kysely<any>): Promise<void> {
await db.schema
.alterTable('event_stats')
.addColumn('quotes_count', 'integer', (col) => col.notNull().defaultTo(0))
.execute();
}
export async function down(db: Kysely<any>): Promise<void> {
await db.schema.alterTable('event_stats').dropColumn('quotes_count').execute();
}

View File

@ -11,6 +11,7 @@ export interface AuthorStats {
export interface EventStats { export interface EventStats {
replies_count: number; replies_count: number;
reposts_count: number; reposts_count: number;
quotes_count: number;
reactions: Record<string, number>; reactions: Record<string, number>;
} }

View File

@ -297,6 +297,7 @@ async function gatherEventStats(events: DittoEvent[]): Promise<DittoTables['even
reposts_count: Math.max(0, row.reposts_count), reposts_count: Math.max(0, row.reposts_count),
replies_count: Math.max(0, row.replies_count), replies_count: Math.max(0, row.replies_count),
reactions_count: Math.max(0, row.reactions_count), reactions_count: Math.max(0, row.reactions_count),
quotes_count: Math.max(0, row.quotes_count),
reactions: row.reactions, reactions: row.reactions,
})); }));
} }

View File

@ -3,7 +3,7 @@ import { Kysely, UpdateObject } from 'kysely';
import { SetRequired } from 'type-fest'; import { SetRequired } from 'type-fest';
import { DittoTables } from '@/db/DittoTables.ts'; import { DittoTables } from '@/db/DittoTables.ts';
import { findReplyTag, getTagSet } from '@/utils/tags.ts'; import { findQuoteTag, findReplyTag, getTagSet } from '@/utils/tags.ts';
import { Conf } from '@/config.ts'; import { Conf } from '@/config.ts';
interface UpdateStatsOpts { interface UpdateStatsOpts {
@ -34,14 +34,24 @@ export async function updateStats({ event, kysely, store, x = 1 }: UpdateStatsOp
async function handleEvent1(kysely: Kysely<DittoTables>, event: NostrEvent, x: number): Promise<void> { async function handleEvent1(kysely: Kysely<DittoTables>, event: NostrEvent, x: number): Promise<void> {
await updateAuthorStats(kysely, event.pubkey, ({ notes_count }) => ({ notes_count: Math.max(0, notes_count + x) })); await updateAuthorStats(kysely, event.pubkey, ({ notes_count }) => ({ notes_count: Math.max(0, notes_count + x) }));
const inReplyToId = findReplyTag(event.tags)?.[1]; const replyId = findReplyTag(event.tags)?.[1];
if (inReplyToId) { const quoteId = findQuoteTag(event.tags)?.[1];
if (replyId) {
await updateEventStats( await updateEventStats(
kysely, kysely,
inReplyToId, replyId,
({ replies_count }) => ({ replies_count: Math.max(0, replies_count + x) }), ({ replies_count }) => ({ replies_count: Math.max(0, replies_count + x) }),
); );
} }
if (quoteId) {
await updateEventStats(
kysely,
quoteId,
({ quotes_count }) => ({ quotes_count: Math.max(0, quotes_count + x) }),
);
}
} }
/** Update stats for kind 3 event. */ /** Update stats for kind 3 event. */
@ -208,6 +218,7 @@ export async function updateEventStats(
replies_count: 0, replies_count: 0,
reposts_count: 0, reposts_count: 0,
reactions_count: 0, reactions_count: 0,
quotes_count: 0,
reactions: '{}', reactions: '{}',
}; };

View File

@ -125,6 +125,7 @@ async function renderStatus(event: DittoEvent, opts: RenderStatusOpts): Promise<
pleroma: { pleroma: {
emoji_reactions: reactions, emoji_reactions: reactions,
expires_at: !isNaN(expiresAt.getTime()) ? expiresAt.toISOString() : undefined, expires_at: !isNaN(expiresAt.getTime()) ? expiresAt.toISOString() : undefined,
quotes_count: event.event_stats?.quotes_count ?? 0,
}, },
}; };
} }