From 5a6c658204f56a7d58dbdc4e023c2d0e863d1af4 Mon Sep 17 00:00:00 2001
From: Evgeny Gerashchenko <Evgeny.Gerashchenko@jetbrains.com>
Date: Tue, 7 Feb 2012 17:06:30 +0400
Subject: [PATCH] Added completion in 'anonymous' live template.

---
 idea/resources/liveTemplates/Kotlin.xml       |  2 +-
 idea/src/META-INF/plugin.xml                  |  1 +
 .../jetbrains/jet/plugin/JetBundle.properties |  3 +-
 .../macro/JetAnonymousSuperMacro.java         | 94 +++++++++++++++++++
 .../macro/JetPsiElementResult.java            | 10 +-
 5 files changed, 103 insertions(+), 7 deletions(-)
 create mode 100644 idea/src/org/jetbrains/jet/plugin/liveTemplates/macro/JetAnonymousSuperMacro.java

diff --git a/idea/resources/liveTemplates/Kotlin.xml b/idea/resources/liveTemplates/Kotlin.xml
index 02f7b4aa6ff..7e63db35cc4 100644
--- a/idea/resources/liveTemplates/Kotlin.xml
+++ b/idea/resources/liveTemplates/Kotlin.xml
@@ -119,7 +119,7 @@
     </template>
     <template resource-bundle="org.jetbrains.jet.plugin.JetBundle" key="livetemplate.description.anonymous"
               name="anonymous" toReformat="true" toShortenFQNames="true" value="object : $SUPERTYPE$ {&#10;$END$&#10;}">
-        <variable alwaysStopAt="true" defaultValue="" expression="" name="SUPERTYPE" />
+        <variable alwaysStopAt="true" defaultValue="&quot;Object&quot;" expression="anonymousSuper()" name="SUPERTYPE" />
         <context>
             <option name="KOTLIN_STATEMENT" value="true"/>
             <option name="KOTLIN_EXPRESSION" value="true"/>
diff --git a/idea/src/META-INF/plugin.xml b/idea/src/META-INF/plugin.xml
index de4c8aaa88d..d6ad3fa5d43 100644
--- a/idea/src/META-INF/plugin.xml
+++ b/idea/src/META-INF/plugin.xml
@@ -96,6 +96,7 @@
       <defaultLiveTemplatesProvider implementation="org.jetbrains.jet.plugin.liveTemplates.JetLiveTemplatesProvider"/>
       <liveTemplateMacro implementation="org.jetbrains.jet.plugin.liveTemplates.macro.JetVariableOfTypeMacro"/>
       <liveTemplateMacro implementation="org.jetbrains.jet.plugin.liveTemplates.macro.JetFunctionParametersMacro"/>
+      <liveTemplateMacro implementation="org.jetbrains.jet.plugin.liveTemplates.macro.JetAnonymousSuperMacro"/>
 
       <annotator language="jet" implementationClass="org.jetbrains.jet.plugin.annotations.SoftKeywordsAnnotator"/>
       <annotator language="jet" implementationClass="org.jetbrains.jet.plugin.annotations.LabelsAnnotator"/>
diff --git a/idea/src/org/jetbrains/jet/plugin/JetBundle.properties b/idea/src/org/jetbrains/jet/plugin/JetBundle.properties
index ab80854cb53..711bf9a54d8 100644
--- a/idea/src/org/jetbrains/jet/plugin/JetBundle.properties
+++ b/idea/src/org/jetbrains/jet/plugin/JetBundle.properties
@@ -41,4 +41,5 @@ livetemplate.description.exfun=Extension function
 livetemplate.description.exval=Extension read-only property
 livetemplate.description.exvar=Extension read-write property
 macro.variable.of.type=kotlinVariableOfType(Type)
