import sys
import os
from PyQt5.QtWidgets import (
    QApplication, QMainWindow, QLabel, QLineEdit, QPushButton, QFileDialog,
    QVBoxLayout, QHBoxLayout, QWidget, QMessageBox, QComboBox
)
from PyQt5.QtGui import QPixmap, QIcon
from PyQt5.QtCore import Qt
from PyQt5.QtCore import QMimeData
from PyQt5.QtGui import QDragEnterEvent, QDropEvent
from PIL import Image

# 日志文件
LOG_FILE = "conversion_history.log"

class ImageToICOConverter(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setWindowTitle("圖片轉(zhuǎn)ICO工具")
        self.setGeometry(100, 100, 355, 360)
        self.setAcceptDrops(True)  # 啟用拖拽功能
        self.initUI()

    def initUI(self):
        # 主布局
        main_layout = QVBoxLayout()

        # 圖片選擇
        file_layout = QHBoxLayout()
        self.image_path_input = QLineEdit(self)
        browse_button = QPushButton("瀏覽", self)
        browse_button.clicked.connect(self.choose_image_file)
        file_layout.addWidget(QLabel("選擇圖片文件:"))
        file_layout.addWidget(self.image_path_input)
        file_layout.addWidget(browse_button)
        main_layout.addLayout(file_layout)

        # 輸出路徑選擇
        output_layout = QHBoxLayout()
        self.output_path_input = QLineEdit(self)
        save_button = QPushButton("保存", self)
        save_button.clicked.connect(self.choose_output_path)
        output_layout.addWidget(QLabel("選擇輸出路徑:"))
        output_layout.addWidget(self.output_path_input)
        output_layout.addWidget(save_button)
        main_layout.addLayout(output_layout)

        # 圖標(biāo)尺寸選擇（單選）
        size_layout = QHBoxLayout()
        self.size_combo = QComboBox(self)
        self.size_combo.addItems(["16", "32", "48", "64", "128", "256"])
        size_layout.addWidget(QLabel("選擇圖標(biāo)尺寸:"))
        size_layout.addWidget(self.size_combo)
        main_layout.addLayout(size_layout)

        # 圖片預(yù)覽
        self.preview_label = QLabel("圖片預(yù)覽", self)
        self.preview_label.setAlignment(Qt.AlignCenter)
        self.preview_label.setStyleSheet("background-color: lightgray; border: 1px solid black;")
        self.preview_label.setFixedSize(200, 200)
        preview_layout = QVBoxLayout()
        preview_layout.addWidget(self.preview_label, alignment=Qt.AlignCenter)
        main_layout.addLayout(preview_layout)

        # 轉(zhuǎn)換按鈕
        convert_button = QPushButton("轉(zhuǎn)換為ICO", self)
        convert_button.clicked.connect(self.convert_to_ico)
        main_layout.addWidget(convert_button, alignment=Qt.AlignCenter)

        # 設(shè)置中央窗口
        central_widget = QWidget()
        central_widget.setLayout(main_layout)
        self.setCentralWidget(central_widget)

    def choose_image_file(self):
        file_path, _ = QFileDialog.getOpenFileName(self, "選擇圖片文件", "", "圖片文件 (*.png *.jpg *.jpeg *.bmp *.gif)")
        if file_path:
            self.image_path_input.setText(file_path)
            self.show_preview(file_path)

    def choose_output_path(self):
        output_path, _ = QFileDialog.getSaveFileName(self, "選擇輸出路徑", "", "ICO文件 (*.ico)")
        if output_path:
            self.output_path_input.setText(output_path)

    def show_preview(self, image_path):
        try:
            pixmap = QPixmap(image_path)
            pixmap = pixmap.scaled(200, 200, Qt.KeepAspectRatio, Qt.SmoothTransformation)
            self.preview_label.setPixmap(pixmap)
        except Exception as e:
            QMessageBox.critical(self, "預(yù)覽錯(cuò)誤", f"無法加載圖片預(yù)覽: {str(e)}")

    def convert_to_ico(self):
        image_path = self.image_path_input.text()
        output_path = self.output_path_input.text()

        if not image_path:
            QMessageBox.critical(self, "錯(cuò)誤", "請選擇源圖片文件")
            return

        if not output_path:
            output_path = os.path.splitext(image_path)[0] + ".ico"
            self.output_path_input.setText(output_path)

        try:
            img = Image.open(image_path)

            # 獲取用戶選擇的圖標(biāo)尺寸
            size = int(self.size_combo.currentText())
            sizes = [(size, size)]

            img = img.convert("RGBA")
            img.save(output_path, format="ICO", sizes=sizes)

            with open(LOG_FILE, "a") as log_file:
                log_file.write(f"Converted: {image_path} -> {output_path}\n")

            QMessageBox.information(self, "成功", f"圖片已成功轉(zhuǎn)換并保存為 {output_path}")
        except Exception as e:
            QMessageBox.critical(self, "轉(zhuǎn)換錯(cuò)誤", f"轉(zhuǎn)換過程中發(fā)生錯(cuò)誤: {str(e)}")

    def dragEnterEvent(self, event: QDragEnterEvent):
        if event.mimeData().hasUrls():
            event.accept()
        else:
            event.ignore()

    def dropEvent(self, event: QDropEvent):
        mime_data: QMimeData = event.mimeData()
        if mime_data.hasUrls():
            file_path = mime_data.urls()[0].toLocalFile()
            self.image_path_input.setText(file_path)
            self.show_preview(file_path)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = ImageToICOConverter()
    window.show()
    sys.exit(app.exec_())