fix popups: heavy/medium: restore state of factory after popup creation + delegate...

[idea/community.git] / platform / platform-api / src / com / intellij / openapi / ui / popup / util / PopupUtil.java

diff --git a/platform/platform-api/src/com/intellij/openapi/ui/popup/util/PopupUtil.java b/platform/platform-api/src/com/intellij/openapi/ui/popup/util/PopupUtil.java
index b8d9f8f87e14c86c6298c650f839f46893e8d675..02a92ed721f0bf21a61da1297e02c62410fac1c5 100644 (file)
--- a/platform/platform-api/src/com/intellij/openapi/ui/popup/util/PopupUtil.java
+++ b/platform/platform-api/src/com/intellij/openapi/ui/popup/util/PopupUtil.java
@@ -15,13 +15,17 @@
  */
 package com.intellij.openapi.ui.popup.util;
 
+import com.intellij.openapi.diagnostic.Logger;
 import com.intellij.openapi.ui.popup.JBPopup;
+import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
 import javax.swing.*;
 import java.awt.*;
+import java.lang.reflect.Method;
 
 public class PopupUtil {
+  private static final Logger LOG = Logger.getInstance("#com.intellij.openapi.ui.popup.util.PopupUtil");
 
   private PopupUtil() {
   }
@@ -46,4 +50,29 @@ public class PopupUtil {
     }
   }
 
+  public static void setPopupType(@NotNull final PopupFactory factory, final int type) {
+    try {
+      final Method method = PopupFactory.class.getDeclaredMethod("setPopupType", int.class);
+      method.setAccessible(true);
+      method.invoke(factory, type);
+    }
+    catch (Throwable e) {
+      LOG.error(e);
+    }
+  }
+
+  public static int getPopupType(@NotNull final PopupFactory factory) {
+    try {
+      final Method method = PopupFactory.class.getDeclaredMethod("getPopupType");
+      method.setAccessible(true);
+      final Object result = method.invoke(factory);
+      return result instanceof Integer ? (Integer) result : -1;
+    }
+    catch (Throwable e) {
+      LOG.error(e);
+    }
+
+    return -1;
+  }
+
 }
\ No newline at end of file