From 360264117ee62d92cd58da90fcc9f61567ec0885 Mon Sep 17 00:00:00 2001
From: Toke Stuart Jepsen <tokejepsen@gmail.com>
Date: Wed, 22 Jul 2020 17:07:07 +0100
Subject: [PATCH] Adding "audioMain" to Nuke reviews.

- also added get_latest_version method to library.
---
 pype/api.py                                 |  6 ++--
 pype/lib.py                                 | 37 +++++++++++++++++++++
 pype/plugins/nuke/publish/collect_review.py | 18 ++++++++++
 3 files changed, 59 insertions(+), 2 deletions(-)

diff --git a/pype/api.py b/pype/api.py
index 44a31f2626..ce13688d13 100644
--- a/pype/api.py
+++ b/pype/api.py
@@ -40,7 +40,8 @@ from .lib import (
     get_version_from_path,
     get_last_version_from_path,
     modified_environ,
-    add_tool_to_environment
+    add_tool_to_environment,
+    get_latest_version
 )
 
 # Special naming case for subprocess since its a built-in method.
@@ -85,5 +86,6 @@ __all__ = [
     "modified_environ",
     "add_tool_to_environment",
 
-    "subprocess"
+    "subprocess",
+    "get_latest_version"
 ]
diff --git a/pype/lib.py b/pype/lib.py
index 87808e53f5..29e02ffd57 100644
--- a/pype/lib.py
+++ b/pype/lib.py
@@ -1387,3 +1387,40 @@ def ffprobe_streams(path_to_file):
     popen_output = popen.communicate()[0]
     log.debug("FFprobe output: {}".format(popen_output))
     return json.loads(popen_output)["streams"]
+
+
+def get_latest_version(asset_name, subset_name):
+    """Retrieve latest version from `asset_name`, and `subset_name`.
+
+    Args:
+        asset_name (str): Name of asset.
+        subset_name (str): Name of subset.
+    """
+    # Get asset
+    asset_name = io.find_one(
+        {"type": "asset", "name": asset_name}, projection={"name": True}
+    )
+
+    subset = io.find_one(
+        {"type": "subset", "name": subset_name, "parent": asset_name["_id"]},
+        projection={"_id": True, "name": True},
+    )
+
+    # Check if subsets actually exists.
+    assert subset, "No subsets found."
+
+    # Get version
+    version_projection = {
+        "name": True,
+        "parent": True,
+    }
+
+    version = io.find_one(
+        {"type": "version", "parent": subset["_id"]},
+        projection=version_projection,
+        sort=[("name", -1)],
+    )
+
+    assert version, "No version found, this is a bug"
+
+    return version
diff --git a/pype/plugins/nuke/publish/collect_review.py b/pype/plugins/nuke/publish/collect_review.py
index c95c94541d..e7e8da19a1 100644
--- a/pype/plugins/nuke/publish/collect_review.py
+++ b/pype/plugins/nuke/publish/collect_review.py
@@ -1,4 +1,7 @@
 import pyblish.api
+import pype.api
+from avalon import io, api
+
 import nuke
 
 
@@ -23,6 +26,21 @@ class CollectReview(pyblish.api.InstancePlugin):
         if not node["review"].value():
             return
 
+        # Add audio to instance if it exists.
+        try:
+            version = pype.api.get_latest_version(
+                instance.context.data["assetEntity"]["name"], "audioMain"
+            )
+            representation = io.find_one(
+                {"type": "representation", "parent": version["_id"]}
+            )
+            instance.data["audio"] = [{
+                "offset": 0,
+                "filename": api.get_representation_path(representation)
+            }]
+        except AssertionError:
+            pass
+
         instance.data["families"].append("review")
         instance.data['families'].append('ftrack')