DankMaterialShell/core/internal/greeter/installer_test.go

package greeter

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

func writeTestFile(t *testing.T, path string, content string) {
	t.Helper()
	if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
		t.Fatalf("failed to create parent dir for %s: %v", path, err)
	}
	if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
		t.Fatalf("failed to write %s: %v", path, err)
	}
}

func TestResolveGreeterThemeSyncState(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name                    string
		settingsJSON            string
		sessionJSON             string
		wantSourcePath          string
		wantResolvedWallpaper   string
		wantDynamicOverrideUsed bool
	}{
		{
			name: "dynamic theme with greeter wallpaper override uses generated greeter colors",
			settingsJSON: `{
  "currentThemeName": "dynamic",
  "greeterWallpaperPath": "Pictures/blue.jpg",
  "matugenScheme": "scheme-tonal-spot",
  "iconTheme": "Papirus"
}`,
			sessionJSON:             `{"isLightMode":true}`,
			wantSourcePath:          filepath.Join(".cache", "DankMaterialShell", "greeter-colors", "dms-colors.json"),
			wantResolvedWallpaper:   filepath.Join("Pictures", "blue.jpg"),
			wantDynamicOverrideUsed: true,
		},
		{
			name: "dynamic theme without override uses desktop colors",
			settingsJSON: `{
  "currentThemeName": "dynamic",
  "greeterWallpaperPath": ""
}`,
			sessionJSON:             `{"isLightMode":false}`,
			wantSourcePath:          filepath.Join(".cache", "DankMaterialShell", "dms-colors.json"),
			wantResolvedWallpaper:   "",
			wantDynamicOverrideUsed: false,
		},
		{
			name: "non-dynamic theme keeps desktop colors even with override wallpaper",
			settingsJSON: `{
  "currentThemeName": "purple",
  "greeterWallpaperPath": "/tmp/blue.jpg"
}`,
			sessionJSON:             `{"isLightMode":false}`,
			wantSourcePath:          filepath.Join(".cache", "DankMaterialShell", "dms-colors.json"),
			wantResolvedWallpaper:   "/tmp/blue.jpg",
			wantDynamicOverrideUsed: false,
		},
	}

	for _, tt := range tests {
		tt := tt
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			homeDir := t.TempDir()
			writeTestFile(t, filepath.Join(homeDir, ".config", "DankMaterialShell", "settings.json"), tt.settingsJSON)
			writeTestFile(t, filepath.Join(homeDir, ".local", "state", "DankMaterialShell", "session.json"), tt.sessionJSON)

			state, err := resolveGreeterThemeSyncState(homeDir)
			if err != nil {
				t.Fatalf("resolveGreeterThemeSyncState returned error: %v", err)
			}

			if got := state.effectiveColorsSource(homeDir); got != filepath.Join(homeDir, tt.wantSourcePath) {
				t.Fatalf("effectiveColorsSource = %q, want %q", got, filepath.Join(homeDir, tt.wantSourcePath))
			}

			wantResolvedWallpaper := tt.wantResolvedWallpaper
			if wantResolvedWallpaper != "" && !filepath.IsAbs(wantResolvedWallpaper) {
				wantResolvedWallpaper = filepath.Join(homeDir, wantResolvedWallpaper)
			}
			if state.ResolvedGreeterWallpaperPath != wantResolvedWallpaper {
				t.Fatalf("ResolvedGreeterWallpaperPath = %q, want %q", state.ResolvedGreeterWallpaperPath, wantResolvedWallpaper)
			}

			if state.UsesDynamicWallpaperOverride != tt.wantDynamicOverrideUsed {
				t.Fatalf("UsesDynamicWallpaperOverride = %v, want %v", state.UsesDynamicWallpaperOverride, tt.wantDynamicOverrideUsed)
			}
		})
	}
}

func TestUpsertInitialSession(t *testing.T) {
	t.Parallel()

	baseConfig := `[terminal]
vt = 1

[default_session]
user = "greeter"
command = "/usr/bin/dms-greeter --command niri"
`

	t.Run("inserts initial session", func(t *testing.T) {
		t.Parallel()
		got := upsertInitialSession(baseConfig, "alice", "/var/cache/dms-greeter", true)
		if !strings.Contains(got, "[initial_session]") {
			t.Fatalf("expected [initial_session] section, got:\n%s", got)
		}
		if !strings.Contains(got, `user = "alice"`) {
			t.Fatalf("expected alice user in initial session, got:\n%s", got)
		}
		if !strings.Contains(got, `dms greeter launch-session --from-memory --cache-dir`) {
			t.Fatalf("expected stable launch-session command, got:\n%s", got)
		}
		if strings.Contains(got, `exec niri`) {
			t.Fatalf("initial session must not bake the desktop Exec command, got:\n%s", got)
		}
	})

	t.Run("updates existing initial session", func(t *testing.T) {
		t.Parallel()
		existing := baseConfig + `
[initial_session]
user = "bob"
command = "old-command"
`
		got := upsertInitialSession(existing, "alice", "/var/cache/dms-greeter", true)
		if strings.Contains(got, `user = "bob"`) {
			t.Fatalf("expected bob to be replaced, got:\n%s", got)
		}
		if !strings.Contains(got, `dms greeter launch-session --from-memory`) {
			t.Fatalf("expected launch-session command, got:\n%s", got)
		}
	})

	t.Run("removes initial session when disabled", func(t *testing.T) {
		t.Parallel()
		existing := baseConfig + `
[initial_session]
user = "alice"
command = "niri"
`
		got := upsertInitialSession(existing, "", "", false)
		if strings.Contains(got, "[initial_session]") {
			t.Fatalf("expected initial session removed, got:\n%s", got)
		}
		if !strings.Contains(got, "[default_session]") {
			t.Fatalf("expected default session preserved, got:\n%s", got)
		}
	})
}

