stacker.news/worker/streak.js

import { sendUserNotification } from '../api/webPush'
import { FOUND_BLURBS, LOST_BLURBS } from '../lib/constants'

const STREAK_THRESHOLD = 100

export function computeStreaks ({ models }) {
  return async function () {
    console.log('computing streaks')

    // get all eligible users in the last day
    // if the user doesn't have an active streak, add one
    // if they have an active streak but didn't maintain it, end it
    const endingStreaks = await models.$queryRaw`
      WITH day_streaks (id) AS (
        SELECT "userId"
        FROM
        ((SELECT "userId", floor(sum("ItemAct".msats)/1000) as sats_spent
            FROM "ItemAct"
            WHERE (created_at AT TIME ZONE 'UTC' AT TIME ZONE 'America/Chicago')::date >= (now() AT TIME ZONE 'America/Chicago' - interval '1 day')::date
            GROUP BY "userId")
        UNION ALL
        (SELECT "userId", sats as sats_spent
            FROM "Donation"
            WHERE (created_at AT TIME ZONE 'UTC' AT TIME ZONE 'America/Chicago')::date >= (now() AT TIME ZONE 'America/Chicago' - interval '1 day')::date
        )) spending
        GROUP BY "userId"
        HAVING sum(sats_spent) >= 100
      ), existing_streaks (id, started_at) AS (
        SELECT "userId", "startedAt"
        FROM "Streak"
        WHERE "Streak"."endedAt" IS NULL
      ), new_streaks (id) AS (
        SELECT day_streaks.id
        FROM day_streaks
        LEFT JOIN existing_streaks ON existing_streaks.id = day_streaks.id
        WHERE existing_streaks.id IS NULL
      ), ending_streaks (id) AS (
        SELECT existing_streaks.id
        FROM existing_streaks
        LEFT JOIN day_streaks ON existing_streaks.id = day_streaks.id
        WHERE day_streaks.id IS NULL
      ), extending_streaks (id, started_at) AS (
        SELECT existing_streaks.id, existing_streaks.started_at
        FROM existing_streaks
        JOIN day_streaks ON existing_streaks.id = day_streaks.id
      ),
      -- a bunch of mutations
      streak_insert AS (
        INSERT INTO "Streak" ("userId", "startedAt", created_at, updated_at)
        SELECT id, (now() AT TIME ZONE 'America/Chicago' - interval '1 day')::date, now_utc(), now_utc()
        FROM new_streaks
      ), user_update_new_streaks AS (
        UPDATE users SET streak = 1 FROM new_streaks WHERE new_streaks.id = users.id
      ), user_update_end_streaks AS (
        UPDATE users SET streak = NULL FROM ending_streaks WHERE ending_streaks.id = users.id
      ), user_update_extend_streaks AS (
        UPDATE users
        SET streak = (now() AT TIME ZONE 'America/Chicago')::date - extending_streaks.started_at
        FROM extending_streaks WHERE extending_streaks.id = users.id
      )
      UPDATE "Streak"
      SET "endedAt" = (now() AT TIME ZONE 'America/Chicago' - interval '1 day')::date, updated_at = now_utc()
      FROM ending_streaks
      WHERE ending_streaks.id = "Streak"."userId" AND "endedAt" IS NULL
      RETURNING "Streak".id, ending_streaks."id" AS "userId"`

    Promise.allSettled(
      endingStreaks.map(({ id, userId }) => {
        const index = id % LOST_BLURBS.length
        const blurb = LOST_BLURBS[index]
        return sendUserNotification(userId, {
          title: 'you lost your cowboy hat',
          body: blurb,
          tag: 'STREAK-LOST'
        }).catch(console.error)
      })
    )

    console.log('done computing streaks')
  }
}

export function checkStreak ({ models }) {
  return async function ({ data: { id } }) {
    console.log('checking streak', id)

    // if user is actively streaking skip
    let streak = await models.streak.findFirst({
      where: {
        userId: Number(id),
        endedAt: null
      }
    })

    if (streak) {
      console.log('done checking streak', id)
      return
    }

    [streak] = await models.$queryRaw`
      WITH streak_started (id) AS (
          SELECT "userId"
          FROM
          ((SELECT "userId", floor(sum("ItemAct".msats)/1000) as sats_spent
              FROM "ItemAct"
              WHERE (created_at AT TIME ZONE 'UTC' AT TIME ZONE 'America/Chicago')::date >= (now() AT TIME ZONE 'America/Chicago')::date
              AND "userId" = ${Number(id)}
              GROUP BY "userId")
          UNION ALL
          (SELECT "userId", sats as sats_spent
              FROM "Donation"
              WHERE (created_at AT TIME ZONE 'UTC' AT TIME ZONE 'America/Chicago')::date >= (now() AT TIME ZONE 'America/Chicago')::date
              AND "userId" = ${Number(id)}
          )) spending
            GROUP BY "userId"
            HAVING sum(sats_spent) >= ${STREAK_THRESHOLD}
      ), user_start_streak AS (
        UPDATE users SET streak = 0 FROM streak_started WHERE streak_started.id = users.id
      )
      INSERT INTO "Streak" ("userId", "startedAt", created_at, updated_at)
      SELECT id, (now() AT TIME ZONE 'America/Chicago')::date, now_utc(), now_utc()
      FROM streak_started
      RETURNING "Streak".id`

    console.log('done checking streak', id)

    if (!streak) return

    // new streak started for user
    const index = streak.id % FOUND_BLURBS.length
    const blurb = FOUND_BLURBS[index]
    sendUserNotification(id, {
      title: 'you found a cowboy hat',
      body: blurb,
      tag: 'STREAK-FOUND'
    }).catch(console.error)
  }
}