From 8d60739fe43b96ea227e2af6b3d7889504448ae6 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Fri, 9 Sep 2022 16:55:24 +0200
Subject: [PATCH] implemented functions for building actions

---
 .../hosts/maya/api/new_template_builder.py    | 47 ++++++++++++++++++-
 1 file changed, 46 insertions(+), 1 deletion(-)

diff --git a/openpype/hosts/maya/api/new_template_builder.py b/openpype/hosts/maya/api/new_template_builder.py
index b28cc80cd1..43e92c59e2 100644
--- a/openpype/hosts/maya/api/new_template_builder.py
+++ b/openpype/hosts/maya/api/new_template_builder.py
@@ -5,7 +5,7 @@ from maya import cmds
 
 from openpype.client import get_representations
 from openpype.lib import attribute_definitions
-from openpype.pipeline import legacy_io
+from openpype.pipeline import legacy_io, registered_host
 from openpype.pipeline.workfile.build_template_exceptions import (
     TemplateAlreadyImported
 )
@@ -14,6 +14,9 @@ from openpype.pipeline.workfile.new_template_loader import (
     PlaceholderPlugin,
     PlaceholderItem,
 )
+from openpype.tools.workfile_template_build import (
+    WorkfileBuildPlaceholderDialog,
+)
 
 from .lib import read, imprint
 
@@ -566,3 +569,45 @@ class LoadPlaceholder(PlaceholderItem):
 
     def load_succeed(self, container):
         self.parent_in_hierarchy(container)
+
+
+def build_workfile_template():
+    builder = MayaTemplateLoader(registered_host())
+    builder.build_template()
+
+
+def update_workfile_template():
+    builder = MayaTemplateLoader(registered_host())
+    builder.update_build_template()
+
+
+def create_placeholder():
+    host = registered_host()
+    builder = MayaTemplateLoader(host)
+    window = WorkfileBuildPlaceholderDialog(host, builder)
+    window.exec_()
+
+
+def update_placeholder():
+    host = registered_host()
+    builder = MayaTemplateLoader(host)
+    placeholder_items_by_id = {
+        placeholder_item.scene_identifier: placeholder_item
+        for placeholder_item in builder.get_placeholders()
+    }
+    placeholder_items = []
+    for node_name in cmds.ls(selection=True, long=True):
+        if node_name in placeholder_items_by_id:
+            placeholder_items.append(placeholder_items_by_id[node_name])
+
+    # TODO show UI at least
+    if len(placeholder_items) == 0:
+        raise ValueError("No node selected")
+
+    if len(placeholder_items) > 1:
+        raise ValueError("Too many selected nodes")
+
+    placeholder_item = placeholder_items[0]
+    window = WorkfileBuildPlaceholderDialog(host, builder)
+    window.set_update_mode(placeholder_item)
+    window.exec_()