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

api/http/handler/status/handler.go

@@ -6,6 +6,7 @@ import (
 	"github.com/gorilla/mux"
 	httperror "github.com/portainer/libhttp/error"
 	portainer "github.com/portainer/portainer/api"
+	"github.com/portainer/portainer/api/dataservices"
 	"github.com/portainer/portainer/api/demo"
 	"github.com/portainer/portainer/api/http/security"
 )
@@ -13,21 +14,28 @@ import (
 // Handler is the HTTP handler used to handle status operations.
 type Handler struct {
 	*mux.Router
-	Status      *portainer.Status
+	status      *portainer.Status
+	dataStore   dataservices.DataStore
 	demoService *demo.Service
 }
 
 // NewHandler creates a handler to manage status operations.
-func NewHandler(bouncer *security.RequestBouncer, status *portainer.Status, demoService *demo.Service) *Handler {
+func NewHandler(bouncer *security.RequestBouncer, status *portainer.Status, demoService *demo.Service, dataStore dataservices.DataStore) *Handler {
 	h := &Handler{
 		Router:      mux.NewRouter(),
-		Status:      status,
+		dataStore:   dataStore,
 		demoService: demoService,
+		status:      status,
 	}
+
 	h.Handle("/status",
 		bouncer.PublicAccess(httperror.LoggerHandler(h.statusInspect))).Methods(http.MethodGet)
 	h.Handle("/status/version",
 		bouncer.AuthenticatedAccess(http.HandlerFunc(h.version))).Methods(http.MethodGet)
+	h.Handle("/status/nodes",
+		bouncer.AuthenticatedAccess(httperror.LoggerHandler(h.statusNodesCount))).Methods(http.MethodGet)
+	h.Handle("/status/system",
+		bouncer.AuthenticatedAccess(httperror.LoggerHandler(h.statusSystem))).Methods(http.MethodGet)
 
 	return h
 }

api/http/handler/status/status_inspect.go

@@ -24,7 +24,7 @@ type status struct {
 // @router /status [get]
 func (handler *Handler) statusInspect(w http.ResponseWriter, r *http.Request) *httperror.HandlerError {
 	return response.JSON(w, &status{
-		Status:          handler.Status,
+		Status:          handler.status,
 		DemoEnvironment: handler.demoService.Details(),
 	})
 }

api/http/handler/status/status_nodes_count.go

@@ -0,0 +1,42 @@
+package status
+
+import (
+	"net/http"
+
+	httperror "github.com/portainer/libhttp/error"
+	"github.com/portainer/libhttp/response"
+	statusutil "github.com/portainer/portainer/api/internal/nodes"
+	"github.com/portainer/portainer/api/internal/snapshot"
+)
+
+type nodesCountResponse struct {
+	Nodes int `json:"nodes"`
+}
+
+// @id statusNodesCount
+// @summary Retrieve the count of nodes
+// @description **Access policy**: authenticated
+// @security ApiKeyAuth
+// @security jwt
+// @tags status
+// @produce json
+// @success 200 {object} nodesCountResponse "Success"
+// @failure 500 "Server error"
+// @router /status/nodes [get]
+func (handler *Handler) statusNodesCount(w http.ResponseWriter, r *http.Request) *httperror.HandlerError {
+	endpoints, err := handler.dataStore.Endpoint().Endpoints()
+	if err != nil {
+		return httperror.InternalServerError("Failed to get environment list", err)
+	}
+
+	for i := range endpoints {
+		err = snapshot.FillSnapshotData(handler.dataStore, &endpoints[i])
+		if err != nil {
+			return httperror.InternalServerError("Unable to add snapshot data", err)
+		}
+	}
+
+	nodes := statusutil.NodesCount(endpoints)
+
+	return response.JSON(w, &nodesCountResponse{Nodes: nodes})
+}

api/http/handler/status/status_system.go

@@ -0,0 +1,60 @@
+package status
+
+import (
+	"net/http"
+
+	httperror "github.com/portainer/libhttp/error"
+	"github.com/portainer/libhttp/response"
+	"github.com/portainer/portainer/api/internal/endpointutils"
+	"github.com/portainer/portainer/api/platform"
+)
+
+type systemInfoResponse struct {
+	Platform    platform.ContainerPlatform `json:"platform"`
+	EdgeAgents  int                        `json:"edgeAgents"`
+	EdgeDevices int                        `json:"edgeDevices"`
+	Agents      int                        `json:"agents"`
+}
+
+// @id statusSystem
+// @summary Retrieve system info
+// @description **Access policy**: authenticated
+// @security ApiKeyAuth
+// @security jwt
+// @tags status
+// @produce json
+// @success 200 {object} systemInfoResponse "Success"
+// @failure 500 "Server error"
+// @router /status/system [get]
+func (handler *Handler) statusSystem(w http.ResponseWriter, r *http.Request) *httperror.HandlerError {
+	environments, err := handler.dataStore.Endpoint().Endpoints()
+	if err != nil {
+		return httperror.InternalServerError("Failed to get environment list", err)
+	}
+
+	agents := 0
+	edgeAgents := 0
+	edgeDevices := 0
+
+	for _, environment := range environments {
+		if endpointutils.IsAgentEndpoint(&environment) {
+			agents++
+		}
+
+		if endpointutils.IsEdgeEndpoint(&environment) {
+			edgeAgents++
+		}
+
+		if environment.IsEdgeDevice {
+			edgeDevices++
+		}
+
+	}
+
+	return response.JSON(w, &systemInfoResponse{
+		EdgeAgents:  edgeAgents,
+		EdgeDevices: edgeDevices,
+		Agents:      agents,
+		Platform:    platform.DetermineContainerPlatform(),
+	})
+}