Add support for projects in nexus endpoint commands (#98)

Author: asmclean

temporalcloudcli/commands.gen.go

@@ -4625,6 +4625,7 @@ type CloudNexusEndpointCreateCommand struct {
 	ClientOptions
 	AsyncOperationOptions
 	Name            string
+	ProjectId       string
 	TargetNamespace string
 	TargetTaskQueue string
 	AllowNamespace  []string
@@ -4646,6 +4647,7 @@ func NewCloudNexusEndpointCreateCommand(cctx *CommandContext, parent *CloudNexus
 	s.Command.Args = cobra.NoArgs
 	s.Command.Flags().StringVar(&s.Name, "name", "", "The name of the Nexus Endpoint to create. Required.")
 	_ = cobra.MarkFlagRequired(s.Command.Flags(), "name")
+	s.Command.Flags().StringVar(&s.ProjectId, "project-id", "", "The ID of the project to create the Nexus Endpoint in.")
 	s.Command.Flags().StringVar(&s.TargetNamespace, "target-namespace", "", "The namespace in which a handler worker will be polling for Nexus tasks. Required.")
 	_ = cobra.MarkFlagRequired(s.Command.Flags(), "target-namespace")
 	s.Command.Flags().StringVar(&s.TargetTaskQueue, "target-task-queue", "", "The task queue on which a handler worker will be polling for Nexus tasks. Required.")
@@ -4733,6 +4735,7 @@ type CloudNexusEndpointListCommand struct {
 	Parent  *CloudNexusEndpointCommand
 	Command cobra.Command
 	ClientOptions
+	ProjectId string
 	PageSize  int
 	PageToken string
 }
@@ -4745,6 +4748,7 @@ func NewCloudNexusEndpointListCommand(cctx *CommandContext, parent *CloudNexusEn
 	s.Command.Short = "List Nexus Endpoints"
 	s.Command.Long = "List Nexus Endpoint configurations on the Cloud Account."
 	s.Command.Args = cobra.NoArgs
+	s.Command.Flags().StringVar(&s.ProjectId, "project-id", "", "Filter Nexus Endpoints by project ID.")
 	s.Command.Flags().IntVar(&s.PageSize, "page-size", 0, "Number of endpoints to return per page. If no page size is provided, it will default to 100. A maximum of 1000 endpoints can be fetched at a time.")
 	s.Command.Flags().StringVar(&s.PageToken, "page-token", "", "Token for retrieving the next page of results. Initial value is empty string.")
 	s.ClientOptions.BuildFlags(s.Command.Flags())

temporalcloudcli/commands.nexus.go

@@ -52,6 +52,7 @@ func (c *CloudNexusEndpointListCommand) run(cctx *CommandContext, _ []string) er
 	res, err := client.GetNexusEndpoints(cctx, &cloudservice.GetNexusEndpointsRequest{
 		PageSize:  int32(c.PageSize),
 		PageToken: c.PageToken,
+		ProjectId: c.ProjectId,
 	})
 	if err != nil {
 		return err
@@ -66,7 +67,7 @@ func (c *CloudNexusEndpointListCommand) run(cctx *CommandContext, _ []string) er
 			NextPageToken: res.NextPageToken,
 		},
 		printer.PrintResourceOptions{
-			Fields:     []string{"Id", "State"},
+			Fields:     []string{"Id", "ProjectId", "State"},
 			SpecFields: []string{"Name", "Description"},
 		},
 		printer.TableOptions{},
@@ -132,6 +133,7 @@ func (c *CloudNexusEndpointCreateCommand) run(cctx *CommandContext, _ []string)
 			PolicySpecs: policySpecs,
 		},
 		AsyncOperationId: c.AsyncOperationId,
+		ProjectId:        c.ProjectId,
 	})
 	return cctx.GetPoller(client, c.AsyncOperationOptions).HandleCreateAsyncOperationResponse(cctx, resp, err)
 }

temporalcloudcli/commands.nexus_test.go

@@ -180,6 +180,27 @@ func TestListNexusEndpoints(t *testing.T) {
 				NextPageToken: "tok-def",
 			},
 		},
+		{
+			name: "WithProjectID",
+			cmd: temporalcloudcli.CloudNexusEndpointListCommand{
+				ProjectId: "project-123",
+			},
+			cloudClientExpectations: func(c *cloudmock.MockCloudServiceClient) {
+				c.EXPECT().
+					GetNexusEndpoints(mock.Anything, &cloudservice.GetNexusEndpointsRequest{
+						ProjectId: "project-123",
+					}, mock.Anything).
+					Return(&cloudservice.GetNexusEndpointsResponse{
+						Endpoints: []*nexusv1.Endpoint{testEndpoint},
+					}, nil)
+			},
+			expectedJsonOutput: struct {
+				Endpoints     []*nexusv1.Endpoint
+				NextPageToken string
+			}{
+				Endpoints: []*nexusv1.Endpoint{testEndpoint},
+			},
+		},
 		{
 			name: "APIError",
 			cmd:  temporalcloudcli.CloudNexusEndpointListCommand{},
@@ -219,6 +240,7 @@ func TestCreateNexusEndpoint(t *testing.T) {
 			name: "Success",
 			cmd: temporalcloudcli.CloudNexusEndpointCreateCommand{
 				Name:            "my-endpoint",
+				ProjectId:       "project-123",
 				TargetNamespace: "ns-123",
 				TargetTaskQueue: "my-tq",
 				AllowNamespace:  []string{"caller-ns"},
@@ -230,7 +252,8 @@ func TestCreateNexusEndpoint(t *testing.T) {
 							req.Spec.TargetSpec.GetWorkerTargetSpec().NamespaceId == "ns-123" &&
 							req.Spec.TargetSpec.GetWorkerTargetSpec().TaskQueue == "my-tq" &&
 							len(req.Spec.PolicySpecs) == 1 &&
-							req.Spec.PolicySpecs[0].GetAllowedCloudNamespacePolicySpec().NamespaceId == "caller-ns"
+							req.Spec.PolicySpecs[0].GetAllowedCloudNamespacePolicySpec().NamespaceId == "caller-ns" &&
+							req.ProjectId == "project-123"
 					}), mock.Anything).
 					Return(&cloudservice.CreateNexusEndpointResponse{
 						EndpointId:     "ep-new",
@@ -500,7 +523,6 @@ func TestUpdateNexusEndpoint(t *testing.T) {
 				Endpoints: []*nexusv1.Endpoint{existingEndpoint},
 			}, nil)
 	}
-
 	tests := []struct {
 		name                    string
 		setupCmd                func(*temporalcloudcli.CloudNexusEndpointUpdateCommand)

temporalcloudcli/commands.yml

@@ -1868,6 +1868,10 @@ commands:
     option-sets:
       - client
     options:
+      - name: project-id
+        type: string
+        description: |
+          Filter Nexus Endpoints by project ID.
       - name: page-size
         type: int
         description: |
@@ -1918,6 +1922,10 @@ commands:
         required: true
         description: |
           The name of the Nexus Endpoint to create.
+      - name: project-id
+        type: string
+        description: |
+          The ID of the project to create the Nexus Endpoint in.
       - name: target-namespace
         type: string
         required: true