use registry for status

cmd/windows_agent/service.go

@@ -46,11 +46,11 @@ func (p *agentService) startPing() {
 		var pingResp PingResp
 		pingErr := agent.ProxmoxHTTPRequest(http.MethodGet, "/api2/json/ping", nil, &pingResp)
 		if pingErr != nil {
-			utils.SetEnvironment("PBS_AGENT_STATUS", fmt.Sprintf("Error - (%s)", pingErr.Error()))
+			agent.SetStatus(fmt.Sprintf("Error - (%s)", pingErr.Error()))
 		} else if !pingResp.Data.Pong {
-			utils.SetEnvironment("PBS_AGENT_STATUS", "Error - server did not return expected data")
+			agent.SetStatus("Error - server did not return expected data")
 		} else {
-			utils.SetEnvironment("PBS_AGENT_STATUS", "Connected")
+			agent.SetStatus("Connected")
 		}
 	}
 
@@ -60,7 +60,7 @@ func (p *agentService) startPing() {
 		retryWait := utils.WaitChan(time.Second * 5)
 		select {
 		case <-p.ctx.Done():
-			utils.SetEnvironment("PBS_AGENT_STATUS", "Agent service is not running")
+			agent.SetStatus("Agent service is not running")
 			return
 		case <-retryWait:
 			ping()
@@ -69,10 +69,10 @@ func (p *agentService) startPing() {
 }
 
 func (p *agentService) run() {
-	utils.SetEnvironment("PBS_AGENT_STATUS", "Starting")
+	agent.SetStatus("Starting")
 	logger, err := syslog.InitializeLogger(p.svc)
 	if err != nil {
-		utils.SetEnvironment("PBS_AGENT_STATUS", fmt.Sprintf("Failed to initialize logger -> %s", err.Error()))
+		agent.SetStatus(fmt.Sprintf("Failed to initialize logger -> %s", err.Error()))
 		return
 	}
 

cmd/windows_agent/systray.go

@@ -16,6 +16,7 @@ import (
 
 	"github.com/getlantern/systray"
 	"github.com/kardianos/service"
+	"github.com/sonroyaalmerol/pbs-plus/internal/agent"
 	"github.com/sonroyaalmerol/pbs-plus/internal/syslog"
 	"github.com/sonroyaalmerol/pbs-plus/internal/utils"
 	"golang.org/x/sys/windows"
@@ -107,8 +108,8 @@ func (p *agentTray) onReady(url string) func() {
 		go func(ctx context.Context, status *systray.MenuItem, url *string) {
 			setStatus := func() {
 				if url != nil && *url != "" {
-					svcStatus, ok := os.LookupEnv("PBS_AGENT_STATUS")
-					if !ok {
+					svcStatus, err := agent.GetStatus()
+					if err != nil {
 						status.SetTitle("Status: Agent service not running")
 					} else {
 						status.SetTitle(fmt.Sprintf("Status: %s", svcStatus))

internal/agent/systray_comm.go

@@ -0,0 +1,26 @@
+package agent
+
+import (
+	"golang.org/x/sys/windows/registry"
+)
+
+func SetStatus(status string) error {
+	key, _, err := registry.CreateKey(registry.LOCAL_MACHINE, `Software\PBSPlus`, registry.ALL_ACCESS)
+	if err == nil {
+		defer key.Close()
+		err := key.SetStringValue("Status", status)
+		return err
+	}
+	return err
+}
+
+func GetStatus() (string, error) {
+	key, err := registry.OpenKey(registry.LOCAL_MACHINE, `Software\PBSPlus`, registry.QUERY_VALUE)
+	if err == nil {
+		defer key.Close()
+		regStatus, _, err := key.GetStringValue("Status")
+		return regStatus, err
+	}
+
+	return "", err
+}