get user home
This commit is contained in:
parent
ce489a73e7
commit
c5f3942ed7
16
conf/wide.go
16
conf/wide.go
|
@ -161,14 +161,22 @@ func initWide(confPath, confIP, confPort, confServer, confLogLevel, confStaticSe
|
||||||
logger.Debugf("${pwd} [%s]", Wide.WD)
|
logger.Debugf("${pwd} [%s]", Wide.WD)
|
||||||
|
|
||||||
// User Home
|
// User Home
|
||||||
|
userHome := ""
|
||||||
user, err := user.Current()
|
user, err := user.Current()
|
||||||
if nil != err {
|
if nil == err {
|
||||||
logger.Error("Can't get user's home, please report this issue to developer")
|
userHome = user.HomeDir
|
||||||
|
} else {
|
||||||
|
// cross compile support for darwin
|
||||||
|
home, er := util.OS.Home()
|
||||||
|
if nil != er {
|
||||||
|
logger.Error("Can't get user's home, please report this issue to developer", err, er)
|
||||||
|
|
||||||
os.Exit(-1)
|
os.Exit(-1)
|
||||||
|
}
|
||||||
|
|
||||||
|
userHome = home
|
||||||
}
|
}
|
||||||
|
|
||||||
userHome := user.HomeDir
|
|
||||||
logger.Debugf("${user.home} [%s]", userHome)
|
logger.Debugf("${user.home} [%s]", userHome)
|
||||||
|
|
||||||
// Playground Directory
|
// Playground Directory
|
||||||
|
|
|
@ -25,6 +25,7 @@ func TestGetAPIPath(t *testing.T) {
|
||||||
apiPath := Go.GetAPIPath()
|
apiPath := Go.GetAPIPath()
|
||||||
|
|
||||||
v := runtime.Version()[2:]
|
v := runtime.Version()[2:]
|
||||||
|
v = v[:3]
|
||||||
|
|
||||||
verNum, err := strconv.ParseFloat(v, 64)
|
verNum, err := strconv.ParseFloat(v, 64)
|
||||||
|
|
||||||
|
|
52
util/os.go
52
util/os.go
|
@ -15,10 +15,13 @@
|
||||||
package util
|
package util
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
type myos struct{}
|
type myos struct{}
|
||||||
|
@ -38,3 +41,52 @@ func (*myos) Pwd() string {
|
||||||
|
|
||||||
return filepath.Dir(pwd)
|
return filepath.Dir(pwd)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Home 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 (*myos) Home() (string, error) {
|
||||||
|
if OS.IsWindows() {
|
||||||
|
return homeWindows()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Unix-like system, so just assume Unix
|
||||||
|
return homeUnix()
|
||||||
|
}
|
||||||
|
|
||||||
|
func homeUnix() (string, error) {
|
||||||
|
// First prefer the HOME environmental variable
|
||||||
|
if home := os.Getenv("HOME"); home != "" {
|
||||||
|
return home, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// If that fails, try the shell
|
||||||
|
var stdout bytes.Buffer
|
||||||
|
cmd := exec.Command("sh", "-c", "eval echo ~$USER")
|
||||||
|
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 homeWindows() (string, error) {
|
||||||
|
drive := os.Getenv("HOMEDRIVE")
|
||||||
|
path := os.Getenv("HOMEPATH")
|
||||||
|
home := drive + path
|
||||||
|
if drive == "" || path == "" {
|
||||||
|
home = os.Getenv("USERPROFILE")
|
||||||
|
}
|
||||||
|
if home == "" {
|
||||||
|
return "", errors.New("HOMEDRIVE, HOMEPATH, and USERPROFILE are blank")
|
||||||
|
}
|
||||||
|
|
||||||
|
return home, nil
|
||||||
|
}
|
||||||
|
|
|
@ -36,3 +36,14 @@ func TestPwd(t *testing.T) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestHome(t *testing.T) {
|
||||||
|
home, err := OS.Home()
|
||||||
|
if nil != err {
|
||||||
|
t.Error("Can not get user home")
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Log(home)
|
||||||
|
}
|
Loading…
Reference in New Issue