Let setup wizard enable IM channels

scripts/wizard/steps/channels.py

@@ -0,0 +1,46 @@
+"""Step: browser-connectable IM channel enablement."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from wizard.ui import ask_multi_choice, print_header, print_info, print_success
+
+
+CHANNEL_CONNECTION_OPTIONS: tuple[tuple[str, str, str], ...] = (
+    ("telegram", "Telegram", "direct messages through your DeerFlow bot"),
+    ("slack", "Slack", "workspace messages and mentions"),
+    ("discord", "Discord", "server messages through your DeerFlow bot"),
+    ("feishu", "Feishu / Lark", "messages through your DeerFlow app"),
+    ("dingtalk", "DingTalk", "Stream Push messages through your DeerFlow bot"),
+    ("wechat", "WeChat", "iLink messages through your DeerFlow bot"),
+    ("wecom", "WeCom", "messages through your DeerFlow AI bot"),
+)
+
+
+@dataclass
+class ChannelConnectionsStepResult:
+    enabled_providers: list[str]
+
+
+def run_channels_step(step_label: str = "Step 4/5") -> ChannelConnectionsStepResult:
+    print_header(f"{step_label} · IM Channels (optional)")
+    print_info("Choose which IM channels should appear in the DeerFlow sidebar and Settings.")
+    print_info("Credentials can be entered later from the browser with Connect or Modify.")
+    print()
+
+    options = [f"{display_name}  —  {description}" for _, display_name, description in CHANNEL_CONNECTION_OPTIONS]
+    selected = ask_multi_choice(
+        "Enable channels (comma-separated numbers, 'all', or Enter for none)",
+        options,
+        default=[],
+    )
+    enabled_providers = [CHANNEL_CONNECTION_OPTIONS[idx][0] for idx in selected]
+
+    if enabled_providers:
+        display_names = [CHANNEL_CONNECTION_OPTIONS[idx][1] for idx in selected]
+        print_success(f"Enabled channels: {', '.join(display_names)}")
+    else:
+        print_info("No IM channels selected; channel connections will stay disabled.")
+
+    return ChannelConnectionsStepResult(enabled_providers=enabled_providers)

scripts/wizard/ui.py

@@ -224,6 +224,49 @@ def ask_choice(prompt: str, options: list[str], default: int | None = None) -> i
     return _ask_choice_with_numbers(prompt, options, default=default)
 
 
+def ask_multi_choice(prompt: str, options: list[str], default: list[int] | None = None) -> list[int]:
+    """Present a numbered multi-select menu and return 0-based indexes."""
+    has_default = default is not None
+    default_indexes = list(default or [])
+    for i, opt in enumerate(options, 1):
+        marker = f" {green('*')}" if has_default and i - 1 in default_indexes else "  "
+        print(f"{marker} {i}. {opt}")
+    print()
+
+    suffix = ""
+    if default_indexes:
+        suffix = f" [{','.join(str(idx + 1) for idx in default_indexes)}]"
+    elif has_default:
+        suffix = " [none]"
+
+    while True:
+        raw = input(f"{prompt}{suffix}: ").strip().lower()
+        if raw == "" and has_default:
+            return default_indexes
+        if raw in {"none", "no", "n", "skip"}:
+            return []
+        if raw == "all":
+            return list(range(len(options)))
+
+        parts = [part.strip() for part in raw.replace(" ", ",").split(",") if part.strip()]
+        selected: list[int] = []
+        valid = bool(parts)
+        for part in parts:
+            if not part.isdigit():
+                valid = False
+                break
+            idx = int(part) - 1
+            if not 0 <= idx < len(options):
+                valid = False
+                break
+            if idx not in selected:
+                selected.append(idx)
+        if valid:
+            return selected
+
+        print(f"  Enter comma-separated numbers between 1 and {len(options)}, 'all', or 'none'.")
+
+
 def ask_text(prompt: str, default: str = "", required: bool = False) -> str:
     """Ask for a text value, returning default if the user presses Enter."""
     suffix = f" [{default}]" if default else ""

scripts/wizard/writer.py

@@ -12,6 +12,16 @@ from typing import Any
 
 import yaml
 
+CHANNEL_CONNECTION_PROVIDERS: tuple[str, ...] = (
+    "telegram",
+    "slack",
+    "discord",
+    "feishu",
+    "dingtalk",
+    "wechat",
+    "wecom",
+)
+
 
 def _project_root() -> Path:
     return Path(__file__).resolve().parents[2]
@@ -151,6 +161,18 @@ def _make_model_config_name(model_name: str) -> str:
     return base.replace(".", "-")
 
 
+def _build_channel_connections_config(enabled_providers: list[str]) -> dict[str, Any]:
+    selected = set(enabled_providers)
+    unknown = selected.difference(CHANNEL_CONNECTION_PROVIDERS)
+    if unknown:
+        raise ValueError(f"Unknown channel connection provider(s): {', '.join(sorted(unknown))}")
+
+    return {
+        "enabled": bool(selected),
+        **{provider: {"enabled": provider in selected} for provider in CHANNEL_CONNECTION_PROVIDERS},
+    }
+
+
 def build_minimal_config(
     *,
     provider_use: str,
@@ -170,6 +192,7 @@ def build_minimal_config(
     allow_host_bash: bool = False,
     include_bash_tool: bool = False,
     include_write_tools: bool = True,
+    channel_connection_providers: list[str] | None = None,
     config_version: int = 5,
     base_config: dict[str, Any] | None = None,
 ) -> str:
@@ -219,6 +242,8 @@ def build_minimal_config(
     else:
         sandbox_config.pop("allow_host_bash", None)
     data["sandbox"] = sandbox_config
+    if channel_connection_providers is not None:
+        data["channel_connections"] = _build_channel_connections_config(channel_connection_providers)
 
     header = (
         f"# DeerFlow Configuration\n"
@@ -250,6 +275,7 @@ def write_config_yaml(
     allow_host_bash: bool = False,
     include_bash_tool: bool = False,
     include_write_tools: bool = True,
+    channel_connection_providers: list[str] | None = None,
 ) -> None:
     """Write (or overwrite) config.yaml with a minimal working configuration."""
     # Read config_version from config.example.yaml if present
@@ -284,6 +310,7 @@ def write_config_yaml(
         allow_host_bash=allow_host_bash,
         include_bash_tool=include_bash_tool,
         include_write_tools=include_write_tools,
+        channel_connection_providers=channel_connection_providers,
         config_version=config_version,
         base_config=example_defaults,
     )