parent
a3ea303ab8
commit
62d406e0de
3
Makefile
3
Makefile
|
@ -30,6 +30,9 @@ curl-uptime:
|
|||
curl-droplets:
|
||||
curl --silent http://localhost:8080/droplets
|
||||
|
||||
curl-writeconfig:
|
||||
curl --silent http://localhost:8080/writeconfig
|
||||
|
||||
# this is for release builds using the go.mod files
|
||||
release-build:
|
||||
@echo ${REDOMOD}
|
||||
|
|
|
@ -12,8 +12,7 @@ import (
|
|||
|
||||
func readConfigFile() {
|
||||
me.cluster = new(pb.Cluster)
|
||||
homeDir, _ := os.UserHomeDir()
|
||||
fullname := filepath.Join(homeDir, ".config/virtigo.json")
|
||||
fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "virtigo.json")
|
||||
pfile, err := os.ReadFile(fullname)
|
||||
if err != nil {
|
||||
log.Info("open config file :", err)
|
||||
|
@ -57,8 +56,7 @@ func readConfigFile() {
|
|||
}
|
||||
|
||||
func writeConfigFile() {
|
||||
homeDir, _ := os.UserHomeDir()
|
||||
fullname := filepath.Join(homeDir, ".config/virtigo.json")
|
||||
fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "virtigo.json")
|
||||
cfgfile, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE, 0666)
|
||||
defer cfgfile.Close()
|
||||
if err != nil {
|
||||
|
@ -67,4 +65,19 @@ func writeConfigFile() {
|
|||
}
|
||||
json := me.cluster.FormatJSON()
|
||||
fmt.Fprintln(cfgfile, json)
|
||||
log.Info("Write:", fullname, "OK")
|
||||
}
|
||||
|
||||
func writeConfigFileDroplets() {
|
||||
fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "droplets.text")
|
||||
cfgfile, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE, 0666)
|
||||
defer cfgfile.Close()
|
||||
if err != nil {
|
||||
log.Info("open config file :", err)
|
||||
return
|
||||
}
|
||||
// text := me.cluster.Droplets.FormatTEXT()
|
||||
text := me.cluster.FormatTEXT()
|
||||
fmt.Fprintln(cfgfile, text)
|
||||
log.Info("Write:", fullname, "OK")
|
||||
}
|
||||
|
|
1
http.go
1
http.go
|
@ -58,6 +58,7 @@ func okHandler(w http.ResponseWriter, r *http.Request) {
|
|||
|
||||
if tmp == "/writeconfig" {
|
||||
writeConfigFile()
|
||||
writeConfigFileDroplets()
|
||||
fmt.Fprintln(w, "OK")
|
||||
return
|
||||
}
|
||||
|
|
6
main.go
6
main.go
|
@ -6,6 +6,7 @@ import (
|
|||
"embed"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
"go.wit.com/dev/alexflint/arg"
|
||||
|
@ -18,6 +19,11 @@ var Version string
|
|||
var resources embed.FS
|
||||
|
||||
func main() {
|
||||
if os.Getenv("VIRTIGO_HOME") == "" {
|
||||
homeDir, _ := os.UserHomeDir()
|
||||
fullpath := filepath.Join(homeDir, ".config/virtigo")
|
||||
os.Setenv("VIRTIGO_HOME", fullpath)
|
||||
}
|
||||
pp := arg.MustParse(&argv)
|
||||
|
||||
if !argv.Uptime {
|
||||
|
|
Loading…
Reference in New Issue