Feat/karbon clusters update

meta/runtime.yml

@@ -95,3 +95,4 @@ action_groups:
   - ntnx_ndb_maintenance_window
   - ntnx_ndb_maintenance_windows_info
   - ntnx_ndb_slas
+  - ntnx_karbon_clusters_node_pools

plugins/module_utils/karbon/node_pools.py

@@ -0,0 +1,176 @@
+# This file is part of Ansible
+# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
+from __future__ import absolute_import, division, print_function
+
+__metaclass__ = type
+
+from copy import deepcopy
+
+from ..prism.subnets import get_subnet_uuid
+from .clusters import Cluster
+
+
+class NodePool(Cluster):
+    kind = "cluster"
+
+    def __init__(self, module, resource_type="/v1-alpha.1/k8s/clusters"):
+        super(NodePool, self).__init__(module, resource_type=resource_type)
+        self.build_spec_methods = {}
+
+    def _get_default_pool_spec(self):
+        return deepcopy(
+            {
+                "name": "",
+                "num_instances": 0,
+                "ahv_config": {
+                    "cpu": 0,
+                    "disk_mib": 0,
+                    "memory_mib": 0,
+                    "network_uuid": "",
+                    "iscsi_network_uuid": "",
+                },
+            }
+        )
+
+    def _build_pool_spec(self, payload, config):
+        payload["name"] = config["node_pool_name"]
+        if config.get("pool_config"):
+            payload["num_instances"] = config["pool_config"]["num_instances"]
+            payload["ahv_config"]["cpu"] = config["pool_config"]["cpu"]
+            payload["ahv_config"]["memory_mib"] = (
+                config["pool_config"]["memory_gb"] * 1024
+            )
+            payload["ahv_config"]["disk_mib"] = config["pool_config"]["disk_gb"] * 1024
+        subnet_uuid, err = get_subnet_uuid(config["node_subnet"], self.module)
+        if err:
+            return None, err
+        payload["ahv_config"]["network_uuid"] = subnet_uuid
+        if config.get("node_iscsi_subnet"):
+            iscsi_subnet_uuid, err = get_subnet_uuid(
+                config["node_iscsi_subnet"], self.module
+            )
+            if err:
+                return None, err
+            payload["ahv_config"]["iscsi_network_uuid"] = iscsi_subnet_uuid
+
+        return payload, None
+
+    def get_pool_spec(self):
+        default_pool_spec = self._get_default_pool_spec()
+        error = self.validate_pool_resources()
+        if error:
+            return None, error
+        spec, error = self._build_pool_spec(default_pool_spec, self.module.params)
+        if error:
+            return None, error
+        return spec, None
+
+    def get_labels_spec(self):
+        return deepcopy(
+            {
+                "add_labels": self.module.params.get("add_labels"),
+                "remove_labels": self.module.params.get("remove_labels"),
+            }
+        )
+
+    def add_node_pool(self, cluster_name, data=None):
+
+        endpoint = "add-node-pool"
+        resp = self.update(
+            data=data,
+            uuid=cluster_name,
+            method="POST",
+            endpoint=endpoint,
+        )
+        return resp
+
+    def get_nodes_count(self, cluster_name, pool_name):
+        nodes_count = 0
+        pool = self.get_node_pool(cluster_name, pool_name)
+        if pool:
+            nodes_count = len(pool.get("nodes"))
+        return nodes_count
+
+    def remove_pool_nodes(self, cluster_name, pool_name):
+        nodes_count = self.get_nodes_count(cluster_name, pool_name)
+        resp = {}
+        if nodes_count:
+            spec = {"count": nodes_count}
+            endpoint = "node-pools/{0}/remove-nodes".format(pool_name)
+            resp = self.update(
+                data=spec, uuid=cluster_name, endpoint=endpoint, method="POST"
+            )
+        return resp
+
+    def remove_node_pool(self, cluster_name, pool_name):
+
+        endpoint = "node-pools/{0}".format(pool_name)
+        resp = self.delete(
+            uuid=cluster_name,
+            endpoint=endpoint,
+        )
+        return resp
+
+    def read_node_pools(self, cluster_name):
+
+        endpoint = "node-pools"
+        resp = self.read(
+            uuid=cluster_name,
+            endpoint=endpoint,
+        )
+        return resp
+
+    def get_node_pool(self, cluster_name, pool_name):
+        node_pools = self.read_node_pools(cluster_name)
+        for pool in node_pools:
+            if pool.get("name") == pool_name:
+                return pool
+        return None
+
+    def update_nodes_count(self, cluster_name, pool_name, actual_count, expected_count):
+        residual_count = expected_count - actual_count
+        spec = {"count": abs(residual_count)}
+        if residual_count > 0:
+            resp = self.add_node(cluster_name, pool_name, spec)
+        else:
+            resp = self.remove_node(cluster_name, pool_name, spec)
+        return resp
+
+    def add_node(self, cluster_name, pool_name, data=None):
+
+        endpoint = "node-pools/{0}/add-nodes".format(pool_name)
+        resp = self.update(
+            data=data,
+            uuid=cluster_name,
+            endpoint=endpoint,
+            method="POST",
+        )
+        return resp
+
+    def remove_node(self, cluster_name, pool_name, data=None):
+
+        endpoint = "node-pools/{0}/remove-nodes".format(pool_name)
+        resp = self.update(
+            data=data,
+            uuid=cluster_name,
+            endpoint=endpoint,
+            method="POST",
+        )
+        return resp
+
+    def update_labels(self, cluster_name, pool_name, data=None, raise_error=True):
+
+        endpoint = "node-pools/{0}/update-labels".format(pool_name)
+        resp = self.update(
+            data=data,
+            uuid=cluster_name,
+            endpoint=endpoint,
+            method="POST",
+            raise_error=raise_error,
+        )
+        return resp
+
+    def validate_pool_resources(self):
+        if not self.module.params.get("node_subnet"):
+            return "missing required arguments: node_subnet"
+        return None