From 7bfc956964a1440c3b75373c597e17687f745edb Mon Sep 17 00:00:00 2001
From: Rhys Sullivan <39114868+RhysSullivan@users.noreply.github.com>
Date: Sat, 2 Dec 2023 19:39:32 -0800
Subject: [PATCH] update generator

---
 packages/generator/src/bin.ts           |   2 +-
 packages/generator/src/core/generate.ts | 148 ++++++++++++++++++++++--
 2 files changed, 142 insertions(+), 8 deletions(-)

diff --git a/packages/generator/src/bin.ts b/packages/generator/src/bin.ts
index 831fe18..24ef4cb 100644
--- a/packages/generator/src/bin.ts
+++ b/packages/generator/src/bin.ts
@@ -82,7 +82,7 @@ function posthogPropertyTypeToGeneric(
 
 void fetchFromPosthog("event_definitions").then(async (data) => {
   const events: AnalyticsEvent[] = [];
-  for await (const event of data.results.slice(0, 2)) {
+  for await (const event of data.results.slice(0, 20)) {
     const definitions = await fetchFromPosthog("property_definitions", {
       queryParams: {
         event_names: [event.name],
diff --git a/packages/generator/src/core/generate.ts b/packages/generator/src/core/generate.ts
index a4d0ba3..d54d873 100644
--- a/packages/generator/src/core/generate.ts
+++ b/packages/generator/src/core/generate.ts
@@ -9,16 +9,150 @@ export async function generateTypeScriptFile(events: AnalyticsEvent[]) {
   }, {} as Record<string, AnalyticsEvent>);
   const typeScriptFileContent = `
     const events = ${JSON.stringify(eventMap, null, 2)} as const;
-    type AnalyticsEvents = typeof events;
 
-type InsightOptions<T extends AnalyticsEvents> = {
-  filters: [];
-  breakdownBy: T[number]["name"];
+
+    
+type AnalyticsEvents = typeof events;
+type AnalyticsEventNames = keyof AnalyticsEvents;
+
+type AnalyticsEvent = AnalyticsEvents[keyof AnalyticsEvents];
+
+type SamplingOptions =
+  | "TotalCount"
+  | "CountPerUser"
+  | "UniqueUsers"
+  | "WeeklyActiveUsers"
+  | "MonthlyActiveUsers"
+  | "UniqueSessions";
+
+type Filter<T extends AnalyticsEventNames> = {
+  name: AnalyticsEvents[T]["properties"][number]["name"];
+  compare:
+    | "equals"
+    | "does_not_equal"
+    | "contains"
+    | "does_not_contain"
+    | "matches_regex"
+    | "does_not_match_regex"
+    | "greater_than"
+    | "less_than"
+    | "is_set"
+    | "is_not_set";
+  value: string | number | boolean;
+};
+
+type SeriesEntry<T extends AnalyticsEventNames> = {
+  name: T;
+  sampling?: SamplingOptions;
+  where?: Filter<T> | Filter<T>[];
+};
+
+type InsightOptions<T extends AnalyticsEventNames> = {
+  events: SeriesEntry<T> | SeriesEntry<T>[];
+  filters?: [];
+  breakdownBy?: AnalyticsEvents[T]["properties"][number]["name"];
+};
+
+type ExecuteOptions<T extends AnalyticsEventNames> = {
+  groupBy?: "day" | "hour" | "week" | "month";
+  filterMatch: "all" | "any";
+  type:
+    | "line"
+    | "bar"
+    | "area"
+    | "cumulative-line"
+    | "number"
+    | "pie"
+    | "bar-total"
+    | "table"
+    | "world";
+  breakdownBy?: AnalyticsEvents[T]["properties"][number]["name"];
 };
 
-function fetchInsight<T extends AnalyticsEvents[number]["name"]>(
-  eventName: T
-) {}
+function addFilter<T extends AnalyticsEventNames>(
+  series: SeriesEntry<T>,
+  existingFilters?: Filter<T> | Filter<T>[]
+) {
+  const existing = Array.isArray(existingFilters)
+    ? existingFilters
+    : existingFilters
+    ? [existingFilters]
+    : [];
+  return {
+    addFilter: (filter: Filter<T>) => addFilter(series, [...existing, filter]),
+    execute: (opts: ExecuteOptions<T>) => execute([...existing, series], opts),
+  };
+}
+
+function execute<T extends AnalyticsEventNames>(
+  series: SeriesEntry<T>[],
+  opts: ExecuteOptions<T>
+) {
+  console.log("series", series);
+  return [
+    {
+      hello: "world",
+    },
+  ];
+}
+
+function addSeries<
+  Current extends AnalyticsEventNames,
+  Existing extends AnalyticsEventNames,
+  Next extends AnalyticsEventNames,
+  ExecOpts extends ExecuteOptions<Current | Existing | Next>,
+  FilterOpts extends Filter<Current | Existing | Next>
+>(
+  series: SeriesEntry<Current>,
+  existingSeries?: SeriesEntry<Existing> | SeriesEntry<Existing>[]
+) {
+  const existing = Array.isArray(existingSeries)
+    ? existingSeries
+    : existingSeries
+    ? [existingSeries]
+    : [];
+  return {
+    addSeries: (series: SeriesEntry<Next>) =>
+      addSeries(series, [...existing, series]),
+    addFilter: (opts: FilterOpts) => addFilter(series, opts),
+    execute: (opts: ExecOpts) => execute([...existing, series], opts),
+  };
+}
+
+function buildQuery() {
+  return {
+    addSeries,
+  };
+}
+
+const data = buildQuery()
+  .addSeries({
+    name: "User Grant Consent",
+    sampling: "TotalCount",
+    where: {
+      name: "Answer Overflow Account Id",
+      compare: "equals",
+      value: 123,
+    },
+  })
+  .addSeries({
+    name: "$autocapture",
+    where: {
+      name: "Answer Overflow Account Id",
+      compare: "equals",
+      value: 123,
+    },
+  }).addFilter({
+name: "Server"
+  })
+  .execute({
+    breakdownBy: "Answer Overflow Account Id",
+    type: "line",
+    filterMatch: "all",
+  });
+
+console.log(data);
+
     `;
 
   // write file to dist/events.ts