chore: format

Author: jesse-michael-han

morphcloud/api.py

@@ -1575,6 +1575,18 @@ async def astop(self, instance_id: str) -> None:
         )
         response.raise_for_status()
 
+    def reboot(self, instance_id: str) -> None:
+        """Reboot an instance by its ID."""
+        response = self._client._http_client.post(f"/instance/{instance_id}/reboot")
+        response.raise_for_status()
+
+    async def areboot(self, instance_id: str) -> None:
+        """Reboot an instance by its ID."""
+        response = await self._client._async_http_client.post(
+            f"/instance/{instance_id}/reboot"
+        )
+        response.raise_for_status()
+
     def boot(
         self,
         snapshot_id: str,

morphcloud/cli.py

@@ -1308,6 +1308,35 @@ def resume_instance(instance_id):
         handle_api_error(e)
 
 
+@instance.command("reboot")
+@click.argument("instance_id")
+def reboot_instance(instance_id):
+    """Reboot a running instance."""
+    client = get_client()
+    try:
+        instance_obj = client.instances.get(instance_id)
+        with Spinner(
+            text=f"Rebooting instance {instance_id}...",
+            success_text=f"Instance rebooted: {instance_id}",
+            success_emoji="🔄",
+        ):
+            instance_obj.reboot()
+    except api.ApiError as e:
+        if e.status_code == 404:
+            click.echo(f"Error: Instance '{instance_id}' not found.", err=True)
+            sys.exit(1)
+        elif e.status_code == 409:
+            click.echo(
+                f"Error: Instance '{instance_id}' cannot be rebooted ({e.response_body}).",
+                err=True,
+            )
+            sys.exit(1)
+        else:
+            handle_api_error(e)
+    except Exception as e:
+        handle_api_error(e)
+
+
 @instance.command("get")
 @click.argument("instance_id")
 def get_instance(instance_id):