From c3fca896d48f82026aea0f81055a996c366ea920 Mon Sep 17 00:00:00 2001
From: Roy Nieterau <roy_nieterau@hotmail.com>
Date: Mon, 9 Jan 2023 11:16:23 +0100
Subject: [PATCH] Implement plug-ins to support workfile publishing

---
 .../plugins/publish/collect_current_file.py   | 17 ++++++++++++
 .../collect_workfile_representation.py        | 26 +++++++++++++++++++
 .../plugins/publish/increment_workfile.py     | 23 ++++++++++++++++
 .../plugins/publish/save_workfile.py          | 23 ++++++++++++++++
 4 files changed, 89 insertions(+)
 create mode 100644 openpype/hosts/substancepainter/plugins/publish/collect_current_file.py
 create mode 100644 openpype/hosts/substancepainter/plugins/publish/collect_workfile_representation.py
 create mode 100644 openpype/hosts/substancepainter/plugins/publish/increment_workfile.py
 create mode 100644 openpype/hosts/substancepainter/plugins/publish/save_workfile.py

diff --git a/openpype/hosts/substancepainter/plugins/publish/collect_current_file.py b/openpype/hosts/substancepainter/plugins/publish/collect_current_file.py
new file mode 100644
index 0000000000..dac493bbf1
--- /dev/null
+++ b/openpype/hosts/substancepainter/plugins/publish/collect_current_file.py
@@ -0,0 +1,17 @@
+import pyblish.api
+
+from openpype.pipeline import registered_host
+
+
+class CollectCurrentFile(pyblish.api.ContextPlugin):
+    """Inject the current working file into context"""
+
+    order = pyblish.api.CollectorOrder - 0.49
+    label = "Current Workfile"
+    hosts = ["substancepainter"]
+
+    def process(self, context):
+        host = registered_host()
+        path = host.get_current_workfile()
+        context.data["currentFile"] = path
+        self.log.debug(f"Current workfile: {path}")
\ No newline at end of file
diff --git a/openpype/hosts/substancepainter/plugins/publish/collect_workfile_representation.py b/openpype/hosts/substancepainter/plugins/publish/collect_workfile_representation.py
new file mode 100644
index 0000000000..563c2d4c07
--- /dev/null
+++ b/openpype/hosts/substancepainter/plugins/publish/collect_workfile_representation.py
@@ -0,0 +1,26 @@
+import os
+import pyblish.api
+
+
+class CollectWorkfileRepresentation(pyblish.api.InstancePlugin):
+    """Create a publish representation for the current workfile instance."""
+
+    order = pyblish.api.CollectorOrder
+    label = "Workfile representation"
+    hosts = ['substancepainter']
+    families = ["workfile"]
+
+    def process(self, instance):
+
+        context = instance.context
+        current_file = context.data["currentFile"]
+
+        folder, file = os.path.split(current_file)
+        filename, ext = os.path.splitext(file)
+
+        instance.data['representations'] = [{
+            'name': ext.lstrip("."),
+            'ext': ext.lstrip("."),
+            'files': file,
+            "stagingDir": folder,
+        }]
diff --git a/openpype/hosts/substancepainter/plugins/publish/increment_workfile.py b/openpype/hosts/substancepainter/plugins/publish/increment_workfile.py
new file mode 100644
index 0000000000..b45d66fbb1
--- /dev/null
+++ b/openpype/hosts/substancepainter/plugins/publish/increment_workfile.py
@@ -0,0 +1,23 @@
+import pyblish.api
+
+from openpype.lib import version_up
+from openpype.pipeline import registered_host
+
+
+class IncrementWorkfileVersion(pyblish.api.ContextPlugin):
+    """Increment current workfile version."""
+
+    order = pyblish.api.IntegratorOrder + 1
+    label = "Increment Workfile Version"
+    optional = True
+    hosts = ["substancepainter"]
+
+    def process(self, context):
+
+        assert all(result["success"] for result in context.data["results"]), (
+            "Publishing not successful so version is not increased.")
+
+        host = registered_host()
+        path = context.data["currentFile"]
+        self.log.info(f"Incrementing current workfile to: {path}")
+        host.save_workfile(version_up(path))
diff --git a/openpype/hosts/substancepainter/plugins/publish/save_workfile.py b/openpype/hosts/substancepainter/plugins/publish/save_workfile.py
new file mode 100644
index 0000000000..5e86785e0d
--- /dev/null
+++ b/openpype/hosts/substancepainter/plugins/publish/save_workfile.py
@@ -0,0 +1,23 @@
+import pyblish.api
+
+from openpype.pipeline import registered_host
+
+
+class SaveCurrentWorkfile(pyblish.api.ContextPlugin):
+    """Save current workfile"""
+
+    label = "Save current workfile"
+    order = pyblish.api.ExtractorOrder - 0.49
+    hosts = ["substancepainter"]
+
+    def process(self, context):
+
+        host = registered_host()
+        assert context.data['currentFile'] == host.get_current_workfile()
+
+        if host.has_unsaved_changes():
+            self.log.info("Saving current file..")
+            host.save_workfile()
+        else:
+            self.log.debug("Skipping workfile save because there are no "
+                           "unsaved changes.")