1 files changed, 65 insertions, 0 deletions
diff --git a/scripts/seed/generators/revenue.ts b/scripts/seed/generators/revenue.ts
new file mode 100644
index 0000000..deea9e6
--- /dev/null
+++ b/scripts/seed/generators/revenue.ts
@@ -0,0 +1,65 @@
+import { uuid, randomFloat } from '../utils.js';
+import type { EventData } from './events.js';
+
+export interface RevenueConfig {
+  eventName: string;
+  minAmount: number;
+  maxAmount: number;
+  currency: string;
+  weight: number;
+}
+
+export interface RevenueData {
+  id: string;
+  websiteId: string;
+  sessionId: string;
+  eventId: string;
+  eventName: string;
+  currency: string;
+  revenue: number;
+  createdAt: Date;
+}
+
+export function generateRevenue(event: EventData, config: RevenueConfig): RevenueData | null {
+  if (event.eventName !== config.eventName) {
+    return null;
+  }
+
+  if (Math.random() > config.weight) {
+    return null;
+  }
+
+  const revenue = randomFloat(config.minAmount, config.maxAmount);
+
+  return {
+    id: uuid(),
+    websiteId: event.websiteId,
+    sessionId: event.sessionId,
+    eventId: event.id,
+    eventName: event.eventName!,
+    currency: config.currency,
+    revenue: Math.round(revenue * 100) / 100, // Round to 2 decimal places
+    createdAt: event.createdAt,
+  };
+}
+
+export function generateRevenueForEvents(
+  events: EventData[],
+  configs: RevenueConfig[],
+): RevenueData[] {
+  const revenueEntries: RevenueData[] = [];
+
+  for (const event of events) {
+    if (!event.eventName) continue;
+
+    for (const config of configs) {
+      const revenue = generateRevenue(event, config);
+      if (revenue) {
+        revenueEntries.push(revenue);
+        break; // Only one revenue per event
+      }
+    }
+  }
+
+  return revenueEntries;
+}