feat(system/upgrade): add upgrade banner [EE-4564] (#8046)

api/internal/nodes/nodes.go

@@ -0,0 +1,35 @@
+package status
+
+import (
+	portainer "github.com/portainer/portainer/api"
+)
+
+// NodesCount returns the total node number of all environments
+func NodesCount(endpoints []portainer.Endpoint) int {
+	nodes := 0
+	for _, env := range endpoints {
+		nodes += countNodes(&env)
+	}
+
+	return nodes
+}
+
+func countNodes(endpoint *portainer.Endpoint) int {
+	if len(endpoint.Snapshots) == 1 {
+		return max(endpoint.Snapshots[0].NodeCount, 1)
+	}
+
+	if len(endpoint.Kubernetes.Snapshots) == 1 {
+		return max(endpoint.Kubernetes.Snapshots[0].NodeCount, 1)
+	}
+
+	return 1
+}
+
+func max(a, b int) int {
+	if a > b {
+		return a
+	}
+
+	return b
+}