feat(baremetal): support overriding default SSH keys on rescue (#1205)

Author: scaleway-bot

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

@@ -529,13 +529,15 @@ async def reboot_server(
         server_id: str,
         zone: Optional[ScwZone] = None,
         boot_type: Optional[ServerBootType] = None,
+        ssh_key_ids: Optional[List[str]] = None,
     ) -> Server:
         """
         Reboot an Elastic Metal server.
         Reboot the Elastic Metal server associated with the ID, use the `boot_type` `rescue` to reboot the server in rescue mode.
         :param server_id: ID of the server to reboot.
         :param zone: Zone to target. If none is passed will use default zone from the config.
         :param boot_type: The type of boot.
+        :param ssh_key_ids: Additional SSH public key IDs to configure on rescue image.
         :return: :class:`Server <Server>`
 
         Usage:
@@ -557,6 +559,7 @@ async def reboot_server(
                     server_id=server_id,
                     zone=zone,
                     boot_type=boot_type,
+                    ssh_key_ids=ssh_key_ids,
                 ),
                 self.client,
             ),
@@ -571,13 +574,15 @@ async def start_server(
         server_id: str,
         zone: Optional[ScwZone] = None,
         boot_type: Optional[ServerBootType] = None,
+        ssh_key_ids: Optional[List[str]] = None,
     ) -> Server:
         """
         Start an Elastic Metal server.
         Start the server associated with the ID.
         :param server_id: ID of the server to start.
         :param zone: Zone to target. If none is passed will use default zone from the config.
         :param boot_type: The type of boot.
+        :param ssh_key_ids: Additional SSH public key IDs to configure on rescue image.
         :return: :class:`Server <Server>`
 
         Usage:
@@ -599,6 +604,7 @@ async def start_server(
                     server_id=server_id,
                     zone=zone,
                     boot_type=boot_type,
+                    ssh_key_ids=ssh_key_ids,
                 ),
                 self.client,
             ),

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

@@ -2001,6 +2001,9 @@ def marshal_RebootServerRequest(
     if request.boot_type is not None:
         output["boot_type"] = request.boot_type
 
+    if request.ssh_key_ids is not None:
+        output["ssh_key_ids"] = request.ssh_key_ids
+
     return output
 
 
@@ -2025,6 +2028,9 @@ def marshal_StartServerRequest(
     if request.boot_type is not None:
         output["boot_type"] = request.boot_type
 
+    if request.ssh_key_ids is not None:
+        output["ssh_key_ids"] = request.ssh_key_ids
+
     return output
 
 

scaleway-async/scaleway_async/baremetal/v1/types.py

@@ -1725,6 +1725,11 @@ class RebootServerRequest:
     The type of boot.
     """
 
+    ssh_key_ids: Optional[List[str]] = field(default_factory=list)
+    """
+    Additional SSH public key IDs to configure on rescue image.
+    """
+
 
 @dataclass
 class SetServerPrivateNetworksResponse:
@@ -1766,6 +1771,11 @@ class StartServerRequest:
     The type of boot.
     """
 
+    ssh_key_ids: Optional[List[str]] = field(default_factory=list)
+    """
+    Additional SSH public key IDs to configure on rescue image.
+    """
+
 
 @dataclass
 class StopBMCAccessRequest:

scaleway/scaleway/baremetal/v1/api.py

@@ -529,13 +529,15 @@ def reboot_server(
         server_id: str,
         zone: Optional[ScwZone] = None,
         boot_type: Optional[ServerBootType] = None,
+        ssh_key_ids: Optional[List[str]] = None,
     ) -> Server:
         """
         Reboot an Elastic Metal server.
         Reboot the Elastic Metal server associated with the ID, use the `boot_type` `rescue` to reboot the server in rescue mode.
         :param server_id: ID of the server to reboot.
         :param zone: Zone to target. If none is passed will use default zone from the config.
         :param boot_type: The type of boot.
+        :param ssh_key_ids: Additional SSH public key IDs to configure on rescue image.
         :return: :class:`Server <Server>`
 
         Usage:
@@ -557,6 +559,7 @@ def reboot_server(
                     server_id=server_id,
                     zone=zone,
                     boot_type=boot_type,
+                    ssh_key_ids=ssh_key_ids,
                 ),
                 self.client,
             ),
@@ -571,13 +574,15 @@ def start_server(
         server_id: str,
         zone: Optional[ScwZone] = None,
         boot_type: Optional[ServerBootType] = None,
+        ssh_key_ids: Optional[List[str]] = None,
     ) -> Server:
         """
         Start an Elastic Metal server.
         Start the server associated with the ID.
         :param server_id: ID of the server to start.
         :param zone: Zone to target. If none is passed will use default zone from the config.
         :param boot_type: The type of boot.
+        :param ssh_key_ids: Additional SSH public key IDs to configure on rescue image.
         :return: :class:`Server <Server>`
 
         Usage:
@@ -599,6 +604,7 @@ def start_server(
                     server_id=server_id,
                     zone=zone,
                     boot_type=boot_type,
+                    ssh_key_ids=ssh_key_ids,
                 ),
                 self.client,
             ),

scaleway/scaleway/baremetal/v1/marshalling.py

@@ -2001,6 +2001,9 @@ def marshal_RebootServerRequest(
     if request.boot_type is not None:
         output["boot_type"] = request.boot_type
 
+    if request.ssh_key_ids is not None:
+        output["ssh_key_ids"] = request.ssh_key_ids
+
     return output
 
 
@@ -2025,6 +2028,9 @@ def marshal_StartServerRequest(
     if request.boot_type is not None:
         output["boot_type"] = request.boot_type
 
+    if request.ssh_key_ids is not None:
+        output["ssh_key_ids"] = request.ssh_key_ids
+
     return output
 
 

scaleway/scaleway/baremetal/v1/types.py

@@ -1725,6 +1725,11 @@ class RebootServerRequest:
     The type of boot.
     """
 
+    ssh_key_ids: Optional[List[str]] = field(default_factory=list)
+    """
+    Additional SSH public key IDs to configure on rescue image.
+    """
+
 
 @dataclass
 class SetServerPrivateNetworksResponse:
@@ -1766,6 +1771,11 @@ class StartServerRequest:
     The type of boot.
     """
 
+    ssh_key_ids: Optional[List[str]] = field(default_factory=list)
+    """
+    Additional SSH public key IDs to configure on rescue image.
+    """
+
 
 @dataclass
 class StopBMCAccessRequest: