factor shared part

Co-Authored-By: Niklas Eicker <git@nikl.me>

Author: mockersf

crates/bevy_app/src/plugin_group.rs

@@ -40,24 +40,9 @@ impl PluginGroupBuilder {
         }
     }
 
-    // Removes a previous ordering of a plugin that has just been added at `added_at` index
-    fn remove_when_adding<T: Plugin>(&mut self, added_at: usize) {
-        if let Some(to_remove) = self
-            .order
-            .iter()
-            .enumerate()
-            .find(|(i, ty)| *i != added_at && **ty == TypeId::of::<T>())
-            .map(|(i, _)| i)
-        {
-            self.order.remove(to_remove);
-        }
-    }
-
-    /// Adds the plugin [`Plugin`] at the end of this [`PluginGroupBuilder`]. If the plugin was
-    /// already in the group, it is removed from its previous place.
-    pub fn add<T: Plugin>(&mut self, plugin: T) -> &mut Self {
-        let target_index = self.order.len();
-        self.order.push(TypeId::of::<T>());
+    // Insert the new plugin as enabled, and removes its previous ordering if it was
+    // already present
+    fn upsert_plugin_state<T: Plugin>(&mut self, plugin: T, added_at_index: usize) {
         if let Some(entry) = self.plugins.insert(
             TypeId::of::<T>(),
             PluginEntry {
@@ -71,9 +56,24 @@ impl PluginGroupBuilder {
                     entry.plugin.name()
                 );
             }
-            self.remove_when_adding::<T>(target_index);
+            if let Some(to_remove) = self
+                .order
+                .iter()
+                .enumerate()
+                .find(|(i, ty)| *i != added_at_index && **ty == TypeId::of::<T>())
+                .map(|(i, _)| i)
+            {
+                self.order.remove(to_remove);
+            }
         }
+    }
 
+    /// Adds the plugin [`Plugin`] at the end of this [`PluginGroupBuilder`]. If the plugin was
+    /// already in the group, it is removed from its previous place.
+    pub fn add<T: Plugin>(&mut self, plugin: T) -> &mut Self {
+        let target_index = self.order.len();
+        self.order.push(TypeId::of::<T>());
+        self.upsert_plugin_state(plugin, target_index);
         self
     }
 
@@ -83,21 +83,7 @@ impl PluginGroupBuilder {
     pub fn add_before<Target: Plugin, T: Plugin>(&mut self, plugin: T) -> &mut Self {
         let target_index = self.index_of::<Target>();
         self.order.insert(target_index, TypeId::of::<T>());
-        if let Some(entry) = self.plugins.insert(
-            TypeId::of::<T>(),
-            PluginEntry {
-                plugin: Box::new(plugin),
-                enabled: true,
-            },
-        ) {
-            if entry.enabled {
-                warn!(
-                    "You are replacing plugin '{}' that was not disabled.",
-                    entry.plugin.name()
-                );
-            }
-            self.remove_when_adding::<T>(target_index);
-        }
+        self.upsert_plugin_state(plugin, target_index);
         self
     }
 
@@ -107,21 +93,7 @@ impl PluginGroupBuilder {
     pub fn add_after<Target: Plugin, T: Plugin>(&mut self, plugin: T) -> &mut Self {
         let target_index = self.index_of::<Target>() + 1;
         self.order.insert(target_index, TypeId::of::<T>());
-        if let Some(entry) = self.plugins.insert(
-            TypeId::of::<T>(),
-            PluginEntry {
-                plugin: Box::new(plugin),
-                enabled: true,
-            },
-        ) {
-            if entry.enabled {
-                warn!(
-                    "You are replacing plugin '{}' that was not disabled.",
-                    entry.plugin.name()
-                );
-            }
-            self.remove_when_adding::<T>(target_index);
-        }
+        self.upsert_plugin_state(plugin, target_index);
         self
     }