runNotificationFunctions.ts

import axios from "axios"
import * as admin from "firebase-admin"
import { FunctionName } from "functions/src"
import { uniq } from "lodash"
import { z } from "zod"
import { Script } from "./types"
import { performance } from "perf_hooks"

admin.initializeApp()
const db = admin.firestore()

const notificationFunctions: {
  [K in FunctionName]?: K
} = {
  httpsDeliverNotifications: "httpsDeliverNotifications",
  httpsDeliverAdminNotifications: "httpsDeliverAdminNotifications",
  httpsCleanupNotifications: "httpsCleanupNotifications"
}

const Args = z.object({
  interval: z.number().default(5),
  targets: z
    .string()
    .transform(s => {
      const t = s
        .split(",")
        .map(s => s.trim())
        .map(name => {
          const s = notificationFunctions[name as FunctionName]
          if (!s) throw Error(`Invalid function ${name}`)
          return s
        })
      return uniq(t)
    })
    .default(Object.keys(notificationFunctions).join(",")) as z.ZodType<
    Array<FunctionName>,
    any,
    string
  >
})

export const script: Script = async ({ env, args }) => {
  if (env !== "local") throw Error("only local supported")

  const { interval, targets } = Args.parse(args)
  const intervalMs = interval * 1e3

  // Fetch all events
  const events = await db.collection("events").get()

  // Loop through the documents in the events collection
  for (const event of events.docs) {
    // Extract the id of the event
    const topicEventId = event.id
    console.log("Topic event id", topicEventId)
    for (const target of targets) {
      console.log("Running", target)
      const start = performance.now()
      await axios.post(
        `http://localhost:5001/demo-dtp/us-central1/${target}`, // modified line
        {
          topicEventId: topicEventId
        }
      )
      const remaining = Math.max(0, intervalMs - (performance.now() - start))
      if (remaining) {
        console.log(`pausing ${(remaining * 1e-3).toFixed(1)} s`)
        await new Promise(r => setTimeout(r, remaining))
      }
    }
  }
}