From fc10b26ea0610dc5d32ed7c2ce285dac8fdef9eb Mon Sep 17 00:00:00 2001
From: Simone Barbieri <simone.barbieri@hey.com>
Date: Thu, 8 Dec 2022 11:20:24 +0000
Subject: [PATCH] Implemented creator and extractor for uasset

---
 .../unreal/plugins/create/create_uasset.py    | 61 +++++++++++++++++++
 .../unreal/plugins/publish/extract_uasset.py  | 45 ++++++++++++++
 openpype/plugins/publish/integrate.py         |  3 +-
 3 files changed, 108 insertions(+), 1 deletion(-)
 create mode 100644 openpype/hosts/unreal/plugins/create/create_uasset.py
 create mode 100644 openpype/hosts/unreal/plugins/publish/extract_uasset.py

diff --git a/openpype/hosts/unreal/plugins/create/create_uasset.py b/openpype/hosts/unreal/plugins/create/create_uasset.py
new file mode 100644
index 0000000000..ee584ac00c
--- /dev/null
+++ b/openpype/hosts/unreal/plugins/create/create_uasset.py
@@ -0,0 +1,61 @@
+"""Create UAsset."""
+from pathlib import Path
+
+import unreal
+
+from openpype.hosts.unreal.api import pipeline
+from openpype.pipeline import LegacyCreator
+
+
+class CreateUAsset(LegacyCreator):
+    """UAsset."""
+
+    name = "UAsset"
+    label = "UAsset"
+    family = "uasset"
+    icon = "cube"
+
+    root = "/Game/OpenPype"
+    suffix = "_INS"
+
+    def __init__(self, *args, **kwargs):
+        super(CreateUAsset, self).__init__(*args, **kwargs)
+
+    def process(self):
+        ar = unreal.AssetRegistryHelpers.get_asset_registry()
+
+        subset = self.data["subset"]
+        path = f"{self.root}/PublishInstances/"
+
+        unreal.EditorAssetLibrary.make_directory(path)
+
+        selection = []
+        if (self.options or {}).get("useSelection"):
+            sel_objects = unreal.EditorUtilityLibrary.get_selected_assets()
+            selection = [a.get_path_name() for a in sel_objects]
+
+            if len(selection) != 1:
+                raise RuntimeError("Please select only one object.")
+
+            obj = selection[0]
+
+            asset = ar.get_asset_by_object_path(obj).get_asset()
+            sys_path = unreal.SystemLibrary.get_system_path(asset)
+
+            if not sys_path:
+                raise RuntimeError(
+                    f"{Path(obj).name} is not on the disk. Likely it needs to"
+                    "be saved first.")
+
+            if Path(sys_path).suffix != ".uasset":
+                raise RuntimeError(f"{Path(sys_path).name} is not a UAsset.")
+
+        unreal.log("selection: {}".format(selection))
+        container_name = f"{subset}{self.suffix}"
+        pipeline.create_publish_instance(
+            instance=container_name, path=path)
+
+        data = self.data.copy()
+        data["members"] = selection
+
+        pipeline.imprint(f"{path}/{container_name}", data)
diff --git a/openpype/hosts/unreal/plugins/publish/extract_uasset.py b/openpype/hosts/unreal/plugins/publish/extract_uasset.py
new file mode 100644
index 0000000000..99279e38a1
--- /dev/null
+++ b/openpype/hosts/unreal/plugins/publish/extract_uasset.py
@@ -0,0 +1,45 @@
+from pathlib import Path
+import shutil
+
+import unreal
+from unreal import EditorLevelLibrary as ell
+from unreal import EditorAssetLibrary as eal
+
+from openpype.client import get_representation_by_name
+from openpype.pipeline import legacy_io, publish
+
+
+class ExtractUAsset(publish.Extractor):
+    """Extract a UAsset."""
+
+    label = "Extract UAsset"
+    hosts = ["unreal"]
+    families = ["uasset"]
+    optional = True
+
+    def process(self, instance):
+        ar = unreal.AssetRegistryHelpers.get_asset_registry()
+
+        self.log.info("Performing extraction..")
+
+        staging_dir = self.staging_dir(instance)
+        filename = "{}.uasset".format(instance.name)
+
+        obj = instance[0]
+
+        asset = ar.get_asset_by_object_path(obj).get_asset()
+        sys_path = unreal.SystemLibrary.get_system_path(asset)
+        filename = Path(sys_path).name
+
+        shutil.copy(sys_path, staging_dir)
+
+        if "representations" not in instance.data:
+            instance.data["representations"] = []
+
+        representation = {
+            'name': 'uasset',
+            'ext': 'uasset',
+            'files': filename,
+            "stagingDir": staging_dir,
+        }
+        instance.data["representations"].append(representation)
diff --git a/openpype/plugins/publish/integrate.py b/openpype/plugins/publish/integrate.py
index 6a85a87129..6efff8440c 100644
--- a/openpype/plugins/publish/integrate.py
+++ b/openpype/plugins/publish/integrate.py
@@ -130,7 +130,8 @@ class IntegrateAsset(pyblish.api.InstancePlugin):
                 "mvUsdComposition",
                 "mvUsdOverride",
                 "simpleUnrealTexture",
-                "online"
+                "online",
+                "uasset"
                 ]
 
     default_template_name = "publish"