mirror of
https://github.com/zu1k/nali.git
synced 2025-02-03 02:42:44 +08:00
34 lines
611 B
Go
34 lines
611 B
Go
package constant
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
var (
|
|
// WorkDirPath database home path
|
|
WorkDirPath string
|
|
)
|
|
|
|
func init() {
|
|
WorkDirPath = os.Getenv("NALI_HOME")
|
|
if WorkDirPath == "" {
|
|
WorkDirPath = os.Getenv("NALI_DB_HOME")
|
|
}
|
|
if WorkDirPath == "" {
|
|
homeDir, err := os.UserHomeDir()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
WorkDirPath = filepath.Join(homeDir, ".nali")
|
|
}
|
|
if _, err := os.Stat(WorkDirPath); os.IsNotExist(err) {
|
|
if err := os.MkdirAll(WorkDirPath, 0777); err != nil {
|
|
log.Fatal("can not create", WorkDirPath, ", use bin dir instead")
|
|
}
|
|
}
|
|
|
|
_ = os.Chdir(WorkDirPath)
|
|
}
|