IJPL-178449 Sometimes toolbar shows only main menu after Zen and DFM modes until the next click

GitOrigin-RevId: 050ede69672c523758fef17a21389aa1d1425a70

platform/platform-impl/src/com/intellij/openapi/wm/impl/customFrameDecorations/header/toolbar/ToolbarFrameHeader.kt

@@ -319,11 +319,11 @@ internal class ToolbarFrameHeader(
         toolbarPlaceholder.add(newToolbar, BorderLayout.CENTER)
       }
 
+      mainMenuWithButton.recalculateWidth(newToolbar)
       newToolbar.revalidate()
       newToolbar.repaint()
       toolbarPlaceholder.revalidate()
       toolbarPlaceholder.repaint()
-      mainMenuWithButton.recalculateWidth(toolbar)
       this@ToolbarFrameHeader.revalidate()
       this@ToolbarFrameHeader.repaint()
     }

platform/platform-impl/src/com/intellij/openapi/wm/impl/headertoolbar/MainMenuWithButton.kt

@@ -22,7 +22,6 @@ import org.jetbrains.annotations.ApiStatus
 import java.util.concurrent.ConcurrentLinkedDeque
 import javax.swing.Icon
 import javax.swing.JFrame
-import javax.swing.JMenu
 
 class MainMenuWithButton(
   val coroutineScope: CoroutineScope, private val frame: JFrame,
@@ -101,9 +100,9 @@ class MainMenuWithButton(
           toolbarMainMenu.add(item)
         }
         toolbarMainMenu.rootMenuItems.forEach { it.updateUI() }
-        toolbar.revalidate()
-        toolbar.repaint()
       }
+      toolbar.revalidate()
+      toolbar.repaint()
     }
   }