settings_window.py

# encoding = utf-8
import json
import logging
from tkinter import *
from tkinter.ttk import *

from i18n import I18n
import multiplatform
import widgets


class SettingsWindow(Toplevel):
    """设置窗口,使用分页(Notebook)布局并复用 `widgets.py` 中的封装控件。

    Pages:
    - 常规 (下载目录、语言)
    - 外观 (主题)
    """

    def __init__(self, master=None):
        super().__init__(master)
        self.title(I18n().get("settings_window.title"))
        self.geometry("520x380")
        self.iconbitmap("assets/icon.ico")
        self.resizable(False, False)
        self.transient(master)
        self.protocol("WM_DELETE_WINDOW", self.withdraw)
        # Notebook（分页）
        self.notebook = Notebook(self)
        self.notebook.pack(fill=BOTH, expand=True, padx=12, pady=(12, 6))

        # --- 常规页 ---
        general_frame = Frame(self)

        # 下载目录: 使用 widgets.FileChooser（封装控件）
        default_path = str(multiplatform.get_default_download_dir())
        # 注意: widgets.LabelWidget 会自动将自己的 StringVar 作为 subwidget 的 textvariable
        # 因此不要在 subwidget_kwargs 中再次传入 textvariable，避免重复参数错误。
        self.download_dir_widget = widgets.LabelWidget(general_frame, widgets.FileChooser,
                                                       label_text=I18n().get("settings_window.download_directory"),
                                                       subwidget_kwargs={
                                                           "choose_type": "folder",
                                                       })
        # 将内部 StringVar 作为窗口属性以便后续读取/保存
        self._download_dir = self.download_dir_widget.value
        self._download_dir.set(default_path)
        self.download_dir_widget.pack(side=TOP, fill=X, expand=False,)

        # 语言: 从 locales 动态读取可用语言
        i18n_inst = I18n()
        langs = i18n_inst.available_languages() or [i18n_inst.default_lang]
        self.language_widget = widgets.LabelWidget(general_frame, Combobox,
                                                   label_text=I18n().get("settings_window.language"),
                                                   widget_width=20,
                                                   subwidget_kwargs={
                                                       "values": langs,
                                                   })
        # 初始化语言值
        try:
            self.language_widget.value.set(i18n_inst.current_lang)
        except Exception:
            pass
        self.language_widget.pack(side=TOP, fill=X, expand=False,)

        # 下载完成后自动打开下载目录（默认）
        self.open_dir_after_finish_var = BooleanVar(value=True)
        self.open_dir_after_finish_cb = Checkbutton(
            general_frame,
            text=I18n().get("settings_window.open_dir_after_finish"),
            variable=self.open_dir_after_finish_var,
        )
        self.open_dir_after_finish_cb.pack(side=TOP, fill=X, expand=False, pady=2)

        self.notebook.add(general_frame, text=I18n().get("settings_window.general_tab"))

        # --- 外观页 ---
        # appearance_frame = Frame(self)
        # self.theme_widget = widgets.LabelWidget(appearance_frame, Combobox,
        #                                         label_text=I18n().get("settings_window.theme"),
        #                                         widget_width=20,
        #                                         subwidget_kwargs={
        #                                             "values": ["light", "dark"],
        #                                         })
        # try:
        #     self.theme_widget.value.set("dark")
        # except Exception:
        #     pass
        # self.theme_widget.pack(fill=X, expand=False,)

        # self.notebook.add(appearance_frame, text=I18n().get("settings_window.appearance_tab"))

        # --- 下载页 ---
        download_frame = Frame(self)

        # 基本下载选项（使用原生 Checkbutton，移除附带的 Entry）
        self.download_captions_var = BooleanVar(value=True)
        self.download_captions = Checkbutton(download_frame,
                            text=I18n().get("settings_window.download_captions"),
                            variable=self.download_captions_var)
        self.download_captions.pack(side=TOP, fill=X, expand=False, pady=2)

        self.merge_parts_var = BooleanVar(value=True)
        self.merge_parts = Checkbutton(download_frame,
                           text=I18n().get("settings_window.merge_video_parts"),
                           variable=self.merge_parts_var)
        self.merge_parts.pack(side=TOP, fill=X, expand=False, pady=2)

        self.download_m3u8_var = BooleanVar(value=False)
        self.download_m3u8 = Checkbutton(download_frame,
                         text=I18n().get("settings_window.download_m3u8_video"),
                         variable=self.download_m3u8_var)
        self.download_m3u8.pack(side=TOP, fill=X, expand=False, pady=2)

        # 自动重命名 / 忽略 SSL / 强制下载 / 跳过已存在
        # 自动重命名: 只有 True/False，使用 Checkbutton 更合适
        self.auto_rename_var = BooleanVar(value=True)
        self.auto_rename_cb = Checkbutton(download_frame,
                         text=I18n().get("settings_window.auto_rename"),
                         variable=self.auto_rename_var)
        self.auto_rename_cb.pack(side=TOP, fill=X, expand=False, pady=2)

        self.ignore_ssl_var = BooleanVar(value=False)
        self.ignore_ssl = Checkbutton(download_frame,
                          text=I18n().get("settings_window.ignore_ssl_errors"),
                          variable=self.ignore_ssl_var)
        self.ignore_ssl.pack(side=TOP, fill=X, expand=False, pady=2)

        self.forced_download_var = BooleanVar(value=False)
        self.forced_download = Checkbutton(download_frame,
                           text=I18n().get("settings_window.forced_download"),
                           variable=self.forced_download_var)
        self.forced_download.pack(side=TOP, fill=X, expand=False, pady=2)

        self.skip_downloaded_var = BooleanVar(value=False)
        self.skip_downloaded = Checkbutton(download_frame,
                          text=I18n().get("settings_window.skip_downloaded_video"),
                          variable=self.skip_downloaded_var)
        self.skip_downloaded.pack(side=TOP, fill=X, expand=False, pady=2)

        # 密码/格式/itag 支持（只有开关，已移除附带的 Entry）
        self.video_password_var = BooleanVar(value=False)
        self.video_password = Checkbutton(download_frame,
                         text=I18n().get("settings_window.download_video_password_power"),
                         variable=self.video_password_var)
        self.video_password.pack(side=TOP, fill=X, expand=False, pady=2)

        self.debug_var = BooleanVar(value=False)
        self.debug_mode = Checkbutton(download_frame, text=I18n().get("settings_window.debug_show"), variable=self.debug_var)
        self.debug_mode.pack(side=TOP, fill=X, expand=False, pady=2)

        self.notebook.add(download_frame, text=I18n().get("settings_window.download_tab") if I18n().get("settings_window.download_tab") else "Download")

        # --- 代理页 ---
        proxy_frame = Frame(self)
        # use widgets.CheckbtnWidget to keep .picked and .entry attributes
        self.proxy_power = widgets.CheckbtnWidget(proxy_frame, Frame,
                              check_text=I18n().get("settings_window.proxy_power"),
                              widget_width=40)
        self.proxy_power.picked.set(False)
        self.proxy_power.pack(side=TOP, fill=X, expand=False, pady=2)

        self.proxy_type = widgets.LabelWidget(proxy_frame, Combobox,
                                              label_text=I18n().get("settings_window.proxy_type"),
                                              widget_width=20,
                                              subwidget_kwargs={"values": ["Http", "Socks5"]})
        self.proxy_type.value.set("Socks5")
        self.proxy_type.pack(side=TOP, fill=X, expand=False, pady=2)

        self.proxy_host = widgets.LabelWidget(proxy_frame, Entry,
                                              label_text=I18n().get("settings_window.proxy_host"),
                                              widget_width=40)
        self.proxy_host.pack(side=TOP, fill=X, expand=False, pady=2)

        self.proxy_login = widgets.CheckbtnWidget(proxy_frame, Entry,
                                                  check_text=I18n().get("settings_window.proxy_login_power"),
                                                  widget_width=40)
        self.proxy_login.picked.set(False)
        self.proxy_login.pack(side=TOP, fill=X, expand=False, pady=2)

        # 根据 proxy_power 开关启用/禁用代理子控件
        def _trace_var(var, cb):
            try:
                var.trace_add('write', lambda *a: cb())
            except Exception:
                var.trace('w', lambda *a: cb())

        def _set_proxy_controls():
            enabled = bool(self.proxy_power.picked.get())
            try:
                # Combobox: 使用 'readonly' 或 'disabled'
                self.proxy_type.subwidget.config(state='readonly' if enabled else 'disabled')
            except Exception:
                pass
            try:
                self.proxy_host.subwidget.config(state='normal' if enabled else 'disabled')
            except Exception:
                pass
            try:
                self.proxy_login.check.config(state='normal' if enabled else 'disabled')
                # 只有在 proxy_login.picked 且 proxy enabled 时才启用 entry
                login_enabled = enabled and bool(self.proxy_login.picked.get())
                self.proxy_login.entry.config(state='normal' if login_enabled else 'disabled')
            except Exception:
                logging.warning("Failed to set proxy controls state", exc_info=True)

        def _on_proxy_login_changed():
            # 根据 proxy 和 proxy_login 两个开关决定 login entry 是否可用
            login_enabled = bool(self.proxy_power.picked.get()) and bool(self.proxy_login.picked.get())
            try:
                self.proxy_login.entry.config(state='normal' if login_enabled else 'disabled')
            except Exception:
                logging.warning("Failed to set proxy login entry state", exc_info=True)

        _trace_var(self.proxy_power.picked, _set_proxy_controls)
        _trace_var(self.proxy_login.picked, _on_proxy_login_changed)
        # 初始状态
        _set_proxy_controls()

        self.notebook.add(proxy_frame, text=I18n().get("settings_window.proxy_tab") if I18n().get("settings_window.proxy_tab") else "Proxy")

        # --- 批量页 ---
        # batch_frame = Frame(self)
        # self.batch_power = widgets.CheckbtnWidget(batch_frame, Entry,
        #                                           check_text=I18n().get("settings_window.batch_download_power"),
        #                                           widget_width=40)
        # self.batch_power.picked.set(False)
        # self.batch_power.pack(side=TOP, fill=X, expand=True, pady=2)

        # self.batch_parallel = widgets.CheckbtnWidget(batch_frame, Entry,
        #                                              check_text=I18n().get("settings_window.batch_download_parallel"),
        #                                              widget_width=40)
        # self.batch_parallel.picked.set(True)
        # self.batch_parallel.pack(side=TOP, fill=X, expand=True, pady=2)

        # self.batch_urls = widgets.LabelWidget(batch_frame, Entry,
        #                                       label_text=I18n().get("settings_window.batch_download_urls"),
        #                                       widget_width=60)
        # self.batch_urls.pack(side=TOP, fill=X, expand=True, pady=2)

        # self.notebook.add(batch_frame, text=I18n().get("settings_window.batch_tab") if I18n().get("settings_window.batch_tab") else "Batch")

        # # --- 通知页 ---
        # notify_frame = Frame(self)
        # # 使用普通的 Checkbutton（也可替换为 widgets.CheckbtnWidget）
        # self.notifications_var = BooleanVar(value=True)
        # self.notifications_cb = Checkbutton(notify_frame, text=I18n().get("settings_window.notifications"),
        #                                     variable=self.notifications_var)
        # self.notifications_cb.pack(anchor=W, pady=6)

        # # 声音开关示例
        # self.sound_var = BooleanVar(value=False)
        # self.sound_cb = Checkbutton(notify_frame, text=I18n().get("settings_window.sound"), variable=self.sound_var)
        # self.sound_cb.pack(anchor=W, pady=2)

        # self.notebook.add(notify_frame, text=I18n().get("settings_window.notifications_tab"))

        # 底部按钮行：保存 / 关闭
        btn_frame = Frame(self)
        btn_frame.pack(fill=X, padx=12, pady=(0, 12))

        self.save_btn = Button(btn_frame, text=I18n().get("settings_window.save"), command=self.save_settings)
        self.save_btn.pack(side=RIGHT, padx=(6, 0))

        self.close_btn = Button(btn_frame, text=I18n().get("settings_window.close"), command=self.close_window)
        self.close_btn.pack(side=RIGHT)

        # 加载当前配置（若有）
        self.load_settings()

    def deiconify(self):
        """Show the settings window and acquire a local grab so it acts modal."""
        super().deiconify()
        self.lift()
        self.focus_force()
        self.load_settings()  # 每次显示时加载最新设置
        try:
            self.grab_set()
        except Exception:
            logging.debug("Failed to set grab on settings window", exc_info=True)

    def withdraw(self):
        """Release any grab and hide the window."""
        try:
            self.grab_release()
        except Exception:
            logging.debug("Failed to release grab on settings window", exc_info=True)
        
        super().withdraw()

    def save_settings(self):
        download_dir = self._download_dir.get() if hasattr(self, "_download_dir") else None
        language = None
        try:
            language = self.language_widget.value.get()
        except Exception:
            logging.warning("Failed to get language setting", exc_info=True)
        # theme = None
        # try:
        #     theme = self.theme_widget.value.get()
        # except Exception:
        #     pass
        # 收集下载选项
        download_opts = {
            "download_captions": bool(self.download_captions_var.get()),
            "merge_video_parts": bool(self.merge_parts_var.get()),
            "download_m3u8_video": bool(self.download_m3u8_var.get()),
            "auto_rename": bool(self.auto_rename_var.get()),
            "ignore_ssl_errors": bool(self.ignore_ssl_var.get()),
            "forced_download": bool(self.forced_download_var.get()),
            "skip_downloaded_video": bool(self.skip_downloaded_var.get()),
            "video_password_power": bool(self.video_password_var.get()),
            "video_password_value": "",
            "open_dir_after_finish": bool(self.open_dir_after_finish_var.get()),
            "debug": bool(self.debug_var.get()),
        }

        # 代理选项
        proxy_opts = {
            "proxy_power": bool(self.proxy_power.picked.get()),
            "proxy_type": self.proxy_type.value.get(),
            "proxy_host": self.proxy_host.value.get(),
            "proxy_login_power": bool(self.proxy_login.picked.get()),
            "proxy_login_value": self.proxy_login.value.get() if self.proxy_login.picked.get() else "",
        }

        # 批量选项
        # batch_opts = {}
        # if hasattr(self, 'batch_power') and hasattr(self, 'batch_parallel') and hasattr(self, 'batch_urls'):
        #     batch_opts = {
        #         "batch_power": bool(self.batch_power.picked.get()),
        #         "parallel": bool(self.batch_parallel.picked.get()),
        #         "urls": self.batch_urls.value.get(),
        #     }

        with open("config.json", "w") as f:
            json.dump({
                "download_dir": download_dir,
                "language": language,
                # "theme": theme,
                "download_options": download_opts,
                "proxy_options": proxy_opts,
                # "batch_options": batch_opts,
            }, f, indent=4)
        self.close_window()

    def load_settings(self):
        try:
            with open("config.json", "r") as f:
                config = json.load(f)
                # 加载下载目录
                if "download_dir" in config and hasattr(self, "_download_dir"):
                    self._download_dir.set(config["download_dir"])
                # 加载语言
                if "language" in config and hasattr(self, "language_widget"):
                    self.language_widget.value.set(config["language"])
                # 加载下载选项
                download_opts = config.get("download_options", {})
                if download_opts:
                    self.download_captions_var.set(download_opts.get("download_captions", True))
                    self.merge_parts_var.set(download_opts.get("merge_video_parts", True))
                    self.download_m3u8_var.set(download_opts.get("download_m3u8_video", False))
                    self.auto_rename_var.set(download_opts.get("auto_rename", True))
                    self.ignore_ssl_var.set(download_opts.get("ignore_ssl_errors", False))
                    self.forced_download_var.set(download_opts.get("forced_download", False))
                    self.skip_downloaded_var.set(download_opts.get("skip_downloaded_video", False))
                    self.video_password_var.set(download_opts.get("video_password_power", False))
                    self.open_dir_after_finish_var.set(download_opts.get("open_dir_after_finish", False))
                    self.debug_var.set(download_opts.get("debug", False))
                # 加载代理选项
                proxy_opts = config.get("proxy_options", {})
                if proxy_opts:
                    self.proxy_power.picked.set(proxy_opts.get("proxy_power", False))
                    self.proxy_type.value.set(proxy_opts.get("proxy_type", "Socks5"))
                    self.proxy_host.value.set(proxy_opts.get("proxy_host", ""))
                    self.proxy_login.picked.set(proxy_opts.get("proxy_login_power", False))
                    self.proxy_login.value.set(proxy_opts.get("proxy_login_value", ""))
        except Exception:
            logging.info("Failed to load settings, using defaults", exc_info=True)

    def close_window(self):
        self.withdraw()


if __name__ == '__main__':
    root = Tk()
    # root.withdraw()  # Hide the main window
    settings_window = SettingsWindow(root)
    root.mainloop()