Add support for XDG_DATA_HOME

samcoe · samcoe · commit e6ad8a10d0bd · 2021-06-02T09:46:14.000-04:00
diff --git a/internal/config/config_file.go b/internal/config/config_file.go
@@ -17,6 +17,7 @@ const (
 	GH_CONFIG_DIR   = "GH_CONFIG_DIR"
 	XDG_CONFIG_HOME = "XDG_CONFIG_HOME"
 	XDG_STATE_HOME  = "XDG_STATE_HOME"
+	XDG_DATA_HOME   = "XDG_DATA_HOME"
 	APP_DATA        = "AppData"
 	LOCAL_APP_DATA  = "LocalAppData"
 )
@@ -70,6 +71,24 @@ func StateDir() string {
 	return path
 }
 
+// Data path precedence
+// 1. XDG_DATA_HOME
+// 2. LocalAppData (windows only)
+// 3. HOME
+func DataDir() string {
+	var path string
+	if a := os.Getenv(XDG_DATA_HOME); a != "" {
+		path = filepath.Join(a, "gh")
+	} else if b := os.Getenv(LOCAL_APP_DATA); runtime.GOOS == "windows" && b != "" {
+		path = filepath.Join(b, "GitHub CLI")
+	} else {
+		c, _ := os.UserHomeDir()
+		path = filepath.Join(c, ".local", "share", "gh")
+	}
+
+	return path
+}
+
 var errSamePath = errors.New("same path")
 var errNotExist = errors.New("not exist")
 
diff --git a/internal/config/config_file_test.go b/internal/config/config_file_test.go
@@ -484,3 +484,68 @@ func Test_autoMigrateStateDir_migration(t *testing.T) {
 	assert.Equal(t, 1, len(files))
 	assert.Equal(t, "state.yml", files[0].Name())
 }
+
+func Test_DataDir(t *testing.T) {
+	tempDir := t.TempDir()
+
+	tests := []struct {
+		name        string
+		onlyWindows bool
+		env         map[string]string
+		output      string
+	}{
+		{
+			name: "HOME/USERPROFILE specified",
+			env: map[string]string{
+				"XDG_DATA_HOME":   "",
+				"GH_CONFIG_DIR":   "",
+				"XDG_CONFIG_HOME": "",
+				"LocalAppData":    "",
+				"USERPROFILE":     tempDir,
+				"HOME":            tempDir,
+			},
+			output: filepath.Join(tempDir, ".local", "share", "gh"),
+		},
+		{
+			name: "XDG_DATA_HOME specified",
+			env: map[string]string{
+				"XDG_DATA_HOME": tempDir,
+			},
+			output: filepath.Join(tempDir, "gh"),
+		},
+		{
+			name:        "LocalAppData specified",
+			onlyWindows: true,
+			env: map[string]string{
+				"LocalAppData": tempDir,
+			},
+			output: filepath.Join(tempDir, "GitHub CLI"),
+		},
+		{
+			name:        "XDG_DATA_HOME and LocalAppData specified",
+			onlyWindows: true,
+			env: map[string]string{
+				"XDG_DATA_HOME": tempDir,
+				"LocalAppData":  tempDir,
+			},
+			output: filepath.Join(tempDir, "gh"),
+		},
+	}
+
+	for _, tt := range tests {
+		if tt.onlyWindows && runtime.GOOS != "windows" {
+			continue
+		}
+		t.Run(tt.name, func(t *testing.T) {
+			if tt.env != nil {
+				for k, v := range tt.env {
+					old := os.Getenv(k)
+					os.Setenv(k, v)
+					defer os.Setenv(k, old)
+				}
+			}
+
+			assert.Equal(t, tt.output, DataDir())
+		})
+	}
+}
