Converted to Kotlin
This commit is contained in:
Valentin Kipyatkov
-109
@@ -1,109 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro;
|
|
||||||
|
|
||||||
import com.intellij.codeInsight.template.Template;
|
|
||||||
import com.intellij.codeInsight.template.TemplateEditingAdapter;
|
|
||||||
import com.intellij.codeInsight.template.impl.TemplateManagerImpl;
|
|
||||||
import com.intellij.codeInsight.template.impl.TemplateState;
|
|
||||||
import com.intellij.openapi.editor.Editor;
|
|
||||||
import com.intellij.openapi.project.Project;
|
|
||||||
import com.intellij.openapi.util.Key;
|
|
||||||
import com.intellij.openapi.util.TextRange;
|
|
||||||
import com.intellij.psi.PsiDocumentManager;
|
|
||||||
import com.intellij.psi.PsiElement;
|
|
||||||
import com.intellij.psi.PsiFile;
|
|
||||||
import org.jetbrains.kotlin.descriptors.ClassDescriptor;
|
|
||||||
import org.jetbrains.kotlin.descriptors.ClassKind;
|
|
||||||
import org.jetbrains.kotlin.descriptors.ConstructorDescriptor;
|
|
||||||
import org.jetbrains.kotlin.descriptors.DeclarationDescriptor;
|
|
||||||
import org.jetbrains.kotlin.idea.caches.resolve.ResolutionUtils;
|
|
||||||
import org.jetbrains.kotlin.idea.core.overrideImplement.ImplementMembersHandler;
|
|
||||||
import org.jetbrains.kotlin.psi.KtReferenceExpression;
|
|
||||||
import org.jetbrains.kotlin.resolve.BindingContext;
|
|
||||||
import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode;
|
|
||||||
|
|
||||||
class AnonymousTemplateEditingListener extends TemplateEditingAdapter {
|
|
||||||
private KtReferenceExpression classRef;
|
|
||||||
private ClassDescriptor classDescriptor;
|
|
||||||
private final Editor editor;
|
|
||||||
private final PsiFile psiFile;
|
|
||||||
|
|
||||||
private static final Key<AnonymousTemplateEditingListener> LISTENER_KEY = Key.create("kotlin.AnonymousTemplateEditingListener");
|
|
||||||
|
|
||||||
public AnonymousTemplateEditingListener(PsiFile psiFile, Editor editor) {
|
|
||||||
this.psiFile = psiFile;
|
|
||||||
this.editor = editor;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void currentVariableChanged(TemplateState templateState, Template template, int oldIndex, int newIndex) {
|
|
||||||
assert templateState.getTemplate() != null;
|
|
||||||
TextRange variableRange = templateState.getVariableRange("SUPERTYPE");
|
|
||||||
if (variableRange == null) return;
|
|
||||||
PsiElement name = psiFile.findElementAt(variableRange.getStartOffset());
|
|
||||||
if (name != null && name.getParent() instanceof KtReferenceExpression) {
|
|
||||||
KtReferenceExpression ref = (KtReferenceExpression) name.getParent();
|
|
||||||
DeclarationDescriptor descriptor = ResolutionUtils.analyze(ref, BodyResolveMode.FULL).get(BindingContext.REFERENCE_TARGET, ref);
|
|
||||||
if (descriptor instanceof ClassDescriptor) {
|
|
||||||
classRef = ref;
|
|
||||||
classDescriptor = (ClassDescriptor) descriptor;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void templateFinished(Template template, boolean brokenOff) {
|
|
||||||
editor.putUserData(LISTENER_KEY, null);
|
|
||||||
if (brokenOff) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (classDescriptor != null) {
|
|
||||||
if (classDescriptor.getKind() == ClassKind.CLASS) {
|
|
||||||
int placeToInsert = classRef.getTextRange().getEndOffset();
|
|
||||||
PsiDocumentManager.getInstance(psiFile.getProject()).getDocument(psiFile).insertString(placeToInsert, "()");
|
|
||||||
|
|
||||||
boolean hasConstructorsParameters = false;
|
|
||||||
for (ConstructorDescriptor cd : classDescriptor.getConstructors()) {
|
|
||||||
// TODO check for visibility
|
|
||||||
hasConstructorsParameters |= cd.getValueParameters().size() != 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (hasConstructorsParameters) {
|
|
||||||
editor.getCaretModel().moveToOffset(placeToInsert + 1);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
new ImplementMembersHandler().invoke(psiFile.getProject(), editor, psiFile, true);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void registerListener(Editor editor, Project project) {
|
|
||||||
if (editor.getUserData(LISTENER_KEY) != null) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
PsiFile psiFile = PsiDocumentManager.getInstance(project).getPsiFile(editor.getDocument());
|
|
||||||
assert psiFile != null;
|
|
||||||
TemplateState templateState = TemplateManagerImpl.getTemplateState(editor);
|
|
||||||
if (templateState != null) {
|
|
||||||
AnonymousTemplateEditingListener listener = new AnonymousTemplateEditingListener(psiFile, editor);
|
|
||||||
editor.putUserData(LISTENER_KEY, listener);
|
|
||||||
templateState.addTemplateStateListener(listener);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
+96
@@ -0,0 +1,96 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro
|
||||||
|
|
||||||
|
import com.intellij.codeInsight.template.Template
|
||||||
|
import com.intellij.codeInsight.template.TemplateEditingAdapter
|
||||||
|
import com.intellij.codeInsight.template.impl.TemplateManagerImpl
|
||||||
|
import com.intellij.codeInsight.template.impl.TemplateState
|
||||||
|
import com.intellij.openapi.editor.Editor
|
||||||
|
import com.intellij.openapi.project.Project
|
||||||
|
import com.intellij.openapi.util.Key
|
||||||
|
import com.intellij.psi.PsiDocumentManager
|
||||||
|
import com.intellij.psi.PsiFile
|
||||||
|
import org.jetbrains.kotlin.descriptors.ClassDescriptor
|
||||||
|
import org.jetbrains.kotlin.descriptors.ClassKind
|
||||||
|
import org.jetbrains.kotlin.idea.caches.resolve.analyze
|
||||||
|
import org.jetbrains.kotlin.idea.core.overrideImplement.ImplementMembersHandler
|
||||||
|
import org.jetbrains.kotlin.psi.KtReferenceExpression
|
||||||
|
import org.jetbrains.kotlin.resolve.BindingContext
|
||||||
|
import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode
|
||||||
|
|
||||||
|
internal class AnonymousTemplateEditingListener(private val psiFile: PsiFile, private val editor: Editor) : TemplateEditingAdapter() {
|
||||||
|
private var classRef: KtReferenceExpression? = null
|
||||||
|
private var classDescriptor: ClassDescriptor? = null
|
||||||
|
|
||||||
|
override fun currentVariableChanged(templateState: TemplateState?, template: Template?, oldIndex: Int, newIndex: Int) {
|
||||||
|
assert(templateState!!.template != null)
|
||||||
|
val variableRange = templateState.getVariableRange("SUPERTYPE") ?: return
|
||||||
|
val name = psiFile.findElementAt(variableRange.startOffset)
|
||||||
|
if (name != null && name.parent is KtReferenceExpression) {
|
||||||
|
val ref = name.parent as KtReferenceExpression
|
||||||
|
val descriptor = ref.analyze(BodyResolveMode.FULL).get(BindingContext.REFERENCE_TARGET, ref)
|
||||||
|
if (descriptor is ClassDescriptor) {
|
||||||
|
classRef = ref
|
||||||
|
classDescriptor = descriptor
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun templateFinished(template: Template?, brokenOff: Boolean) {
|
||||||
|
editor.putUserData(LISTENER_KEY, null)
|
||||||
|
if (brokenOff) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if (classDescriptor != null) {
|
||||||
|
if (classDescriptor!!.kind == ClassKind.CLASS) {
|
||||||
|
val placeToInsert = classRef!!.textRange.endOffset
|
||||||
|
PsiDocumentManager.getInstance(psiFile.project).getDocument(psiFile)!!.insertString(placeToInsert, "()")
|
||||||
|
|
||||||
|
var hasConstructorsParameters = false
|
||||||
|
for (cd in classDescriptor!!.constructors) {
|
||||||
|
// TODO check for visibility
|
||||||
|
hasConstructorsParameters = hasConstructorsParameters or (cd.valueParameters.size != 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
if (hasConstructorsParameters) {
|
||||||
|
editor.caretModel.moveToOffset(placeToInsert + 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ImplementMembersHandler().invoke(psiFile.project, editor, psiFile, true)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
companion object {
|
||||||
|
|
||||||
|
private val LISTENER_KEY = Key.create<AnonymousTemplateEditingListener>("kotlin.AnonymousTemplateEditingListener")
|
||||||
|
|
||||||
|
fun registerListener(editor: Editor, project: Project) {
|
||||||
|
if (editor.getUserData(LISTENER_KEY) != null) return
|
||||||
|
|
||||||
|
val psiFile = PsiDocumentManager.getInstance(project).getPsiFile(editor.document)!!
|
||||||
|
val templateState = TemplateManagerImpl.getTemplateState(editor)
|
||||||
|
if (templateState != null) {
|
||||||
|
val listener = AnonymousTemplateEditingListener(psiFile, editor)
|
||||||
|
editor.putUserData(LISTENER_KEY, listener)
|
||||||
|
templateState.addTemplateStateListener(listener)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,44 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro;
|
|
||||||
|
|
||||||
import com.intellij.openapi.project.Project;
|
|
||||||
import org.jetbrains.annotations.NotNull;
|
|
||||||
import org.jetbrains.kotlin.descriptors.VariableDescriptor;
|
|
||||||
import org.jetbrains.kotlin.idea.KotlinBundle;
|
|
||||||
import org.jetbrains.kotlin.idea.core.IterableTypesDetector;
|
|
||||||
|
|
||||||
public class AnyVariableMacro extends BaseKotlinVariableMacro {
|
|
||||||
@Override
|
|
||||||
public String getName() {
|
|
||||||
return "kotlinVariable";
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getPresentableName() {
|
|
||||||
return KotlinBundle.message("macro.variable.of.type");
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected boolean isSuitable(
|
|
||||||
@NotNull VariableDescriptor variableDescriptor,
|
|
||||||
@NotNull Project project,
|
|
||||||
@NotNull IterableTypesDetector iterableTypesDetector
|
|
||||||
) {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,39 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro
|
||||||
|
|
||||||
|
import com.intellij.openapi.project.Project
|
||||||
|
import org.jetbrains.kotlin.descriptors.VariableDescriptor
|
||||||
|
import org.jetbrains.kotlin.idea.KotlinBundle
|
||||||
|
import org.jetbrains.kotlin.idea.core.IterableTypesDetector
|
||||||
|
|
||||||
|
class AnyVariableMacro : BaseKotlinVariableMacro() {
|
||||||
|
override fun getName(): String {
|
||||||
|
return "kotlinVariable"
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun getPresentableName(): String {
|
||||||
|
return KotlinBundle.message("macro.variable.of.type")
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun isSuitable(
|
||||||
|
variableDescriptor: VariableDescriptor,
|
||||||
|
project: Project,
|
||||||
|
iterableTypesDetector: IterableTypesDetector): Boolean {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,154 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro;
|
|
||||||
|
|
||||||
import com.intellij.codeInsight.lookup.LookupElement;
|
|
||||||
import com.intellij.codeInsight.lookup.LookupElementBuilder;
|
|
||||||
import com.intellij.codeInsight.template.Expression;
|
|
||||||
import com.intellij.codeInsight.template.ExpressionContext;
|
|
||||||
import com.intellij.codeInsight.template.Macro;
|
|
||||||
import com.intellij.codeInsight.template.Result;
|
|
||||||
import com.intellij.openapi.project.Project;
|
|
||||||
import com.intellij.psi.PsiDocumentManager;
|
|
||||||
import com.intellij.psi.PsiElement;
|
|
||||||
import com.intellij.psi.PsiFile;
|
|
||||||
import com.intellij.psi.PsiNamedElement;
|
|
||||||
import com.intellij.util.containers.ContainerUtil;
|
|
||||||
import org.jetbrains.annotations.NotNull;
|
|
||||||
import org.jetbrains.annotations.Nullable;
|
|
||||||
import org.jetbrains.kotlin.descriptors.DeclarationDescriptor;
|
|
||||||
import org.jetbrains.kotlin.descriptors.ReceiverParameterDescriptor;
|
|
||||||
import org.jetbrains.kotlin.descriptors.VariableDescriptor;
|
|
||||||
import org.jetbrains.kotlin.idea.caches.resolve.ResolutionUtils;
|
|
||||||
import org.jetbrains.kotlin.idea.core.IterableTypesDetection;
|
|
||||||
import org.jetbrains.kotlin.idea.core.IterableTypesDetector;
|
|
||||||
import org.jetbrains.kotlin.idea.resolve.ResolutionFacade;
|
|
||||||
import org.jetbrains.kotlin.idea.util.ExtensionUtils;
|
|
||||||
import org.jetbrains.kotlin.idea.util.ScopeUtils;
|
|
||||||
import org.jetbrains.kotlin.psi.*;
|
|
||||||
import org.jetbrains.kotlin.resolve.BindingContext;
|
|
||||||
import org.jetbrains.kotlin.resolve.DescriptorToSourceUtils;
|
|
||||||
import org.jetbrains.kotlin.resolve.DescriptorUtils;
|
|
||||||
import org.jetbrains.kotlin.resolve.bindingContextUtil.BindingContextUtilsKt;
|
|
||||||
import org.jetbrains.kotlin.resolve.calls.smartcasts.DataFlowInfo;
|
|
||||||
import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode;
|
|
||||||
import org.jetbrains.kotlin.resolve.scopes.DescriptorKindFilter;
|
|
||||||
import org.jetbrains.kotlin.resolve.scopes.MemberScope;
|
|
||||||
import org.jetbrains.kotlin.resolve.scopes.LexicalScope;
|
|
||||||
import org.jetbrains.kotlin.resolve.scopes.utils.ScopeUtilsKt;
|
|
||||||
|
|
||||||
import java.util.*;
|
|
||||||
|
|
||||||
public abstract class BaseKotlinVariableMacro extends Macro {
|
|
||||||
@Nullable
|
|
||||||
private KtNamedDeclaration[] getVariables(Expression[] params, ExpressionContext context) {
|
|
||||||
if (params.length != 0) return null;
|
|
||||||
|
|
||||||
Project project = context.getProject();
|
|
||||||
PsiDocumentManager.getInstance(project).commitAllDocuments();
|
|
||||||
|
|
||||||
PsiFile psiFile = PsiDocumentManager.getInstance(project).getPsiFile(context.getEditor().getDocument());
|
|
||||||
if (!(psiFile instanceof KtFile)) return null;
|
|
||||||
|
|
||||||
KtExpression contextExpression = findContextExpression(psiFile, context.getStartOffset());
|
|
||||||
if (contextExpression == null) return null;
|
|
||||||
|
|
||||||
ResolutionFacade resolutionFacade = ResolutionUtils.getResolutionFacade(contextExpression);
|
|
||||||
|
|
||||||
BindingContext bindingContext = resolutionFacade.analyze(contextExpression, BodyResolveMode.FULL);
|
|
||||||
LexicalScope scope = ScopeUtils.getResolutionScope(contextExpression, bindingContext, resolutionFacade);
|
|
||||||
|
|
||||||
IterableTypesDetector detector = resolutionFacade.getIdeService(IterableTypesDetection.class).createDetector(scope);
|
|
||||||
|
|
||||||
DataFlowInfo dataFlowInfo = BindingContextUtilsKt.getDataFlowInfo(bindingContext, contextExpression);
|
|
||||||
|
|
||||||
List<VariableDescriptor> filteredDescriptors = new ArrayList<VariableDescriptor>();
|
|
||||||
for (DeclarationDescriptor declarationDescriptor : getAllVariables(scope)) {
|
|
||||||
if (declarationDescriptor instanceof VariableDescriptor) {
|
|
||||||
VariableDescriptor variableDescriptor = (VariableDescriptor) declarationDescriptor;
|
|
||||||
|
|
||||||
if (variableDescriptor.getExtensionReceiverParameter() != null
|
|
||||||
&& ExtensionUtils.substituteExtensionIfCallableWithImplicitReceiver(
|
|
||||||
variableDescriptor, scope, bindingContext, dataFlowInfo).isEmpty()) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (isSuitable(variableDescriptor, project, detector)) {
|
|
||||||
filteredDescriptors.add(variableDescriptor);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
List<KtNamedDeclaration> declarations = new ArrayList<KtNamedDeclaration>();
|
|
||||||
for (DeclarationDescriptor declarationDescriptor : filteredDescriptors) {
|
|
||||||
PsiElement declaration = DescriptorToSourceUtils.descriptorToDeclaration(declarationDescriptor);
|
|
||||||
assert declaration == null || declaration instanceof PsiNamedElement;
|
|
||||||
|
|
||||||
if (declaration instanceof KtProperty || declaration instanceof KtParameter) {
|
|
||||||
declarations.add((KtNamedDeclaration) declaration);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return declarations.toArray(new KtNamedDeclaration[declarations.size()]);
|
|
||||||
}
|
|
||||||
|
|
||||||
private static Collection<DeclarationDescriptor> getAllVariables(LexicalScope scope) {
|
|
||||||
Collection<DeclarationDescriptor> result = ContainerUtil.newArrayList();
|
|
||||||
result.addAll(ScopeUtilsKt.collectDescriptorsFiltered(scope, DescriptorKindFilter.VARIABLES, MemberScope.Companion.getALL_NAME_FILTER()));
|
|
||||||
for (ReceiverParameterDescriptor implicitReceiver : ScopeUtilsKt.getImplicitReceiversHierarchy(scope)) {
|
|
||||||
result.addAll(DescriptorUtils.getAllDescriptors(implicitReceiver.getType().getMemberScope()));
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
protected abstract boolean isSuitable(
|
|
||||||
@NotNull VariableDescriptor variableDescriptor,
|
|
||||||
@NotNull Project project,
|
|
||||||
@NotNull IterableTypesDetector iterableTypesDetector
|
|
||||||
);
|
|
||||||
|
|
||||||
@Nullable
|
|
||||||
private static KtExpression findContextExpression(PsiFile psiFile, int startOffset) {
|
|
||||||
PsiElement e = psiFile.findElementAt(startOffset);
|
|
||||||
while (e != null) {
|
|
||||||
if (e instanceof KtExpression) {
|
|
||||||
return (KtExpression) e;
|
|
||||||
}
|
|
||||||
e = e.getParent();
|
|
||||||
}
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public Result calculateResult(@NotNull Expression[] params, ExpressionContext context) {
|
|
||||||
KtNamedDeclaration[] vars = getVariables(params, context);
|
|
||||||
if (vars == null || vars.length == 0) return null;
|
|
||||||
return new KotlinPsiElementResult(vars[0]);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public LookupElement[] calculateLookupItems(@NotNull Expression[] params, ExpressionContext context) {
|
|
||||||
PsiNamedElement[] vars = getVariables(params, context);
|
|
||||||
if (vars == null || vars.length < 2) return null;
|
|
||||||
Set<LookupElement> set = new LinkedHashSet<LookupElement>();
|
|
||||||
for (PsiNamedElement var : vars) {
|
|
||||||
set.add(LookupElementBuilder.create(var));
|
|
||||||
}
|
|
||||||
return set.toArray(new LookupElement[set.size()]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,141 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro
|
||||||
|
|
||||||
|
import com.intellij.codeInsight.lookup.LookupElement
|
||||||
|
import com.intellij.codeInsight.lookup.LookupElementBuilder
|
||||||
|
import com.intellij.codeInsight.template.Expression
|
||||||
|
import com.intellij.codeInsight.template.ExpressionContext
|
||||||
|
import com.intellij.codeInsight.template.Macro
|
||||||
|
import com.intellij.codeInsight.template.Result
|
||||||
|
import com.intellij.openapi.project.Project
|
||||||
|
import com.intellij.psi.PsiDocumentManager
|
||||||
|
import com.intellij.psi.PsiFile
|
||||||
|
import com.intellij.psi.PsiNamedElement
|
||||||
|
import com.intellij.util.containers.ContainerUtil
|
||||||
|
import org.jetbrains.kotlin.descriptors.DeclarationDescriptor
|
||||||
|
import org.jetbrains.kotlin.descriptors.VariableDescriptor
|
||||||
|
import org.jetbrains.kotlin.idea.caches.resolve.getResolutionFacade
|
||||||
|
import org.jetbrains.kotlin.idea.core.IterableTypesDetection
|
||||||
|
import org.jetbrains.kotlin.idea.core.IterableTypesDetector
|
||||||
|
import org.jetbrains.kotlin.idea.util.getResolutionScope
|
||||||
|
import org.jetbrains.kotlin.idea.util.substituteExtensionIfCallableWithImplicitReceiver
|
||||||
|
import org.jetbrains.kotlin.psi.*
|
||||||
|
import org.jetbrains.kotlin.resolve.DescriptorToSourceUtils
|
||||||
|
import org.jetbrains.kotlin.resolve.DescriptorUtils
|
||||||
|
import org.jetbrains.kotlin.resolve.bindingContextUtil.getDataFlowInfo
|
||||||
|
import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode
|
||||||
|
import org.jetbrains.kotlin.resolve.scopes.DescriptorKindFilter
|
||||||
|
import org.jetbrains.kotlin.resolve.scopes.LexicalScope
|
||||||
|
import org.jetbrains.kotlin.resolve.scopes.MemberScope
|
||||||
|
import org.jetbrains.kotlin.resolve.scopes.utils.collectDescriptorsFiltered
|
||||||
|
import org.jetbrains.kotlin.resolve.scopes.utils.getImplicitReceiversHierarchy
|
||||||
|
import java.util.*
|
||||||
|
|
||||||
|
abstract class BaseKotlinVariableMacro : Macro() {
|
||||||
|
private fun getVariables(params: Array<Expression>, context: ExpressionContext): Array<KtNamedDeclaration>? {
|
||||||
|
if (params.size != 0) return null
|
||||||
|
|
||||||
|
val project = context.project
|
||||||
|
PsiDocumentManager.getInstance(project).commitAllDocuments()
|
||||||
|
|
||||||
|
val psiFile = PsiDocumentManager.getInstance(project).getPsiFile(context.editor!!.document)
|
||||||
|
if (psiFile !is KtFile) return null
|
||||||
|
|
||||||
|
val contextExpression = findContextExpression(psiFile, context.startOffset) ?: return null
|
||||||
|
|
||||||
|
val resolutionFacade = contextExpression.getResolutionFacade()
|
||||||
|
|
||||||
|
val bindingContext = resolutionFacade.analyze(contextExpression, BodyResolveMode.FULL)
|
||||||
|
val scope = contextExpression.getResolutionScope(bindingContext, resolutionFacade)
|
||||||
|
|
||||||
|
val detector = resolutionFacade.getIdeService(IterableTypesDetection::class.java).createDetector(scope)
|
||||||
|
|
||||||
|
val dataFlowInfo = bindingContext.getDataFlowInfo(contextExpression)
|
||||||
|
|
||||||
|
val filteredDescriptors = ArrayList<VariableDescriptor>()
|
||||||
|
for (declarationDescriptor in getAllVariables(scope)) {
|
||||||
|
if (declarationDescriptor is VariableDescriptor) {
|
||||||
|
|
||||||
|
if (declarationDescriptor.extensionReceiverParameter != null && declarationDescriptor.substituteExtensionIfCallableWithImplicitReceiver(scope, bindingContext, dataFlowInfo).isEmpty()) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isSuitable(declarationDescriptor, project, detector)) {
|
||||||
|
filteredDescriptors.add(declarationDescriptor)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
val declarations = ArrayList<KtNamedDeclaration>()
|
||||||
|
for (declarationDescriptor in filteredDescriptors) {
|
||||||
|
val declaration = DescriptorToSourceUtils.descriptorToDeclaration(declarationDescriptor)
|
||||||
|
assert(declaration == null || declaration is PsiNamedElement)
|
||||||
|
|
||||||
|
if (declaration is KtProperty) {
|
||||||
|
declarations.add(declaration)
|
||||||
|
}
|
||||||
|
else if (declaration is KtParameter) {
|
||||||
|
declarations.add(declaration)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return declarations.toTypedArray()
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun getAllVariables(scope: LexicalScope): Collection<DeclarationDescriptor> {
|
||||||
|
val result = ContainerUtil.newArrayList<DeclarationDescriptor>()
|
||||||
|
result.addAll(scope.collectDescriptorsFiltered(DescriptorKindFilter.VARIABLES, MemberScope.ALL_NAME_FILTER))
|
||||||
|
for (implicitReceiver in scope.getImplicitReceiversHierarchy()) {
|
||||||
|
result.addAll(DescriptorUtils.getAllDescriptors(implicitReceiver.type.memberScope))
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
protected abstract fun isSuitable(
|
||||||
|
variableDescriptor: VariableDescriptor,
|
||||||
|
project: Project,
|
||||||
|
iterableTypesDetector: IterableTypesDetector): Boolean
|
||||||
|
|
||||||
|
private fun findContextExpression(psiFile: PsiFile, startOffset: Int): KtExpression? {
|
||||||
|
var e = psiFile.findElementAt(startOffset)
|
||||||
|
while (e != null) {
|
||||||
|
if (e is KtExpression) {
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
e = e.parent
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun calculateResult(params: Array<Expression>, context: ExpressionContext): Result? {
|
||||||
|
val vars = getVariables(params, context)
|
||||||
|
if (vars == null || vars.size == 0) return null
|
||||||
|
return KotlinPsiElementResult(vars[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun calculateLookupItems(params: Array<Expression>, context: ExpressionContext): Array<LookupElement>? {
|
||||||
|
val vars = getVariables(params, context)
|
||||||
|
if (vars == null || vars.size < 2) return null
|
||||||
|
val set = LinkedHashSet<LookupElement>()
|
||||||
|
for (`var` in vars) {
|
||||||
|
set.add(LookupElementBuilder.create(`var`))
|
||||||
|
}
|
||||||
|
return set.toArray<LookupElement>(arrayOfNulls<LookupElement>(set.size))
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,74 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro;
|
|
||||||
|
|
||||||
import com.intellij.codeInsight.template.*;
|
|
||||||
import com.intellij.openapi.project.Project;
|
|
||||||
import com.intellij.psi.PsiDocumentManager;
|
|
||||||
import com.intellij.psi.PsiElement;
|
|
||||||
import com.intellij.psi.PsiFile;
|
|
||||||
import org.jetbrains.annotations.NotNull;
|
|
||||||
import org.jetbrains.kotlin.idea.KotlinBundle;
|
|
||||||
import org.jetbrains.kotlin.psi.KtFunction;
|
|
||||||
import org.jetbrains.kotlin.psi.KtParameter;
|
|
||||||
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
public class FunctionParametersMacro extends Macro {
|
|
||||||
@Override
|
|
||||||
public String getName() {
|
|
||||||
return "functionParameters";
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getPresentableName() {
|
|
||||||
return KotlinBundle.message("macro.fun.parameters");
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public Result calculateResult(@NotNull Expression[] params, ExpressionContext context) {
|
|
||||||
Project project = context.getProject();
|
|
||||||
int templateStartOffset = context.getTemplateStartOffset();
|
|
||||||
int offset = templateStartOffset > 0 ? context.getTemplateStartOffset() - 1 : context.getTemplateStartOffset();
|
|
||||||
|
|
||||||
PsiDocumentManager.getInstance(project).commitAllDocuments();
|
|
||||||
|
|
||||||
PsiFile file = PsiDocumentManager.getInstance(project).getPsiFile(context.getEditor().getDocument());
|
|
||||||
if (file == null) return null;
|
|
||||||
PsiElement place = file.findElementAt(offset);
|
|
||||||
while (place != null){
|
|
||||||
if (place instanceof KtFunction) {
|
|
||||||
List<Result> result = new ArrayList<Result>();
|
|
||||||
for (KtParameter param : ((KtFunction) place).getValueParameters()) {
|
|
||||||
String name = param.getName();
|
|
||||||
assert name != null;
|
|
||||||
result.add(new TextResult(name));
|
|
||||||
}
|
|
||||||
return new ListResult(result);
|
|
||||||
}
|
|
||||||
place = place.getParent();
|
|
||||||
}
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean isAcceptableInContext(TemplateContextType context) {
|
|
||||||
return context instanceof JavaCodeContextType;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,60 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro
|
||||||
|
|
||||||
|
import com.intellij.codeInsight.template.*
|
||||||
|
import com.intellij.psi.PsiDocumentManager
|
||||||
|
import org.jetbrains.kotlin.idea.KotlinBundle
|
||||||
|
import org.jetbrains.kotlin.psi.KtFunction
|
||||||
|
import java.util.*
|
||||||
|
|
||||||
|
class FunctionParametersMacro : Macro() {
|
||||||
|
override fun getName(): String {
|
||||||
|
return "functionParameters"
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun getPresentableName(): String {
|
||||||
|
return KotlinBundle.message("macro.fun.parameters")
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun calculateResult(params: Array<Expression>, context: ExpressionContext): Result? {
|
||||||
|
val project = context.project
|
||||||
|
val templateStartOffset = context.templateStartOffset
|
||||||
|
val offset = if (templateStartOffset > 0) context.templateStartOffset - 1 else context.templateStartOffset
|
||||||
|
|
||||||
|
PsiDocumentManager.getInstance(project).commitAllDocuments()
|
||||||
|
|
||||||
|
val file = PsiDocumentManager.getInstance(project).getPsiFile(context.editor!!.document) ?: return null
|
||||||
|
var place = file.findElementAt(offset)
|
||||||
|
while (place != null) {
|
||||||
|
if (place is KtFunction) {
|
||||||
|
val result = ArrayList<Result>()
|
||||||
|
for (param in place.valueParameters) {
|
||||||
|
result.add(TextResult(param.name!!))
|
||||||
|
}
|
||||||
|
return ListResult(result)
|
||||||
|
}
|
||||||
|
place = place.parent
|
||||||
|
}
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun isAcceptableInContext(context: TemplateContextType?): Boolean {
|
||||||
|
return context is JavaCodeContextType
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@@ -1,46 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro;
|
|
||||||
|
|
||||||
import com.intellij.openapi.project.Project;
|
|
||||||
import org.jetbrains.annotations.NotNull;
|
|
||||||
import org.jetbrains.kotlin.descriptors.VariableDescriptor;
|
|
||||||
import org.jetbrains.kotlin.idea.KotlinBundle;
|
|
||||||
import org.jetbrains.kotlin.idea.core.IterableTypesDetector;
|
|
||||||
|
|
||||||
public class IterableVariableMacro extends BaseKotlinVariableMacro {
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getName() {
|
|
||||||
return "kotlinIterableVariable";
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getPresentableName() {
|
|
||||||
return KotlinBundle.message("macro.iterable.variable");
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected boolean isSuitable(
|
|
||||||
@NotNull VariableDescriptor variableDescriptor,
|
|
||||||
@NotNull Project project,
|
|
||||||
@NotNull IterableTypesDetector iterableTypesDetector
|
|
||||||
) {
|
|
||||||
//TODO: smart-casts
|
|
||||||
return iterableTypesDetector.isIterable(variableDescriptor.getType(), null);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,41 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro
|
||||||
|
|
||||||
|
import com.intellij.openapi.project.Project
|
||||||
|
import org.jetbrains.kotlin.descriptors.VariableDescriptor
|
||||||
|
import org.jetbrains.kotlin.idea.KotlinBundle
|
||||||
|
import org.jetbrains.kotlin.idea.core.IterableTypesDetector
|
||||||
|
|
||||||
|
class IterableVariableMacro : BaseKotlinVariableMacro() {
|
||||||
|
|
||||||
|
override fun getName(): String {
|
||||||
|
return "kotlinIterableVariable"
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun getPresentableName(): String {
|
||||||
|
return KotlinBundle.message("macro.iterable.variable")
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun isSuitable(
|
||||||
|
variableDescriptor: VariableDescriptor,
|
||||||
|
project: Project,
|
||||||
|
iterableTypesDetector: IterableTypesDetector): Boolean {
|
||||||
|
//TODO: smart-casts
|
||||||
|
return iterableTypesDetector.isIterable(variableDescriptor.type, null)
|
||||||
|
}
|
||||||
|
}
|
||||||
+6
-11
@@ -14,18 +14,13 @@
|
|||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
package org.jetbrains.kotlin.idea.liveTemplates.macro;
|
package org.jetbrains.kotlin.idea.liveTemplates.macro
|
||||||
|
|
||||||
import com.intellij.codeInsight.template.JavaPsiElementResult;
|
import com.intellij.codeInsight.template.JavaPsiElementResult
|
||||||
import com.intellij.psi.PsiNamedElement;
|
import com.intellij.psi.PsiNamedElement
|
||||||
|
|
||||||
public class KotlinPsiElementResult extends JavaPsiElementResult {
|
class KotlinPsiElementResult(element: PsiNamedElement) : JavaPsiElementResult(element) {
|
||||||
public KotlinPsiElementResult(PsiNamedElement element) {
|
override fun toString(): String {
|
||||||
super(element);
|
return (element as PsiNamedElement).name ?: ""
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String toString() {
|
|
||||||
return ((PsiNamedElement) getElement()).getName();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,44 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro;
|
|
||||||
|
|
||||||
import com.intellij.openapi.project.Project;
|
|
||||||
import org.jetbrains.annotations.NotNull;
|
|
||||||
import org.jetbrains.kotlin.descriptors.VariableDescriptor;
|
|
||||||
import org.jetbrains.kotlin.idea.KotlinBundle;
|
|
||||||
import org.jetbrains.kotlin.idea.core.IterableTypesDetector;
|
|
||||||
|
|
||||||
public class SuggestVariableNameMacro extends BaseKotlinVariableMacro {
|
|
||||||
@Override
|
|
||||||
public String getName() {
|
|
||||||
return "kotlinSuggestVariableName";
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getPresentableName() {
|
|
||||||
return KotlinBundle.message("macro.suggest.variable.name");
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected boolean isSuitable(
|
|
||||||
@NotNull VariableDescriptor variableDescriptor,
|
|
||||||
@NotNull Project project,
|
|
||||||
@NotNull IterableTypesDetector iterableTypesDetector
|
|
||||||
) {
|
|
||||||
return variableDescriptor.getType().isMarkedNullable();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,39 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2015 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.kotlin.idea.liveTemplates.macro
|
||||||
|
|
||||||
|
import com.intellij.openapi.project.Project
|
||||||
|
import org.jetbrains.kotlin.descriptors.VariableDescriptor
|
||||||
|
import org.jetbrains.kotlin.idea.KotlinBundle
|
||||||
|
import org.jetbrains.kotlin.idea.core.IterableTypesDetector
|
||||||
|
|
||||||
|
class SuggestVariableNameMacro : BaseKotlinVariableMacro() {
|
||||||
|
override fun getName(): String {
|
||||||
|
return "kotlinSuggestVariableName"
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun getPresentableName(): String {
|
||||||
|
return KotlinBundle.message("macro.suggest.variable.name")
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun isSuitable(
|
||||||
|
variableDescriptor: VariableDescriptor,
|
||||||
|
project: Project,
|
||||||
|
iterableTypesDetector: IterableTypesDetector): Boolean {
|
||||||
|
return variableDescriptor.type.isMarkedNullable
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user