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

Extracted JetDecompiledData from JetContentBasedFileSubstitutor. It saves parsed Kotlin tree and text to user data of ClsFileImpl

Browse Source
This commit is contained in:
Evgeny Gerashchenko
2012-03-02 18:34:25 +04:00
parent 54a781edb6
commit 72fbdfa65e
2 changed files with 193 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/jet/plugin/JetContentBasedFileSubstitutor.java
+8 -104
View File
@@ -16,24 +16,14 @@
package org.jetbrains.jet.plugin; package org.jetbrains.jet.plugin;
import com.intellij.ide.highlighter.JavaClassFileType;
import com.intellij.lang.Language; import com.intellij.lang.Language;
import com.intellij.openapi.fileTypes.ContentBasedClassFileProcessor; import com.intellij.openapi.fileTypes.ContentBasedClassFileProcessor;
import com.intellij.openapi.fileTypes.FileTypeManager;
import com.intellij.openapi.fileTypes.SyntaxHighlighter; import com.intellij.openapi.fileTypes.SyntaxHighlighter;
import com.intellij.openapi.project.Project; import com.intellij.openapi.project.Project;
import com.intellij.openapi.vfs.VirtualFile; import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.*; import com.intellij.psi.PsiFile;
import com.intellij.psi.impl.compiled.ClsAnnotationImpl; import com.intellij.psi.impl.compiled.ClsFileImpl;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jetbrains.jet.lang.JetSemanticServices;
import org.jetbrains.jet.lang.descriptors.*;
import org.jetbrains.jet.lang.resolve.BindingTraceContext;
import org.jetbrains.jet.lang.resolve.java.JavaDescriptorResolver;
import org.jetbrains.jet.lang.resolve.java.JavaSemanticServices;
import org.jetbrains.jet.resolve.DescriptorRenderer;
import java.util.Arrays;
/** /**
* @author Evgeny Gerashchenko * @author Evgeny Gerashchenko
@@ -42,102 +32,18 @@ import java.util.Arrays;
public class JetContentBasedFileSubstitutor implements ContentBasedClassFileProcessor { public class JetContentBasedFileSubstitutor implements ContentBasedClassFileProcessor {
@Override @Override
public boolean isApplicable(Project project, VirtualFile vFile) { public boolean isApplicable(Project project, VirtualFile file) {
if (!FileTypeManager.getInstance().isFileOfType(vFile, JavaClassFileType.INSTANCE)) { return JetDecompiledData.getClsFileIfKotlin(project, file) != null;
return false;
}
PsiClassOwner clsFile = (PsiClassOwner) PsiManager.getInstance(project).findFile(vFile);
if (clsFile == null) {
return false;
}
// TODO multiple?
if (clsFile.getClasses().length != 1) {
throw new AssertionError("Multiple classes in file: " + Arrays.toString(clsFile.getClasses()));
}
PsiClass psiClass = clsFile.getClasses()[0];
// TODO add better check
if (psiClass.getName().equals("namespace")) {
return true;
}
PsiModifierList modifierList = psiClass.getModifierList();
if (modifierList != null) {
for (PsiAnnotation annotation : modifierList.getAnnotations()) {
if (annotation instanceof ClsAnnotationImpl) {
if (((ClsAnnotationImpl) annotation).getStub().getText().startsWith("@jet.runtime.typeinfo.JetClass")) {
return true;
}
}
}
}
return false;
} }
@NotNull @NotNull
@Override @Override
public String obtainFileText(Project project, VirtualFile file) { public String obtainFileText(Project project, VirtualFile file) {
StringBuilder builder = new StringBuilder(); ClsFileImpl clsFile = JetDecompiledData.getClsFileIfKotlin(project, file);
JavaSemanticServices jss = new JavaSemanticServices(
project,
JetSemanticServices.createSemanticServices(project),
new BindingTraceContext());
PsiClassOwner clsFile = (PsiClassOwner) PsiManager.getInstance(project).findFile(file);
if (clsFile != null) { if (clsFile != null) {
builder.append(PsiBundle.message("psi.decompiled.text.header")); return JetDecompiledData.getDecompiledData(clsFile).getText();
builder.append("\n\n");
String packageName = clsFile.getPackageName();
if (packageName == null) {
packageName = "";
}
if (packageName.length() > 0) {
builder.append("package ").append(packageName).append("\n\n");
}
PsiClass psiClass = clsFile.getClasses()[0];
JavaDescriptorResolver jdr = jss.getDescriptorResolver();
if (psiClass.getName().equals("namespace")) { // TODO better check for namespace
NamespaceDescriptor nd = jdr.resolveNamespace(packageName);
if (nd != null) {
for (DeclarationDescriptor member : nd.getMemberScope().getAllDescriptors()) {
if (member instanceof ClassDescriptor && member.getName().equals("namespace") || member instanceof NamespaceDescriptor) {
continue;
}
appendMemberDescriptor(builder, member);
}
}
} else {
ClassDescriptor cd = jdr.resolveClass(psiClass);
if (cd != null) {
builder.append(DescriptorRenderer.COMPACT.render(cd));
builder.append(" {\n");
for (DeclarationDescriptor member : cd.getDefaultType().getMemberScope().getAllDescriptors()) {
if (member.getContainingDeclaration() == cd) {
builder.append(" ");
appendMemberDescriptor(builder, member);
}
}
builder.append("}");
}
}
} }
return builder.toString(); return "";
}
private static void appendMemberDescriptor(StringBuilder builder, DeclarationDescriptor member) {
String decompiledComment = "/* " + PsiBundle.message("psi.decompiled.method.body") + " */";
builder.append(DescriptorRenderer.COMPACT.render(member));
if (member instanceof FunctionDescriptor) {
builder.append(" { ").append(decompiledComment).append(" }");
} else if (member instanceof PropertyDescriptor) {
builder.append(" ").append(decompiledComment);
}
builder.append("\n\n");
} }
@Override @Override
@@ -154,9 +60,7 @@ public class JetContentBasedFileSubstitutor implements ContentBasedClassFileProc
@NotNull @NotNull
@Override @Override
public PsiFile getDecompiledPsiFile(PsiFile psiFile) { public PsiFile getDecompiledPsiFile(PsiFile psiFile) {
Project project = psiFile.getProject(); return JetDecompiledData.getDecompiledData((ClsFileImpl) psiFile).getJetFile();
String text = obtainFileText(project, psiFile.getVirtualFile());
return PsiFileFactory.getInstance(project).createFileFromText("", JetLanguage.INSTANCE, text);
} }
} }
idea/src/org/jetbrains/jet/plugin/JetDecompiledData.java
+185
View File
@@ -0,0 +1,185 @@
/*
* Copyright 2000-2012 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;
import com.intellij.ide.highlighter.JavaClassFileType;
import com.intellij.openapi.fileTypes.FileTypeManager;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.Key;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.*;
import com.intellij.psi.impl.compiled.ClsAnnotationImpl;
import com.intellij.psi.impl.compiled.ClsFileImpl;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.jet.lang.JetSemanticServices;
import org.jetbrains.jet.lang.descriptors.*;
import org.jetbrains.jet.lang.psi.JetFile;
import org.jetbrains.jet.lang.resolve.BindingTraceContext;
import org.jetbrains.jet.lang.resolve.java.JavaDescriptorResolver;
import org.jetbrains.jet.lang.resolve.java.JavaSemanticServices;
import org.jetbrains.jet.resolve.DescriptorRenderer;
import java.util.Arrays;
/**
* @author Evgeny Gerashchenko
* @since 3/2/12
*/
public class JetDecompiledData {
private JetFile myJetFile;
private String myText;
// private Map<ClsElementImpl, JetElement> myClsElementsToJetElements = new HashMap<ClsElementImpl, JetElement>();
private static final Object LOCK = new String("decompiled data lock");
private static final Key<JetDecompiledData> USER_DATA_KEY = new Key<JetDecompiledData>("USER_DATA_KEY");
private JetDecompiledData() {
}
@NotNull
public JetFile getJetFile() {
return myJetFile;
}
public String getText() {
return myText;
}
@Nullable
public static ClsFileImpl getClsFileIfKotlin(@NotNull Project project, @NotNull VirtualFile vFile) {
if (!FileTypeManager.getInstance().isFileOfType(vFile, JavaClassFileType.INSTANCE)) {
return null;
}
PsiFile psiFile = PsiManager.getInstance(project).findFile(vFile);
if (!(psiFile instanceof ClsFileImpl)) {
return null;
}
ClsFileImpl clsFile = (ClsFileImpl) psiFile;
// TODO multiple?
if (clsFile.getClasses().length != 1) {
throw new AssertionError("Multiple classes in file: " + Arrays.toString(clsFile.getClasses()));
}
PsiClass psiClass = clsFile.getClasses()[0];
// TODO add better check
if (psiClass.getName().equals("namespace")) {
return clsFile;
}
PsiModifierList modifierList = psiClass.getModifierList();
if (modifierList != null) {
for (PsiAnnotation annotation : modifierList.getAnnotations()) {
if (annotation instanceof ClsAnnotationImpl) {
if (((ClsAnnotationImpl) annotation).getStub().getText().startsWith("@jet.runtime.typeinfo.JetClass")) {
return clsFile;
}
}
}
}
return null;
}
@NotNull
public static JetDecompiledData getDecompiledData(@NotNull ClsFileImpl clsFile) {
synchronized (LOCK) {
if (clsFile.getUserData(USER_DATA_KEY) == null) {
clsFile.putUserData(USER_DATA_KEY, decompileData(clsFile));
}
JetDecompiledData decompiledData = clsFile.getUserData(USER_DATA_KEY);
assert decompiledData != null;
return decompiledData;
}
}
@NotNull
private static JetDecompiledData decompileData(@NotNull ClsFileImpl clsFile) {
JetDecompiledData result = new JetDecompiledData();
Project project = clsFile.getProject();
result.myText = getJetText(project, clsFile.getVirtualFile());
result.myJetFile = (JetFile) PsiFileFactory.getInstance(project).createFileFromText(clsFile.getName(), JetLanguage.INSTANCE, result.myText);
return result;
}
@NotNull
private static String getJetText(Project project, VirtualFile file) {
StringBuilder builder = new StringBuilder();
JavaSemanticServices jss = new JavaSemanticServices(
project,
JetSemanticServices.createSemanticServices(project),
new BindingTraceContext());
PsiClassOwner clsFile = (PsiClassOwner) PsiManager.getInstance(project).findFile(file);
if (clsFile != null) {
builder.append(PsiBundle.message("psi.decompiled.text.header"));
builder.append("\n\n");
String packageName = clsFile.getPackageName();
if (packageName == null) {
packageName = "";
}
if (packageName.length() > 0) {
builder.append("package ").append(packageName).append("\n\n");
}
PsiClass psiClass = clsFile.getClasses()[0];
JavaDescriptorResolver jdr = jss.getDescriptorResolver();
if (psiClass.getName().equals("namespace")) { // TODO better check for namespace
NamespaceDescriptor nd = jdr.resolveNamespace(packageName);
if (nd != null) {
for (DeclarationDescriptor member : nd.getMemberScope().getAllDescriptors()) {
if (member instanceof ClassDescriptor && member.getName().equals("namespace") || member instanceof NamespaceDescriptor) {
continue;
}
appendMemberDescriptor(builder, member);
}
}
} else {
ClassDescriptor cd = jdr.resolveClass(psiClass);
if (cd != null) {
builder.append(DescriptorRenderer.COMPACT.render(cd));
builder.append(" {\n");
for (DeclarationDescriptor member : cd.getDefaultType().getMemberScope().getAllDescriptors()) {
if (member.getContainingDeclaration() == cd) {
builder.append(" ");
appendMemberDescriptor(builder, member);
}
}
builder.append("}");
}
}
}
return builder.toString();
}
private static void appendMemberDescriptor(StringBuilder builder, DeclarationDescriptor member) {
String decompiledComment = "/* " + PsiBundle.message("psi.decompiled.method.body") + " */";
builder.append(DescriptorRenderer.COMPACT.render(member));
if (member instanceof FunctionDescriptor) {
builder.append(" { ").append(decompiledComment).append(" }");
} else if (member instanceof PropertyDescriptor) {
builder.append(" ").append(decompiledComment);
}
builder.append("\n\n");
}
}