Fix service count toast (#630)

Notifiarr · Nov 4, 2023 · 344ae09 · 344ae09
2 parents 7509798 + 9218aa2
commit 344ae09
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/pkg/client/tray.go b/pkg/client/tray.go
@@ -272,7 +272,7 @@ func (c *Client) notifiarrMenuActions() {
 	menu["syncqp"].Click(func() { c.triggers.CFSync.SyncSonarrRP(website.EventUser) })
 	menu["svcs_prod"].Click(func() {
 		c.Print("[user requested] Checking services and sending results to Notifiarr.")
-		ui.Notify("Running and sending %d Service Checks.", len(c.Config.Service))
+		ui.Notify("Running and sending %d Service Checks.", c.Config.Services.SvcCount())
 		c.Config.Services.RunChecks(website.EventUser)
 	})
 	menu["plex_prod"].Click(func() { c.triggers.PlexCron.Send(website.EventUser) })
@@ -323,7 +323,7 @@ func (c *Client) debugMenu() {
 	menu["svcs_log"] = debug.AddSubMenuItem("Log Service Checks", "check all services and log results")
 	menu["svcs_log"].Click(func() {
 		c.Print("[user requested] Checking services and logging results.")
-		ui.Notify("Running and logging %d Service Checks.", len(c.Config.Service))
+		ui.Notify("Running and logging %d Service Checks.", c.Config.Services.SvcCount())
 		c.Config.Services.RunChecks("log")
 	})
 

diff --git a/pkg/services/services.go b/pkg/services/services.go
@@ -234,3 +234,8 @@ func (c *Config) Stop() {
 	c.done = nil
 	c.stopChan = nil
 }
+
+// SvcCount returns the count of services being monitored.
+func (c *Config) SvcCount() int {
+	return len(c.services)
+}