From 6d124e7bb0bd7b566d74e9b1101c625db526a742 Mon Sep 17 00:00:00 2001
From: Francis Cao <franciscao@gmail.com>
Date: Tue, 22 Jul 2025 11:10:37 -0700
Subject: [PATCH] fix export report route

---
 .../(main)/reports/[reportId]/ReportBody.tsx  |  4 +++-
 .../api/websites/[websiteId]/export/route.ts  | 20 +++++++++----------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/src/app/(main)/reports/[reportId]/ReportBody.tsx b/src/app/(main)/reports/[reportId]/ReportBody.tsx
index 7b47c848..fe2f8c91 100644
--- a/src/app/(main)/reports/[reportId]/ReportBody.tsx
+++ b/src/app/(main)/reports/[reportId]/ReportBody.tsx
@@ -12,7 +12,9 @@ export function ReportBody({ children }) {
 
   return (
     <div className={styles.body}>
-      <DownloadButton filename={report.name} data={report.data} />
+      {report.type !== 'revenue' && report.type !== 'attribution' && (
+        <DownloadButton filename={report.name} data={report.data} />
+      )}
       {children}
     </div>
   );
diff --git a/src/app/api/websites/[websiteId]/export/route.ts b/src/app/api/websites/[websiteId]/export/route.ts
index d8b403b4..534fd229 100644
--- a/src/app/api/websites/[websiteId]/export/route.ts
+++ b/src/app/api/websites/[websiteId]/export/route.ts
@@ -5,7 +5,7 @@ import { getRequestFilters, parseRequest } from '@/lib/request';
 import { unauthorized, json } from '@/lib/response';
 import { canViewWebsite } from '@/lib/auth';
 import { pagingParams } from '@/lib/schema';
-import { getEventMetrics, getWebsiteEvents } from '@/queries';
+import { getEventMetrics, getPageviewMetrics, getSessionMetrics } from '@/queries';
 
 export async function GET(
   request: Request,
@@ -32,8 +32,6 @@ export async function GET(
 
   const startDate = new Date(+startAt);
   const endDate = new Date(+endAt);
-  const limit = 10;
-  const offset = 0;
 
   const filters = {
     ...(await getRequestFilters(query)),
@@ -42,13 +40,13 @@ export async function GET(
   };
 
   const [events, pages, referrers, browsers, os, devices, countries] = await Promise.all([
-    getWebsiteEvents(websiteId, { startDate, endDate }, query),
-    getEventMetrics(websiteId, 'url', filters, limit, offset),
-    getEventMetrics(websiteId, 'referrer', filters, limit, offset),
-    getEventMetrics(websiteId, 'browser', filters, limit, offset),
-    getEventMetrics(websiteId, 'os', filters, limit, offset),
-    getEventMetrics(websiteId, 'device', filters, limit, offset),
-    getEventMetrics(websiteId, 'country', filters, limit, offset),
+    getEventMetrics(websiteId, 'event', filters),
+    getPageviewMetrics(websiteId, 'url', filters),
+    getPageviewMetrics(websiteId, 'referrer', filters),
+    getSessionMetrics(websiteId, 'browser', filters),
+    getSessionMetrics(websiteId, 'os', filters),
+    getSessionMetrics(websiteId, 'device', filters),
+    getSessionMetrics(websiteId, 'country', filters),
   ]);
 
   const zip = new JSZip();
@@ -60,7 +58,7 @@ export async function GET(
     });
   };
 
-  zip.file('events.csv', parse(events?.data));
+  zip.file('events.csv', parse(events));
   zip.file('pages.csv', parse(pages));
   zip.file('referrers.csv', parse(referrers));
   zip.file('browsers.csv', parse(browsers));