#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
自定义组件 - 文件选择控件等公共组件
"""

import os

from PyQt5.QtWidgets import (
    QWidget, QHBoxLayout, QLabel, QLineEdit, QPushButton, QFileDialog,
)
from PyQt5.QtCore import Qt


class DirSelectWidget(QWidget):
    """目录选择组件"""
    def __init__(self, label_text, parent=None):
        """
        初始化目录选择组件

        Args:
            label_text: 标签文本
            parent: 父控件
        """
        super().__init__(parent)
        self.init_ui(label_text)

    def init_ui(self, label_text):
        layout = QHBoxLayout()
        layout.setContentsMargins(0, 0, 0, 0)

        self.label = QLabel(label_text)
        self.label.setMinimumWidth(120)
        self.line_edit = QLineEdit()
        self.line_edit.setPlaceholderText("请选择目录...")
        self.browse_btn = QPushButton("浏览...")
        self.browse_btn.setMaximumWidth(80)
        self.browse_btn.clicked.connect(self.browse_dir)

        layout.addWidget(self.label)
        layout.addWidget(self.line_edit, 1)
        layout.addWidget(self.browse_btn)

        self.setLayout(layout)

    def browse_dir(self):
        """浏览目录 - 智能记忆上次选择位置"""
        current_text = self.line_edit.text().strip()
        initial_dir = ""

        # 最高优先级：输入框已有路径存在
        if current_text:
            if os.path.isdir(current_text):
                initial_dir = current_text
            else:
                dir_path = os.path.dirname(current_text)
                if dir_path and os.path.exists(dir_path):
                    initial_dir = dir_path

        # 调用目录选择对话框
        dir_path = QFileDialog.getExistingDirectory(
            self, "选择目录", initial_dir
        )
        if dir_path:
            self.line_edit.setText(dir_path)

    def get_path(self):
        """获取路径"""
        return self.line_edit.text()

    def set_path(self, path):
        """设置路径"""
        self.line_edit.setText(str(path))


class FileSelectWidget(QWidget):
    """文件选择组件"""
    def __init__(self, label_text, file_filter="All Files (*.*)", mode="open", parent=None):
        """
        初始化文件选择组件

        Args:
            label_text: 标签文本
            file_filter: 文件过滤器
            mode: 选择模式 - "open"(打开文件) 或 "save"(保存文件)
            parent: 父控件
        """
        super().__init__(parent)
        self.file_filter = file_filter
        self.mode = mode  # "open" 或 "save"
        self.init_ui(label_text)

    def init_ui(self, label_text):
        layout = QHBoxLayout()
        layout.setContentsMargins(0, 0, 0, 0)

        self.label = QLabel(label_text)
        self.label.setMinimumWidth(120)
        self.line_edit = QLineEdit()
        placeholder = "请选择保存路径..." if self.mode == "save" else "请选择文件..."
        self.line_edit.setPlaceholderText(placeholder)
        self.browse_btn = QPushButton("浏览...")
        self.browse_btn.setMaximumWidth(80)
        self.browse_btn.clicked.connect(self.browse_file)

        layout.addWidget(self.label)
        layout.addWidget(self.line_edit, 1)
        layout.addWidget(self.browse_btn)

        self.setLayout(layout)

    def browse_file(self):
        """浏览文件 - 智能记忆上次选择位置"""
        current_text = self.line_edit.text().strip()
        initial_dir = ""

        # 最高优先级：输入框已有路径存在
        if current_text:
            if os.path.isdir(current_text):
                initial_dir = current_text
            else:
                dir_path = os.path.dirname(current_text)
                if dir_path and os.path.exists(dir_path):
                    initial_dir = dir_path

        if self.mode == "save":
            file_path, _ = QFileDialog.getSaveFileName(
                self, "保存文件", initial_dir, self.file_filter
            )
        else:
            file_path, _ = QFileDialog.getOpenFileName(
                self, "选择文件", initial_dir, self.file_filter
            )
        if file_path:
            self.line_edit.setText(file_path)

    def get_path(self):
        """获取路径"""
        return self.line_edit.text()

    def set_path(self, path):
        """设置路径"""
        self.line_edit.setText(str(path))