From c8fb00c9c81c9a60f0436dbcb43b546060cf6664 Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakubjezek001@gmail.com>
Date: Mon, 30 Jan 2023 12:43:40 +0100
Subject: [PATCH] global: expanding staging dir maker abstraction

so it supports `OPENPYPE_TEMP_DIR` with anatomy formatting keys
---
 openpype/pipeline/publish/lib.py | 55 ++++++++++++++++++++++++++++----
 1 file changed, 49 insertions(+), 6 deletions(-)

diff --git a/openpype/pipeline/publish/lib.py b/openpype/pipeline/publish/lib.py
index c76671fa39..5591acf57d 100644
--- a/openpype/pipeline/publish/lib.py
+++ b/openpype/pipeline/publish/lib.py
@@ -10,7 +10,11 @@ import six
 import pyblish.plugin
 import pyblish.api
 
-from openpype.lib import Logger, filter_profiles
+from openpype.lib import (
+    Logger,
+    filter_profiles,
+    StringTemplate
+)
 from openpype.settings import (
     get_project_settings,
     get_system_settings,
@@ -623,12 +627,51 @@ def get_instance_staging_dir(instance):
     Returns:
         str: Path to staging dir of instance.
     """
+    staging_dir = instance.data.get('stagingDir', None)
+    openpype_temp_dir = os.getenv("OPENPYPE_TEMP_DIR")
 
-    staging_dir = instance.data.get("stagingDir")
     if not staging_dir:
-        staging_dir = os.path.normpath(
-            tempfile.mkdtemp(prefix="pyblish_tmp_")
-        )
-        instance.data["stagingDir"] = staging_dir
+        custom_temp_dir = None
+        if openpype_temp_dir:
+            if "{" in openpype_temp_dir:
+                anatomy = instance.context.data["anatomy"]
+                # get anatomy formating data
+                # so template formating is supported
+                anatomy_data = copy.deepcopy(instance.context.data["anatomyData"])
+                anatomy_data["root"] = anatomy.roots
+                """Template path formating is supporting:
+                - optional key formating
+                - available tokens:
+                    - root[work | <root name key>]
+                    - project[name | code]
+                    - asset
+                    - hierarchy
+                    - task
+                    - username
+                    - app
+                """
+                custom_temp_dir = StringTemplate.format_template(
+                    openpype_temp_dir, anatomy_data)
+                custom_temp_dir = os.path.normpath(custom_temp_dir)
+                # create the dir in case it doesnt exists
+                os.makedirs(os.path.dirname(custom_temp_dir))
+            elif os.path.exists(openpype_temp_dir):
+                custom_temp_dir = openpype_temp_dir
+
+
+        if custom_temp_dir:
+            staging_dir = os.path.normpath(
+                tempfile.mkdtemp(
+                    prefix="pyblish_tmp_",
+                    dir=custom_temp_dir
+                )
+            )
+        else:
+            staging_dir = os.path.normpath(
+                tempfile.mkdtemp(prefix="pyblish_tmp_")
+            )
+        instance.data['stagingDir'] = staging_dir
+
+    instance.context.data["cleanupFullPaths"].append(staging_dir)
 
     return staging_dir