# 推薦使用Anaconda創(chuàng)建虛擬環(huán)境
conda create -n wechat python=3.8
conda install -c anaconda pyqt=5.15

class WeChatFinder(QThread):
    def search_registry(self):
        # 從HKCU讀取安裝路徑
        with winreg.OpenKey(winreg.HKEY_CURRENT_USER, r"Software\Tencent\WeChat") as key:
            ...
    
    def search_path(self, path):
        # 使用os.walk進(jìn)行深度搜索
        for root, dirs, files in os.walk(path):
            if "WeChat.exe" in files:
                return os.path.join(root, "WeChat.exe")

def launch_wechat_instance(self, exe_path):
    try:
        # 根據(jù)模式選擇啟動(dòng)方式
        if mode == "sandbox":
            subprocess.Popen(["sandboxie", exe_path])
        else:
            subprocess.Popen([exe_path], creationflags=subprocess.CREATE_NEW_PROCESS_GROUP)
    except subprocess.CalledProcessError as e:
        logging.error(f"進(jìn)程創(chuàng)建失敗: {e.output.decode('gbk')}")

import sys
import os
import winreg
import logging
import ctypes
import json
from PyQt5.QtWidgets import (QApplication, QWidget, QVBoxLayout, QHBoxLayout, 
                            QPushButton, QLabel, QLineEdit, QSpinBox, 
                            QListWidget, QMessageBox, QProgressBar, QFileDialog,
                            QCheckBox, QComboBox)
from PyQt5.QtCore import Qt, QThread, pyqtSignal, QSettings
from PyQt5.QtGui import QIcon, QFont

class WeChatFinder(QThread):
    found_wechat = pyqtSignal(str)
    search_complete = pyqtSignal()
    progress_update = pyqtSignal(int)
    
    def __init__(self):
        super().__init__()
        self._is_running = True
        
    def run(self):
        try:
            # 從注冊(cè)表查找
            self.search_registry()
            
            # 從常見路徑查找
            common_paths = self.get_common_paths()
            total_paths = len(common_paths)
            
            for i, path in enumerate(common_paths):
                if not self._is_running:
                    break
                self.search_path(path)
                self.progress_update.emit(int((i+1)/total_paths*100))
                
        except Exception as e:
            logging.error(f"搜索微信時(shí)出錯(cuò): {str(e)}", exc_info=True)
        finally:
            self.search_complete.emit()
    
    def search_registry(self):
        try:
            with winreg.OpenKey(winreg.HKEY_CURRENT_USER, r"Software\Tencent\WeChat") as key:
                install_path, _ = winreg.QueryValueEx(key, "InstallPath")
                if install_path:
                    exe_path = os.path.join(install_path, "WeChat.exe")
                    if os.path.exists(exe_path):
                        self.found_wechat.emit(exe_path)
        except WindowsError:
            pass
    
    def search_path(self, path):
        if os.path.exists(path):
            for root, dirs, files in os.walk(path):
                if not self._is_running:
                    break
                if "WeChat.exe" in files:
                    exe_path = os.path.join(root, "WeChat.exe")
                    if self.verify_wechat_exe(exe_path):
                        self.found_wechat.emit(exe_path)
                    return
    
    def get_common_paths(self):
        return [
            os.path.expanduser("~") + "\\AppData\\Local\\Tencent\\WeChat",
            "C:\\Program Files (x86)\\Tencent\\WeChat",
            "D:\\Program Files\\Tencent\\WeChat",
            "C:\\Program Files\\Tencent\\WeChat",
            "D:\\Program Files (x86)\\Tencent\\WeChat",
            os.path.join(os.environ.get("ProgramFiles", ""), "Tencent", "WeChat"),
            os.path.join(os.environ.get("ProgramFiles(x86)", ""), "Tencent", "WeChat")
        ]
    
    def verify_wechat_exe(self, exe_path):
        # 這里可以添加驗(yàn)證微信簽名等安全檢查
        return True
    
    def stop(self):
        self._is_running = False

