52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
package homedir
|
|
|
|
import (
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/opencontainers/runc/libcontainer/user"
|
|
)
|
|
|
|
// Key returns the env var name for the user's home dir based on
|
|
// the platform being run on
|
|
func Key() string {
|
|
if runtime.GOOS == "windows" {
|
|
return "USERPROFILE"
|
|
}
|
|
return "HOME"
|
|
}
|
|
|
|
// Get returns the home directory of the current user with the help of
|
|
// environment variables depending on the target operating system.
|
|
// Returned path should be used with "path/filepath" to form new paths.
|
|
func Get() string {
|
|
home := os.Getenv(Key())
|
|
if home == "" && runtime.GOOS != "windows" {
|
|
if u, err := user.CurrentUser(); err == nil {
|
|
return u.Home
|
|
}
|
|
}
|
|
return home
|
|
}
|
|
|
|
// GetWithSudoUser returns the home directory of the user who called sudo (if
|
|
// available, retrieved from $SUDO_USER). It fallbacks to Get if any error occurs.
|
|
// Returned path should be used with "path/filepath" to form new paths.
|
|
func GetWithSudoUser() string {
|
|
sudoUser := os.Getenv("SUDO_USER")
|
|
if sudoUser != "" {
|
|
if user, err := user.LookupUser(sudoUser); err == nil {
|
|
return user.Home
|
|
}
|
|
}
|
|
return Get()
|
|
}
|
|
|
|
// GetShortcutString returns the string that is shortcut to user's home directory
|
|
// in the native shell of the platform running on.
|
|
func GetShortcutString() string {
|
|
if runtime.GOOS == "windows" {
|
|
return "%USERPROFILE%" // be careful while using in format functions
|
|
}
|
|
return "~"
|
|
}
|