skip cleaning up devices of pro plan customers

cleanup.ts

@@ -1,5 +1,5 @@
 import AWS = require('aws-sdk')
-import { getDevicesOfThing } from './db'
+import { getDevicesOfThing, getUserRecord, postponeTokenDeletion } from './db'
 import { handleBackchannelBulkUndiscover } from './handler'
 
 AWS.config.update({ region: process.env.VSH_IOT_REGION })
@@ -148,6 +148,16 @@ async function totallyCleanUpThing(thingName) {
   const data: any = await describeThing(thingName)
   const { userId } = data.attributes
 
+  //skip cleaning up things belonging to pro plan customers:
+  const account = await getUserRecord(userId)
+  if (account.plan === 'pro') {
+    await postponeTokenDeletion(userId, 60)
+
+    throw new Error(
+      `skipping PRO plan customer ${userId}, thingId ${thingName}`
+    )
+  }
+
   const devices = await getDevicesOfThing(userId, thingName)
   console.log('devices::', devices.length)
 

db.ts

@@ -59,6 +59,33 @@ export function upsertTokens(
   })
 }
 
+export function postponeTokenDeletion(
+  userId: string,
+  daysFromNow: number
+): Promise<any> {
+  const params = {
+    TableName: 'VSH',
+    Key: {
+      PK: `USER#${userId}`,
+      SK: 'TOKEN',
+    },
+    UpdateExpression: 'set deleteAtUnixTime = :ttl',
+    ExpressionAttributeValues: {
+      ':ttl': dayjs().add(daysFromNow, 'day').unix(),
+    },
+  }
+
+  return new Promise((resolve, reject) => {
+    docClient.update(params, function (err, data) {
+      if (err) {
+        return reject(err)
+      } else {
+        return resolve(data)
+      }
+    })
+  })
+}
+
 export function updateUserRecord(partialUser: PartialUserRecord): Promise<any> {
   const updateRec = { ...partialUser }