[intention-preview] CreateClassFromUsageFix: preview; tests for preview

GitOrigin-RevId: 2eb36ee066743ffcacf4fae2968ba9d6d9c8bcff

java/java-impl/src/com/intellij/codeInsight/daemon/impl/quickfix/CreateClassFromUsageBaseFix.java

@@ -19,6 +19,7 @@ import com.intellij.codeInsight.ExpectedTypeInfo;
 import com.intellij.codeInsight.ExpectedTypesProvider;
 import com.intellij.codeInsight.daemon.QuickFixBundle;
 import com.intellij.codeInsight.intention.impl.BaseIntentionAction;
+import com.intellij.codeInsight.intention.preview.IntentionPreviewInfo;
 import com.intellij.codeInspection.util.IntentionName;
 import com.intellij.ide.scratch.ScratchUtil;
 import com.intellij.openapi.diagnostic.Logger;
@@ -45,6 +46,9 @@ public abstract class CreateClassFromUsageBaseFix extends BaseIntentionAction {
     myRefElement = SmartPointerManager.getInstance(refElement.getProject()).createSmartPsiElementPointer(refElement);
   }
 
+  @Override
+  public abstract @NotNull IntentionPreviewInfo generatePreview(@NotNull Project project, @NotNull Editor editor, @NotNull PsiFile file);
+
   protected abstract @IntentionName String getText(String varName);
 
   private boolean isAvailableInContext(@NotNull final PsiJavaCodeReferenceElement element) {

java/java-impl/src/com/intellij/codeInsight/daemon/impl/quickfix/CreateClassFromUsageFix.java

@@ -2,17 +2,19 @@
 package com.intellij.codeInsight.daemon.impl.quickfix;
 
 import com.intellij.codeInsight.FileModificationService;
+import com.intellij.codeInsight.intention.preview.IntentionPreviewInfo;
 import com.intellij.codeInspection.CommonQuickFixBundle;
+import com.intellij.ide.highlighter.JavaFileType;
 import com.intellij.ide.util.PsiNavigationSupport;
 import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.editor.Editor;
 import com.intellij.openapi.fileEditor.ex.IdeDocumentHistory;
 import com.intellij.openapi.project.Project;
+import com.intellij.openapi.util.text.StringUtil;
 import com.intellij.pom.Navigatable;
-import com.intellij.psi.PsiClass;
-import com.intellij.psi.PsiDocumentManager;
-import com.intellij.psi.PsiFile;
-import com.intellij.psi.PsiJavaCodeReferenceElement;
+import com.intellij.psi.*;
+import com.intellij.psi.codeStyle.CodeStyleManager;
+import com.intellij.psi.util.PsiTreeUtil;
 import com.intellij.util.IncorrectOperationException;
 import org.jetbrains.annotations.NotNull;
 
@@ -22,6 +24,23 @@ public class CreateClassFromUsageFix extends CreateClassFromUsageBaseFix {
     super(kind, refElement);
   }
 
+  @Override
+  public @NotNull IntentionPreviewInfo generatePreview(@NotNull Project project, @NotNull Editor editor, @NotNull PsiFile file) {
+    PsiJavaCodeReferenceElement element = getRefElement();
+    if (element == null) return IntentionPreviewInfo.EMPTY;
+    element = PsiTreeUtil.findSameElementInCopy(element, file);
+    String superClassName = getSuperClassName(element);
+    PsiClass aClass = myKind.create(JavaPsiFacade.getElementFactory(project), element.getReferenceName());
+    if (StringUtil.isNotEmpty(superClassName) &&
+        (myKind != CreateClassKind.ENUM || !superClassName.equals(CommonClassNames.JAVA_LANG_ENUM)) &&
+        (myKind != CreateClassKind.RECORD || !superClassName.equals(CommonClassNames.JAVA_LANG_RECORD))) {
+      CreateFromUsageUtils.setupSuperClassReference(aClass, superClassName);
+    }
+    CreateFromUsageBaseFix.setupGenericParameters(aClass, element);
+    CodeStyleManager.getInstance(project).reformat(aClass);
+    return new IntentionPreviewInfo.CustomDiff(JavaFileType.INSTANCE, "", aClass.getText());
+  }
+
   @Override
   public String getText(String varName) {
     return CommonQuickFixBundle.message("fix.create.title.x", myKind.getDescriptionAccusative(), varName);

java/java-impl/src/com/intellij/codeInsight/daemon/impl/quickfix/CreateFromUsageBaseFix.java

@@ -417,15 +417,16 @@ public abstract class CreateFromUsageBaseFix extends BaseIntentionAction {
       }
     }
     int idx = 0;
+    PsiTypeParameterList typeParameterList = Objects.requireNonNull(targetClass.getTypeParameterList());
     for (PsiType type : ref.getTypeParameters()) {
       final PsiClass psiClass = PsiUtil.resolveClassInType(type);
       if (psiClass instanceof PsiTypeParameter) {
-        targetClass.getTypeParameterList().add(factory.createTypeParameterFromText(psiClass.getName(), null));
+        typeParameterList.add(factory.createTypeParameterFromText(psiClass.getName(), null));
       } else {
         while (true) {
           final @NonNls String paramName = idx > 0 ? "T" + idx : "T";
           if (typeParamNames.add(paramName)) {
-            targetClass.getTypeParameterList().add(factory.createTypeParameterFromText(paramName, null));
+            typeParameterList.add(factory.createTypeParameterFromText(paramName, null));
             break;
           }
           idx++;

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/after1.java

@@ -1,4 +1,4 @@
-// "Create class 'Abc'" "true"
+// "Create class 'Abc'" "true-preview"
 public class Test {
   public foo(int ppp) {
     Abc.foo();

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterCast.java

@@ -1,4 +1,4 @@
-// "Create class 'ClassB'" "true"
+// "Create class 'ClassB'" "true-preview"
 public class ClassA
 {
   public void mA(Object arg)

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterClassUsedInWildcard.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 import java.util.Collection;
 public class Test {
   public Collection<? extends Foo> getSomething() {

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterEnum.java

@@ -1,4 +1,4 @@
-// "Create enum 'Foo'" "true"
+// "Create enum 'Foo'" "true-preview"
 public class Test {
   void f(Class<? extends Enum> e) {}
   {

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterEnumInstanceofCapture.java

@@ -1,4 +1,4 @@
-// "Create enum 'Foo'" "true"
+// "Create enum 'Foo'" "true-preview"
 public interface Test {
   default void foo(java.util.List<? extends Test> l){
     if (l.get(0) instanceof Foo) {}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterEnumUsedInCall.java

@@ -1,4 +1,4 @@
-// "Create enum 'Foo'" "true"
+// "Create enum 'Foo'" "true-preview"
 public class Test {
   void f(I i) {}
   {

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterExtends.java

@@ -1,4 +1,4 @@
-// "Create class 'abc'" "true"
+// "Create class 'abc'" "true-preview"
 public class Test extends abc {
 }
 

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterFieldType.java

@@ -1,4 +1,4 @@
-// "Create class 'Abc'" "true"
+// "Create class 'Abc'" "true-preview"
 public class Test {
   private Abc fld;
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterInNewArgs.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 public class Test {
   void foo() {
     new Test(Foo.FIELD);

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterMethodType.java

@@ -1,4 +1,4 @@
-// "Create class 'abc'" "true"
+// "Create class 'abc'" "true-preview"
 public class Test {
   private abc foo() {}
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterParameterType.java

@@ -1,4 +1,4 @@
-// "Create class 'Abc'" "true"
+// "Create class 'Abc'" "true-preview"
 public class Test {
   private void foo(Abc param) {}
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterPermits.java

@@ -1,4 +1,4 @@
-// "Create class 'abc'" "true"
+// "Create class 'abc'" "true-preview"
 public class Test permits abc {
 }
 

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterSameNamedField.java

@@ -1,4 +1,4 @@
-// "Create class 'Abc'" "true"
+// "Create class 'Abc'" "true-preview"
 public class Test {
   Abc Abc;
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterSuperInsideInner.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 class MyTest {
   {
     Class<? extends Throwable> c = Bar.Foo.class;

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterTypeParameter.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 public class Test {
   List l = new ArrayList<Foo>();
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/afterTypeParameterCreated.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 public class Test {
   <R> void foo(Foo<R, String> f){}
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/before1.java

@@ -1,4 +1,4 @@
-// "Create class 'Abc'" "true"
+// "Create class 'Abc'" "true-preview"
 public class Test {
   public foo(int ppp) {
     <caret>Abc.foo();

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeCast.java

@@ -1,4 +1,4 @@
-// "Create class 'ClassB'" "true"
+// "Create class 'ClassB'" "true-preview"
 public class ClassA
 {
   public void mA(Object arg)

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeClassUsedInWildcard.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 import java.util.Collection;
 public class Test {
   public Collection<? extends Fo<caret>o> getSomething() {

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeEnum.java

@@ -1,4 +1,4 @@
-// "Create enum 'Foo'" "true"
+// "Create enum 'Foo'" "true-preview"
 public class Test {
   void f(Class<? extends Enum> e) {}
   {

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeEnumInstanceofCapture.java

@@ -1,4 +1,4 @@
-// "Create enum 'Foo'" "true"
+// "Create enum 'Foo'" "true-preview"
 public interface Test {
   default void foo(java.util.List<? extends Test> l){
     if (l.get(0) instanceof Fo<caret>o) {}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeEnumUsedInCall.java

@@ -1,4 +1,4 @@
-// "Create enum 'Foo'" "true"
+// "Create enum 'Foo'" "true-preview"
 public class Test {
   void f(I i) {}
   {

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeExtends.java

@@ -1,3 +1,3 @@
-// "Create class 'abc'" "true"
+// "Create class 'abc'" "true-preview"
 public class Test extends <caret>abc {
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeFieldType.java

@@ -1,4 +1,4 @@
-// "Create class 'Abc'" "true"
+// "Create class 'Abc'" "true-preview"
 public class Test {
   private <caret>Abc fld;
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeInNewArgs.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 public class Test {
   void foo() {
     new Test(F<caret>oo.FIELD);

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeMethodType.java

@@ -1,4 +1,4 @@
-// "Create class 'abc'" "true"
+// "Create class 'abc'" "true-preview"
 public class Test {
   private <caret>abc foo() {}
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeParameterType.java

@@ -1,4 +1,4 @@
-// "Create class 'Abc'" "true"
+// "Create class 'Abc'" "true-preview"
 public class Test {
   private void foo(<caret>Abc param) {}
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforePermits.java

@@ -1,3 +1,3 @@
-// "Create class 'abc'" "true"
+// "Create class 'abc'" "true-preview"
 public class Test permits <caret>abc {
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeSameNamedField.java

@@ -1,4 +1,4 @@
-// "Create class 'Abc'" "true"
+// "Create class 'Abc'" "true-preview"
 public class Test {
   <caret>Abc Abc;
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeSuperInsideInner.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 class MyTest {
   {
     Class<? extends Throwable> c = Bar.Fo<caret>o.class;

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeTypeParameter.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 public class Test {
   List l = new ArrayList<<caret>Foo>();
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/beforeTypeParameterCreated.java

@@ -1,4 +1,4 @@
-// "Create class 'Foo'" "true"
+// "Create class 'Foo'" "true-preview"
 public class Test {
   <R> void foo(Fo<caret>o<R, String> f){}
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/preview1.java

@@ -0,0 +1,2 @@
+public class Abc {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewCast.java

@@ -0,0 +1,2 @@
+public class ClassB {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewClassUsedInWildcard.java

@@ -0,0 +1,2 @@
+public class Foo {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewEnum.java

@@ -0,0 +1 @@
+public enum Foo {}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewEnumInstanceofCapture.java

@@ -0,0 +1 @@
+public enum Foo {}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewEnumUsedInCall.java

@@ -0,0 +1 @@
+public enum Foo implements I {}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewExtends.java

@@ -0,0 +1,2 @@
+public class abc {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewFieldType.java

@@ -0,0 +1,2 @@
+public class Abc {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewInNewArgs.java

@@ -0,0 +1,2 @@
+public class Foo {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewMethodType.java

@@ -0,0 +1,2 @@
+public class abc {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewParameterType.java

@@ -0,0 +1,2 @@
+public class Abc {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewPermits.java

@@ -0,0 +1,2 @@
+public class abc extends Test {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewSameNamedField.java

@@ -0,0 +1,2 @@
+public class Abc {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewSuperInsideInner.java

@@ -0,0 +1,2 @@
+public class Foo extends Throwable {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewTypeParameter.java

@@ -0,0 +1,2 @@
+public class Foo {
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createClassFromUsage/previewTypeParameterCreated.java

@@ -0,0 +1,2 @@
+public class Foo<R, T> {
+}