47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
# Copyright (C) 2025 AIDC-AI
|
|
# Licensed under the MIT License.
|
|
|
|
"""
|
|
Authentication utilities for ComfyUI Copilot
|
|
"""
|
|
|
|
from typing import Optional
|
|
from .globals import set_comfyui_copilot_api_key, get_comfyui_copilot_api_key, set_session_id
|
|
|
|
|
|
def extract_and_store_api_key(request) -> Optional[str]:
|
|
"""
|
|
Extract Bearer token from Authorization header and store it in globals
|
|
|
|
Args:
|
|
request: The aiohttp request object
|
|
|
|
Returns:
|
|
The extracted API key if successful, None otherwise
|
|
"""
|
|
try:
|
|
session_id = request.get('session_id')
|
|
if session_id:
|
|
set_session_id(session_id)
|
|
|
|
auth_header = request.headers.get('Authorization')
|
|
if auth_header and auth_header.startswith('Bearer '):
|
|
api_key = auth_header[7:] # Remove 'Bearer ' prefix
|
|
set_comfyui_copilot_api_key(api_key)
|
|
print(f"ComfyUI Copilot API key extracted and stored: {api_key[:12]}...")
|
|
|
|
# Verify it's stored correctly
|
|
stored_key = get_comfyui_copilot_api_key()
|
|
if stored_key == api_key:
|
|
print("API key verification: ✓ Successfully stored in globals")
|
|
else:
|
|
print("API key verification: ✗ Storage failed")
|
|
|
|
return api_key
|
|
else:
|
|
print("No valid Authorization header found")
|
|
return None
|
|
except Exception as e:
|
|
print(f"Error extracting API key: {str(e)}")
|
|
return None
|