From aeeaa523f9bcf27d421c13b02a606dfb4d4c22fb Mon Sep 17 00:00:00 2001
From: Liam <liam@fascinated.cc>
Date: Sat, 21 Oct 2023 22:25:41 +0100
Subject: [PATCH] make "You" button in navbar show up w/o refreshing page

---
 src/app/player/[id]/page.tsx           | 14 ++++++++---
 src/components/Navbar.tsx              |  2 +-
 src/store/settingsStore.ts             | 34 +++++++++++++-------------
 src/utils/{useState.ts => useStore.ts} |  5 +++-
 4 files changed, 33 insertions(+), 22 deletions(-)
 rename src/utils/{useState.ts => useStore.ts} (81%)

diff --git a/src/app/player/[id]/page.tsx b/src/app/player/[id]/page.tsx
index 16a714e..68a55ea 100644
--- a/src/app/player/[id]/page.tsx
+++ b/src/app/player/[id]/page.tsx
@@ -13,6 +13,7 @@ import { ScoresaberPlayerScore } from "@/schemas/scoresaber/playerScore";
 import { useSettingsStore } from "@/store/settingsStore";
 import { formatNumber } from "@/utils/number";
 import { fetchScores, getPlayerInfo } from "@/utils/scoresaber/api";
+import useStore from "@/utils/useStore";
 import {
   ClockIcon,
   GlobeAsiaAustraliaIcon,
@@ -59,7 +60,13 @@ const sortTypes: { [key: string]: SortType } = {
 const DEFAULT_SORT_TYPE = sortTypes.top;
 
 export default function Player({ params }: { params: { id: string } }) {
-  const settingsStore = useSettingsStore();
+  const settingsStore = useStore(useSettingsStore, (state) => {
+    return {
+      userId: state.userId,
+      setUserId: state.setUserId,
+      refreshProfile: state.refreshProfile,
+    };
+  });
   const searchParams = useSearchParams();
   const router = useRouter();
 
@@ -135,7 +142,8 @@ export default function Player({ params }: { params: { id: string } }) {
   );
 
   function claimProfile() {
-    settingsStore.setUserId(params.id);
+    settingsStore?.setUserId(params.id);
+    settingsStore?.refreshProfile();
     toast.success("Successfully claimed profile");
   }
 
@@ -206,7 +214,7 @@ export default function Player({ params }: { params: { id: string } }) {
 
               {/* Settings Buttons */}
               <div className="absolute right-3 top-16 flex justify-end md:relative md:right-3 md:top-0 md:mt-2 md:justify-center">
-                {settingsStore.userId !== params.id && (
+                {settingsStore?.userId !== params.id && (
                   <button>
                     <HomeIcon
                       title="Set as your Profile"
diff --git a/src/components/Navbar.tsx b/src/components/Navbar.tsx
index 96392db..89a3ab1 100644
--- a/src/components/Navbar.tsx
+++ b/src/components/Navbar.tsx
@@ -1,7 +1,7 @@
 "use client";
 
 import { useSettingsStore } from "@/store/settingsStore";
-import { useStore } from "@/utils/useState";
+import useStore from "@/utils/useStore";
 import {
   CogIcon,
   MagnifyingGlassIcon,
diff --git a/src/store/settingsStore.ts b/src/store/settingsStore.ts
index 57b945c..4399ba7 100644
--- a/src/store/settingsStore.ts
+++ b/src/store/settingsStore.ts
@@ -2,7 +2,7 @@
 
 import { getPlayerInfo } from "@/utils/scoresaber/api";
 import { create } from "zustand";
-import { persist } from "zustand/middleware";
+import { createJSONStorage, persist } from "zustand/middleware";
 
 interface SettingsStore {
   userId: string | undefined;
@@ -10,6 +10,7 @@ interface SettingsStore {
 
   setUserId: (userId: string) => void;
   setProfilePicture: (profilePicture: string) => void;
+  refreshProfile: () => void;
 }
 
 export const useSettingsStore = create<SettingsStore>()(
@@ -22,26 +23,25 @@ export const useSettingsStore = create<SettingsStore>()(
         set({ userId });
       },
       setProfilePicture: (profilePicture: string) => set({ profilePicture }),
+      async refreshProfile() {
+        const id = useSettingsStore.getState().userId;
+        if (!id) return;
+
+        const profile = await getPlayerInfo(id);
+        if (profile == undefined || profile == null) return;
+
+        useSettingsStore.setState({
+          userId: profile.id,
+          profilePicture: profile.profilePicture,
+        });
+        console.log("Updated profile:", profile.id);
+      },
     }),
     {
       name: "settings",
-      getStorage: () => localStorage,
+      storage: createJSONStorage(() => localStorage),
     },
   ),
 );
 
-async function refreshProfile() {
-  const id = useSettingsStore.getState().userId;
-  if (!id) return;
-
-  const profile = await getPlayerInfo(id);
-  if (profile == undefined || profile == null) return;
-
-  useSettingsStore.setState({
-    userId: profile.id,
-    profilePicture: profile.profilePicture,
-  });
-  console.log("Updated profile:", profile.id);
-}
-
-refreshProfile();
+useSettingsStore.getState().refreshProfile();
diff --git a/src/utils/useState.ts b/src/utils/useStore.ts
similarity index 81%
rename from src/utils/useState.ts
rename to src/utils/useStore.ts
index 04ab104..ab3ed36 100644
--- a/src/utils/useState.ts
+++ b/src/utils/useStore.ts
@@ -1,6 +1,7 @@
+// useStore.ts
 import { useEffect, useState } from "react";
 
-export const useStore = <T, F>(
+const useStore = <T, F>(
   store: (callback: (state: T) => unknown) => unknown,
   callback: (state: T) => F,
 ) => {
@@ -13,3 +14,5 @@ export const useStore = <T, F>(
 
   return data;
 };
+
+export default useStore;