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

gcloud auth application-default login/google-cloud-sdk/lib/surface/mps/instances/__init__.py

@@ -0,0 +1,30 @@
+# -*- 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.
+
+"""The instances command group for the Marketplace Solutions CLI."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.calliope import base
+
+
+@base.Hidden
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class Instances(base.Group):
+  """Manage instances in MarketplaceSolutions."""
+
+  pass

gcloud auth application-default login/google-cloud-sdk/lib/surface/mps/instances/create.py

@@ -0,0 +1,92 @@
+# -*- 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.
+"""Marketplace Solution instance create command."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.mps.mps_client import MpsClient
+from googlecloudsdk.api_lib.util import waiter
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.mps import flags
+from googlecloudsdk.core import log
+from googlecloudsdk.core import properties
+from googlecloudsdk.core import resources
+
+DETAILED_HELP = {
+    'DESCRIPTION':
+        """
+          Create a Marketplace Solution Instance.
+        """,
+    'EXAMPLES':
+        """
+          To create an instance called ``my-instance'' in region ``us-central1'', with
+          requested boot image of AIX72_ORD_Cloud, 2 Gib of memory, an s922 system type,
+          a shared core type, and 0.25 cores, run:
+
+          $ {command} my-instance  --region=us-central1 --boot-image-name=AIX72_ORD_Cloud --memory-gib=2 --network-attachment-name=dev-net --system-type=s922 --virtual-cpu-type=UNCAPPED_SHARED --virtual-cpu-cores=0.25
+
+    """,
+}
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class Create(base.CreateCommand):
+  """Create a Marketplace Solution Instance."""
+
+  @staticmethod
+  def Args(parser):
+    """Register flags for this command."""
+    flags.AddInstanceArgToParser(parser, positional=True)
+    flags.AddInstanceBootImageNameArgToParse(parser=parser)
+    flags.AddInstanceMemoryGibArgToParse(parser=parser)
+    flags.AddInstanceNetworkAttachmentNameArgToParse(parser=parser)
+    flags.AddInstanceSystemTypeArgToParse(parser=parser)
+    flags.AddInstanceVirtualCpuCoresArgToParse(parser=parser)
+    flags.AddInstanceVirtualCpuTypeArgToParse(parser=parser)
+
+  def Run(self, args):
+    # pylint: disable=line-too-long
+    instance = args.CONCEPTS.instance.Parse()
+    client = MpsClient()
+    product = properties.VALUES.mps.product.Get(required=True)
+
+    op_ref = client.CreateInstance(product,
+                                   instance_resource=instance,
+                                   boot_image_name=args.boot_image_name,
+                                   system_type=args.system_type,
+                                   memory_gib=args.memory_gib,
+                                   network_attachment_names=args.network_attachment_name,
+                                   virtual_cpu_cores=args.virtual_cpu_cores,
+                                   virtual_cpu_type=args.virtual_cpu_type)
+
+    if op_ref.done:
+      log.CreatedResource(instance.Name(), kind='Instance')
+      return op_ref
+
+    op_resource = resources.REGISTRY.ParseRelativeName(
+        op_ref.name,
+        collection='marketplacesolutions.projects.locations.operations',
+        api_version='v1alpha1')
+    poller = waiter.CloudOperationPollerNoResources(client.operation_service)
+    res = waiter.WaitFor(
+        poller, op_resource,
+        'Waiting for operation [{}] to complete'.format(op_ref.name))
+    log.CreatedResource(instance.Name(), kind='Instance')
+    return res
+
+
+Create.detailed_help = DETAILED_HELP

gcloud auth application-default login/google-cloud-sdk/lib/surface/mps/instances/delete.py