class WeChatLauncher(QWidget):
    def __init__(self):
        super().__init__()
        self.settings = QSettings("WeChatMultiLauncher", "Settings")
        self.init_logging()
        self.initUI()
        self.load_settings()
        self.start_wechat_finder()
        
    def init_logging(self):
        logging.basicConfig(
            filename='wechat_launcher.log',
            level=logging.INFO,
            format='%(asctime)s - %(levelname)s - %(message)s'
        )
    
    def initUI(self):
        self.setWindowTitle('?? 微信多開助手')
        self.setWindowIcon(QIcon('wechat.ico'))
        self.setFixedSize(700, 600)
        
        main_layout = QVBoxLayout()
        main_layout.setSpacing(15)
        main_layout.setContentsMargins(20, 20, 20, 20)
        
        # 標(biāo)題
        title_label = QLabel('?? 微信多開助手')
        title_label.setFont(QFont('Microsoft YaHei', 18, QFont.Bold))
        title_label.setAlignment(Qt.AlignCenter)
        main_layout.addWidget(title_label)
        
        # 分隔線
        main_layout.addWidget(self.create_separator())
        
        # 微信路徑部分
        main_layout.addLayout(self.create_path_section())
        
        # 多開設(shè)置部分
        main_layout.addLayout(self.create_settings_section())
        
        # 高級(jí)選項(xiàng)
        main_layout.addLayout(self.create_advanced_section())
        
        # 進(jìn)度條
        self.progress_bar = QProgressBar()
        self.progress_bar.setRange(0, 100)
        self.progress_bar.setValue(0)
        self.progress_bar.setTextVisible(True)
        self.progress_bar.hide()
        main_layout.addWidget(self.progress_bar)
        
        # 按鈕部分
        main_layout.addLayout(self.create_button_section())
        
        # 狀態(tài)欄
        self.status_label = QLabel('? 正在搜索微信安裝路徑...')
        self.status_label.setAlignment(Qt.AlignCenter)
        main_layout.addWidget(self.status_label)
        
        self.setLayout(main_layout)
        
    def create_separator(self):
        separator = QLabel('━' * 50)
        separator.setAlignment(Qt.AlignCenter)
        return separator
    
    def create_path_section(self):
        path_layout = QVBoxLayout()
        
        path_title = QLabel('?? 微信路徑:')
        path_title.setFont(QFont('Microsoft YaHei', 10))
        path_layout.addWidget(path_title)
        
        self.path_list = QListWidget()
        self.path_list.setStyleSheet("QListWidget { border: 1px solid #ccc; border-radius: 5px; }")
        path_layout.addWidget(self.path_list)
        
        # 手動(dòng)輸入路徑
        manual_layout = QHBoxLayout()
        manual_label = QLabel('?? 手動(dòng)指定路徑:')
        self.manual_path_input = QLineEdit()
        self.manual_path_input.setPlaceholderText('輸入WeChat.exe完整路徑...')
        browse_btn = QPushButton('瀏覽...')
        browse_btn.clicked.connect(self.browse_for_wechat)
        manual_layout.addWidget(manual_label)
        manual_layout.addWidget(self.manual_path_input)
        manual_layout.addWidget(browse_btn)
        path_layout.addLayout(manual_layout)
        
        return path_layout
    
    def create_settings_section(self):
        settings_layout = QHBoxLayout()
        
        settings_label = QLabel('?? 多開設(shè)置:')
        settings_layout.addWidget(settings_label)
        
        count_label = QLabel('啟動(dòng)數(shù)量:')
        self.count_spin = QSpinBox()
        self.count_spin.setRange(1, 20)
        self.count_spin.setValue(2)
        settings_layout.addWidget(count_label)
        settings_layout.addWidget(self.count_spin)
        
        self.minimize_check = QCheckBox('啟動(dòng)后最小化')
        settings_layout.addWidget(self.minimize_check)
        
        settings_layout.addStretch()
        
        return settings_layout
    
    def create_advanced_section(self):
        advanced_layout = QHBoxLayout()
        
        advanced_label = QLabel('?? 高級(jí)選項(xiàng):')
        advanced_layout.addWidget(advanced_label)
        
        mode_label = QLabel('多開模式:')
        self.mode_combo = QComboBox()
        self.mode_combo.addItems(['普通多開', '沙盒模式', '不同配置'])
        advanced_layout.addWidget(mode_label)
        advanced_layout.addWidget(self.mode_combo)
        
        self.admin_check = QCheckBox('以管理員身份運(yùn)行')
        advanced_layout.addWidget(self.admin_check)
        
        advanced_layout.addStretch()
        
        return advanced_layout
    
    def create_button_section(self):
        button_layout = QHBoxLayout()
        
        self.launch_btn = QPushButton('?? 啟動(dòng)微信')
        self.launch_btn.setStyleSheet("QPushButton { background-color: #07C160; color: white; font-weight: bold; }")
        self.launch_btn.clicked.connect(self.launch_multiple_wechats)
        button_layout.addWidget(self.launch_btn)
        
        refresh_btn = QPushButton('?? 重新搜索')
        refresh_btn.clicked.connect(self.start_wechat_finder)
        button_layout.addWidget(refresh_btn)
        
        settings_btn = QPushButton('?? 設(shè)置')
        settings_btn.clicked.connect(self.show_settings)
        button_layout.addWidget(settings_btn)
        
        exit_btn = QPushButton('? 退出')
        exit_btn.clicked.connect(self.close)
        button_layout.addWidget(exit_btn)
        
        return button_layout
    
    def load_settings(self):
        # 加載保存的設(shè)置
        self.count_spin.setValue(self.settings.value("launch_count", 2, type=int))
        self.minimize_check.setChecked(self.settings.value("minimize", False, type=bool))
        self.admin_check.setChecked(self.settings.value("admin", False, type=bool))
        self.mode_combo.setCurrentIndex(self.settings.value("mode", 0, type=int))
        
        last_path = self.settings.value("last_path", "")
        if last_path:
            self.manual_path_input.setText(last_path)
    
    def save_settings(self):
        # 保存當(dāng)前設(shè)置
        self.settings.setValue("launch_count", self.count_spin.value())
        self.settings.setValue("minimize", self.minimize_check.isChecked())
        self.settings.setValue("admin", self.admin_check.isChecked())
        self.settings.setValue("mode", self.mode_combo.currentIndex())
        
        selected_path = self.get_selected_path()
        if selected_path:
            self.settings.setValue("last_path", selected_path)
    
    def start_wechat_finder(self):
        self.path_list.clear()
        self.status_label.setText('? 正在搜索微信安裝路徑...')
        self.progress_bar.show()
        self.progress_bar.setValue(0)
        
        self.finder = WeChatFinder()
        self.finder.found_wechat.connect(self.add_wechat_path)
        self.finder.search_complete.connect(self.on_search_complete)
        self.finder.progress_update.connect(self.progress_bar.setValue)
        self.finder.start()
        
    def add_wechat_path(self, path):
        if path not in [self.path_list.item(i).text() for i in range(self.path_list.count())]:
            self.path_list.addItem(path)
            
    def on_search_complete(self):
        self.progress_bar.hide()
        if self.path_list.count() == 0:
            self.status_label.setText('?? 未找到微信安裝路徑，請(qǐng)手動(dòng)指定')
        else:
            self.status_label.setText(f'? 找到 {self.path_list.count()} 個(gè)微信安裝路徑')
            self.path_list.setCurrentRow(0)
    
    def browse_for_wechat(self):
        file_path, _ = QFileDialog.getOpenFileName(
            self, 
            "選擇WeChat.exe", 
            "", 
            "Executable Files (*.exe)"
        )
        
        if file_path:
            self.manual_path_input.setText(file_path)
            if file_path not in [self.path_list.item(i).text() for i in range(self.path_list.count())]:
                self.path_list.addItem(file_path)
                self.path_list.setCurrentRow(self.path_list.count()-1)
    
    def get_selected_path(self):
        if self.path_list.currentItem():
            return self.path_list.currentItem().text()
        elif self.manual_path_input.text() and os.path.exists(self.manual_path_input.text()):
            return self.manual_path_input.text()
        return None
    
    def is_admin(self):
        try:
            return ctypes.windll.shell32.IsUserAnAdmin()
        except:
            return False
    
    def launch_multiple_wechats(self):
        selected_path = self.get_selected_path()
        if not selected_path:
            self.show_error("請(qǐng)先選擇或輸入有效的微信路徑！")
            return
        
        if self.admin_check.isChecked() and not self.is_admin():
            self.show_error("請(qǐng)以管理員身份運(yùn)行此程序！")
            return
        
        count = self.count_spin.value()
        self.show_loading(f"正在啟動(dòng) {count} 個(gè)微信實(shí)例...")
        
        success_count = 0
        for i in range(count):
            if self.launch_wechat_instance(selected_path):
                success_count += 1
                self.progress_bar.setValue(int((i+1)/count*100))
                QApplication.processEvents()
            else:
                break
        
        self.show_success(f"已啟動(dòng) {success_count} 個(gè)微信實(shí)例")
        if self.minimize_check.isChecked():
            self.showMinimized()
        
        self.save_settings()
    
    def launch_wechat_instance(self, exe_path):
        try:
            if self.mode_combo.currentIndex() == 1:  # 沙盒模式
                # 這里可以添加沙盒啟動(dòng)邏輯
                subprocess.Popen([exe_path])
            elif self.mode_combo.currentIndex() == 2:  # 不同配置
                # 這里可以添加不同配置啟動(dòng)邏輯
                subprocess.Popen([exe_path])
            else:  # 普通模式
                subprocess.Popen([exe_path])
            return True
        except Exception as e:
            logging.error(f"啟動(dòng)微信失敗: {str(e)}", exc_info=True)
            self.show_error(f"啟動(dòng)微信失敗: {str(e)}")
            return False
    
    def show_error(self, message):
        QMessageBox.critical(self, "錯(cuò)誤", message)
        self.status_label.setText(f'? {message}')
        self.progress_bar.hide()
    
    def show_loading(self, message):
        self.status_label.setText(f'? {message}')
        self.progress_bar.show()
        self.progress_bar.setValue(0)
    
    def show_success(self, message):
        self.status_label.setText(f'? {message}')
        self.progress_bar.hide()
    
    def show_settings(self):
        # 這里可以擴(kuò)展為更詳細(xì)的設(shè)置對(duì)話框
        QMessageBox.information(self, "設(shè)置", "更多設(shè)置功能將在未來版本中添加")
    
    def closeEvent(self, event):
        self.save_settings()
        if hasattr(self, 'finder') and self.finder.isRunning():
            self.finder.stop()
            self.finder.wait()
        event.accept()

if __name__ == '__main__':
    app = QApplication(sys.argv)
    app.setStyle('Fusion')
    
    # 設(shè)置全局字體
    font = QFont('Microsoft YaHei', 10)
    app.setFont(font)
    
    launcher = WeChatLauncher()
    launcher.show()
    sys.exit(app.exec_())