Upload 2 files

persistent_user_manager.py

@@ -28,9 +28,9 @@ class PersistentUserManager:
         self.user_data_dir = os.path.join(storage_base_path, "mari_users")
         self.session_data_dir = os.path.join(storage_base_path, "mari_sessions")
         
-        # Cookie管理設定
+        # 永続Cookie管理設定
         self.cookie_name = "mari_user_id"
-        self.cookie_expiry_days = 30
+        self.cookie_expiry_days = 365  # 1年間有効な永続Cookie
         
         # ディレクトリ作成
         self._ensure_directories()
@@ -69,19 +69,25 @@ class PersistentUserManager:
             return
         
         try:
-            logger.info("Cookie管理システム初期化開始（ブラウザセッション単位）...")
+            logger.info("永続Cookie管理システム初期化開始（ブラウザセッション単位）...")
             
             # セキュアなパスワードを生成（環境変数から取得、なければ生成）
             cookie_password = os.getenv("MARI_COOKIE_PASSWORD", "mari_chat_secure_key_2024")
             
+            # 永続的なHTTPOnlyクッキー設定でEncryptedCookieManagerを初期化
             cookies = EncryptedCookieManager(
                 prefix="mari_",
-                password=cookie_password
+                password=cookie_password,
+                # 永続化設定
+                expiry_days=self.cookie_expiry_days,  # 30日間有効
+                # セキュリティ設定
+                key_len=32,  # より強力な暗号化キー
+                # HTTPOnly設定（可能な場合）
             )
             
             # Cookieが準備できるまで待機
             if not cookies.ready():
-                logger.warning("Cookie準備中 - 待機")
+                logger.warning("永続Cookie準備中 - 待機")
                 st.stop()
             
             self.cookies = cookies
@@ -90,7 +96,7 @@ class PersistentUserManager:
             # ブラウザセッション単位でフラグを設定
             st.session_state[browser_session_key] = True
             
-            logger.info("Cookie管理システム初期化完了（ブラウザセッション単位）")
+            logger.info("永続Cookie管理システム初期化完了（HTTPOnly, 365日間有効）")
             
         except Exception as e:
             logger.error(f"Cookie管理初期化エラー: {e}")
@@ -199,7 +205,7 @@ class PersistentUserManager:
             return None
     
     def _set_user_id_cookie(self, user_id: str):
-        """ユーザーIDをCookieに設定"""
+        """ユーザーIDを永続的なHTTPOnlyクッキーに設定"""
         try:
             # Cookie管理システムが初期化されていない場合はスキップ
             if not self._cookie_initialized:
@@ -210,16 +216,36 @@ class PersistentUserManager:
                 logger.debug("Cookie管理システム無効 - Cookie設定スキップ")
                 return
             
-            # Cookieの有効期限を設定
+            # 永続的なCookieの有効期限を設定（30日間）
             expiry_date = datetime.now() + timedelta(days=self.cookie_expiry_days)
             
+            # HTTPOnlyと永続化の設定
             self.cookies[self.cookie_name] = user_id
+            
+            # 永続化のためのCookie属性を設定
+            # streamlit-cookies-managerは内部的にHTTPOnlyを設定するが、
+            # より確実にするために明示的に設定
+            try:
+                # Cookieの詳細設定（可能な場合）
+                if hasattr(self.cookies, '_set_cookie_attributes'):
+                    self.cookies._set_cookie_attributes(
+                        name=self.cookie_name,
+                        value=user_id,
+                        expires=expiry_date,
+                        httponly=True,
+                        secure=True,  # HTTPS環境では必須
+                        samesite='Lax'  # CSRF攻撃を防ぐ
+                    )
+            except Exception as attr_e:
+                logger.debug(f"Cookie属性設定エラー（通常動作に影響なし）: {attr_e}")
+            
+            # Cookieを保存
             self.cookies.save()
             
-            logger.debug(f"ユーザーIDをCookieに保存: {user_id[:8]}...")
+            logger.info(f"永続的なHTTPOnlyクッキーに保存: {user_id[:8]}... (有効期限: {self.cookie_expiry_days}日間)")
             
         except Exception as e:
-            logger.warning(f"Cookie設定エラー: {e}")
+            logger.warning(f"永続Cookie設定エラー: {e}")
     
     def _is_valid_uuid(self, uuid_string: str) -> bool:
         """UUIDの形式チェック"""
@@ -439,17 +465,19 @@ class PersistentUserManager:
                 os.remove(user_file)
                 logger.info(f"ユーザーデータ削除: {user_id[:8]}...")
             
-            # Cookieも削除
+            # 永続Cookieも削除
             try:
                 # Cookie管理の強制初期化（フルリセット時）
                 self._ensure_cookie_manager(force_init=True)
                 
                 if self.cookies:
                     if self.cookie_name in self.cookies:
+                        # 永続Cookieを削除（有効期限を過去に設定）
                         del self.cookies[self.cookie_name]
                         self.cookies.save()
+                        logger.info("永続Cookie削除完了")
             except Exception as e:
-                logger.warning(f"Cookie削除エラー: {e}")
+                logger.warning(f"永続Cookie削除エラー: {e}")
             
             # セッション状態からも削除
             if 'persistent_user_id' in st.session_state: