Upload 2 files

persistent_user_manager.py

@@ -56,12 +56,21 @@ class PersistentUserManager:
             os.makedirs(self.session_data_dir, exist_ok=True)
             logger.warning(f"フォールバック: ローカルディレクトリを使用 {self.user_data_dir}")
     
-    def _ensure_cookie_manager(self):
-        """Cookie管理システムの遅延初期化"""
-        if self._cookie_initialized:
+    def _ensure_cookie_manager(self, force_init: bool = False):
+        """Cookie管理システムの初期化（初回またはフルリセット時のみ）"""
+        # 既に初期化済みで、強制初期化でない場合はスキップ
+        if self._cookie_initialized and not force_init:
+            return
+        
+        # アプリ全体でのCookie初期化状態をチェック（グローバルフラグ）
+        if not force_init and st.session_state.get('cookie_manager_initialized', False):
+            logger.debug("Cookie管理システム既に初期化済み - スキップ")
+            self._cookie_initialized = True
             return
         
         try:
+            logger.info("Cookie管理システム初期化開始...")
+            
             # セキュアなパスワードを生成（環境変数から取得、なければ生成）
             cookie_password = os.getenv("MARI_COOKIE_PASSWORD", "mari_chat_secure_key_2024")
             
@@ -76,63 +85,86 @@ class PersistentUserManager:
             
             self.cookies = cookies
             self._cookie_initialized = True
-            logger.info("Cookie管理システム遅延初期化完了")
+            
+            # グローバルフラグを設定（アプリ全体で一度だけ初期化）
+            st.session_state.cookie_manager_initialized = True
+            
+            logger.info("Cookie管理システム初期化完了")
             
         except Exception as e:
             logger.error(f"Cookie管理初期化エラー: {e}")
             # フォールバック: セッション状態のみ使用
             self.cookies = None
             self._cookie_initialized = True
+            st.session_state.cookie_manager_initialized = True
     
-    def get_or_create_user_id(self) -> str:
+    def get_or_create_user_id(self, force_reset: bool = False) -> str:
         """
         ユーザーIDを取得または新規作成（Cookie連携・重複防止）
         
+        Args:
+            force_reset: フルリセット時のみTrue
+        
         Returns:
             ユーザーID
         """
         try:
-            # 最優先: 既にStreamlitセッション状態にuser_idがある場合はそれを使用
-            if 'user_id' in st.session_state:
+            # 最優先: 既にStreamlitセッション状態にuser_idがある場合はそれを使用（フルリセット以外）
+            if not force_reset and 'user_id' in st.session_state:
                 existing_id = st.session_state.user_id
                 if existing_id and self._is_valid_uuid(existing_id):
                     logger.debug(f"Streamlitセッション状態からユーザーID使用: {existing_id[:8]}...")
                     return existing_id
             
-            # 重複チェック防止: 既にセッション状態にある場合はそれを使用
-            if hasattr(st.session_state, 'persistent_user_id_checked'):
+            # 重複チェック防止: 既にセッション状態にある場合はそれを使用（フルリセット以外）
+            if not force_reset and hasattr(st.session_state, 'persistent_user_id_checked'):
                 existing_id = st.session_state.get('persistent_user_id')
                 if existing_id and self._is_valid_uuid(existing_id):
                     logger.debug(f"既にチェック済みのユーザーID使用: {existing_id[:8]}...")
                     return existing_id
             
-            # 1. CookieからユーザーIDを取得
-            user_id = self._get_user_id_from_cookie()
-            
-            if user_id and self._is_valid_user_id(user_id):
-                # 有効なユーザーIDが存在
-                self._update_user_access_time(user_id)
-                st.session_state.persistent_user_id = user_id
+            # Cookie初期化は初回またはフルリセット時のみ
+            if not self._cookie_initialized or force_reset:
+                # 1. CookieからユーザーIDを取得
+                user_id = self._get_user_id_from_cookie()
+                
+                if user_id and self._is_valid_user_id(user_id):
+                    # 有効なユーザーIDが存在
+                    self._update_user_access_time(user_id)
+                    st.session_state.persistent_user_id = user_id
+                    st.session_state.persistent_user_id_checked = True
+                    logger.info(f"CookieからユーザーID取得: {user_id[:8]}...")
+                    return user_id
+                
+                # 2. セッション状態から取得を試行（Cookieが無効な場合のみ）
+                if not force_reset:
+                    user_id = st.session_state.get('persistent_user_id')
+                    if user_id and self._is_valid_user_id(user_id):
+                        # セッション状態にあるIDを使用してCookieを更新
+                        self._set_user_id_cookie(user_id)
+                        self._update_user_access_time(user_id)
+                        st.session_state.persistent_user_id_checked = True
+                        logger.info(f"セッション状態からユーザーID復元: {user_id[:8]}...")
+                        return user_id
+                
+                # 3. 新規ユーザーIDを作成
+                user_id = self._create_new_user()
                 st.session_state.persistent_user_id_checked = True
-                logger.info(f"CookieからユーザーID取得: {user_id[:8]}...")
+                logger.info(f"新規ユーザーID作成: {user_id[:8]}...")
                 return user_id
