""" UI交互模块 - 处理用户界面和交互逻辑 """ import os from .git_operations import GitOperations from .remote_manager import RemoteManager from .utils import OutputFormatter, InputValidator, Colors, PlatformUtils class GitManagerUI: """Git管理器UI""" def __init__(self, work_dir: str = ""): """ 初始化UI Args: work_dir: 工作目录（默认为当前目录） """ self.git_ops = GitOperations(work_dir) self.remote_mgr = RemoteManager() def select_work_directory(self): """让用户选择工作目录""" from .utils import Colors print(f"\n{Colors.BRIGHT_CYAN}{'=' * 60}") print(f"{Colors.BRIGHT_MAGENTA}{Colors.BOLD} QuickGit - 萌芽一键Git管理工具 v1.0{Colors.ENDC}") print(f"{Colors.BRIGHT_CYAN}{'=' * 60}{Colors.ENDC}") # 获取当前目录 current_dir = os.getcwd() # 显示平台信息和路径示例 platform = PlatformUtils.get_platform_name() print(f"{Colors.BRIGHT_YELLOW}当前平台:{Colors.ENDC} {Colors.WHITE}{platform}{Colors.ENDC}") print(f"{Colors.BRIGHT_YELLOW}脚本目录:{Colors.ENDC} {Colors.WHITE}{current_dir}{Colors.ENDC}") print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}") # 显示路径输入提示 OutputFormatter.info("请输入要管理的Git仓库目录") # 根据平台显示不同的示例 if PlatformUtils.is_windows(): OutputFormatter.tip("Windows路径示例: C:\\Users\\YourName\\project") OutputFormatter.tip("或使用相对路径: .\\myproject") else: OutputFormatter.tip("Linux/macOS路径示例: /home/yourname/project") OutputFormatter.tip("或使用相对路径: ./myproject 或 ~/project") print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}") # 获取用户输入的目录 work_dir = InputValidator.get_directory( f"{Colors.BRIGHT_CYAN}>> 请输入目录路径{Colors.ENDC}", default=current_dir ) # 切换到工作目录 try: self.git_ops.change_directory(work_dir) OutputFormatter.success(f"已切换到工作目录: {work_dir}") except Exception as e: OutputFormatter.error(f"切换目录失败: {str(e)}") return False return True def show_welcome(self): """显示欢迎信息""" from .utils import Colors # 简洁的标题 print(f"\n{Colors.BRIGHT_CYAN}{'=' * 60}") print(f"{Colors.BRIGHT_MAGENTA}{Colors.BOLD} QuickGit - 萌芽一键Git管理工具 v1.0{Colors.ENDC}") print(f"{Colors.BRIGHT_CYAN}{'=' * 60}{Colors.ENDC}") # 显示当前状态 is_repo = self.git_ops.is_git_repo() repo_status = f"{Colors.BRIGHT_GREEN}[已初始化]{Colors.ENDC}" if is_repo else f"{Colors.BRIGHT_RED}[未初始化]{Colors.ENDC}" print(f"{Colors.BRIGHT_YELLOW}当前目录:{Colors.ENDC} {Colors.WHITE}{self.git_ops.current_dir}{Colors.ENDC}") print(f"{Colors.BRIGHT_YELLOW}Git状态:{Colors.ENDC} {repo_status}") print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}") def show_main_menu(self): """显示主菜单""" print(f"\n{Colors.BRIGHT_MAGENTA}{Colors.BOLD}>> 主菜单{Colors.ENDC}") print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}") OutputFormatter.menu_item(1, "初始化Git仓库") OutputFormatter.menu_item(2, "提交并推送更改") OutputFormatter.menu_item(3, "从远程仓库拉取") OutputFormatter.menu_item(4, "查看仓库状态") OutputFormatter.menu_item(5, "管理远程仓库") OutputFormatter.menu_item(6, "退出程序") print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}") # 显示永久提示 OutputFormatter.tip("提交代码前建议先拉取最新代码，减少代码冲突") OutputFormatter.tip("使用SSH进行Git提交更方便快捷和安全") print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}") def handle_init_repo(self): """处理初始化仓库""" OutputFormatter.header("初始化Git仓库") if self.git_ops.init_repo(): # 配置远程仓库 self.remote_mgr.configure_remotes_interactive() def handle_commit_and_push(self): """处理提交并推送""" OutputFormatter.header("提交并推送更改") if not self.git_ops.is_git_repo(): OutputFormatter.error("当前目录不是Git仓库，请先初始化") return # 检查是否有更改 OutputFormatter.status('running', "检查文件更改中...") if not self.git_ops.has_changes(): OutputFormatter.warning("没有文件更改，无需提交") return # 显示更改 print(f"\n{Colors.BRIGHT_YELLOW}>> 当前更改的文件：{Colors.ENDC}") _, status = self.git_ops.get_status() print(f"{Colors.CYAN}{status}{Colors.ENDC}") print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}") # 输入提交信息 commit_msg = InputValidator.get_input( f"{Colors.BRIGHT_CYAN}>> 请输入提交信息 (回车使用默认): {Colors.ENDC}", allow_empty=True ) # 添加并提交 if not self.git_ops.add_all(): return if not self.git_ops.commit(commit_msg or ""): return # 推送到远程仓库 selected_remotes = self.remote_mgr.select_remote_for_push() for remote in selected_remotes: self.git_ops.push(remote) def handle_pull(self): """处理拉取""" OutputFormatter.header("从远程仓库拉取") if not self.git_ops.is_git_repo(): OutputFormatter.error("当前目录不是Git仓库") return remote = self.remote_mgr.select_remote_for_pull() if remote: self.git_ops.pull(remote) def handle_show_status(self): """处理显示状态""" OutputFormatter.header("仓库状态") if not self.git_ops.is_git_repo(): OutputFormatter.error("当前目录不是Git仓库") return self.git_ops.show_status() # 显示远程仓库 self.remote_mgr.show_remotes() def handle_manage_remotes(self): """处理远程仓库管理""" OutputFormatter.header("管理远程仓库") if not self.git_ops.is_git_repo(): OutputFormatter.error("当前目录不是Git仓库") return while True: print(f"\n{Colors.BRIGHT_BLUE}{Colors.BOLD}>> 远程仓库管理{Colors.ENDC}") print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}") OutputFormatter.menu_item(1, "查看远程仓库") OutputFormatter.menu_item(2, "添加GitHub远程仓库") OutputFormatter.menu_item(3, "添加Gitea远程仓库") OutputFormatter.menu_item(4, "删除远程仓库") OutputFormatter.menu_item(5, "返回主菜单") print(f"{Colors.CYAN}{'-' * 60}{Colors.ENDC}") choice = InputValidator.get_choice( f"{Colors.BRIGHT_CYAN}>> 请选择 [1-5]: {Colors.ENDC}", range(1, 6) ) if choice == 1: self.remote_mgr.show_remotes() elif choice == 2: self.remote_mgr.add_github_remote() elif choice == 3: self.remote_mgr.add_gitea_remote() elif choice == 4: self.remote_mgr.remove_remote() elif choice == 5: break def run(self): """运行主程序""" # 首先选择工作目录 if not self.select_work_directory(): return # 显示欢迎信息 self.show_welcome() while True: self.show_main_menu() choice = InputValidator.get_choice( f"{Colors.BRIGHT_MAGENTA}>> 请输入选项 [1-6]: {Colors.ENDC}", range(1, 7) ) if choice == 1: self.handle_init_repo() elif choice == 2: self.handle_commit_and_push() elif choice == 3: self.handle_pull() elif choice == 4: self.handle_show_status() elif choice == 5: self.handle_manage_remotes() elif choice == 6: print(f"\n{Colors.BRIGHT_GREEN}{'=' * 60}") print(f"{Colors.BRIGHT_GREEN}{Colors.BOLD} 感谢使用QuickGit！祝您工作愉快！{Colors.ENDC}") print(f"{Colors.BRIGHT_GREEN}{'=' * 60}{Colors.ENDC}\n") break