package cloudinit

import (
	"os"
	"path/filepath"
	"strings"
	"testing"
)

func TestBuildExtraFlags(t *testing.T) {
	tests := []struct {
		name string
		cfg  Config
		want string
	}{
		{
			name: "empty",
			cfg:  Config{},
			want: "",
		},
		{
			name: "extra flags only",
			cfg: Config{
				KubeSolo: KubeSoloConfig{ExtraFlags: "--disable traefik"},
			},
			want: "--disable traefik",
		},
		{
			name: "extra sans only",
			cfg: Config{
				KubeSolo: KubeSoloConfig{
					ExtraSANs: []string{"node.local", "192.168.1.100"},
				},
			},
			want: "--apiserver-extra-sans node.local --apiserver-extra-sans 192.168.1.100",
		},
		{
			name: "flags and sans",
			cfg: Config{
				KubeSolo: KubeSoloConfig{
					ExtraFlags: "--disable servicelb",
					ExtraSANs:  []string{"edge.local"},
				},
			},
			want: "--disable servicelb --apiserver-extra-sans edge.local",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := buildExtraFlags(&tt.cfg)
			if got != tt.want {
				t.Errorf("buildExtraFlags() = %q, want %q", got, tt.want)
			}
		})
	}
}

func TestApplyKubeSolo(t *testing.T) {
	dir := t.TempDir()
	tr := true
	cfg := &Config{
		KubeSolo: KubeSoloConfig{
			ExtraFlags:   "--disable traefik",
			LocalStorage: &tr,
			ExtraSANs:    []string{"test.local"},
		},
	}

	if err := ApplyKubeSolo(cfg, dir); err != nil {
		t.Fatalf("ApplyKubeSolo error: %v", err)
	}

	// Check extra-flags file
	flagsData, err := os.ReadFile(filepath.Join(dir, "extra-flags"))
	if err != nil {
		t.Fatalf("reading extra-flags: %v", err)
	}
	flags := strings.TrimSpace(string(flagsData))
	if !strings.Contains(flags, "--disable traefik") {
		t.Errorf("extra-flags missing '--disable traefik': %q", flags)
	}
	if !strings.Contains(flags, "--apiserver-extra-sans test.local") {
		t.Errorf("extra-flags missing SANs: %q", flags)
	}

	// Check config.yaml
	configData, err := os.ReadFile(filepath.Join(dir, "config.yaml"))
	if err != nil {
		t.Fatalf("reading config.yaml: %v", err)
	}
	config := string(configData)
	if !strings.Contains(config, "local-storage: true") {
		t.Errorf("config.yaml missing local-storage: %q", config)
	}
	if !strings.Contains(config, "data-dir: /var/lib/kubesolo") {
		t.Errorf("config.yaml missing data-dir: %q", config)
	}
}

func TestApplyKubeSoloNoFlags(t *testing.T) {
	dir := t.TempDir()
	cfg := &Config{}

	if err := ApplyKubeSolo(cfg, dir); err != nil {
		t.Fatalf("ApplyKubeSolo error: %v", err)
	}

	// extra-flags should not exist when empty
	if _, err := os.Stat(filepath.Join(dir, "extra-flags")); !os.IsNotExist(err) {
		t.Error("extra-flags file should not exist when no flags configured")
	}

	// config.yaml should still be created with defaults
	if _, err := os.Stat(filepath.Join(dir, "config.yaml")); err != nil {
		t.Error("config.yaml should be created even with empty config")
	}
}