Bytecode viewer, initial

2011-11-23 21:29:36 +04:00
parent 9a25b1a843
commit 6eea1716d3
2 changed files with 165 additions and 0 deletions
@@ -58,5 +58,8 @@
      <codeInsight.overrideMethod language="jet" implementationClass="org.jetbrains.jet.plugin.codeInsight.OverrideMethodsHandler"/>

      <java.elementFinder implementation="org.jetbrains.jet.plugin.java.JavaElementFinder"/>
+      <toolWindow id="CodeWindow"
+                  factoryClass="org.jetbrains.jet.plugin.internal.codewindow.BytecodeToolwindow$Factory"
+                  anchor="right"/>
  </extensions>
 </idea-plugin>
@@ -0,0 +1,162 @@
+/*
+ * @author max
+ */
+package org.jetbrains.jet.plugin.internal.codewindow;
+
+import com.intellij.ide.highlighter.JavaFileType;
+import com.intellij.openapi.application.Result;
+import com.intellij.openapi.command.WriteCommandAction;
+import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.editor.EditorFactory;
+import com.intellij.openapi.editor.ex.EditorEx;
+import com.intellij.openapi.fileEditor.FileEditorManager;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.util.Comparing;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.openapi.wm.ToolWindow;
+import com.intellij.openapi.wm.ToolWindowFactory;
+import com.intellij.psi.PsiFile;
+import com.intellij.psi.PsiManager;
+import com.intellij.ui.content.ContentFactory;
+import com.intellij.util.Alarm;
+import org.jetbrains.jet.codegen.ClassBuilderFactory;
+import org.jetbrains.jet.codegen.ClassFileFactory;
+import org.jetbrains.jet.codegen.GenerationState;
+import org.jetbrains.jet.lang.psi.JetFile;
+import org.jetbrains.jet.lang.resolve.AnalyzingUtils;
+
+import javax.swing.*;
+import java.awt.*;
+
+public class BytecodeToolwindow extends JPanel {
+    private static final int UPDATE_DELAY = 500;
+    private final Editor myEditor;
+    private final Alarm myUpdateAlarm;
+    private Location myCurrentLocation;
+    private final Project myProject;
+    
+
+    public BytecodeToolwindow(Project project) {
+        super(new BorderLayout());
+        myProject = project;
+        myEditor = EditorFactory.getInstance().createEditor(EditorFactory.getInstance().createDocument(""), project, JavaFileType.INSTANCE, true);
+        add(myEditor.getComponent());
+        myUpdateAlarm = new Alarm(Alarm.ThreadToUse.SWING_THREAD);
+        myUpdateAlarm.addRequest(new Runnable() {
+            @Override
+            public void run() {
+                Location location = Location.fromEditor(FileEditorManager.getInstance(myProject).getSelectedTextEditor());
+                if (!Comparing.equal(location, myCurrentLocation)) {
+                    myCurrentLocation = location;
+                    updateBytecode(location);
+                }
+                myUpdateAlarm.addRequest(this, UPDATE_DELAY);
+            }
+        }, UPDATE_DELAY);
+    }
+
+    private void updateBytecode(Location location) {
+        Editor editor = location.editor;
+
+        if (editor == null) {
+            setText("");
+        }
+        else {
+            VirtualFile vFile = ((EditorEx) editor).getVirtualFile();
+            if (vFile == null) {
+                setText("");
+                return;
+            }
+
+            PsiFile psiFile = PsiManager.getInstance(myProject).findFile(vFile);
+            if (!(psiFile instanceof JetFile)) {
+                setText("");
+                return;
+            }
+
+            setText(generateToText((JetFile) psiFile));
+        }
+    }
+
+    private void setText(final String text) {
+        new WriteCommandAction(myProject) {
+            @Override
+            protected void run(Result result) throws Throwable {
+                myEditor.getDocument().setText(text);
+            }
+        }.execute();
+    }
+
+    protected String generateToText(JetFile file) {
+        GenerationState state = new GenerationState(myProject, ClassBuilderFactory.TEXT);
+        try {
+            AnalyzingUtils.checkForSyntacticErrors(file);
+            state.compile(file);
+        } catch (Exception e) {
+            return e.toString();
+        }
+
+
+        StringBuilder answer = new StringBuilder();
+
+        final ClassFileFactory factory = state.getFactory();
+        for (String filename : factory.files()) {
+            answer.append("// ================ ");
+            answer.append(filename);
+            answer.append(" =================\n");
+            answer.append(factory.asText(filename)).append("\n\n");
+        }
+
+        return answer.toString();
+    }
+    
+
+    public static class Factory implements ToolWindowFactory {
+        @Override
+        public void createToolWindowContent(Project project, ToolWindow toolWindow) {
+            toolWindow.getContentManager().addContent(ContentFactory.SERVICE.getInstance().createContent(new BytecodeToolwindow(project), "", false));
+        }
+    }
+    
+    private static class Location {
+        final Editor editor;
+        final long modificationStamp;
+        final int startOffset;
+        final int endOffset;
+
+        private Location(Editor editor) {
+            this.editor = editor;
+            modificationStamp = editor != null ? editor.getDocument().getModificationStamp() : 0;
+            startOffset = editor != null ? editor.getSelectionModel().getSelectionStart() : 0;
+            endOffset = editor != null ? editor.getSelectionModel().getSelectionEnd() : 0;
+        }
+
+        public static Location fromEditor(Editor editor) {
+            return new Location(editor);
+        }
+
+        @Override
+        public boolean equals(Object o) {
+            if (this == o) return true;
+            if (!(o instanceof Location)) return false;
+
+            Location location = (Location) o;
+
+            if (endOffset != location.endOffset) return false;
+            if (modificationStamp != location.modificationStamp) return false;
+            if (startOffset != location.startOffset) return false;
+            if (editor != null ? !editor.equals(location.editor) : location.editor != null) return false;
+
+            return true;
+        }
+
+        @Override
+        public int hashCode() {
+            int result = editor != null ? editor.hashCode() : 0;
+            result = 31 * result + (int) (modificationStamp ^ (modificationStamp >>> 32));
+            result = 31 * result + startOffset;
+            result = 31 * result + endOffset;
+            return result;
+        }
+    }    
+}