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

gcloud auth application-default login/google-cloud-sdk/lib/googlecloudsdk/api_lib/vmware/sddc/clusters.py

@@ -0,0 +1,85 @@
+# -*- coding: utf-8 -*- # # Copyright 2020 Google LLC. All Rights Reserved.
+# Copyright 2020 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.
+"""Cloud vmware sddc Clusters client."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from apitools.base.py import list_pager
+from googlecloudsdk.api_lib.vmware.sddc import util
+from googlecloudsdk.command_lib.vmware.sddc import flags
+
+
+class ClustersClient(util.VmwareClientBase):
+  """cloud vmware Clusters client."""
+
+  def __init__(self):
+    super(ClustersClient, self).__init__()
+    self.service = self.client.projects_locations_clusterGroups_clusters
+
+  def Get(self, resource):
+    request = self.messages.SddcProjectsLocationsClusterGroupsClustersGetRequest(
+        name=resource.RelativeName())
+    return self.service.Get(request)
+
+  def Create(self, resource, node_count, node_type, zone, labels=None):
+    parent = resource.Parent().RelativeName()
+    cluster_id = resource.Name()
+    cluster = self.messages.Cluster(
+        nodeCount=node_count, defaultZone=zone, nodeType=node_type)
+    flags.AddLabelsToMessage(labels, cluster)
+
+    request = self.messages.SddcProjectsLocationsClusterGroupsClustersCreateRequest(
+        parent=parent,
+        cluster=cluster,
+        clusterId=cluster_id,
+        managementCluster=True)
+
+    return self.service.Create(request)
+
+  def Delete(self, resource):
+    request = self.messages.SddcProjectsLocationsClusterGroupsClustersDeleteRequest(
+        name=resource.RelativeName())
+    return self.service.Delete(request)
+
+  def List(self, cluster_group_resource):
+    cluster_group = cluster_group_resource.RelativeName()
+    request = (
+        self.messages.SddcProjectsLocationsClusterGroupsClustersListRequest(
+            parent=cluster_group
+        )
+    )
+    return list_pager.YieldFromList(
+        self.service,
+        request,
+        batch_size_attribute='pageSize',
+        field='clusters')
+
+  def AddNodes(self, resource, node_count):
+    cluster = self.Get(resource)
+    request = self.messages.SddcProjectsLocationsClusterGroupsClustersAddNodesRequest(
+        cluster=resource.RelativeName(),
+        addNodesRequest=self.messages.AddNodesRequest(
+            nodeCount=cluster.nodeCount + node_count))
+    return self.service.AddNodes(request)
+
+  def RemoveNodes(self, resource, node_count):
+    cluster = self.Get(resource)
+    request = self.messages.SddcProjectsLocationsClusterGroupsClustersRemoveNodesRequest(
+        cluster=resource.RelativeName(),
+        removeNodesRequest=self.messages.RemoveNodesRequest(
+            nodeCount=cluster.nodeCount - node_count))
+    return self.service.RemoveNodes(request)

gcloud auth application-default login/google-cloud-sdk/lib/googlecloudsdk/api_lib/vmware/sddc/ipaddresses.py

@@ -0,0 +1,87 @@
+# -*- coding: utf-8 -*- # # Copyright 2020 Google LLC. All Rights Reserved.
+# Copyright 2020 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.
+"""Cloud vmware IPAdresses client."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from apitools.base.py import list_pager
+from googlecloudsdk.api_lib.vmware.sddc import util
+from googlecloudsdk.command_lib.vmware.sddc import flags
+import six.moves.urllib.parse
+
+
+class IPAddressesClient(util.VmwareClientBase):
+  """cloud vmware sddc ip addresses client."""
+
+  def __init__(self):
+    super(IPAddressesClient, self).__init__()
+    self.service = self.client.projects_locations_clusterGroups_ipAddresses
+
+  def Create(self, resource, internal_ip, labels=None):
+    ip_address = self.messages.IpAddress(internalIp=internal_ip)
+    flags.AddLabelsToMessage(labels, ip_address)
+
+    request = self.messages.SddcProjectsLocationsClusterGroupsIpAddressesCreateRequest(
+        ipAddress=ip_address,
+        ipAddressId=resource.Name(),
+        parent=self.GetResourcePath(
+            resource, resource_path=resource.Parent().RelativeName()))
+
+    return self.service.Create(request)
+
+  def Delete(self, resource):
+    request = self.messages.SddcProjectsLocationsClusterGroupsIpAddressesDeleteRequest(
+        name=self.GetResourcePath(
+            resource, resource_path=resource.RelativeName()))
+    return self.service.Delete(request)
+
+  def Get(self, resource):
+    request = self.messages.SddcProjectsLocationsClusterGroupsIpAddressesGetRequest(
+        name=self.GetResourcePath(
+            resource, resource_path=resource.RelativeName()))
+    return self.service.Get(request)
+
+  def GetResourcePath(self,
+                      resource,
+                      resource_path,
+                      encoded_cluster_groups_id=False):
+    result = six.text_type(resource_path)
+
+    if '/' not in resource.clusterGroupsId:
+      return result
+
+    cluster_groups_id = resource.clusterGroupsId.split('/').pop()
+    cluster_groups_id_path = six.text_type(resource.clusterGroupsId)
+
+    if encoded_cluster_groups_id:
+      cluster_groups_id_path = six.moves.urllib.parse.quote(
+          cluster_groups_id_path, safe='')
+
+    return result.replace(cluster_groups_id_path, cluster_groups_id)
+
+  def List(self, resource):
+    ip_name = resource.RelativeName()
+    request = (
+        self.messages.SddcProjectsLocationsClusterGroupsIpAddressesListRequest(
+            parent=ip_name
+        )
+    )
+    return list_pager.YieldFromList(
+        self.service,
+        request,
+        batch_size_attribute='pageSize',
+        field='ipAddresses')

