fix: use checkbox style for setup alias selection (#153)

Author: RexWzh

CHANGELOG.md

@@ -46,6 +46,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/), and this
 - `skills/feishu/` 补回 `SKILL.zh.md`，避免技能资产检查在 CI 中因缺少中文入口文件失败
 - `chattool gh pr-merge` 新增可选的 `--check` 开关，用于在合并前显式检查 check runs 与 workflow runs，避免再次误把带红 CI 的 PR 当成可安全合并
 - `chattool skill install` 不再强制要求 skill frontmatter 包含 `version`，只校验 `name` 与 `description`
+- `chattool setup alias` 的自定义多选现在统一走 `utils/tui.py` 的 checkbox 封装，交互里显式显示 `☑/☐` 勾选态，不再只靠高亮区分选中项
 
 ### Added
 - 新增根目录 `Dockerfile.playground`，用于直接构建一个最小的 ChatTool Playground 镜像；镜像在 `/opt/venv` 中安装 ChatTool，容器启动后会线性执行 `chattool setup playground -> chattool env set CHATTOOL_SKILLS_DIR -> chattool setup alias`

src/chattool/setup/alias.py

@@ -4,7 +4,7 @@
 import click
 
 from chattool.utils.custom_logger import setup_logger
-from chattool.utils.tui import ask_select, is_interactive_available
+from chattool.utils.tui import ask_checkbox, ask_select, is_interactive_available
 
 logger = setup_logger("setup_alias")
 
@@ -86,16 +86,20 @@ def select_aliases_interactively(default_selected):
         return list(ALIAS_MAP.keys())
     if preset == "Unselect all":
         return []
-    import questionary
+    from chattool.utils.tui import create_choice
 
     choices = [
-        questionary.Choice(title=f"{name} => {cmd}", value=name, checked=name in default_selected)
+        create_choice(
+            title=f"{name} => {cmd}",
+            value=name,
+            checked=name in default_selected,
+        )
         for name, cmd in ALIAS_MAP.items()
     ]
-    selected = questionary.checkbox(
+    selected = ask_checkbox(
         "Select aliases (Space to toggle, A select/unselect all)",
         choices=choices,
-    ).ask()
+    )
     return selected or []
 
 

src/chattool/utils/tui.py

@@ -1,4 +1,9 @@
+from contextlib import contextmanager
+
+
 BACK_VALUE = "__BACK__"
+CHECKBOX_SELECTED_INDICATOR = "☑"
+CHECKBOX_UNSELECTED_INDICATOR = "☐"
 
 def get_style():
     """Custom style for questionary."""
@@ -47,10 +52,10 @@ def get_separator():
     import questionary
     return questionary.Separator()
 
-def create_choice(title, value):
+def create_choice(title, value, checked=False):
     """Return a questionary Choice."""
     import questionary
-    return questionary.Choice(title=title, value=value)
+    return questionary.Choice(title=title, value=value, checked=checked)
 
 def ask_select(message, choices, style=None):
     """Ask a selection question with escape back support."""
@@ -64,6 +69,45 @@ def ask_select(message, choices, style=None):
         use_arrow_keys=True
     ))
 
+
+@contextmanager
+def checkbox_indicator_style():
+    """Temporarily render questionary checkboxes as checkbox glyphs."""
+    import questionary.constants as constants
+    import questionary.prompts.common as common
+
+    old_selected = constants.INDICATOR_SELECTED
+    old_unselected = constants.INDICATOR_UNSELECTED
+    old_common_selected = common.INDICATOR_SELECTED
+    old_common_unselected = common.INDICATOR_UNSELECTED
+
+    constants.INDICATOR_SELECTED = CHECKBOX_SELECTED_INDICATOR
+    constants.INDICATOR_UNSELECTED = CHECKBOX_UNSELECTED_INDICATOR
+    common.INDICATOR_SELECTED = CHECKBOX_SELECTED_INDICATOR
+    common.INDICATOR_UNSELECTED = CHECKBOX_UNSELECTED_INDICATOR
+    try:
+        yield
+    finally:
+        constants.INDICATOR_SELECTED = old_selected
+        constants.INDICATOR_UNSELECTED = old_unselected
+        common.INDICATOR_SELECTED = old_common_selected
+        common.INDICATOR_UNSELECTED = old_common_unselected
+
+
+def ask_checkbox(message, choices, style=None, instruction=None):
+    """Ask a checkbox question with shared style and escape back support."""
+    import questionary
+    if style is None:
+        style = get_style()
+    with checkbox_indicator_style():
+        return ask_with_escape_back(questionary.checkbox(
+            message,
+            choices=choices,
+            style=style,
+            use_arrow_keys=True,
+            instruction=instruction,
+        ))
+
 def ask_text(message, default="", password=False, style=None):
     """Ask a text question with escape back support."""
     import questionary

tests/test_setup_alias.py

@@ -8,6 +8,7 @@
     render_alias_block,
     resolve_shell,
     resolve_shell_rc,
+    select_aliases_interactively,
     setup_alias,
 )
 
@@ -70,3 +71,24 @@ def test_setup_alias_dry_run_does_not_write(tmp_path, monkeypatch, capsys):
     out = capsys.readouterr().out
     assert "[dry-run] target shell rc:" in out
     assert "alias chatenv='chattool env'" in out
+
+
+def test_select_aliases_interactively_custom_uses_checkbox(monkeypatch):
+    monkeypatch.setattr("chattool.setup.alias.ask_select", lambda *args, **kwargs: "Custom")
+
+    captured = {}
+
+    def fake_ask_checkbox(message, choices, style=None, instruction=None):
+        captured["message"] = message
+        captured["choices"] = choices
+        return ["chatgh"]
+
+    monkeypatch.setattr("chattool.setup.alias.ask_checkbox", fake_ask_checkbox)
+    selected = select_aliases_interactively(["chatenv", "chatgh"])
+
+    assert selected == ["chatgh"]
+    assert captured["message"] == "Select aliases (Space to toggle, A select/unselect all)"
+    checked = {choice.value: choice.checked for choice in captured["choices"]}
+    assert checked["chatenv"] is True
+    assert checked["chatgh"] is True
+    assert checked["chatdns"] is False