2 * Back 2 Browser Bytecode Translator
3 * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, version 2 of the License.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. Look for COPYING file in the top folder.
16 * If not, see http://opensource.org/licenses/GPL-2.0.
18 package org.apidesign.bck2brwsr.htmlpage;
20 import java.io.IOException;
21 import java.io.InputStream;
22 import java.io.OutputStreamWriter;
23 import java.io.StringWriter;
24 import java.io.Writer;
25 import java.util.ArrayList;
26 import java.util.Collection;
27 import java.util.Collections;
28 import java.util.HashMap;
29 import java.util.LinkedHashSet;
30 import java.util.List;
31 import java.util.Locale;
34 import javax.annotation.processing.AbstractProcessor;
35 import javax.annotation.processing.Completion;
36 import javax.annotation.processing.Completions;
37 import javax.annotation.processing.Processor;
38 import javax.annotation.processing.RoundEnvironment;
39 import javax.annotation.processing.SupportedAnnotationTypes;
40 import javax.lang.model.element.AnnotationMirror;
41 import javax.lang.model.element.Element;
42 import javax.lang.model.element.ElementKind;
43 import javax.lang.model.element.ExecutableElement;
44 import javax.lang.model.element.Modifier;
45 import javax.lang.model.element.PackageElement;
46 import javax.lang.model.element.TypeElement;
47 import javax.lang.model.element.VariableElement;
48 import javax.lang.model.type.MirroredTypeException;
49 import javax.lang.model.type.TypeKind;
50 import javax.lang.model.type.TypeMirror;
51 import javax.lang.model.util.Types;
52 import javax.tools.Diagnostic;
53 import javax.tools.FileObject;
54 import javax.tools.StandardLocation;
55 import org.apidesign.bck2brwsr.htmlpage.api.ComputedProperty;
56 import org.apidesign.bck2brwsr.htmlpage.api.Model;
57 import org.apidesign.bck2brwsr.htmlpage.api.On;
58 import org.apidesign.bck2brwsr.htmlpage.api.OnFunction;
59 import org.apidesign.bck2brwsr.htmlpage.api.Page;
60 import org.apidesign.bck2brwsr.htmlpage.api.Property;
61 import org.openide.util.lookup.ServiceProvider;
63 /** Annotation processor to process an XHTML page and generate appropriate
66 * @author Jaroslav Tulach <jtulach@netbeans.org>
68 @ServiceProvider(service=Processor.class)
69 @SupportedAnnotationTypes({
70 "org.apidesign.bck2brwsr.htmlpage.api.Model",
71 "org.apidesign.bck2brwsr.htmlpage.api.Page",
72 "org.apidesign.bck2brwsr.htmlpage.api.OnFunction",
73 "org.apidesign.bck2brwsr.htmlpage.api.On"
75 public final class PageProcessor extends AbstractProcessor {
77 public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
79 for (Element e : roundEnv.getElementsAnnotatedWith(Model.class)) {
80 if (!processModel(e)) {
84 for (Element e : roundEnv.getElementsAnnotatedWith(Page.class)) {
85 if (!processPage(e)) {
92 private InputStream openStream(String pkg, String name) throws IOException {
94 FileObject fo = processingEnv.getFiler().getResource(
95 StandardLocation.SOURCE_PATH, pkg, name);
96 return fo.openInputStream();
97 } catch (IOException ex) {
98 return processingEnv.getFiler().getResource(StandardLocation.CLASS_OUTPUT, pkg, name).openInputStream();
102 private boolean processModel(Element e) {
104 Model m = e.getAnnotation(Model.class);
108 String pkg = findPkgName(e);
110 String className = m.className();
112 StringWriter body = new StringWriter();
113 List<String> propsGetSet = new ArrayList<>();
114 Map<String, Collection<String>> propsDeps = new HashMap<>();
115 if (!generateComputedProperties(body, m.properties(), e.getEnclosedElements(), propsGetSet, propsDeps)) {
118 if (!generateProperties(e, body, m.properties(), propsGetSet, propsDeps)) {
121 FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + className, e);
122 w = new OutputStreamWriter(java.openOutputStream());
124 w.append("package " + pkg + ";\n");
125 w.append("import org.apidesign.bck2brwsr.htmlpage.api.*;\n");
126 w.append("import org.apidesign.bck2brwsr.htmlpage.KOList;\n");
127 w.append("final class ").append(className).append(" {\n");
128 w.append(" private Object json;\n");
129 w.append(" private boolean locked;\n");
130 w.append(" private org.apidesign.bck2brwsr.htmlpage.Knockout ko;\n");
131 w.append(body.toString());
136 } catch (IOException ex) {
137 processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "Can't create " + className + ".java", e);
143 private boolean processPage(Element e) {
145 Page p = e.getAnnotation(Page.class);
149 String pkg = findPkgName(e);
152 try (InputStream is = openStream(pkg, p.xhtml())) {
153 pp = ProcessPage.readPage(is);
155 } catch (IOException iOException) {
156 processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "Can't read " + p.xhtml(), e);
161 String className = p.className();
162 if (className.isEmpty()) {
163 int indx = p.xhtml().indexOf('.');
164 className = p.xhtml().substring(0, indx);
167 StringWriter body = new StringWriter();
168 List<String> propsGetSet = new ArrayList<>();
169 List<String> functions = new ArrayList<>();
170 Map<String, Collection<String>> propsDeps = new HashMap<>();
171 if (!generateComputedProperties(body, p.properties(), e.getEnclosedElements(), propsGetSet, propsDeps)) {
174 if (!generateProperties(e, body, p.properties(), propsGetSet, propsDeps)) {
177 if (!generateFunctions(e, body, className, e.getEnclosedElements(), functions)) {
181 FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + className, e);
182 w = new OutputStreamWriter(java.openOutputStream());
184 w.append("package " + pkg + ";\n");
185 w.append("import org.apidesign.bck2brwsr.htmlpage.api.*;\n");
186 w.append("import org.apidesign.bck2brwsr.htmlpage.KOList;\n");
187 w.append("final class ").append(className).append(" {\n");
188 w.append(" private boolean locked;\n");
189 if (!initializeOnClick(className, (TypeElement) e, w, pp)) {
192 for (String id : pp.ids()) {
193 String tag = pp.tagNameForId(id);
194 String type = type(tag);
195 w.append(" ").append("public final ").
196 append(type).append(' ').append(cnstnt(id)).append(" = new ").
197 append(type).append("(\"").append(id).append("\");\n");
200 w.append(" private org.apidesign.bck2brwsr.htmlpage.Knockout ko;\n");
201 w.append(body.toString());
202 if (!propsGetSet.isEmpty()) {
203 w.write("public " + className + " applyBindings() {\n");
204 w.write(" ko = org.apidesign.bck2brwsr.htmlpage.Knockout.applyBindings(");
205 w.write(className + ".class, this, ");
206 w.write("new String[] {\n");
208 for (String n : propsGetSet) {
213 w.write(" \"" + n + "\"");
217 w.write("\n }, new String[] {\n");
219 for (String n : functions) {
224 w.write("\n });\n return this;\n}\n");
226 w.write("public void triggerEvent(Element e, OnEvent ev) {\n");
227 w.write(" org.apidesign.bck2brwsr.htmlpage.Knockout.triggerEvent(e.getId(), ev.getElementPropertyName());\n");
234 } catch (IOException ex) {
235 processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "Can't create " + className + ".java", e);
241 private static String type(String tag) {
242 if (tag.equals("title")) {
245 if (tag.equals("button")) {
248 if (tag.equals("input")) {
251 if (tag.equals("canvas")) {
254 if (tag.equals("img")) {
260 private static String cnstnt(String id) {
261 return id.toUpperCase(Locale.ENGLISH).replace('.', '_').replace('-', '_');
264 private boolean initializeOnClick(
265 String className, TypeElement type, Writer w, ProcessPage pp
266 ) throws IOException {
268 TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
269 { //for (Element clazz : pe.getEnclosedElements()) {
270 // if (clazz.getKind() != ElementKind.CLASS) {
273 w.append(" public ").append(className).append("() {\n");
274 StringBuilder dispatch = new StringBuilder();
276 for (Element method : type.getEnclosedElements()) {
277 On oc = method.getAnnotation(On.class);
279 for (String id : oc.id()) {
281 processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "id = " + id + " not found in HTML page.");
285 if (pp.tagNameForId(id) == null) {
286 processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "id = " + id + " does not exist in the HTML page. Found only " + pp.ids(), method);
290 ExecutableElement ee = (ExecutableElement)method;
291 CharSequence params = wrapParams(ee, id, className, "ev", null);
292 if (!ee.getModifiers().contains(Modifier.STATIC)) {
293 processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "@On method has to be static", ee);
297 if (ee.getModifiers().contains(Modifier.PRIVATE)) {
298 processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "@On method can't be private", ee);
302 w.append(" OnEvent." + oc.event()).append(".of(").append(cnstnt(id)).
303 append(").perform(new OnDispatch(" + dispatchCnt + "));\n");
306 append(" case ").append(dispatchCnt).append(": ").
307 append(type.getSimpleName().toString()).
308 append('.').append(ee.getSimpleName()).append("(").
310 append("); break;\n");
317 if (dispatchCnt > 0) {
318 w.append("class OnDispatch implements OnHandler {\n");
319 w.append(" private final int dispatch;\n");
320 w.append(" OnDispatch(int d) { dispatch = d; }\n");
321 w.append(" public void onEvent(Object ev) {\n");
322 w.append(" switch (dispatch) {\n");
335 public Iterable<? extends Completion> getCompletions(
336 Element element, AnnotationMirror annotation,
337 ExecutableElement member, String userText
339 if (!userText.startsWith("\"")) {
340 return Collections.emptyList();
343 Element cls = findClass(element);
344 Page p = cls.getAnnotation(Page.class);
345 String pkg = findPkgName(cls);
348 InputStream is = openStream(pkg, p.xhtml());
349 pp = ProcessPage.readPage(is);
351 } catch (IOException iOException) {
352 return Collections.emptyList();
355 List<Completion> cc = new ArrayList<>();
356 userText = userText.substring(1);
357 for (String id : pp.ids()) {
358 if (id.startsWith(userText)) {
359 cc.add(Completions.of("\"" + id + "\"", id));
365 private static Element findClass(Element e) {
369 Page p = e.getAnnotation(Page.class);
373 return e.getEnclosingElement();
376 private boolean generateProperties(
378 Writer w, Property[] properties,
379 Collection<String> props, Map<String,Collection<String>> deps
380 ) throws IOException {
382 for (Property p : properties) {
384 tn = typeName(where, p);
385 String[] gs = toGetSet(p.name(), tn, p.array());
388 w.write("private KOList<" + tn + "> prop_" + p.name() + " = new KOList<" + tn + ">(\""
390 final Collection<String> dependants = deps.get(p.name());
391 if (dependants != null) {
392 for (String depProp : dependants) {
400 w.write("public java.util.List<" + tn + "> " + gs[0] + "() {\n");
401 w.write(" if (locked) throw new IllegalStateException();\n");
402 w.write(" prop_" + p.name() + ".assign(ko);\n");
403 w.write(" return prop_" + p.name() + ";\n");
405 w.write("private Object[] " + gs[0] + "ToArray() {\n");
406 w.write(" return " + gs[0] + "().toArray(new Object[0]);\n");
409 w.write("private " + tn + " prop_" + p.name() + ";\n");
410 w.write("public " + tn + " " + gs[0] + "() {\n");
411 w.write(" if (locked) throw new IllegalStateException();\n");
412 w.write(" return prop_" + p.name() + ";\n");
414 w.write("public void " + gs[1] + "(" + tn + " v) {\n");
415 w.write(" if (locked) throw new IllegalStateException();\n");
416 w.write(" prop_" + p.name() + " = v;\n");
417 w.write(" if (ko != null) {\n");
418 w.write(" ko.valueHasMutated(\"" + p.name() + "\");\n");
419 final Collection<String> dependants = deps.get(p.name());
420 if (dependants != null) {
421 for (String depProp : dependants) {
422 w.write(" ko.valueHasMutated(\"" + depProp + "\");\n");
437 private boolean generateComputedProperties(
438 Writer w, Property[] fixedProps,
439 Collection<? extends Element> arr, Collection<String> props,
440 Map<String,Collection<String>> deps
441 ) throws IOException {
443 for (Element e : arr) {
444 if (e.getKind() != ElementKind.METHOD) {
447 if (e.getAnnotation(ComputedProperty.class) == null) {
450 ExecutableElement ee = (ExecutableElement)e;
451 final TypeMirror rt = ee.getReturnType();
452 final Types tu = processingEnv.getTypeUtils();
453 TypeMirror ert = tu.erasure(rt);
454 String tn = ert.toString();
455 boolean array = false;
456 if (tn.equals("java.util.List")) {
460 final String sn = ee.getSimpleName().toString();
461 String[] gs = toGetSet(sn, tn, array);
463 w.write("public " + tn + " " + gs[0] + "() {\n");
464 w.write(" if (locked) throw new IllegalStateException();\n");
466 for (VariableElement pe : ee.getParameters()) {
467 final String dn = pe.getSimpleName().toString();
469 if (!verifyPropName(pe, dn, fixedProps)) {
473 final String dt = pe.asType().toString();
474 String[] call = toGetSet(dn, dt, false);
475 w.write(" " + dt + " arg" + (++arg) + " = ");
476 w.write(call[0] + "();\n");
478 Collection<String> depends = deps.get(dn);
479 if (depends == null) {
480 depends = new LinkedHashSet<>();
481 deps.put(dn, depends);
486 w.write(" locked = true;\n");
487 w.write(" return " + e.getEnclosingElement().getSimpleName() + '.' + e.getSimpleName() + "(");
489 for (int i = 1; i <= arg; i++) {
495 w.write(" } finally {\n");
496 w.write(" locked = false;\n");
501 w.write("private Object[] " + gs[0] + "ToArray() {\n");
502 w.write(" return " + gs[0] + "().toArray(new Object[0]);\n");
505 props.add(e.getSimpleName().toString());
514 private static String[] toGetSet(String name, String type, boolean array) {
515 String n = Character.toUpperCase(name.charAt(0)) + name.substring(1);
516 String bck2brwsrType = "L" + type.replace('.', '_') + "_2";
517 if ("int".equals(type)) {
520 if ("double".equals(type)) {
524 if ("boolean".equals(type)) {
528 final String nu = n.replace('.', '_');
530 return new String[] {
533 "get" + nu + "ToArray___3Ljava_lang_Object_2",
540 pref + nu + "__" + bck2brwsrType,
541 "set" + nu + "__V" + bck2brwsrType
545 private String typeName(Element where, Property p) {
547 boolean isModel = false;
549 ret = p.type().getName();
550 } catch (MirroredTypeException ex) {
551 TypeMirror tm = processingEnv.getTypeUtils().erasure(ex.getTypeMirror());
552 final Element e = processingEnv.getTypeUtils().asElement(tm);
553 final Model m = e == null ? null : e.getAnnotation(Model.class);
555 ret = findPkgName(e) + '.' + m.className();
562 String bt = findBoxedType(ret);
567 if (!isModel && !"java.lang.String".equals(ret)) {
568 String bt = findBoxedType(ret);
570 processingEnv.getMessager().printMessage(
571 Diagnostic.Kind.ERROR,
572 "Only primitive types supported in the mapping. Not " + ret,
580 private static String findBoxedType(String ret) {
581 if (ret.equals("boolean")) {
582 return Boolean.class.getName();
584 if (ret.equals("byte")) {
585 return Byte.class.getName();
587 if (ret.equals("short")) {
588 return Short.class.getName();
590 if (ret.equals("char")) {
591 return Character.class.getName();
593 if (ret.equals("int")) {
594 return Integer.class.getName();
596 if (ret.equals("long")) {
597 return Long.class.getName();
599 if (ret.equals("float")) {
600 return Float.class.getName();
602 if (ret.equals("double")) {
603 return Double.class.getName();
608 private boolean verifyPropName(Element e, String propName, Property[] existingProps) {
609 StringBuilder sb = new StringBuilder();
611 for (Property property : existingProps) {
612 if (property.name().equals(propName)) {
617 sb.append(property.name());
621 processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR,
622 propName + " is not one of known properties: " + sb
628 private static String findPkgName(Element e) {
630 if (e.getKind() == ElementKind.PACKAGE) {
631 return ((PackageElement)e).getQualifiedName().toString();
633 e = e.getEnclosingElement();
637 private boolean generateFunctions(
638 Element clazz, StringWriter body, String className,
639 List<? extends Element> enclosedElements, List<String> functions
641 for (Element m : enclosedElements) {
642 if (m.getKind() != ElementKind.METHOD) {
645 ExecutableElement e = (ExecutableElement)m;
646 OnFunction onF = e.getAnnotation(OnFunction.class);
650 if (!e.getModifiers().contains(Modifier.STATIC)) {
651 processingEnv.getMessager().printMessage(
652 Diagnostic.Kind.ERROR, "@OnFunction method needs to be static", e
656 if (e.getModifiers().contains(Modifier.PRIVATE)) {
657 processingEnv.getMessager().printMessage(
658 Diagnostic.Kind.ERROR, "@OnFunction method cannot be private", e
662 if (e.getReturnType().getKind() != TypeKind.VOID) {
663 processingEnv.getMessager().printMessage(
664 Diagnostic.Kind.ERROR, "@OnFunction method should return void", e
668 String n = e.getSimpleName().toString();
669 body.append("private void ").append(n).append("(Object data, Object ev) {\n");
670 body.append(" ").append(clazz.getSimpleName()).append(".").append(n).append("(");
671 body.append(wrapParams(e, null, className, "ev", "data"));
675 functions.add('\"' + n + '\"');
676 functions.add('\"' + n + "__VLjava_lang_Object_2Ljava_lang_Object_2" + '\"');
681 private CharSequence wrapParams(
682 ExecutableElement ee, String id, String className, String evName, String dataName
684 TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
685 StringBuilder params = new StringBuilder();
686 boolean first = true;
687 for (VariableElement ve : ee.getParameters()) {
692 String toCall = null;
693 if (ve.asType() == stringType) {
694 if (ve.getSimpleName().contentEquals("id")) {
695 params.append('"').append(id).append('"');
698 toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toString";
700 if (ve.asType().getKind() == TypeKind.DOUBLE) {
701 toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toDouble";
703 if (ve.asType().getKind() == TypeKind.INT) {
704 toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toInt";
707 if (toCall != null) {
708 params.append(toCall).append('(');
709 if (dataName != null && ve.getSimpleName().contentEquals("data")) {
710 params.append(dataName);
711 params.append(", null");
713 params.append(evName);
714 params.append(", \"");
715 params.append(ve.getSimpleName().toString());
721 String rn = ve.asType().toString();
722 int last = rn.lastIndexOf('.');
724 rn = rn.substring(last + 1);
726 if (rn.equals(className)) {
727 params.append(className).append(".this");
730 processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR,
731 "@On method can only accept String named 'id' or " + className + " arguments",