Translate JetSimpleNameReference to Kotlin
This commit is contained in:
Alexey Sedunov
@@ -1,102 +0,0 @@
|
||||
/*
|
||||
* Copyright 2010-2013 JetBrains s.r.o.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.jetbrains.jet.plugin.references;
|
||||
|
||||
import com.intellij.openapi.project.Project;
|
||||
import com.intellij.openapi.util.TextRange;
|
||||
import com.intellij.psi.PsiElement;
|
||||
import com.intellij.psi.tree.IElementType;
|
||||
import com.intellij.psi.util.PsiTreeUtil;
|
||||
import com.intellij.util.IncorrectOperationException;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.jet.lang.psi.*;
|
||||
import org.jetbrains.jet.lang.psi.psiUtil.PsiUtilPackage;
|
||||
import org.jetbrains.jet.lang.resolve.name.FqName;
|
||||
import org.jetbrains.jet.lexer.JetTokens;
|
||||
import org.jetbrains.jet.plugin.codeInsight.CodeInsightPackage;
|
||||
import org.jetbrains.jet.plugin.codeInsight.ShortenReferences;
|
||||
import org.jetbrains.jet.plugin.refactoring.RefactoringPackage;
|
||||
|
||||
public class JetSimpleNameReference extends JetSimpleReference<JetSimpleNameExpression> {
|
||||
public JetSimpleNameReference(@NotNull JetSimpleNameExpression jetSimpleNameExpression) {
|
||||
super(jetSimpleNameExpression);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public TextRange getRangeInElement() {
|
||||
return new TextRange(0, getElement().getTextLength());
|
||||
}
|
||||
|
||||
@Override
|
||||
public PsiElement handleElementRename(String newElementName) throws IncorrectOperationException {
|
||||
IElementType type = getExpression().getReferencedNameElementType();
|
||||
Project project = getExpression().getProject();
|
||||
PsiElement element;
|
||||
if (JetTokens.FIELD_IDENTIFIER == type) {
|
||||
element = JetPsiFactory.createFieldIdentifier(project, newElementName);
|
||||
}
|
||||
else if (JetTokens.LABEL_IDENTIFIER == type) {
|
||||
element = JetPsiFactory.createClassLabel(project, newElementName);
|
||||
}
|
||||
else {
|
||||
element = JetPsiFactory.createNameIdentifier(project, newElementName);
|
||||
}
|
||||
return getExpression().getReferencedNameElement().replace(element);
|
||||
}
|
||||
|
||||
// By default reference binding is delayed
|
||||
@NotNull
|
||||
@Override
|
||||
public PsiElement bindToElement(@NotNull PsiElement element) {
|
||||
return bindToElement(element, false);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public PsiElement bindToElement(@NotNull PsiElement element, boolean bindImmediately) {
|
||||
FqName fqName = PsiUtilPackage.getFqName(element);
|
||||
return fqName != null ? bindToFqName(fqName, bindImmediately) : getExpression();
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public PsiElement bindToFqName(@NotNull FqName fqName, boolean forceImmediateBinding) {
|
||||
JetSimpleNameExpression currentExpression = getExpression();
|
||||
|
||||
JetElement qualifier = RefactoringPackage.changeQualifiedName(currentExpression, fqName);
|
||||
JetSimpleNameExpression newExpression = (JetSimpleNameExpression) PsiUtilPackage.getQualifiedElementSelector(qualifier);
|
||||
assert newExpression != null : "No selector in qualified element";
|
||||
|
||||
//noinspection unchecked
|
||||
boolean needToShorten =
|
||||
PsiTreeUtil.getParentOfType(currentExpression, JetImportDirective.class, JetPackageDirective.class) == null;
|
||||
if (needToShorten) {
|
||||
if (forceImmediateBinding) {
|
||||
ShortenReferences.instance$.process(PsiUtilPackage.getOutermostNonInterleavingQualifiedElement(newExpression));
|
||||
}
|
||||
else {
|
||||
CodeInsightPackage.addElementToShorteningWaitSet(newExpression.getProject(), newExpression);
|
||||
}
|
||||
}
|
||||
|
||||
return newExpression;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return JetSimpleNameReference.class.getSimpleName() + ": " + getExpression().getText();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,78 @@
|
||||
/*
|
||||
* Copyright 2010-2014 JetBrains s.r.o.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.jetbrains.jet.plugin.references
|
||||
|
||||
import com.intellij.openapi.util.TextRange
|
||||
import com.intellij.psi.PsiElement
|
||||
import com.intellij.psi.util.PsiTreeUtil
|
||||
import org.jetbrains.jet.lang.psi.*
|
||||
import org.jetbrains.jet.lang.resolve.name.FqName
|
||||
import org.jetbrains.jet.lexer.JetTokens
|
||||
import org.jetbrains.jet.plugin.codeInsight.ShortenReferences
|
||||
import org.jetbrains.jet.lang.psi.psiUtil.getFqName
|
||||
import org.jetbrains.jet.plugin.refactoring.changeQualifiedName
|
||||
import org.jetbrains.jet.lang.psi.psiUtil.getQualifiedElementSelector
|
||||
import org.jetbrains.jet.lang.psi.psiUtil.getOutermostNonInterleavingQualifiedElement
|
||||
import org.jetbrains.jet.plugin.codeInsight.addElementToShorteningWaitSet
|
||||
|
||||
public class JetSimpleNameReference(
|
||||
jetSimpleNameExpression: JetSimpleNameExpression
|
||||
) : JetSimpleReference<JetSimpleNameExpression>(jetSimpleNameExpression) {
|
||||
|
||||
override fun getRangeInElement(): TextRange = TextRange(0, getElement().getTextLength())
|
||||
|
||||
public override fun handleElementRename(newElementName: String?): PsiElement? {
|
||||
if (newElementName == null) return null;
|
||||
|
||||
val project = expression.getProject()
|
||||
val element = when (expression.getReferencedNameElementType()) {
|
||||
JetTokens.FIELD_IDENTIFIER -> JetPsiFactory.createFieldIdentifier(project, newElementName)
|
||||
JetTokens.LABEL_IDENTIFIER -> JetPsiFactory.createClassLabel(project, newElementName)
|
||||
else -> JetPsiFactory.createNameIdentifier(project, newElementName)
|
||||
}
|
||||
|
||||
return expression.getReferencedNameElement().replace(element)
|
||||
}
|
||||
|
||||
// By default reference binding is delayed
|
||||
override fun bindToElement(element: PsiElement): PsiElement = bindToElement(element, false)
|
||||
|
||||
public fun bindToElement(element: PsiElement, bindImmediately: Boolean): PsiElement {
|
||||
return element.getFqName()?.let { fqName -> bindToFqName(fqName, bindImmediately) } ?: expression
|
||||
}
|
||||
|
||||
public fun bindToFqName(fqName: FqName, forceImmediateBinding: Boolean): PsiElement {
|
||||
val newExpression = expression.changeQualifiedName(fqName).getQualifiedElementSelector() as JetSimpleNameExpression
|
||||
|
||||
val needToShorten =
|
||||
PsiTreeUtil.getParentOfType(expression, javaClass<JetImportDirective>(), javaClass<JetPackageDirective>()) == null
|
||||
if (needToShorten) {
|
||||
if (forceImmediateBinding) {
|
||||
ShortenReferences.process(newExpression.getOutermostNonInterleavingQualifiedElement())
|
||||
}
|
||||
else {
|
||||
newExpression.getProject().addElementToShorteningWaitSet(newExpression)
|
||||
}
|
||||
}
|
||||
|
||||
return newExpression
|
||||
}
|
||||
|
||||
override fun toString(): String {
|
||||
return javaClass<JetSimpleNameReference>().getSimpleName() + ": " + expression.getText()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user