API: Admin endpoint cluster id filter for listing nodes (#2641)

Adds support to query nodes in non-default clusters

Author: sitole

packages/api/internal/api/api.gen.go

@@ -14,8 +14,9 @@ import (
 	"github.com/e2b-dev/infra/packages/shared/pkg/telemetry"
 )
 
-func (a *APIStore) GetNodes(c *gin.Context) {
-	result, err := a.orchestrator.AdminNodes()
+func (a *APIStore) GetNodes(c *gin.Context, params api.GetNodesParams) {
+	clusterID := clusters.WithClusterFallback(params.ClusterID)
+	result, err := a.orchestrator.AdminNodes(clusterID)
 	if err != nil {
 		telemetry.ReportCriticalError(c.Request.Context(), "error when getting nodes", err)
 		a.sendAPIStoreError(c, http.StatusInternalServerError, "Error when getting nodes")

packages/api/internal/handlers/admin.go

@@ -9,12 +9,11 @@ import (
 	"github.com/e2b-dev/infra/packages/api/internal/api"
 )
 
-func (o *Orchestrator) AdminNodes() ([]*api.Node, error) {
+func (o *Orchestrator) AdminNodes(clusterID uuid.UUID) ([]*api.Node, error) {
 	var result []*api.Node
 
 	for _, n := range o.nodes.Items() {
-		// Skip all nodes that are not running in local (Nomad) cluster
-		if !n.IsNomadManaged() {
+		if n.ClusterID != clusterID {
 			continue
 		}
 

packages/api/internal/orchestrator/admin.go

@@ -291,14 +291,14 @@ components:
           additionalProperties:
             type: array
             items:
-              $ref: '#/components/schemas/SandboxNetworkRule'
+              $ref: "#/components/schemas/SandboxNetworkRule"
 
     SandboxNetworkRule:
       type: object
       description: Transform rule applied to egress requests matching a domain pattern.
       properties:
         transform:
-          $ref: '#/components/schemas/SandboxNetworkTransform'
+          $ref: "#/components/schemas/SandboxNetworkTransform"
 
     SandboxNetworkTransform:
       type: object
@@ -2431,7 +2431,7 @@ paths:
                   additionalProperties:
                     type: array
                     items:
-                      $ref: '#/components/schemas/SandboxNetworkRule'
+                      $ref: "#/components/schemas/SandboxNetworkRule"
                 allow_internet_access:
                   type: boolean
                   description:
@@ -3091,6 +3091,14 @@ paths:
       tags: [admin]
       security:
         - AdminTokenAuth: []
+      parameters:
+        - in: query
+          name: clusterID
+          description: Identifier of the cluster
+          required: false
+          schema:
+            type: string
+            format: uuid
       responses:
         "200":
           description: Successfully returned all nodes