Infect inactive users after 36 hours

Author: ekzyis

lib/constants.js

@@ -215,3 +215,4 @@ export const WALLET_RETRY_TIMEOUT_MS = 60_000 // 1 minute
 export const BECH32_CHARSET = 'qpzry9x8gf2tvdw0s3jn54khce6mua7l'
 
 export const HALLOWEEN_IMMUNITY_HOURS = 6
+export const HALLOWEEN_INACTIVITY_TIMEOUT_HOURS = 36

prisma/migrations/20251022202342_halloween/migration.sql

@@ -9,13 +9,7 @@ CREATE TABLE "Infection" (
     "infecteeId" INTEGER NOT NULL,
     "infectorId" INTEGER,
 
-    CONSTRAINT "Infection_pkey" PRIMARY KEY ("id"),
-    CONSTRAINT "Infection_patient_zero_check" CHECK (
-        CASE
-            WHEN id = 1 THEN true
-            ELSE "infectorId" IS NOT NULL
-        END
-    )
+    CONSTRAINT "Infection_pkey" PRIMARY KEY ("id")
 );
 
 -- CreateIndex
@@ -39,3 +33,21 @@ ALTER TABLE "Infection" ADD CONSTRAINT "Infection_itemActId_fkey" FOREIGN KEY ("
 -- anon is patient zero
 INSERT INTO "Infection" ("infecteeId") VALUES (27);
 UPDATE users SET infected = true WHERE id = 27;
+
+CREATE OR REPLACE FUNCTION schedule_halloween()
+RETURNS INTEGER
+LANGUAGE plpgsql
+AS $$
+DECLARE
+BEGIN
+    -- every hour
+    INSERT INTO pgboss.schedule (name, cron, timezone)
+    VALUES ('halloween', '0 * * * *', 'America/Chicago') ON CONFLICT DO NOTHING;
+    return 0;
+EXCEPTION WHEN OTHERS THEN
+    return 0;
+END;
+$$;
+
+SELECT schedule_halloween();
+DROP FUNCTION IF EXISTS schedule_halloween;

worker/halloween.js

@@ -0,0 +1,31 @@
+import { datePivot } from '@/lib/time'
+import { HALLOWEEN_INACTIVITY_TIMEOUT_HOURS } from '@/lib/constants'
+import { notifyInfected } from '@/lib/webPush'
+
+export async function halloween ({ models }) {
+  const inactiveUsers = await models.$queryRaw`
+    WITH
+        inactive_users AS (
+            SELECT id FROM "users"
+            WHERE id NOT IN (
+                SELECT "userId" FROM "Item"
+                WHERE created_at > ${datePivot(new Date(), { hours: -HALLOWEEN_INACTIVITY_TIMEOUT_HOURS })}
+                AND ("invoiceActionState" IS NULL OR "invoiceActionState" = 'PAID')
+            )
+        ),
+        new_infections AS (
+            INSERT INTO "Infection" ("infecteeId")
+            SELECT id FROM inactive_users
+            ON CONFLICT ("infecteeId") DO NOTHING
+            RETURNING "infecteeId"
+        )
+    UPDATE "users"
+    SET infected = true
+    WHERE id IN (SELECT "infecteeId" FROM new_infections)
+    RETURNING id
+  `
+
+  await Promise.allSettled(
+    inactiveUsers.map(user => notifyInfected(user.id).catch(console.error))
+  )
+}

worker/index.js

@@ -38,6 +38,8 @@ import { payingActionConfirmed, payingActionFailed } from './payingAction'
 import { autoDropBolt11s } from './autoDropBolt11'
 import { postToSocial } from './socialPoster'
 import { untrackOldItems } from './untrackOldItems'
+import { halloween } from './halloween'
+
 // WebSocket polyfill
 import ws from 'isomorphic-ws'
 if (typeof WebSocket === 'undefined') {
@@ -144,6 +146,7 @@ async function work () {
   await boss.work('thisDay', jobWrapper(thisDay))
   await boss.work('socialPoster', jobWrapper(postToSocial))
   await boss.work('untrackOldItems', jobWrapper(untrackOldItems))
+  await boss.work('halloween', jobWrapper(halloween))
 
   console.log('working jobs')
 }