feat(k8s): add support for GetNodeMetadata (scaleway#612)

Co-authored-by: Laure-di <[email protected]>

Author: scaleway-bot

scaleway-async/scaleway_async/k8s/v1/__init__.py

@@ -34,19 +34,23 @@
 from .types import Version
 from .types import Cluster
 from .types import Node
+from .types import NodeMetadataCoreV1Taint
 from .types import UpdateClusterRequestAutoUpgrade
 from .types import UpdateClusterRequestAutoscalerConfig
 from .types import UpdateClusterRequestOpenIDConnectConfig
 from .types import UpdatePoolRequestUpgradePolicy
+from .types import AuthExternalNodeRequest
 from .types import CreateClusterRequest
 from .types import CreateExternalNodeRequest
 from .types import CreatePoolRequest
 from .types import DeleteClusterRequest
 from .types import DeleteNodeRequest
 from .types import DeletePoolRequest
 from .types import ExternalNode
+from .types import ExternalNodeAuth
 from .types import GetClusterKubeConfigRequest
 from .types import GetClusterRequest
+from .types import GetNodeMetadataRequest
 from .types import GetNodeRequest
 from .types import GetPoolRequest
 from .types import GetVersionRequest
@@ -65,6 +69,7 @@
 from .types import ListVersionsRequest
 from .types import ListVersionsResponse
 from .types import MigrateClusterToRoutedIPsRequest
+from .types import NodeMetadata
 from .types import RebootNodeRequest
 from .types import ReplaceNodeRequest
 from .types import ResetClusterAdminTokenRequest
@@ -110,19 +115,23 @@
     "Version",
     "Cluster",
     "Node",
+    "NodeMetadataCoreV1Taint",
     "UpdateClusterRequestAutoUpgrade",
     "UpdateClusterRequestAutoscalerConfig",
     "UpdateClusterRequestOpenIDConnectConfig",
     "UpdatePoolRequestUpgradePolicy",
+    "AuthExternalNodeRequest",
     "CreateClusterRequest",
     "CreateExternalNodeRequest",
     "CreatePoolRequest",
     "DeleteClusterRequest",
     "DeleteNodeRequest",
     "DeletePoolRequest",
     "ExternalNode",
+    "ExternalNodeAuth",
     "GetClusterKubeConfigRequest",
     "GetClusterRequest",
+    "GetNodeMetadataRequest",
     "GetNodeRequest",
     "GetPoolRequest",
     "GetVersionRequest",
@@ -141,6 +150,7 @@
     "ListVersionsRequest",
     "ListVersionsResponse",
     "MigrateClusterToRoutedIPsRequest",
+    "NodeMetadata",
     "RebootNodeRequest",
     "ReplaceNodeRequest",
     "ResetClusterAdminTokenRequest",

scaleway-async/scaleway_async/k8s/v1/api.py

@@ -37,6 +37,7 @@
     CreatePoolRequest,
     CreatePoolRequestUpgradePolicy,
     ExternalNode,
+    ExternalNodeAuth,
     ListClusterAvailableTypesResponse,
     ListClusterAvailableVersionsResponse,
     ListClusterTypesResponse,
@@ -45,6 +46,7 @@
     ListPoolsResponse,
     ListVersionsResponse,
     Node,
+    NodeMetadata,
     Pool,
     SetClusterTypeRequest,
     UpdateClusterRequest,
@@ -68,13 +70,15 @@
     unmarshal_Cluster,
     unmarshal_Node,
     unmarshal_ExternalNode,
+    unmarshal_ExternalNodeAuth,
     unmarshal_ListClusterAvailableTypesResponse,
     unmarshal_ListClusterAvailableVersionsResponse,
     unmarshal_ListClusterTypesResponse,
     unmarshal_ListClustersResponse,
     unmarshal_ListNodesResponse,
     unmarshal_ListPoolsResponse,
     unmarshal_ListVersionsResponse,
+    unmarshal_NodeMetadata,
     marshal_CreateClusterRequest,
     marshal_CreatePoolRequest,
     marshal_SetClusterTypeRequest,
@@ -1151,6 +1155,70 @@ async def delete_pool(
         self._throw_on_error(res)
         return unmarshal_Pool(res.json())
 
+    async def get_node_metadata(
+        self,
+        *,
+        region: Optional[Region] = None,
+    ) -> NodeMetadata:
+        """
+        Fetch node metadata.
+        Rerieve metadata to instantiate a Kapsule/Kosmos node. This method is not intended to be called by end users but rather programmatically by the node-installer.
+        :param region: Region to target. If none is passed will use default region from the config.
+        :return: :class:`NodeMetadata <NodeMetadata>`
+
+        Usage:
+        ::
+
+            result = await api.get_node_metadata()
+        """
+
+        param_region = validate_path_param(
+            "region", region or self.client.default_region
+        )
+
+        res = self._request(
+            "GET",
+            f"/k8s/v1/regions/{param_region}/nodes/metadata",
+        )
+
+        self._throw_on_error(res)
+        return unmarshal_NodeMetadata(res.json())
+
+    async def auth_external_node(
+        self,
+        *,
+        pool_id: str,
+        region: Optional[Region] = None,
+    ) -> ExternalNodeAuth:
+        """
+        Authenticate Kosmos external node.
+        Creates a newer Kosmos node and returns its token. This method is not intended to be called by end users but rather programmatically by the node-installer.
+        :param pool_id: Pool the node will be attached to.
+        :param region: Region to target. If none is passed will use default region from the config.
+        :return: :class:`ExternalNodeAuth <ExternalNodeAuth>`
+
+        Usage:
+        ::
+
+            result = await api.auth_external_node(
+                pool_id="example",
+            )
+        """
+
+        param_region = validate_path_param(
+            "region", region or self.client.default_region
+        )
+        param_pool_id = validate_path_param("pool_id", pool_id)
+
+        res = self._request(
+            "POST",
+            f"/k8s/v1/regions/{param_region}/pools/{param_pool_id}/external-nodes/auth",
+            body={},
+        )
+
+        self._throw_on_error(res)
+        return unmarshal_ExternalNodeAuth(res.json())
+
     async def create_external_node(
         self,
         *,

scaleway-async/scaleway_async/k8s/v1/marshalling.py

@@ -23,6 +23,7 @@
     Node,
     ExternalNodeCoreV1Taint,
     ExternalNode,
+    ExternalNodeAuth,
     ClusterType,
     ListClusterAvailableTypesResponse,
     ListClusterAvailableVersionsResponse,
@@ -31,6 +32,8 @@
     ListNodesResponse,
     ListPoolsResponse,
     ListVersionsResponse,
+    NodeMetadataCoreV1Taint,
+    NodeMetadata,
     CreateClusterRequestPoolConfigUpgradePolicy,
     CreateClusterRequestAutoUpgrade,
     CreateClusterRequestAutoscalerConfig,
@@ -652,6 +655,25 @@ def unmarshal_ExternalNode(data: Any) -> ExternalNode:
     return ExternalNode(**args)
 
 
+def unmarshal_ExternalNodeAuth(data: Any) -> ExternalNodeAuth:
+    if not isinstance(data, dict):
+        raise TypeError(
+            "Unmarshalling the type 'ExternalNodeAuth' failed as data isn't a dictionary."
+        )
+
+    args: Dict[str, Any] = {}
+
+    field = data.get("node_token", None)
+    if field is not None:
+        args["node_token"] = field
+
+    field = data.get("api_url", None)
+    if field is not None:
+        args["api_url"] = field
+
+    return ExternalNodeAuth(**args)
+
+
 def unmarshal_ClusterType(data: Any) -> ClusterType:
     if not isinstance(data, dict):
         raise TypeError(
@@ -844,6 +866,100 @@ def unmarshal_ListVersionsResponse(data: Any) -> ListVersionsResponse:
     return ListVersionsResponse(**args)
 
 
+def unmarshal_NodeMetadataCoreV1Taint(data: Any) -> NodeMetadataCoreV1Taint:
+    if not isinstance(data, dict):
+        raise TypeError(
+            "Unmarshalling the type 'NodeMetadataCoreV1Taint' failed as data isn't a dictionary."
+        )
+
+    args: Dict[str, Any] = {}
+
+    field = data.get("key", None)
+    if field is not None:
+        args["key"] = field
+
+    field = data.get("value", None)
+    if field is not None:
+        args["value"] = field
+
+    field = data.get("effect", None)
+    if field is not None:
+        args["effect"] = field
+
+    return NodeMetadataCoreV1Taint(**args)
+
+
+def unmarshal_NodeMetadata(data: Any) -> NodeMetadata:
+    if not isinstance(data, dict):
+        raise TypeError(
+            "Unmarshalling the type 'NodeMetadata' failed as data isn't a dictionary."
+        )
+
+    args: Dict[str, Any] = {}
+
+    field = data.get("id", None)
+    if field is not None:
+        args["id"] = field
+
+    field = data.get("name", None)
+    if field is not None:
+        args["name"] = field
+
+    field = data.get("cluster_url", None)
+    if field is not None:
+        args["cluster_url"] = field
+
+    field = data.get("cluster_ca", None)
+    if field is not None:
+        args["cluster_ca"] = field
+
+    field = data.get("credential_provider_config", None)
+    if field is not None:
+        args["credential_provider_config"] = field
+
+    field = data.get("pool_version", None)
+    if field is not None:
+        args["pool_version"] = field
+
+    field = data.get("kubelet_config", None)
+    if field is not None:
+        args["kubelet_config"] = field
+
+    field = data.get("node_labels", None)
+    if field is not None:
+        args["node_labels"] = field
+
+    field = data.get("node_taints", None)
+    if field is not None:
+        args["node_taints"] = (
+            [unmarshal_NodeMetadataCoreV1Taint(v) for v in field]
+            if field is not None
+            else None
+        )
+
+    field = data.get("private_network_mode", None)
+    if field is not None:
+        args["private_network_mode"] = field
+
+    field = data.get("kapsule_iface_mac", None)
+    if field is not None:
+        args["kapsule_iface_mac"] = field
+
+    field = data.get("full_isolation", None)
+    if field is not None:
+        args["full_isolation"] = field
+
+    field = data.get("has_gpu", None)
+    if field is not None:
+        args["has_gpu"] = field
+
+    field = data.get("external_ip", None)
+    if field is not None:
+        args["external_ip"] = field
+
+    return NodeMetadata(**args)
+
+
 def marshal_MaintenanceWindow(
     request: MaintenanceWindow,
     defaults: ProfileDefaults,