Cleanup DB at the end of scripts

Author: pimterry

scripts/add-trial-subscription.ts

@@ -2,7 +2,7 @@
 
 import moment from 'moment';
 import { getUsersByEmail, createUser, updateUserMetadata } from '../api/src/user-data-facade';
-import { initializeDbConnection } from '../api/src/db/database';
+import { closeDatabase, initializeDbConnection } from '../api/src/db/database';
 
 // Add a trial subscription in Auth0 for the target user. Occasionally useful
 // for actual custom trials in some cases, but mostly for open-source contributors
@@ -23,7 +23,7 @@ import { initializeDbConnection } from '../api/src/db/database';
 
     console.log(`Adding ${duration.asDays()} day subscription for ${email}`);
 
-    await initializeDbConnection();
+    const db = await initializeDbConnection();
     const users = await getUsersByEmail(email);
 
     let userId: string;
@@ -47,10 +47,11 @@ import { initializeDbConnection } from '../api/src/db/database';
         return process.exit(1);
     }
 
-    updateUserMetadata(userId!, {
+    await updateUserMetadata(userId!, {
         subscription_status: 'trialing',
         subscription_sku: 'pro-monthly',
         subscription_expiry: Date.now() + duration.asMilliseconds(),
         subscription_quantity: 1
     });
+    await closeDatabase(db);
 })();

scripts/cancel-account.ts

@@ -1,7 +1,7 @@
 #!./node_modules/.bin/tsx
 
 import { getUsersByEmail, updateUserMetadata } from '../api/src/user-data-facade';
-import { initializeDbConnection } from '../api/src/db/database';
+import { initializeDbConnection, closeDatabase } from '../api/src/db/database';
 
 // Cancel an account - closing the subscription and setting the expiry date to
 // now. This immediately ends any active subscription, useful for refunds or
@@ -13,7 +13,7 @@ import { initializeDbConnection } from '../api/src/db/database';
     const email = process.argv[2];
     console.log(`Cancelling account for ${email}`);
 
-    await initializeDbConnection();
+    const db = await initializeDbConnection();
     const users = await getUsersByEmail(email);
 
     if (users.length !== 1) {
@@ -27,4 +27,5 @@ import { initializeDbConnection } from '../api/src/db/database';
         subscription_status: 'deleted',
         subscription_expiry: Date.now()
     });
+    await closeDatabase(db);
 })();