DeepSeek-Reasonix/internal/frontend/cli/cli_test.go

package cli

import (
	"bufio"
	"bytes"
	"context"
	"errors"
	"io"
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"reasonix/internal/state/sessionstore"
	"reflect"
	"strings"
	"sync/atomic"
	"testing"

	"reasonix/internal/assembly/boot"
	"reasonix/internal/base/i18n"
	"reasonix/internal/base/testenv"
	"reasonix/internal/contract/config"
	"reasonix/internal/contract/event"
	"reasonix/internal/contract/provider"
	"reasonix/internal/platform/notify"
	"reasonix/internal/platform/telemetry"
	"reasonix/internal/session/control"
)

func TestChdirTo(t *testing.T) {
	orig, err := os.Getwd()
	if err != nil {
		t.Fatal(err)
	}

	if rc := chdirTo(""); rc != 0 {
		t.Fatalf(`chdirTo("") = %d, want 0`, rc)
	}
	if cwd, _ := os.Getwd(); cwd != orig {
		t.Fatalf(`chdirTo("") moved cwd to %q`, cwd)
	}

	tmp := testenv.TempDir(t)
	// Restore CWD before TempDir's RemoveAll runs (LIFO ordering): Windows can't
	// delete a directory that is still the process working directory.
	t.Cleanup(func() { _ = os.Chdir(orig) })
	if rc := chdirTo(tmp); rc != 0 {
		t.Fatalf("chdirTo(tmp) = %d, want 0", rc)
	}
	got, _ := filepath.EvalSymlinks(mustGetwd(t))
	want, _ := filepath.EvalSymlinks(tmp)
	if got != want {
		t.Fatalf("cwd = %q, want %q", got, want)
	}

	if rc := chdirTo(filepath.Join(tmp, "does-not-exist")); rc != 2 {
		t.Fatalf("chdirTo(missing) = %d, want 2", rc)
	}
}

func TestModelForResumePathUsesStoredModelWhenAvailable(t *testing.T) {
	dir := testenv.TempDir(t)
	path := filepath.Join(dir, "session.jsonl")
	session := sessionstore.NewSession("sys")
	session.Add(provider.Message{Role: provider.RoleUser, Content: "hello"})
	if err := session.Save(path); err != nil {
		t.Fatal(err)
	}
	if err := sessionstore.SetBranchModelPreserveUpdated(path, "saved/model"); err != nil {
		t.Fatal(err)
	}
	cfg := &config.Config{
		DefaultModel: "default/model",
		Providers: []config.ProviderEntry{
			{Name: "default", Kind: "openai", BaseURL: "https://default.invalid/v1", Model: "model"},
			{Name: "saved", Kind: "openai", BaseURL: "https://saved.invalid/v1", Model: "model"},
		},
	}

	if got := modelForResumePath("", path, cfg); got != "saved/model" {
		t.Fatalf("modelForResumePath = %q, want saved/model", got)
	}
	if got := modelForResumePath("explicit/model", path, cfg); got != "explicit/model" {
		t.Fatalf("explicit model was overwritten: %q", got)
	}
	if got := modelForResumePath("", filepath.Join(dir, "missing.jsonl"), cfg); got != "" {
		t.Fatalf("missing session model = %q, want empty fallback", got)
	}
	cfg.Providers = cfg.Providers[:1]
	if got := modelForResumePath("", path, cfg); got != "" {
		t.Fatalf("unknown stored model = %q, want empty fallback", got)
	}
}

func TestServeRejectsUnknownAuthMode(t *testing.T) {
	isolateCLIConfigHome(t)

	errOut := captureStderr(t, func() {
		if rc := runServe([]string{"--auth", "tokne", "--addr", "127.0.0.1:0"}, "v1.20.0"); rc != 1 {
			t.Fatalf("serve --auth tokne rc = %d, want 1", rc)
		}
	})
	if !strings.Contains(errOut, "auth mode must be none, token, or password") {
		t.Fatalf("serve --auth tokne stderr = %q, want auth mode validation", errOut)
	}
}

func TestServePasswordAuthRequiresPasswordMaterial(t *testing.T) {
	isolateCLIConfigHome(t)

	errOut := captureStderr(t, func() {
		if rc := runServe([]string{"--auth", "password", "--addr", "127.0.0.1:0"}, "v1.20.0"); rc != 1 {
			t.Fatalf("serve --auth password without password rc = %d, want 1", rc)
		}
	})
	if !strings.Contains(errOut, "auth mode password requires --password or serve.password_hash") {
		t.Fatalf("serve --auth password stderr = %q, want password material validation", errOut)
	}
}

func mustGetwd(t *testing.T) string {
	t.Helper()
	cwd, err := os.Getwd()
	if err != nil {
		t.Fatal(err)
	}
	return cwd
}

func isolateCLIConfigHome(t *testing.T) string {
	t.Helper()
	home := testenv.TempDir(t)
	t.Setenv("HOME", home)
	// Keep tests on the default-path code path while preventing a caller's
	// higher-priority REASONIX_HOME from escaping this temporary home.
	t.Setenv("REASONIX_HOME", "")
	if err := os.Unsetenv("REASONIX_HOME"); err != nil {
		t.Fatalf("unset REASONIX_HOME: %v", err)
	}
	t.Setenv("REASONIX_CREDENTIALS_STORE", "file")
	t.Setenv("USERPROFILE", home)
	t.Setenv("XDG_CONFIG_HOME", filepath.Join(home, ".config"))
	t.Setenv("AppData", filepath.Join(home, "AppData"))
	t.Chdir(testenv.TempDir(t))
	return home
}

func TestIsolateCLIConfigHomeOverridesExistingReasonixHome(t *testing.T) {
	externalHome := testenv.TempDir(t)
	t.Setenv("REASONIX_HOME", externalHome)

	home := isolateCLIConfigHome(t)

	got := config.UserConfigPath()
	rel, err := filepath.Rel(home, got)
	if err != nil || rel == ".." || strings.HasPrefix(rel, ".."+string(filepath.Separator)) {
		t.Fatalf("UserConfigPath() = %q, outside isolated home %q", got, home)
	}
}

func TestMCPMigrationWaitsForCLIWorkspace(t *testing.T) {
	isolateCLIConfigHome(t)
	cwd := mustGetwd(t)
	if err := os.WriteFile(filepath.Join(cwd, "reasonix.toml"), []byte(`
[[plugins]]
name = "cwd-project"
command = "cwd-project-bin"
`), 0o644); err != nil {
		t.Fatal(err)
	}

	migrateLegacyConfigForCLI()
	if cfg := config.LoadForEdit(config.UserConfigPath()); hasPluginNamed(cfg, "cwd-project") {
		t.Fatalf("early CLI legacy migration imported the cwd project plugin: %+v", cfg.Plugins)
	}

	migrateMCPConfigForCLIWorkspace()
	if cfg := config.LoadForEdit(config.UserConfigPath()); !hasPluginNamed(cfg, "cwd-project") {
		t.Fatalf("workspace-aware CLI migration did not import project plugin: %+v", cfg.Plugins)
	}
}

func hasPluginNamed(cfg *config.Config, name string) bool {
	if cfg == nil {
		return false
	}
	for _, plugin := range cfg.Plugins {
		if plugin.Name == name {
			return true
		}
	}
	return false
}

