Fix T92505: previewing specific node outputs did not work anymore

This was missing from my refactor in rB5bfe09df2244cb9de0b6554a378eecef77b1e75d.

1 files changed, 8 insertions, 0 deletions

diff --git a/source/blender/editors/space_node/node_relationships.cc b/source/blender/editors/space_node/node_relationships.cc
index 459608a67ea..76aad684b4c 100644
--- a/source/blender/editors/space_node/node_relationships.cc
+++ b/source/blender/editors/space_node/node_relationships.cc
@@ -768,6 +768,14 @@ static const NodeRef *get_existing_viewer(const NodeTreeRef &tree)
 static const OutputSocketRef *find_output_socket_to_be_viewed(const NodeRef *active_viewer_node,
                                                               const NodeRef &node_to_view)
 {
+  /* Check if any of the output sockets is selected, which is the case when the user just clicked
+   * on the socket. */
+  for (const OutputSocketRef *output_socket : node_to_view.outputs()) {
+    if (output_socket->bsocket()->flag & SELECT) {
+      return output_socket;
+    }
+  }
+
   const OutputSocketRef *last_socket_linked_to_viewer = nullptr;
   if (active_viewer_node != nullptr) {
     for (const OutputSocketRef *output_socket : node_to_view.outputs()) {