azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Converted to Kotlin (step 1)

Browse Source
This commit is contained in:
Valentin Kipyatkov
2015-04-28 16:04:26 +03:00
parent 995b82ae62
commit ea62fafa69
1 changed files with 41 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/kotlin/idea/intentions/ReconstructTypeInCastOrIsAction.java
+41 -48
View File
@@ -14,74 +14,67 @@
* limitations under the License. * limitations under the License.
*/ */
package org.jetbrains.kotlin.idea.intentions; package org.jetbrains.kotlin.idea.intentions
import com.intellij.codeInsight.intention.PsiElementBaseIntentionAction; import com.intellij.codeInsight.intention.PsiElementBaseIntentionAction
import com.intellij.openapi.editor.Editor; import com.intellij.openapi.editor.Editor
import com.intellij.openapi.project.Project; import com.intellij.openapi.project.Project
import com.intellij.psi.PsiElement; import com.intellij.psi.PsiElement
import com.intellij.psi.util.PsiTreeUtil; import com.intellij.psi.util.PsiTreeUtil
import org.jetbrains.annotations.NotNull; import org.jetbrains.kotlin.idea.JetBundle
import org.jetbrains.kotlin.idea.JetBundle; import org.jetbrains.kotlin.idea.caches.resolve.*
import org.jetbrains.kotlin.idea.caches.resolve.ResolvePackage; import org.jetbrains.kotlin.idea.util.IdeDescriptorRenderers
import org.jetbrains.kotlin.idea.util.IdeDescriptorRenderers; import org.jetbrains.kotlin.idea.util.ShortenReferences
import org.jetbrains.kotlin.idea.util.ShortenReferences; import org.jetbrains.kotlin.psi.*
import org.jetbrains.kotlin.psi.*; import org.jetbrains.kotlin.resolve.BindingContext
import org.jetbrains.kotlin.resolve.BindingContext; import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode
import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode; import org.jetbrains.kotlin.types.JetType
import org.jetbrains.kotlin.types.JetType;
import static org.jetbrains.kotlin.psi.PsiPackage.JetPsiFactory; public class ReconstructTypeInCastOrIsAction : PsiElementBaseIntentionAction() {
override fun getFamilyName(): String {
public class ReconstructTypeInCastOrIsAction extends PsiElementBaseIntentionAction { return JetBundle.message("replace.by.reconstructed.type.family.name")
@NotNull
@Override
public String getFamilyName() {
return JetBundle.message("replace.by.reconstructed.type.family.name");
} }
@Override override fun invoke(project: Project, editor: Editor, element: PsiElement) {
public void invoke(@NotNull Project project, Editor editor, @NotNull PsiElement element) { val typeRef = PsiTreeUtil.getTopmostParentOfType<JetTypeReference>(element, javaClass<JetTypeReference>())
JetTypeReference typeRef = PsiTreeUtil.getTopmostParentOfType(element, JetTypeReference.class); assert(typeRef != null) { "Must be checked by isAvailable(): " + element }
assert typeRef != null : "Must be checked by isAvailable(): " + element;
JetType type = getReconstructedType(typeRef); val type = getReconstructedType(typeRef)
JetTypeReference newType = JetPsiFactory(typeRef).createType(IdeDescriptorRenderers.SOURCE_CODE.renderType(type)); val newType = JetPsiFactory(typeRef).createType(IdeDescriptorRenderers.SOURCE_CODE.renderType(type))
JetTypeReference replaced = (JetTypeReference) typeRef.replace(newType); val replaced = typeRef!!.replace(newType) as JetTypeReference
ShortenReferences.DEFAULT.process(replaced); ShortenReferences.DEFAULT.process(replaced)
} }
@Override override fun isAvailable(project: Project, editor: Editor, element: PsiElement): Boolean {
public boolean isAvailable(@NotNull Project project, Editor editor, @NotNull PsiElement element) { val typeRef = PsiTreeUtil.getTopmostParentOfType<JetTypeReference>(element, javaClass<JetTypeReference>())
JetTypeReference typeRef = PsiTreeUtil.getTopmostParentOfType(element, JetTypeReference.class); if (typeRef == null) return false
if (typeRef == null) return false;
// Only user types (like Foo) are interesting // Only user types (like Foo) are interesting
JetTypeElement typeElement = typeRef.getTypeElement(); val typeElement = typeRef.getTypeElement()
if (!(typeElement instanceof JetUserType)) return false; if (typeElement !is JetUserType) return false
// If there are generic arguments already, there's nothing to reconstruct // If there are generic arguments already, there's nothing to reconstruct
if (!((JetUserType) typeElement).getTypeArguments().isEmpty()) return false; if (!typeElement.getTypeArguments().isEmpty()) return false
// We must be on the RHS of as/as?/is/!is or inside an is/!is-condition in when() // We must be on the RHS of as/as?/is/!is or inside an is/!is-condition in when()
JetExpression outerExpression = PsiTreeUtil.getParentOfType(typeRef, JetExpression.class); val outerExpression = PsiTreeUtil.getParentOfType<JetExpression>(typeRef, javaClass<JetExpression>())
if (!(outerExpression instanceof JetBinaryExpressionWithTypeRHS)) { if (outerExpression !is JetBinaryExpressionWithTypeRHS) {
JetWhenConditionIsPattern outerIsCondition = PsiTreeUtil.getParentOfType(typeRef, JetWhenConditionIsPattern.class); val outerIsCondition = PsiTreeUtil.getParentOfType<JetWhenConditionIsPattern>(typeRef, javaClass<JetWhenConditionIsPattern>())
if (outerIsCondition == null) return false; if (outerIsCondition == null) return false
} }
JetType type = getReconstructedType(typeRef); val type = getReconstructedType(typeRef)
if (type == null || type.isError()) return false; if (type == null || type.isError()) return false
// No type parameters expected => nothing to reconstruct // No type parameters expected => nothing to reconstruct
if (type.getConstructor().getParameters().isEmpty()) return false; if (type.getConstructor().getParameters().isEmpty()) return false
setText(JetBundle.message("replace.by.reconstructed.type", type)); setText(JetBundle.message("replace.by.reconstructed.type", type))
return true; return true
} }
private static JetType getReconstructedType(JetTypeReference typeRef) { private fun getReconstructedType(typeRef: JetTypeReference): JetType? {
return ResolvePackage.analyze(typeRef, BodyResolveMode.FULL).get(BindingContext.TYPE, typeRef); return typeRef.analyze(BodyResolveMode.FULL).get<JetTypeReference, JetType>(BindingContext.TYPE, typeRef)
} }
} }