load and save PB() funcs
This commit is contained in:
parent
0e805acc67
commit
77f7f39c25
31
load.go
31
load.go
|
@ -53,6 +53,37 @@ func ConfigLoad(pb proto.Message, argname string, protoname string) error {
|
||||||
return log.Errorf("could not load config file")
|
return log.Errorf("could not load config file")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func LoadPB(pb proto.Message, argname string, protoname string) (string, error) {
|
||||||
|
var fullname string
|
||||||
|
if strings.HasPrefix(argname, "/") {
|
||||||
|
fullname = filepath.Join(argname, protoname+".pb")
|
||||||
|
} else {
|
||||||
|
homeDir, err := os.UserHomeDir()
|
||||||
|
if err != nil {
|
||||||
|
log.Warn("ConfigLoad() UserHomeDir() err", err)
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
fullname = filepath.Join(homeDir, ".config", argname, protoname+".pb")
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := loadFile(fullname)
|
||||||
|
if err != nil {
|
||||||
|
log.Warn("LoadPB()", fullname, err)
|
||||||
|
// set pb.Filename that was attempted
|
||||||
|
return fullname, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Unmarshal()
|
||||||
|
if err = proto.Unmarshal(data, pb); err != nil {
|
||||||
|
log.Warn("LoadPB() file", fullname)
|
||||||
|
log.Warn("LoadPB() Unmarshal() err", err)
|
||||||
|
return fullname, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return fullname, nil
|
||||||
|
}
|
||||||
|
|
||||||
func loadTEXT(pb proto.Message, fullname string) error {
|
func loadTEXT(pb proto.Message, fullname string) error {
|
||||||
var data []byte
|
var data []byte
|
||||||
var err error
|
var err error
|
||||||
|
|
23
save.go
23
save.go
|
@ -19,6 +19,29 @@ func ConfigSave(pb proto.Message) error {
|
||||||
return saveTEXT(pb, "")
|
return saveTEXT(pb, "")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func SavePB(pb proto.Message, fullname string) error {
|
||||||
|
if !strings.HasSuffix(fullname, ".pb") {
|
||||||
|
// todo: append .text here?
|
||||||
|
return log.Errorf("%s needs to end in '.pb'", fullname)
|
||||||
|
}
|
||||||
|
|
||||||
|
dir, name := filepath.Split(fullname)
|
||||||
|
if name == "" {
|
||||||
|
return fmt.Errorf("filename was blank")
|
||||||
|
}
|
||||||
|
if err := os.MkdirAll(dir, os.ModePerm); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := proto.Marshal(pb)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Infof("ConfigSave() filename=%s %d\n", fullname, len(data))
|
||||||
|
return configWrite(fullname, data)
|
||||||
|
}
|
||||||
|
|
||||||
func ConfigSaveWithHeader(pb proto.Message, header string) error {
|
func ConfigSaveWithHeader(pb proto.Message, header string) error {
|
||||||
var final error
|
var final error
|
||||||
if err := saveTEXT(pb, header); err != nil {
|
if err := saveTEXT(pb, header); err != nil {
|
||||||
|
|
Loading…
Reference in New Issue