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/__init__.py

@@ -0,0 +1,46 @@
+# -*- 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`."""
+
+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
+
+
+@base.Deprecate(
+    is_removed=False,
+    warning=(
+        "This command is deprecated. See"
+        " https://cloud.google.com/kubernetes-engine/multi-cloud/docs/aws/deprecations/deprecation-announcement"
+        " for more details."
+    ),
+    error=(
+        "This command has been removed. See"
+        " https://cloud.google.com/kubernetes-engine/multi-cloud/docs/aws/deprecations/deprecation-announcement"
+        " for more details."
+    ),
+)
+@base.UniverseCompatible
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Aws(base.Group):
+  """Deploy and manage clusters of machines on AWS for running containers."""
+
+  category = base.COMPUTE_CATEGORY
+
+  def Filter(self, context, args):
+    del context, args

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/clusters/__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 clusters`."""
+
+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 Clusters(base.Group):
+  """Create and manage Anthos clusters on AWS."""
+
+  @staticmethod
+  def Args(parser):
+    pass

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

@@ -0,0 +1,132 @@
+# -*- 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 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 arg_parsers
+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 cluster named ``my-cluster'' managed in location ``us-west1'',
+run:
+
+$ {command} my-cluster --location=us-west1 --aws-region=AWS_REGION --cluster-version=CLUSTER_VERSION --database-encryption-kms-key-arn=KMS_KEY_ARN --iam-instance-profile=IAM_INSTANCE_PROFILE --pod-address-cidr-blocks=POD_ADDRESS_CIDR_BLOCKS --role-arn=ROLE_ARN --service-address-cidr-blocks=SERVICE_ADDRESS_CIDR_BLOCKS --subnet-ids=SUBNET_ID --vpc-id=VPC_ID
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class Create(base.CreateCommand):
+  """Create an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    """Registers flags for this command."""
+    resource_args.AddAwsClusterResourceArg(parser, 'to create')
+
+    parser.add_argument(
+        '--subnet-ids',
+        required=True,
+        type=arg_parsers.ArgList(),
+        metavar='SUBNET_ID',
+        help=(
+            'Subnet ID of an existing VNET to use for the cluster control'
+            ' plane.'
+        ),
+    )
+
+    flags.AddPodAddressCidrBlocks(parser)
+    flags.AddServiceAddressCidrBlocks(parser)
+    flags.AddClusterVersion(parser)
+    flags.AddRootVolumeSize(parser)
+    flags.AddMainVolumeSize(parser)
+    flags.AddValidateOnly(parser, 'cluster to create')
+    flags.AddFleetProject(parser)
+    flags.AddTags(parser, 'cluster')
+    flags.AddAdminUsers(parser)
+    flags.AddAdminGroups(parser)
+    flags.AddDescription(parser)
+    flags.AddAnnotations(parser)
+    flags.AddLogging(parser)
+    flags.AddMonitoringConfig(parser, True)
+    flags.AddBinauthzEvaluationMode(parser)
+
+    aws_flags.AddAwsRegion(parser)
+    aws_flags.AddIamInstanceProfile(parser)
+    aws_flags.AddInstanceType(parser)
+    aws_flags.AddSshEC2KeyPair(parser)
+    aws_flags.AddConfigEncryptionKmsKeyArn(parser)
+    aws_flags.AddDatabaseEncryptionKmsKeyArn(parser)
+    aws_flags.AddRoleArn(parser)
+    aws_flags.AddRoleSessionName(parser)
+    aws_flags.AddVpcId(parser)
+    aws_flags.AddSecurityGroupIds(parser, kind='control plane')
+    aws_flags.AddPerNodePoolSGRules(parser)
+    aws_flags.AddRootVolumeType(parser)
+    aws_flags.AddRootVolumeIops(parser)
+    aws_flags.AddRootVolumeThroughput(parser)
+    aws_flags.AddRootVolumeKmsKeyArn(parser)
+    aws_flags.AddMainVolumeType(parser)
+    aws_flags.AddMainVolumeIops(parser)
+    aws_flags.AddMainVolumeThroughput(parser)
+    aws_flags.AddMainVolumeKmsKeyArn(parser)
+    aws_flags.AddProxyConfig(parser)
+
+    base.ASYNC_FLAG.AddToParser(parser)
+
+    parser.display_info.AddFormat(constants.AWS_CLUSTERS_FORMAT)
+
+  def Run(self, args):
+    """Runs the create command."""
+    location = resource_args.ParseAwsClusterResourceArg(args).locationsId
+    with endpoint_util.GkemulticloudEndpointOverride(location):
+      cluster_ref = resource_args.ParseAwsClusterResourceArg(args)
+      cluster_client = api_util.ClustersClient()
+      message = command_util.ClusterMessage(
+          cluster_ref.awsClustersId,
+          action='Creating',
+          kind=constants.AWS,
+          region=args.aws_region,
+      )
+      return command_util.Create(
+          resource_ref=cluster_ref,
+          resource_client=cluster_client,
+          args=args,
+          message=message,
+          kind=constants.AWS_CLUSTER_KIND,
+      )
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class CreateAlpha(Create):
+  """Create an Anthos cluster on AWS."""
+
+  @staticmethod
+  def Args(parser, track=base.ReleaseTrack.ALPHA):
+    """Registers alpha track flags for this command."""
+    Create.Args(parser)
+    aws_flags.AddInstancePlacement(parser)

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

@@ -0,0 +1,72 @@
+# -*- 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 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 cluster named ``my-cluster'' managed in location ``us-west1'',
+run:
+
+$ {command} my-cluster --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Delete(base.DeleteCommand):
+  """Delete an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    """Register flags for this command."""
+    resource_args.AddAwsClusterResourceArg(parser, 'to delete')
+
+    flags.AddValidateOnly(parser, 'cluster to delete')
+    flags.AddAllowMissing(parser, 'cluster')
+    flags.AddIgnoreErrors(parser, constants.AWS, 'cluster')
+
+    base.ASYNC_FLAG.AddToParser(parser)
+
+  def Run(self, args):
+    """Runs the delete command."""
+    location = resource_args.ParseAwsClusterResourceArg(args).locationsId
+    with endpoint_util.GkemulticloudEndpointOverride(location):
+      cluster_ref = resource_args.ParseAwsClusterResourceArg(args)
+      cluster_client = api_util.ClustersClient()
+      cluster = cluster_client.Get(cluster_ref)
+      message = command_util.ClusterMessage(
+          cluster_ref.awsClustersId,
+          kind=constants.AWS,
+          region=cluster.awsRegion,
+      )
+      command_util.DeleteWithIgnoreErrors(
+          args,
+          cluster_client,
+          cluster_ref,
+          message,
+          constants.AWS_CLUSTER_KIND,
+      )

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

@@ -0,0 +1,63 @@
+# -*- 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 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 cluster named ``my-cluster'' managed in location ``us-west1'',
+run:
+
+$ {command} my-cluster --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Describe(base.DescribeCommand):
+  """Describe an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    """Registers flags for this command."""
+    resource_args.AddAwsClusterResourceArg(parser, 'to describe')
+
+  def Run(self, args):
+    """Runs the describe command."""
+    self._upgrade_hint = None
+    cluster_ref = resource_args.ParseAwsClusterResourceArg(args)
+    with endpoint_util.GkemulticloudEndpointOverride(cluster_ref.locationsId):
+      cluster_client = api_util.ClustersClient()
+      cluster_info = cluster_client.Get(cluster_ref)
+      self._upgrade_hint = versions.upgrade_hint_cluster(
+          cluster_ref, cluster_info, constants.AWS
+      )
+      return cluster_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/clusters/get_credentials.py

@@ -0,0 +1,95 @@
+# -*- 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 get credentials of 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.api_lib.container.gkemulticloud import util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+from googlecloudsdk.command_lib.container.gkemulticloud import flags
+from googlecloudsdk.command_lib.container.gkemulticloud import kubeconfig
+from googlecloudsdk.core import log
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class GetCredentials(base.Command):
+  """Get credentials of an Anthos cluster on AWS."""
+
+  detailed_help = {
+      'EXAMPLES': kubeconfig.COMMAND_EXAMPLE,
+      'DESCRIPTION': kubeconfig.COMMAND_DESCRIPTION.format(
+          cluster_type='Anthos cluster on AWS'
+      ),
+  }
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAwsClusterResourceArg(parser, 'to get credentials')
+    flags.AddAuthProviderCmdPath(parser)
+    flags.AddPrivateEndpoint(parser)
+
+  def Run(self, args):
+    """Runs the get-credentials command."""
+    with endpoint_util.GkemulticloudEndpointOverride(
+        resource_args.ParseAwsClusterResourceArg(args).locationsId,
+        self.ReleaseTrack(),
+    ):
+      cluster_ref = resource_args.ParseAwsClusterResourceArg(args)
+      cluster_client = api_util.ClustersClient()
+
+      log.status.Print('Fetching cluster endpoint and auth data.')
+      resp = cluster_client.Get(cluster_ref)
+
+      if (
+          resp.state
+          != util.GetMessagesModule().GoogleCloudGkemulticloudV1AwsCluster.StateValueValuesEnum.RUNNING
+      ):
+        log.warning(
+            kubeconfig.NOT_RUNNING_MSG.format(cluster_ref.awsClustersId)
+        )
+
+      if not args.private_endpoint and kubeconfig.ConnectGatewayInNodePools(
+          resp, cluster_ref.awsClustersId
+      ):
+        kubeconfig.CheckClusterHasNodePools(
+            api_util.NodePoolsClient(), cluster_ref
+        )
+
+      kubeconfig.ValidateClusterVersion(resp, cluster_ref.awsClustersId)
+      context = kubeconfig.GenerateContext(
+          'aws',
+          cluster_ref.projectsId,
+          cluster_ref.locationsId,
+          cluster_ref.awsClustersId,
+      )
+      cmd_args = kubeconfig.GenerateAuthProviderCmdArgs(
+          'aws',
+          cluster_ref.awsClustersId,
+          cluster_ref.locationsId,
+          cluster_ref.projectsId,
+      )
+      kubeconfig.GenerateKubeconfig(
+          resp,
+          cluster_ref.awsClustersId,
+          context,
+          args.auth_provider_cmd_path,
+          cmd_args,
+          args.private_endpoint,
+      )

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

@@ -0,0 +1,74 @@
+# -*- 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 Anthos clusters 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 lists all clusters managed in location ``us-west1'', run:
+
+$ {command} --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class List(base.ListCommand):
+  """List Anthos clusters on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddLocationResourceArg(parser, 'to list')
+    parser.display_info.AddFormat(constants.AWS_CLUSTERS_FORMAT)
+
+  def Run(self, args):
+    """Runs the list command."""
+    self._upgrade_hint = None
+    location_ref = args.CONCEPTS.location.Parse()
+    with endpoint_util.GkemulticloudEndpointOverride(location_ref.locationsId):
+      cluster_client = api_util.ClustersClient()
+      items, is_empty = cluster_client.List(
+          location_ref, args.page_size, args.limit
+      )
+      if is_empty:
+        return items
+      platform = constants.AWS
+      cluster_info_table, end_of_life_flag = (
+          versions.generate_cluster_versions_table(
+              location_ref,
+              platform,
+              items,
+          )
+      )
+      if end_of_life_flag:
+        self._upgrade_hint = versions.upgrade_hint_cluster_list(platform)
+      return cluster_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/clusters/print_access_token.py

@@ -0,0 +1,51 @@
+# -*- 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 print access tokens for 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 endpoint_util
+from googlecloudsdk.command_lib.container.gkemulticloud import flags
+from googlecloudsdk.command_lib.container.gkemulticloud import kubeconfig
+
+
+@base.Hidden
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class PrintAccessToken(base.Command):
+  """Generate an access token for an Anthos cluster on AWS."""
+
+  @staticmethod
+  def Args(parser):
+    """Register flags for this command."""
+    resource_args.AddAwsClusterResourceArg(parser, 'to access')
+    flags.AddExecCredential(parser)
+
+  def Run(self, args):
+    """Runs the command."""
+    cluster_ref = args.CONCEPTS.cluster.Parse()
+    with endpoint_util.GkemulticloudEndpointOverride(cluster_ref.locationsId):
+      cluster_client = api_util.ClustersClient()
+      response = cluster_client.GenerateAccessToken(cluster_ref)
+      if args.exec_credential:
+        return kubeconfig.ExecCredential(
+            expiration_timestamp=response.expirationTime,
+            access_token=response.accessToken,
+        )
+      return response

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/clusters/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 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 cluster named ``my-cluster'' managed in location ``us-west1'', run:
+
+$ {command} my-cluster --location=us-west1 --cluster-version=CLUSTER_VERSION
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class Update(base.UpdateCommand):
+  """Update an Anthos cluster on AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAwsClusterResourceArg(parser, 'to update')
+    flags.AddClusterVersion(parser, required=False)
+    flags.AddValidateOnly(parser, 'update of the cluster')
+    flags.AddAdminUsers(parser, create=False)
+    flags.AddAdminGroups(parser)
+    flags.AddRootVolumeSize(parser)
+    flags.AddDescriptionForUpdate(parser)
+    flags.AddAnnotationsForUpdate(parser, 'cluster')
+    flags.AddTagsForUpdate(parser, 'control plane')
+    flags.AddLogging(parser)
+    flags.AddMonitoringConfig(parser)
+    flags.AddBinauthzEvaluationMode(parser)
+    aws_flags.AddInstanceType(parser)
+    aws_flags.AddRoleArn(parser, required=False)
+    aws_flags.AddRoleSessionName(parser)
+    aws_flags.AddConfigEncryptionKmsKeyArn(parser, required=False)
+    aws_flags.AddSecurityGroupFlagsForUpdate(parser, 'control plane')
+    aws_flags.AddPerNodePoolSGRulesForUpdate(parser)
+    aws_flags.AddProxyConfigForUpdate(parser, 'control plane')
+    aws_flags.AddRootVolumeKmsKeyArn(parser)
+    aws_flags.AddRootVolumeType(parser)
+    aws_flags.AddRootVolumeIops(parser)
+    aws_flags.AddRootVolumeThroughput(parser)
+    aws_flags.AddSshEC2KeyPairForUpdate(parser)
+    aws_flags.AddIamInstanceProfile(parser, required=False)
+
+    base.ASYNC_FLAG.AddToParser(parser)
+    parser.display_info.AddFormat(constants.AWS_CLUSTERS_FORMAT)
+
+  def Run(self, args):
+    """Runs the update command."""
+    location = resource_args.ParseAwsClusterResourceArg(args).locationsId
+    with endpoint_util.GkemulticloudEndpointOverride(location):
+      cluster_ref = resource_args.ParseAwsClusterResourceArg(args)
+      cluster_client = api_util.ClustersClient()
+      message = command_util.ClusterMessage(
+          cluster_ref.awsClustersId, action='Updating'
+      )
+      return command_util.Update(
+          resource_ref=cluster_ref,
+          resource_client=cluster_client,
+          args=args,
+          message=message,
+          kind=constants.AWS_CLUSTER_KIND,
+      )
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class UpdateAlpha(Update):
+  """Update an Anthos cluster on AWS."""
+
+  @staticmethod
+  def Args(parser, track=base.ReleaseTrack.ALPHA):
+    """Registers alpha track flags for this command."""
+    Update.Args(parser)

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

@@ -0,0 +1,57 @@
+# -*- 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 get Anthos Multi-Cloud server configuration for AWS."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import locations 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.core import log
+
+
+_EXAMPLES = """
+To return supported AWS regions and valid versions in location ``us-west1'', run:
+
+$ {command} --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class GetServerConfig(base.Command):
+  """Get Anthos Multi-Cloud server configuration for AWS."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddLocationResourceArg(parser, 'to get server configuration')
+    parser.display_info.AddFormat(constants.AWS_SERVER_CONFIG_FORMAT)
+
+  def Run(self, args):
+    """Runs the get-server-config command."""
+    location_ref = args.CONCEPTS.location.Parse()
+    with endpoint_util.GkemulticloudEndpointOverride(location_ref.locationsId):
+      log.status.Print(
+          'Fetching server config for {location}'.format(
+              location=location_ref.locationsId
+          )
+      )
+      client = api_util.LocationsClient()
+      return client.GetAwsServerConfig(location_ref)

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)

gcloud auth application-default login/google-cloud-sdk/lib/surface/container/aws/operations/__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 operations`."""
+
+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 Operations(base.Group):
+  """Manage Anthos Multi-Cloud long running operations for AWS."""
+
+  @staticmethod
+  def Args(parser):
+    pass

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

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2022 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 cancel an operation."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import operations as op_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 endpoint_util
+from googlecloudsdk.core import log
+
+_EXAMPLES = """
+To cancel an operation in location ``us-west1'', run:
+
+$ {command} OPERATION_ID --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Cancel(base.Command):
+  """Cancel an operation."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    """Registers flags for this command."""
+    resource_args.AddOperationResourceArg(parser, 'to cancel')
+
+  def Run(self, args):
+    """Runs the cancel command."""
+    with endpoint_util.GkemulticloudEndpointOverride(
+        resource_args.ParseOperationResourceArg(args).locationsId,
+        self.ReleaseTrack(),
+    ):
+      op_client = op_api_util.OperationsClient()
+      op_ref = resource_args.ParseOperationResourceArg(args)
+      op = op_client.Get(op_ref)
+      command_util.CancelOperationPrompt(op.name)
+      op_client.Cancel(op_ref)
+      log.status.Print(command_util.CancelOperationMessage(op.name, 'aws'))
+      return op_client.Get(op_ref)

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

@@ -0,0 +1,52 @@
+# -*- 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 an operation."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import operations as op_api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+
+_EXAMPLES = """
+To describe an operation in location ``us-west1'', run:
+
+$ {command} OPERATION_ID --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Describe(base.DescribeCommand):
+  """Describe an operation."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    """Registers flags for this command."""
+    resource_args.AddOperationResourceArg(parser, 'to describe')
+
+  def Run(self, args):
+    """Runs the describe command."""
+    with endpoint_util.GkemulticloudEndpointOverride(
+        resource_args.ParseOperationResourceArg(args).locationsId,
+        self.ReleaseTrack(),
+    ):
+      op_client = op_api_util.OperationsClient()
+      op_ref = resource_args.ParseOperationResourceArg(args)
+      return op_client.Get(op_ref)

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

@@ -0,0 +1,61 @@
+# -*- 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.
+"""Commmand to list operations."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import operations as op_api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+from googlecloudsdk.command_lib.container.gkemulticloud import operations
+
+_EXAMPLES = """
+To list all operations in location ``us-west1'', run:
+
+$ {command} --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class List(base.ListCommand):
+  """List operations."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    """Registers flags for this command."""
+    resource_args.AddLocationResourceArg(parser, 'to list operations')
+    operations.AddFormat(parser)
+
+  def Run(self, args):
+    """Runs the describe command."""
+    release_track = self.ReleaseTrack()
+    location_ref = args.CONCEPTS.location.Parse()
+    with endpoint_util.GkemulticloudEndpointOverride(
+        location_ref.locationsId, release_track
+    ):
+      op_client = op_api_util.OperationsClient()
+      items, empty = op_client.List(
+          location_ref, args.page_size, args.limit, parent_field='name'
+      )
+      if not empty:
+        # ListOperations returns AWS, Azure, and attached operations.
+        # Add a filter for AWS operations.
+        operations.AddFilter(args, 'aws')
+      return items

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

@@ -0,0 +1,56 @@
+# -*- 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 wait for an operation to complete."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.container.gkemulticloud import operations as op_api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.container.aws import resource_args
+from googlecloudsdk.command_lib.container.gkemulticloud import endpoint_util
+
+_EXAMPLES = """
+To wait for an operation in location ``us-west1'' to complete, run:
+
+$ {command} OPERATION_ID --location=us-west1
+"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Describe(base.DescribeCommand):
+  """Wait for an operation to complete."""
+
+  detailed_help = {'EXAMPLES': _EXAMPLES}
+
+  @staticmethod
+  def Args(parser):
+    """Registers flags for this command."""
+    resource_args.AddOperationResourceArg(parser, 'to wait for')
+
+  def Run(self, args):
+    """Runs the wait command."""
+    with endpoint_util.GkemulticloudEndpointOverride(
+        resource_args.ParseOperationResourceArg(args).locationsId,
+        self.ReleaseTrack(),
+    ):
+      op_client = op_api_util.OperationsClient()
+      op_ref = resource_args.ParseOperationResourceArg(args)
+      op_client.Wait(
+          op_ref,
+          'Waiting for operation {} to complete'.format(op_ref.RelativeName()),
+      )
+      return op_client.Get(op_ref)