load and save PB() funcsv0.0.5

1 files changed, 31 insertions, 0 deletions

diff --git a/load.go b/load.go
index 427ae5e..1df5665 100644
--- a/load.go
+++ b/load.go
@@ -53,6 +53,37 @@ func ConfigLoad(pb proto.Message, argname string, protoname string) error {
 	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 {
 	var data []byte
 	var err error