@@ -0,0 +1,75 @@
+# -*- 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.
+"""Marketplace Solution instance delete command."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.mps.mps_client import MpsClient
+from googlecloudsdk.api_lib.util import waiter
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.mps import flags
+from googlecloudsdk.core import log
+from googlecloudsdk.core import properties
+from googlecloudsdk.core import resources
+
+DETAILED_HELP = {
+    'DESCRIPTION':
+        """
+          Delete a Marketplace Solution instance.
+        """,
+    'EXAMPLES':
+        """
+          To delete an instance called ``my-instance'' in region
+          ``us-central1'', run:
+
+          $ {command} my-instance  --region=us-central1
+        """,
+}
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class Delete(base.DeleteCommand):
+  """Delete a Marketplace Solution instance."""
+
+  @staticmethod
+  def Args(parser):
+    """Register flags for this command."""
+    flags.AddInstanceArgToParser(parser, positional=True)
+
+  def Run(self, args):
+    instance = args.CONCEPTS.instance.Parse()
+    client = MpsClient()
+    product = properties.VALUES.mps.product.Get(required=True)
+
+    op_ref = client.DeleteInstance(product, instance)
+    if op_ref.done:
+      log.DeletedResource(instance.Name(), kind='Instance')
+      return op_ref
+
+    op_resource = resources.REGISTRY.ParseRelativeName(
+        op_ref.name,
+        collection='marketplacesolutions.projects.locations.operations',
+        api_version='v1alpha1')
+    poller = waiter.CloudOperationPollerNoResources(client.operation_service)
+    res = waiter.WaitFor(
+        poller, op_resource,
+        'Waiting for operation [{}] to complete'.format(op_ref.name))
+    log.DeletedResource(instance.Name(), kind='Instance')
+    return res
+
+
+Delete.detailed_help = DETAILED_HELP

gcloud auth application-default login/google-cloud-sdk/lib/surface/mps/instances/describe.py

@@ -0,0 +1,54 @@
+# -*- 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.
+"""'Marketplace Solutions instances describe command."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.mps.mps_client import MpsClient
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.mps import flags
+from googlecloudsdk.core import properties
+
+
+DETAILED_HELP = {
+    'DESCRIPTION': """
+          Describe a Marketplace Solutions instance.
+        """,
+    'EXAMPLES': """
+          To get a description of an instance called ``my-instance'' in
+          project ``my-project'' and region ``us-central1'', run:
+
+          $ {command} my-instance --project=my-project --region=us-central1
+    """,
+}
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class Describe(base.DescribeCommand):
+  """Describe a Marketplace Solutions instance."""
+  detailed_help = DETAILED_HELP
+
+  @staticmethod
+  def Args(parser):
+    """Register flags for this command."""
+    flags.AddInstanceArgToParser(parser, positional=True)
+
+  def Run(self, args):
+    instance = args.CONCEPTS.instance.Parse()
+    product = properties.VALUES.mps.product.Get(required=True)
+    client = MpsClient()
+    return client.GetInstance(product, instance)

gcloud auth application-default login/google-cloud-sdk/lib/surface/mps/instances/list.py