-macro.fun.parameters=functionParameters()
\ No newline at end of file
+macro.fun.parameters=functionParameters()
+macro.fun.anonymousSuper=anonymousSuper()
\ No newline at end of file
diff --git a/idea/src/org/jetbrains/jet/plugin/liveTemplates/macro/JetAnonymousSuperMacro.java b/idea/src/org/jetbrains/jet/plugin/liveTemplates/macro/JetAnonymousSuperMacro.java
new file mode 100644
index 00000000000..001a217bddc
--- /dev/null
+++ b/idea/src/org/jetbrains/jet/plugin/liveTemplates/macro/JetAnonymousSuperMacro.java
@@ -0,0 +1,94 @@
+package org.jetbrains.jet.plugin.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.psi.util.PsiTreeUtil;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
+import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor;
+import org.jetbrains.jet.lang.psi.JetExpression;
+import org.jetbrains.jet.lang.psi.JetFile;
+import org.jetbrains.jet.lang.resolve.BindingContext;
+import org.jetbrains.jet.lang.resolve.scopes.JetScope;
+import org.jetbrains.jet.plugin.JetBundle;
+import org.jetbrains.jet.plugin.compiler.WholeProjectAnalyzerFacade;
+
+import java.util.ArrayList;
+import java.util.LinkedHashSet;
+import java.util.List;
+import java.util.Set;
+
+/**
+ * @author Evgeny Gerashchenko
+ * @since 2/2/12
+ */
+public class JetAnonymousSuperMacro extends Macro {
+    @Override
+    public String getName() {
+        return "anonymousSuper";
+    }
+
+    @Override
+    public String getPresentableName() {
+        return JetBundle.message("macro.fun.anonymousSuper");
+    }
+
+    @Override
+    public Result calculateResult(@NotNull Expression[] params, ExpressionContext context) {
+        final PsiNamedElement[] vars = getSupertypes(params, context);
+        if (vars == null || vars.length == 0) return null;
+        return new JetPsiElementResult(vars[0]);
+    }
+
+    @Override
+    public LookupElement[] calculateLookupItems(@NotNull Expression[] params, ExpressionContext context) {
+        final PsiNamedElement[] vars = getSupertypes(params, context);
+        if (vars == null || vars.length < 2) return null;
+        final Set<LookupElement> set = new LinkedHashSet<LookupElement>();
+        for (PsiNamedElement var : vars) {
+            set.add(LookupElementBuilder.create(var));
+        }
+        return set.toArray(new LookupElement[set.size()]);
+    }
+
+    @Nullable
+    private static PsiNamedElement[] getSupertypes(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 JetFile)) return null;
+
+        BindingContext bc = WholeProjectAnalyzerFacade.analyzeProjectWithCacheOnAFile((JetFile) psiFile);
+        JetExpression expression = PsiTreeUtil.getParentOfType(psiFile.findElementAt(context.getStartOffset()), JetExpression.class);
+        JetScope scope = bc.get(BindingContext.RESOLUTION_SCOPE, expression);
+        if (scope == null) {
+            return null;
+        }
+        
+        List<PsiNamedElement> result = new ArrayList<PsiNamedElement>();
+
+        for (DeclarationDescriptor descriptor : scope.getAllDescriptors()) {
+            if (descriptor instanceof ClassDescriptor && ((ClassDescriptor) descriptor).getModality().isOverridable()) {
+                PsiElement declaration = bc.get(BindingContext.DESCRIPTOR_TO_DECLARATION, descriptor);
+                if (declaration != null) {
+                    result.add((PsiNamedElement) declaration);
+                }
+            }
+        }
+        
+        return result.toArray(new PsiNamedElement[result.size()]);
+    }
+}
diff --git a/idea/src/org/jetbrains/jet/plugin/liveTemplates/macro/JetPsiElementResult.java b/idea/src/org/jetbrains/jet/plugin/liveTemplates/macro/JetPsiElementResult.java
index b5b3a8f22b1..cc833661bb9 100644
--- a/idea/src/org/jetbrains/jet/plugin/liveTemplates/macro/JetPsiElementResult.java
+++ b/idea/src/org/jetbrains/jet/plugin/liveTemplates/macro/JetPsiElementResult.java
@@ -1,19 +1,19 @@
 package org.jetbrains.jet.plugin.liveTemplates.macro;
 
-import com.intellij.codeInsight.template.PsiElementResult;
-import org.jetbrains.jet.lang.psi.JetNamedDeclaration;
+import com.intellij.codeInsight.template.JavaPsiElementResult;
+import com.intellij.psi.PsiNamedElement;
 
 /**
  * @author Evgeny Gerashchenko
  * @since 1/30/12
  */
-public class JetPsiElementResult extends PsiElementResult {
-    public JetPsiElementResult(JetNamedDeclaration element) {
+public class JetPsiElementResult extends JavaPsiElementResult {
+    public JetPsiElementResult(PsiNamedElement element) {
         super(element);
     }
 
     @Override
     public String toString() {
-        return ((JetNamedDeclaration) getElement()).getName();
+        return ((PsiNamedElement) getElement()).getName();
     }
 }