#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import sys
import os
import pandas as pd
import time
import logging
from datetime import datetime
from PyQt6.QtWidgets import (
    QApplication, QWidget, QPushButton, QVBoxLayout, QLabel, 
    QFileDialog, QLineEdit, QTextEdit, QProgressBar, QMessageBox,
    QCheckBox, QToolTip, QMenu, QHBoxLayout
)
from PyQt6.QtCore import Qt, QThread, pyqtSignal, QPropertyAnimation, QEasingCurve
from PyQt6.QtGui import QIcon, QPixmap

logging.basicConfig(filename='scanner.log', level=logging.INFO)

class PathTextEdit(QTextEdit):
    """支持拖放文件夾并顯示純路徑的自定義文本框"""
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setAcceptDrops(True)
        self.setPlaceholderText("拖拽文件夾到此處或點(diǎn)擊瀏覽按鈕")
        self.setMinimumHeight(80)
    
    def dragEnterEvent(self, event):
        if any(url.isLocalFile() and os.path.isdir(url.toLocalFile())
               for url in event.mimeData().urls()):
            self.setStyleSheet("border: 2px dashed #0078D7; background-color: #F0F7FF;")
            event.acceptProposedAction()
        else:
            event.ignore()
    
    def dragLeaveEvent(self, event):
        self.setStyleSheet("border: 2px dashed #9E9E9E; background-color: #FAFAFA;")
        event.accept()
    
    def dropEvent(self, event):
        self.setStyleSheet("border: 1px solid #CCCCCC; background-color: white;")
        valid_paths = []
        for url in event.mimeData().urls():
            if url.isLocalFile():
                path = url.toLocalFile()
                if os.path.isdir(path):
                    valid_paths.append(path)
        
        if valid_paths:
            self.setText("\n".join(valid_paths))
            event.accept()
        else:
            QToolTip.showText(event.pos(), "請拖入有效的文件夾", self)
            event.ignore()
    
    def contextMenuEvent(self, event):
        menu = QMenu(self)
        clear_action = menu.addAction(QIcon("icons/clear.png"), "清空") if os.path.exists("icons/clear.png") else menu.addAction("清空")
        paste_action = menu.addAction(QIcon("icons/paste.png"), "粘貼路徑") if os.path.exists("icons/paste.png") else menu.addAction("粘貼路徑")
        action = menu.exec(event.globalPos())
        if action == clear_action:
            self.clear()
        elif action == paste_action:
            self.paste()

class DirectoryScanner(QThread):
    progress = pyqtSignal(int, int, float)
    completed = pyqtSignal(list)
    error = pyqtSignal(str)
    
    def __init__(self, directory, file_types):
        super().__init__()
        self.directory = directory
        self.file_types = file_types
        self._is_running = True
    
    def stop(self):
        self._is_running = False
    
    def run(self):
        try:
            file_data = []
            total_files = sum(len(files) for _, _, files in os.walk(self.directory))
            processed_files = 0
            start_time = time.time()
            
            for root, _, files in os.walk(self.directory):
                if not self._is_running:
                    break
                
                for file in files:
                    if self.file_types and not any(file.lower().endswith(ft.lower()) for ft in self.file_types if ft):
                        continue
                    
                    try:
                        file_path = os.path.join(root, file)
                        file_stat = os.stat(file_path)
                        file_data.append([
                            root, file, os.path.splitext(file)[1], os.path.getsize(file_path),
                            time.ctime(file_stat.st_mtime), time.ctime(file_stat.st_ctime), file_path
                        ])
                    except Exception as e:
                        logging.error(f"Error processing {file_path}: {str(e)}")
                        continue
                    
                    processed_files += 1
                    elapsed_time = time.time() - start_time
                    estimated_total_time = (elapsed_time / processed_files) * total_files if processed_files else 0
                    remaining_time = max(0, estimated_total_time - elapsed_time)
                    self.progress.emit(processed_files, total_files, remaining_time)
            
            if self._is_running:
                self.completed.emit(file_data)
        except Exception as e:
            self.error.emit(f"掃描過程中出錯(cuò): {str(e)}")

