From 465101c6398fcc1fcd6cca0e5ecd278dae2a4640 Mon Sep 17 00:00:00 2001
From: Simone Barbieri <simone.barbieri@hey.com>
Date: Fri, 20 Oct 2023 12:22:44 +0100
Subject: [PATCH] Added another action to delete unused assets

---
 openpype/hosts/unreal/api/pipeline.py         |  2 +-
 .../plugins/inventory/delete_unused_assets.py | 31 +++++++++++++++++++
 .../unreal/plugins/inventory/update_actors.py |  4 +--
 3 files changed, 34 insertions(+), 3 deletions(-)
 create mode 100644 openpype/hosts/unreal/plugins/inventory/delete_unused_assets.py

diff --git a/openpype/hosts/unreal/api/pipeline.py b/openpype/hosts/unreal/api/pipeline.py
index 35b218e629..0bb19ec601 100644
--- a/openpype/hosts/unreal/api/pipeline.py
+++ b/openpype/hosts/unreal/api/pipeline.py
@@ -750,7 +750,7 @@ def replace_geometry_cache_actors(old_assets, new_assets, selected):
                 comp.set_geometry_cache(new_mesh)
 
 
-def delete_previous_asset_if_unused(container, asset_content):
+def delete_asset_if_unused(container, asset_content):
     ar = unreal.AssetRegistryHelpers.get_asset_registry()
 
     references = set()
diff --git a/openpype/hosts/unreal/plugins/inventory/delete_unused_assets.py b/openpype/hosts/unreal/plugins/inventory/delete_unused_assets.py
new file mode 100644
index 0000000000..f63476b3c7
--- /dev/null
+++ b/openpype/hosts/unreal/plugins/inventory/delete_unused_assets.py
@@ -0,0 +1,31 @@
+import unreal
+
+from openpype.hosts.unreal.api.pipeline import delete_asset_if_unused
+from openpype.pipeline import InventoryAction
+
+
+
+class DeleteUnusedAssets(InventoryAction):
+    """Delete all the assets that are not used in any level.
+    """
+
+    label = "Delete Unused Assets"
+    icon = "trash"
+    color = "red"
+    order = 1
+
+    def process(self, containers):
+        allowed_families = ["model", "rig"]
+
+        for container in containers:
+            container_dir = container.get("namespace")
+            if container.get("family") not in allowed_families:
+                unreal.log_warning(
+                    f"Container {container_dir} is not supported.")
+                continue
+
+            asset_content = unreal.EditorAssetLibrary.list_assets(
+                container_dir, recursive=True, include_folder=False
+            )
+
+            delete_asset_if_unused(container, asset_content)
diff --git a/openpype/hosts/unreal/plugins/inventory/update_actors.py b/openpype/hosts/unreal/plugins/inventory/update_actors.py
index 2b012cf22c..6bc576716c 100644
--- a/openpype/hosts/unreal/plugins/inventory/update_actors.py
+++ b/openpype/hosts/unreal/plugins/inventory/update_actors.py
@@ -5,7 +5,7 @@ from openpype.hosts.unreal.api.pipeline import (
     replace_static_mesh_actors,
     replace_skeletal_mesh_actors,
     replace_geometry_cache_actors,
-    delete_previous_asset_if_unused,
+    delete_asset_if_unused,
 )
 from openpype.pipeline import InventoryAction
 
@@ -60,7 +60,7 @@ def update_assets(containers, selected):
 
             unreal.EditorLevelLibrary.save_current_level()
 
-            delete_previous_asset_if_unused(sa_cont, old_content)
+            delete_asset_if_unused(sa_cont, old_content)
 
 
 class UpdateAllActors(InventoryAction):