Extracted JetDecompiledData from JetContentBasedFileSubstitutor. It saves parsed Kotlin tree and text to user data of ClsFileImpl
This commit is contained in:
Evgeny Gerashchenko
@@ -16,24 +16,14 @@
|
||||
|
||||
package org.jetbrains.jet.plugin;
|
||||
|
||||
import com.intellij.ide.highlighter.JavaClassFileType;
|
||||
import com.intellij.lang.Language;
|
||||
import com.intellij.openapi.fileTypes.ContentBasedClassFileProcessor;
|
||||
import com.intellij.openapi.fileTypes.FileTypeManager;
|
||||
import com.intellij.openapi.fileTypes.SyntaxHighlighter;
|
||||
import com.intellij.openapi.project.Project;
|
||||
import com.intellij.openapi.vfs.VirtualFile;
|
||||
import com.intellij.psi.*;
|
||||
import com.intellij.psi.impl.compiled.ClsAnnotationImpl;
|
||||
import com.intellij.psi.PsiFile;
|
||||
import com.intellij.psi.impl.compiled.ClsFileImpl;
|
||||
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
|
||||
@@ -42,102 +32,18 @@ import java.util.Arrays;
|
||||
public class JetContentBasedFileSubstitutor implements ContentBasedClassFileProcessor {
|
||||
|
||||
@Override
|
||||
public boolean isApplicable(Project project, VirtualFile vFile) {
|
||||
if (!FileTypeManager.getInstance().isFileOfType(vFile, JavaClassFileType.INSTANCE)) {
|
||||
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;
|
||||
public boolean isApplicable(Project project, VirtualFile file) {
|
||||
return JetDecompiledData.getClsFileIfKotlin(project, file) != null;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Override
|
||||
public String obtainFileText(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);
|
||||
ClsFileImpl clsFile = JetDecompiledData.getClsFileIfKotlin(project, 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 JetDecompiledData.getDecompiledData(clsFile).getText();
|
||||
}
|
||||
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");
|
||||
return "";
|
||||
}
|
||||
|
||||
@Override
|
||||
@@ -154,9 +60,7 @@ public class JetContentBasedFileSubstitutor implements ContentBasedClassFileProc
|
||||
@NotNull
|
||||
@Override
|
||||
public PsiFile getDecompiledPsiFile(PsiFile psiFile) {
|
||||
Project project = psiFile.getProject();
|
||||
String text = obtainFileText(project, psiFile.getVirtualFile());
|
||||
return PsiFileFactory.getInstance(project).createFileFromText("", JetLanguage.INSTANCE, text);
|
||||
return JetDecompiledData.getDecompiledData((ClsFileImpl) psiFile).getJetFile();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -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");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user