fix: updated storagecontext

power-pay-frontend/src/hooks/StorageContext.tsx

@@ -1,20 +1,6 @@
 import { createContext, PropsWithChildren, useEffect } from 'react';
 import { useStorage } from './useStorage';
 
-interface StorageService<T> {
-  getItem: (key: string) => Promise<T | undefined>;
-  setItem: (key: string, value: T) => Promise<boolean>;
-  removeItem: (key: string) => Promise<boolean>;
-  clear: () => Promise<boolean>;
-}
-
-// Create the StorageContextData interface.
-interface StorageContextData<T> {
-  getItem: (key: string) => Promise<unknown | undefined>;
-  setItem: (key: string, value: T) => Promise<boolean>;
-  removeItem: (key: string) => Promise<boolean>;
-  clear: () => Promise<boolean>;
-}
 
 
 // Creating  a fake implementation of the StorageContextData interface.
@@ -38,51 +24,71 @@ const fakeStorage: StorageContextData<unknown> = {
 };
 
 
+// Create the StorageContextData interface.
+interface StorageContextData<T> {
+  getItem: (key: string) => Promise<unknown | undefined>;
+  setItem: (key: string, value: T) => Promise<boolean>;
+  removeItem: (key: string) => Promise<boolean>;
+  clear: () => Promise<boolean>;
+}
+
+interface StorageService<T> {
+  getItem: (key: string) => Promise<T | undefined>;
+  setItem: (key: string, value: T) => Promise<boolean>;
+  removeItem: (key: string) => Promise<boolean>;
+  clear: () => Promise<boolean>;
+}
+
+
 // Create the StorageContext with default functions for getItem and setItem.
-const StorageContext = createContext<StorageService<unknown> | undefined>(fakeStorage);
+const StorageContext = createContext<StorageService<unknown> | undefined>(undefined);
 
 // StorageProvider: A React component that wraps the application and provides the storage context with getItem and setItem methods.
 export function StorageProvider<T>({ initialValue, children }: PropsWithChildren<{ initialValue?: T }>) {
-  const [storedValue, setStoredValue] = useStorage<T | undefined>({ initialValue: { ...initialValue } });
+  const [storedValue, setStoredValue] = useStorage<T | undefined>({ key: 'your_key_here', initialValue });
 
-// Define a global constant for the key.
+  // Define a global constant for the key.
   const STORAGE_KEY = 'key';
 
   // Initializes the state with the value from the local storage if it exists.
   useEffect(() => {
-    const item = localStorage.getItem('STORAGE_KEY');
+    const item = localStorage.getItem(STORAGE_KEY);
     if (item) {
       setStoredValue(JSON.parse(item));
     }
-  }, []);
+  }, [setStoredValue]);
 
   // Updates the local storage whenever the state changes.
   useEffect(() => {
-    if (storedValue) {
-      setItemWrapper(STORAGE_KEY, storedValue);
+    if (storedValue !== undefined) {
+      localStorage.setItem(STORAGE_KEY, JSON.stringify(storedValue));
     }
   }, [storedValue]);
 
-   // Defining a wrapper function for the setItem method that updates the local storage and the state with the new value.
-   const setItemWrapper = async (key: string, value: T | undefined) => {
-    fakeStorage.setItem(key, value);
+  // Defining a wrapper function for the setItem method that updates the local storage and the state with the new value.
+  const setItemWrapper = async (key: string, value: T | undefined) => {
+    if (value !== undefined) {
+      localStorage.setItem(key, JSON.stringify(value));
+    } else {
+      localStorage.removeItem(key);
+    }
     setStoredValue(value);
     return true;
   };
 
-// Destructure getItem and setItem before using them in the StorageContext.Provider value prop.
-const contextValue: StorageService<T> = {
-  getItem: fakeStorage.getItem,
-  setItem: setItemWrapper,
-  removeItem: fakeStorage.removeItem,
-  clear: fakeStorage.clear,
-};
-return (
-  <StorageContext.Provider value={contextValue as StorageService<unknown>}>
-    {children}
-  </StorageContext.Provider>
-);
-}
+  // Destructure getItem and setItem before using them in the StorageContext.Provider value prop.
+  const contextValue: StorageService<T> = {
+    getItem: (key) => fakeStorage.getItem(key),
+    setItem: (key, value) => setItemWrapper(key, value),
+    removeItem: (key) => fakeStorage.removeItem(key),
+    clear: () => fakeStorage.clear(),
+  };
 
+  return (
+    <StorageContext.Provider value={contextValue as StorageService<unknown>}>
+      {children}
+    </StorageContext.Provider>
+  );
+}
 
-export default StorageContext;
+export default StorageContext;