136 lines
3.0 KiB
Go
136 lines
3.0 KiB
Go
package main
|
|
|
|
// functions to import and export the protobuf
|
|
// data to and from config files
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"go.wit.com/log"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func (m *Portmaps) ConfigSave() error {
|
|
if m == nil {
|
|
return fmt.Errorf("ConfigSave() nil")
|
|
}
|
|
s := m.FormatTEXT()
|
|
log.Info("proto.Marshal() worked len", len(s))
|
|
configWrite([]byte(s))
|
|
return nil
|
|
}
|
|
|
|
func ConfigLoad() *Portmaps {
|
|
if os.Getenv("CLOUD_HOME") == "" {
|
|
homeDir, _ := os.UserHomeDir()
|
|
fullpath := filepath.Join(homeDir, ".config/cloud")
|
|
os.Setenv("CLOUD_HOME", fullpath)
|
|
}
|
|
|
|
var data []byte
|
|
var err error
|
|
if data, err = loadFile("gus.text"); err != nil {
|
|
log.Warn("gus.text failed to load", err)
|
|
// something went wrong loading the file
|
|
return nil
|
|
}
|
|
|
|
if data == nil {
|
|
return nil
|
|
}
|
|
p := new(Portmaps)
|
|
if err = p.UnmarshalTEXT(data); err != nil {
|
|
log.Warn("unmarshal failed on gus.text config file", err)
|
|
return nil
|
|
}
|
|
|
|
log.Log(INFO, "gus.ConfigLoad() has", p.Len(), "port mappings")
|
|
return p
|
|
}
|
|
|
|
func (m *Portmaps) ConfigLoad() error {
|
|
if m == nil {
|
|
return errors.New("It's not safe to run ConfigLoad() on a nil ?")
|
|
}
|
|
if os.Getenv("CLOUD_HOME") == "" {
|
|
homeDir, _ := os.UserHomeDir()
|
|
fullpath := filepath.Join(homeDir, ".config/cloud")
|
|
os.Setenv("CLOUD_HOME", fullpath)
|
|
}
|
|
|
|
var data []byte
|
|
var err error
|
|
if data, err = loadFile("gus.text"); err != nil {
|
|
// something went wrong loading the file
|
|
return err
|
|
}
|
|
|
|
if data != nil {
|
|
if err = proto.Unmarshal(data, m); err != nil {
|
|
log.Warn("broken gus.text config file", "gus.text")
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
log.Log(INFO, "gus.ConfigLoad() has", m.Len(), "port mappings")
|
|
return nil
|
|
}
|
|
|
|
func loadFile(filename string) ([]byte, error) {
|
|
homeDir, err := os.UserHomeDir()
|
|
p := filepath.Join(homeDir, ".config/cloud")
|
|
fullname := filepath.Join(p, filename)
|
|
data, err := os.ReadFile(fullname)
|
|
if errors.Is(err, os.ErrNotExist) {
|
|
// if file does not exist, just return nil. this
|
|
return nil, nil
|
|
}
|
|
if err != nil {
|
|
// log.Info("open config file :", err)
|
|
return nil, err
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
func (m *Portmaps) loadFile(fname string) ([]byte, error) {
|
|
fullname := filepath.Join(os.Getenv("CLOUD_HOME"), fname)
|
|
|
|
data, err := os.ReadFile(fullname)
|
|
if err != nil {
|
|
// log.Info("open config file :", err)
|
|
return nil, err
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
func configWrite(data []byte) error {
|
|
homeDir, err := os.UserHomeDir()
|
|
p := filepath.Join(homeDir, ".config/cloud")
|
|
fname := filepath.Join(p, "gus.text")
|
|
cfgfile, err := os.OpenFile(fname, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
|
|
defer cfgfile.Close()
|
|
if err != nil {
|
|
log.Warn("open config file :", err)
|
|
return err
|
|
}
|
|
cfgfile.Write(data)
|
|
return nil
|
|
}
|
|
|
|
func (m *Portmaps) configWrite(fname string, data []byte) error {
|
|
fullname := filepath.Join(os.Getenv("CLOUD_HOME"), fname)
|
|
|
|
cfgfile, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
|
|
defer cfgfile.Close()
|
|
if err != nil {
|
|
log.Warn("open config file :", err)
|
|
return err
|
|
}
|
|
cfgfile.Write(data)
|
|
return nil
|
|
}
|