class DirectoryScannerApp(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()
        self.setup_animations()
    
    def initUI(self):
        self.setWindowTitle("目錄結(jié)構(gòu)生成器")
        self.setGeometry(100, 100, 650, 550)
        
        # 主布局
        self.main_layout = QVBoxLayout()
        self.main_layout.setSpacing(15)
        self.main_layout.setContentsMargins(20, 20, 20, 20)
        
        # 標(biāo)題區(qū)域
        title_layout = QHBoxLayout()
        self.title_icon = QLabel()
        if os.path.exists("icons/folder.png"):
            self.title_icon.setPixmap(QPixmap("icons/folder.png").scaled(32, 32))
        title_layout.addWidget(self.title_icon)
        
        self.title_label = QLabel("目錄結(jié)構(gòu)生成器")
        self.title_label.setStyleSheet("font-size: 18pt; font-weight: bold; color: #0078D7;")
        title_layout.addWidget(self.title_label)
        title_layout.addStretch()
        self.main_layout.addLayout(title_layout)
        
        # 目錄選擇部分
        self.dir_group = QWidget()
        dir_layout = QVBoxLayout()
        
        self.dir_label = QLabel("選擇掃描目錄:")
        dir_layout.addWidget(self.dir_label)
        
        self.directory_input = PathTextEdit()
        dir_layout.addWidget(self.directory_input)
        
        self.browse_button = QPushButton(" 瀏覽目錄")
        if os.path.exists("icons/folder_open.png"):
            self.browse_button.setIcon(QIcon("icons/folder_open.png"))
        self.browse_button.clicked.connect(self.browse_directory)
        dir_layout.addWidget(self.browse_button)
        
        self.dir_group.setLayout(dir_layout)
        self.main_layout.addWidget(self.dir_group)
        
        # 過濾選項(xiàng)部分
        self.filter_group = QWidget()
        filter_layout = QVBoxLayout()
        
        self.filter_label = QLabel("文件類型過濾:")
        filter_layout.addWidget(self.filter_label)
        
        self.filter_input = QLineEdit()
        self.filter_input.setPlaceholderText("例如: .txt,.docx,.xlsx")
        filter_layout.addWidget(self.filter_input)
        
        self.filter_group.setLayout(filter_layout)
        self.main_layout.addWidget(self.filter_group)
        
        # 選項(xiàng)設(shè)置
        self.options_group = QWidget()
        options_layout = QVBoxLayout()
        
        self.checkbox_size = QCheckBox("顯示友好文件大小（KB/MB）")
        self.checkbox_open = QCheckBox("導(dǎo)出后自動打開文件")
        options_layout.addWidget(self.checkbox_size)
        options_layout.addWidget(self.checkbox_open)
        
        self.options_group.setLayout(options_layout)
        self.main_layout.addWidget(self.options_group)
        
        # 操作按鈕
        self.generate_button = QPushButton(" 生成Excel文件")
        if os.path.exists("icons/excel.png"):
            self.generate_button.setIcon(QIcon("icons/excel.png"))
        self.generate_button.clicked.connect(self.generate_excel)
        self.main_layout.addWidget(self.generate_button)
        
        # 進(jìn)度顯示
        self.progress_bar = QProgressBar()
        self.progress_bar.setTextVisible(True)
        self.main_layout.addWidget(self.progress_bar)
        
        # 狀態(tài)欄
        self.status_bar = QHBoxLayout()
        self.status_icon = QLabel()
        if os.path.exists("icons/info.png"):
            self.status_icon.setPixmap(QPixmap("icons/info.png").scaled(16, 16))
        self.status_bar.addWidget(self.status_icon)
        
        self.status_label = QLabel("就緒")
        self.status_bar.addWidget(self.status_label)
        self.status_bar.addStretch()
        
        self.version_label = QLabel(f"探客白澤 ? {datetime.now().year}")
        self.status_bar.addWidget(self.version_label)
        self.main_layout.addLayout(self.status_bar)
        
        self.setLayout(self.main_layout)
        self.apply_stylesheet()
    
    def setup_animations(self):
        self.animations = {}
        for btn in [self.browse_button, self.generate_button]:
            animation = QPropertyAnimation(btn, b"geometry")
            animation.setDuration(200)
            animation.setEasingCurve(QEasingCurve.Type.OutQuad)
            self.animations[btn] = animation
            
            def make_animator(button):
                return lambda: self.animate_button(button)
            
            btn.clicked.connect(make_animator(btn))
    
    def animate_button(self, button):
        animation = self.animations[button]
        orig_rect = button.geometry()
        animation.setStartValue(orig_rect.adjusted(0, 5, 0, 5))
        animation.setEndValue(orig_rect)
        animation.start()
    
    def apply_stylesheet(self):
        self.setStyleSheet("""
            /* 主窗口樣式 */
            QWidget {
                font-family: 'Microsoft YaHei', 'Segoe UI';
                font-size: 10pt;
                background-color: #F5F5F5;
            }
            
            /* 分組框樣式 */
            QWidget#dir_group, QWidget#filter_group, QWidget#options_group {
                background-color: white;
                border-radius: 6px;
                padding: 12px;
            }
            
            /* 標(biāo)簽樣式 */
            QLabel {
                color: #333333;
                font-weight: 500;
            }
            
            /* 輸入框樣式 */
            QTextEdit, QLineEdit {
                background-color: white;
                border: 1px solid #CCCCCC;
                border-radius: 4px;
                padding: 8px;
                selection-background-color: #0078D7;
            }
            
            QTextEdit:hover, QLineEdit:hover {
                border: 1px solid #0078D7;
            }
            
            /* 按鈕基礎(chǔ)樣式 */
            QPushButton {
                color: white;
                border: none;
                border-radius: 6px;
                padding: 10px 18px;
                min-width: 100px;
                font-weight: 500;
            }
            
            /* 瀏覽按鈕樣式 */
            QPushButton#browse_button {
                background-color: #4CAF50;
            }
            
            QPushButton#browse_button:hover {
                background-color: #3E8E41;
            }
            
            QPushButton#browse_button:pressed {
                background-color: #2E7D32;
            }
            
            /* 生成按鈕樣式 */
            QPushButton#generate_button {
                background-color: #FF5722;
                font-weight: bold;
            }
            
            QPushButton#generate_button:hover {
                background-color: #E64A19;
            }
            
            QPushButton#generate_button:pressed {
                background-color: #BF360C;
            }
            
            /* 進(jìn)度條樣式 */
            QProgressBar {
                border: 1px solid #CCCCCC;
                border-radius: 4px;
                text-align: center;
                height: 24px;
            }
            
            QProgressBar::chunk {
                background-color: #4CAF50;
                border-radius: 3px;
            }
            
            /* 復(fù)選框樣式 */
            QCheckBox {
                spacing: 8px;
                color: #333333;
            }
            
            QCheckBox::indicator {
                width: 18px;
                height: 18px;
            }
            
            QCheckBox::indicator:unchecked {
                border: 1px solid #CCCCCC;
                background: white;
                border-radius: 3px;
            }
            
            QCheckBox::indicator:checked {
                border: 1px solid #0078D7;
                background: #0078D7;
                border-radius: 3px;
            }
            
            /* 狀態(tài)欄樣式 */
            QLabel#version_label {
                color: #666666;
                font-size: 9pt;
            }
        """)
        
        # 設(shè)置對象名用于樣式選擇
        self.browse_button.setObjectName("browse_button")
        self.generate_button.setObjectName("generate_button")
        self.version_label.setObjectName("version_label")
        self.dir_group.setObjectName("dir_group")
        self.filter_group.setObjectName("filter_group")
        self.options_group.setObjectName("options_group")
    
    def browse_directory(self):
        directory = QFileDialog.getExistingDirectory(self, "選擇目錄")
        if directory:
            self.directory_input.setText(directory)
            self.status_label.setText(f"已選擇目錄: {directory[:50]}..." if len(directory) > 50 else directory)
    
    def generate_excel(self):
        directory = self.directory_input.toPlainText().strip()
        file_types = [ft.strip() for ft in self.filter_input.text().split(',') if ft.strip()]
        
        if not directory or not os.path.isdir(directory):
            QMessageBox.warning(self, "錯(cuò)誤", "請選擇有效的目錄！")
            return
        
        default_name = f"目錄結(jié)構(gòu)_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx"
        save_path, _ = QFileDialog.getSaveFileName(
            self, "保存 Excel 文件", 
            default_name, 
            "Excel Files (*.xlsx)"
        )
        
        if not save_path:
            return
        
        self.scanner = DirectoryScanner(directory, file_types)
        self.scanner.progress.connect(self.update_progress)
        self.scanner.completed.connect(lambda data: self.save_to_excel(data, save_path))
        self.scanner.error.connect(self.show_error)
        self.generate_button.setEnabled(False)
        self.status_label.setText("正在掃描目錄...")
        self.status_icon.setPixmap(QPixmap("icons/loading.png").scaled(16, 16)) if os.path.exists("icons/loading.png") else None
        self.scanner.start()
    
    def update_progress(self, processed, total, remaining_time):
        self.progress_bar.setMaximum(total)
        self.progress_bar.setValue(processed)
        
        progress_percent = processed / total * 100
        status_text = (
            f"掃描進(jìn)度: {progress_percent:.1f}% | "
            f"已處理: {processed}/{total} | "
            f"剩余時(shí)間: {remaining_time:.0f}秒"
        )
        self.status_label.setText(status_text)
    
    def save_to_excel(self, file_data, save_path):
        try:
            df = pd.DataFrame(file_data, columns=[
                "目錄路徑", "文件名", "類型", "大小(B)", "修改日期", "創(chuàng)建時(shí)間", "文件位置"
            ])
            
            # 添加人性化大小顯示
            if self.checkbox_size.isChecked():
                df["大小"] = df["大小(B)"].apply(
                    lambda x: f"{x/1024:.2f}KB" if x < 1024*1024 else f"{x/1024/1024:.2f}MB"
                )
            
            # 添加超鏈接
            df["文件位置"] = df["文件位置"].apply(lambda x: f'=HYPERLINK("{x}", "?? 打開")')
            
            with pd.ExcelWriter(save_path, engine='xlsxwriter') as writer:
                df.to_excel(writer, index=False, sheet_name='目錄結(jié)構(gòu)')
                
                workbook = writer.book
                worksheet = writer.sheets['目錄結(jié)構(gòu)']
                
                # 設(shè)置列寬
                col_widths = {
                    "目錄路徑": 40, "文件名": 30, "類型": 10, 
                    "大小(B)": 15, "大小": 15, "修改日期": 20,
                    "創(chuàng)建時(shí)間": 20, "文件位置": 15
                }
                
                for idx, col in enumerate(df.columns):
                    worksheet.set_column(idx, idx, col_widths.get(col, 15))
                
                # 添加凍結(jié)窗格和自動篩選
                worksheet.autofilter(0, 0, 0, len(df.columns)-1)
                worksheet.freeze_panes(1, 0)
            
            self.status_label.setText(f"導(dǎo)出成功: {os.path.basename(save_path)}")
            self.status_icon.setPixmap(QPixmap("icons/success.png").scaled(16, 16)) if os.path.exists("icons/success.png") else None
            
            if self.checkbox_open.isChecked():
                try:
                    if sys.platform == "win32":
                        os.startfile(save_path)
                    elif sys.platform == "darwin":
                        os.system(f'open "{save_path}"')
                    else:
                        os.system(f'xdg-open "{save_path}"')
                except Exception as e:
                    logging.error(f"無法打開文件: {str(e)}")
        
        except Exception as e:
            self.show_error(f"生成Excel時(shí)出錯(cuò):\n{str(e)}")
            logging.error(f"Excel導(dǎo)出錯(cuò)誤: {str(e)}")
        
        finally:
            self.generate_button.setEnabled(True)
            self.progress_bar.reset()
    
    def show_error(self, message):
        QMessageBox.critical(self, "錯(cuò)誤", message)
        self.status_label.setText("操作失敗")
        self.status_icon.setPixmap(QPixmap("icons/error.png").scaled(16, 16)) if os.path.exists("icons/error.png") else None
        self.generate_button.setEnabled(True)
    
    def closeEvent(self, event):
        if hasattr(self, 'scanner') and self.scanner.isRunning():
            self.scanner.stop()
            self.scanner.wait()
        event.accept()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    app.setStyle('Fusion')
    
    # 設(shè)置應(yīng)用程序圖標(biāo)
    if os.path.exists("icons/app_icon.png"):
        app.setWindowIcon(QIcon("icons/app_icon.png"))
    
    window = DirectoryScannerApp()
    window.show()
    sys.exit(app.exec())