"""
主GUI界面
Etsy客户营销工具的主窗口
"""

import tkinter as tk
from tkinter import ttk, messagebox, filedialog
from datetime import datetime, timedelta
from typing import List
import threading

from ..core.config import Config
from ..core.database import Database
from ..etsy.etsy_manager import EtsyManager
from ..core.excel_exporter import ExcelExporter


class MainWindow:
    """主窗口类"""

    def __init__(self, root):
        self.root = root
        self.config = Config()
        self.database = Database()
        self.etsy_manager = None
        self.excel_exporter = ExcelExporter(self.database)

        self.setup_window()
        self.create_widgets()
        self.load_config_data()

        # 绑定窗口关闭事件
        self.root.protocol("WM_DELETE_WINDOW", self.on_closing)

    def on_closing(self):
        """窗口关闭事件处理"""
        from ..etsy.chrome_controller import ChromeController

        # 停止浏览器
        chrome = ChromeController()
        if chrome.is_running:
            chrome.stop_browser()

        # 关闭窗口
        self.root.destroy()

    def setup_window(self):
        """设置窗口属性"""
        self.root.title("Etsy客户营销工具 v1.0")
        self.root.geometry("800x700")
        self.root.resizable(True, True)

        # 设置图标（如果有的话）
        try:
            self.root.iconbitmap("assets/icon.ico")
        except:
            pass

    def create_widgets(self):
        """创建所有界面组件"""
        # 创建主框架
        main_frame = ttk.Frame(self.root, padding="10")
        main_frame.grid(row=0, column=0, sticky=(tk.W, tk.E, tk.N, tk.S))

        # 配置网格权重
        self.root.columnconfigure(0, weight=1)
        self.root.rowconfigure(0, weight=1)
        main_frame.columnconfigure(1, weight=1)

        row = 0

        # 标题
        title_label = ttk.Label(main_frame, text="Etsy客户营销工具", font=("Arial", 16, "bold"))
        title_label.grid(row=row, column=0, columnspan=2, pady=(0, 20))
        row += 1

        # 日期范围选择
        ttk.Label(main_frame, text="日期范围:").grid(row=row, column=0, sticky=tk.W, pady=5)
        date_frame = ttk.Frame(main_frame)
        date_frame.grid(row=row, column=1, sticky=(tk.W, tk.E), pady=5)

        self.start_date_var = tk.StringVar(value=(datetime.now() - timedelta(days=30)).strftime('%Y-%m-%d'))
        self.end_date_var = tk.StringVar(value=datetime.now().strftime('%Y-%m-%d'))

        ttk.Label(date_frame, text="从:").pack(side=tk.LEFT)
        self.start_date_entry = ttk.Entry(date_frame, textvariable=self.start_date_var, width=12)
        self.start_date_entry.pack(side=tk.LEFT, padx=5)

        ttk.Label(date_frame, text="到:").pack(side=tk.LEFT, padx=(10, 0))
        self.end_date_entry = ttk.Entry(date_frame, textvariable=self.end_date_var, width=12)
        self.end_date_entry.pack(side=tk.LEFT, padx=5)

        row += 1

        # 用户标签选择
        ttk.Label(main_frame, text="排除标签:").grid(row=row, column=0, sticky=(tk.W, tk.N), pady=5)
        tags_frame = ttk.Frame(main_frame)
        tags_frame.grid(row=row, column=1, sticky=(tk.W, tk.E), pady=5)

        # 标签列表框架
        self.tags_listbox_frame = ttk.Frame(tags_frame)
        self.tags_listbox_frame.pack(fill=tk.BOTH, expand=True)

        # 标签列表
        self.tags_listbox = tk.Listbox(self.tags_listbox_frame, selectmode=tk.MULTIPLE, height=6)
        tags_scrollbar = ttk.Scrollbar(self.tags_listbox_frame, orient=tk.VERTICAL, command=self.tags_listbox.yview)
        self.tags_listbox.configure(yscrollcommand=tags_scrollbar.set)

        self.tags_listbox.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
        tags_scrollbar.pack(side=tk.RIGHT, fill=tk.Y)

        # 刷新标签按钮
        ttk.Button(tags_frame, text="刷新标签", command=self.refresh_tags).pack(pady=5)

        row += 1

        # 营销消息设置
        ttk.Label(main_frame, text="营销消息:").grid(row=row, column=0, sticky=(tk.W, tk.N), pady=5)
        messages_frame = ttk.Frame(main_frame)
        messages_frame.grid(row=row, column=1, sticky=(tk.W, tk.E), pady=5)

        # 消息列表
        self.messages_listbox = tk.Listbox(messages_frame, height=8)
        messages_scrollbar = ttk.Scrollbar(messages_frame, orient=tk.VERTICAL, command=self.messages_listbox.yview)
        self.messages_listbox.configure(yscrollcommand=messages_scrollbar.set)

        self.messages_listbox.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
        messages_scrollbar.pack(side=tk.RIGHT, fill=tk.Y)

        # 消息操作按钮
        messages_btn_frame = ttk.Frame(messages_frame)
        messages_btn_frame.pack(fill=tk.X, pady=5)

        ttk.Button(messages_btn_frame, text="添加消息", command=self.add_message).pack(side=tk.LEFT, padx=2)
        ttk.Button(messages_btn_frame, text="编辑消息", command=self.edit_message).pack(side=tk.LEFT, padx=2)
        ttk.Button(messages_btn_frame, text="删除消息", command=self.delete_message).pack(side=tk.LEFT, padx=2)

        row += 1

        # 状态检查和控制按钮
        control_frame = ttk.LabelFrame(main_frame, text="控制面板", padding="10")
        control_frame.grid(row=row, column=0, columnspan=2, sticky=(tk.W, tk.E), pady=10)

        # 第一行按钮
        btn_row1 = ttk.Frame(control_frame)
        btn_row1.pack(fill=tk.X, pady=2)

        self.check_login_btn = ttk.Button(btn_row1, text="检查登录状态", command=self.check_login_status)
        self.check_login_btn.pack(side=tk.LEFT, padx=5)

        self.reset_marketing_btn = ttk.Button(btn_row1, text="重置营销消息", command=self.reset_marketing_data)
        self.reset_marketing_btn.pack(side=tk.LEFT, padx=5)

        ttk.Button(btn_row1, text="导出记录", command=self.export_records).pack(side=tk.LEFT, padx=5)

        # 第二行按钮
        btn_row2 = ttk.Frame(control_frame)
        btn_row2.pack(fill=tk.X, pady=2)

        self.start_conversation_btn = ttk.Button(btn_row2, text="启动对话营销", command=self.start_conversation_marketing)
        self.start_conversation_btn.pack(side=tk.LEFT, padx=5)

        self.start_order_btn = ttk.Button(btn_row2, text="启动订单营销", command=self.start_order_marketing)
        self.start_order_btn.pack(side=tk.LEFT, padx=5)

        self.stop_btn = ttk.Button(btn_row2, text="停止营销", command=self.stop_marketing, state=tk.DISABLED)
        self.stop_btn.pack(side=tk.LEFT, padx=5)

        row += 1

        # 状态显示
        status_frame = ttk.LabelFrame(main_frame, text="状态信息", padding="10")
        status_frame.grid(row=row, column=0, columnspan=2, sticky=(tk.W, tk.E), pady=10)

        self.status_text = tk.Text(status_frame, height=8, wrap=tk.WORD)
        status_scrollbar = ttk.Scrollbar(status_frame, orient=tk.VERTICAL, command=self.status_text.yview)
        self.status_text.configure(yscrollcommand=status_scrollbar.set)

        self.status_text.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
        status_scrollbar.pack(side=tk.RIGHT, fill=tk.Y)

        row += 1

        # 进度条
        self.progress = ttk.Progressbar(main_frame, mode='indeterminate')
        self.progress.grid(row=row, column=0, columnspan=2, sticky=(tk.W, tk.E), pady=5)

    def load_config_data(self):
        """加载配置数据"""
        # 加载营销消息
        messages = self.config.get_marketing_messages()
        for msg in messages:
            self.messages_listbox.insert(tk.END, msg[:50] + "..." if len(msg) > 50 else msg)

        # 加载用户标签
        self.refresh_tags()

        self.log_status("程序启动完成")

    def log_status(self, message: str):
        """添加状态日志"""
        timestamp = datetime.now().strftime("%H:%M:%S")
        full_message = f"[{timestamp}] {message}\n"

        self.status_text.insert(tk.END, full_message)
        self.status_text.see(tk.END)
        self.root.update_idletasks()

    def refresh_tags(self):
        """刷新用户标签"""
        def refresh_task():
            try:
                self.progress.start()
                self.log_status("刷新标签中...")

                if not self.etsy_manager:
                    self.etsy_manager = EtsyManager()

                tags = self.etsy_manager.get_user_tags()
                if tags:
                    self.tags_listbox.delete(0, tk.END)
                    for tag_name, count in tags.items():
                        display_text = f"{tag_name} ({count})"
                        self.tags_listbox.insert(tk.END, display_text)

                    self.database.update_user_tags(tags)
                    self.log_status(f"标签刷新完成，共{len(tags)}个标签")
                else:
                    self.log_status("标签刷新失败，请检查网络连接")

            except Exception as e:
                self.log_status(f"标签刷新异常: {str(e)}")
            finally:
                self.progress.stop()

        threading.Thread(target=refresh_task, daemon=True).start()

    def add_message(self):
        """添加营销消息"""
        dialog = MessageDialog(self.root, "添加营销消息", "")
        if dialog.result:
            self.config.add_marketing_message(dialog.result)
            self.config.save()

            display_text = dialog.result[:50] + "..." if len(dialog.result) > 50 else dialog.result
            self.messages_listbox.insert(tk.END, display_text)
            self.log_status("营销消息已添加")

    def edit_message(self):
        """编辑营销消息"""
        selection = self.messages_listbox.curselection()
        if not selection:
            messagebox.showwarning("警告", "请先选择要编辑的消息")
            return

        index = selection[0]
        messages = self.config.get_marketing_messages()
        if index < len(messages):
            old_message = messages[index]
            dialog = MessageDialog(self.root, "编辑营销消息", old_message)
            if dialog.result:
                messages[index] = dialog.result
                self.config.set("marketing_messages", messages)
                self.config.save()

                display_text = dialog.result[:50] + "..." if len(dialog.result) > 50 else dialog.result
                self.messages_listbox.delete(index)
                self.messages_listbox.insert(index, display_text)
                self.log_status("营销消息已更新")

    def delete_message(self):
        """删除营销消息"""
        selection = self.messages_listbox.curselection()
        if not selection:
            messagebox.showwarning("警告", "请先选择要删除的消息")
            return

        if messagebox.askyesno("确认删除", "确定要删除选中的营销消息吗？"):
            index = selection[0]
            messages = self.config.get_marketing_messages()
            if index < len(messages):
                del messages[index]
                self.config.set("marketing_messages", messages)
                self.config.save()

                self.messages_listbox.delete(index)
                self.log_status("营销消息已删除")

    def check_login_status(self):
        """检查Etsy登录状态"""
        def check_task():
            try:
                self.progress.start()
                self.check_login_btn.configure(state=tk.DISABLED)
                self.log_status("检查登录状态中...")

                if not self.etsy_manager:
                    self.etsy_manager = EtsyManager()

                status = self.etsy_manager.check_login_status()
                if status["logged_in"]:
                    self.log_status("✓ Etsy登录状态正常")
                    if status["message_access"]:
                        self.log_status("✓ 消息访问正常")
                    if status["tag_access"]:
                        self.log_status("✓ 标签获取正常")
                else:
                    self.log_status("✗ Etsy未登录，请先在浏览器中登录Etsy")

            except Exception as e:
                self.log_status(f"登录状态检查异常: {str(e)}")
            finally:
                self.progress.stop()
                self.check_login_btn.configure(state=tk.NORMAL)

        threading.Thread(target=check_task, daemon=True).start()

    def reset_marketing_data(self):
        """重置营销数据"""
        result = messagebox.askyesnocancel(
            "确认重置",
            "将会忽略历史的发送用户和订单，重新发送新的营销消息，\n"
            "请仔细核查以避免重复发送对客户产生困扰，是否确认？"
        )

        if result:
            self.database.clear_sent_records()
            self.config.reset_marketing_data()
            self.config.save()
            self.log_status("营销数据已重置")

    def get_excluded_tags(self) -> List[str]:
        """获取选中的排除标签"""
        selected_indices = self.tags_listbox.curselection()
        excluded_tags = []

        for index in selected_indices:
            tag_text = self.tags_listbox.get(index)
            # 提取标签名称（去掉计数部分）
            tag_name = tag_text.split(" (")[0]
            excluded_tags.append(tag_name)

        return excluded_tags

    def validate_settings(self) -> bool:
        """验证设置"""
        messages = self.config.get_marketing_messages()
        if len(messages) < 5:
            messagebox.showerror("错误", "营销消息至少需要5条，当前只有{}条".format(len(messages)))
            return False

        try:
            datetime.strptime(self.start_date_var.get(), '%Y-%m-%d')
            datetime.strptime(self.end_date_var.get(), '%Y-%m-%d')
        except ValueError:
            messagebox.showerror("错误", "日期格式不正确，请使用YYYY-MM-DD格式")
            return False

        return True

    def start_conversation_marketing(self):
        """启动对话营销"""
        if not self.validate_settings():
            return

        def marketing_task():
            try:
                self.progress.start()
                self._set_marketing_buttons_state(False)
                self.log_status("启动对话营销...")

                if not self.etsy_manager:
                    self.etsy_manager = EtsyManager()

                excluded_tags = self.get_excluded_tags()
                self.etsy_manager.start_conversation_marketing(
                    start_date=self.start_date_var.get(),
                    end_date=self.end_date_var.get(),
                    excluded_tags=excluded_tags,
                    status_callback=self.log_status
                )

            except Exception as e:
                self.log_status(f"对话营销异常: {str(e)}")
            finally:
                self.progress.stop()
                self._set_marketing_buttons_state(True)

        threading.Thread(target=marketing_task, daemon=True).start()

    def start_order_marketing(self):
        """启动订单营销"""
        if not self.validate_settings():
            return

        def marketing_task():
            try:
                self.progress.start()
                self._set_marketing_buttons_state(False)
                self.log_status("启动订单营销...")

                if not self.etsy_manager:
                    self.etsy_manager = EtsyManager()

                excluded_tags = self.get_excluded_tags()
                self.etsy_manager.start_order_marketing(
                    start_date=self.start_date_var.get(),
                    end_date=self.end_date_var.get(),
                    excluded_tags=excluded_tags,
                    status_callback=self.log_status
                )

            except Exception as e:
                self.log_status(f"订单营销异常: {str(e)}")
            finally:
                self.progress.stop()
                self._set_marketing_buttons_state(True)

        threading.Thread(target=marketing_task, daemon=True).start()

    def stop_marketing(self):
        """停止营销"""
        if self.etsy_manager:
            self.etsy_manager.stop_marketing()
            self.log_status("营销已停止")

    def _set_marketing_buttons_state(self, enabled: bool):
        """设置营销按钮状态"""
        state = tk.NORMAL if enabled else tk.DISABLED
        self.start_conversation_btn.configure(state=state)
        self.start_order_btn.configure(state=state)
        self.stop_btn.configure(state=tk.DISABLED if enabled else tk.NORMAL)

    def export_records(self):
        """导出记录"""
        file_path = filedialog.asksaveasfilename(
            defaultextension=".xlsx",
            filetypes=[("Excel files", "*.xlsx"), ("All files", "*.*")]
        )

        if file_path:
            try:
                success = self.excel_exporter.export_to_excel(
                    file_path,
                    self.start_date_var.get(),
                    self.end_date_var.get()
                )

                if success:
                    self.log_status(f"记录已导出到: {file_path}")
                    messagebox.showinfo("成功", "记录导出成功")
                else:
                    messagebox.showerror("错误", "记录导出失败")

            except Exception as e:
                self.log_status(f"导出异常: {str(e)}")
                messagebox.showerror("错误", f"导出失败: {str(e)}")