gcloud auth application-default login/google-cloud-sdk/lib/googlecloudsdk/api_lib/vmware/sddc/privateclouds.py

@@ -0,0 +1,97 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2020 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.
+"""Cloud vmware sddc Privateclouds client."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from apitools.base.py import list_pager
+from googlecloudsdk.api_lib.vmware.sddc import util
+from googlecloudsdk.command_lib.vmware.sddc import flags
+
+
+class PrivatecloudsClient(util.VmwareClientBase):
+  """cloud vmware privateclouds client."""
+
+  def __init__(self):
+    super(PrivatecloudsClient, self).__init__()
+    self.service = self.client.projects_locations_clusterGroups
+
+  def Get(self, resource):
+    request = self.messages.SddcProjectsLocationsClusterGroupsGetRequest(
+        name=resource.RelativeName())
+    return self.service.Get(request)
+
+  def Create(
+      self,
+      resource,
+      vpc_network,
+      management_ip_range,
+      workload_ip_range,
+      labels=None,
+      description=None,
+  ):
+    parent = resource.Parent().RelativeName()
+    cluster_group_id = resource.Name()
+    cluster_group = self.messages.ClusterGroup(description=description)
+    flags.AddLabelsToMessage(labels, cluster_group)
+    network_config = self.messages.NetworkConfig(
+        network=vpc_network,
+        managementCidr=management_ip_range,
+        workloadCidr=workload_ip_range)
+    cluster_group.networkConfig = network_config
+    request = self.messages.SddcProjectsLocationsClusterGroupsCreateRequest(
+        parent=parent,
+        clusterGroup=cluster_group,
+        clusterGroupId=cluster_group_id)
+    return self.service.Create(request)
+
+  def Update(self,
+             resource,
+             labels=None,
+             description=None,
+             external_ip_access=None):
+    cluster_group = self.Get(resource)
+    update_mask = ['labels']
+    if labels is not None:
+      flags.AddLabelsToMessage(labels, cluster_group)
+    if description is not None:
+      cluster_group.description = description
+      update_mask.append('description')
+    if external_ip_access is not None:
+      cluster_group.networkConfig.externalIpAccess = external_ip_access
+      update_mask.append('network_config.external_ip_access')
+    request = self.messages.SddcProjectsLocationsClusterGroupsPatchRequest(
+        clusterGroup=cluster_group,
+        name=resource.RelativeName(),
+        updateMask=','.join(update_mask))
+    return self.service.Patch(request)
+
+  def Delete(self, resource):
+    request = self.messages.SddcProjectsLocationsClusterGroupsDeleteRequest(
+        name=resource.RelativeName())
+    return self.service.Delete(request)
+
+  def List(self, location_resource):
+    location = location_resource.RelativeName()
+    request = self.messages.SddcProjectsLocationsClusterGroupsListRequest(
+        parent=location
+    )
+    return list_pager.YieldFromList(
+        self.service,
+        request,
+        batch_size_attribute='pageSize',
+        field='clusterGroups')

gcloud auth application-default login/google-cloud-sdk/lib/googlecloudsdk/api_lib/vmware/sddc/util.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2020 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.
+"""Cloud vmware sddc API utilities."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.util import apis
+from googlecloudsdk.api_lib.util import waiter
+from googlecloudsdk.core import resources
+
+_DEFAULT_API_VERSION = 'v1alpha1'
+
+
+class VmwareClientBase(object):
+  """Base class for vwmare API client wrappers."""
+
+  def __init__(self, api_version=_DEFAULT_API_VERSION):
+    self._client = apis.GetClientInstance('sddc', api_version)
+    self._messages = apis.GetMessagesModule('sddc', api_version)
+    self.service = None
+    self.operations_service = self.client.projects_locations_operations
+
+  @property
+  def client(self):
+    return self._client
+
+  @property
+  def messages(self):
+    return self._messages
+
+  def WaitForOperation(self, operation, message, is_delete=False):
+    operation_ref = resources.REGISTRY.Parse(
+        operation.name, collection='sddc.projects.locations.operations')
+    if is_delete:
+      poller = waiter.CloudOperationPollerNoResources(self.operations_service)
+    else:
+      poller = waiter.CloudOperationPoller(self.service,
+                                           self.operations_service)
+    return waiter.WaitFor(poller, operation_ref, message)