Node Graph: Always show context menu (#56876)

* Ensure context menu always appears

* Close the context menu after selecting the alternative layout

Co-authored-by: Hamas Shafiq <hamas.shafiq@grafana.com>

public/app/plugins/panel/nodeGraph/useContextMenu.tsx

@@ -28,17 +28,25 @@ export function useContextMenu(
 
   const onNodeOpen = useCallback(
     (event: MouseEvent<SVGElement>, node: NodeDatum) => {
-      const extraNodeItem = config.gridLayout
+      let label = 'Show in Grid layout';
-        ? [
+      let showGridLayout = true;
-            {
+
-              label: 'Show in Graph layout',
+      if (config.gridLayout) {
-              onClick: (node: NodeDatum) => {
+        label = 'Show in Graph layout';
-                setFocusedNodeId(node.id);
+        showGridLayout = false;
-                setConfig({ ...config, gridLayout: false });
+      }
-              },
+
-            },
+      const extraNodeItem = [
-          ]
+        {
-        : undefined;
+          label: label,
+          onClick: (node: NodeDatum) => {
+            setFocusedNodeId(node.id);
+            setConfig({ ...config, gridLayout: showGridLayout });
+            setMenu(undefined);
+          },
+        },
+      ];
+
       const renderer = getItemsRenderer(getLinks(nodes, node.dataFrameRowIndex), node, extraNodeItem);
 
       if (renderer) {