fix: respect ZDOTDIR for zsh init

Author: brosu

README.md

@@ -105,7 +105,7 @@ After running `wt init`, restart your shell or run:
 
 ```bash
 source ~/.bashrc   # for bash
-source ~/.zshrc    # for zsh
+source "${ZDOTDIR:-$HOME}/.zshrc"    # for zsh
 ```
 
 Shell integration enables:

init.go

@@ -32,7 +32,7 @@ var initCmd = &cobra.Command{
 
 Automatically detects your shell and updates the appropriate config file:
   - bash: ~/.bashrc
-  - zsh:  ~/.zshrc
+  - zsh:  ~/.zshrc (or $ZDOTDIR/.zshrc if ZDOTDIR is set)
   - powershell: $PROFILE (Windows only)
 
 The configuration is wrapped in markers so it can be safely updated or removed.
@@ -135,6 +135,14 @@ func getShellConfigPath(shell string) string {
 		}
 		return bashrc
 	case "zsh":
+		// Respect ZDOTDIR if set: zsh reads its startup files from $ZDOTDIR (default: $HOME).
+		// This avoids writing to the wrong (potentially unused or broken) ~/.zshrc.
+		if zdotdir := strings.TrimSpace(os.Getenv("ZDOTDIR")); zdotdir != "" {
+			if !filepath.IsAbs(zdotdir) {
+				zdotdir = filepath.Join(home, zdotdir)
+			}
+			return filepath.Join(filepath.Clean(zdotdir), ".zshrc")
+		}
 		return filepath.Join(home, ".zshrc")
 	case "powershell":
 		// Check $PROFILE env var first (works for both Windows PowerShell 5.1 and PowerShell Core)
@@ -258,9 +266,9 @@ func installShellConfig(configPath, shell string, dryRun, noPrompt bool) error {
 		fmt.Println("To activate, run:")
 		switch shell {
 		case "bash":
-			fmt.Println("  source ~/.bashrc")
+			fmt.Printf("  source %s\n", configPath)
 		case "zsh":
-			fmt.Println("  source ~/.zshrc")
+			fmt.Printf("  source %s\n", configPath)
 		case "powershell":
 			fmt.Println("  . $PROFILE")
 		}

init_test.go

@@ -97,6 +97,11 @@ func TestGetShellConfigPath(t *testing.T) {
 		t.Fatalf("Failed to get home dir: %v", err)
 	}
 
+	// Ensure tests are stable even when the developer machine has ZDOTDIR set.
+	origZdotdir := os.Getenv("ZDOTDIR")
+	os.Setenv("ZDOTDIR", "")
+	t.Cleanup(func() { os.Setenv("ZDOTDIR", origZdotdir) })
+
 	tests := []struct {
 		name  string
 		shell string
@@ -119,6 +124,25 @@ func TestGetShellConfigPath(t *testing.T) {
 	}
 }
 
+func TestGetShellConfigPathZshRespectsZdotdir(t *testing.T) {
+	home, err := os.UserHomeDir()
+	if err != nil {
+		t.Fatalf("Failed to get home dir: %v", err)
+	}
+
+	orig := os.Getenv("ZDOTDIR")
+	t.Cleanup(func() { os.Setenv("ZDOTDIR", orig) })
+
+	zdotdir := filepath.Join(home, ".config", "zsh")
+	os.Setenv("ZDOTDIR", zdotdir)
+
+	got := getShellConfigPath("zsh")
+	want := filepath.Join(zdotdir, ".zshrc")
+	if got != want {
+		t.Fatalf("getShellConfigPath(%q) = %q, want %q", "zsh", got, want)
+	}
+}
+
 func TestGetShellConfigContent(t *testing.T) {
 	tests := []struct {
 		name     string

scripts/postinstall.sh

@@ -33,6 +33,6 @@ echo "wt has been installed successfully!"
 echo ""
 echo "To activate shell integration in your current session:"
 echo "  source ~/.bashrc   # for bash"
-echo "  source ~/.zshrc    # for zsh"
+echo "  source \"${ZDOTDIR:-$HOME}/.zshrc\"    # for zsh"
 echo ""
 echo "Or simply start a new terminal session."