refactor: made budget backward compatible

stakwork · Oct 17, 2023 · 68c3cec · 68c3cec
1 parent 432f6f7
commit 68c3cec
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 6 deletions.
diff --git a/src/components/AddNodeModal/index.tsx b/src/components/AddNodeModal/index.tsx
@@ -85,7 +85,7 @@ const handleSubmit = async (
   close: () => void,
   sourceType: string,
   successCallback: () => void,
-  setBudget: (value: number) => void,
+  setBudget: (value: number | null) => void,
 ): Promise<void> => {
   const body: { [index: string]: unknown } = {}
 

diff --git a/src/components/Stats/index.tsx b/src/components/Stats/index.tsx
@@ -37,7 +37,11 @@ export const Stats = () => {
   const [stats, setStats] = useState<TStats | null>(null)
   const [budget] = useUserStore((s) => [s.budget])
 
-  function formatBudget(value: number) {
+  function formatBudget(value: number | null) {
+    if (value === null) {
+      return '?'
+    }
+
     const stringBudget = value.toLocaleString()
 
     const splittedBudget = stringBudget.split(',')

diff --git a/src/stores/useUserStore/index.ts b/src/stores/useUserStore/index.ts
@@ -5,8 +5,8 @@ export type UserStore = {
   pubKey: string
   setIsAdmin: (val: boolean) => void
   setPubKey: (val: string) => void
-  budget: number
-  setBudget: (val: number) => void
+  budget: number | null
+  setBudget: (val: number | null) => void
 }
 
 const defaultData: Omit<UserStore, 'setIsAdmin' | 'setPubKey' | 'setBudget'> = {

diff --git a/src/utils/setBudget/index.ts b/src/utils/setBudget/index.ts
@@ -1,6 +1,6 @@
 import * as sphinx from 'sphinx-bridge-kevkevinpal'
 
-export async function updateBudget(setBudget: (value: number) => void) {
+export async function updateBudget(setBudget: (value: number | null) => void) {
   // eslint-disable-next-line @typescript-eslint/ban-ts-comment
   // @ts-ignore
   await sphinx.enable()
@@ -9,7 +9,9 @@ export async function updateBudget(setBudget: (value: number) => void) {
   // @ts-ignore
   const budget = await sphinx.getBudget()
 
-  if (budget?.budget) {
+  if (budget?.msg === 'Invalid Action') {
+    setBudget(null)
+  } else if (budget?.budget) {
     setBudget(budget.budget)
   }
 }