@@ -0,0 +1,102 @@
+# -*- 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.
+
+"""'Marketplace Solutions instances list command."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+import json
+
+from googlecloudsdk.api_lib.mps.mps_client import MpsClient
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.mps import flags
+from googlecloudsdk.core import properties
+from googlecloudsdk.core.resource import resource_projector
+
+DETAILED_HELP = {
+    'DESCRIPTION':
+        """
+          List Marketplace Solutions instances in a project.
+        """,
+    'EXAMPLES':
+        """
+          To list instances in the region within the project ``us-central1'', run:
+
+            $ {command} --region=us-central1
+
+          Or:
+
+          To list all instances in the project, run:
+
+            $ {command}
+    """,
+}
+
+INSTANCE_LIST_FORMAT = """ table(
+        name.segment(-1):label=NAME,
+        name.segment(-5):label=PROJECT,
+        name.segment(-3):label=REGION,
+        systemType,
+        virtualCpuCores,
+        memoryGib,
+        osImage,
+        state
+    )"""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class List(base.ListCommand):
+  """List Marketplace Solution instances in a project."""
+  detailed_help = DETAILED_HELP
+
+  @staticmethod
+  def Args(parser):
+    """Register flags for this command."""
+    # Remove unsupported default List flags.
+    base.FILTER_FLAG.RemoveFromParser(parser)
+    base.PAGE_SIZE_FLAG.RemoveFromParser(parser)
+    base.SORT_BY_FLAG.RemoveFromParser(parser)
+    base.URI_FLAG.RemoveFromParser(parser)
+
+    flags.AddRegionArgToParser(parser)
+    # The default format picks out the components of the relative name:
+    # given projects/myproject/locations/us-central1/clusterGroups/my-test
+    # it takes -1 (my-test), -3 (us-central1), and -5 (myproject).
+
+    parser.display_info.AddFormat(INSTANCE_LIST_FORMAT)
+
+  def Run(self, args):
+
+    region = args.CONCEPTS.region.Parse()
+    client = MpsClient()
+    product = properties.VALUES.mps.product.Get(required=True)
+
+    if region is None:
+      project = properties.VALUES.core.project.Get(required=True)
+      return (self.synthesizesInstanceInfo(ins)
+              for ins in client.AggregateListInstances(
+                  project, product, limit=args.limit))
+    return (self.synthesizesInstanceInfo(ins)
+            for ins in client.ListInstances(product, region))
+
+  def synthesizesInstanceInfo(self, ins):
+    out = resource_projector.MakeSerializable(ins)
+    out['osImage'] = ins.osImage.version
+    # We dump jsons here because when we use the built-in serialization, it
+    # sometimes adds a 'u' character before the strings and the tests break.
+    out['osImage'] = json.dumps(out['osImage'])
+    return out

gcloud auth application-default login/google-cloud-sdk/lib/surface/mps/instances/update.py

@@ -0,0 +1,93 @@
+# -*- 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.
+"""Bare Metal Solution instance update command."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.mps.mps_client import MpsClient
+from googlecloudsdk.api_lib.util import waiter
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.mps import flags
+from googlecloudsdk.core import exceptions
+from googlecloudsdk.core import log
+from googlecloudsdk.core import properties
+from googlecloudsdk.core import resources
+
+DETAILED_HELP = {
+    'DESCRIPTION':
+        """
+          Update a Marketplace Solution instance.
+
+          This call returns immediately, but the update operation may take
+          several minutes to complete. To check if the operation is complete,
+          use the `describe` command for the instance.
+        """,
+    'EXAMPLES':
+        """
+          To update an instance called ``my-instance'' in region ``us-central1'',
+          to 3 memoryGib and 0.5 virtualCpuCores, run:
+
+          $ {command} my-instance update --region=us-central1 --memory_gib=3
+          --virtual_cpu_cores=0.5
+        """,
+}
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class Update(base.UpdateCommand):
+  """Update a Marketplace Solution instance."""
+
+  @staticmethod
+  def Args(parser):
+    """Register flags for this command."""
+    flags.AddInstanceArgToParser(parser, positional=True)
+    flags.AddInstanceMemoryGibArgToParse(parser, required=False)
+    flags.AddInstanceVirtualCpuCoresArgToParse(parser, required=False)
+
+  def Run(self, args):
+    client = MpsClient()
+    instance = args.CONCEPTS.instance.Parse()
+    product = properties.VALUES.mps.product.Get(required=True)
+
+    memory_gib = getattr(args, 'memory_gib', None)
+    virtual_cpu_cores = getattr(args, 'virtual_cpu_cores', None)
+    if memory_gib is None and virtual_cpu_cores is None:
+      raise exceptions.Error('At least one of '
+                             '`--memory-gib` or'
+                             '`--virtual-cpu-cores` '
+                             'is required')
+    op_ref = client.UpdateInstance(
+        product=product,
+        instance_resource=instance, memory_gib=memory_gib,
+        virtual_cpu_cores=virtual_cpu_cores)
+
+    if op_ref.done:
+      log.UpdatedResource(instance.Name(), kind='Instance')
+      return op_ref
+
+    op_resource = resources.REGISTRY.ParseRelativeName(
+        op_ref.name,
+        collection='marketplacesolutions.projects.locations.operations',
+        api_version='v1alpha1')
+    poller = waiter.CloudOperationPollerNoResources(client.operation_service)
+    res = waiter.WaitFor(
+        poller, op_resource,
+        'Waiting for operation [{}] to complete'.format(op_ref.name))
+    log.UpdatedResource(instance.Name(), kind='instance')
+    return res
+
+Update.detailed_help = DETAILED_HELP