feat: vanish optimization (#446)

Author: vikashsiwach

migrations/20260410_230000_add_is_vanished_to_users_table.js

@@ -0,0 +1,22 @@
+exports.up = async function (knex) {
+  await knex.schema.alterTable('users', (table) => {
+    table.boolean('is_vanished').notNullable().defaultTo(false)
+  })
+
+  await knex.raw(`
+    UPDATE users u
+    SET is_vanished = true
+    WHERE EXISTS (
+      SELECT 1 FROM events e
+      WHERE e.event_pubkey = u.pubkey
+        AND e.event_kind = 62
+        AND e.deleted_at IS NULL
+    )
+  `)
+}
+
+exports.down = function (knex) {
+  return knex.schema.alterTable('users', (table) => {
+    table.dropColumn('is_vanished')
+  })
+}

package-lock.json

@@ -63,6 +63,8 @@ export interface IUserRepository {
   findByPubkey(pubkey: Pubkey, client?: DatabaseClient): Promise<User | undefined>
   upsert(user: Partial<User>, client?: DatabaseClient): Promise<number>
   getBalanceByPubkey(pubkey: Pubkey, client?: DatabaseClient): Promise<bigint>
+  isVanished(pubkey: Pubkey, client?: DatabaseClient): Promise<boolean>
+  setVanished(pubkey: Pubkey, vanished: boolean, client?: DatabaseClient): Promise<number>
   admitUser(pubkey: Pubkey, admittedAt: Date, client?: DatabaseClient): Promise<void>
 }
 

src/@types/repositories.ts

@@ -3,6 +3,7 @@ import { Pubkey } from './base'
 export interface User {
   pubkey: Pubkey
   isAdmitted: boolean
+  isVanished: boolean
   balance: bigint
   tosAcceptedAt?: Date | null
   createdAt: Date
@@ -12,6 +13,7 @@ export interface User {
 export interface DBUser {
   pubkey: Buffer
   is_admitted: boolean
+  is_vanished: boolean
   balance: bigint
   created_at: Date
   updated_at: Date

src/@types/user.ts

@@ -1,12 +1,15 @@
+import { getMasterDbClient, getReadReplicaDbClient } from '../../database/client'
 import { createSettings } from '../settings-factory'
-import { getMasterDbClient } from '../../database/client'
+import { EventRepository } from '../../repositories/event-repository'
 import { GetSubmissionCheckController } from '../../controllers/admission/get-admission-check-controller'
 import { slidingWindowRateLimiterFactory } from '../rate-limiter-factory'
 import { UserRepository } from '../../repositories/user-repository'
 
 export const createGetAdmissionCheckController = () => {
   const dbClient = getMasterDbClient()
-  const userRepository = new UserRepository(dbClient)
+  const readReplicaDbClient = getReadReplicaDbClient()
+  const eventRepository = new EventRepository(dbClient, readReplicaDbClient)
+  const userRepository = new UserRepository(dbClient, eventRepository)
 
   return new GetSubmissionCheckController(
     userRepository,

src/factories/controllers/get-admission-check-controller-factory.ts

@@ -1,14 +1,17 @@
+import { getMasterDbClient, getReadReplicaDbClient } from '../../database/client'
 import { createPaymentsService } from '../payments-service-factory'
 import { createSettings } from '../settings-factory'
-import { getMasterDbClient } from '../../database/client'
+import { EventRepository } from '../../repositories/event-repository'
 import { IController } from '../../@types/controllers'
 import { PostInvoiceController } from '../../controllers/invoices/post-invoice-controller'
 import { slidingWindowRateLimiterFactory } from '../rate-limiter-factory'
 import { UserRepository } from '../../repositories/user-repository'
 
 export const createPostInvoiceController = (): IController => {
   const dbClient = getMasterDbClient()
-  const userRepository = new UserRepository(dbClient)
+  const readReplicaDbClient = getReadReplicaDbClient()
+  const eventRepository = new EventRepository(dbClient, readReplicaDbClient)
+  const userRepository = new UserRepository(dbClient, eventRepository)
   const paymentsService = createPaymentsService()
 
   return new PostInvoiceController(

src/factories/controllers/post-invoice-controller-factory.ts

@@ -1,11 +1,11 @@
+import { IEventRepository, IUserRepository } from '../@types/repositories'
 import { isDeleteEvent, isEphemeralEvent, isGiftWrapEvent, isParameterizedReplaceableEvent, isReplaceableEvent, isRequestToVanishEvent } from '../utils/event'
 import { DefaultEventStrategy } from '../handlers/event-strategies/default-event-strategy'
 import { DeleteEventStrategy } from '../handlers/event-strategies/delete-event-strategy'
 import { EphemeralEventStrategy } from '../handlers/event-strategies/ephemeral-event-strategy'
 import { Event } from '../@types/event'
 import { Factory } from '../@types/base'
 import { GiftWrapEventStrategy } from '../handlers/event-strategies/gift-wrap-event-strategy'
-import { IEventRepository } from '../@types/repositories'
 import { IEventStrategy } from '../@types/message-handlers'
 import { IWebSocketAdapter } from '../@types/adapters'
 import { ParameterizedReplaceableEventStrategy } from '../handlers/event-strategies/parameterized-replaceable-event-strategy'
@@ -14,10 +14,11 @@ import { VanishEventStrategy } from '../handlers/event-strategies/vanish-event-s
 
 export const eventStrategyFactory = (
   eventRepository: IEventRepository,
+  userRepository: IUserRepository,
 ): Factory<IEventStrategy<Event, Promise<void>>, [Event, IWebSocketAdapter]> =>
   ([event, adapter]: [Event, IWebSocketAdapter]) => {
     if (isRequestToVanishEvent(event)) {
-      return new VanishEventStrategy(adapter, eventRepository)
+      return new VanishEventStrategy(adapter, eventRepository, userRepository)
     } else if (isGiftWrapEvent(event)) {
       return new GiftWrapEventStrategy(adapter, eventRepository)
     } else if (isReplaceableEvent(event)) {

src/factories/event-strategy-factory.ts

@@ -18,7 +18,7 @@ export const messageHandlerFactory = (
       {
         return new EventMessageHandler(
           adapter,
-          eventStrategyFactory(eventRepository),
+          eventStrategyFactory(eventRepository, userRepository),
           eventRepository,
           userRepository,
           createSettings,

src/factories/message-handler-factory.ts

@@ -10,9 +10,9 @@ export const createPaymentsService = () => {
   const dbClient = getMasterDbClient()
   const rrDbClient = getReadReplicaDbClient()
   const invoiceRepository = new InvoiceRepository(dbClient)
-  const userRepository = new UserRepository(dbClient)
-  const paymentsProcessor = createPaymentsProcessor()
   const eventRepository = new EventRepository(dbClient, rrDbClient)
+  const userRepository = new UserRepository(dbClient, eventRepository)
+  const paymentsProcessor = createPaymentsProcessor()
 
   return new PaymentsService(
     dbClient,

src/factories/payments-service-factory.ts

@@ -8,7 +8,7 @@ export const staticMirroringWorkerFactory = () => {
   const dbClient = getMasterDbClient()
   const readReplicaDbClient = getReadReplicaDbClient()
   const eventRepository = new EventRepository(dbClient, readReplicaDbClient)
-  const userRepository = new UserRepository(dbClient)
+  const userRepository = new UserRepository(dbClient, eventRepository)
 
   return new StaticMirroringWorker(
     eventRepository,