mirror of
https://github.com/google/gops.git
synced 2024-11-24 08:22:25 +02:00
ed67c5b345
On Go ≥ 1.13 the os package has os.ConfigDir (https://golang.org/pkg/os/#UserConfigDir) which returns the default root directory to use for user-specific configuration data. Use this in internal.ConfigDir if available, while still making sure that any directory specified in GOPS_CONFIG_DIR takes precedence.
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
// Copyright 2017 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package internal
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/user"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
)
|
|
|
|
const gopsConfigDirEnvKey = "GOPS_CONFIG_DIR"
|
|
|
|
func ConfigDir() (string, error) {
|
|
if configDir := os.Getenv(gopsConfigDirEnvKey); configDir != "" {
|
|
return configDir, nil
|
|
}
|
|
|
|
if osUserConfigDir := getOSUserConfigDir(); osUserConfigDir != "" {
|
|
return osUserConfigDir, nil
|
|
}
|
|
|
|
if runtime.GOOS == "windows" {
|
|
return filepath.Join(os.Getenv("APPDATA"), "gops"), nil
|
|
}
|
|
|
|
if xdgConfigDir := os.Getenv("XDG_CONFIG_HOME"); xdgConfigDir != "" {
|
|
return filepath.Join(xdgConfigDir, "gops"), nil
|
|
}
|
|
|
|
homeDir := guessUnixHomeDir()
|
|
if homeDir == "" {
|
|
return "", errors.New("unable to get current user home directory: os/user lookup failed; $HOME is empty")
|
|
}
|
|
return filepath.Join(homeDir, ".config", "gops"), nil
|
|
}
|
|
|
|
func guessUnixHomeDir() string {
|
|
usr, err := user.Current()
|
|
if err == nil {
|
|
return usr.HomeDir
|
|
}
|
|
return os.Getenv("HOME")
|
|
}
|
|
|
|
func PIDFile(pid int) (string, error) {
|
|
gopsdir, err := ConfigDir()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return fmt.Sprintf("%s/%d", gopsdir, pid), nil
|
|
}
|
|
|
|
func GetPort(pid int) (string, error) {
|
|
portfile, err := PIDFile(pid)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
b, err := ioutil.ReadFile(portfile)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
port := strings.TrimSpace(string(b))
|
|
return port, nil
|
|
}
|