DC-4894 Move PostHog to worker

create-db-worker/src/analytics.ts

@@ -0,0 +1,60 @@
+class EventCaptureError extends Error {
+	constructor(
+		public readonly event: string,
+		public readonly statusCode: number,
+		public readonly statusText: string,
+	) {
+		super(`Failed to submit PostHog event '${event}': ${statusCode} ${statusText}`);
+	}
+}
+
+interface AnalyticsProperties {
+	[key: string]: any;
+}
+
+class PosthogEventCapture {
+	constructor(private env: { POSTHOG_API_HOST?: string; POSTHOG_API_KEY?: string }) {}
+
+	async capture(eventName: string, properties: AnalyticsProperties = {}) {
+		const host = this.env.POSTHOG_API_HOST?.replace(/\/+$/, '');
+		const key = this.env.POSTHOG_API_KEY;
+
+		if (!host || !key) {
+			return;
+		}
+
+		const POSTHOG_CAPTURE_URL = `${host}/capture`;
+		const POSTHOG_KEY = key;
+
+		const payload = {
+			api_key: POSTHOG_KEY,
+			event: eventName,
+			distinct_id: crypto.randomUUID(),
+			properties: {
+				$process_person_profile: false,
+				...properties,
+			},
+		};
+
+		try {
+			const response = await fetch(POSTHOG_CAPTURE_URL, {
+				method: 'POST',
+				headers: {
+					'Content-Type': 'application/json',
+				},
+				body: JSON.stringify(payload),
+			});
+
+			if (!response.ok) {
+				throw new EventCaptureError(eventName, response.status, response.statusText);
+			}
+
+			console.log(`${eventName}: Success`);
+		} catch (error) {
+			console.error(`${eventName}: Failed - ${error instanceof Error ? error.message : String(error)}`);
+			throw error;
+		}
+	}
+}
+
+export { PosthogEventCapture, EventCaptureError };

create-db-worker/src/index.ts

@@ -1,16 +1,20 @@
 import DeleteDbWorkflow from './delete-workflow';
-
+import { PosthogEventCapture } from './analytics';
 interface Env {
 	INTEGRATION_TOKEN: string;
 	DELETE_DB_WORKFLOW: Workflow;
 	CREATE_DB_RATE_LIMITER: RateLimit;
 	CREATE_DB_DATASET: AnalyticsEngineDataset;
+	POSTHOG_API_KEY?: string;
+	POSTHOG_API_HOST?: string;
 }
 
 export { DeleteDbWorkflow };
 
 export default {
 	async fetch(request: Request, env: Env, ctx: ExecutionContext): Promise<Response> {
+		const analytics = new PosthogEventCapture(env);
+
 		// --- Rate limiting ---
 		const { success } = await env.CREATE_DB_RATE_LIMITER.limit({ key: request.url });
 
@@ -56,16 +60,44 @@ export default {
 			});
 		}
 
+		// --- Analytics endpoint ---
+		if (url.pathname === '/analytics' && request.method === 'POST') {
+			let body: any = {};
+			try {
+				body = await request.json();
+			} catch {
+				return new Response('Invalid JSON body', { status: 400 });
+			}
+
+			const { eventName, properties } = body;
+			if (!eventName) {
+				return new Response('Missing eventName in request body', { status: 400 });
+			}
+
+			if (!env.POSTHOG_API_HOST || !env.POSTHOG_API_KEY) {
+				return new Response(JSON.stringify({ status: 'disabled' }), {
+					status: 204,
+					headers: { 'Content-Type': 'application/json' },
+				});
+			}
+
+			ctx.waitUntil(analytics.capture(eventName, properties || {}));
+			return new Response(JSON.stringify({ status: 'queued', event: eventName }), {
+				status: 202,
+				headers: { 'Content-Type': 'application/json' },
+			});
+		}
+
 		// --- Create new project ---
 		if (url.pathname === '/create' && request.method === 'POST') {
-			let body: { region?: string; name?: string } = {};
+			let body: { region?: string; name?: string; analytics?: { eventName?: string; properties?: any } } = {};
 			try {
 				body = await request.json();
 			} catch {
 				return new Response('Invalid JSON body', { status: 400 });
 			}
 
-			const { region, name } = body;
+			const { region, name, analytics: analyticsData } = body;
 			if (!region || !name) {
 				return new Response('Missing region or name in request body', { status: 400 });
 			}
@@ -96,7 +128,13 @@ export default {
 						indexes: ['create_db'],
 					});
 
-					await Promise.all([workflowPromise, analyticsPromise]);
+					const posthogPromise = analyticsData?.eventName
+						? analytics
+								.capture(analyticsData.eventName, analyticsData.properties || {})
+								.catch((e) => console.error('Error sending PostHog analytics:', e))
+						: Promise.resolve();
+
+					await Promise.all([workflowPromise, analyticsPromise, posthogPromise]);
 				} catch (e) {
 					console.error('Error in background tasks:', e);
 				}

create-db-worker/wrangler.jsonc

@@ -4,6 +4,7 @@
 	"main": "src/index.ts",
 	"account_id": "16b32bbb36161aca01a6357a37bc453e",
 	"compatibility_date": "2025-06-27",
