mirror of
https://github.com/Omni-guides/Jackify.git
synced 2026-01-17 19:47:00 +01:00
283 lines
14 KiB
Python
283 lines
14 KiB
Python
"""
|
|
Additional Tasks Menu Handler for Jackify CLI Frontend
|
|
Extracted from src.modules.menu_handler.MenuHandler.show_additional_tasks_menu()
|
|
"""
|
|
|
|
import time
|
|
|
|
from jackify.shared.colors import (
|
|
COLOR_SELECTION, COLOR_RESET, COLOR_ACTION, COLOR_PROMPT, COLOR_INFO, COLOR_DISABLED, COLOR_WARNING
|
|
)
|
|
from jackify.shared.ui_utils import print_jackify_banner, print_section_header, clear_screen
|
|
|
|
class AdditionalMenuHandler:
|
|
"""
|
|
Handles the Additional Tasks menu (MO2, NXM Handling & Recovery)
|
|
Extracted from legacy MenuHandler class
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.logger = None # Will be set by CLI when needed
|
|
|
|
def _clear_screen(self):
|
|
"""Clear the terminal screen with AppImage compatibility"""
|
|
clear_screen()
|
|
|
|
def show_additional_tasks_menu(self, cli_instance):
|
|
"""Show the Additional Tasks & Tools submenu"""
|
|
while True:
|
|
self._clear_screen()
|
|
print_jackify_banner()
|
|
print_section_header("Additional Tasks & Tools")
|
|
print(f"{COLOR_INFO}Nexus Authentication, TTW Install & more{COLOR_RESET}\n")
|
|
|
|
print(f"{COLOR_SELECTION}1.{COLOR_RESET} Nexus Mods Authorization")
|
|
print(f" {COLOR_ACTION}→ Authorize with Nexus using OAuth or manage API key{COLOR_RESET}")
|
|
print(f"{COLOR_SELECTION}2.{COLOR_RESET} Tale of Two Wastelands (TTW) Installation")
|
|
print(f" {COLOR_ACTION}→ Install TTW using TTW_Linux_Installer{COLOR_RESET}")
|
|
print(f"{COLOR_SELECTION}3.{COLOR_RESET} Coming Soon...")
|
|
print(f"{COLOR_SELECTION}0.{COLOR_RESET} Return to Main Menu")
|
|
selection = input(f"\n{COLOR_PROMPT}Enter your selection (0-3): {COLOR_RESET}").strip()
|
|
|
|
if selection.lower() == 'q': # Allow 'q' to re-display menu
|
|
continue
|
|
if selection == "1":
|
|
self._execute_nexus_authorization(cli_instance)
|
|
elif selection == "2":
|
|
self._execute_ttw_install(cli_instance)
|
|
elif selection == "3":
|
|
print(f"\n{COLOR_INFO}More features coming soon!{COLOR_RESET}")
|
|
input("\nPress Enter to return to menu...")
|
|
elif selection == "0":
|
|
break
|
|
else:
|
|
print("Invalid selection. Please try again.")
|
|
time.sleep(1)
|
|
|
|
def _execute_legacy_install_mo2(self, cli_instance):
|
|
"""LEGACY BRIDGE: Execute MO2 installation"""
|
|
# LEGACY BRIDGE: Use legacy imports until backend migration complete
|
|
if hasattr(cli_instance, 'menu') and hasattr(cli_instance.menu, 'mo2_handler'):
|
|
cli_instance.menu.mo2_handler.install_mo2()
|
|
else:
|
|
print(f"{COLOR_INFO}MO2 handler not available - this will be implemented in Phase 2.3{COLOR_RESET}")
|
|
input("\nPress Enter to continue...")
|
|
|
|
def _execute_legacy_recovery_menu(self, cli_instance):
|
|
"""LEGACY BRIDGE: Execute recovery menu"""
|
|
# This will be handled by the RecoveryMenuHandler
|
|
from .recovery_menu import RecoveryMenuHandler
|
|
|
|
recovery_handler = RecoveryMenuHandler()
|
|
recovery_handler.logger = self.logger
|
|
recovery_handler.show_recovery_menu(cli_instance)
|
|
|
|
def _execute_ttw_install(self, cli_instance):
|
|
"""Execute TTW installation using TTW_Linux_Installer handler"""
|
|
from ....backend.handlers.ttw_installer_handler import TTWInstallerHandler
|
|
from ....backend.models.configuration import SystemInfo
|
|
from ....shared.colors import COLOR_ERROR, COLOR_WARNING, COLOR_SUCCESS, COLOR_INFO, COLOR_PROMPT
|
|
from pathlib import Path
|
|
|
|
system_info = SystemInfo(is_steamdeck=cli_instance.system_info.is_steamdeck)
|
|
ttw_installer_handler = TTWInstallerHandler(
|
|
steamdeck=system_info.is_steamdeck,
|
|
verbose=cli_instance.verbose,
|
|
filesystem_handler=cli_instance.filesystem_handler,
|
|
config_handler=cli_instance.config_handler
|
|
)
|
|
|
|
# First check if TTW_Linux_Installer is installed
|
|
if not ttw_installer_handler.ttw_installer_installed:
|
|
print(f"\n{COLOR_WARNING}TTW_Linux_Installer is not installed. Installing TTW_Linux_Installer first...{COLOR_RESET}")
|
|
success, message = ttw_installer_handler.install_ttw_installer()
|
|
if not success:
|
|
print(f"{COLOR_ERROR}Failed to install TTW_Linux_Installer. Cannot proceed with TTW installation.{COLOR_RESET}")
|
|
print(f"{COLOR_ERROR}Error: {message}{COLOR_RESET}")
|
|
input("Press Enter to return to menu...")
|
|
return
|
|
|
|
# Check for required games
|
|
detected_games = ttw_installer_handler.path_handler.find_vanilla_game_paths()
|
|
required_games = ['Fallout 3', 'Fallout New Vegas']
|
|
missing_games = [game for game in required_games if game not in detected_games]
|
|
if missing_games:
|
|
print(f"\n{COLOR_ERROR}Missing required games: {', '.join(missing_games)}")
|
|
print(f"TTW requires both Fallout 3 and Fallout New Vegas to be installed.{COLOR_RESET}")
|
|
input("Press Enter to return to menu...")
|
|
return
|
|
|
|
# Prompt for TTW .mpi file
|
|
print(f"\n{COLOR_PROMPT}TTW Installer File (.mpi){COLOR_RESET}")
|
|
mpi_path = input(f"{COLOR_PROMPT}Path to TTW .mpi file: {COLOR_RESET}").strip()
|
|
if not mpi_path:
|
|
print(f"{COLOR_WARNING}No .mpi file specified. Cancelling.{COLOR_RESET}")
|
|
input("Press Enter to return to menu...")
|
|
return
|
|
|
|
mpi_path = Path(mpi_path).expanduser()
|
|
if not mpi_path.exists() or not mpi_path.is_file():
|
|
print(f"{COLOR_ERROR}TTW .mpi file not found: {mpi_path}{COLOR_RESET}")
|
|
input("Press Enter to return to menu...")
|
|
return
|
|
|
|
# Prompt for output directory
|
|
print(f"\n{COLOR_PROMPT}TTW Installation Directory{COLOR_RESET}")
|
|
default_output = Path.home() / "ModdedGames" / "TTW"
|
|
output_path = input(f"{COLOR_PROMPT}TTW install directory (Enter for default: {default_output}): {COLOR_RESET}").strip()
|
|
if not output_path:
|
|
output_path = default_output
|
|
else:
|
|
output_path = Path(output_path).expanduser()
|
|
|
|
# Run TTW installation
|
|
print(f"\n{COLOR_INFO}Starting TTW installation workflow...{COLOR_RESET}")
|
|
success, message = ttw_installer_handler.install_ttw_backend(mpi_path, output_path)
|
|
|
|
if success:
|
|
print(f"\n{COLOR_SUCCESS}TTW installation completed successfully!{COLOR_RESET}")
|
|
print(f"{COLOR_INFO}TTW installed to: {output_path}{COLOR_RESET}")
|
|
else:
|
|
print(f"\n{COLOR_ERROR}TTW installation failed.{COLOR_RESET}")
|
|
print(f"{COLOR_ERROR}Error: {message}{COLOR_RESET}")
|
|
input("Press Enter to return to menu...")
|
|
|
|
def _execute_nexus_authorization(self, cli_instance):
|
|
"""Execute Nexus authorization menu (OAuth or API key)"""
|
|
from ....backend.services.nexus_auth_service import NexusAuthService
|
|
from ....backend.services.api_key_service import APIKeyService
|
|
from ....shared.colors import COLOR_ERROR, COLOR_SUCCESS
|
|
|
|
auth_service = NexusAuthService()
|
|
|
|
while True:
|
|
self._clear_screen()
|
|
print_jackify_banner()
|
|
print_section_header("Nexus Mods Authorization")
|
|
|
|
# Get current auth status
|
|
authenticated, method, username = auth_service.get_auth_status()
|
|
|
|
if authenticated:
|
|
if method == 'oauth':
|
|
print(f"\n{COLOR_SUCCESS}Status: Authorized via OAuth{COLOR_RESET}")
|
|
if username:
|
|
print(f"{COLOR_INFO}Logged in as: {username}{COLOR_RESET}")
|
|
elif method == 'api_key':
|
|
print(f"\n{COLOR_WARNING}Status: Using API Key (Legacy){COLOR_RESET}")
|
|
print(f"{COLOR_INFO}Consider switching to OAuth for better security{COLOR_RESET}")
|
|
else:
|
|
print(f"\n{COLOR_WARNING}Status: Not Authorized{COLOR_RESET}")
|
|
print(f"{COLOR_INFO}You need to authorize to download mods from Nexus{COLOR_RESET}")
|
|
|
|
print(f"\n{COLOR_SELECTION}1.{COLOR_RESET} Authorize with Nexus (OAuth)")
|
|
print(f" {COLOR_ACTION}→ Opens browser for secure authorization{COLOR_RESET}")
|
|
|
|
if method == 'oauth':
|
|
print(f"{COLOR_SELECTION}2.{COLOR_RESET} Revoke OAuth Authorization")
|
|
print(f" {COLOR_ACTION}→ Remove OAuth token{COLOR_RESET}")
|
|
|
|
print(f"{COLOR_SELECTION}3.{COLOR_RESET} Set API Key (Legacy Fallback)")
|
|
print(f" {COLOR_ACTION}→ Manually enter Nexus API key{COLOR_RESET}")
|
|
|
|
if authenticated:
|
|
print(f"{COLOR_SELECTION}4.{COLOR_RESET} Clear All Authentication")
|
|
print(f" {COLOR_ACTION}→ Remove both OAuth and API key{COLOR_RESET}")
|
|
|
|
print(f"{COLOR_SELECTION}0.{COLOR_RESET} Return to Additional Tasks Menu")
|
|
|
|
selection = input(f"\n{COLOR_PROMPT}Enter your selection: {COLOR_RESET}").strip()
|
|
|
|
if selection == "1":
|
|
# OAuth authorization
|
|
print(f"\n{COLOR_INFO}Starting OAuth authorization...{COLOR_RESET}")
|
|
print(f"{COLOR_WARNING}Your browser will open shortly.{COLOR_RESET}")
|
|
print(f"{COLOR_WARNING}Please check your browser and authorize Jackify.{COLOR_RESET}")
|
|
print(f"\n{COLOR_INFO}Note: Your browser may ask permission to open 'xdg-open' or{COLOR_RESET}")
|
|
print(f"{COLOR_INFO}Jackify's protocol handler - please click 'Open' or 'Allow'.{COLOR_RESET}")
|
|
|
|
input(f"\n{COLOR_PROMPT}Press Enter to open browser...{COLOR_RESET}")
|
|
|
|
# Perform OAuth authorization
|
|
def show_message(msg):
|
|
print(f"\n{COLOR_INFO}{msg}{COLOR_RESET}")
|
|
|
|
success = auth_service.authorize_oauth(show_browser_message_callback=show_message)
|
|
|
|
if success:
|
|
print(f"\n{COLOR_SUCCESS}OAuth authorization successful!{COLOR_RESET}")
|
|
# Get username
|
|
_, _, username = auth_service.get_auth_status()
|
|
if username:
|
|
print(f"{COLOR_INFO}Authorized as: {username}{COLOR_RESET}")
|
|
else:
|
|
print(f"\n{COLOR_ERROR}OAuth authorization failed.{COLOR_RESET}")
|
|
print(f"{COLOR_INFO}You can try again or use API key as fallback.{COLOR_RESET}")
|
|
|
|
input(f"\n{COLOR_PROMPT}Press Enter to continue...{COLOR_RESET}")
|
|
|
|
elif selection == "2" and method == 'oauth':
|
|
# Revoke OAuth
|
|
print(f"\n{COLOR_WARNING}Are you sure you want to revoke OAuth authorization?{COLOR_RESET}")
|
|
confirm = input(f"{COLOR_PROMPT}Type 'yes' to confirm: {COLOR_RESET}").strip().lower()
|
|
|
|
if confirm == 'yes':
|
|
if auth_service.revoke_oauth():
|
|
print(f"\n{COLOR_SUCCESS}OAuth authorization revoked.{COLOR_RESET}")
|
|
else:
|
|
print(f"\n{COLOR_ERROR}Failed to revoke OAuth authorization.{COLOR_RESET}")
|
|
else:
|
|
print(f"\n{COLOR_INFO}Cancelled.{COLOR_RESET}")
|
|
|
|
input(f"\n{COLOR_PROMPT}Press Enter to continue...{COLOR_RESET}")
|
|
|
|
elif selection == "3":
|
|
# Set API key
|
|
print(f"\n{COLOR_INFO}Enter your Nexus API Key{COLOR_RESET}")
|
|
print(f"{COLOR_INFO}(Get it from: https://www.nexusmods.com/users/myaccount?tab=api){COLOR_RESET}")
|
|
|
|
api_key = input(f"\n{COLOR_PROMPT}API Key: {COLOR_RESET}").strip()
|
|
|
|
if api_key:
|
|
if auth_service.save_api_key(api_key):
|
|
print(f"\n{COLOR_SUCCESS}API key saved successfully.{COLOR_RESET}")
|
|
|
|
# Optionally validate
|
|
print(f"\n{COLOR_INFO}Validating API key...{COLOR_RESET}")
|
|
valid, result = auth_service.validate_api_key(api_key)
|
|
|
|
if valid:
|
|
print(f"{COLOR_SUCCESS}API key validated successfully!{COLOR_RESET}")
|
|
print(f"{COLOR_INFO}Username: {result}{COLOR_RESET}")
|
|
else:
|
|
print(f"{COLOR_WARNING}Warning: API key validation failed: {result}{COLOR_RESET}")
|
|
print(f"{COLOR_INFO}Key saved, but may not work correctly.{COLOR_RESET}")
|
|
else:
|
|
print(f"\n{COLOR_ERROR}Failed to save API key.{COLOR_RESET}")
|
|
else:
|
|
print(f"\n{COLOR_INFO}Cancelled.{COLOR_RESET}")
|
|
|
|
input(f"\n{COLOR_PROMPT}Press Enter to continue...{COLOR_RESET}")
|
|
|
|
elif selection == "4" and authenticated:
|
|
# Clear all authentication
|
|
print(f"\n{COLOR_WARNING}Are you sure you want to clear ALL authentication?{COLOR_RESET}")
|
|
print(f"{COLOR_WARNING}This will remove both OAuth token and API key.{COLOR_RESET}")
|
|
confirm = input(f"{COLOR_PROMPT}Type 'yes' to confirm: {COLOR_RESET}").strip().lower()
|
|
|
|
if confirm == 'yes':
|
|
if auth_service.clear_all_auth():
|
|
print(f"\n{COLOR_SUCCESS}All authentication cleared.{COLOR_RESET}")
|
|
else:
|
|
print(f"\n{COLOR_INFO}No authentication to clear.{COLOR_RESET}")
|
|
else:
|
|
print(f"\n{COLOR_INFO}Cancelled.{COLOR_RESET}")
|
|
|
|
input(f"\n{COLOR_PROMPT}Press Enter to continue...{COLOR_RESET}")
|
|
|
|
elif selection == "0":
|
|
break
|
|
else:
|
|
print(f"\n{COLOR_ERROR}Invalid selection.{COLOR_RESET}")
|
|
time.sleep(1)
|