chore: import upstream snapshot with attribution

2026-07-13 13:32:57 +08:00
commit cd420f9332
4811 changed files with 884702 additions and 0 deletions
@@ -0,0 +1,46 @@
+import { z } from "zod";
+import { singleton } from "~/utils/singleton";
+import type { ZodSubscriber } from "../utils/zodPubSub.server";
+import { ZodPubSub } from "../utils/zodPubSub.server";
+import { env } from "~/env.server";
+import { Gauge } from "prom-client";
+import { metricsRegister } from "~/metrics.server";
+
+const messageCatalog = {
+  WORKER_CREATED: z.object({
+    environmentId: z.string(),
+    environmentType: z.string(),
+    createdAt: z.coerce.date(),
+    taskCount: z.number(),
+    type: z.union([z.literal("local"), z.literal("deployed")]),
+  }),
+};
+
+export type ProjectSubscriber = ZodSubscriber<typeof messageCatalog>;
+
+export const projectPubSub = singleton("projectPubSub", initializeProjectPubSub);
+
+function initializeProjectPubSub() {
+  const pubSub = new ZodPubSub({
+    redis: {
+      port: env.PUBSUB_REDIS_PORT,
+      host: env.PUBSUB_REDIS_HOST,
+      username: env.PUBSUB_REDIS_USERNAME,
+      password: env.PUBSUB_REDIS_PASSWORD,
+      tlsDisabled: env.PUBSUB_REDIS_TLS_DISABLED === "true",
+      clusterMode: env.PUBSUB_REDIS_CLUSTER_MODE_ENABLED === "1",
+    },
+    schema: messageCatalog,
+  });
+
+  new Gauge({
+    name: "project_pub_sub_subscribers",
+    help: "Number of project pub sub subscribers",
+    collect() {
+      this.set(pubSub.subscriberCount);
+    },
+    registers: [metricsRegister],
+  });
+
+  return pubSub;
+}