# ComfyUI_DropSendNode/setup_dropbox_node.py import os import requests import webbrowser import uuid import json from dotenv import load_dotenv, dotenv_values import urllib.parse from .dropbox_auth_manager import DropboxAuthManager from .oauth_handler import OAuthCallbackHandler, get_server_base_url class DropboxSetupNode: @classmethod def _check_credentials_non_blocking(cls): """Check for credentials without blocking operations""" try: node_dir = os.path.dirname(__file__) # Check .env file env_path = os.path.join(node_dir, ".env") if os.path.exists(env_path): from dotenv import dotenv_values env_vars = dotenv_values(env_path) if all([env_vars.get("DROPBOX_APP_KEY"), env_vars.get("DROPBOX_APP_SECRET"), env_vars.get("DROPBOX_REFRESH_TOKEN")]): return True # Check display_only with environment variables display_marker_path = os.path.join(node_dir, ".dropbox_display_complete") if os.path.exists(display_marker_path): display_only_env_set = all([ os.getenv("DROPBOX_APP_KEY"), os.getenv("DROPBOX_APP_SECRET"), os.getenv("DROPBOX_REFRESH_TOKEN") ]) if display_only_env_set: return True return False except Exception: return False @classmethod def INPUT_TYPES(cls): # Check if credentials are already stored (non-blocking) try: print(f"[DropboxSetup] INPUT_TYPES: Checking for credentials...") is_connected = cls._check_credentials_non_blocking() print(f"[DropboxSetup] INPUT_TYPES check - is_connected: {is_connected}") except Exception as e: print(f"[DropboxSetup] INPUT_TYPES error: {e}") is_connected = False # Base inputs that are always shown inputs = { "required": { "dropbox_dest_folder": ("STRING", {"default": "/Apps/ComfyUI_Output_Files"}), }, "optional": {} } if is_connected: # If connected, only show reconnect option inputs["optional"]["reconnect"] = ("BOOLEAN", { "label": "Reconnect Dropbox (clears credentials & refreshes UI)", "default": False }) else: # If not connected, show setup fields inputs["required"].update({ "app_key": ("STRING", {"default": "", "multiline": False}), "app_secret": ("STRING", {"default": "", "multiline": False}), "auth_code": ("STRING", {"default": "", "multiline": False}), }) inputs["optional"]["reconnect"] = ("BOOLEAN", { "label": "Reset stored credentials", "default": False }) # Simple storage options without keyring complexity default_storage = "env_file" storage_options = ["env_file", "display_only"] inputs["optional"]["storage_method"] = (storage_options, { "label": "Credential Storage Method", "default": default_storage }) return inputs RETURN_TYPES = ("STRING",) OUTPUT_NODE = True FUNCTION = "setup" def setup(self, dropbox_dest_folder, app_key=None, app_secret=None, auth_code=None, reconnect=False, storage_method="env_file"): try: print(f"[DropboxSetup] Called with:") print(f" app_key: '{app_key}' (type: {type(app_key)}, bool: {bool(app_key)})") print(f" app_secret: '{app_secret}' (type: {type(app_secret)}, bool: {bool(app_secret)})") print(f" auth_code: '{auth_code}' (type: {type(auth_code)}, bool: {bool(auth_code)})") print(f" reconnect: {reconnect}") print(f" storage_method: {storage_method}") # Initialize auth manager auth_manager = DropboxAuthManager() print(f"[DropboxSetup] Auth manager initialized") # Handle reconnect/reset request if reconnect: print("[DropboxSetup] Reconnect requested - clearing all credentials") # Clear all credentials (skip token revocation to avoid delays) print("[DropboxSetup] Clearing all stored credentials") auth_manager.reset(revoke_token=False) # Manually clear all credential files node_dir = os.path.dirname(__file__) # Clear .env file if it exists env_path = os.path.join(node_dir, ".env") if os.path.exists(env_path): print(f"[DropboxSetup] Removing .env file: {env_path}") os.remove(env_path) # Clear display_only marker if it exists display_marker_path = os.path.join(node_dir, ".dropbox_display_complete") if os.path.exists(display_marker_path): print(f"[DropboxSetup] Removing display_only marker: {display_marker_path}") os.remove(display_marker_path) # Send WebSocket message to trigger ComfyUI refresh after clearing credentials try: from server import PromptServer message_data = { "type": "dropbox_reconnect_complete", "success": True, "message": "Credentials cleared - ComfyUI will refresh to show auth fields" } PromptServer.instance.send_sync("dropbox_reconnect_complete", message_data) print(f"[DropboxSetup] Sent WebSocket notification for reconnect completion") except Exception as e: print(f"[DropboxSetup] Warning: Could not send WebSocket notification: {e}") message = "Dropbox credentials cleared. ComfyUI will refresh to show auth fields..." print(f"[DropboxSetup] {message}") return { "ui": {"text": [message]}, "result": (message,) } # Check if already connected using the same non-blocking method as INPUT_TYPES is_connected = self._check_credentials_non_blocking() if is_connected: print("[DropboxSetup] Already connected - testing stored credentials") # Only access keyring now when we know we need to test credentials try: # This will trigger keyring access but only when we know credentials exist access_token = auth_manager.get_access_token() message = "Dropbox already connected using stored credentials. Ready to upload files." print(f"[DropboxSetup] {message}") return { "ui": {"text": [message]}, "result": (message,) } except Exception as e: message = f"Warning: Stored credentials found but invalid: {e}. Use 'reconnect' to reset." print(f"[DropboxSetup] {message}") return { "ui": {"text": [message]}, "result": (message,) } # Check for display_only environment variables (user set them up after OAuth flow) display_only_env_set = all([ os.getenv("DROPBOX_APP_KEY"), os.getenv("DROPBOX_APP_SECRET"), os.getenv("DROPBOX_REFRESH_TOKEN") ]) if display_only_env_set: # Check if this came from display_only flow node_dir = os.path.dirname(__file__) display_marker_path = os.path.join(node_dir, ".dropbox_display_complete") if os.path.exists(display_marker_path): message = "Dropbox connected using environment variables (display_only setup). Ready to upload files." print(f"[DropboxSetup] {message}") return { "ui": {"text": [message]}, "result": (message,) } else: # Legacy environment variables (not from display_only) message = "Dropbox credentials found in system environment variables. Ready to upload files." print(f"[DropboxSetup] {message}") return { "ui": {"text": [message]}, "result": (message,) } # Check for RunPod secrets runpod_env_set = all([ os.getenv("RUNPOD_SECRET_DROPBOX_ACCESS_TOKEN"), os.getenv("RUNPOD_SECRET_DROPBOX_REFRESH_TOKEN") ]) if runpod_env_set: return ("Warning: Detected RunPod secrets. Using those instead.",) # New setup flow using DropboxAuthManager print(f"[DropboxSetup] Starting new setup flow") # Clean up the inputs first app_key_clean = app_key.strip() if app_key else "" app_secret_clean = app_secret.strip() if app_secret else "" auth_code_clean = auth_code.strip() if auth_code else "" print(f"[DropboxSetup] Cleaned inputs:") print(f" app_key_clean: '{app_key_clean}' (len: {len(app_key_clean)})") print(f" app_secret_clean: '{app_secret_clean}' (len: {len(app_secret_clean)})") print(f" auth_code_clean: '{auth_code_clean}' (len: {len(auth_code_clean)})") # Check if we have app credentials if not app_key_clean or not app_secret_clean: message = "Error: Missing App Key or App Secret. Please provide both." print(f"[DropboxSetup] {message}") return (message,) # If no auth code, generate OAuth URL for automatic popup flow if not auth_code_clean: print(f"[DropboxSetup] No auth code provided - generating OAuth URL for popup flow") auth_temp = DropboxAuthManager(app_key=app_key_clean) # Automatic OAuth flow with popup callback session_id = str(uuid.uuid4()) base_url = get_server_base_url() print(f"[DropboxSetup] Detected base URL: {base_url}") callback_url = f"{base_url}/oauth/dropbox/callback" print(f"[DropboxSetup] Using callback URL: {callback_url}") oauth_url = auth_temp.get_oauth_url(redirect_uri=callback_url, state=session_id, force_reapprove=True) # Set up OAuth session for callback handling oauth_handler = OAuthCallbackHandler() oauth_handler.start_oauth_session(session_id, app_key_clean, app_secret_clean, storage_method=storage_method, dropbox_folder=dropbox_dest_folder, original_redirect_uri=callback_url) try: print(f"[DropboxSetup] Setting up automatic OAuth popup...") # JavaScript will automatically open a popup window message = f"Dropbox OAuth Ready!\n\nClick the link below to authorize with Dropbox:\n\n{oauth_url}\n\nA popup window will open, and after authorization, ComfyUI will refresh automatically!" print(f"[DropboxSetup] Session ID: {session_id}") print(f"[DropboxSetup] Callback URL: {callback_url}") print(f"[DropboxSetup] OAuth URL ready for popup: {oauth_url}") except Exception as e: print(f"[DropboxSetup] Error setting up OAuth: {e}") message = f"Dropbox Authorization:\n\nPlease visit this URL to authorize:\n{oauth_url}\n\nAfter authorization, ComfyUI will refresh automatically!" print(f"[DropboxSetup] OAuth URL: {oauth_url}") # Use ComfyUI's dynamic return format for better UI integration return { "ui": {"text": [message]}, "result": (message,) } # Exchange auth code for refresh token using DropboxAuthManager print(f"[DropboxSetup] Attempting to exchange auth code") auth_manager_setup = DropboxAuthManager(app_key_clean, app_secret_clean) # Get the tokens without storing them yet # Always use automatic OAuth flow with callback URL base_url = get_server_base_url() callback_url = f"{base_url}/oauth/dropbox/callback" print(f"[DropboxSetup] Using automatic OAuth with redirect_uri: {callback_url}") result = auth_manager_setup.exchange_auth_code_raw(auth_code_clean, redirect_uri=callback_url) refresh_token = result.get("refresh_token") print(f"[DropboxSetup] Auth code exchange successful") print(f"[DropboxSetup] Using storage method: {storage_method}") # Handle different storage methods if storage_method == "env_file": # Store in .env file node_dir = os.path.dirname(__file__) env_path = os.path.join(node_dir, ".env") with open(env_path, "w") as f: f.write(f"DROPBOX_APP_KEY={app_key_clean}\n") f.write(f"DROPBOX_APP_SECRET={app_secret_clean}\n") f.write(f"DROPBOX_REFRESH_TOKEN={refresh_token}\n") f.write(f"DROPBOX_FOLDER={dropbox_dest_folder}\n") message = "Dropbox connected successfully! Credentials saved to .env file." elif storage_method == "display_only": # Display credentials for manual copying and create completion marker node_dir = os.path.dirname(__file__) display_marker_path = os.path.join(node_dir, ".dropbox_display_complete") with open(display_marker_path, "w") as f: f.write("display_only_setup_completed") message = f"""Dropbox Connected Successfully! ===================================================================== ENVIRONMENT VARIABLES - Copy & Paste Ready ===================================================================== DROPBOX_APP_KEY={app_key_clean} DROPBOX_APP_SECRET={app_secret_clean} DROPBOX_REFRESH_TOKEN={refresh_token} DROPBOX_FOLDER={dropbox_dest_folder} ===================================================================== Perfect for RunPod, Docker, and production environments! These credentials are ready to use immediately. =====================================================================""" else: # Fallback to env_file storage node_dir = os.path.dirname(__file__) env_path = os.path.join(node_dir, ".env") with open(env_path, "w") as f: f.write(f"DROPBOX_APP_KEY={app_key_clean}\n") f.write(f"DROPBOX_APP_SECRET={app_secret_clean}\n") f.write(f"DROPBOX_REFRESH_TOKEN={refresh_token}\n") f.write(f"DROPBOX_FOLDER={dropbox_dest_folder}\n") message = "Dropbox connected successfully! Credentials saved to .env file." print(f"[DropboxSetup] {message}") # For display_only, ensure credentials are prominently shown in console if storage_method == "display_only": print("\n" + "="*80) print("DROPBOX CREDENTIALS READY FOR PRODUCTION - COPY FROM CONSOLE") print("="*80) print(f"DROPBOX_APP_KEY={app_key_clean}") print(f"DROPBOX_APP_SECRET={app_secret_clean}") print(f"DROPBOX_REFRESH_TOKEN={refresh_token}") print(f"DROPBOX_FOLDER={dropbox_dest_folder}") print("="*80) print("Copy the lines above to your environment variables!") print("Perfect for RunPod, Docker, and production environments!") print("Note: ComfyUI won't auto-refresh - refresh manually after copying!") print("="*80 + "\n") # Use ComfyUI's dynamic return format for better UI integration return { "ui": {"text": [message]}, "result": (message,) } except Exception as e: message = f"Error: Setup failed: {e}" print(f"[DropboxSetup] ERROR: {message}") return { "ui": {"text": [message]}, "result": (message,) } # Required mappings for ComfyUI NODE_CLASS_MAPPINGS = {"DropboxSetupNode": DropboxSetupNode} NODE_DISPLAY_NAME_MAPPINGS = {"DropboxSetupNode": "Dropbox AutoUploader Setup"}