From e108952e111c2f47ce808bb99382c5f7110affaf Mon Sep 17 00:00:00 2001
From: Aditya Hegde <adityahegderocks@gmail.com>
Date: Fri, 22 Mar 2024 14:14:50 +0530
Subject: [PATCH] fix: disable bookmark for anonymous users (#4406)

* Disable bookmark for anonymous users

* fix: lint

---------

Co-authored-by: Alexander Thor <alec.karlsson@gmail.com>
---
 web-admin/src/features/bookmarks/selectors.ts |  8 +++--
 ...=> DashboardBookmarksStateProvider.svelte} |  2 +-
 .../navigation/TopNavigationBar.svelte        |  4 ++-
 .../[project]/[dashboard]/+page.svelte        | 30 ++++++++++++++-----
 4 files changed, 32 insertions(+), 12 deletions(-)
 rename web-admin/src/features/dashboards/{DashboardAdminStateProvider.svelte => DashboardBookmarksStateProvider.svelte} (95%)

diff --git a/web-admin/src/features/bookmarks/selectors.ts b/web-admin/src/features/bookmarks/selectors.ts
index f3e2cd51455..f0d372311b2 100644
--- a/web-admin/src/features/bookmarks/selectors.ts
+++ b/web-admin/src/features/bookmarks/selectors.ts
@@ -1,4 +1,5 @@
 import {
+  createAdminServiceGetCurrentUser,
   createAdminServiceListBookmarks,
   type V1Bookmark,
 } from "@rilldata/web-admin/client";
@@ -46,8 +47,9 @@ export function getBookmarks(
       useProjectId(orgName, projectName),
       useMetricsView(instanceId, metricsViewName),
       createQueryServiceMetricsViewSchema(instanceId, metricsViewName),
+      createAdminServiceGetCurrentUser(),
     ],
-    ([projectId, metricsViewResp, schemaResp], set) =>
+    ([projectId, metricsViewResp, schemaResp, userResp], set) =>
       createAdminServiceListBookmarks(
         {
           projectId: projectId.data,
@@ -60,7 +62,9 @@ export function getBookmarks(
               !!projectId?.data &&
               !!metricsViewName &&
               !metricsViewResp.isFetching &&
-              !schemaResp.isFetching,
+              !schemaResp.isFetching &&
+              userResp.isSuccess &&
+              !!userResp.data.user,
             select: (resp) => {
               const bookmarks: Bookmarks = {
                 home: undefined,
diff --git a/web-admin/src/features/dashboards/DashboardAdminStateProvider.svelte b/web-admin/src/features/dashboards/DashboardBookmarksStateProvider.svelte
similarity index 95%
rename from web-admin/src/features/dashboards/DashboardAdminStateProvider.svelte
rename to web-admin/src/features/dashboards/DashboardBookmarksStateProvider.svelte
index d02546cddad..e6670b124ae 100644
--- a/web-admin/src/features/dashboards/DashboardAdminStateProvider.svelte
+++ b/web-admin/src/features/dashboards/DashboardBookmarksStateProvider.svelte
@@ -21,7 +21,7 @@
     metricViewName,
   );
 
-  const dashboardStoreReady = createDashboardStateSync(
+  $: dashboardStoreReady = createDashboardStateSync(
     getStateManagers(),
     homeBookmark,
   );
diff --git a/web-admin/src/features/navigation/TopNavigationBar.svelte b/web-admin/src/features/navigation/TopNavigationBar.svelte
index cfb1ca29d94..3fbb33d75b0 100644
--- a/web-admin/src/features/navigation/TopNavigationBar.svelte
+++ b/web-admin/src/features/navigation/TopNavigationBar.svelte
@@ -60,7 +60,9 @@
     {/if}
     {#if onDashboardPage}
       <LastRefreshedDate {dashboard} />
-      <Bookmarks />
+      {#if $user.isSuccess && $user.data.user}
+        <Bookmarks />
+      {/if}
       <ShareDashboardButton />
     {/if}
     {#if $user.isSuccess}
diff --git a/web-admin/src/routes/[organization]/[project]/[dashboard]/+page.svelte b/web-admin/src/routes/[organization]/[project]/[dashboard]/+page.svelte
index bfe35465756..318df516b48 100644
--- a/web-admin/src/routes/[organization]/[project]/[dashboard]/+page.svelte
+++ b/web-admin/src/routes/[organization]/[project]/[dashboard]/+page.svelte
@@ -1,10 +1,12 @@
 <script lang="ts">
   import { page } from "$app/stores";
   import {
+    createAdminServiceGetCurrentUser,
     createAdminServiceGetProject,
     V1DeploymentStatus,
   } from "@rilldata/web-admin/client";
-  import DashboardAdminStateProvider from "@rilldata/web-admin/features/dashboards/DashboardAdminStateProvider.svelte";
+  import DashboardBookmarksStateProvider from "@rilldata/web-admin/features/dashboards/DashboardBookmarksStateProvider.svelte";
+  import DashboardStateProvider from "@rilldata/web-common/features/dashboards/stores/DashboardStateProvider.svelte";
   import { getDashboardsForProject } from "@rilldata/web-admin/features/dashboards/listing/selectors";
   import { invalidateDashboardsQueries } from "@rilldata/web-admin/features/projects/invalidations";
   import ProjectErrored from "@rilldata/web-admin/features/projects/ProjectErrored.svelte";
@@ -29,6 +31,8 @@
   $: projectName = $page.params.project;
   $: dashboardName = $page.params.dashboard;
 
+  const user = createAdminServiceGetCurrentUser();
+
   $: project = createAdminServiceGetProject(orgName, projectName);
 
   $: projectDeploymentStatus = useProjectDeploymentStatus(orgName, projectName); // polls
@@ -102,13 +106,23 @@
   {:else}
     {#key dashboardName}
       <StateManagersProvider metricsViewName={dashboardName}>
-        <DashboardAdminStateProvider metricViewName={dashboardName}>
-          <DashboardURLStateProvider metricViewName={dashboardName}>
-            <DashboardThemeProvider>
-              <Dashboard metricViewName={dashboardName} leftMargin={"48px"} />
-            </DashboardThemeProvider>
-          </DashboardURLStateProvider>
-        </DashboardAdminStateProvider>
+        {#if $user.isSuccess && $user.data.user}
+          <DashboardBookmarksStateProvider metricViewName={dashboardName}>
+            <DashboardURLStateProvider metricViewName={dashboardName}>
+              <DashboardThemeProvider>
+                <Dashboard metricViewName={dashboardName} leftMargin={"48px"} />
+              </DashboardThemeProvider>
+            </DashboardURLStateProvider>
+          </DashboardBookmarksStateProvider>
+        {:else}
+          <DashboardStateProvider metricViewName={dashboardName}>
+            <DashboardURLStateProvider metricViewName={dashboardName}>
+              <DashboardThemeProvider>
+                <Dashboard metricViewName={dashboardName} leftMargin={"48px"} />
+              </DashboardThemeProvider>
+            </DashboardURLStateProvider>
+          </DashboardStateProvider>
+        {/if}
       </StateManagersProvider>
     {/key}
   {/if}