From ea54b0dc2512eeb428b1f3ea702a10a55ac6ccf3 Mon Sep 17 00:00:00 2001
From: "clement.hector" <clement.hector@fixstudio.com>
Date: Mon, 9 May 2022 10:28:09 +0200
Subject: [PATCH] refactor

---
 openpype/hosts/nuke/startup/menu.py           | 31 +++++++++++++++++++
 .../defaults/project_settings/nuke.json       |  4 +++
 .../projects_schema/schema_project_maya.json  |  2 +-
 .../projects_schema/schema_project_nuke.json  |  4 +++
 ...riptsmenu.json => schema_scriptsmenu.json} |  0
 5 files changed, 40 insertions(+), 1 deletion(-)
 rename openpype/settings/entities/schemas/projects_schema/schemas/{schema_maya_scriptsmenu.json => schema_scriptsmenu.json} (100%)

diff --git a/openpype/hosts/nuke/startup/menu.py b/openpype/hosts/nuke/startup/menu.py
index 9ed43b2110..dee1d9d868 100644
--- a/openpype/hosts/nuke/startup/menu.py
+++ b/openpype/hosts/nuke/startup/menu.py
@@ -1,5 +1,7 @@
 import nuke
+import os
 
+import avalon.api
 from openpype.api import Logger
 from openpype.pipeline import install_host
 from openpype.hosts.nuke import api
@@ -9,6 +11,7 @@ from openpype.hosts.nuke.api.lib import (
     WorkfileSettings,
     dirmap_file_name_filter
 )
+from openpype.settings import get_project_settings
 
 log = Logger.get_logger(__name__)
 
@@ -28,3 +31,31 @@ nuke.addOnScriptLoad(WorkfileSettings().set_context_settings)
 nuke.addFilenameFilter(dirmap_file_name_filter)
 
 log.info('Automatic syncing of write file knob to script version')
+
+
+def add_scripts_menu():
+    try:
+        from scriptsmenu import launchfornuke
+    except ImportError:
+        log.warning(
+            "Skipping studio.menu install, because "
+            "'scriptsmenu' module seems unavailable."
+        )
+        return
+
+    # load configuration of custom menu
+    project_settings = get_project_settings(os.getenv("AVALON_PROJECT"))
+    config = project_settings["nuke"]["scriptsmenu"]["definition"]
+    _menu = project_settings["nuke"]["scriptsmenu"]["name"]
+
+    if not config:
+        log.warning("Skipping studio menu, no definition found.")
+        return
+
+    # run the launcher for Maya menu
+    studio_menu = launchfornuke.main(title=_menu.title())
+
+    # apply configuration
+    studio_menu.build_from_configuration(studio_menu, config)
+
+add_scripts_menu()
diff --git a/openpype/settings/defaults/project_settings/nuke.json b/openpype/settings/defaults/project_settings/nuke.json
index 128d440732..a9d284873c 100644
--- a/openpype/settings/defaults/project_settings/nuke.json
+++ b/openpype/settings/defaults/project_settings/nuke.json
@@ -15,6 +15,10 @@
             "destination-path": []
         }
     },
+    "scriptsmenu": {
+        "name": "OpenPype Tools",
+        "definition": []
+    },
     "create": {
         "CreateWriteRender": {
             "fpath_template": "{work}/renders/nuke/{subset}/{subset}.{frame}.{ext}",
diff --git a/openpype/settings/entities/schemas/projects_schema/schema_project_maya.json b/openpype/settings/entities/schemas/projects_schema/schema_project_maya.json
index cc70516c72..0c7943447b 100644
--- a/openpype/settings/entities/schemas/projects_schema/schema_project_maya.json
+++ b/openpype/settings/entities/schemas/projects_schema/schema_project_maya.json
@@ -49,7 +49,7 @@
         },
         {
             "type": "schema",
-            "name": "schema_maya_scriptsmenu"
+            "name": "schema_scriptsmenu"
         },
         {
             "type": "schema",
diff --git a/openpype/settings/entities/schemas/projects_schema/schema_project_nuke.json b/openpype/settings/entities/schemas/projects_schema/schema_project_nuke.json
index bc572cbdc8..1ae4efd8ea 100644
--- a/openpype/settings/entities/schemas/projects_schema/schema_project_nuke.json
+++ b/openpype/settings/entities/schemas/projects_schema/schema_project_nuke.json
@@ -79,6 +79,10 @@
                 }
             ]
         },
+        {
+            "type": "schema",
+            "name": "schema_scriptsmenu"
+        },
         {
             "type": "dict",
             "collapsible": true,
diff --git a/openpype/settings/entities/schemas/projects_schema/schemas/schema_maya_scriptsmenu.json b/openpype/settings/entities/schemas/projects_schema/schemas/schema_scriptsmenu.json
similarity index 100%
rename from openpype/settings/entities/schemas/projects_schema/schemas/schema_maya_scriptsmenu.json
rename to openpype/settings/entities/schemas/projects_schema/schemas/schema_scriptsmenu.json