flaten set method

Author: Syed Raza Abbas

kinde_sdk/auth/oauth.py

@@ -58,7 +58,7 @@ def __init__(
             storage_config = config.get("storage", {"type": "memory"})  # Default to "memory"
         elif storage_config is None:
             storage_config = {"type": "memory"}  # Default to in-memory storage if no config is provided
-        # storage = StorageFactory.create_storage(storage_config)
+        
         storage_manager = StorageManager()
         storage_manager.initialize(storage_config)
 
@@ -150,12 +150,12 @@ async def generate_auth_url(
         # Generate state if not provided
         state = login_options.get(LoginOptions.STATE, generate_random_string(32))
         search_params["state"] = state
-        self.session_manager.storage_manager.set("state", {"value": state})
+        self.session_manager.storage_manager.setItems("state", {"value": state})
 
         # Generate nonce if not provided
         nonce = login_options.get(LoginOptions.NONCE, generate_random_string(16))
         search_params["nonce"] = nonce
-        self.session_manager.storage_manager.set("nonce", {"value": nonce})
+        self.session_manager.storage_manager.setItems("nonce", {"value": nonce})
 
         # Handle PKCE
         code_verifier = ""
@@ -166,7 +166,7 @@ async def generate_auth_url(
             pkce_data = await generate_pkce_pair(52)  # Use 52 chars to match JS implementation
             code_verifier = pkce_data["code_verifier"]
             search_params["code_challenge"] = pkce_data["code_challenge"]
-            self.session_manager.storage_manager.set("code_verifier", {"value": code_verifier})
+            self.session_manager.storage_manager.setItems("code_verifier", {"value": code_verifier})
 
         # Set code challenge method
         code_challenge_method = login_options.get(LoginOptions.CODE_CHALLENGE_METHOD, "S256")

kinde_sdk/auth/user_session.py

@@ -77,7 +77,7 @@ def _save_to_storage(self, user_id: str):
             }
             # Store with user: prefix to make it user-specific but device-independent
             # if you want device-specific sessions, remove the "user:" prefix
-            self.storage_manager.set(user_id, serialized_data)
+            self.storage_manager.setItems(user_id, serialized_data)
 
     # def _load_from_storage(self, user_id: str) -> bool:
     #     """Load session data from storage if not already in memory."""

kinde_sdk/core/storage/local_storage.py

@@ -32,6 +32,16 @@ def set(self, key: str, value: Dict) -> None:
         """
         self.storage.setItem(key, json.dumps(value))
 
+    def set_flat(self, data: str) -> None:
+        """
+        Store data associated with the given key.
+        
+        Args:
+            key (str): The key to store the data under.
+            value (Dict): The data to store.
+        """
+        self.storage.set(json.dumps(data))
+
     def delete(self, key: str) -> None:
         """
         Delete data associated with the given key.

kinde_sdk/core/storage/memory_storage.py

@@ -5,6 +5,7 @@
 class MemoryStorage(StorageInterface):
     def __init__(self):
         self._storage = {}
+        self._flat_storage = []
 
     def get(self, key: str) -> Optional[Dict]:
         """
@@ -28,6 +29,16 @@ def set(self, key: str, value: Dict) -> None:
         """
         self._storage[key] = value
 
+    def set_flat(self, data: str) -> None:
+        """
+        Store data associated with the given key.
+        
+        Args:
+            key (str): The key to store the data under.
+            value (Dict): The data to store.
+        """
+        self._flat_storage = data
+
     def delete(self, key: str) -> None:
         """
         Delete data associated with the given key.

kinde_sdk/core/storage/storage_interface.py

@@ -27,6 +27,17 @@ def set(self, key: str, value: Dict) -> None:
         """
         pass
 
+    @abstractmethod
+    def set_flat(self, value: str) -> None:
+        """
+        Store data associated with the given key.
+        
+        Args:
+            key (str): The key to store the data under.
+            value (Dict): The data to store.
+        """
+        pass
+
     @abstractmethod
     def delete(self, key: str) -> None:
         """

kinde_sdk/core/storage/storage_manager.py

@@ -66,7 +66,7 @@ def get_device_id(self) -> str:
             else:
                 # Generate a new device ID
                 self._device_id = str(uuid.uuid4())
-                self.set("_device_id", {"value": self._device_id, "timestamp": time.time()})
+                self.setItems("_device_id", {"value": self._device_id, "timestamp": time.time()})
 
         return self._device_id
 
@@ -128,7 +128,7 @@ def get(self, key: str) -> Optional[Dict]:
         namespaced_key = self._get_namespaced_key(key)
         return self._storage.get(namespaced_key)
 
-    def set(self, key: str, value: Dict) -> None:
+    def setItems(self, key: str, value: Dict) -> None:
         """
         Store data in storage.
         
@@ -143,6 +143,21 @@ def set(self, key: str, value: Dict) -> None:
         self._storage.set(namespaced_key, value)
 
 
+    def set(self, access_token : str) -> None:
+        """
+        Store data in storage.
+        
+        Args:
+            key (str): The key to store under.
+            value (Dict): The data to store.
+        """
+        if self._storage is None:
+            self.initialize()
+            
+        # namespaced_key = self._get_namespaced_key(key)
+        self._storage.set_flat(access_token)
+
+
     def delete(self, key: str) -> None:
         """
         Delete data from storage by key.