import os
import sys
import re
import json
import fnmatch
from PyQt5.QtWidgets import (
    QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit,
    QPushButton, QRadioButton, QButtonGroup, QGroupBox, QFileDialog, QTextEdit,
    QComboBox, QMessageBox, QCheckBox, QListWidget, QListWidgetItem, QInputDialog
)
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QFont, QPalette, QColor

# 獲取腳本所在目錄
if getattr(sys, 'frozen', False):
    # 如果是打包后的可執(zhí)行文件
    APP_DIR = os.path.dirname(sys.executable)
else:
    # 如果是腳本文件
    APP_DIR = os.path.dirname(os.path.abspath(__file__))

# 修改配置文件路徑為腳本所在目錄
CONFIG_FILE = os.path.join(APP_DIR, "config.json")

class FileSearchApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("目錄/文件遞歸檢索工具")
        self.setGeometry(300, 300, 800, 650)
        
        # 初始化變量
        self.ignore_configs = []
        self.current_ignore_config = {"name": "默認配置", "patterns": []}
        self.load_config()
        
        # 創(chuàng)建UI
        self.init_ui()

class FileSearchApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("目錄/文件遞歸檢索工具 V1.0 by:Sunf10wer")
        self.setGeometry(300, 300, 800, 650)
        
        # 初始化變量
        self.ignore_configs = []
        self.current_ignore_config = {"name": "默認配置", "patterns": []}
        self.load_config()
        
        # 創(chuàng)建UI
        self.init_ui()
        
    def init_ui(self):
        # 主布局
        main_widget = QWidget()
        main_layout = QVBoxLayout()
        main_widget.setLayout(main_layout)
        self.setCentralWidget(main_widget)
        
        # 設置標題樣式
        title_label = QLabel("目錄/文件遞歸檢索工具")
        title_font = QFont("Arial", 16, QFont.Bold)
        title_label.setFont(title_font)
        title_label.setAlignment(Qt.AlignCenter)
        title_label.setStyleSheet("color: #FFFFFF; padding: 10px;")
        main_layout.addWidget(title_label)
        
        # 目錄選擇
        dir_layout = QHBoxLayout()
        dir_label = QLabel("目標目錄:")
        self.dir_entry = QLineEdit()
        self.dir_entry.setPlaceholderText("請選擇或輸入要檢索的目錄...")
        browse_button = QPushButton("瀏覽...")
        browse_button.clicked.connect(self.browse_directory)
        
        dir_layout.addWidget(dir_label)
        dir_layout.addWidget(self.dir_entry, 4)
        dir_layout.addWidget(browse_button, 1)
        main_layout.addLayout(dir_layout)
        
        # 輸出文件選擇
        output_layout = QHBoxLayout()
        output_label = QLabel("輸出文件:")
        self.output_entry = QLineEdit()
        self.output_entry.setPlaceholderText("輸出文件名...")
        output_browse_button = QPushButton("瀏覽...")
        output_browse_button.clicked.connect(self.browse_output_file)
        
        output_layout.addWidget(output_label)
        output_layout.addWidget(self.output_entry, 4)
        output_layout.addWidget(output_browse_button, 1)
        main_layout.addLayout(output_layout)
        
        # 檢索類型選擇
        search_type_group = QGroupBox("檢索類型")
        search_layout = QHBoxLayout()
        
        self.folder_radio = QRadioButton("僅文件夾")
        self.file_radio = QRadioButton("僅文件")
        self.both_radio = QRadioButton("文件和文件夾")
        self.both_radio.setChecked(True)
        
        self.search_type_group = QButtonGroup()
        self.search_type_group.addButton(self.folder_radio)
        self.search_type_group.addButton(self.file_radio)
        self.search_type_group.addButton(self.both_radio)
        
        # 文件后綴過濾
        suffix_layout = QHBoxLayout()
        suffix_label = QLabel("文件后綴(用分號分隔):")
        self.suffix_entry = QLineEdit()
        self.suffix_entry.setPlaceholderText("例如: .txt; .py; .jpg")
        
        suffix_layout.addWidget(suffix_label)
        suffix_layout.addWidget(self.suffix_entry)
        
        search_layout.addWidget(self.folder_radio)
        search_layout.addWidget(self.file_radio)
        search_layout.addWidget(self.both_radio)
        search_layout.addStretch()
        
        search_type_group.setLayout(search_layout)
        main_layout.addWidget(search_type_group)
        main_layout.addLayout(suffix_layout)
        
        # 屏蔽詞管理
        ignore_group = QGroupBox("屏蔽詞管理")
        ignore_layout = QVBoxLayout()
        
        # 屏蔽詞配置選擇
        config_layout = QHBoxLayout()
        config_label = QLabel("當前配置:")
        self.config_combo = QComboBox()
        self.config_combo.setMinimumWidth(150)
        self.config_combo.currentIndexChanged.connect(self.config_selected)
        
        new_config_btn = QPushButton("新建配置")
        new_config_btn.clicked.connect(self.create_new_config)
        
        config_layout.addWidget(config_label)
        config_layout.addWidget(self.config_combo, 1)
        config_layout.addWidget(new_config_btn)
        
        # 屏蔽詞列表
        ignore_list_layout = QVBoxLayout()
        list_label = QLabel("屏蔽詞列表(支持通配符，如 *.tmp; backup_*)")
        self.ignore_list = QListWidget()
        self.ignore_list.setAlternatingRowColors(True)
        
        add_btn = QPushButton("添加屏蔽詞")
        add_btn.clicked.connect(self.add_ignore_pattern)
        remove_btn = QPushButton("移除選中")
        remove_btn.clicked.connect(self.remove_selected_pattern)
        
        list_btn_layout = QHBoxLayout()
        list_btn_layout.addWidget(add_btn)
        list_btn_layout.addWidget(remove_btn)
        
        ignore_list_layout.addWidget(list_label)
        ignore_list_layout.addWidget(self.ignore_list)
        ignore_list_layout.addLayout(list_btn_layout)
        
        ignore_layout.addLayout(config_layout)
        ignore_layout.addLayout(ignore_list_layout)
        ignore_group.setLayout(ignore_layout)
        main_layout.addWidget(ignore_group)
        
        # 操作按鈕
        button_layout = QHBoxLayout()
        self.search_btn = QPushButton("開始檢索")
        self.search_btn.setStyleSheet(
            "background-color: #3498db; color: white; font-weight: bold; padding: 8px;"
        )
        self.search_btn.clicked.connect(self.start_search)
        
        export_btn = QPushButton("導出配置")
        export_btn.clicked.connect(self.export_config)
        
        button_layout.addStretch()
        button_layout.addWidget(self.search_btn)
        button_layout.addWidget(export_btn)
        button_layout.addStretch()
        
        main_layout.addLayout(button_layout)
        
        # 狀態(tài)欄
        self.status_bar = self.statusBar()
        self.status_label = QLabel("就緒")
        self.status_bar.addWidget(self.status_label)
        
        # 更新UI
        self.update_config_combo()
        self.update_ignore_list()
        
        # 連接信號
        self.dir_entry.textChanged.connect(self.update_output_filename)
        
    def load_config(self):
        """從配置文件加載屏蔽詞配置"""
        if os.path.exists(CONFIG_FILE):
            try:
                with open(CONFIG_FILE, 'r', encoding='utf-8') as f:
                    self.ignore_configs = json.load(f)
                # 確保至少有一個默認配置
                if not any(cfg['name'] == '默認配置' for cfg in self.ignore_configs):
                    self.ignore_configs.insert(0, {"name": "默認配置", "patterns": []})
            except:
                self.ignore_configs = [{"name": "默認配置", "patterns": []}]
        else:
            self.ignore_configs = [{"name": "默認配置", "patterns": []}]
        
        self.current_ignore_config = self.ignore_configs[0]
    
    def save_config(self):
        """保存屏蔽詞配置到文件"""
        try:
            with open(CONFIG_FILE, 'w', encoding='utf-8') as f:
                json.dump(self.ignore_configs, f, ensure_ascii=False, indent=2)
            return True
        except Exception as e:
            QMessageBox.critical(self, "保存錯誤", f"保存配置時出錯: {str(e)}")
            return False
    
    def update_config_combo(self):
        """更新配置下拉框"""
        self.config_combo.clear()
        for config in self.ignore_configs:
            self.config_combo.addItem(config['name'])
        
        # 選擇當前配置
        current_index = next(
            (i for i, config in enumerate(self.ignore_configs) 
             if config['name'] == self.current_ignore_config['name']),
            0
        )
        self.config_combo.setCurrentIndex(current_index)
    
    def update_ignore_list(self):
        """更新屏蔽詞列表"""
        self.ignore_list.clear()
        for pattern in self.current_ignore_config['patterns']:
            self.ignore_list.addItem(pattern)
    
    def config_selected(self, index):
        """配置選擇改變事件"""
        if 0 <= index < len(self.ignore_configs):
            self.current_ignore_config = self.ignore_configs[index]
            self.update_ignore_list()
    
    def create_new_config(self):
        """創(chuàng)建新的屏蔽詞配置"""
        name, ok = QInputDialog.getText(
            self, "新建配置", "輸入配置名稱:", 
            text=f"配置{len(self.ignore_configs)+1}"
        )
        if ok and name:
            # 檢查名稱是否已存在
            if any(cfg['name'] == name for cfg in self.ignore_configs):
                QMessageBox.warning(self, "名稱沖突", f"配置名 '{name}' 已存在！")
                return
            
            new_config = {"name": name, "patterns": []}
            self.ignore_configs.append(new_config)
            self.current_ignore_config = new_config
            self.update_config_combo()
            self.update_ignore_list()
            self.save_config()
    
    def add_ignore_pattern(self):
        """添加屏蔽詞"""
        pattern, ok = QInputDialog.getText(
            self, "添加屏蔽詞", "請輸入屏蔽詞(支持通配符):"
        )
        if ok and pattern:
            if pattern not in self.current_ignore_config['patterns']:
                self.current_ignore_config['patterns'].append(pattern)
                self.update_ignore_list()
                self.save_config()
    
    def remove_selected_pattern(self):
        """移除選中的屏蔽詞"""
        selected_items = self.ignore_list.selectedItems()
        if not selected_items:
            return
        
        for item in selected_items:
            pattern = item.text()
            if pattern in self.current_ignore_config['patterns']:
                self.current_ignore_config['patterns'].remove(pattern)
        
        self.update_ignore_list()
        self.save_config()
    
    def browse_directory(self):
        """瀏覽目錄"""
        directory = QFileDialog.getExistingDirectory(self, "選擇目錄")
        if directory:
            self.dir_entry.setText(directory)
            self.update_output_filename()
    
    def browse_output_file(self):
        """瀏覽輸出文件"""
        # 獲取默認輸出路徑
        default_path = self.output_entry.text() or self.get_default_output_path()
        
        # 確保默認路徑包含文件名
        if os.path.isdir(default_path):
            default_path = os.path.join(default_path, self.get_default_filename())
        
        # 打開文件對話框
        file_path, _ = QFileDialog.getSaveFileName(
            self, "保存結果", default_path, "文本文件 (*.txt)"
        )
        
        if file_path:
            # 確保文件擴展名正確
            if not file_path.endswith('.txt'):
                file_path += '.txt'
            self.output_entry.setText(file_path)
    
    def get_default_output_path(self):
        """獲取默認輸出目錄"""
        # 優(yōu)先使用目標目錄所在位置
        if self.dir_entry.text():
            return os.path.dirname(self.dir_entry.text())
        
        # 使用當前工作目錄作為備選
        return os.getcwd()
    
    def update_output_filename(self):
        """當目錄改變時更新默認輸出文件名"""
        # 僅當輸出框為空時更新
        if not self.output_entry.text():
            self.output_entry.setText(self.get_default_output_path())
    
    def get_default_filename(self):
        """獲取默認文件名"""
        directory = self.dir_entry.text()
        if directory:
            # 獲取目錄名作為文件名基礎
            base_name = os.path.basename(directory) or "root"
            return f"{base_name}_檢索結果.txt"
        return "檢索結果.txt"
    
    def export_config(self):
        """導出當前配置"""
        if not self.current_ignore_config['patterns']:
            QMessageBox.information(self, "導出配置", "當前配置沒有屏蔽詞！")
            return
        
        file_path, _ = QFileDialog.getSaveFileName(
            self, "導出配置", "", "JSON文件 (*.json)"
        )
        if file_path:
            if not file_path.endswith('.json'):
                file_path += '.json'
            
            try:
                with open(file_path, 'w', encoding='utf-8') as f:
                    json.dump(self.current_ignore_config, f, ensure_ascii=False, indent=2)
                QMessageBox.information(self, "導出成功", f"配置已導出到:\n{file_path}")
            except Exception as e:
                QMessageBox.critical(self, "導出錯誤", f"導出配置時出錯: {str(e)}")
    
    def start_search(self):
        """開始檢索"""
        # 獲取輸入參數
        target_dir = self.dir_entry.text().strip()
        if not target_dir or not os.path.isdir(target_dir):
            QMessageBox.warning(self, "目錄錯誤", "請選擇有效的目標目錄！")
            return
        
        # 獲取輸出文件路徑
        output_file = self.output_entry.text().strip()
        if not output_file:
            # 如果沒有指定輸出文件，使用默認文件名
            output_file = os.path.join(
                self.get_default_output_path(),
                self.get_default_filename()
            )
            self.output_entry.setText(output_file)
        
        if not output_file.endswith('.txt'):
            output_file += '.txt'
        
        # 確保輸出目錄存在
        output_dir = os.path.dirname(output_file)
        if output_dir and not os.path.exists(output_dir):
            try:
                os.makedirs(output_dir)
            except Exception as e:
                QMessageBox.critical(self, "目錄錯誤", f"無法創(chuàng)建輸出目錄: {str(e)}")
                return
        
        # 檢查檢索類型
        search_folders = self.folder_radio.isChecked()
        search_files = self.file_radio.isChecked()
        search_both = self.both_radio.isChecked()
        
        file_extensions = []
        if search_files or search_both:
            ext_str = self.suffix_entry.text().strip()
            if ext_str:
                file_extensions = [ext.strip().lower() for ext in ext_str.split(';') if ext.strip()]
        
        # 獲取屏蔽詞
        ignore_patterns = self.current_ignore_config['patterns']
        
        # 執(zhí)行檢索
        self.status_label.setText("正在檢索...")
        QApplication.processEvents()  # 更新UI
        
        try:
            # 獲取層級結構的檢索結果
            results = []
            self.recursive_traverse(
                target_dir, 
                target_dir, 
                results, 
                0,
                search_folders, 
                search_files,
                search_both,
                file_extensions, 
                ignore_patterns
            )
            
            # 寫入文件
            with open(output_file, 'w', encoding='utf-8') as f:
                # 寫入頭部信息
                f.write(f"檢索目錄: {target_dir}\n")
                
                if search_folders:
                    f.write(f"檢索類型: 僅文件夾\n")
                elif search_files:
                    f.write(f"檢索類型: 僅文件\n")
                else:
                    f.write(f"檢索類型: 文件和文件夾\n")
                
                if (search_files or search_both) and file_extensions:
                    f.write(f"文件后綴: {', '.join(file_extensions)}\n")
                if ignore_patterns:
                    f.write(f"屏蔽配置: {self.current_ignore_config['name']}\n")
                    f.write(f"屏蔽詞: {', '.join(ignore_patterns)}\n")
                
                f.write("\n" + "=" * 70 + "\n\n")
                
                # 寫入層級結構
                total_items = 0
                total_folders = 0
                total_files = 0
                
                for item in results:
                    # 根據層級深度添加縮進
                    indent = "│   " * (item['depth'] - 1)
                    prefix = "├── " if item['depth'] > 0 else ""
                    
                    # 添加文件夾/文件標識
                    if item['type'] == 'folder':
                        line = f"{indent}{prefix}?? {item['name']}/"
                        total_folders += 1
                    else:
                        line = f"{indent}{prefix}?? {item['name']}"
                        total_files += 1
                    
                    f.write(line + "\n")
                    total_items += 1
                
                # 添加統(tǒng)計信息
                f.write("\n" + "=" * 70 + "\n\n")
                f.write(f"統(tǒng)計信息:\n")
                f.write(f"總項目數: {total_items}\n")
                f.write(f"文件夾數: {total_folders}\n")
                f.write(f"文件數: {total_files}\n")
            
            self.status_label.setText(f"檢索完成！找到 {total_items} 個項目，結果已保存到: {output_file}")
            QMessageBox.information(self, "完成", 
                f"檢索完成！\n"
                f"總項目數: {total_items}\n"
                f"文件夾數: {total_folders}\n"
                f"文件數: {total_files}\n"
                f"結果已保存到:\n{output_file}"
            )
        except Exception as e:
            self.status_label.setText("檢索出錯")
            QMessageBox.critical(self, "錯誤", f"檢索過程中出錯: {str(e)}")
    
    def recursive_traverse(self, root_dir, current_dir, results, depth, 
                           search_folders, search_files, search_both,
                           file_extensions, ignore_patterns):
        """遞歸遍歷目錄，保持實際目錄結構"""
        try:
            # 獲取當前目錄下的條目
            entries = os.listdir(current_dir)
        except Exception as e:
            # 跳過無權訪問的目錄
            return
        
        # 排序條目
        entries.sort(key=lambda s: s.lower())
        
        # 獲取當前目錄相對于根目錄的相對路徑
        rel_dir = os.path.relpath(current_dir, root_dir)
        
        # 如果是根目錄，添加根目錄項
        if current_dir == root_dir:
            results.append({
                'name': os.path.basename(root_dir) or os.path.splitdrive(root_dir)[0],
                'path': root_dir,
                'depth': depth,
                'type': 'folder'
            })
        
        # 處理文件夾
        folders = [e for e in entries if os.path.isdir(os.path.join(current_dir, e))]
        for folder in folders:
            folder_path = os.path.join(current_dir, folder)
            rel_path = os.path.relpath(folder_path, root_dir)
            
            # 檢查是否在屏蔽列表中
            if self.is_ignored(rel_path, ignore_patterns):
                continue
            
            # 添加到結果（如果需要檢索文件夾）
            if search_folders or search_both:
                results.append({
                    'name': folder,
                    'path': folder_path,
                    'depth': depth + 1,
                    'type': 'folder'
                })
            
            # 遞歸處理子目錄
            self.recursive_traverse(
                root_dir, 
                folder_path, 
                results, 
                depth + 1,
                search_folders, 
                search_files,
                search_both,
                file_extensions, 
                ignore_patterns
            )
        
        # 處理文件
        files = [e for e in entries if os.path.isfile(os.path.join(current_dir, e))]
        for file in files:
            file_path = os.path.join(current_dir, file)
            rel_path = os.path.relpath(file_path, root_dir)
            
            # 檢查是否在屏蔽列表中
            if self.is_ignored(rel_path, ignore_patterns):
                continue
            
            # 檢查文件后綴
            if (search_files or search_both) and file_extensions:
                ext = os.path.splitext(file)[1].lower()
                if ext not in file_extensions:
                    continue
            
            # 添加到結果
            if search_files or search_both:
                results.append({
                    'name': file,
                    'path': file_path,
                    'depth': depth + 1,
                    'type': 'file'
                })
    
    def is_ignored(self, path, patterns):
        """檢查路徑是否與任何屏蔽模式匹配"""
        for pattern in patterns:
            if fnmatch.fnmatch(path, pattern):
                return True
            if pattern in path:
                return True
        return False


if __name__ == "__main__":
    app = QApplication([])
    app.setStyle("Fusion")
    
    # 設置應用樣式
    palette = QPalette()
    palette.setColor(QPalette.Window, QColor(53, 53, 53))
    palette.setColor(QPalette.WindowText, Qt.white)
    palette.setColor(QPalette.Base, QColor(35, 35, 35))
    palette.setColor(QPalette.AlternateBase, QColor(53, 53, 53))
    palette.setColor(QPalette.ToolTipBase, Qt.white)
    palette.setColor(QPalette.ToolTipText, Qt.white)
    palette.setColor(QPalette.Text, Qt.white)
    palette.setColor(QPalette.Button, QColor(53, 53, 53))
    palette.setColor(QPalette.ButtonText, Qt.white)
    palette.setColor(QPalette.BrightText, Qt.red)
    palette.setColor(QPalette.Highlight, QColor(142, 45, 197).lighter())
    palette.setColor(QPalette.HighlightedText, Qt.black)
    app.setPalette(palette)
    
    window = FileSearchApp()
    window.show()
    app.exec_()

pyinstaller --onefile -w so.py