You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
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"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/user"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
const gopsConfigDirEnvKey = "GOPS_CONFIG_DIR"
|
|
|
|
func ConfigDir() (string, error) {
|
|
if configDir := os.Getenv(gopsConfigDirEnvKey); configDir != "" {
|
|
return configDir, nil
|
|
}
|
|
|
|
if userConfigDir, err := os.UserConfigDir(); err == nil {
|
|
return filepath.Join(userConfigDir, "gops"), 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 filepath.Join(gopsdir, strconv.Itoa(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
|
|
}
|