package system

import (
	"context"
	"os"
	"path/filepath"
	"testing"

	"ws_usernode/internal/config"
)

func testCfg() config.SystemConfig {
	cfg := config.Default()
	return cfg.System
}

func TestDryRunDoesNotExecute(t *testing.T) {
	cfg := testCfg()
	cfg.DryRun = true
	m := New(cfg)
	ctx := context.Background()

	// dry-run 下创建/删除不应报错（只打印计划）
	if err := m.CreateUser(ctx, Account{Username: "ext_zhangsan"}); err != nil {
		t.Fatalf("dry-run create: %v", err)
	}
	if err := m.RemoveUser(ctx, "ext_zhangsan"); err != nil {
		t.Fatalf("dry-run remove: %v", err)
	}
	if err := m.SetLock(ctx, "ext_zhangsan", true); err != nil {
		t.Fatalf("dry-run lock: %v", err)
	}
	if err := m.SyncAuthorizedKeys(ctx, "ext_zhangsan", nil); err != nil {
		t.Fatalf("dry-run keys: %v", err)
	}
	// 参数校验在 dry-run 下仍然生效
	if err := m.CreateUser(ctx, Account{Username: "ext_..bad"}); err == nil {
		t.Fatal("expected validation error for illegal account name")
	}
}

func TestExistsParsesPasswd(t *testing.T) {
	// 用临时 passwd 文件验证解析逻辑
	dir := t.TempDir()
	p := filepath.Join(dir, "passwd")
	content := "root:x:0:0:root:/root:/bin/sh\n" +
		"ext_zhangsan:x:1001:1001::/home/ext_zhangsan:/bin/sh\n"
	if err := os.WriteFile(p, []byte(content), 0o600); err != nil {
		t.Fatal(err)
	}
	old := passwdPath
	passwdPath = p
	t.Cleanup(func() { passwdPath = old })

	m := New(testCfg())
	ctx := context.Background()
	// 已带前缀与未带前缀都会命中同一账号
	for _, name := range []string{"ext_zhangsan", "zhangsan"} {
		ok, err := m.Exists(ctx, name)
		if err != nil {
			t.Fatalf("exists(%q): %v", name, err)
		}
		if !ok {
			t.Fatalf("user %q should exist", name)
		}
	}
	// 不存在的账号返回 false
	if ok, err := m.Exists(ctx, "ghost_xyz"); err != nil || ok {
		t.Fatalf("exists(ghost) = %v/%v, want false/nil", ok, err)
	}
	// 非法账号名直接报错
	if _, err := m.Exists(ctx, "bad..name"); err == nil {
		t.Fatal("expected validation error")
	}
}

func TestPrefixNormalization(t *testing.T) {
	cfg := testCfg()
	lm := &localManager{cfg: cfg}
	// sysName 逻辑：已带前缀不重复加
	if got := lm.sysName("ext_x"); got != "ext_x" {
		t.Fatalf("sysName(ext_x) = %q", got)
	}
	if got := lm.sysName("x"); got != "ext_x" {
		t.Fatalf("sysName(x) = %q", got)
	}
}