feat: Add new gcloud commands, API clients, and third-party libraries across various services.

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/node_pools/__init__.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google LLC. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Command group `gcloud container aws node-pools`."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.projects import util
+from googlecloudsdk.core import log
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class NodePools(base.Group):
+  """Create and manage node pools in an Anthos cluster on AWS."""
+
+  @staticmethod
+  def Args(parser):
+    pass

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/node_pools/create.py

@@ -0,0 +1,113 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google LLC. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Command to create a node pool in an Anthos cluster on AWS."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import aws as api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import flags as aws_flags
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import command_util
+from googlecloudsdk.command_lib.container.gkemulticloud import constants
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+from googlecloudsdk.command_lib.container.gkemulticloud import flags
+
+# Command needs to be in one line for the docgen tool to format properly.
+_EXAMPLES = """
+To create a node pool named ``my-node-pool'' in a cluster named ``my-cluster''
+managed in location ``us-west1'', run:
+
+$ {command} my-node-pool --cluster=my-cluster --location=us-west1 --iam-instance-profile=IAM_INSTANCE_PROFILE --node-version=NODE_VERSION --subnet-id=SUBNET_ID
+"""
+
+
+@base.DefaultUniverseOnly
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class Create(base.CreateCommand):
+  """Create a node pool in an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser, track=base.ReleaseTrack.GA):
+    resource_args.AddAwsNodePoolResourceArg(parser, 'to create')
+    flags.AddNodeVersion(parser)
+    flags.AddSubnetID(parser, 'the node pool')
+    flags.AddAutoscaling(parser)
+    flags.AddMaxPodsPerNode(parser)
+    flags.AddRootVolumeSize(parser)
+    flags.AddValidateOnly(parser, 'node pool to create')
+    flags.AddTags(parser, 'node pool')
+    flags.AddNodeLabels(parser)
+    flags.AddNodeTaints(parser)
+    flags.AddAnnotations(parser, 'node pool')
+    flags.AddEnableAutoRepair(parser, True)
+    flags.AddMaxSurgeUpdate(parser)
+    flags.AddMaxUnavailableUpdate(parser, for_create=True)
+
+    aws_flags.AddOnDemandOrSpotInstanceType(parser, kind='node pool')
+    aws_flags.AddSshEC2KeyPair(parser, kind='node pool')
+    aws_flags.AddIamInstanceProfile(parser, kind='node pool')
+    aws_flags.AddSecurityGroupIds(parser, kind='node pool')
+    aws_flags.AddRootVolumeType(parser)
+    aws_flags.AddRootVolumeIops(parser)
+    aws_flags.AddRootVolumeThroughput(parser)
+    aws_flags.AddRootVolumeKmsKeyArn(parser)
+    aws_flags.AddProxyConfig(parser)
+    aws_flags.AddConfigEncryptionKmsKeyArn(parser)
+    aws_flags.AddAutoScalingMetricsCollection(parser)
+    aws_flags.AddKubeletConfigCpuManagerPolicy(parser)
+    aws_flags.AddKubeletConfigCpuCfsQuota(parser)
+    aws_flags.AddKubeletConfigCpuCfsQuotaPeriod(parser)
+    aws_flags.AddKubeletConfigPodPidsLimit(parser)
+
+    base.ASYNC_FLAG.AddToParser(parser)
+
+    parser.display_info.AddFormat(constants.AWS_NODEPOOLS_FORMAT)
+
+  def Run(self, args):
+    """Runs the create command."""
+    location = resource_args.ParseAwsNodePoolResourceArg(args).locationsId
+    with endpoint_util.GkemulticloudEndpointOverride(location):
+      node_pool_ref = resource_args.ParseAwsNodePoolResourceArg(args)
+      node_pool_client = api_util.NodePoolsClient()
+      message = command_util.NodePoolMessage(
+          node_pool_ref.awsNodePoolsId,
+          action='Creating',
+          cluster=node_pool_ref.awsClustersId,
+      )
+      return command_util.Create(
+          resource_ref=node_pool_ref,
+          resource_client=node_pool_client,
+          args=args,
+          message=message,
+          kind=constants.AWS_NODEPOOL_KIND,
+      )
+
+
+@base.DefaultUniverseOnly
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class CreateAlpha(Create):
+  """Create a node pool in an Anthos cluster on AWS."""
+
+  @staticmethod
+  def Args(parser):
+    """Registers alpha track flags for this command."""
+    Create.Args(parser, base.ReleaseTrack.ALPHA)
+    aws_flags.AddInstancePlacement(parser)
+    flags.AddImageType(parser)

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/node_pools/delete.py

@@ -0,0 +1,68 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google LLC. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Command to delete a node pool in an Anthos cluster on AWS."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import aws as api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import command_util
+from googlecloudsdk.command_lib.container.gkemulticloud import constants
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+from googlecloudsdk.command_lib.container.gkemulticloud import flags
+
+_EXAMPLES = """
+To delete a node pool named ``my-node-pool'' in a cluster named ``my-cluster''
+managed in location ``us-west1'', run:
+
+$ {command} my-node-pool --cluster=my-cluster --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Delete(base.DeleteCommand):
+  """Delete a node pool in an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAwsNodePoolResourceArg(parser, 'to delete')
+
+    flags.AddValidateOnly(parser, 'node pool to delete')
+    flags.AddAllowMissing(parser, 'node pool')
+    flags.AddIgnoreErrors(parser, constants.AWS, 'node pool')
+
+    base.ASYNC_FLAG.AddToParser(parser)
+
+  def Run(self, args):
+    """Runs the delete command."""
+    location = resource_args.ParseAwsNodePoolResourceArg(args).locationsId
+    with endpoint_util.GkemulticloudEndpointOverride(location):
+      node_pool_ref = resource_args.ParseAwsNodePoolResourceArg(args)
+      node_pool_client = api_util.NodePoolsClient()
+      message = command_util.NodePoolMessage(
+          node_pool_ref.awsNodePoolsId, cluster=node_pool_ref.awsClustersId
+      )
+      command_util.DeleteWithIgnoreErrors(
+          args,
+          node_pool_client,
+          node_pool_ref,
+          message,
+          constants.AWS_NODEPOOL_KIND,
+      )

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/node_pools/describe.py

@@ -0,0 +1,62 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google LLC. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Command to describe a node pool in an Anthos cluster on AWS."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import aws as api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import constants
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+from googlecloudsdk.command_lib.container.gkemulticloud import versions
+from googlecloudsdk.core import log
+
+_EXAMPLES = """
+To describe a node pool named ``my-node-pool'' in a cluster named
+``my-cluster'' managed in location ``us-west1'', run:
+
+$ {command} my-node-pool --cluster=my-cluster --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Describe(base.DescribeCommand):
+  """Describe a node pool in an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAwsNodePoolResourceArg(parser, 'to describe')
+
+  def Run(self, args):
+    """Runs the describe command."""
+    self._upgrade_hint = None
+    node_pool_ref = resource_args.ParseAwsNodePoolResourceArg(args)
+    with endpoint_util.GkemulticloudEndpointOverride(node_pool_ref.locationsId):
+      node_pool_client = api_util.NodePoolsClient()
+      node_pool_info = node_pool_client.Get(node_pool_ref)
+      self._upgrade_hint = versions.upgrade_hint_node_pool(
+          node_pool_ref, node_pool_info, constants.AWS
+      )
+
+      return node_pool_info
+
+  def Epilog(self, results_were_displayed):
+    if self._upgrade_hint:
+      log.status.Print(self._upgrade_hint)

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/node_pools/list.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google LLC. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Command to list node pools in an Anthos cluster on AWS."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import aws as api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import constants
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+from googlecloudsdk.command_lib.container.gkemulticloud import versions
+from googlecloudsdk.core import log
+
+
+_EXAMPLES = """
+To list all node pools in a cluster named ``my-cluster''
+managed in location ``us-west1'', run:
+
+$ {command} --cluster=my-cluster --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class List(base.ListCommand):
+  """List node pools in an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAwsClusterResourceArg(parser, 'to list', positional=False)
+    parser.display_info.AddFormat(constants.AWS_NODEPOOLS_FORMAT)
+
+  def Run(self, args):
+    """Runs the list command."""
+    self._upgrade_hint = None
+    cluster_ref = args.CONCEPTS.cluster.Parse()
+    with endpoint_util.GkemulticloudEndpointOverride(cluster_ref.locationsId):
+      node_pool_client = api_util.NodePoolsClient()
+      items, is_empty = node_pool_client.List(
+          cluster_ref, args.page_size, args.limit
+      )
+      if is_empty:
+        return items
+
+      platform = constants.AWS
+      node_pool_info_table, end_of_life_flag = (
+          versions.generate_node_pool_versions_table(
+              cluster_ref,
+              platform,
+              items,
+          )
+      )
+      if end_of_life_flag:
+        self._upgrade_hint = versions.upgrade_hint_node_pool_list(
+            platform, cluster_ref
+        )
+      return node_pool_info_table
+
+  def Epilog(self, results_were_displayed):
+    super(List, self).Epilog(results_were_displayed)
+    if self._upgrade_hint:
+      log.status.Print(self._upgrade_hint)

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/node_pools/rollback.py

@@ -0,0 +1,64 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2023 Google LLC. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Command to rollback a node pool in an Anthos cluster on AWS."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import aws as api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import command_util
+from googlecloudsdk.command_lib.container.gkemulticloud import constants
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+from googlecloudsdk.command_lib.container.gkemulticloud import flags
+
+_EXAMPLES = """
+To roll back a canceled or failed update in node pool named ``my-node-pool''
+in a cluster named ``my-cluster'' managed in location ``us-west1'', run:
+
+$ {command} my-node-pool --cluster=my-cluster --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Rollback(base.Command):
+  """Rollback a node pool in an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAwsNodePoolResourceArg(parser, 'to rollback')
+    flags.AddRespectPodDisruptionBudget(parser)
+    base.ASYNC_FLAG.AddToParser(parser)
+
+  def Run(self, args):
+    """Runs the rollback command."""
+    location = resource_args.ParseAwsNodePoolResourceArg(args).locationsId
+    with endpoint_util.GkemulticloudEndpointOverride(location):
+      node_pool_ref = resource_args.ParseAwsNodePoolResourceArg(args)
+      node_pool_client = api_util.NodePoolsClient()
+      message = command_util.NodePoolMessage(
+          node_pool_ref.awsNodePoolsId, cluster=node_pool_ref.awsClustersId
+      )
+      return command_util.Rollback(
+          resource_ref=node_pool_ref,
+          resource_client=node_pool_client,
+          message=message,
+          args=args,
+          kind=constants.AWS_NODEPOOL_KIND,
+      )

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/node_pools/update.py

@@ -0,0 +1,100 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google LLC. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Command to update a node pool in an Anthos cluster on AWS."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import aws as api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import flags as aws_flags
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import command_util
+from googlecloudsdk.command_lib.container.gkemulticloud import constants
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+from googlecloudsdk.command_lib.container.gkemulticloud import flags
+
+# Command needs to be in one line for the docgen tool to format properly.
+_EXAMPLES = """
+To update a node pool named ``my-node-pool'' in a cluster named ``my-cluster''
+managed in location ``us-west1'', run:
+
+$ {command} my-node-pool --cluster=my-cluster --location=us-west1 --node-version=NODE_VERSION
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class Update(base.UpdateCommand):
+  """Update a node pool in an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAwsNodePoolResourceArg(parser, 'to update')
+    flags.AddNodeVersion(parser, required=False)
+    flags.AddValidateOnly(parser, 'node pool to update')
+    flags.AddAutoscaling(parser, required=False)
+    flags.AddAnnotationsForUpdate(parser, 'node pool')
+    flags.AddTagsForUpdate(parser, 'node pool')
+    flags.AddRootVolumeSize(parser)
+    flags.AddNodeLabelsForUpdate(parser)
+    flags.AddEnableAutoRepair(parser, False)
+    flags.AddMaxSurgeUpdate(parser)
+    flags.AddMaxUnavailableUpdate(parser)
+    aws_flags.AddSecurityGroupFlagsForUpdate(parser, 'node pool')
+    aws_flags.AddRootVolumeKmsKeyArn(parser)
+    aws_flags.AddRootVolumeType(parser)
+    aws_flags.AddRootVolumeIops(parser)
+    aws_flags.AddRootVolumeThroughput(parser)
+    aws_flags.AddConfigEncryptionKmsKeyArn(parser, required=False)
+    aws_flags.AddProxyConfigForUpdate(parser, 'node pool')
+    aws_flags.AddSshEC2KeyPairForUpdate(parser, 'node pool')
+    aws_flags.AddIamInstanceProfile(parser, kind='node pool', required=False)
+    aws_flags.AddAutoScalingMetricsCollectionForUpdate(parser)
+    aws_flags.AddInstanceType(parser, kind='node pool')
+    base.ASYNC_FLAG.AddToParser(parser)
+
+    parser.display_info.AddFormat(constants.AWS_NODEPOOLS_FORMAT)
+
+  def Run(self, args):
+    """Runs the update command."""
+    location = resource_args.ParseAwsNodePoolResourceArg(args).locationsId
+    with endpoint_util.GkemulticloudEndpointOverride(location):
+      node_pool_ref = resource_args.ParseAwsNodePoolResourceArg(args)
+      node_pool_client = api_util.NodePoolsClient()
+      message = command_util.NodePoolMessage(
+          node_pool_ref.awsNodePoolsId,
+          action='Updating',
+          cluster=node_pool_ref.awsClustersId,
+      )
+      return command_util.Update(
+          resource_ref=node_pool_ref,
+          resource_client=node_pool_client,
+          args=args,
+          message=message,
+          kind=constants.AWS_NODEPOOL_KIND,
+      )
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class UpdateAlpha(Update):
+  """Update a node pool in an Anthos cluster on AWS."""
+
+  @staticmethod
+  def Args(parser, track=base.ReleaseTrack.ALPHA):
+    """Registers alpha track flags for this command."""
+    Update.Args(parser)