From 42063e5f8850e18019db5656c5688353d45e2896 Mon Sep 17 00:00:00 2001
From: Tulir Asokan <tulir@maunium.net>
Date: Sun, 19 Sep 2021 23:46:33 -0400
Subject: [PATCH] Use xdg-user-dir to find download directory

Fixes #157
Closes #219
---
 main.go | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index a87c87f..4792599 100644
--- a/main.go
+++ b/main.go
@@ -20,6 +20,7 @@ import (
 	"errors"
 	"fmt"
 	"os"
+	"os/exec"
 	"path/filepath"
 	"runtime"
 	"strings"
@@ -128,7 +129,7 @@ func UserDataDir() (dir string, err error) {
 	if dir == "" {
 		dir = os.Getenv("HOME")
 		if dir == "" {
-			return "", errors.New("neither $XDG_CACHE_HOME nor $HOME are defined")
+			return "", errors.New("neither $XDG_DATA_HOME nor $HOME are defined")
 		}
 		dir = filepath.Join(dir, ".local", "share")
 	}
@@ -136,7 +137,24 @@ func UserDataDir() (dir string, err error) {
 	return
 }
 
+func getXDGUserDir(name string) (dir string, err error) {
+	cmd := exec.Command("xdg-user-dir", name)
+	var out strings.Builder
+	cmd.Stdout = &out
+	err = cmd.Run()
+	dir = strings.TrimSpace(out.String())
+	return
+}
+
 func UserDownloadDir() (dir string, err error) {
+	dir = os.Getenv("GOMUKS_DOWNLOAD_HOME")
+	if dir != "" {
+		return
+	}
+	dir, _ = getXDGUserDir("DOWNLOAD")
+	if dir != "" {
+		return
+	}
 	dir, err = os.UserHomeDir()
 	dir = filepath.Join(dir, "Downloads")
 	return