__main__.py

"""ExcelTools - Tabbed PyQt6 GUI for Excel search and merge operations."""

import os
import sys
import threading

from PyQt6.QtWidgets import (
    QApplication, QMainWindow, QTabWidget, QWidget, QVBoxLayout, QHBoxLayout,
    QPushButton, QLabel, QLineEdit, QFileDialog, QProgressBar,
    QTableWidget, QTableWidgetItem, QCheckBox, QHeaderView, QMessageBox,
    QAbstractItemView, QFrame,
)
from PyQt6.QtCore import QObject, pyqtSignal, pyqtSlot, Qt

from exceltools.search import search_file
from exceltools.merge import get_file_info, merge_files


# ═══════════════════════════════════════════════════════════
#  Search Tab
# ═══════════════════════════════════════════════════════════

class SearchTab(QWidget):
    def __init__(self):
        super().__init__()
        self.search_thread = None
        self._build_ui()

    def _build_ui(self):
        layout = QVBoxLayout()

        # Folder row
        folder_row = QHBoxLayout()
        folder_row.addWidget(QLabel("搜索目录:"))
        self.folder_edit = QLineEdit()
        folder_row.addWidget(self.folder_edit)
        browse_btn = QPushButton("浏览...")
        browse_btn.clicked.connect(self._browse_folder)
        folder_row.addWidget(browse_btn)
        layout.addLayout(folder_row)

        # Search row
        search_row = QHBoxLayout()
        search_row.addWidget(QLabel("搜索内容:"))
        self.search_edit = QLineEdit()
        self.search_edit.returnPressed.connect(self._start_search)
        search_row.addWidget(self.search_edit)

        self.search_btn = QPushButton("搜索")
        self.search_btn.clicked.connect(self._start_search)
        search_row.addWidget(self.search_btn)
        layout.addLayout(search_row)

        # Options row
        options_row = QHBoxLayout()
        self.exact_cb = QCheckBox("精确匹配（单元格内容与搜索词完全一致）")
        options_row.addWidget(self.exact_cb)
        self.wildcard_cb = QCheckBox("启用通配符（* 任意字符, ? 单个字符）")
        options_row.addWidget(self.wildcard_cb)
        options_row.addStretch()
        layout.addLayout(options_row)

        # Progress
        self.progress = QProgressBar()
        layout.addWidget(self.progress)

        # Results table
        self.results = QTableWidget(0, 4)
        self.results.setHorizontalHeaderLabels(["文件", "工作表", "单元格", "单元格内容"])
        self.results.horizontalHeader().setSectionResizeMode(0, QHeaderView.ResizeMode.Interactive)
        self.results.horizontalHeader().setSectionResizeMode(1, QHeaderView.ResizeMode.Interactive)
        self.results.horizontalHeader().setSectionResizeMode(2, QHeaderView.ResizeMode.Interactive)
        self.results.horizontalHeader().setSectionResizeMode(3, QHeaderView.ResizeMode.Stretch)
        self.results.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
        self.results.setEditTriggers(QAbstractItemView.EditTrigger.NoEditTriggers)
        self.results.verticalHeader().setVisible(False)
        layout.addWidget(self.results)

        self.setLayout(layout)

    def _browse_folder(self):
        folder = QFileDialog.getExistingDirectory(self, "选择搜索目录")
        if folder:
            self.folder_edit.setText(folder)

    def _start_search(self):
        folder = self.folder_edit.text().strip()
        keyword = self.search_edit.text().strip()
        if not folder or not keyword:
            return

        if self.search_thread and self.search_thread.is_alive():
            self.search_thread.stop()

        self.results.setRowCount(0)
        self.progress.setValue(0)

        self.search_thread = SearchThread(
            folder, keyword, self.exact_cb.isChecked(), self.wildcard_cb.isChecked()
        )
        self.search_thread.search_progress.connect(self.progress.setValue)
        self.search_thread.search_complete.connect(self._show_results)
        self.search_thread.start()

    @pyqtSlot(list)
    def _show_results(self, matches):
        self.results.setRowCount(len(matches))
        for i, m in enumerate(matches):
            self.results.setItem(i, 0, QTableWidgetItem(os.path.basename(m["File"])))
            self.results.setItem(i, 1, QTableWidgetItem(m["Sheet"]))
            self.results.setItem(i, 2, QTableWidgetItem(m["Cell"]))
            self.results.setItem(i, 3, QTableWidgetItem(m["Value"]))


class SearchThread(QObject, threading.Thread):
    search_complete = pyqtSignal(list)
    search_progress = pyqtSignal(int)

    def __init__(self, folder, keyword, exact_match, wildcard):
        super().__init__()
        self.folder = folder
        self.keyword = keyword
        self.exact_match = exact_match
        self.wildcard = wildcard
        self._running = True

    def run(self):
        matches = []
        excel_files = []
        for root, _, files in os.walk(self.folder):
            for f in files:
                if f.startswith("~$"):
                    continue
                if f.endswith((".xlsx", ".xls")):
                    excel_files.append(os.path.join(root, f))

        total = len(excel_files)
        for idx, path in enumerate(excel_files):
            if not self._running:
                break
            for sheet, coord, value in search_file(path, self.keyword, self.exact_match, self.wildcard):
                if not self._running:
                    break
                matches.append({"File": path, "Sheet": sheet, "Cell": coord, "Value": value})
            self.search_progress.emit(int(((idx + 1) / total) * 100) if total else 100)

        self.search_complete.emit(matches)

    def stop(self):
        self._running = False


