137 lines
2.8 KiB
Go
137 lines
2.8 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/user"
|
|
"path"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/kardianos/osext"
|
|
)
|
|
|
|
// MakeName creates a node name that follows the ethereum convention
|
|
// for such names. It adds the operation system name and Go runtime version
|
|
// the name.
|
|
func MakeName(name, version string) string {
|
|
return fmt.Sprintf("%s/v%s/%s/%s", name, version, runtime.GOOS, runtime.Version())
|
|
}
|
|
|
|
func ExpandHomePath(p string) (path string) {
|
|
path = p
|
|
|
|
// Check in case of paths like "/something/~/something/"
|
|
if len(path) > 1 && path[:2] == "~/" {
|
|
usr, _ := user.Current()
|
|
dir := usr.HomeDir
|
|
|
|
path = strings.Replace(p, "~", dir, 1)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func FileExist(filePath string) bool {
|
|
_, err := os.Stat(filePath)
|
|
if err != nil && os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func ReadAllFile(filePath string) (string, error) {
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
data, err := ioutil.ReadAll(file)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return string(data), nil
|
|
}
|
|
|
|
func WriteFile(filePath string, content []byte) error {
|
|
fh, err := os.OpenFile(filePath, os.O_TRUNC|os.O_RDWR|os.O_CREATE, os.ModePerm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer fh.Close()
|
|
|
|
_, err = fh.Write(content)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func AbsolutePath(Datadir string, filename string) string {
|
|
if path.IsAbs(filename) {
|
|
return filename
|
|
}
|
|
return path.Join(Datadir, filename)
|
|
}
|
|
|
|
func DefaultAssetPath() string {
|
|
var assetPath string
|
|
pwd, _ := os.Getwd()
|
|
srcdir := path.Join(os.Getenv("GOPATH"), "src", "github.com", "ethereum", "go-ethereum", "cmd", "mist")
|
|
|
|
// If the current working directory is the go-ethereum dir
|
|
// assume a debug build and use the source directory as
|
|
// asset directory.
|
|
if pwd == srcdir {
|
|
assetPath = path.Join(pwd, "assets")
|
|
} else {
|
|
switch runtime.GOOS {
|
|
case "darwin":
|
|
// Get Binary Directory
|
|
exedir, _ := osext.ExecutableFolder()
|
|
assetPath = filepath.Join(exedir, "../Resources")
|
|
case "linux":
|
|
assetPath = "/usr/share/mist"
|
|
case "windows":
|
|
assetPath = "./assets"
|
|
default:
|
|
assetPath = "."
|
|
}
|
|
}
|
|
|
|
// Check if the assetPath exists. If not, try the source directory
|
|
// This happens when binary is run from outside cmd/mist directory
|
|
if _, err := os.Stat(assetPath); os.IsNotExist(err) {
|
|
assetPath = path.Join(srcdir, "assets")
|
|
}
|
|
|
|
return assetPath
|
|
}
|
|
|
|
func DefaultDataDir() string {
|
|
usr, _ := user.Current()
|
|
if runtime.GOOS == "darwin" {
|
|
return path.Join(usr.HomeDir, "Library/Ethereum")
|
|
} else if runtime.GOOS == "windows" {
|
|
return path.Join(usr.HomeDir, "AppData/Roaming/Ethereum")
|
|
} else {
|
|
return path.Join(usr.HomeDir, ".ethereum")
|
|
}
|
|
}
|
|
|
|
func IsWindows() bool {
|
|
return runtime.GOOS == "windows"
|
|
}
|
|
|
|
func WindonizePath(path string) string {
|
|
if string(path[0]) == "/" && IsWindows() {
|
|
path = path[1:]
|
|
}
|
|
return path
|
|
}
|