func TestMetadataCommandsDoNotProbeTerminalTheme(t *testing.T) {
	defer func(prev func() (terminalRGB, bool)) { terminalProbe = prev }(terminalProbe)
	terminalProbe = func() (terminalRGB, bool) {
		t.Fatal("metadata command should not query terminal background")
		return terminalRGB{}, false
	}

	out := captureStdout(t, func() {
		if rc := Run([]string{"version"}, "test-version"); rc != 0 {
			t.Fatalf("version rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "reasonix test-version") {
		t.Fatalf("version output = %q", out)
	}

	out = captureStdout(t, func() {
		if rc := Run([]string{"help"}, "test-version"); rc != 0 {
			t.Fatalf("help rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "Usage:") && !strings.Contains(out, "用法：") {
		t.Fatalf("help output missing usage:\n%s", out)
	}
	if !strings.Contains(out, "reasonix run [--model NAME] [--max-steps N] [-c|--continue] [--resume PATH] [--copy] [--output-format FORMAT] <task>") {
		t.Fatalf("help output missing run resume flags:\n%s", out)
	}
}

func TestRunDispatchesACPLongFlagAlias(t *testing.T) {
	out, errOut := captureCLIOutput(t, func() {
		if rc := Run([]string{"--acp", "-h"}, "test-version"); rc != 0 {
			t.Fatalf("Run --acp -h rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "Usage of acp:") {
		t.Fatalf("--acp should dispatch to the ACP command, got stdout:\n%s", out)
	}
	if errOut != "" {
		t.Fatalf("--acp help wrote stderr: %q", errOut)
	}
	if strings.Contains(out, "unknown command") {
		t.Fatalf("--acp should not be treated as an unknown command:\n%s", out)
	}
}

func TestRunReportsFlagParseErrors(t *testing.T) {
	isolateCLIConfigHome(t)

	tests := []struct {
		name string
		args []string
		want string
	}{
		{name: "run unknown flag", args: []string{"run", "--unknown"}, want: "unknown flag: --unknown"},
		{name: "run invalid value", args: []string{"run", "--max-steps=invalid"}, want: "invalid argument \"invalid\" for \"--max-steps\" flag"},
		{name: "run missing value", args: []string{"run", "--model"}, want: "flag needs an argument: --model"},
		{name: "serve unknown flag", args: []string{"serve", "--unknown"}, want: "flag provided but not defined: -unknown"},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			stderr := captureStderr(t, func() {
				if rc := Run(tt.args, "test-version"); rc != 2 {
					t.Fatalf("Run(%q) rc = %d, want 2", tt.args, rc)
				}
			})
			if !strings.Contains(stderr, tt.want) {
				t.Fatalf("Run(%q) stderr = %q, want %q", tt.args, stderr, tt.want)
			}
			if strings.Contains(stderr, "Usage of") {
				t.Fatalf("Run(%q) should print a concise error, got:\n%s", tt.args, stderr)
			}
		})
	}
}

func TestSubcommandHelpReturnsSuccess(t *testing.T) {
	isolateCLIConfigHome(t)

	tests := []struct {
		name string
		args []string
		want string
	}{
		{name: "run", args: []string{"run", "--help"}, want: "Usage of run:"},
		{name: "serve", args: []string{"serve", "--help"}, want: "Usage of serve:"},
		{name: "upgrade", args: []string{"upgrade", "--help"}, want: "Usage of upgrade:"},
		{name: "remote connect", args: []string{"remote", "connect", "--help"}, want: "Usage of remote connect:"},
		{name: "remote add before name", args: []string{"remote", "add", "--help"}, want: remoteAddUsage},
		{name: "remote add before target", args: []string{"remote", "add", "box", "--help"}, want: remoteAddUsage},
		{name: "remote serve before action", args: []string{"remote", "serve", "--help"}, want: remoteServeUsage},
		{name: "remote serve before name", args: []string{"remote", "serve", "start", "--help"}, want: remoteServeUsage},
		{name: "subagent create", args: []string{"subagent", "create", "--help"}, want: subagentUsageText},
		{name: "subagent edit", args: []string{"subagent", "edit", "--help"}, want: subagentUsageText},
		{name: "subagent delete", args: []string{"subagent", "delete", "--help"}, want: subagentUsageText},
		{name: "subagent try", args: []string{"subagent", "try", "--help"}, want: subagentUsageText},
		{name: "subagent run", args: []string{"subagent", "run", "--help"}, want: subagentUsageText},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			stdout, stderr := captureCLIOutput(t, func() {
				if rc := Run(tt.args, "test-version"); rc != 0 {
					t.Fatalf("Run(%q) rc = %d, want 0", tt.args, rc)
				}
			})
			if !strings.Contains(stdout, tt.want) {
				t.Fatalf("Run(%q) help missing %q:\n%s", tt.args, tt.want, stdout)
			}
			if stderr != "" {
				t.Fatalf("Run(%q) help wrote stderr: %q", tt.args, stderr)
			}
			if strings.Contains(stdout, "help requested") {
				t.Fatalf("Run(%q) reported help as an error:\n%s", tt.args, stdout)
			}
		})
	}
}

func TestRunPrintAliasDispatchesRunFlags(t *testing.T) {
	isolateCLIConfigHome(t)
	out, errOut := captureCLIOutput(t, func() {
		if rc := Run([]string{"-p", "-h"}, "test-version"); rc != 0 {
			t.Fatalf("Run(-p -h) rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "Usage of run:") {
		t.Fatalf("-p should dispatch to one-shot run flags, got:\n%s", out)
	}
	if errOut != "" {
		t.Fatalf("-p help wrote stderr: %q", errOut)
	}
}

func TestParsePermissionModeClaudeAliases(t *testing.T) {
	tests := map[string]cliPermissionMode{
		"ask":               {approval: control.ToolApprovalAsk},
		"manual":            {approval: control.ToolApprovalAsk},
		"acceptEdits":       {approval: control.ToolApprovalAsk, allow: []string{"write_file", "edit_file", "multi_edit", "move_file", "notebook_edit", "delete_range", "delete_symbol"}},
		"dontAsk":           {approval: control.ToolApprovalDontAsk},
		"plan":              {approval: control.ToolApprovalAsk, plan: true},
		"bypassPermissions": {approval: control.ToolApprovalYolo},
	}
	for input, want := range tests {
		got, err := parsePermissionMode(input)
		if err != nil || !reflect.DeepEqual(got, want) {
			t.Errorf("parsePermissionMode(%q) = (%+v, %v), want %+v", input, got, err, want)
		}
	}
}

func TestResolveRunPermissionModeRequiresExplicitAuto(t *testing.T) {
	if got, err := resolveRunPermissionMode("ask", false, false); err != nil || got != "ask" {
		t.Fatalf("default run permission mode = (%q, %v), want ask", got, err)
	}
	if got, err := resolveRunPermissionMode("ask", true, false); err != nil || got != "auto" {
		t.Fatalf("-y run permission mode = (%q, %v), want auto", got, err)
	}
	if got, err := resolveRunPermissionMode("dontAsk", true, true); err == nil || got != "" {
		t.Fatalf("combined permission flags = (%q, %v), want conflict", got, err)
	}
}

func TestRunMigratesLegacyConfigBeforeConfigOnlyCommands(t *testing.T) {
	isolateCLIConfigHome(t)
	legacyPath := filepath.Join(filepath.Dir(config.UserConfigPath()), "reasonix.toml")
	if err := os.MkdirAll(filepath.Dir(legacyPath), 0o755); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(legacyPath, []byte(`
default_model = "deepseek-flash"

[[plugins]]
name = "legacy-cli"
command = "legacy-bin"
`), 0o644); err != nil {
		t.Fatal(err)
	}

	out := captureStdout(t, func() {
		if rc := Run([]string{"mcp", "list"}, "test-version"); rc != 0 {
			t.Fatalf("mcp list rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "legacy-cli") {
		t.Fatalf("mcp list should include migrated legacy config:\n%s", out)
	}

	body, err := os.ReadFile(config.UserConfigPath())
	if err != nil {
		t.Fatalf("read migrated user config: %v", err)
	}
	for _, want := range []string{`config_version = 6`, `[desktop]`, `name    = "legacy-cli"`} {
		if !strings.Contains(string(body), want) {
			t.Fatalf("migrated config missing %q:\n%s", want, body)
		}
	}
}

func TestRunAppliesUserConfigUpgradesOnStartup(t *testing.T) {
	isolateCLIConfigHome(t)
	path := config.UserConfigPath()
	if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(path, []byte("config_version = 2\ndefault_model = \"deepseek-flash\"\n"), 0o644); err != nil {
		t.Fatal(err)
	}

	captureStdout(t, func() {
		if rc := Run([]string{"mcp", "list"}, "test-version"); rc != 0 {
			t.Fatalf("mcp list rc = %d, want 0", rc)
		}
	})

	body, err := os.ReadFile(path)
	if err != nil {
		t.Fatalf("read upgraded user config: %v", err)
	}
	if !strings.Contains(string(body), "config_version = 6") {
		t.Fatalf("CLI startup should apply user config upgrades:\n%s", body)
	}
}

func TestRunMetadataCommandsDoNotMigrateLegacyConfig(t *testing.T) {
	isolateCLIConfigHome(t)
	legacyPath := filepath.Join(filepath.Dir(config.UserConfigPath()), "reasonix.toml")
	if err := os.MkdirAll(filepath.Dir(legacyPath), 0o755); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(legacyPath, []byte(`default_model = "deepseek-flash"`), 0o644); err != nil {
		t.Fatal(err)
	}

	out := captureStdout(t, func() {
		if rc := Run([]string{"version"}, "test-version"); rc != 0 {
			t.Fatalf("version rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "reasonix test-version") {
		t.Fatalf("version output = %q", out)
	}
	if _, err := os.Stat(config.UserConfigPath()); !os.IsNotExist(err) {
		t.Fatalf("version should not migrate legacy config, stat err=%v", err)
	}
}

func TestConfigLoadIgnoresRetiredAutoPlan(t *testing.T) {
	isolateCLIConfigHome(t)
	if err := os.WriteFile("reasonix.toml", []byte("[agent]\nauto_plan = \"on\"\nauto_plan_classifier = \"deepseek-flash\"\n"), 0o644); err != nil {
		t.Fatalf("write project config: %v", err)
	}

	cfg, err := config.Load()
	if err != nil {
		t.Fatalf("load config: %v", err)
	}
	if cfg.Agent.AutoPlan != "off" || cfg.Agent.AutoPlanClassifier != "" {
		t.Fatalf("retired auto-plan config = (%q, %q), want off/empty", cfg.Agent.AutoPlan, cfg.Agent.AutoPlanClassifier)
	}
}

func TestConfigAutoPlanCompatibilityCommandKeepsOffAsNoOp(t *testing.T) {
	isolateCLIConfigHome(t)
	path := config.UserConfigPath()
	cfg := config.Default()
	cfg.Agent.Temperature = 0.4
	if err := cfg.SaveTo(path); err != nil {
		t.Fatalf("write user config: %v", err)
	}
	before, err := os.ReadFile(path)
	if err != nil {
		t.Fatalf("read user config before command: %v", err)
	}

	out := captureStdout(t, func() {
		if rc := Run([]string{"config", "auto-plan", "off"}, "test-version"); rc != 0 {
			t.Fatalf("config auto-plan off rc = %d, want 0", rc)
		}
	})
	if out != "auto_plan = \"off\"\n" {
		t.Fatalf("config auto-plan off output = %q", out)
	}
	after, err := os.ReadFile(path)
	if err != nil {
		t.Fatalf("read user config after command: %v", err)
	}
	if !bytes.Equal(after, before) {
		t.Fatalf("config auto-plan off must not rewrite user config\nbefore:\n%s\nafter:\n%s", before, after)
	}

	out = captureStdout(t, func() {
		if rc := Run([]string{"config", "auto-plan"}, "test-version"); rc != 0 {
			t.Fatalf("config auto-plan query rc = %d, want 0", rc)
		}
	})
	if out != "auto_plan = \"off\"\n" {
		t.Fatalf("config auto-plan query output = %q", out)
	}
}

func TestConfigAutoPlanCompatibilityCommandRejectsEnable(t *testing.T) {
	isolateCLIConfigHome(t)

	errOut := captureStderr(t, func() {
		if rc := Run([]string{"config", "auto-plan", "on"}, "test-version"); rc != 2 {
			t.Fatalf("config auto-plan on rc = %d, want 2", rc)
		}
	})
	if !strings.Contains(errOut, "automatic plan mode has been retired") {
		t.Fatalf("config auto-plan on stderr = %q", errOut)
	}
}

func TestConfigReasoningLanguageCommandWritesUserConfig(t *testing.T) {
	isolateCLIConfigHome(t)

	out := captureStdout(t, func() {
		if rc := Run([]string{"config", "reasoning-language", "zh"}, "test-version"); rc != 0 {
			t.Fatalf("config reasoning-language rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, `reasoning_language = "zh"`) {
		t.Fatalf("config reasoning-language output = %q", out)
	}
	cfg := config.LoadForEdit(config.UserConfigPath())
	if cfg.Agent.ReasoningLanguage != "zh" || cfg.ReasoningLanguage() != "zh" {
		t.Fatalf("saved reasoning_language = %q/%q, want zh", cfg.Agent.ReasoningLanguage, cfg.ReasoningLanguage())
	}
}

func TestConfigReasoningLanguageLocalCreatesMinimalProjectOverride(t *testing.T) {
	isolateCLIConfigHome(t)

	userCfg := config.Default()
	userCfg.DefaultModel = "mimo-pro"
	if err := userCfg.SaveTo(config.UserConfigPath()); err != nil {
		t.Fatalf("write user config: %v", err)
	}

	out := captureStdout(t, func() {
		if rc := Run([]string{"config", "reasoning-language", "--local", "en"}, "test-version"); rc != 0 {
			t.Fatalf("config reasoning-language --local rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, `reasoning_language = "en"`) {
		t.Fatalf("config reasoning-language --local output = %q", out)
	}

	body, err := os.ReadFile("reasonix.toml")
	if err != nil {
		t.Fatalf("read project config: %v", err)
	}
	if strings.Contains(string(body), "default_model") {
		t.Fatalf("project reasoning-language override should not pin default_model:\n%s", body)
	}
	if !strings.Contains(string(body), "[agent]") || !strings.Contains(string(body), `reasoning_language = "en"`) {
		t.Fatalf("project config missing reasoning_language override:\n%s", body)
	}

	cfg, err := config.Load()
	if err != nil {
		t.Fatalf("load merged config: %v", err)
	}
	if cfg.DefaultModel != "mimo-pro" {
		t.Fatalf("default_model = %q, want global mimo-pro", cfg.DefaultModel)
	}
	if cfg.ReasoningLanguage() != "en" {
		t.Fatalf("reasoning_language = %q, want local en", cfg.ReasoningLanguage())
	}
}

func TestConfigReasoningLanguageRejectsAliases(t *testing.T) {
	isolateCLIConfigHome(t)

	errOut := captureStderr(t, func() {
		if rc := Run([]string{"config", "reasoning-language", "中文"}, "test-version"); rc != 2 {
			t.Fatalf("config reasoning-language alias rc = %d, want 2", rc)
		}
	})
	if !strings.Contains(errOut, "must be auto|zh|en") {
		t.Fatalf("config reasoning-language alias stderr = %q", errOut)
	}
}

func TestConfigCompactRatioCommandWritesUserConfigAndReportsSource(t *testing.T) {
	isolateCLIConfigHome(t)
	userCfg := config.Default()
	userCfg.Agent.Temperature = 0.42
	if err := userCfg.SaveTo(config.UserConfigPath()); err != nil {
		t.Fatalf("write user config: %v", err)
	}

	out := captureStdout(t, func() {
		if rc := Run([]string{"config", "compact-ratio", "75.5"}, "test-version"); rc != 0 {
			t.Fatalf("config compact-ratio rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "compact_ratio = 75.5%") || !strings.Contains(out, "user:") {
		t.Fatalf("config compact-ratio output = %q", out)
	}
	cfg := config.LoadForEdit(config.UserConfigPath())
	if got := cfg.Agent.CompactRatio; got != 0.755 {
		t.Fatalf("saved compact ratio = %v, want 0.755", got)
	}
	if got := cfg.Agent.Temperature; got != 0.42 {
		t.Fatalf("compact-ratio update changed temperature to %v, want 0.42", got)
	}

	out = captureStdout(t, func() {
		if rc := Run([]string{"config", "compact-ratio"}, "test-version"); rc != 0 {
			t.Fatalf("config compact-ratio query rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "compact_ratio = 75.5%") || !strings.Contains(out, "user:") {
		t.Fatalf("config compact-ratio query output = %q", out)
	}
}

func TestConfigCompactRatioQueryReportsBuiltInDefault(t *testing.T) {
	isolateCLIConfigHome(t)

	out := captureStdout(t, func() {
		if rc := Run([]string{"config", "compact-ratio"}, "test-version"); rc != 0 {
			t.Fatalf("config compact-ratio query rc = %d, want 0", rc)
		}
	})
	if out != "compact_ratio = 85% (built-in default)\n" {
		t.Fatalf("config compact-ratio query output = %q", out)
	}
}

func TestConfigCompactRatioLocalCreatesMinimalProjectOverride(t *testing.T) {
	isolateCLIConfigHome(t)

	userCfg := config.Default()
	userCfg.DefaultModel = "mimo-pro"
	if err := userCfg.SaveTo(config.UserConfigPath()); err != nil {
		t.Fatalf("write user config: %v", err)
	}

	out := captureStdout(t, func() {
		if rc := Run([]string{"config", "compact-ratio", "--local", "70"}, "test-version"); rc != 0 {
			t.Fatalf("config compact-ratio --local rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "compact_ratio = 70%") || !strings.Contains(out, "project:") {
		t.Fatalf("config compact-ratio --local output = %q", out)
	}

	body, err := os.ReadFile("reasonix.toml")
	if err != nil {
		t.Fatalf("read project config: %v", err)
	}
	if strings.Contains(string(body), "default_model") {
		t.Fatalf("project compact-ratio override should not pin default_model:\n%s", body)
	}
	if !strings.Contains(string(body), "[agent]") || !strings.Contains(string(body), "compact_ratio = 0.7") {
		t.Fatalf("project config missing compact_ratio override:\n%s", body)
	}

	cfg, err := config.Load()
	if err != nil {
		t.Fatalf("load merged config: %v", err)
	}
	if cfg.DefaultModel != "mimo-pro" {
		t.Fatalf("default_model = %q, want global mimo-pro", cfg.DefaultModel)
	}
	if cfg.Agent.CompactRatio != 0.7 {
		t.Fatalf("compact ratio = %v, want local 0.7", cfg.Agent.CompactRatio)
	}

	out = captureStdout(t, func() {
		if rc := Run([]string{"config", "compact-ratio"}, "test-version"); rc != 0 {
			t.Fatalf("config compact-ratio query rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, "compact_ratio = 70%") || !strings.Contains(out, "project:") {
		t.Fatalf("project compact-ratio query output = %q", out)
	}
}

func TestConfigCompactRatioRejectsValuesOutsideEditableRange(t *testing.T) {
	isolateCLIConfigHome(t)

	for _, value := range []string{"64", "86", "NaN", "+Inf", "not-a-number"} {
		t.Run(value, func(t *testing.T) {
			errOut := captureStderr(t, func() {
				if rc := Run([]string{"config", "compact-ratio", value}, "test-version"); rc != 2 {
					t.Fatalf("config compact-ratio %s rc = %d, want 2", value, rc)
				}
			})
			if !strings.Contains(errOut, "percentage between 65 and 85") {
				t.Fatalf("config compact-ratio %s stderr = %q", value, errOut)
			}
		})
	}
	if _, err := os.Stat(config.UserConfigPath()); !os.IsNotExist(err) {
		t.Fatalf("invalid compact ratio wrote user config, stat err=%v", err)
	}
}

func TestConfigCurrencyCommandWritesUserConfig(t *testing.T) {
	isolateCLIConfigHome(t)

	out := captureStdout(t, func() {
		if rc := Run([]string{"config", "currency", "CNY"}, "test-version"); rc != 0 {
			t.Fatalf("config currency rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, `currency = "CNY"`) || !strings.Contains(out, "display: CNY") {
		t.Fatalf("config currency output = %q", out)
	}
	cfg := config.LoadForEdit(config.UserConfigPath())
	if got := cfg.DesktopCurrency(); got != "CNY" {
		t.Fatalf("saved currency = %q, want CNY", got)
	}
	if got := cfg.DisplayCurrencyPref(); got != "CNY" {
		t.Fatalf("display pref = %q, want CNY", got)
	}
}

func TestConfigCurrencyAutoRemainsUnresolved(t *testing.T) {
	isolateCLIConfigHome(t)
	i18n.DetectLanguage("zh-TW")
	t.Cleanup(func() { i18n.DetectLanguage("en") })

	out := captureStdout(t, func() {
		if rc := configCurrencyCommand([]string{"auto"}); rc != 0 {
			t.Fatalf("config currency auto rc = %d, want 0", rc)
		}
	})
	if !strings.Contains(out, `currency = "auto"`) || !strings.Contains(out, "display: ,") {
		t.Fatalf("config currency auto output = %q", out)
	}
	cfg := config.LoadForEdit(config.UserConfigPath())
	if got := cfg.DesktopCurrency(); got != "" {
		t.Fatalf("auto should clear saved currency, got %q", got)
	}
}

func TestConfigCurrencyRejectsProjectScope(t *testing.T) {
	isolateCLIConfigHome(t)
	errOut := captureStderr(t, func() {
		if rc := Run([]string{"config", "currency", "--local", "USD"}, "test-version"); rc != 2 {
			t.Fatalf("config currency --local rc = %d, want 2", rc)
		}
	})
	if !strings.Contains(errOut, "user-level only") {
		t.Fatalf("config currency --local stderr = %q", errOut)
	}
	if _, err := os.Stat("reasonix.toml"); !os.IsNotExist(err) {
		t.Fatalf("config currency --local wrote project config, stat err=%v", err)
	}
}

func TestProvidersWithMissingKeysOnlyChecksActiveDefaultModel(t *testing.T) {
	cfg := config.Default()
	t.Setenv("DEEPSEEK_API_KEY", "")
	t.Setenv("MIMO_API_KEY", "")

	missing := providersWithMissingKeys(cfg)
	if len(missing) != 1 {
		t.Fatalf("missing providers = %+v, want only active default model provider", missing)
	}
	if missing[0].APIKeyEnv != "DEEPSEEK_API_KEY" {
		t.Fatalf("missing key env = %q, want DEEPSEEK_API_KEY", missing[0].APIKeyEnv)
	}
}

func TestProvidersWithMissingKeysIgnoresUnusedBuiltInPresets(t *testing.T) {
	cfg := config.Default()
	t.Setenv("DEEPSEEK_API_KEY", "test-key")
	t.Setenv("MIMO_API_KEY", "")

	if missing := providersWithMissingKeys(cfg); len(missing) != 0 {
		t.Fatalf("missing providers = %+v, want none when only the configured default is keyed", missing)
	}
}

func TestProvidersWithMissingKeysIncludesReferencedSecondaryModels(t *testing.T) {
	cfg := config.Default()
	cfg.Providers = append(cfg.Providers,
		config.ProviderEntry{Name: "mimo-pro", Kind: "openai", BaseURL: "https://token-plan-cn.xiaomimimo.com/v1", Model: "mimo-v2.5-pro", APIKeyEnv: "MIMO_API_KEY"},
		config.ProviderEntry{Name: "mimo-flash", Kind: "openai", BaseURL: "https://token-plan-cn.xiaomimimo.com/v1", Model: "mimo-v2.5", APIKeyEnv: "MIMO_API_KEY"},
	)
	cfg.Agent.PlannerModel = "mimo-pro"
	cfg.Agent.SubagentModel = "mimo-flash"
	cfg.Agent.SubagentModels = map[string]string{
		"review": "mimo-pro/mimo-v2.5-pro",
	}
	t.Setenv("DEEPSEEK_API_KEY", "test-key")
	t.Setenv("MIMO_API_KEY", "")

	missing := providersWithMissingKeys(cfg)
	if len(missing) != 1 {
		t.Fatalf("missing providers = %+v, want MiMo once", missing)
	}
	if missing[0].APIKeyEnv != "MIMO_API_KEY" {
		t.Fatalf("missing key env = %q, want MIMO_API_KEY", missing[0].APIKeyEnv)
	}
}

type cliRecordSink struct {
	events []event.Kind
}

func (s *cliRecordSink) Emit(e event.Event) {
	s.events = append(s.events, e.Kind)
}

type cliRecordSender struct {
	messages []notify.Message
}

func (s *cliRecordSender) Send(m notify.Message) error {
	s.messages = append(s.messages, m)
	return nil
}

func TestWithNotificationsWrapsCLISinkWithConfiguredSender(t *testing.T) {
	inner := &cliRecordSink{}
	sender := &cliRecordSender{}
	calls := 0
	prev := newNotificationSender
	newNotificationSender = func() notify.Sender {
		calls++
		return sender
	}
	t.Cleanup(func() { newNotificationSender = prev })

	cfg := config.Default()
	cfg.Notifications.Enabled = true

	wrapped := withNotifications(inner, cfg)
	wrapped.Emit(event.Event{Kind: event.TurnDone})

	if calls != 1 {
		t.Fatalf("newNotificationSender calls = %d, want 1", calls)
	}
	if len(inner.events) != 1 || inner.events[0] != event.TurnDone {
		t.Fatalf("forwarded events = %v, want [TurnDone]", inner.events)
	}
	if len(sender.messages) != 1 {
		t.Fatalf("notifications = %d, want 1", len(sender.messages))
	}
	if sender.messages[0].Body != i18n.M.NotifyTurnDone {
		t.Fatalf("notification body = %q, want %q", sender.messages[0].Body, i18n.M.NotifyTurnDone)
	}
}

func TestConfigTelemetryCommandRoundTripAndOptOutCleanup(t *testing.T) {
	isolateCLIConfigHome(t)
	out := captureStdout(t, func() {
		if rc := configTelemetryCommand(nil); rc != 0 {
			t.Fatalf("config telemetry query rc = %d", rc)
		}
	})
	if !strings.Contains(out, `cli_metrics = "auto"`) {
		t.Fatalf("default telemetry query = %q", out)
	}
	if rc := configTelemetryCommand([]string{"on"}); rc != 0 {
		t.Fatalf("config telemetry on rc = %d", rc)
	}
	cfg, err := config.Load()
	if err != nil || cfg.CLITelemetryMode() != "on" {
		t.Fatalf("saved telemetry mode = %q, err = %v", cfg.CLITelemetryMode(), err)
	}
	pending := filepath.Join(config.ReasonixHomeDir(), "cli-telemetry-pending")
	if err := os.MkdirAll(pending, 0o700); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(pending, "pending.json"), []byte("{}"), 0o600); err != nil {
		t.Fatal(err)
	}
	if rc := configTelemetryCommand([]string{"off"}); rc != 0 {
		t.Fatalf("config telemetry off rc = %d", rc)
	}
	if _, err := os.Stat(pending); !errors.Is(err, os.ErrNotExist) {
		t.Fatalf("opt-out did not remove pending queue: %v", err)
	}
}

func TestConfigTelemetryCommandReportsOptOutCleanupFailure(t *testing.T) {
	isolateCLIConfigHome(t)
	previous := cleanupCLITelemetry
	t.Cleanup(func() { cleanupCLITelemetry = previous })
	cleanupCLITelemetry = func(string) error { return errors.New("cleanup denied") }

	errOut := captureStderr(t, func() {
		if rc := configTelemetryCommand([]string{"off"}); rc != 1 {
			t.Fatalf("config telemetry off rc = %d, want 1", rc)
		}
	})
	if !strings.Contains(errOut, "telemetry disabled") || !strings.Contains(errOut, "cleanup denied") {
		t.Fatalf("cleanup failure stderr = %q", errOut)
	}
	cfg, err := config.Load()
	if err != nil || cfg.CLITelemetryMode() != "off" {
		t.Fatalf("saved telemetry mode = %q, err = %v", cfg.CLITelemetryMode(), err)
	}
}

func TestCLITelemetryConsentDefaultsYesAndPromptsOnlyOnce(t *testing.T) {
	isolateCLIConfigHome(t)
	clearCLITelemetryPolicyEnv(t)
	t.Cleanup(func() { i18n.DetectLanguage("en") })
	i18n.DetectLanguage("en")

	previousStart := startCLITelemetryReporter
	t.Cleanup(func() { startCLITelemetryReporter = previousStart })
	want := &telemetry.Reporter{}
	starts := 0
	startCLITelemetryReporter = func(opts telemetry.Options) *telemetry.Reporter {
		starts++
		saved, err := config.LoadForEditReadOnlyStrict(config.UserConfigPath())
		if err != nil || !saved.CLITelemetryConfigured() || saved.CLITelemetryMode() != "auto" {
			t.Fatalf("telemetry started before consent was saved: mode=%q configured=%v err=%v", saved.CLITelemetryMode(), saved.CLITelemetryConfigured(), err)
		}
		return want
	}

	cfg := config.Default()
	var out, errOut bytes.Buffer
	got := startCLITelemetryWithIO(cfg, telemetry.Options{
		Version: "v1.20.0", Interactive: true, CLIMode: "tui",
	}, strings.NewReader("\n"), &out, &errOut)
	if got != want || starts != 1 {
		t.Fatalf("first start = %p, calls=%d; want %p, 1", got, starts, want)
	}
	if !strings.Contains(out.String(), "crash.reasonix.io") || !strings.Contains(out.String(), "[Y/n]:") || !strings.Contains(out.String(), "reasonix config telemetry off") {
		t.Fatalf("consent prompt is incomplete: %q", out.String())
	}
	if errOut.Len() != 0 {
		t.Fatalf("unexpected consent stderr: %q", errOut.String())
	}
	if !cfg.CLITelemetryConfigured() || cfg.CLITelemetryMode() != "auto" {
		t.Fatalf("runtime config was not synchronized: mode=%q configured=%v", cfg.CLITelemetryMode(), cfg.CLITelemetryConfigured())
	}

	var secondOut bytes.Buffer
	if got := startCLITelemetryWithIO(cfg, telemetry.Options{
		Version: "v1.20.0", Interactive: true, CLIMode: "tui",
	}, strings.NewReader("n\n"), &secondOut, &errOut); got != want {
		t.Fatalf("second start = %p, want %p", got, want)
	}
	if secondOut.Len() != 0 || starts != 2 {
		t.Fatalf("saved decision prompted again: output=%q calls=%d", secondOut.String(), starts)
	}
}

func TestCLITelemetryConsentNoDisablesAndCleansPending(t *testing.T) {
	isolateCLIConfigHome(t)
	clearCLITelemetryPolicyEnv(t)

	previousStart := startCLITelemetryReporter
	t.Cleanup(func() { startCLITelemetryReporter = previousStart })
	starts := 0
	startCLITelemetryReporter = func(telemetry.Options) *telemetry.Reporter {
		starts++
		return &telemetry.Reporter{}
	}
	home := config.ReasonixHomeDir()
	pending := filepath.Join(home, "cli-telemetry-pending")
	if err := os.MkdirAll(pending, 0o700); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(pending, "pending.json"), []byte("{}"), 0o600); err != nil {
		t.Fatal(err)
	}

	cfg := config.Default()
	var out, errOut bytes.Buffer
	if got := startCLITelemetryWithIO(cfg, telemetry.Options{
		Version: "v1.20.0", Interactive: true, CLIMode: "tui",
	}, strings.NewReader("n\n"), &out, &errOut); got != nil {
		t.Fatalf("declined telemetry returned reporter %p", got)
	}
	if starts != 0 {
		t.Fatalf("declined telemetry started upload %d times", starts)
	}
	if cfg.CLITelemetryMode() != "off" || !cfg.CLITelemetryConfigured() {
		t.Fatalf("decline was not saved in runtime config: mode=%q configured=%v", cfg.CLITelemetryMode(), cfg.CLITelemetryConfigured())
	}
	if _, err := os.Stat(pending); !errors.Is(err, os.ErrNotExist) {
		t.Fatalf("decline did not clear pending queue: %v", err)
	}
	saved, err := config.LoadForEditReadOnlyStrict(config.UserConfigPath())
	if err != nil || saved.CLITelemetryMode() != "off" || !saved.CLITelemetryConfigured() {
		t.Fatalf("saved decline = mode %q configured=%v err=%v", saved.CLITelemetryMode(), saved.CLITelemetryConfigured(), err)
	}
}

func TestCLITelemetryConsentSaveFailureDoesNotUpload(t *testing.T) {
	isolateCLIConfigHome(t)
	clearCLITelemetryPolicyEnv(t)

	previousSave := persistCLITelemetryConsent
	previousStart := startCLITelemetryReporter
	t.Cleanup(func() {
		persistCLITelemetryConsent = previousSave
		startCLITelemetryReporter = previousStart
	})
	persistCLITelemetryConsent = func(string) error { return errors.New("read-only config") }
	starts := 0
	startCLITelemetryReporter = func(telemetry.Options) *telemetry.Reporter {
		starts++
		return &telemetry.Reporter{}
	}

	cfg := config.Default()
	var out, errOut bytes.Buffer
	if got := startCLITelemetryWithIO(cfg, telemetry.Options{
		Version: "v1.20.0", Interactive: true, CLIMode: "tui",
	}, strings.NewReader("\n"), &out, &errOut); got != nil {
		t.Fatalf("save failure returned reporter %p", got)
	}
	if starts != 0 || cfg.CLITelemetryConfigured() {
		t.Fatalf("save failure started=%d configured=%v", starts, cfg.CLITelemetryConfigured())
	}
	if !strings.Contains(errOut.String(), "read-only config") {
		t.Fatalf("save failure was not explained: %q", errOut.String())
	}
}

func TestConfiguredCLITelemetryDoesNotPromptAgain(t *testing.T) {
	isolateCLIConfigHome(t)
	clearCLITelemetryPolicyEnv(t)
	previousSave := persistCLITelemetryConsent
	previousStart := startCLITelemetryReporter
	t.Cleanup(func() {
		persistCLITelemetryConsent = previousSave
		startCLITelemetryReporter = previousStart
	})
	persistCalls := 0
	persistCLITelemetryConsent = func(string) error {
		persistCalls++
		return nil
	}
	want := &telemetry.Reporter{}
	startCalls := 0
	startCLITelemetryReporter = func(opts telemetry.Options) *telemetry.Reporter {
		startCalls++
		if telemetry.Enabled(opts.Mode, opts.Version, opts.Interactive) {
			return want
		}
		return nil
	}

	for _, mode := range []string{"auto", "on", "off"} {
		cfg := config.Default()
		if err := cfg.SetCLITelemetryMode(mode); err != nil {
			t.Fatal(err)
		}
		var out bytes.Buffer
		got := startCLITelemetryWithIO(cfg, telemetry.Options{
			Version: "v1.20.0", Interactive: true, CLIMode: "tui",
		}, strings.NewReader("n\n"), &out, io.Discard)
		if out.Len() != 0 {
			t.Fatalf("configured mode %q prompted again: %q", mode, out.String())
		}
		if mode == "off" && got != nil {
			t.Fatalf("configured off returned reporter %p", got)
		}
		if mode != "off" && got != want {
			t.Fatalf("configured %s returned %p, want %p", mode, got, want)
		}
	}
	if persistCalls != 0 || startCalls != 3 {
		t.Fatalf("configured modes persisted=%d started=%d, want 0 and 3", persistCalls, startCalls)
	}
}

func TestUndecidedCLITelemetryDoesNotPromptOrUploadWhenIneligible(t *testing.T) {
	for _, tc := range []struct {
		name        string
		version     string
		interactive bool
		envKey      string
		envValue    string
	}{
		{name: "noninteractive", version: "v1.20.0"},
		{name: "development", version: "dev", interactive: true},
		{name: "CI", version: "v1.20.0", interactive: true, envKey: "CI", envValue: "1"},
		{name: "do not track", version: "v1.20.0", interactive: true, envKey: "DO_NOT_TRACK", envValue: "1"},
		{name: "environment opt out", version: "v1.20.0", interactive: true, envKey: "REASONIX_TELEMETRY", envValue: "0"},
	} {
		t.Run(tc.name, func(t *testing.T) {
			isolateCLIConfigHome(t)
			clearCLITelemetryPolicyEnv(t)
			if tc.envKey != "" {
				t.Setenv(tc.envKey, tc.envValue)
			}
			cfg, err := config.LoadForRootReadOnly(".")
			if err != nil {
				t.Fatal(err)
			}
			var out, errOut bytes.Buffer
			if got := startCLITelemetryWithIO(cfg, telemetry.Options{
				Version: tc.version, Interactive: tc.interactive, CLIMode: "tui",
			}, strings.NewReader("\n"), &out, &errOut); got != nil {
				t.Fatalf("ineligible telemetry returned reporter %p", got)
			}
			if out.Len() != 0 || errOut.Len() != 0 {
				t.Fatalf("ineligible telemetry wrote output: stdout=%q stderr=%q", out.String(), errOut.String())
			}
			if _, err := os.Stat(config.UserConfigPath()); !errors.Is(err, os.ErrNotExist) {
				t.Fatalf("ineligible invocation wrote config: %v", err)
			}
		})
	}
}

func TestLegacySafeModeEnvDoesNotAlterConfiguredCLITelemetry(t *testing.T) {
	isolateCLIConfigHome(t)
	clearCLITelemetryPolicyEnv(t)
	t.Setenv("REASONIX_SAFE_MODE", "1")
	cfg := config.Default()
	if err := cfg.SetCLITelemetryMode("auto"); err != nil {
		t.Fatal(err)
	}
	previousStart := startCLITelemetryReporter
	t.Cleanup(func() { startCLITelemetryReporter = previousStart })
	want := &telemetry.Reporter{}
	startCLITelemetryReporter = func(telemetry.Options) *telemetry.Reporter { return want }
	if got := startCLITelemetryWithIO(cfg, telemetry.Options{
		Version: "v1.20.0", Interactive: true, CLIMode: "tui",
	}, strings.NewReader(""), io.Discard, io.Discard); got != want {
		t.Fatalf("telemetry reporter = %p, want %p", got, want)
	}
}

func TestCLITelemetryConsentPromptIsLocalized(t *testing.T) {
	isolateCLIConfigHome(t)
	clearCLITelemetryPolicyEnv(t)
	previousSave := persistCLITelemetryConsent
	previousStart := startCLITelemetryReporter
	t.Cleanup(func() {
		persistCLITelemetryConsent = previousSave
		startCLITelemetryReporter = previousStart
		i18n.DetectLanguage("en")
	})
	persistCLITelemetryConsent = func(string) error { return nil }
	startCLITelemetryReporter = func(telemetry.Options) *telemetry.Reporter { return nil }

	for _, lang := range []string{"en", "zh", "zh-TW"} {
		i18n.DetectLanguage(lang)
		var out bytes.Buffer
		startCLITelemetryWithIO(config.Default(), telemetry.Options{
			Version: "v1.20.0", Interactive: true, CLIMode: "tui",
		}, strings.NewReader("\n"), &out, io.Discard)
		for _, required := range []string{"crash.reasonix.io", "reasonix config telemetry off", "[Y/n]:"} {
			if !strings.Contains(out.String(), required) {
				t.Fatalf("%s consent prompt missing %q: %q", lang, required, out.String())
			}
		}
	}
}

func clearCLITelemetryPolicyEnv(t *testing.T) {
	t.Helper()
	for _, key := range []string{
		"DO_NOT_TRACK", "REASONIX_TELEMETRY", "REASONIX_SAFE_MODE", "CI", "CONTINUOUS_INTEGRATION",
		"GITHUB_ACTIONS", "GITLAB_CI", "BUILDKITE", "CIRCLECI", "JENKINS_URL",
		"TEAMCITY_VERSION", "TF_BUILD",
	} {
		t.Setenv(key, "")
	}
}

func TestSetupOverwritePromptShowsYNDefault(t *testing.T) {
	t.Cleanup(func() { i18n.DetectLanguage("en") })
	for _, lang := range []string{"en", "zh"} {
		i18n.DetectLanguage(lang)
		var out bytes.Buffer
		if confirmReconfigureExistingConfig("config.toml", bufio.NewScanner(strings.NewReader("\n")), &out) {
			t.Fatalf("%s empty overwrite answer should keep existing config", lang)
		}
		if !strings.Contains(out.String(), "[y/N]:") {
			t.Fatalf("%s overwrite prompt should show explicit [y/N] default, got %q", lang, out.String())
		}
	}
}

// TestConfigureKeys verifies that a shared api_key_env (each vendor's SKUs use
// the same env var) is asked only once, and entered keys become env lines.
func TestConfigureKeys(t *testing.T) {
	// Force a clean baseline: any DEEPSEEK_API_KEY in the
	// process env (e.g. inherited from the test runner) would be picked up
	// by the new "reuse existing" path and the prompt would be skipped,
	// making the assertion below noisy.
	t.Setenv("DEEPSEEK_API_KEY", "")

	selected := config.Default().Providers

	input := "ds-key\n"
	env := configureKeys(selected, strings.NewReader(input), io.Discard)

	if len(env) != 1 {
		t.Fatalf("env = %v (want 1: DeepSeek asked once)", env)
	}
	if env[0] != "DEEPSEEK_API_KEY=ds-key" {
		t.Errorf("env[0] = %q", env[0])
	}
}

// TestConfigureKeysReusesExistingEnv covers the "user already typed the key
// in the URL-fetch flow, don't ask again" path. When the env var is set
// (either from .env or from a prior os.Setenv in the wizard), configureKeys
// must NOT consume from the input stream — otherwise the user's next typed
// line bleeds into the next provider's prompt. It also must include the
// existing value in envLines so the value is re-pinned into .env on
// re-runs of setup.
func TestConfigureKeysReusesExistingEnv(t *testing.T) {
	t.Setenv("DEEPSEEK_API_KEY", "preset-ds-key")

	selected := config.Default().Providers
	var output bytes.Buffer
	env := configureKeys(selected, strings.NewReader("\n"), &output)

	if len(env) != 1 {
		t.Fatalf("env = %v (want 1: DeepSeek reused)", env)
	}
	if env[0] != "DEEPSEEK_API_KEY=preset-ds-key" {
		t.Errorf("env[0] = %q, want re-pinned existing value", env[0])
	}
	if !strings.Contains(output.String(), "DEEPSEEK_API_KEY") {
		t.Errorf("expected a 'reusing' confirmation for DEEPSEEK_API_KEY, got:\n%s", output.String())
	}
}

func TestConfigureKeysCanResetExistingEnv(t *testing.T) {
	t.Setenv("DEEPSEEK_API_KEY", "stale-ds-key")

	selected := config.Default().Providers
	var output bytes.Buffer
	env := configureKeys(selected, strings.NewReader("y\nfresh-ds-key\n"), &output)

	if len(env) != 1 {
		t.Fatalf("env = %v (want 1: DeepSeek reset)", env)
	}
	if env[0] != "DEEPSEEK_API_KEY=fresh-ds-key" {
		t.Errorf("env[0] = %q, want freshly entered value", env[0])
	}
	if !strings.Contains(output.String(), "[y/N]:") || !strings.Contains(output.String(), "DEEPSEEK_API_KEY") {
		t.Errorf("expected a reset confirmation for DEEPSEEK_API_KEY, got:\n%s", output.String())
	}
}

// TestConfigureKeysAllSetDefaultsToReusingInput ensures that when every env var
// is already populated, pressing Enter at each confirmation keeps the values.
func TestConfigureKeysAllSetDefaultsToReusingInput(t *testing.T) {
	t.Setenv("DEEPSEEK_API_KEY", "ds")

	selected := config.Default().Providers
	env := configureKeys(selected, strings.NewReader("\n"), io.Discard)
	if len(env) != 1 {
		t.Errorf("env = %v, want 1 (DeepSeek reused)", env)
	}
}

// TestAppendEnvUpsertReplacesExistingKey covers the bug where re-running the
// wizard with a corrected key would append a second line for the same env
// var. Without dedupe, different dotenv readers can disagree on which
// assignment wins, leaving stale keys hard to diagnose.
func TestAppendEnvUpsertReplacesExistingKey(t *testing.T) {
	t.Setenv("DEEPSEEK_API_KEY", "") // also covers the os.Setenv pin path
	p := filepath.Join(testenv.TempDir(t), ".env")
	os.WriteFile(p, []byte("# initial\nDEEPSEEK_API_KEY=stale\nMIMO_API_KEY=keepme\n"), 0o600)

	if err := appendEnv(p, []string{"DEEPSEEK_API_KEY=fresh"}); err != nil {
		t.Fatalf("appendEnv: %v", err)
	}
	got, _ := os.ReadFile(p)
	want := "# initial\nMIMO_API_KEY=keepme\nDEEPSEEK_API_KEY=fresh\n"
	if string(got) != want {
		t.Errorf("after upsert =\n%s\nwant =\n%s", got, want)
	}
	if got := os.Getenv("DEEPSEEK_API_KEY"); got != "fresh" {
		t.Errorf("process env DEEPSEEK_API_KEY = %q, want %q (upsert should pin in-process)", got, "fresh")
	}
}

// TestAppendEnvUpsertHandlesExportPrefix proves `export FOO=...` style lines
// also get replaced, since users might hand-edit .env in shell-friendly form.
func TestAppendEnvUpsertHandlesExportPrefix(t *testing.T) {
	t.Setenv("FOO", "")
	p := filepath.Join(testenv.TempDir(t), ".env")
	os.WriteFile(p, []byte("export FOO=old\nKEEP=yes\n"), 0o600)
	if err := appendEnv(p, []string{"FOO=new"}); err != nil {
		t.Fatalf("appendEnv: %v", err)
	}
	got, _ := os.ReadFile(p)
	if !strings.Contains(string(got), "FOO=new") || strings.Contains(string(got), "FOO=old") {
		t.Errorf("export-prefixed line not replaced:\n%s", got)
	}
}

// TestGroupByFamily verifies the wizard collapses the whole default preset into
// the single "deepseek" family, in the order each provider appears. The member
// list is derived from the preset rather than pinned: adding a provider to the
// family is an ordinary change, and pinning the count only dates the test.
func TestGroupByFamily(t *testing.T) {
	providers := config.Default().Providers
	if len(providers) < 2 {
		t.Fatalf("default preset has %d providers; the grouping assertion is vacuous", len(providers))
	}
	order, members, info := groupByFamily(providers)

	if !reflect.DeepEqual(order, []string{"deepseek"}) {
		t.Fatalf("family order = %v, want [deepseek]", order)
	}
	wantMembers := make([]int, len(providers))
	for i := range wantMembers {
		wantMembers[i] = i
	}
	if !reflect.DeepEqual(members["deepseek"], wantMembers) {
		t.Errorf("deepseek members = %v, want every provider in order %v", members["deepseek"], wantMembers)
	}
	if info["deepseek"].name != "DeepSeek" {
		t.Errorf("display name = %q", info["deepseek"].name)
	}
}

// TestFetchOrFallbackLiveReturns covers the happy path: a live /models call
// succeeds and its result wins over the preset's static list. We can't run
// the real probe (no key) so the FetchModels call is expected to 401 and the
// fallback path runs; the assertion below is that fallback works (static
// list returned) and that an empty base URL short-circuits to the static
// list with no network call.
func TestFetchOrFallback(t *testing.T) {
	t.Run("empty base URL returns static list", func(t *testing.T) {
		probe := config.ProviderEntry{
			BaseURL: "",
			Models:  []string{"preset-a", "preset-b"},
		}
		got := fetchOrFallback(&probe, "Test")
		if !reflect.DeepEqual(got, []string{"preset-a", "preset-b"}) {
			t.Errorf("got %v, want preset-a/b", got)
		}
	})

	t.Run("no key set returns static list (offline first-run)", func(t *testing.T) {
		t.Setenv("REASONIX_FETCH_TEST_KEY", "")
		probe := config.ProviderEntry{
			BaseURL:   "http://127.0.0.1:1", // unreachable, no listener
			APIKeyEnv: "REASONIX_FETCH_TEST_KEY",
			Models:    []string{"preset-a"},
		}
		got := fetchOrFallback(&probe, "Test")
		if !reflect.DeepEqual(got, []string{"preset-a"}) {
			t.Errorf("got %v, want preset-a", got)
		}
	})
}

// TestFetchModelListCompatWalksCandidates covers the wizard's custom-provider
// model probe. Previously the probe was a single URL (baseURL+"/models"),
// which worked for OpenAI vendors with a /v1 base URL but silently failed
// for Anthropic-style root URLs (no /v1) and Anthropic-compatible proxies
// (a /v1 base URL but a /v1/messages endpoint). The new helper walks
// BuildModelFetchURLs's candidate list — root + /v1 + known compat
// suffixes — so the same probe now succeeds for both shapes, matching
// what the conversation-time client URL will actually be.
func TestFetchModelListCompatWalksCandidates(t *testing.T) {
	t.Run("anthropic root form resolves via v1 fallback", func(t *testing.T) {
		var gotPath atomic.Value
		gotPath.Store("")
		srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			gotPath.Store(r.URL.Path)
			if r.URL.Path == "/v1/models" {
				w.Header().Set("Content-Type", "application/json")
				_, _ = io.WriteString(w, `{"data":[{"id":"claude-test"}]}`)
				return
			}
			w.WriteHeader(http.StatusNotFound)
		}))
		defer srv.Close()

		models, err := fetchModelListCompat(context.Background(), srv.URL, "k")
		if err != nil {
			t.Fatalf("fetchModelListCompat: %v", err)
		}
		if !reflect.DeepEqual(models, []string{"claude-test"}) {
			t.Errorf("models = %v, want [claude-test]", models)
		}
		if got := gotPath.Load().(string); got != "/v1/models" {
			t.Errorf("probe path = %q, want /v1/models (root form should fall through to v1 candidate)", got)
		}
	})

	t.Run("versioned v1 base URL hits models directly", func(t *testing.T) {
		var gotPath atomic.Value
		gotPath.Store("")
		srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			gotPath.Store(r.URL.Path)
			w.Header().Set("Content-Type", "application/json")
			_, _ = io.WriteString(w, `{"data":[{"id":"model-a"}]}`)
		}))
		defer srv.Close()

		models, err := fetchModelListCompat(context.Background(), srv.URL+"/v1", "k")
		if err != nil {
			t.Fatalf("fetchModelListCompat: %v", err)
		}
		if !reflect.DeepEqual(models, []string{"model-a"}) {
			t.Errorf("models = %v, want [model-a]", models)
		}
		if got := gotPath.Load().(string); got != "/v1/models" {
			t.Errorf("probe path = %q, want /v1/models", got)
		}
	})

	t.Run("endpoint-miss on every candidate returns empty (manual flow)", func(t *testing.T) {
		srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
			w.WriteHeader(http.StatusNotFound)
		}))
		defer srv.Close()

		models, err := fetchModelListCompat(context.Background(), srv.URL, "k")
		if err != nil {
			t.Fatalf("expected graceful empty result on all-miss, got err: %v", err)
		}
		if len(models) != 0 {
			t.Errorf("expected empty models on all-miss, got %v", models)
		}
	})

	t.Run("non-404 network error short-circuits with the real error", func(t *testing.T) {
		// Point at a closed port — connection refused, not a 404.
		models, err := fetchModelListCompat(context.Background(), "http://127.0.0.1:1", "k")
		if err == nil {
			t.Fatalf("expected error for unreachable host, got models=%v", models)
		}
	})
}

// TestFamilyStaticModels proves the offline fallback unions every member of a
// family (the flash + pro SKUs), not just the first — the regression that left
// users with only flash when the live /models probe failed.
func TestFamilyStaticModels(t *testing.T) {
	providers := []config.ProviderEntry{
		{Name: "deepseek-flash", Model: "deepseek-v4-flash"},
		{Name: "deepseek-pro", Model: "deepseek-v4-pro"},
		{Name: "mimo-flash", Model: "mimo-v2.5"},
	}
	got := familyStaticModels(providers, []int{0, 1})
	want := []string{"deepseek-v4-flash", "deepseek-v4-pro"}
	if !reflect.DeepEqual(got, want) {
		t.Errorf("got %v, want %v", got, want)
	}
}

func TestFamilyStaticModelsDedupes(t *testing.T) {
	providers := []config.ProviderEntry{
		{Name: "a", Models: []string{"x", "y"}},
		{Name: "b", Models: []string{"y", "z"}},
	}
	got := familyStaticModels(providers, []int{0, 1})
	if !reflect.DeepEqual(got, []string{"x", "y", "z"}) {
		t.Errorf("got %v, want x/y/z deduped", got)
	}
}

// TestBuildFamilyEntriesSplitsPricing proves flash and pro land in separate
// entries carrying their own price, rather than collapsing into one entry that
// would bill pro at flash's rate.
func TestBuildFamilyEntriesSplitsPricing(t *testing.T) {
	flash := config.ProviderEntry{Name: "deepseek-flash", BaseURL: "https://api.deepseek.com", Model: "deepseek-v4-flash", Price: &provider.Pricing{Input: 1, Output: 2}}
	pro := config.ProviderEntry{Name: "deepseek-pro", BaseURL: "https://api.deepseek.com", Model: "deepseek-v4-pro", Price: &provider.Pricing{Input: 3, Output: 6}}
	got := buildFamilyEntries(flash, []config.ProviderEntry{flash, pro}, []string{"deepseek-v4-flash", "deepseek-v4-pro"})
	if len(got) != 2 {
		t.Fatalf("got %d entries, want 2", len(got))
	}
	byName := map[string]config.ProviderEntry{}
	for _, e := range got {
		byName[e.Name] = e
	}
	if e := byName["deepseek-flash"]; e.Model != "deepseek-v4-flash" || e.Price == nil || e.Price.Output != 2 {
		t.Errorf("flash entry wrong: %+v (price %+v)", e, e.Price)
	}
	if e := byName["deepseek-pro"]; e.Model != "deepseek-v4-pro" || e.Price == nil || e.Price.Output != 6 {
		t.Errorf("pro entry wrong: %+v (price %+v)", e, e.Price)
	}
}

// TestBuildFamilyEntriesUnknownModelUsesProbe puts a live-only SKU (no matching
// preset) under the probe entry rather than dropping it.
func TestBuildFamilyEntriesUnknownModelUsesProbe(t *testing.T) {
	flash := config.ProviderEntry{Name: "deepseek-flash", Model: "deepseek-v4-flash", Price: &provider.Pricing{Input: 1}}
	got := buildFamilyEntries(flash, []config.ProviderEntry{flash}, []string{"deepseek-v4-flash", "deepseek-v9-experimental"})
	if len(got) != 1 || got[0].Name != "deepseek-flash" {
		t.Fatalf("got %+v, want one deepseek-flash entry", got)
	}
	if !reflect.DeepEqual(got[0].Models, []string{"deepseek-v4-flash", "deepseek-v9-experimental"}) {
		t.Errorf("Models = %v, want both under the probe entry", got[0].Models)
	}
}

// TestBuildFamilyEntry covers the three observable behaviors:
//   - The selected models land in the entry's Models field, with Model
//     pointed at the first one so legacy single-model lookups still work.
//   - A preset Default that points to a model the user didn't pick is
//     reset to the first selected model (otherwise resolve-by-default
//     would silently break).
//   - A preset Default that IS in the selection is preserved.
func TestBuildFamilyEntry(t *testing.T) {
	t.Run("default reset when not in selection", func(t *testing.T) {
		probe := config.ProviderEntry{
			Name: "deepseek", Kind: "openai",
			BaseURL: "https://api.deepseek.com",
			Models:  []string{"deepseek-v4-flash", "deepseek-v4-pro"},
			Default: "deepseek-v4-pro",
		}
		got := buildFamilyEntry(probe, []string{"deepseek-v4-flash"})
		if got.Model != "deepseek-v4-flash" {
			t.Errorf("Model = %q, want deepseek-v4-flash", got.Model)
		}
		if got.Default != "deepseek-v4-flash" {
			t.Errorf("Default = %q, want reset to first selected", got.Default)
		}
		if !reflect.DeepEqual(got.Models, []string{"deepseek-v4-flash"}) {
			t.Errorf("Models = %v", got.Models)
		}
		if got.BaseURL != "https://api.deepseek.com" {
			t.Errorf("BaseURL lost: %q", got.BaseURL)
		}
	})

	t.Run("default preserved when in selection", func(t *testing.T) {
		probe := config.ProviderEntry{
			Name: "deepseek", Default: "deepseek-v4-pro",
			BaseURL: "https://api.deepseek.com",
		}
		got := buildFamilyEntry(probe, []string{"deepseek-v4-flash", "deepseek-v4-pro"})
		if got.Default != "deepseek-v4-pro" {
			t.Errorf("Default = %q, want preserved", got.Default)
		}
	})

	t.Run("empty default filled from first selected", func(t *testing.T) {
		probe := config.ProviderEntry{Name: "x", BaseURL: "u"}
		got := buildFamilyEntry(probe, []string{"alpha", "beta"})
		if got.Default != "alpha" {
			t.Errorf("Default = %q, want alpha", got.Default)
		}
	})
}

// TestProviderSlug covers the host-derivation rules and the sha1 fallback
// for unparseable URLs. The exact format isn't load-bearing — what matters
// is that the slug (a) starts with the kind prefix, (b) is stable across
// calls with the same URL, and (c) never produces the bare "custom" /
// "anthropic" magic names that would collide with the wizard menu items.
func TestProviderSlug(t *testing.T) {
	cases := []struct {
		name, kind, url, want string
	}{
		{"standard host with port", "custom", "https://token.sensenova.cn/v1", "custom-token-sensenova-cn"},
		{"api subdomain", "custom", "https://api.openai.com/v1", "custom-api-openai-com"},
		{"www stripped", "custom", "https://www.example.com/v1", "custom-example-com"},
		{"port preserved", "custom", "http://localhost:11434/v1", "custom-localhost-11434"},
		{"anthropic kind", "anthropic", "https://api.anthropic.com", "anthropic-api-anthropic-com"},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			if got := providerSlug(tc.kind, tc.url); got != tc.want {
				t.Errorf("providerSlug(%q, %q) = %q, want %q", tc.kind, tc.url, got, tc.want)
			}
		})
	}

	t.Run("stable across calls", func(t *testing.T) {
		a := providerSlug("custom", "https://token.sensenova.cn/v1")
		b := providerSlug("custom", "https://token.sensenova.cn/v1")
		if a != b {
			t.Errorf("not stable: %q vs %q", a, b)
		}
		if a == "custom" {
			t.Error("slug degenerated to bare magic name — collision risk")
		}
	})

	t.Run("sha1 fallback for unparseable URL", func(t *testing.T) {
		got := providerSlug("custom", "://not a url::://")
		if !strings.HasPrefix(got, "custom-") || got == "custom" {
			t.Errorf("fallback slug = %q, want custom-<hex>", got)
		}
		// sha1 is 40 hex chars; we take 4 bytes (8 hex chars).
		if len(got) != len("custom-")+8 {
			t.Errorf("fallback slug = %q, want 8 hex chars after prefix", got)
		}
	})

	t.Run("sha1 fallback for non-ascii host", func(t *testing.T) {
		got := providerSlug("custom", "https://例子.测试/v1")
		if !strings.HasPrefix(got, "custom-") || got == "custom-" {
			t.Errorf("fallback slug = %q, want custom-<hex>", got)
		}
		if len(got) != len("custom-")+8 {
			t.Errorf("fallback slug = %q, want 8 hex chars after prefix", got)
		}
	})
}

func TestAPIKeyEnvFromProviderName(t *testing.T) {
	cases := []struct {
		name, providerName, want string
	}{
		{"custom host slug", "custom-token-sensenova-cn", "CUSTOM_TOKEN_SENSENOVA_CN_API_KEY"},
		{"localhost slug with port", "custom-localhost-11434", "CUSTOM_LOCALHOST_11434_API_KEY"},
		{"desktop-style custom name", "Local Gateway", "LOCAL_GATEWAY_API_KEY"},
		{"digit-leading provider name", "9router", "CUSTOM_9ROUTER_API_KEY"},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			if got := apiKeyEnvFromProviderName(tc.providerName); got != tc.want {
				t.Errorf("apiKeyEnvFromProviderName(%q) = %q, want %q", tc.providerName, got, tc.want)
			}
		})
	}

	t.Run("non-ascii provider names use desktop-compatible hash fallback", func(t *testing.T) {
		if got, want := apiKeyEnvFromProviderName("商汤"), "CUSTOM_d39b9067_API_KEY"; got != want {
			t.Errorf("apiKeyEnvFromProviderName(non-ascii) = %q, want %q", got, want)
		}
		if got := apiKeyEnvFromProviderName("通义千问"); got == "CUSTOM_d39b9067_API_KEY" || got == "CUSTOM_API_KEY" {
			t.Errorf("apiKeyEnvFromProviderName(second non-ascii) = %q, want distinct stable fallback", got)
		}
	})
}

func TestPromptCustomProviderManualDefaultsKeyEnvFromBaseURL(t *testing.T) {
	result, err := promptCustomProviderManualWith(
		bufio.NewScanner(strings.NewReader("sensenova-chat\n\n\n")),
		"openai",
		"https://token.sensenova.cn/v1",
		"",
		"",
	)
	if err != nil {
		t.Fatalf("promptCustomProviderManualWith: %v", err)
	}
	entries := result.entries
	if len(entries) != 1 {
		t.Fatalf("entries = %d, want 1", len(entries))
	}
	if got, want := entries[0].APIKeyEnv, "CUSTOM_TOKEN_SENSENOVA_CN_API_KEY"; got != want {
		t.Errorf("APIKeyEnv = %q, want %q", got, want)
	}
}

func TestPromptCustomProviderManualPreservesExplicitKeyEnv(t *testing.T) {
	result, err := promptCustomProviderManualWith(
		bufio.NewScanner(strings.NewReader("manual-chat\n\n")),
		"openai",
		"https://token.sensenova.cn/v1",
		"CUSTOM_API_KEY",
		"",
	)
	if err != nil {
		t.Fatalf("promptCustomProviderManualWith: %v", err)
	}
	entries := result.entries
	if len(entries) != 1 {
		t.Fatalf("entries = %d, want 1", len(entries))
	}
	if got := entries[0].APIKeyEnv; got != "CUSTOM_API_KEY" {
		t.Errorf("APIKeyEnv = %q, want explicit CUSTOM_API_KEY", got)
	}
}

func TestPromptAPIKeyEnvNameRejectsModelName(t *testing.T) {
	i18n.DetectLanguage("en")
	var out bytes.Buffer
	got := promptAPIKeyEnvName(
		bufio.NewScanner(strings.NewReader("grok-4.5\n\n")),
		&out,
		i18n.M.CustomPromptKeyEnv,
		"CUSTOM_API_YAIROUTER_COM_API_KEY",
	)
	if got != "CUSTOM_API_YAIROUTER_COM_API_KEY" {
		t.Fatalf("key env = %q, want generated default", got)
	}
	if text := out.String(); !strings.Contains(text, "not a valid API Key variable name") || !strings.Contains(text, "do not enter a model name") {
		t.Fatalf("validation guidance missing from prompt output: %q", text)
	}
}

func TestPromptCustomProviderManualAsksForModelBeforeCredentialName(t *testing.T) {
	result, err := promptCustomProviderManualWith(
		bufio.NewScanner(strings.NewReader("grok-4.5\ngrok-4.5\n\n\n")),
		"openai",
		"https://api.example.com/v1",
		"",
		"",
	)
	if err != nil {
		t.Fatalf("promptCustomProviderManualWith: %v", err)
	}
	entries := result.entries
	if got := entries[0].Model; got != "grok-4.5" {
		t.Fatalf("model = %q, want grok-4.5", got)
	}
	if got := entries[0].APIKeyEnv; got != "CUSTOM_API_EXAMPLE_COM_API_KEY" {
		t.Fatalf("APIKeyEnv = %q, want generated default after invalid model-like input", got)
	}
}

func TestPromptCustomProviderStagesExplicitKeyEvenWhenProcessEnvMatches(t *testing.T) {
	const key = "CUSTOM_API_EXAMPLE_COM_API_KEY"
	t.Setenv(key, "same-secret")
	result, err := promptCustomProviderManualWith(
		bufio.NewScanner(strings.NewReader("grok-4.5\n")),
		"openai",
		"https://api.example.com/v1",
		key,
		"same-secret",
	)
	if err != nil {
		t.Fatalf("promptCustomProviderManualWith: %v", err)
	}
	if got := result.credentials[key]; got != "same-secret" {
		t.Fatalf("staged credential = %q, want explicitly entered value", got)
	}
	if got := os.Getenv(key); got != "same-secret" {
		t.Fatalf("prompt changed process environment to %q", got)
	}
	result, err = promptCustomProviderManualWith(
		bufio.NewScanner(strings.NewReader("grok-4.5\n")),
		"openai",
		"https://api.example.com/v1",
		key,
		"new-secret",
	)
	if err != nil {
		t.Fatalf("promptCustomProviderManualWith with replacement key: %v", err)
	}
	if got := result.credentials[key]; got != "new-secret" {
		t.Fatalf("replacement staged credential = %q", got)
	}
	if got := os.Getenv(key); got != "same-secret" {
		t.Fatalf("prompt leaked replacement credential into process environment: %q", got)
	}
}

func TestRepairInvalidProviderKeyEnvs(t *testing.T) {
	original := []config.ProviderEntry{
		{Name: "custom-relay-example-com", APIKeyEnv: "grok-4.5"},
		{Name: "valid", APIKeyEnv: "VALID_API_KEY"},
		{Name: "no-auth"},
	}
	got, repairs := repairInvalidProviderKeyEnvs(original)
	if len(repairs) != 1 {
		t.Fatalf("repairs = %+v, want one", repairs)
	}
	if got[0].APIKeyEnv != "CUSTOM_RELAY_EXAMPLE_COM_API_KEY" {
		t.Fatalf("repaired key env = %q", got[0].APIKeyEnv)
	}
	if repairs[0].old != "grok-4.5" || repairs[0].new != got[0].APIKeyEnv {
		t.Fatalf("repair detail = %+v", repairs[0])
	}
	if got[1].APIKeyEnv != "VALID_API_KEY" || got[2].APIKeyEnv != "" {
		t.Fatalf("valid/no-auth providers changed: %+v", got)
	}
	if original[0].APIKeyEnv != "grok-4.5" {
		t.Fatalf("repair mutated caller input: %+v", original[0])
	}
}

// TestFilterStaleCustomEntries covers the wizard's auto-cleanup of legacy
// "custom" / "anthropic" magic-name entries that previous versions wrote
// into reasonix.toml. These collide with the wizard's own menu items, so
// they're dropped from the providers list before grouping — but the caller
// still gets them back in the dropped slice to surface a warning.
func TestFilterStaleCustomEntries(t *testing.T) {
	in := []config.ProviderEntry{
		{Name: "deepseek", Kind: "openai", BaseURL: "https://api.deepseek.com"},
		{Name: "custom", Kind: "openai", BaseURL: "https://old.example/v1"},                // stale
		{Name: "anthropic", Kind: "anthropic", BaseURL: "https://old.example/v1/messages"}, // stale
		{Name: "mimo-tp", Kind: "openai", BaseURL: "https://token-plan-cn.xiaomimimo.com/v1"},
	}
	kept, dropped := filterStaleCustomEntries(in)
	if len(kept) != 2 {
		t.Errorf("kept = %d entries, want 2: %+v", len(kept), kept)
	}
	if len(dropped) != 2 {
		t.Errorf("dropped = %d entries, want 2: %+v", len(dropped), dropped)
	}
	for _, k := range kept {
		if k.Name == "custom" || k.Name == "anthropic" {
			t.Errorf("magic name leaked through: %q", k.Name)
		}
	}

	t.Run("non-magic names with kind anthropic are kept", func(t *testing.T) {
		// An entry someone deliberately named "claude" (kind=anthropic) must
		// not be touched by the filter — only the bare "anthropic" magic name.
		in := []config.ProviderEntry{
			{Name: "claude", Kind: "anthropic", BaseURL: "https://api.anthropic.com"},
		}
		kept, dropped := filterStaleCustomEntries(in)
		if len(kept) != 1 || len(dropped) != 0 {
			t.Errorf("claude should be kept, got kept=%d dropped=%d", len(kept), len(dropped))
		}
	})

	t.Run("custom kind anthropic is kept", func(t *testing.T) {
		// Name="custom" with kind=anthropic is ambiguous — keep it.
		in := []config.ProviderEntry{
			{Name: "custom", Kind: "anthropic", BaseURL: "https://x"},
		}
		kept, dropped := filterStaleCustomEntries(in)
		if len(kept) != 1 || len(dropped) != 0 {
			t.Errorf("custom+anthropic should be kept (ambiguous), got kept=%d dropped=%d", len(kept), len(dropped))
		}
	})
}

func TestWithBuiltinFamiliesDoesNotAddMissingMimo(t *testing.T) {
	// The user's case: a reasonix.toml that defines only deepseek providers.
	cfg := []config.ProviderEntry{
		{Name: "deepseek-flash", Kind: "openai", BaseURL: "https://api.deepseek.com"},
		{Name: "deepseek-pro", Kind: "openai", BaseURL: "https://api.deepseek.com"},
	}
	merged := withBuiltinFamilies(cfg)
	order, _, info := groupByFamily(merged)
	seen := map[string]bool{}
	for _, k := range order {
		seen[info[k].name] = true
	}
	if !seen["DeepSeek"] {
		t.Fatalf("wizard families = %v, want DeepSeek", order)
	}
	if seen["MiMo (Xiaomi)"] {
		t.Fatalf("wizard families = %v, should not inject MiMo", order)
	}

	// The merge dedupes by name, so a customized entry must survive exactly once.
	// A builtin the user has not named is a genuine addition, not a duplicate, so
	// this counts names rather than members.
	count := map[string]int{}
	for _, p := range merged {
		count[p.Name]++
	}
	for _, p := range cfg {
		if count[p.Name] != 1 {
			t.Fatalf("%s appears %d times; the user's entry was duplicated", p.Name, count[p.Name])
		}
	}
	for name, n := range count {
		if n != 1 {
			t.Fatalf("%s appears %d times in the merged list", name, n)
		}
	}
	if len(groupByFamilyKeys(merged, "deepseek")) < len(cfg) {
		t.Fatalf("the user's deepseek entries did not survive the merge: %v", merged)
	}
}

func TestWithBuiltinFamiliesForLanguageUsesDeepSeekPricing(t *testing.T) {
	// Language no longer rewrites list prices; defaults stay on the frozen USD table.
	providers := withBuiltinFamiliesForLanguage(nil, "zh")
	var flash *config.ProviderEntry
	for i := range providers {
		if providers[i].Name == "deepseek-flash" {
			flash = &providers[i]
			break
		}
	}
	if flash == nil {
		t.Fatal("deepseek-flash provider missing")
	}
	if flash.Price == nil || flash.Price.Output != 0.6 || flash.Price.Currency != "$" {
		t.Fatalf("flash price = %+v, want frozen USD official table", flash.Price)
	}
}

// TestWithBuiltinFamiliesRestoresSiblingEntries covers the re-run scenario:
// a user previously selected only deepseek-v4-flash (saved as deepseek-flash
// with a single model). Re-running `reasonix setup` must still surface the
// sibling deepseek-pro entry so the user can pick deepseek-v4-pro too,
// rather than only showing the previously selected model.
func TestWithBuiltinFamiliesRestoresSiblingEntries(t *testing.T) {
	cfg := []config.ProviderEntry{
		{Name: "deepseek-flash", Kind: "openai", BaseURL: "https://api.deepseek.com", Model: "deepseek-v4-flash", Models: []string{"deepseek-v4-flash"}, APIKeyEnv: "DEEPSEEK_API_KEY"},
	}
	got := withBuiltinFamilies(cfg)

	// deepseek-pro must be restored even though deepseek family already exists.
	var found bool
	for _, p := range got {
		if p.Name == "deepseek-pro" {
			found = true
			break
		}
	}
	if !found {
		t.Fatalf("withBuiltinFamilies(%+v) = %v, want deepseek-pro sibling restored", cfg, namesOf(got))
	}

	// The static model list for the deepseek family must include both SKUs.
	_, members, _ := groupByFamily(got)
	deepseekIdxs := members["deepseek"]
	models := familyStaticModels(got, deepseekIdxs)
	wantModels := map[string]bool{"deepseek-v4-flash": true, "deepseek-v4-pro": true}
	for _, m := range models {
		delete(wantModels, m)
	}
	if len(wantModels) > 0 {
		t.Errorf("familyStaticModels = %v, missing %v", models, wantModels)
	}
}

func namesOf(ps []config.ProviderEntry) []string {
	out := make([]string, len(ps))
	for i, p := range ps {
		out[i] = p.Name
	}
	return out
}

func groupByFamilyKeys(ps []config.ProviderEntry, key string) []int {
	_, members, _ := groupByFamily(ps)
	return members[key]
}

func TestWriteDefaultConfigOmitsLegacyInternalMCPSections(t *testing.T) {
	path := filepath.Join(testenv.TempDir(t), "reasonix.toml")
	if rc := writeDefaultConfig(path); rc != 0 {
		t.Fatalf("writeDefaultConfig rc = %d", rc)
	}
	raw, err := os.ReadFile(path)
	if err != nil {
		t.Fatal(err)
	}
	text := string(raw)
	for _, forbidden := range []string{"[codegraph]", "[builtin_mcp]", "[builtin_mcp_updates]"} {
		if strings.Contains(text, forbidden) {
			t.Fatalf("default config should omit %s:\n%s", forbidden, text)
		}
	}
}

func captureStderr(t *testing.T, fn func()) string {
	t.Helper()
	old := os.Stderr
	r, w, err := os.Pipe()
	if err != nil {
		t.Fatal(err)
	}
	os.Stderr = w
	defer func() { os.Stderr = old }()

	fn()
	if err := w.Close(); err != nil {
		t.Fatal(err)
	}
	data, err := io.ReadAll(r)
	if err != nil {
		t.Fatal(err)
	}
	return string(data)
}

func captureCLIOutput(t *testing.T, fn func()) (stdout, stderr string) {
	t.Helper()
	stderr = captureStderr(t, func() {
		stdout = captureStdout(t, fn)
	})
	return stdout, stderr
}

func TestProvidersWithMissingKeysOnlyReferenced(t *testing.T) {
	t.Setenv("DEEPSEEK_API_KEY", "")
	t.Setenv("MIMO_API_KEY", "")
	cfg := config.Default()

	got := providersWithMissingKeys(cfg)
	envs := map[string]bool{}
	for _, p := range got {
		envs[p.APIKeyEnv] = true
	}
	if !envs["DEEPSEEK_API_KEY"] {
		t.Errorf("the default model's missing key must be prompted, got %v", got)
	}
	if envs["MIMO_API_KEY"] {
		t.Errorf("unreferenced preset keys must not be prompted, got %v", got)
	}
}

func TestProvidersWithMissingKeysIncludesPlannerModel(t *testing.T) {
	t.Setenv("DEEPSEEK_API_KEY", "set")
	t.Setenv("MIMO_API_KEY", "")
	cfg := config.Default()
	cfg.Providers = append(cfg.Providers, config.ProviderEntry{Name: "mimo-pro", Kind: "openai", BaseURL: "https://token-plan-cn.xiaomimimo.com/v1", Model: "mimo-v2.5-pro", APIKeyEnv: "MIMO_API_KEY"})
	cfg.Agent.PlannerModel = "mimo-pro"

	got := providersWithMissingKeys(cfg)
	if len(got) != 1 || got[0].APIKeyEnv != "MIMO_API_KEY" {
		t.Errorf("planner model's missing key must be prompted, got %+v", got)
	}
}

func TestParseRuntimeProfile(t *testing.T) {
	for input, want := range map[string]string{
		"":         boot.TokenModeFull,
		"balanced": boot.TokenModeFull,
		"full":     boot.TokenModeFull,
		"economy":  boot.TokenModeEconomy,
		"delivery": boot.TokenModeDelivery,
	} {
		got, err := parseRuntimeProfile(input)
		if err != nil || got != want {
			t.Errorf("parseRuntimeProfile(%q) = %q, %v; want %q", input, got, err, want)
		}
	}
	if _, err := parseRuntimeProfile("fast"); err == nil {
		t.Fatal("unknown profile should fail")
	}
}