+	"keep_vars": true,
 	"observability": {
 		"enabled": true,
 	},

create-db/index.js

@@ -9,12 +9,29 @@ dotenv.config();
 import { select, spinner, intro, outro, log, cancel } from "@clack/prompts";
 import chalk from "chalk";
 import terminalLink from "terminal-link";
-import { analytics } from "./analytics.js";
 
 const CREATE_DB_WORKER_URL =
-  process.env.CREATE_DB_WORKER_URL || "https://create-db-temp.prisma.io";
+  process.env.CREATE_DB_WORKER_URL.replace(/\/+$/, "") ||
+  "https://create-db-temp.prisma.io";
 const CLAIM_DB_WORKER_URL =
-  process.env.CLAIM_DB_WORKER_URL || "https://create-db.prisma.io";
+  process.env.CLAIM_DB_WORKER_URL.replace(/\/+$/, "") ||
+  "https://create-db.prisma.io";
+
+async function sendAnalyticsToWorker(eventName, properties) {
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), 2000);
+  try {
+    await fetch(`${CREATE_DB_WORKER_URL}/analytics`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ eventName, properties }),
+      signal: controller.signal,
+    });
+  } catch (error) {
+  } finally {
+    clearTimeout(timer);
+  }
+}
 
 async function detectUserLocation() {
   try {
@@ -43,7 +60,6 @@ async function detectUserLocation() {
   }
 }
 
-// Region coordinates (latitude, longitude)
 const REGION_COORDINATES = {
   "ap-southeast-1": { lat: 1.3521, lng: 103.8198 }, // Singapore
   "ap-northeast-1": { lat: 35.6762, lng: 139.6503 }, // Tokyo
@@ -360,14 +376,12 @@ async function promptForRegion(defaultRegion, userAgent) {
   }
 
   try {
-    const analyticsProps = {
+    await sendAnalyticsToWorker("create_db:region_selected", {
       command: CLI_NAME,
       region: region,
       "selection-method": "interactive",
       "user-agent": userAgent,
-    };
-
-    await analytics.capture("create_db:region_selected", analyticsProps);
+    });
   } catch (error) {}
 
   return region;
@@ -407,18 +421,13 @@ async function createDatabase(name, region, userAgent, returnJson = false) {
     }
 
     try {
-      const analyticsProps = {
+      await sendAnalyticsToWorker("create_db:database_creation_failed", {
         command: CLI_NAME,
         region: region,
         "error-type": "rate_limit",
         "status-code": 429,
         "user-agent": userAgent,
-      };
-
-      await analytics.capture(
-        "create_db:database_creation_failed",
-        analyticsProps
-      );
+      });
     } catch (error) {}
 
     process.exit(1);
@@ -442,18 +451,13 @@ async function createDatabase(name, region, userAgent, returnJson = false) {
       s.stop("Unexpected response from create service.");
     }
     try {
-      const analyticsProps = {
+      await sendAnalyticsToWorker("create_db:database_creation_failed", {
         command: CLI_NAME,
         region,
         "error-type": "invalid_json",
         "status-code": resp.status,
         "user-agent": userAgent,
-      };
-
-      await analytics.capture(
-        "create_db:database_creation_failed",
-        analyticsProps
-      );
+      });
     } catch (error) {}
     process.exit(1);
   }
@@ -518,18 +522,13 @@ async function createDatabase(name, region, userAgent, returnJson = false) {
     }
 
     try {
-      const analyticsProps = {
+      await sendAnalyticsToWorker("create_db:database_creation_failed", {
         command: CLI_NAME,
         region: region,
         "error-type": "api_error",
         "error-message": result.error.message,
         "user-agent": userAgent,
-      };
-
-      await analytics.capture(
-        "create_db:database_creation_failed",
-        analyticsProps
-      );
+      });
     } catch (error) {}
     process.exit(1);
   }
@@ -581,6 +580,14 @@ async function createDatabase(name, region, userAgent, returnJson = false) {
       )
     )
   );
+
+  try {
+    await sendAnalyticsToWorker("create_db:database_created", {
+      command: CLI_NAME,
+      region,
+      utm_source: CLI_NAME,
+    });
+  } catch {}
 }
 
 async function main() {
@@ -596,7 +603,7 @@ async function main() {
     }
 
     try {
-      const analyticsProps = {
+      await sendAnalyticsToWorker("create_db:cli_command_ran", {
         command: CLI_NAME,
         "full-command": `${CLI_NAME} ${rawArgs.join(" ")}`.trim(),
         "has-region-flag":
@@ -611,9 +618,7 @@ async function main() {
         platform: process.platform,
         arch: process.arch,
         "user-agent": userAgent,
-      };
-
-      await analytics.capture("create_db:cli_command_ran", analyticsProps);
+      });
     } catch (error) {
       console.error("Error:", error.message);
     }
@@ -640,14 +645,12 @@ async function main() {
       region = flags.region;
 
       try {
-        const analyticsProps = {
+        await sendAnalyticsToWorker("create_db:region_selected", {
           command: CLI_NAME,
           region: region,
           "selection-method": "flag",
           "user-agent": userAgent,
-        };
-
-        await analytics.capture("create_db:region_selected", analyticsProps);
+        });
       } catch (error) {}
     }