func TestStripDesktopExecCodes(t *testing.T) {
	t.Parallel()

	got := stripDesktopExecCodes("niri --session %f")
	want := "niri --session"
	if got != want {
		t.Fatalf("stripDesktopExecCodes = %q, want %q", got, want)
	}
}

func TestResolveGreeterAutoLoginState(t *testing.T) {
	t.Parallel()

	cacheDir := t.TempDir()
	homeDir := t.TempDir()

	writeTestFile(t, filepath.Join(cacheDir, "settings.json"), `{
  "greeterAutoLogin": true,
  "greeterRememberLastUser": true,
  "greeterRememberLastSession": true
}`)
	writeTestFile(t, filepath.Join(cacheDir, ".local/state/memory.json"), `{
  "lastSuccessfulUser": "alice",
  "lastSessionDesktopId": "niri.desktop"
}`)

	enabled, loginUser, sessionID, err := resolveGreeterAutoLoginState(cacheDir, homeDir)
	if err != nil {
		t.Fatalf("resolveGreeterAutoLoginState returned error: %v", err)
	}
	if !enabled || loginUser != "alice" || sessionID != "niri.desktop" {
		t.Fatalf("got enabled=%v user=%q session=%q", enabled, loginUser, sessionID)
	}
}

func TestResolveGreeterAutoLoginStateIgnoresStaleSessionExec(t *testing.T) {
	t.Parallel()

	cacheDir := t.TempDir()
	homeDir := t.TempDir()

	writeTestFile(t, filepath.Join(cacheDir, "settings.json"), `{
  "greeterAutoLogin": true,
  "greeterRememberLastUser": true,
  "greeterRememberLastSession": true
}`)
	writeTestFile(t, filepath.Join(cacheDir, ".local/state/memory.json"), `{
  "lastSuccessfulUser": "alice",
  "lastSessionId": "/nix/store/old-session/share/wayland-sessions/example.desktop",
  "lastSessionExec": "/nix/store/old-session/bin/start-example-session"
}`)

	enabled, loginUser, sessionID, err := resolveGreeterAutoLoginState(cacheDir, homeDir)
	if err != nil {
		t.Fatalf("resolveGreeterAutoLoginState returned error: %v", err)
	}
	if !enabled || loginUser != "alice" || sessionID != "example.desktop" {
		t.Fatalf("got enabled=%v user=%q session=%q", enabled, loginUser, sessionID)
	}

	got := upsertInitialSession("", loginUser, cacheDir, true)
	if strings.Contains(got, "/nix/store/old-session") {
		t.Fatalf("initial session must not include stale store path, got:\n%s", got)
	}
}

func TestResolveGreeterAutoLoginStateIgnoresMemoryFlag(t *testing.T) {
	t.Parallel()

	cacheDir := t.TempDir()
	homeDir := t.TempDir()

	writeTestFile(t, filepath.Join(cacheDir, "settings.json"), `{
  "greeterAutoLogin": false,
  "greeterRememberLastUser": true,
  "greeterRememberLastSession": true
}`)
	writeTestFile(t, filepath.Join(cacheDir, ".local/state/memory.json"), `{
  "autoLoginEnabled": true,
  "lastSuccessfulUser": "alice",
  "lastSessionExec": "niri"
}`)

	enabled, loginUser, sessionID, err := resolveGreeterAutoLoginState(cacheDir, homeDir)
	if err != nil {
		t.Fatalf("resolveGreeterAutoLoginState returned error: %v", err)
	}
	if enabled || loginUser != "" || sessionID != "" {
		t.Fatalf("expected disabled with empty user/session, got enabled=%v user=%q session=%q", enabled, loginUser, sessionID)
	}
}

func TestResolveSessionExecInDirs(t *testing.T) {
	t.Parallel()

	oldDir := filepath.Join(t.TempDir(), "wayland-sessions")
	newDir := filepath.Join(t.TempDir(), "wayland-sessions")
	writeTestFile(t, filepath.Join(oldDir, "example.desktop"), `[Desktop Entry]
Name=Example Session
Exec=/nix/store/old-session/bin/start-example-session
`)
	writeTestFile(t, filepath.Join(newDir, "example.desktop"), `[Desktop Entry]
Name=Example Session
Exec=/run/current-system/sw/bin/start-example-session
`)

	got, err := resolveSessionExecInDirs("example.desktop", []string{newDir, oldDir})
	if err != nil {
		t.Fatalf("resolveSessionExecInDirs returned error: %v", err)
	}
	if got != "/run/current-system/sw/bin/start-example-session" {
		t.Fatalf("resolveSessionExecInDirs = %q", got)
	}
}

func TestClearGreeterAutoLoginMemory(t *testing.T) {
	t.Parallel()

	memoryPath := filepath.Join(t.TempDir(), "memory.json")
	writeTestFile(t, memoryPath, `{
  "autoLoginEnabled": true,
  "lastSuccessfulUser": "alice"
}`)

	if err := clearGreeterAutoLoginMemory(memoryPath, ""); err != nil {
		t.Fatalf("clearGreeterAutoLoginMemory returned error: %v", err)
	}

	data, err := os.ReadFile(memoryPath)
	if err != nil {
		t.Fatalf("failed to read memory file: %v", err)
	}
	if strings.Contains(string(data), "autoLoginEnabled") {
		t.Fatalf("expected autoLoginEnabled removed, got: %s", string(data))
	}
	if !strings.Contains(string(data), "lastSuccessfulUser") {
		t.Fatalf("expected other memory fields preserved, got: %s", string(data))
	}
}