ide/editor/src/main/java/org/apidesign/bck2brwsr/ide/editor/JSEmbeddingProvider.java
1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/ide/editor/src/main/java/org/apidesign/bck2brwsr/ide/editor/JSEmbeddingProvider.java Wed Feb 13 15:21:54 2013 +0100
1.3 @@ -0,0 +1,175 @@
1.4 +/*
1.5 + * To change this template, choose Tools | Templates
1.6 + * and open the template in the editor.
1.7 + */
1.8 +package org.apidesign.bck2brwsr.ide.editor;
1.9 +
1.10 +import com.sun.source.tree.AnnotationTree;
1.11 +import com.sun.source.tree.AssignmentTree;
1.12 +import com.sun.source.tree.CompilationUnitTree;
1.13 +import com.sun.source.tree.ExpressionTree;
1.14 +import com.sun.source.tree.LiteralTree;
1.15 +import com.sun.source.tree.MethodTree;
1.16 +import com.sun.source.util.SourcePositions;
1.17 +import com.sun.source.util.TreePath;
1.18 +import com.sun.source.util.TreePathScanner;
1.19 +import com.sun.source.util.Trees;
1.20 +import java.io.IOException;
1.21 +import java.util.ArrayList;
1.22 +import java.util.Collection;
1.23 +import java.util.Collections;
1.24 +import java.util.List;
1.25 +import java.util.concurrent.atomic.AtomicBoolean;
1.26 +import javax.lang.model.element.TypeElement;
1.27 +import javax.swing.text.Document;
1.28 +import org.netbeans.api.editor.mimelookup.MimeRegistration;
1.29 +import org.netbeans.api.java.source.CompilationInfo;
1.30 +import org.netbeans.api.java.source.JavaParserResultTask;
1.31 +import org.netbeans.api.java.source.JavaSource;
1.32 +import org.netbeans.api.lexer.Language;
1.33 +import org.netbeans.api.lexer.TokenHierarchy;
1.34 +import org.netbeans.api.lexer.TokenSequence;
1.35 +import org.netbeans.modules.parsing.api.Snapshot;
1.36 +import org.netbeans.modules.parsing.spi.Parser;
1.37 +import org.netbeans.modules.parsing.spi.Scheduler;
1.38 +import org.netbeans.modules.parsing.spi.SchedulerEvent;
1.39 +import org.netbeans.modules.parsing.spi.SchedulerTask;
1.40 +import org.netbeans.modules.parsing.spi.TaskFactory;
1.41 +import org.openide.util.Exceptions;
1.42 +
1.43 +/**
1.44 + *
1.45 + * @author Tomas Zezula
1.46 + */
1.47 +public final class JSEmbeddingProvider extends JavaParserResultTask<Parser.Result> {
1.48 +
1.49 + private static final int PRIORITY = 1000;
1.50 + private static final String JS_ANNOTATION = "org.apidesign.bck2brwsr.core.JavaScriptBody"; //NOI18N
1.51 + private static final String BODY = "body"; //NOI18N
1.52 + private static final String JAVA_MIME_TYPE = "text/x-java"; //NOI18N
1.53 + private static final String JAVASCRIPT_MIME_TYPE = "text/javascript"; //NOI18N
1.54 + private final AtomicBoolean canceled = new AtomicBoolean();
1.55 +
1.56 + private JSEmbeddingProvider() {
1.57 + super(JavaSource.Phase.ELEMENTS_RESOLVED);
1.58 + }
1.59 +
1.60 + @Override
1.61 + public int getPriority() {
1.62 + return PRIORITY;
1.63 + }
1.64 +
1.65 + @Override
1.66 + public void cancel() {
1.67 + canceled.set(true);
1.68 + }
1.69 +
1.70 + @Override
1.71 + public Class<? extends Scheduler> getSchedulerClass() {
1.72 + return Scheduler.EDITOR_SENSITIVE_TASK_SCHEDULER;
1.73 + }
1.74 +
1.75 + @Override
1.76 + public void run(Parser.Result t, SchedulerEvent se) {
1.77 + canceled.set(false);
1.78 + final CompilationInfo ci = CompilationInfo.get(t);
1.79 + final CompilationUnitTree cu = ci.getCompilationUnit();
1.80 + final Trees trees = ci.getTrees();
1.81 + final SourcePositions sp = trees.getSourcePositions();
1.82 + final Finder f = new Finder(trees);
1.83 + final List<LiteralTree> result = new ArrayList<LiteralTree>();
1.84 + f.scan(cu, result);
1.85 + if (!result.isEmpty()) {
1.86 + try {
1.87 + final TokenHierarchy<Document> tk = TokenHierarchy.get(ci.getDocument());
1.88 + final Language<?> java = Language.find(JAVA_MIME_TYPE);
1.89 + final Language<?> javaScript = Language.find(JAVASCRIPT_MIME_TYPE);
1.90 + if (java != null && javaScript != null) {
1.91 + final TokenSequence<?> seq = tk.tokenSequence(java);
1.92 + if (seq != null) {
1.93 + for (LiteralTree lt : result) {
1.94 + final int start = (int) sp.getStartPosition(cu, lt);
1.95 + final int end = (int) sp.getEndPosition(cu, lt);
1.96 + seq.move(start);
1.97 + while (seq.moveNext() && seq.offset() < end) {
1.98 + seq.createEmbedding(javaScript, 1, 1, true);
1.99 + }
1.100 + }
1.101 + }
1.102 + }
1.103 + } catch (IOException ioe) {
1.104 + Exceptions.printStackTrace(ioe);
1.105 + }
1.106 + }
1.107 + }
1.108 +
1.109 +
1.110 +
1.111 +
1.112 + private static final class Finder extends TreePathScanner<Void, List<? super LiteralTree>> {
1.113 +
1.114 + private final Trees trees;
1.115 + private CompilationUnitTree cu;
1.116 + private boolean inEmbedding;
1.117 +
1.118 + Finder(final Trees trees) {
1.119 + this.trees = trees;
1.120 + }
1.121 +
1.122 + @Override
1.123 + public Void visitCompilationUnit(
1.124 + final CompilationUnitTree unit,
1.125 + final List p) {
1.126 + this.cu = unit;
1.127 + return super.visitCompilationUnit(unit, p);
1.128 + }
1.129 +
1.130 +
1.131 +
1.132 + @Override
1.133 + public Void visitMethod(
1.134 + final MethodTree m,
1.135 + final List<? super LiteralTree> p) {
1.136 + for (AnnotationTree a : m.getModifiers().getAnnotations()) {
1.137 + final TypeElement ae = (TypeElement) trees.getElement(TreePath.getPath(cu, a.getAnnotationType()));
1.138 + if (ae != null && JS_ANNOTATION.contentEquals(ae.getQualifiedName())) {
1.139 + final List<? extends ExpressionTree> args = a.getArguments();
1.140 + for (ExpressionTree kvp : args) {
1.141 + if (kvp instanceof AssignmentTree) {
1.142 + final AssignmentTree assignemt = (AssignmentTree) kvp;
1.143 + if (BODY.equals(assignemt.getVariable().toString())) {
1.144 + inEmbedding = true;
1.145 + try {
1.146 + scan(assignemt.getExpression(), p);
1.147 + } finally {
1.148 + inEmbedding = false;
1.149 + }
1.150 + }
1.151 + }
1.152 + }
1.153 + }
1.154 + }
1.155 + return null;
1.156 + }
1.157 +
1.158 + @Override
1.159 + public Void visitLiteral(LiteralTree node, List<? super LiteralTree> p) {
1.160 + if (inEmbedding) {
1.161 + p.add(node);
1.162 + }
1.163 + return super.visitLiteral(node, p);
1.164 + }
1.165 +
1.166 + }
1.167 +
1.168 + @MimeRegistration(
1.169 + service = TaskFactory.class,
1.170 + mimeType = JAVA_MIME_TYPE)
1.171 + public static final class Factory extends TaskFactory {
1.172 + @Override
1.173 + public Collection<? extends SchedulerTask> create(Snapshot snpsht) {
1.174 + return Collections.singleton(new JSEmbeddingProvider());
1.175 + }
1.176 + }
1.177 +
1.178 +}