Schedule reminder on revenue share start and purge on manual close

src/mappings/token/index.ts

@@ -20,6 +20,7 @@ import {
   CreatorTokenMarketStarted,
   CreatorTokenMarketStartedEventData,
   CreatorTokenRevenueShareEnded,
+  CreatorTokenRevenueShareEndedReminder,
   CreatorTokenRevenueSharePlanned,
   CreatorTokenRevenueShareStarted,
   CreatorTokenRevenueSplitIssuedEventData,
@@ -41,10 +42,12 @@ import {
   VestedSale,
   VestingSchedule,
 } from '../../model'
+import { BLOCKS_PER_DAY } from '../../server-extension/resolvers/CreatorToken'
 import { getCurrentBlockHeight } from '../../utils/blockHeight'
 import { EventHandlerContext } from '../../utils/events'
 import { criticalError } from '../../utils/misc'
 import { addNotification } from '../../utils/notification'
+import { removeFutureNotification, removeNotification } from '../../utils/notification/helpers'
 import { getChannelOwnerAccount, notifyChannelFollowers, parseChannelTitle } from '../content/utils'
 import { deserializeMetadata, genericEventFields } from '../utils'
 import {
@@ -695,6 +698,26 @@ export async function processRevenueSplitIssuedEvent({
     event,
     endsAt // schedule for end block
   )
+
+  // This event should have dispatch block of ending block plus about 3 days
+  // If user closes the share before its execution, this notification will be removing during share closing mapping
+  const revenueSharedEndedReminderNotification = new CreatorTokenRevenueShareEndedReminder({
+    revenueShareId: revenueShare.id,
+    channelTitle: parseChannelTitle(channel),
+    channelId: channel.id,
+    tokenSymbol: parseCreatorTokenSymbol(token),
+    tokenId: tokenId.toString(),
+  })
+
+  const channelOwnerAccount = await getChannelOwnerAccount(overlay, channel)
+  await addNotification(
+    overlay,
+    channelOwnerAccount,
+    new ChannelRecipient({ channel: channel.id }),
+    revenueSharedEndedReminderNotification,
+    undefined,
+    endsAt + BLOCKS_PER_DAY * 3 // schedule for end block
+  )
 }
 
 export async function processMemberJoinedWhitelistEvent({
@@ -799,6 +822,10 @@ export async function processRevenueSplitFinalizedEvent({
     .getByIdOrFail(token.currentRevenueShareId!)
   revenueShare.finalized = true
   token.currentRevenueShareId = null
+
+  await removeFutureNotification(overlay.getEm(), 'CreatorTokenRevenueShareEndedReminder', 1, {
+    tokenId: token.id,
+  })
 }
 
 export async function processUserParticipatedInSplitEvent({

src/utils/notification/helpers.ts

@@ -1,5 +1,5 @@
 import { Flat } from 'lodash'
-import { EntityManager } from 'typeorm'
+import { EntityManager, FindOptionsWhere } from 'typeorm'
 import {
   Account,
   AccountNotificationPreferences,
@@ -72,6 +72,7 @@ export function defaultNotificationPreferences(): AccountNotificationPreferences
     crtRevenueShareStarted: notificationPrefAllTrue(),
     crtRevenueSharePlanned: notificationPrefAllTrue(),
     crtRevenueShareEnded: notificationPrefAllTrue(),
+    crtRevenueShareEndedReminder: notificationPrefAllTrue(),
   })
 }
 
@@ -144,6 +145,9 @@ export function preferencesForNotification(
       return preferences.crtRevenueSharePlanned
     case 'CreatorTokenRevenueShareEnded':
       return preferences.crtRevenueShareEnded
+    case 'CreatorTokenRevenueShareEndedReminder':
+      return preferences.crtRevenueShareEnded
+
     default: // all the remaining notifications (v2 scope) are not enabled by default
       return new NotificationPreference({ inAppEnabled: false, emailEnabled: false })
   }
@@ -299,6 +303,45 @@ export const addNotification = async (
   }
 }
 
+export async function removeFutureNotification(
+  em: EntityManager,
+  notificationType: Notification['notificationType']['isTypeOf'],
+  currentBlockHeight: number,
+  filters: Record<string, string | number>
+) {
+  const result: { id: string }[] = await em.query(
+    `
+    SELECT
+        id 
+    FROM 
+        public.notification 
+    WHERE
+        notification_type ->> 'isTypeOf'=$1
+      AND
+        dispatch_block > $2
+        ${Object.keys(filters).map((key, idx) => {
+          return `
+      AND
+        notification_type ->> '${key}'=$${idx + 3}
+        `
+        })}
+         `,
+    [notificationType, currentBlockHeight, ...Object.values(filters)]
+  )
+  const notificationToRemove = result[0]
+
+  if (!notificationToRemove) {
+    return
+  }
+
+  await em.getRepository(Notification).delete({
+    id: notificationToRemove.id,
+  })
+  await em.getRepository(NotificationEmailDelivery).delete({
+    notificationId: notificationToRemove.id,
+  })
+}
+
 async function saveNextNotificationId(
   em: EntityManager,
   nextNotificationId: number,