DC-4894 Move PostHog to worker

create-db-worker/src/analytics.ts

@@ -0,0 +1,53 @@
+class EventCaptureError extends Error {
+	constructor(event: string, status: string) {
+		super(`Failed to submit PostHog event '${event}': ${status}`);
+	}
+}
+
+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 POSTHOG_CAPTURE_URL = this.env.POSTHOG_API_HOST + '/capture';
+		const POSTHOG_KEY = this.env.POSTHOG_API_KEY;
+
+		console.log('Sending analytics to PostHog', eventName, properties);
+		console.log('POSTHOG_CAPTURE_URL', POSTHOG_CAPTURE_URL);
+		console.log('POSTHOG_KEY Set?', !!POSTHOG_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.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,52 @@ 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 });
+			}
+
+			try {
+				await analytics.capture(eventName, properties || {});
+				return new Response(JSON.stringify({ status: 'success', event: eventName }), {
+					status: 200,
+					headers: { 'Content-Type': 'application/json' },
+				});
+			} catch (error) {
+				console.error('Analytics error:', error);
+				return new Response(
+					JSON.stringify({
+						status: 'error',
+						event: eventName,
+						error: error instanceof Error ? error.message : String(error),
+					}),
+					{
+						status: 500,
+						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 +136,17 @@ export default {
 						indexes: ['create_db'],
 					});
 
-					await Promise.all([workflowPromise, analyticsPromise]);
+					// Handle PostHog analytics if provided
+					let posthogPromise = Promise.resolve();
+					if (analyticsData && analyticsData.eventName) {
+						try {
+							posthogPromise = analytics.capture(analyticsData.eventName, analyticsData.properties || {});
+						} catch (e) {
+							console.error('Error sending PostHog analytics:', e);
+						}
+					}
+
+					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

@@ -6,7 +6,27 @@ 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";
+
+async function sendAnalyticsToWorker(eventName, properties = {}) {
+  try {
+    const response = await fetch(`${CREATE_DB_WORKER_URL}/analytics`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ eventName, properties }),
+    });
+
+    if (!response.ok) {
+      throw new Error(
+        `Analytics request failed: ${response.status} ${response.statusText}`
+      );
+    }
+
+    const result = await response.json();
+    if (result.status === "success") {
+    } else {
+    }
+  } catch (error) {}
+}
 
 const CREATE_DB_WORKER_URL =
   process.env.CREATE_DB_WORKER_URL || "https://create-db-temp.prisma.io";
@@ -264,7 +284,7 @@ async function promptForRegion(defaultRegion) {
   }
 
   try {
-    await analytics.capture("create_db:region_selected", {
+    await sendAnalyticsToWorker("create_db:region_selected", {
       command: CLI_NAME,
       region: region,
       "selection-method": "interactive",
@@ -284,7 +304,19 @@ async function createDatabase(name, region, returnJson = false) {
   const resp = await fetch(`${CREATE_DB_WORKER_URL}/create`, {
     method: "POST",
     headers: { "Content-Type": "application/json" },
-    body: JSON.stringify({ region, name, utm_source: CLI_NAME }),
+    body: JSON.stringify({
+      region,
+      name,
+      utm_source: CLI_NAME,
+      analytics: {
+        eventName: "create_db:database_created",
+        properties: {
+          command: CLI_NAME,
+          region: region,
+          utm_source: CLI_NAME,
+        },
+      },
+    }),
   });
 
   if (resp.status === 429) {
@@ -304,7 +336,7 @@ async function createDatabase(name, region, returnJson = false) {
     }
 
     try {
-      await analytics.capture("create_db:database_creation_failed", {
+      await sendAnalyticsToWorker("create_db:database_creation_failed", {
         command: CLI_NAME,
         region: region,
         "error-type": "rate_limit",
@@ -333,13 +365,13 @@ async function createDatabase(name, region, returnJson = false) {
       s.stop("Unexpected response from create service.");
     }
     try {
-      await analytics.capture("create_db:database_creation_failed", {
+      await sendAnalyticsToWorker("create_db:database_creation_failed", {
         command: CLI_NAME,
         region,
         "error-type": "invalid_json",
         "status-code": resp.status,
       });
-    } catch {}
+    } catch (error) {}
     process.exit(1);
   }
 
@@ -397,7 +429,7 @@ async function createDatabase(name, region, returnJson = false) {
     }
 
     try {
-      await analytics.capture("create_db:database_creation_failed", {
+      await sendAnalyticsToWorker("create_db:database_creation_failed", {
         command: CLI_NAME,
         region: region,
         "error-type": "api_error",
@@ -460,7 +492,7 @@ async function main() {
   try {
     const rawArgs = process.argv.slice(2);
     try {
-      await analytics.capture("create_db:cli_command_ran", {
+      await sendAnalyticsToWorker("create_db:cli_command_ran", {
         command: CLI_NAME,
         "full-command": `${CLI_NAME} ${rawArgs.join(" ")}`.trim(),
         "has-region-flag":
@@ -499,7 +531,7 @@ async function main() {
       region = flags.region;
 
       try {
-        await analytics.capture("create_db:region_selected", {
+        await sendAnalyticsToWorker("create_db:region_selected", {
           command: CLI_NAME,
           region: region,
           "selection-method": "flag",