# ═══════════════════════════════════════════════════════════
#  Merge Tab
# ═══════════════════════════════════════════════════════════

class MergeTab(QWidget):
    def __init__(self):
        super().__init__()
        self.file_paths = []
        self.merge_thread = None
        self._build_ui()

    def _build_ui(self):
        layout = QVBoxLayout()

        # ── File list table ──
        top_label = QLabel("待合并的文件列表（选中行后可用右侧按钮调整顺序）:")
        top_label.setStyleSheet("color: gray;")
        layout.addWidget(top_label)

        mid_row = QHBoxLayout()

        self.table = QTableWidget(0, 4)
        self.table.setHorizontalHeaderLabels(["文件名", "修改日期", "列数", "行数"])
        self.table.horizontalHeader().setSectionResizeMode(0, QHeaderView.ResizeMode.Stretch)
        self.table.horizontalHeader().setSectionResizeMode(1, QHeaderView.ResizeMode.ResizeToContents)
        self.table.horizontalHeader().setSectionResizeMode(2, QHeaderView.ResizeMode.ResizeToContents)
        self.table.horizontalHeader().setSectionResizeMode(3, QHeaderView.ResizeMode.ResizeToContents)
        self.table.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
        self.table.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection)
        self.table.setEditTriggers(QAbstractItemView.EditTrigger.NoEditTriggers)
        self.table.verticalHeader().setVisible(False)
        mid_row.addWidget(self.table)

        # ── Side buttons ──
        side = QVBoxLayout()
        side.setAlignment(Qt.AlignmentFlag.AlignTop)

        side.addWidget(self._btn("添加文件", self._add_files, 100))
        side.addWidget(self._btn("移除选中", self._remove_selected, 100))
        side.addWidget(self._btn("移除所有", self._remove_all, 100))

        side.addSpacing(8)
        line = QFrame()
        line.setFrameShape(QFrame.Shape.HLine)
        side.addWidget(line)
        side.addSpacing(4)

        side.addWidget(QLabel("排序方式:"))
        side.addWidget(self._btn("上移", self._move_up, 100))
        side.addWidget(self._btn("下移", self._move_down, 100))
        side.addSpacing(4)
        side.addWidget(self._btn("按文件名排序", self._sort_by_name, 100))
        side.addWidget(self._btn("按修改日期排序", self._sort_by_date, 100))

        mid_row.addLayout(side)
        layout.addLayout(mid_row)

        # ── Bottom: output + merge ──
        layout.addWidget(QLabel("输出文件:"))

        out_row = QHBoxLayout()
        self.output_edit = QLineEdit()
        out_row.addWidget(self.output_edit)
        browse_out = QPushButton("浏览...")
        browse_out.clicked.connect(self._choose_output)
        out_row.addWidget(browse_out)
        layout.addLayout(out_row)

        self.skip_cb = QCheckBox("跳过首行（将第一行作为表头，不参与合并）")
        self.skip_cb.setChecked(True)
        self.skip_cb.toggled.connect(self._refresh_table)
        layout.addWidget(self.skip_cb)

        btn_row = QHBoxLayout()
        self.merge_btn = QPushButton("开始合并")
        self.merge_btn.clicked.connect(self._start_merge)
        self.merge_btn.setFixedWidth(100)
        btn_row.addWidget(self.merge_btn)
        self.merge_progress = QProgressBar()
        btn_row.addWidget(self.merge_progress)
        layout.addLayout(btn_row)

        self.status_label = QLabel("就绪")
        self.status_label.setStyleSheet("color: gray;")
        layout.addWidget(self.status_label)

        self.setLayout(layout)

    def _btn(self, text, handler, width=None):
        b = QPushButton(text)
        b.clicked.connect(handler)
        if width:
            b.setFixedWidth(width)
        return b

    # ── File list management ──

    def _add_files(self):
        paths, _ = QFileDialog.getOpenFileNames(
            self, "选择要合并的 Excel 文件",
            "", "Excel Files (*.xlsx *.xls);;All Files (*.*)"
        )
        if paths:
            existing = set(self.file_paths)
            for p in paths:
                if p not in existing:
                    self.file_paths.append(p)
            self._refresh_table()

    def _remove_selected(self):
        rows = set(i.row() for i in self.table.selectedIndexes())
        self.file_paths = [p for i, p in enumerate(self.file_paths) if i not in rows]
        self._refresh_table()

    def _remove_all(self):
        if not self.file_paths:
            return
        reply = QMessageBox.question(
            self, "确认", f"确定要移除所有 {len(self.file_paths)} 个文件吗？",
            QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No, QMessageBox.StandardButton.No
        )
        if reply == QMessageBox.StandardButton.Yes:
            self.file_paths.clear()
            self._refresh_table()

    # ── Ordering ──

    def _selected_rows(self):
        return sorted(set(i.row() for i in self.table.selectedIndexes()))

    def _restore_selection(self, rows):
        self.table.clearSelection()
        for r in rows:
            if r < len(self.file_paths):
                self.table.selectRow(r)

    def _move_up(self):
        rows = self._selected_rows()
        if not rows or rows[0] == 0:
            return
        for r in rows:
            self.file_paths[r], self.file_paths[r - 1] = \
                self.file_paths[r - 1], self.file_paths[r]
        self._refresh_table()
        self._restore_selection([r - 1 for r in rows])

    def _move_down(self):
        rows = self._selected_rows()
        n = len(self.file_paths)
        if not rows or rows[-1] >= n - 1:
            return
        for r in reversed(rows):
            self.file_paths[r], self.file_paths[r + 1] = \
                self.file_paths[r + 1], self.file_paths[r]
        self._refresh_table()
        self._restore_selection([r + 1 for r in rows])

    def _sort_by_name(self):
        selected = set(self._selected_rows())
        self.file_paths.sort(key=lambda p: os.path.basename(p).lower())
        self._refresh_table()

    def _sort_by_date(self):
        self.file_paths.sort(key=lambda p: os.path.getmtime(p))
        self._refresh_table()

    # ── Table refresh ──

    def _refresh_table(self):
        skip = self.skip_cb.isChecked()
        self.table.setRowCount(len(self.file_paths))
        for i, path in enumerate(self.file_paths):
            mod_date, cols, rows = get_file_info(path, skip)
            self.table.setItem(i, 0, QTableWidgetItem(os.path.basename(path)))
            self.table.setItem(i, 1, QTableWidgetItem(mod_date))
            col_item = QTableWidgetItem(str(cols))
            col_item.setTextAlignment(Qt.AlignmentFlag.AlignCenter)
            self.table.setItem(i, 2, col_item)
            row_item = QTableWidgetItem(str(rows))
            row_item.setTextAlignment(Qt.AlignmentFlag.AlignCenter)
            self.table.setItem(i, 3, row_item)
        self.status_label.setText(f"已添加 {len(self.file_paths)} 个文件")

    # ── Output path ──

    def _choose_output(self):
        path, _ = QFileDialog.getSaveFileName(
            self, "选择合并后的文件存储位置", "", "Excel Files (*.xlsx)"
        )
        if path:
            self.output_edit.setText(path)

    # ── Merge ──

    def _start_merge(self):
        if not self.file_paths:
            QMessageBox.warning(self, "提示", "请先添加要合并的文件。")
            return
        output = self.output_edit.text().strip()
        if not output:
            QMessageBox.warning(self, "提示", "请先选择输出文件的位置。")
            return

        self.merge_btn.setEnabled(False)
        self.merge_progress.setValue(0)
        self.status_label.setText("正在合并...")

        self.merge_thread = MergeThread(
            self.file_paths, output, self.skip_cb.isChecked()
        )
        self.merge_thread.progress.connect(self._on_merge_progress)
        self.merge_thread.finished.connect(self._on_merge_done)
        self.merge_thread.start()

    @pyqtSlot(int, str)
    def _on_merge_progress(self, pct, msg):
        self.merge_progress.setValue(pct)
        self.status_label.setText(msg)

    @pyqtSlot(bool, str)
    def _on_merge_done(self, ok, msg):
        self.merge_btn.setEnabled(True)
        if ok:
            QMessageBox.information(self, "完成", msg)
            self.status_label.setText("合并完成")
        else:
            QMessageBox.warning(self, "错误", msg)
            self.status_label.setText("合并失败")
        self.merge_progress.setValue(0)


class MergeThread(QObject, threading.Thread):
    progress = pyqtSignal(int, str)
    finished = pyqtSignal(bool, str)

    def __init__(self, file_paths, output_path, skip_header):
        super().__init__()
        self.file_paths = file_paths
        self.output_path = output_path
        self.skip_header = skip_header

    def run(self):
        ok, msg = merge_files(
            self.file_paths, self.output_path, self.skip_header,
            progress_callback=lambda pct, text: self.progress.emit(pct, text)
        )
        self.finished.emit(ok, msg)


# ═══════════════════════════════════════════════════════════
#  Main Window
# ═══════════════════════════════════════════════════════════

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("ExcelTools")
        self.setGeometry(100, 100, 780, 560)

        self.tabs = QTabWidget()
        self.tabs.addTab(SearchTab(), "Excel 搜索")
        self.tabs.addTab(MergeTab(), "Excel 合并")

        self.setCentralWidget(self.tabs)


def main():
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec())


if __name__ == "__main__":
    main()