[auto import] fix Scala 3 enum case import suggestion #SCL-22747 fixed

Author: timaliberdov

scala/scala-impl/src/org/jetbrains/plugins/scala/autoImport/quickFix/ScalaImportTypeFix.scala

@@ -18,7 +18,7 @@ import org.jetbrains.plugins.scala.lang.psi.api.ScalaFile
 import org.jetbrains.plugins.scala.lang.psi.api.base.ScReference
 import org.jetbrains.plugins.scala.lang.psi.api.base.types.{ScSimpleTypeElement, ScTypeProjection}
 import org.jetbrains.plugins.scala.lang.psi.api.expr.{ScMethodCall, ScSugarCallExpr}
-import org.jetbrains.plugins.scala.lang.psi.api.statements.{ScFunction, ScTypeAlias}
+import org.jetbrains.plugins.scala.lang.psi.api.statements.{ScEnumCase, ScFunction, ScTypeAlias}
 import org.jetbrains.plugins.scala.lang.psi.api.toplevel.templates.ScTemplateBody
 import org.jetbrains.plugins.scala.lang.psi.api.toplevel.typedef._
 import org.jetbrains.plugins.scala.lang.psi.api.toplevel.{ScPackaging, ScTypedDefinition}
@@ -77,6 +77,7 @@ object ScalaImportTypeFix {
 
   @annotation.tailrec
   private[this] def notInner(clazz: PsiClass, ref: PsiElement): Boolean = clazz match {
+    case enumCase: ScEnumCase => notInner(enumCase.enumParent, ref)
     case o: ScObject if o.isSyntheticObject =>
       val companion = getCompanionModule(o)
       companion match {

scala/scala-impl/test/org/jetbrains/plugins/scala/lang/autoImport/ScalaImportTypeFixTest.scala

@@ -500,4 +500,48 @@ class Scala3ImportTypeFixTest extends ScalaImportTypeFixTestBase {
        |""".stripMargin,
     "MyObject.MyEnumInClass"
   )
+
+  def testEnumSingletonCase(): Unit = {
+    val fileText =
+      s"""enum Source:
+         |  case Foo
+         |
+         |object Target:
+         |  val foo = ${CARET}Foo
+         |""".stripMargin
+    val qNameToImport = "Source.Foo"
+
+    checkElementsToImport(
+      fileText,
+      qNameToImport
+    )
+
+    doTest(
+      fileText,
+      expectedText = s"import $qNameToImport\n\n${fileText.replace(CARET, "")}",
+      selected = qNameToImport
+    )
+  }
+
+  def testEnumClassCase(): Unit = {
+    val fileText =
+      s"""enum Source:
+         |  case Foo(i: Int)
+         |
+         |object Target:
+         |  val foo = ${CARET}Foo(2)
+         |""".stripMargin
+    val qNameToImport = "Source.Foo"
+
+    checkElementsToImport(
+      fileText,
+      qNameToImport
+    )
+
+    doTest(
+      fileText,
+      expectedText = s"import $qNameToImport\n\n${fileText.replace(CARET, "")}",
+      selected = qNameToImport
+    )
+  }
 }