style(vm): format lifecycle extension changes

Signed-off-by: Patrick Riel <priel@nvidia.com>

Author: cheese-head

crates/openshell-driver-vm/src/driver.rs

@@ -1,13 +1,13 @@
 // SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 // SPDX-License-Identifier: Apache-2.0
 
+use crate::gpu::{
+    GpuInventory, SubnetAllocator, allocate_vsock_cid, mac_from_sandbox_id, tap_device_name,
+};
 use crate::lifecycle::{
     BackendFeature, GuestInitDropin, LaunchAbortReason, LaunchPlan, LifecycleExtensionRegistry,
     RestoreContext, extension_state_dir,
 };
-use crate::gpu::{
-    GpuInventory, SubnetAllocator, allocate_vsock_cid, mac_from_sandbox_id, tap_device_name,
-};
 use crate::rootfs::{
     clone_or_copy_sparse_file, create_ext4_image_from_dir_with_size, create_rootfs_image_from_dir,
     extract_rootfs_archive_to, prepare_sandbox_rootfs_from_image_root, sandbox_guest_init_path,
@@ -919,9 +919,7 @@ impl VmDriver {
                 sandbox: sandbox.clone(),
                 state_dir: state_dir.clone(),
             };
-            self.lifecycle_extensions
-                .after_restore(&persisted)
-                .await;
+            self.lifecycle_extensions.after_restore(&persisted).await;
         }
         tokio::spawn({
             let driver = self.clone();
@@ -1162,11 +1160,7 @@ impl VmDriver {
             sandbox: sandbox.clone(),
             state_dir: state_dir.clone(),
         };
-        if let Err(err) = self
-            .lifecycle_extensions
-            .before_restore(&persisted)
-            .await
-        {
+        if let Err(err) = self.lifecycle_extensions.before_restore(&persisted).await {
             warn!(
                 sandbox_id = %sandbox.id,
                 sandbox_name = %sandbox.name,
@@ -4566,7 +4560,9 @@ fn write_guest_init_dropin_manifest(
     let guest_path = overlay_upper_path(GUEST_INIT_DROPIN_MANIFEST);
     let contents = render_guest_init_dropin_manifest(dropins);
     write_rootfs_image_file(overlay_disk, &guest_path, &contents).map_err(|err| {
-        Status::internal(format!("write VM guest init drop-in manifest failed: {err}"))
+        Status::internal(format!(
+            "write VM guest init drop-in manifest failed: {err}"
+        ))
     })?;
     set_rootfs_image_file_mode(overlay_disk, &guest_path, 0o644).map_err(|err| {
         Status::internal(format!(
@@ -6377,8 +6373,8 @@ mod tests {
     }
 
     use crate::lifecycle::{
-        BackendFeature, LaunchPlan, LifecycleError, LifecycleExtension,
-        LifecycleExtensionRegistry, LifecycleResult,
+        BackendFeature, LaunchPlan, LifecycleError, LifecycleExtension, LifecycleExtensionRegistry,
+        LifecycleResult,
     };
     use crate::runtime::VmBackend;
 

crates/openshell-driver-vm/src/lifecycle.rs

@@ -247,10 +247,7 @@ impl LaunchPlan {
         }
     }
 
-    pub fn require_backend_features(
-        &mut self,
-        features: impl IntoIterator<Item = BackendFeature>,
-    ) {
+    pub fn require_backend_features(&mut self, features: impl IntoIterator<Item = BackendFeature>) {
         for feature in features {
             self.require_backend_feature(feature);
         }
@@ -412,11 +409,7 @@ pub trait LifecycleExtension: std::fmt::Debug + Send + Sync {
     ///
     /// Invoked in reverse registration order. Errors are logged but do not
     /// propagate.
-    async fn after_delete(
-        &self,
-        _sandbox: &Sandbox,
-        _state_dir: &Path,
-    ) -> LifecycleResult<()> {
+    async fn after_delete(&self, _sandbox: &Sandbox, _state_dir: &Path) -> LifecycleResult<()> {
         Ok(())
     }
 
@@ -426,10 +419,7 @@ pub trait LifecycleExtension: std::fmt::Debug + Send + Sync {
     /// Returning an error causes the driver to skip restoring this
     /// sandbox; the persisted state is left on disk for operator
     /// inspection.
-    async fn before_restore(
-        &self,
-        _sandbox: &RestoreContext,
-    ) -> LifecycleResult<()> {
+    async fn before_restore(&self, _sandbox: &RestoreContext) -> LifecycleResult<()> {
         Ok(())
     }
 
@@ -439,10 +429,7 @@ pub trait LifecycleExtension: std::fmt::Debug + Send + Sync {
     /// Only invoked when restore succeeds. If the restore fails partway
     /// through, [`after_launch_failed`](Self::after_launch_failed)
     /// runs instead.
-    async fn after_restore(
-        &self,
-        _sandbox: &RestoreContext,
-    ) -> LifecycleResult<()> {
+    async fn after_restore(&self, _sandbox: &RestoreContext) -> LifecycleResult<()> {
         Ok(())
     }
 }
@@ -558,9 +545,7 @@ impl LifecycleExtensionRegistry {
             .iter()
             .filter(|ext| match ext.activation() {
                 ExtensionActivation::Global => true,
-                ExtensionActivation::OnRequest { key } => {
-                    sandbox_requested_extension(sandbox, key)
-                }
+                ExtensionActivation::OnRequest { key } => sandbox_requested_extension(sandbox, key),
             })
             .collect()
     }
@@ -649,10 +634,7 @@ impl LifecycleExtensionRegistry {
         }
     }
 
-    pub async fn before_restore(
-        &self,
-        sandbox: &RestoreContext,
-    ) -> LifecycleResult<()> {
+    pub async fn before_restore(&self, sandbox: &RestoreContext) -> LifecycleResult<()> {
         for ext in self.active_for(&sandbox.sandbox) {
             ext.before_restore(sandbox).await?;
         }
@@ -752,9 +734,7 @@ fn validate_extension_name(name: &str) -> LifecycleResult<()> {
     Ok(())
 }
 
-fn validate_descriptor_strings(
-    descriptor: &ExtensionDescriptor,
-) -> LifecycleResult<()> {
+fn validate_descriptor_strings(descriptor: &ExtensionDescriptor) -> LifecycleResult<()> {
     for value in descriptor
         .provides
         .kernel_profiles
@@ -932,11 +912,7 @@ mod tests {
             Ok(())
         }
 
-        async fn after_delete(
-            &self,
-            _sandbox: &Sandbox,
-            _state_dir: &Path,
-        ) -> LifecycleResult<()> {
+        async fn after_delete(&self, _sandbox: &Sandbox, _state_dir: &Path) -> LifecycleResult<()> {
             self.calls
                 .lock()
                 .unwrap()
@@ -1040,10 +1016,7 @@ mod tests {
             .configure_launch(&sandbox, &PathBuf::from("/tmp/state"), &mut plan)
             .await
             .expect_err("scripted failure should propagate");
-        assert!(
-            err.message()
-                .contains("scripted configure_launch failure")
-        );
+        assert!(err.message().contains("scripted configure_launch failure"));
 
         assert_eq!(ext_a.calls(), vec!["a:configure_launch"]);
         assert_eq!(ext_fail.calls(), vec!["boom:configure_launch"]);