class MessageDialog:
    """消息编辑对话框"""

    def __init__(self, parent, title, initial_text=""):
        self.result = None

        self.dialog = tk.Toplevel(parent)
        self.dialog.title(title)
        self.dialog.geometry("500x300")
        self.dialog.transient(parent)
        self.dialog.grab_set()

        # 居中显示
        self.dialog.geometry("+%d+%d" % (parent.winfo_rootx() + 50, parent.winfo_rooty() + 50))

        # 创建界面
        frame = ttk.Frame(self.dialog, padding="10")
        frame.pack(fill=tk.BOTH, expand=True)

        ttk.Label(frame, text="营销消息内容:").pack(anchor=tk.W)

        self.text_widget = tk.Text(frame, wrap=tk.WORD, height=10)
        scrollbar = ttk.Scrollbar(frame, orient=tk.VERTICAL, command=self.text_widget.yview)
        self.text_widget.configure(yscrollcommand=scrollbar.set)

        self.text_widget.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
        scrollbar.pack(side=tk.RIGHT, fill=tk.Y)

        if initial_text:
            self.text_widget.insert(tk.END, initial_text)

        # 按钮
        btn_frame = ttk.Frame(self.dialog)
        btn_frame.pack(fill=tk.X, pady=10)

        ttk.Button(btn_frame, text="确定", command=self.ok_clicked).pack(side=tk.RIGHT, padx=5)
        ttk.Button(btn_frame, text="取消", command=self.cancel_clicked).pack(side=tk.RIGHT)

        self.dialog.protocol("WM_DELETE_WINDOW", self.cancel_clicked)
        self.dialog.wait_window()

    def ok_clicked(self):
        text = self.text_widget.get(1.0, tk.END).strip()
        if text:
            self.result = text
            self.dialog.destroy()
        else:
            messagebox.showwarning("警告", "消息内容不能为空")

    def cancel_clicked(self):
        self.dialog.destroy()