-            
-            # 2. セッション状態から取得を試行（Cookieが無効な場合のみ）
-            user_id = st.session_state.get('persistent_user_id')
-            if user_id and self._is_valid_user_id(user_id):
-                # セッション状態にあるIDを使用してCookieを更新
-                self._set_user_id_cookie(user_id)
-                self._update_user_access_time(user_id)
+            else:
+                # Cookie初期化済みの場合は、セッション状態から取得
+                existing_id = st.session_state.get('persistent_user_id')
+                if existing_id and self._is_valid_uuid(existing_id):
+                    logger.debug(f"Cookie初期化済み - セッション状態からユーザーID使用: {existing_id[:8]}...")
+                    return existing_id
+                
+                # フォールバック: 新規作成
+                user_id = self._create_new_user()
                 st.session_state.persistent_user_id_checked = True
-                logger.info(f"セッション状態からユーザーID復元: {user_id[:8]}...")
+                logger.info(f"フォールバック新規ユーザーID作成: {user_id[:8]}...")
                 return user_id
             
-            # 3. 新規ユーザーIDを作成
-            user_id = self._create_new_user()
-            st.session_state.persistent_user_id_checked = True
-            logger.info(f"新規ユーザーID作成: {user_id[:8]}...")
-            return user_id
-            
         except Exception as e:
             logger.error(f"ユーザーID取得エラー: {e}")
             # フォールバック: 一時的なIDを生成
@@ -144,16 +176,19 @@ class PersistentUserManager:
     def _get_user_id_from_cookie(self) -> Optional[str]:
         """CookieからユーザーIDを取得"""
         try:
-            # Cookie管理の遅延初期化
+            # Cookie管理の初期化（初回のみ）
             self._ensure_cookie_manager()
             
             if self.cookies is None:
+                logger.debug("Cookie管理システム無効 - None返却")
                 return None
             
             user_id = self.cookies.get(self.cookie_name)
             if user_id and self._is_valid_uuid(user_id):
+                logger.debug(f"CookieからユーザーID取得: {user_id[:8]}...")
                 return user_id
             
+            logger.debug("Cookie内に有効なユーザーIDなし")
             return None
             
         except Exception as e:
@@ -163,10 +198,13 @@ class PersistentUserManager:
     def _set_user_id_cookie(self, user_id: str):
         """ユーザーIDをCookieに設定"""
         try:
-            # Cookie管理の遅延初期化
-            self._ensure_cookie_manager()
+            # Cookie管理システムが初期化されていない場合はスキップ
+            if not self._cookie_initialized:
+                logger.debug("Cookie管理システム未初期化 - Cookie設定スキップ")
+                return
             
             if self.cookies is None:
+                logger.debug("Cookie管理システム無効 - Cookie設定スキップ")
                 return
             
             # Cookieの有効期限を設定
@@ -375,8 +413,8 @@ class PersistentUserManager:
             
             # Cookieも削除
             try:
-                # Cookie管理の遅延初期化
-                self._ensure_cookie_manager()
+                # Cookie管理の強制初期化（フルリセット時）
+                self._ensure_cookie_manager(force_init=True)
                 
                 if self.cookies:
                     if self.cookie_name in self.cookies:
@@ -388,6 +426,8 @@ class PersistentUserManager:
             # セッション状態からも削除
             if 'persistent_user_id' in st.session_state:
                 del st.session_state.persistent_user_id
+            if 'persistent_user_id_checked' in st.session_state:
+                del st.session_state.persistent_user_id_checked
             
             return True
             
@@ -395,6 +435,42 @@ class PersistentUserManager:
             logger.error(f"ユーザーデータ削除エラー: {e}")
             return False
     
+    def full_reset_user_session(self) -> str:
+        """
+        フルリセット（Cookie初期化を含む）
+        
+        Returns:
+            新しいユーザーID
+        """
+        try:
+            logger.info("フルリセット開始 - Cookie初期化を含む")
+            
+            # セッション状態をクリア
+            if 'persistent_user_id' in st.session_state:
+                del st.session_state.persistent_user_id
+            if 'persistent_user_id_checked' in st.session_state:
+                del st.session_state.persistent_user_id_checked
+            if 'cookie_manager_initialized' in st.session_state:
+                del st.session_state.cookie_manager_initialized
+            
+            # Cookie初期化フラグをリセット
+            self._cookie_initialized = False
+            self.cookies = None
+            
+            # 新しいユーザーIDを作成（強制リセット）
+            new_user_id = self.get_or_create_user_id(force_reset=True)
+            
+            logger.info(f"フルリセット完了: {new_user_id[:8]}...")
+            return new_user_id
+            
+        except Exception as e:
+            logger.error(f"フルリセットエラー: {e}")
+            # フォールバック
+            import uuid
+            fallback_id = str(uuid.uuid4())
+            st.session_state.persistent_user_id = fallback_id
+            return fallback_id
+    
     def list_all_users(self) -> List[Dict[str, Any]]:
         """
         全ユーザーの一覧を取得（管理用）