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,51 @@ export default {
 			});
 		}
 
+		// --- Analytics endpoint ---
+		if (url.pathname === '/analytics' && request.method === 'POST') {
+			type AnalyticsBody = { eventName?: string; properties?: Record<string, unknown> };
+
+			let body: AnalyticsBody = {};
+
+			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(null, { status: 204 });
+			}
+
+			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 } = {};
+			type CreateDbBody = {
+				region?: string;
+				name?: string;
+				analytics?: { eventName?: string; properties?: Record<string, unknown> };
+			};
+
+			let body: CreateDbBody = {};
+
 			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 +135,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,
 	},