Specify relevant declaration type in "create actual" fix #KT-20325 Fixed

2017-10-03 12:06:59 +03:00
parent 93109c0d0e
commit d85c2dbe1f
10 changed files with 31 additions and 5 deletions
@@ -156,7 +156,19 @@ class CreateActualClassFix(
    generateClassOrObject(project, element, actualNeeded = true)
 }) {

-    override val elementType = if ((element as? KtClass)?.isInterface() == true) "interface" else "class"
+    override val elementType = run {
+        val element = element
+        when (element) {
+            is KtObjectDeclaration -> "object"
+            is KtClass -> when {
+                element.isInterface() -> "interface"
+                element.isEnum() -> "enum class"
+                element.isAnnotation() -> "annotation class"
+                else -> "class"
+            }
+            else -> "class"
+        }
+    }
 }

 class CreateActualPropertyFix(