feat: [SG-43418] Pass task context and revision_id from Loader to Publisher

Replace env-var-based revision ID passing and publisher._set_context() with tk_multi_publish2.show_dialog(context=..., root_item_properties=...) so the Publisher receives task context and am_revision_id directly.

Author: chenm1adsk

hooks/tk-desktop_actions.py

@@ -12,7 +12,6 @@
 Hook that loads defines all the available actions, broken down by publish type.
 """
 
-import os
 from typing import Any
 
 import sgtk
@@ -245,23 +244,6 @@ def _launch_publisher(self, action_name: str, sg_publish_data: dict) -> None:
         else:
             raise TankError(f"Invalid entity type for publish: {entity_type}.")
 
-        # Use different env var naming for project-level vs task-level contexts
-        if task_id is not None:
-            revision_id_env_var = f"TK_FLOWAM_REVISION_ID_{task_id}"
-        else:
-            # For project-level contexts, use project ID
-            project_id = entity_id
-            revision_id_env_var = f"TK_FLOWAM_REVISION_ID_PROJECT_{project_id}"
-
-        if action_name == "publish":
-            revision_id = sg_publish_data.get("sg_flow_revision_id")
-            os.environ[revision_id_env_var] = revision_id
-        else:
-            # Clear possible previously existing publish states from an unfinished publish
-            # (Finished publishes should clear this value)
-            if revision_id_env_var in os.environ:
-                os.environ.pop(revision_id_env_var)
-
         # NOTE: the context should be either a Task or a Project
         entity_ctx = engine.tank.context_from_entity(entity_type, entity_id)
 
@@ -277,15 +259,12 @@ def _launch_publisher(self, action_name: str, sg_publish_data: dict) -> None:
                 "env/includes/desktop/project.yml under 'desktop.project: apps:'\n\n"
             )
 
-        try:
-            # Set context and launch Publisher using the pre-imported module
-            publisher_app._set_context(entity_ctx)
-
-            # For republish action, restrict publisher to single file mode
-            single_file_mode = action_name == "publish"
-            publisher_app.show_publish_dialog(single_file_mode=single_file_mode)
-        except Exception as e:
-            raise TankError(
-                f"Failed to launch Publisher: {e}\n\n"
-                f"The Publisher app was found but failed to start."
-            )
+        tk_multi_publish2 = publisher_app.import_module("tk_multi_publish2")
+        revision_id = sg_publish_data.get("sg_flow_revision_id")
+        tk_multi_publish2.show_dialog(
+            publisher_app,
+            context=entity_ctx,
+            root_item_properties=(
+                {"am_revision_id": revision_id} if revision_id else None
+            ),
+        )