more on Machine.Init()

1 files changed, 22 insertions, 0 deletions

diff --git a/init.go b/init.go
index 94c3cf0..2da1427 100644
--- a/init.go
+++ b/init.go
@@ -6,6 +6,7 @@ import (
 	"os"
 	"os/user"
 	"path/filepath"
+	"strings"
 	"time"
 
 	"go.wit.com/lib/gui/shell"
@@ -133,6 +134,27 @@ func (f *Forge) InitPB() {
 	}
 }
 
+func (f *Forge) InitMachine() {
+	f.Machine = new(zoopb.Machine)
+	if err := f.Machine.ConfigLoad(); err != nil {
+		log.Log(WARN, "zoopb.ConfigLoad() failed", err)
+		f.Machine.InitWit()
+	}
+
+	if f.Config.Username == "" {
+		usr, _ := user.Current()
+		f.Config.Username = usr.Username
+	}
+
+	if f.Machine.Hostname == "" {
+		r, err := shell.RunVerbose([]string{"hostname", "-f"})
+		if err == nil {
+			tmp := strings.Join(r.Stdout, "\n")
+			f.Machine.Hostname = strings.TrimSpace(tmp)
+		}
+	}
+}
+
 // only init's the protobuf. intended to not scan or change anything
 func InitPB() *Forge {
 	f := DetermineGoPath()