chore: respect XDG_CACHE_HOME (#32)

TimothyYe · Nov 1, 2023 · 81ca28e · 81ca28e
1 parent de25f01
commit 81ca28e
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/lib/cache.go b/lib/cache.go
@@ -106,17 +106,27 @@ func ClearCahceFiles() {
 
 func getDictDir() string {
 	dbDir := os.Getenv("YDICT_DB")
+	xdgCacheDir := os.Getenv("XDG_CACHE_HOME")
 	if dbDir == "" {
-		dbDir = filepath.Join(os.Getenv("HOME"), ".ydict")
+		if xdgCacheDir != "" {
+			dbDir = filepath.Join(xdgCacheDir, "ydict")
+		} else {
+			dbDir = filepath.Join(os.Getenv("HOME"), ".cache/ydict")
+		}
 	}
 
 	return dbDir
 }
 
 func getDictDBDir() string {
 	dbDir := os.Getenv("YDICT_DB")
+	xdgCacheDir := os.Getenv("XDG_CACHE_HOME")
 	if dbDir == "" {
-		dbDir = filepath.Join(os.Getenv("HOME"), ".ydict")
+		if xdgCacheDir != "" {
+			dbDir = filepath.Join(xdgCacheDir, "ydict")
+		} else {
+			dbDir = filepath.Join(os.Getenv("HOME"), ".cache/ydict")
+		}
 	}
 
 	ydictDir := filepath.Join(dbDir, "db")