mirror of https://github.com/k3d-io/k3d
iwilltry42
5 years ago
parent
11b937a910
commit
84234e7a98
@ -0,0 +1,61 @@ |
||||
/* |
||||
Copyright © 2019 Thorsten Klein <iwilltry42@gmail.com> |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in |
||||
all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
*/ |
||||
package util |
||||
|
||||
import ( |
||||
"os" |
||||
"path" |
||||
|
||||
homedir "github.com/mitchellh/go-homedir" |
||||
log "github.com/sirupsen/logrus" |
||||
) |
||||
|
||||
// GetConfigDirOrCreate will return the base path of the k3d config directory or create it if it doesn't exist yet
|
||||
// k3d's config directory will be $HOME/.k3d (Unix)
|
||||
func GetConfigDirOrCreate() (string, error) { |
||||
|
||||
// build the path
|
||||
homeDir, err := homedir.Dir() |
||||
if err != nil { |
||||
log.Errorln("Failed to get user's home directory") |
||||
return "", err |
||||
} |
||||
configDir := path.Join(homeDir, ".k3d") |
||||
|
||||
// create directories if necessary
|
||||
if err := createDirIfNotExists(configDir); err != nil { |
||||
log.Errorln("Failed to create config path '%s'", configDir) |
||||
return "", err |
||||
} |
||||
|
||||
return configDir, nil |
||||
|
||||
} |
||||
|
||||
// createDirIfNotExists checks for the existence of a directory and creates it along with all required parents if not.
|
||||
// It returns an error if the directory (or parents) couldn't be created and nil if it worked fine or if the path already exists.
|
||||
func createDirIfNotExists(path string) error { |
||||
if _, err := os.Stat(path); os.IsNotExist(err) { |
||||
return os.MkdirAll(path, os.ModePerm) |
||||
} |
||||
return nil |
||||
} |
||||
@ -0,0 +1,21 @@ |
||||
The MIT License (MIT) |
||||
|
||||
Copyright (c) 2013 Mitchell Hashimoto |
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
||||
of this software and associated documentation files (the "Software"), to deal |
||||
in the Software without restriction, including without limitation the rights |
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
||||
copies of the Software, and to permit persons to whom the Software is |
||||
furnished to do so, subject to the following conditions: |
||||
|
||||
The above copyright notice and this permission notice shall be included in |
||||
all copies or substantial portions of the Software. |
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
||||
THE SOFTWARE. |
||||
@ -0,0 +1,14 @@ |
||||
# go-homedir |
||||
|
||||
This is a Go library for detecting the user's home directory without |
||||
the use of cgo, so the library can be used in cross-compilation environments. |
||||
|
||||
Usage is incredibly simple, just call `homedir.Dir()` to get the home directory |
||||
for a user, and `homedir.Expand()` to expand the `~` in a path to the home |
||||
directory. |
||||
|
||||
**Why not just use `os/user`?** The built-in `os/user` package requires |
||||
cgo on Darwin systems. This means that any Go code that uses that package |
||||
cannot cross compile. But 99% of the time the use for `os/user` is just to |
||||
retrieve the home directory, which we can do for the current user without |
||||
cgo. This library does that, enabling cross-compilation. |
||||
@ -0,0 +1 @@ |
||||
module github.com/mitchellh/go-homedir |
||||
@ -0,0 +1,167 @@ |
||||
package homedir |
||||
|
||||
import ( |
||||
"bytes" |
||||
"errors" |
||||
"os" |
||||
"os/exec" |
||||
"path/filepath" |
||||
"runtime" |
||||
"strconv" |
||||
"strings" |
||||
"sync" |
||||
) |
||||
|
||||
// DisableCache will disable caching of the home directory. Caching is enabled
|
||||
// by default.
|
||||
var DisableCache bool |
||||
|
||||
var homedirCache string |
||||
var cacheLock sync.RWMutex |
||||
|
||||
// Dir returns the home directory for the executing user.
|
||||
//
|
||||
// This uses an OS-specific method for discovering the home directory.
|
||||
// An error is returned if a home directory cannot be detected.
|
||||
func Dir() (string, error) { |
||||
if !DisableCache { |
||||
cacheLock.RLock() |
||||
cached := homedirCache |
||||
cacheLock.RUnlock() |
||||
if cached != "" { |
||||
return cached, nil |
||||
} |
||||
} |
||||
|
||||
cacheLock.Lock() |
||||
defer cacheLock.Unlock() |
||||
|
||||
var result string |
||||
var err error |
||||
if runtime.GOOS == "windows" { |
||||
result, err = dirWindows() |
||||
} else { |
||||
// Unix-like system, so just assume Unix
|
||||
result, err = dirUnix() |
||||
} |
||||
|
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
homedirCache = result |
||||
return result, nil |
||||
} |
||||
|
||||
// Expand expands the path to include the home directory if the path
|
||||
// is prefixed with `~`. If it isn't prefixed with `~`, the path is
|
||||
// returned as-is.
|
||||
func Expand(path string) (string, error) { |
||||
if len(path) == 0 { |
||||
return path, nil |
||||
} |
||||
|
||||
if path[0] != '~' { |
||||
return path, nil |
||||
} |
||||
|
||||
if len(path) > 1 && path[1] != '/' && path[1] != '\\' { |
||||
return "", errors.New("cannot expand user-specific home dir") |
||||
} |
||||
|
||||
dir, err := Dir() |
||||
if err != nil { |
||||
return "", err |
||||
} |
||||
|
||||
return filepath.Join(dir, path[1:]), nil |
||||
} |
||||
|
||||
// Reset clears the cache, forcing the next call to Dir to re-detect
|
||||
// the home directory. This generally never has to be called, but can be
|
||||
// useful in tests if you're modifying the home directory via the HOME
|
||||
// env var or something.
|
||||
func Reset() { |
||||
cacheLock.Lock() |
||||
defer cacheLock.Unlock() |
||||
homedirCache = "" |
||||
} |
||||
|
||||
func dirUnix() (string, error) { |
||||
homeEnv := "HOME" |
||||
if runtime.GOOS == "plan9" { |
||||
// On plan9, env vars are lowercase.
|
||||
homeEnv = "home" |
||||
} |
||||
|
||||
// First prefer the HOME environmental variable
|
||||
if home := os.Getenv(homeEnv); home != "" { |
||||
return home, nil |
||||
} |
||||
|
||||
var stdout bytes.Buffer |
||||
|
||||
// If that fails, try OS specific commands
|
||||
if runtime.GOOS == "darwin" { |
||||
cmd := exec.Command("sh", "-c", `dscl -q . -read /Users/"$(whoami)" NFSHomeDirectory | sed 's/^[^ ]*: //'`) |
||||
cmd.Stdout = &stdout |
||||
if err := cmd.Run(); err == nil { |
||||
result := strings.TrimSpace(stdout.String()) |
||||
if result != "" { |
||||
return result, nil |
||||
} |
||||
} |
||||
} else { |
||||
cmd := exec.Command("getent", "passwd", strconv.Itoa(os.Getuid())) |
||||
cmd.Stdout = &stdout |
||||
if err := cmd.Run(); err != nil { |
||||
// If the error is ErrNotFound, we ignore it. Otherwise, return it.
|
||||
if err != exec.ErrNotFound { |
||||
return "", err |
||||
} |
||||
} else { |
||||
if passwd := strings.TrimSpace(stdout.String()); passwd != "" { |
||||
// username:password:uid:gid:gecos:home:shell
|
||||
passwdParts := strings.SplitN(passwd, ":", 7) |
||||
if len(passwdParts) > 5 { |
||||
return passwdParts[5], nil |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
// If all else fails, try the shell
|
||||
stdout.Reset() |
||||
cmd := exec.Command("sh", "-c", "cd && pwd") |
||||
cmd.Stdout = &stdout |
||||
if err := cmd.Run(); err != nil { |
||||
return "", err |
||||
} |
||||
|
||||
result := strings.TrimSpace(stdout.String()) |
||||
if result == "" { |
||||
return "", errors.New("blank output when reading home directory") |
||||
} |
||||
|
||||
return result, nil |
||||
} |
||||
|
||||
func dirWindows() (string, error) { |
||||
// First prefer the HOME environmental variable
|
||||
if home := os.Getenv("HOME"); home != "" { |
||||
return home, nil |
||||
} |
||||
|
||||
// Prefer standard environment variable USERPROFILE
|
||||
if home := os.Getenv("USERPROFILE"); home != "" { |
||||
return home, nil |
||||
} |
||||
|
||||
drive := os.Getenv("HOMEDRIVE") |
||||
path := os.Getenv("HOMEPATH") |
||||
home := drive + path |
||||
if drive == "" || path == "" { |
||||
return "", errors.New("HOMEDRIVE, HOMEPATH, or USERPROFILE are blank") |
||||
} |
||||
|
||||
return home, nil |
||||
} |
||||
Loading…
Reference in new issue