From 90de8e8ff63eb212aee78229f3cc4b93fa207f4d Mon Sep 17 00:00:00 2001
From: Nate Kelley <nate@buster.so>
Date: Mon, 21 Apr 2025 12:32:17 -0600
Subject: [PATCH] Update MetricContainerHeaderButtons.tsx

---
 .../MetricContainerHeaderButtons.tsx          | 25 +++++++++++++++----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/MetricContainerHeaderButtons/MetricContainerHeaderButtons.tsx b/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/MetricContainerHeaderButtons/MetricContainerHeaderButtons.tsx
index ea5ab0ecd..3fa0b77f3 100644
--- a/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/MetricContainerHeaderButtons/MetricContainerHeaderButtons.tsx
+++ b/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/MetricContainerHeaderButtons/MetricContainerHeaderButtons.tsx
@@ -128,6 +128,16 @@ const EditSQLButton = React.memo(({ metricId }: { metricId: string }) => {
 
   const href = useMemo(() => {
     if (!isSelectedView) {
+      if (chatId) {
+        return createBusterRoute({
+          route: BusterRoutes.APP_CHAT_ID_METRIC_ID_RESULTS,
+          versionNumber: metricVersionNumber,
+          metricId,
+          secondaryView: 'sql-edit',
+          chatId
+        });
+      }
+
       return createBusterRoute({
         route: BusterRoutes.APP_METRIC_ID_RESULTS,
         versionNumber: metricVersionNumber,
@@ -136,6 +146,15 @@ const EditSQLButton = React.memo(({ metricId }: { metricId: string }) => {
       });
     }
 
+    if (chatId) {
+      return createBusterRoute({
+        route: BusterRoutes.APP_CHAT_ID_METRIC_ID_RESULTS,
+        versionNumber: metricVersionNumber,
+        metricId,
+        chatId
+      });
+    }
+
     return createBusterRoute({
       route: BusterRoutes.APP_METRIC_ID_RESULTS,
       versionNumber: metricVersionNumber,
@@ -143,11 +162,6 @@ const EditSQLButton = React.memo(({ metricId }: { metricId: string }) => {
     });
   }, [chatId, metricId, metricVersionNumber, isSelectedView]);
 
-  // const onClickButton = useMemoizedFn(() => {
-  //   const secondaryView = isSelectedView ? null : editableSecondaryView;
-  //   onSetFileView({ secondaryView, fileView: 'results' });
-  // });
-
   const onClickButton = useMemoizedFn(() => {
     onChangePage(href, { shallow: true });
   });
@@ -155,6 +169,7 @@ const EditSQLButton = React.memo(({ metricId }: { metricId: string }) => {
   return (
     <Link
       href={href}
+      prefetch={true}
       onClick={(e) => {
         e.preventDefault();
         e.stopPropagation();