wip

packages/trpc/src/routers/chart.ts

@@ -22,11 +22,10 @@ import {
   getSelectPropertyKey,
   getSettingsForProject,
   onlyReportEvents,
+  sankeyService,
 } from '@openpanel/db';
 import {
   type IChartEvent,
-  zChartEvent,
-  zChartEventFilter,
   zChartInput,
   zChartSeries,
   zCriteria,
@@ -379,6 +378,38 @@ export const chartRouter = createTRPCRouter({
     };
   }),
 
+  sankey: protectedProcedure.input(zChartInput).query(async ({ input }) => {
+    const { timezone } = await getSettingsForProject(input.projectId);
+    const currentPeriod = getChartStartEndDate(input, timezone);
+
+    // Extract sankey options
+    const options = input.options;
+
+    if (!options || options.type !== 'sankey') {
+      throw new Error('Sankey options are required');
+    }
+
+    // Extract start/end events from series based on mode
+    const eventSeries = onlyReportEvents(input.series);
+
+    if (!eventSeries[0]) {
+      throw new Error('Start and end events are required');
+    }
+
+    return sankeyService.getSankey({
+      projectId: input.projectId,
+      startDate: currentPeriod.startDate,
+      endDate: currentPeriod.endDate,
+      steps: options.steps,
+      mode: options.mode,
+      startEvent: eventSeries[0],
+      endEvent: eventSeries[1],
+      exclude: options.exclude || [],
+      include: options.include,
+      timezone,
+    });
+  }),
+
   chart: publicProcedure
     // .use(cacher)
     .input(zChartInput)

packages/trpc/src/routers/report.ts

@@ -59,6 +59,7 @@ export const reportRouter = createTRPCRouter({
           metric: report.metric === 'count' ? 'sum' : report.metric,
           funnelGroup: report.funnelGroup,
           funnelWindow: report.funnelWindow,
+          options: report.options,
         },
       });
     }),
@@ -104,6 +105,7 @@ export const reportRouter = createTRPCRouter({
           metric: report.metric === 'count' ? 'sum' : report.metric,
           funnelGroup: report.funnelGroup,
           funnelWindow: report.funnelWindow,
+          options: report.options,
         },
       });
     }),
@@ -175,6 +177,7 @@ export const reportRouter = createTRPCRouter({
           metric: report.metric,
           funnelGroup: report.funnelGroup,
           funnelWindow: report.funnelWindow,
+          options: report.options,
         },
       });
     }),