From 5c4446ef0c153a10652e0eafa6b7c37cfbf8bf2b Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakub@orbi.tools>
Date: Wed, 26 May 2021 18:30:39 +0200
Subject: [PATCH] nuke: workfile options callback

---
 openpype/hosts/nuke/api/__init__.py |  2 +-
 openpype/hosts/nuke/api/lib.py      | 67 +++++++++++++++++++++++------
 2 files changed, 54 insertions(+), 15 deletions(-)

diff --git a/openpype/hosts/nuke/api/__init__.py b/openpype/hosts/nuke/api/__init__.py
index bd7a95f916..bb60465e23 100644
--- a/openpype/hosts/nuke/api/__init__.py
+++ b/openpype/hosts/nuke/api/__init__.py
@@ -80,7 +80,7 @@ def install():
     # Set context settings.
     nuke.addOnCreate(workfile_settings.set_context_settings, nodeClass="Root")
     nuke.addOnCreate(workfile_settings.set_favorites, nodeClass="Root")
-    nuke.addOnCreate(lib.open_last_workfile, nodeClass="Root")
+    nuke.addOnCreate(lib.process_workfile_options, nodeClass="Root")
     nuke.addOnCreate(lib.launch_workfiles_app, nodeClass="Root")
     menu.install()
 
diff --git a/openpype/hosts/nuke/api/lib.py b/openpype/hosts/nuke/api/lib.py
index e6dde813a7..689b51f6a3 100644
--- a/openpype/hosts/nuke/api/lib.py
+++ b/openpype/hosts/nuke/api/lib.py
@@ -16,6 +16,7 @@ from avalon.nuke import (
 from openpype.api import (
     Logger,
     Anatomy,
+    BuildWorkfile,
     get_version_from_path,
     get_anatomy_settings,
     get_hierarchy,
@@ -1641,23 +1642,61 @@ def launch_workfiles_app():
         workfiles.show(os.environ["AVALON_WORKDIR"])
 
 
-def open_last_workfile():
-    # get state from settings
-    open_last_version = get_current_project_settings()["nuke"].get(
-        "general", {}).get("create_initial_workfile")
+def process_workfile_options():
+    from openpype.lib import env_value_to_bool
+
+    # get state from settings
+    workfile_options = get_current_project_settings()["nuke"].get(
+        "workfile_options", {})
+
+    # get all imortant settings
+    openlv_on = env_value_to_bool(
+        env_key="AVALON_OPEN_LAST_WORKFILE",
+        default=None)
+
+    createfv_on = workfile_options.get(
+        "create_first_version", {}).get("enabled")
+    createfv_template_paths = workfile_options.get(
+        "create_first_version", {}).get("custom_template_paths")
+    createfv_builder = workfile_options.get(
+        "create_first_version", {}).get("builder_on")
 
-    log.info("Opening last workfile...")
     last_workfile_path = os.environ.get("AVALON_LAST_WORKFILE")
 
-    if not os.path.exists(last_workfile_path):
-        # return if none is defined
-        if not open_last_version:
-            return
+    # generate first version in file not existing and feature is enabled
+    if createfv_on and not os.path.exists(last_workfile_path):
+        # get custom template path if any
+        custom_template_path = createfv_template_paths[
+            platform.system().lower()]
 
+        # if custom template is defined
+        if custom_template_path:
+            log.info("Adding nodes from `{}`...".format(
+                custom_template_path
+            ))
+            # import nodes into current script
+            nuke.nodePaste(custom_template_path)
+
+        # if builder at start is defined
+        if createfv_builder:
+            log.info("Building nodes from presets...")
+            # build nodes by defined presets
+            BuildWorkfile().process()
+
+        log.info("Saving script as version `{}`...".format(
+            last_workfile_path
+        ))
+        # safe file as version
         save_file(last_workfile_path)
-    else:
-        # to avoid looping of the callback, remove it!
-        nuke.removeOnCreate(open_last_workfile, nodeClass="Root")
+        return
 
-        # open workfile
-        open_file(last_workfile_path)
+    # skip opening of last version if it is not enabled
+    if not openlv_on or not os.path.exists(last_workfile_path):
+        return
+
+    # to avoid looping of the callback, remove it!
+    nuke.removeOnCreate(process_workfile_options, nodeClass="Root")
+
+    log.info("Opening last workfile...")
+    # open workfile
+    open_file(last_workfile_path)