javaquery/api/src/main/java/org/apidesign/bck2brwsr/htmlpage/PageProcessor.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Thu, 04 Apr 2013 11:45:54 +0200
branchmodel
changeset 929 b43aaf398748
parent 927 1b7c8f6cb621
child 930 e8916518b38d
permissions -rw-r--r--
Assume the class belongs to the same package if it is not yet resolvable
jaroslav@26
     1
/**
jaroslav@106
     2
 * Back 2 Browser Bytecode Translator
jaroslav@106
     3
 * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
jaroslav@26
     4
 *
jaroslav@26
     5
 * This program is free software: you can redistribute it and/or modify
jaroslav@26
     6
 * it under the terms of the GNU General Public License as published by
jaroslav@26
     7
 * the Free Software Foundation, version 2 of the License.
jaroslav@26
     8
 *
jaroslav@26
     9
 * This program is distributed in the hope that it will be useful,
jaroslav@26
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
jaroslav@26
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
jaroslav@26
    12
 * GNU General Public License for more details.
jaroslav@26
    13
 *
jaroslav@26
    14
 * You should have received a copy of the GNU General Public License
jaroslav@26
    15
 * along with this program. Look for COPYING file in the top folder.
jaroslav@26
    16
 * If not, see http://opensource.org/licenses/GPL-2.0.
jaroslav@26
    17
 */
jaroslav@26
    18
package org.apidesign.bck2brwsr.htmlpage;
jaroslav@26
    19
jaroslav@26
    20
import java.io.IOException;
jaroslav@26
    21
import java.io.InputStream;
jaroslav@26
    22
import java.io.OutputStreamWriter;
jaroslav@770
    23
import java.io.StringWriter;
jaroslav@26
    24
import java.io.Writer;
jaroslav@117
    25
import java.util.ArrayList;
jaroslav@491
    26
import java.util.Collection;
jaroslav@117
    27
import java.util.Collections;
jaroslav@498
    28
import java.util.HashMap;
jaroslav@498
    29
import java.util.LinkedHashSet;
jaroslav@117
    30
import java.util.List;
jaroslav@498
    31
import java.util.Map;
jaroslav@26
    32
import java.util.Set;
jaroslav@906
    33
import java.util.WeakHashMap;
jaroslav@26
    34
import javax.annotation.processing.AbstractProcessor;
jaroslav@117
    35
import javax.annotation.processing.Completion;
jaroslav@117
    36
import javax.annotation.processing.Completions;
jaroslav@26
    37
import javax.annotation.processing.Processor;
jaroslav@26
    38
import javax.annotation.processing.RoundEnvironment;
jaroslav@26
    39
import javax.annotation.processing.SupportedAnnotationTypes;
jaroslav@117
    40
import javax.lang.model.element.AnnotationMirror;
jaroslav@26
    41
import javax.lang.model.element.Element;
jaroslav@28
    42
import javax.lang.model.element.ElementKind;
jaroslav@28
    43
import javax.lang.model.element.ExecutableElement;
jaroslav@28
    44
import javax.lang.model.element.Modifier;
jaroslav@26
    45
import javax.lang.model.element.PackageElement;
jaroslav@26
    46
import javax.lang.model.element.TypeElement;
jaroslav@491
    47
import javax.lang.model.element.VariableElement;
jaroslav@490
    48
import javax.lang.model.type.MirroredTypeException;
jaroslav@813
    49
import javax.lang.model.type.TypeKind;
jaroslav@124
    50
import javax.lang.model.type.TypeMirror;
jaroslav@929
    51
import javax.lang.model.util.Elements;
jaroslav@768
    52
import javax.lang.model.util.Types;
jaroslav@26
    53
import javax.tools.Diagnostic;
jaroslav@26
    54
import javax.tools.FileObject;
jaroslav@26
    55
import javax.tools.StandardLocation;
jaroslav@491
    56
import org.apidesign.bck2brwsr.htmlpage.api.ComputedProperty;
jaroslav@770
    57
import org.apidesign.bck2brwsr.htmlpage.api.Model;
jaroslav@435
    58
import org.apidesign.bck2brwsr.htmlpage.api.On;
jaroslav@879
    59
import org.apidesign.bck2brwsr.htmlpage.api.OnFunction;
jaroslav@26
    60
import org.apidesign.bck2brwsr.htmlpage.api.Page;
jaroslav@490
    61
import org.apidesign.bck2brwsr.htmlpage.api.Property;
jaroslav@26
    62
import org.openide.util.lookup.ServiceProvider;
jaroslav@26
    63
jaroslav@26
    64
/** Annotation processor to process an XHTML page and generate appropriate 
jaroslav@26
    65
 * "id" file.
jaroslav@26
    66
 *
jaroslav@26
    67
 * @author Jaroslav Tulach <jtulach@netbeans.org>
jaroslav@26
    68
 */
jaroslav@26
    69
@ServiceProvider(service=Processor.class)
jaroslav@28
    70
@SupportedAnnotationTypes({
jaroslav@770
    71
    "org.apidesign.bck2brwsr.htmlpage.api.Model",
jaroslav@28
    72
    "org.apidesign.bck2brwsr.htmlpage.api.Page",
jaroslav@879
    73
    "org.apidesign.bck2brwsr.htmlpage.api.OnFunction",
jaroslav@435
    74
    "org.apidesign.bck2brwsr.htmlpage.api.On"
jaroslav@28
    75
})
jaroslav@26
    76
public final class PageProcessor extends AbstractProcessor {
jaroslav@906
    77
    private final Map<Element,String> models = new WeakHashMap<>();
jaroslav@26
    78
    @Override
jaroslav@26
    79
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
jaroslav@765
    80
        boolean ok = true;
jaroslav@770
    81
        for (Element e : roundEnv.getElementsAnnotatedWith(Model.class)) {
jaroslav@770
    82
            if (!processModel(e)) {
jaroslav@770
    83
                ok = false;
jaroslav@26
    84
            }
jaroslav@26
    85
        }
jaroslav@543
    86
        for (Element e : roundEnv.getElementsAnnotatedWith(Page.class)) {
jaroslav@770
    87
            if (!processPage(e)) {
jaroslav@765
    88
                ok = false;
jaroslav@543
    89
            }
jaroslav@543
    90
        }
jaroslav@906
    91
        if (roundEnv.processingOver()) {
jaroslav@906
    92
            models.clear();
jaroslav@906
    93
        }
jaroslav@765
    94
        return ok;
jaroslav@26
    95
    }
jaroslav@26
    96
jaroslav@26
    97
    private InputStream openStream(String pkg, String name) throws IOException {
jaroslav@26
    98
        try {
jaroslav@107
    99
            FileObject fo = processingEnv.getFiler().getResource(
jaroslav@107
   100
                StandardLocation.SOURCE_PATH, pkg, name);
jaroslav@26
   101
            return fo.openInputStream();
jaroslav@26
   102
        } catch (IOException ex) {
jaroslav@26
   103
            return processingEnv.getFiler().getResource(StandardLocation.CLASS_OUTPUT, pkg, name).openInputStream();
jaroslav@26
   104
        }
jaroslav@26
   105
    }
jaroslav@770
   106
    
jaroslav@770
   107
    private boolean processModel(Element e) {
jaroslav@770
   108
        boolean ok = true;
jaroslav@770
   109
        Model m = e.getAnnotation(Model.class);
jaroslav@770
   110
        if (m == null) {
jaroslav@770
   111
            return true;
jaroslav@770
   112
        }
jaroslav@770
   113
        String pkg = findPkgName(e);
jaroslav@770
   114
        Writer w;
jaroslav@770
   115
        String className = m.className();
jaroslav@770
   116
        try {
jaroslav@770
   117
            StringWriter body = new StringWriter();
jaroslav@770
   118
            List<String> propsGetSet = new ArrayList<>();
jaroslav@914
   119
            List<String> functions = new ArrayList<>();
jaroslav@770
   120
            Map<String, Collection<String>> propsDeps = new HashMap<>();
jaroslav@770
   121
            if (!generateComputedProperties(body, m.properties(), e.getEnclosedElements(), propsGetSet, propsDeps)) {
jaroslav@770
   122
                ok = false;
jaroslav@770
   123
            }
jaroslav@875
   124
            if (!generateProperties(e, body, m.properties(), propsGetSet, propsDeps)) {
jaroslav@770
   125
                ok = false;
jaroslav@770
   126
            }
jaroslav@914
   127
            if (!generateFunctions(e, body, className, e.getEnclosedElements(), functions)) {
jaroslav@914
   128
                ok = false;
jaroslav@914
   129
            }
jaroslav@770
   130
            FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + className, e);
jaroslav@770
   131
            w = new OutputStreamWriter(java.openOutputStream());
jaroslav@770
   132
            try {
jaroslav@770
   133
                w.append("package " + pkg + ";\n");
jaroslav@770
   134
                w.append("import org.apidesign.bck2brwsr.htmlpage.api.*;\n");
jaroslav@770
   135
                w.append("import org.apidesign.bck2brwsr.htmlpage.KOList;\n");
jaroslav@908
   136
                w.append("import org.apidesign.bck2brwsr.core.JavaScriptOnly;\n");
jaroslav@770
   137
                w.append("final class ").append(className).append(" {\n");
jaroslav@770
   138
                w.append("  private Object json;\n");
jaroslav@770
   139
                w.append("  private boolean locked;\n");
jaroslav@770
   140
                w.append("  private org.apidesign.bck2brwsr.htmlpage.Knockout ko;\n");
jaroslav@770
   141
                w.append(body.toString());
jaroslav@925
   142
                w.append("  private static Class<" + inPckName(e) + "> modelFor() { return null; }\n");
jaroslav@909
   143
                w.append("  public ").append(className).append("() {\n");
jaroslav@909
   144
                w.append("    ko = org.apidesign.bck2brwsr.htmlpage.Knockout.applyBindings(this, ");
jaroslav@909
   145
                writeStringArray(propsGetSet, w);
jaroslav@909
   146
                w.append(", ");
jaroslav@914
   147
                writeStringArray(functions, w);
jaroslav@909
   148
                w.append("    );\n");
jaroslav@909
   149
                w.append("  };\n");
jaroslav@919
   150
                writeToString(m.properties(), w);
jaroslav@770
   151
                w.append("}\n");
jaroslav@770
   152
            } finally {
jaroslav@770
   153
                w.close();
jaroslav@770
   154
            }
jaroslav@770
   155
        } catch (IOException ex) {
jaroslav@770
   156
            processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "Can't create " + className + ".java", e);
jaroslav@770
   157
            return false;
jaroslav@770
   158
        }
jaroslav@770
   159
        return ok;
jaroslav@770
   160
    }
jaroslav@770
   161
    
jaroslav@770
   162
    private boolean processPage(Element e) {
jaroslav@770
   163
        boolean ok = true;
jaroslav@770
   164
        Page p = e.getAnnotation(Page.class);
jaroslav@770
   165
        if (p == null) {
jaroslav@770
   166
            return true;
jaroslav@770
   167
        }
jaroslav@770
   168
        String pkg = findPkgName(e);
jaroslav@770
   169
jaroslav@770
   170
        ProcessPage pp;
jaroslav@770
   171
        try (InputStream is = openStream(pkg, p.xhtml())) {
jaroslav@770
   172
            pp = ProcessPage.readPage(is);
jaroslav@770
   173
            is.close();
jaroslav@770
   174
        } catch (IOException iOException) {
jaroslav@923
   175
            processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "Can't read " + p.xhtml() + " as " + iOException.getMessage(), e);
jaroslav@770
   176
            ok = false;
jaroslav@770
   177
            pp = null;
jaroslav@770
   178
        }
jaroslav@770
   179
        Writer w;
jaroslav@770
   180
        String className = p.className();
jaroslav@770
   181
        if (className.isEmpty()) {
jaroslav@770
   182
            int indx = p.xhtml().indexOf('.');
jaroslav@770
   183
            className = p.xhtml().substring(0, indx);
jaroslav@770
   184
        }
jaroslav@770
   185
        try {
jaroslav@770
   186
            StringWriter body = new StringWriter();
jaroslav@770
   187
            List<String> propsGetSet = new ArrayList<>();
jaroslav@879
   188
            List<String> functions = new ArrayList<>();
jaroslav@770
   189
            Map<String, Collection<String>> propsDeps = new HashMap<>();
jaroslav@770
   190
            if (!generateComputedProperties(body, p.properties(), e.getEnclosedElements(), propsGetSet, propsDeps)) {
jaroslav@770
   191
                ok = false;
jaroslav@770
   192
            }
jaroslav@875
   193
            if (!generateProperties(e, body, p.properties(), propsGetSet, propsDeps)) {
jaroslav@770
   194
                ok = false;
jaroslav@770
   195
            }
jaroslav@879
   196
            if (!generateFunctions(e, body, className, e.getEnclosedElements(), functions)) {
jaroslav@879
   197
                ok = false;
jaroslav@879
   198
            }
jaroslav@770
   199
            
jaroslav@770
   200
            FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + className, e);
jaroslav@770
   201
            w = new OutputStreamWriter(java.openOutputStream());
jaroslav@770
   202
            try {
jaroslav@770
   203
                w.append("package " + pkg + ";\n");
jaroslav@770
   204
                w.append("import org.apidesign.bck2brwsr.htmlpage.api.*;\n");
jaroslav@770
   205
                w.append("import org.apidesign.bck2brwsr.htmlpage.KOList;\n");
jaroslav@770
   206
                w.append("final class ").append(className).append(" {\n");
jaroslav@770
   207
                w.append("  private boolean locked;\n");
jaroslav@770
   208
                if (!initializeOnClick(className, (TypeElement) e, w, pp)) {
jaroslav@770
   209
                    ok = false;
jaroslav@770
   210
                } else {
jaroslav@923
   211
                    if (pp != null) for (String id : pp.ids()) {
jaroslav@770
   212
                        String tag = pp.tagNameForId(id);
jaroslav@770
   213
                        String type = type(tag);
jaroslav@770
   214
                        w.append("  ").append("public final ").
jaroslav@770
   215
                            append(type).append(' ').append(cnstnt(id)).append(" = new ").
jaroslav@770
   216
                            append(type).append("(\"").append(id).append("\");\n");
jaroslav@770
   217
                    }
jaroslav@770
   218
                }
jaroslav@770
   219
                w.append("  private org.apidesign.bck2brwsr.htmlpage.Knockout ko;\n");
jaroslav@770
   220
                w.append(body.toString());
jaroslav@770
   221
                if (!propsGetSet.isEmpty()) {
jaroslav@770
   222
                    w.write("public " + className + " applyBindings() {\n");
jaroslav@770
   223
                    w.write("  ko = org.apidesign.bck2brwsr.htmlpage.Knockout.applyBindings(");
jaroslav@770
   224
                    w.write(className + ".class, this, ");
jaroslav@909
   225
                    writeStringArray(propsGetSet, w);
jaroslav@909
   226
                    w.append(", ");
jaroslav@909
   227
                    writeStringArray(functions, w);
jaroslav@909
   228
                    w.write(");\n  return this;\n}\n");
jaroslav@770
   229
jaroslav@770
   230
                    w.write("public void triggerEvent(Element e, OnEvent ev) {\n");
jaroslav@770
   231
                    w.write("  org.apidesign.bck2brwsr.htmlpage.Knockout.triggerEvent(e.getId(), ev.getElementPropertyName());\n");
jaroslav@770
   232
                    w.write("}\n");
jaroslav@770
   233
                }
jaroslav@770
   234
                w.append("}\n");
jaroslav@770
   235
            } finally {
jaroslav@770
   236
                w.close();
jaroslav@770
   237
            }
jaroslav@770
   238
        } catch (IOException ex) {
jaroslav@770
   239
            processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "Can't create " + className + ".java", e);
jaroslav@770
   240
            return false;
jaroslav@770
   241
        }
jaroslav@770
   242
        return ok;
jaroslav@770
   243
    }
jaroslav@26
   244
jaroslav@871
   245
    private static String type(String tag) {
jaroslav@871
   246
        if (tag.equals("title")) {
jaroslav@871
   247
            return "Title";
jaroslav@871
   248
        }
jaroslav@871
   249
        if (tag.equals("button")) {
jaroslav@871
   250
            return "Button";
jaroslav@871
   251
        }
jaroslav@871
   252
        if (tag.equals("input")) {
jaroslav@871
   253
            return "Input";
jaroslav@871
   254
        }
jaroslav@871
   255
        if (tag.equals("canvas")) {
jaroslav@871
   256
            return "Canvas";
jaroslav@871
   257
        }
jaroslav@871
   258
        if (tag.equals("img")) {
jaroslav@871
   259
            return "Image";
jaroslav@871
   260
        }
jaroslav@871
   261
        return "Element";
jaroslav@871
   262
    }
jaroslav@871
   263
jaroslav@26
   264
    private static String cnstnt(String id) {
jaroslav@892
   265
        return id.replace('.', '_').replace('-', '_');
jaroslav@26
   266
    }
jaroslav@28
   267
jaroslav@510
   268
    private boolean initializeOnClick(
jaroslav@510
   269
        String className, TypeElement type, Writer w, ProcessPage pp
jaroslav@510
   270
    ) throws IOException {
jaroslav@765
   271
        boolean ok = true;
jaroslav@124
   272
        TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jaroslav@489
   273
        { //for (Element clazz : pe.getEnclosedElements()) {
jaroslav@489
   274
          //  if (clazz.getKind() != ElementKind.CLASS) {
jaroslav@489
   275
            //    continue;
jaroslav@489
   276
           // }
jaroslav@512
   277
            w.append("  public ").append(className).append("() {\n");
jaroslav@512
   278
            StringBuilder dispatch = new StringBuilder();
jaroslav@512
   279
            int dispatchCnt = 0;
jaroslav@489
   280
            for (Element method : type.getEnclosedElements()) {
jaroslav@435
   281
                On oc = method.getAnnotation(On.class);
jaroslav@28
   282
                if (oc != null) {
jaroslav@124
   283
                    for (String id : oc.id()) {
jaroslav@765
   284
                        if (pp == null) {
jaroslav@765
   285
                            processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "id = " + id + " not found in HTML page.");
jaroslav@765
   286
                            ok = false;
jaroslav@765
   287
                            continue;
jaroslav@765
   288
                        }
jaroslav@124
   289
                        if (pp.tagNameForId(id) == null) {
jaroslav@435
   290
                            processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "id = " + id + " does not exist in the HTML page. Found only " + pp.ids(), method);
jaroslav@765
   291
                            ok = false;
jaroslav@765
   292
                            continue;
jaroslav@124
   293
                        }
jaroslav@124
   294
                        ExecutableElement ee = (ExecutableElement)method;
jaroslav@879
   295
                        CharSequence params = wrapParams(ee, id, className, "ev", null);
jaroslav@124
   296
                        if (!ee.getModifiers().contains(Modifier.STATIC)) {
jaroslav@435
   297
                            processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "@On method has to be static", ee);
jaroslav@765
   298
                            ok = false;
jaroslav@765
   299
                            continue;
jaroslav@124
   300
                        }
jaroslav@124
   301
                        if (ee.getModifiers().contains(Modifier.PRIVATE)) {
jaroslav@435
   302
                            processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "@On method can't be private", ee);
jaroslav@765
   303
                            ok = false;
jaroslav@765
   304
                            continue;
jaroslav@124
   305
                        }
jaroslav@435
   306
                        w.append("  OnEvent." + oc.event()).append(".of(").append(cnstnt(id)).
jaroslav@512
   307
                            append(").perform(new OnDispatch(" + dispatchCnt + "));\n");
jaroslav@512
   308
jaroslav@512
   309
                        dispatch.
jaroslav@512
   310
                            append("      case ").append(dispatchCnt).append(": ").
jaroslav@512
   311
                            append(type.getSimpleName().toString()).
jaroslav@512
   312
                            append('.').append(ee.getSimpleName()).append("(").
jaroslav@512
   313
                            append(params).
jaroslav@512
   314
                            append("); break;\n");
jaroslav@512
   315
                        
jaroslav@512
   316
                        dispatchCnt++;
jaroslav@512
   317
                    }
jaroslav@28
   318
                }
jaroslav@28
   319
            }
jaroslav@512
   320
            w.append("  }\n");
jaroslav@512
   321
            if (dispatchCnt > 0) {
jaroslav@813
   322
                w.append("class OnDispatch implements OnHandler {\n");
jaroslav@512
   323
                w.append("  private final int dispatch;\n");
jaroslav@512
   324
                w.append("  OnDispatch(int d) { dispatch = d; }\n");
jaroslav@813
   325
                w.append("  public void onEvent(Object ev) {\n");
jaroslav@512
   326
                w.append("    switch (dispatch) {\n");
jaroslav@512
   327
                w.append(dispatch);
jaroslav@512
   328
                w.append("    }\n");
jaroslav@512
   329
                w.append("  }\n");
jaroslav@512
   330
                w.append("}\n");
jaroslav@512
   331
            }
jaroslav@512
   332
            
jaroslav@512
   333
jaroslav@28
   334
        }
jaroslav@765
   335
        return ok;
jaroslav@28
   336
    }
jaroslav@117
   337
jaroslav@117
   338
    @Override
jaroslav@117
   339
    public Iterable<? extends Completion> getCompletions(
jaroslav@117
   340
        Element element, AnnotationMirror annotation, 
jaroslav@117
   341
        ExecutableElement member, String userText
jaroslav@117
   342
    ) {
jaroslav@117
   343
        if (!userText.startsWith("\"")) {
jaroslav@117
   344
            return Collections.emptyList();
jaroslav@117
   345
        }
jaroslav@117
   346
        
jaroslav@117
   347
        Element cls = findClass(element);
jaroslav@117
   348
        Page p = cls.getAnnotation(Page.class);
jaroslav@770
   349
        String pkg = findPkgName(cls);
jaroslav@117
   350
        ProcessPage pp;
jaroslav@117
   351
        try {
jaroslav@117
   352
            InputStream is = openStream(pkg, p.xhtml());
jaroslav@117
   353
            pp = ProcessPage.readPage(is);
jaroslav@117
   354
            is.close();
jaroslav@117
   355
        } catch (IOException iOException) {
jaroslav@117
   356
            return Collections.emptyList();
jaroslav@117
   357
        }
jaroslav@117
   358
        
jaroslav@770
   359
        List<Completion> cc = new ArrayList<>();
jaroslav@117
   360
        userText = userText.substring(1);
jaroslav@117
   361
        for (String id : pp.ids()) {
jaroslav@117
   362
            if (id.startsWith(userText)) {
jaroslav@117
   363
                cc.add(Completions.of("\"" + id + "\"", id));
jaroslav@117
   364
            }
jaroslav@117
   365
        }
jaroslav@117
   366
        return cc;
jaroslav@117
   367
    }
jaroslav@117
   368
    
jaroslav@117
   369
    private static Element findClass(Element e) {
jaroslav@117
   370
        if (e == null) {
jaroslav@117
   371
            return null;
jaroslav@117
   372
        }
jaroslav@117
   373
        Page p = e.getAnnotation(Page.class);
jaroslav@117
   374
        if (p != null) {
jaroslav@117
   375
            return e;
jaroslav@117
   376
        }
jaroslav@117
   377
        return e.getEnclosingElement();
jaroslav@117
   378
    }
jaroslav@490
   379
jaroslav@770
   380
    private boolean generateProperties(
jaroslav@875
   381
        Element where,
jaroslav@770
   382
        Writer w, Property[] properties,
jaroslav@770
   383
        Collection<String> props, Map<String,Collection<String>> deps
jaroslav@492
   384
    ) throws IOException {
jaroslav@770
   385
        boolean ok = true;
jaroslav@490
   386
        for (Property p : properties) {
jaroslav@770
   387
            final String tn;
jaroslav@875
   388
            tn = typeName(where, p);
jaroslav@760
   389
            String[] gs = toGetSet(p.name(), tn, p.array());
jaroslav@490
   390
jaroslav@760
   391
            if (p.array()) {
jaroslav@761
   392
                w.write("private KOList<" + tn + "> prop_" + p.name() + " = new KOList<" + tn + ">(\""
jaroslav@761
   393
                    + p.name() + "\"");
jaroslav@761
   394
                final Collection<String> dependants = deps.get(p.name());
jaroslav@761
   395
                if (dependants != null) {
jaroslav@761
   396
                    for (String depProp : dependants) {
jaroslav@761
   397
                        w.write(", ");
jaroslav@761
   398
                        w.write('\"');
jaroslav@761
   399
                        w.write(depProp);
jaroslav@761
   400
                        w.write('\"');
jaroslav@761
   401
                    }
jaroslav@498
   402
                }
jaroslav@761
   403
                w.write(");\n");
jaroslav@760
   404
                w.write("public java.util.List<" + tn + "> " + gs[0] + "() {\n");
jaroslav@760
   405
                w.write("  if (locked) throw new IllegalStateException();\n");
jaroslav@761
   406
                w.write("  prop_" + p.name() + ".assign(ko);\n");
jaroslav@760
   407
                w.write("  return prop_" + p.name() + ";\n");
jaroslav@760
   408
                w.write("}\n");
jaroslav@760
   409
            } else {
jaroslav@760
   410
                w.write("private " + tn + " prop_" + p.name() + ";\n");
jaroslav@760
   411
                w.write("public " + tn + " " + gs[0] + "() {\n");
jaroslav@760
   412
                w.write("  if (locked) throw new IllegalStateException();\n");
jaroslav@760
   413
                w.write("  return prop_" + p.name() + ";\n");
jaroslav@760
   414
                w.write("}\n");
jaroslav@760
   415
                w.write("public void " + gs[1] + "(" + tn + " v) {\n");
jaroslav@760
   416
                w.write("  if (locked) throw new IllegalStateException();\n");
jaroslav@760
   417
                w.write("  prop_" + p.name() + " = v;\n");
jaroslav@760
   418
                w.write("  if (ko != null) {\n");
jaroslav@760
   419
                w.write("    ko.valueHasMutated(\"" + p.name() + "\");\n");
jaroslav@760
   420
                final Collection<String> dependants = deps.get(p.name());
jaroslav@760
   421
                if (dependants != null) {
jaroslav@760
   422
                    for (String depProp : dependants) {
jaroslav@760
   423
                        w.write("    ko.valueHasMutated(\"" + depProp + "\");\n");
jaroslav@760
   424
                    }
jaroslav@543
   425
                }
jaroslav@760
   426
                w.write("  }\n");
jaroslav@760
   427
                w.write("}\n");
jaroslav@498
   428
            }
jaroslav@492
   429
            
jaroslav@492
   430
            props.add(p.name());
jaroslav@492
   431
            props.add(gs[2]);
jaroslav@492
   432
            props.add(gs[3]);
jaroslav@530
   433
            props.add(gs[0]);
jaroslav@490
   434
        }
jaroslav@770
   435
        return ok;
jaroslav@490
   436
    }
jaroslav@490
   437
jaroslav@498
   438
    private boolean generateComputedProperties(
jaroslav@765
   439
        Writer w, Property[] fixedProps,
jaroslav@765
   440
        Collection<? extends Element> arr, Collection<String> props,
jaroslav@498
   441
        Map<String,Collection<String>> deps
jaroslav@492
   442
    ) throws IOException {
jaroslav@765
   443
        boolean ok = true;
jaroslav@491
   444
        for (Element e : arr) {
jaroslav@491
   445
            if (e.getKind() != ElementKind.METHOD) {
jaroslav@491
   446
                continue;
jaroslav@491
   447
            }
jaroslav@491
   448
            if (e.getAnnotation(ComputedProperty.class) == null) {
jaroslav@491
   449
                continue;
jaroslav@491
   450
            }
jaroslav@491
   451
            ExecutableElement ee = (ExecutableElement)e;
jaroslav@767
   452
            final TypeMirror rt = ee.getReturnType();
jaroslav@768
   453
            final Types tu = processingEnv.getTypeUtils();
jaroslav@768
   454
            TypeMirror ert = tu.erasure(rt);
jaroslav@929
   455
            String tn = fqn(ert, ee);
jaroslav@768
   456
            boolean array = false;
jaroslav@768
   457
            if (tn.equals("java.util.List")) {
jaroslav@768
   458
                array = true;
jaroslav@768
   459
            }
jaroslav@768
   460
            
jaroslav@498
   461
            final String sn = ee.getSimpleName().toString();
jaroslav@768
   462
            String[] gs = toGetSet(sn, tn, array);
jaroslav@498
   463
            
jaroslav@505
   464
            w.write("public " + tn + " " + gs[0] + "() {\n");
jaroslav@500
   465
            w.write("  if (locked) throw new IllegalStateException();\n");
jaroslav@500
   466
            int arg = 0;
jaroslav@491
   467
            for (VariableElement pe : ee.getParameters()) {
jaroslav@498
   468
                final String dn = pe.getSimpleName().toString();
jaroslav@765
   469
                
jaroslav@765
   470
                if (!verifyPropName(pe, dn, fixedProps)) {
jaroslav@765
   471
                    ok = false;
jaroslav@765
   472
                }
jaroslav@765
   473
                
jaroslav@929
   474
                final String dt = fqn(pe.asType(), ee);
jaroslav@760
   475
                String[] call = toGetSet(dn, dt, false);
jaroslav@500
   476
                w.write("  " + dt + " arg" + (++arg) + " = ");
jaroslav@500
   477
                w.write(call[0] + "();\n");
jaroslav@498
   478
                
jaroslav@498
   479
                Collection<String> depends = deps.get(dn);
jaroslav@498
   480
                if (depends == null) {
jaroslav@770
   481
                    depends = new LinkedHashSet<>();
jaroslav@498
   482
                    deps.put(dn, depends);
jaroslav@498
   483
                }
jaroslav@498
   484
                depends.add(sn);
jaroslav@491
   485
            }
jaroslav@500
   486
            w.write("  try {\n");
jaroslav@500
   487
            w.write("    locked = true;\n");
jaroslav@500
   488
            w.write("    return " + e.getEnclosingElement().getSimpleName() + '.' + e.getSimpleName() + "(");
jaroslav@500
   489
            String sep = "";
jaroslav@500
   490
            for (int i = 1; i <= arg; i++) {
jaroslav@500
   491
                w.write(sep);
jaroslav@500
   492
                w.write("arg" + i);
jaroslav@500
   493
                sep = ", ";
jaroslav@500
   494
            }
jaroslav@491
   495
            w.write(");\n");
jaroslav@500
   496
            w.write("  } finally {\n");
jaroslav@500
   497
            w.write("    locked = false;\n");
jaroslav@500
   498
            w.write("  }\n");
jaroslav@491
   499
            w.write("}\n");
jaroslav@768
   500
jaroslav@492
   501
            props.add(e.getSimpleName().toString());
jaroslav@492
   502
            props.add(gs[2]);
jaroslav@492
   503
            props.add(null);
jaroslav@530
   504
            props.add(gs[0]);
jaroslav@491
   505
        }
jaroslav@498
   506
        
jaroslav@765
   507
        return ok;
jaroslav@491
   508
    }
jaroslav@491
   509
jaroslav@760
   510
    private static String[] toGetSet(String name, String type, boolean array) {
jaroslav@491
   511
        String n = Character.toUpperCase(name.charAt(0)) + name.substring(1);
jaroslav@492
   512
        String bck2brwsrType = "L" + type.replace('.', '_') + "_2";
jaroslav@492
   513
        if ("int".equals(type)) {
jaroslav@492
   514
            bck2brwsrType = "I";
jaroslav@492
   515
        }
jaroslav@492
   516
        if ("double".equals(type)) {
jaroslav@492
   517
            bck2brwsrType = "D";
jaroslav@492
   518
        }
jaroslav@492
   519
        String pref = "get";
jaroslav@492
   520
        if ("boolean".equals(type)) {
jaroslav@492
   521
            pref = "is";
jaroslav@492
   522
            bck2brwsrType = "Z";
jaroslav@492
   523
        }
jaroslav@498
   524
        final String nu = n.replace('.', '_');
jaroslav@760
   525
        if (array) {
jaroslav@760
   526
            return new String[] { 
jaroslav@760
   527
                "get" + n,
jaroslav@760
   528
                null,
jaroslav@887
   529
                "get" + nu + "__Ljava_util_List_2",
jaroslav@760
   530
                null
jaroslav@760
   531
            };
jaroslav@760
   532
        }
jaroslav@492
   533
        return new String[]{
jaroslav@492
   534
            pref + n, 
jaroslav@492
   535
            "set" + n, 
jaroslav@498
   536
            pref + nu + "__" + bck2brwsrType,
jaroslav@498
   537
            "set" + nu + "__V" + bck2brwsrType
jaroslav@492
   538
        };
jaroslav@490
   539
    }
jaroslav@490
   540
jaroslav@875
   541
    private String typeName(Element where, Property p) {
jaroslav@764
   542
        String ret;
jaroslav@770
   543
        boolean isModel = false;
jaroslav@911
   544
        boolean isEnum = false;
jaroslav@490
   545
        try {
jaroslav@764
   546
            ret = p.type().getName();
jaroslav@490
   547
        } catch (MirroredTypeException ex) {
jaroslav@765
   548
            TypeMirror tm = processingEnv.getTypeUtils().erasure(ex.getTypeMirror());
jaroslav@770
   549
            final Element e = processingEnv.getTypeUtils().asElement(tm);
jaroslav@770
   550
            final Model m = e == null ? null : e.getAnnotation(Model.class);
jaroslav@770
   551
            if (m != null) {
jaroslav@770
   552
                ret = findPkgName(e) + '.' + m.className();
jaroslav@770
   553
                isModel = true;
jaroslav@906
   554
                models.put(e, m.className());
jaroslav@770
   555
            } else {
jaroslav@770
   556
                ret = tm.toString();
jaroslav@770
   557
            }
jaroslav@911
   558
            TypeMirror enm = processingEnv.getElementUtils().getTypeElement("java.lang.Enum").asType();
jaroslav@911
   559
            enm = processingEnv.getTypeUtils().erasure(enm);
jaroslav@911
   560
            isEnum = processingEnv.getTypeUtils().isSubtype(tm, enm);
jaroslav@543
   561
        }
jaroslav@764
   562
        if (p.array()) {
jaroslav@769
   563
            String bt = findBoxedType(ret);
jaroslav@769
   564
            if (bt != null) {
jaroslav@769
   565
                return bt;
jaroslav@764
   566
            }
jaroslav@764
   567
        }
jaroslav@911
   568
        if (!isModel && !"java.lang.String".equals(ret) && !isEnum) {
jaroslav@770
   569
            String bt = findBoxedType(ret);
jaroslav@770
   570
            if (bt == null) {
jaroslav@875
   571
                processingEnv.getMessager().printMessage(
jaroslav@875
   572
                    Diagnostic.Kind.ERROR, 
jaroslav@875
   573
                    "Only primitive types supported in the mapping. Not " + ret,
jaroslav@875
   574
                    where
jaroslav@875
   575
                );
jaroslav@770
   576
            }
jaroslav@769
   577
        }
jaroslav@764
   578
        return ret;
jaroslav@543
   579
    }
jaroslav@769
   580
    
jaroslav@769
   581
    private static String findBoxedType(String ret) {
jaroslav@876
   582
        if (ret.equals("boolean")) {
jaroslav@876
   583
            return Boolean.class.getName();
jaroslav@876
   584
        }
jaroslav@769
   585
        if (ret.equals("byte")) {
jaroslav@769
   586
            return Byte.class.getName();
jaroslav@769
   587
        }
jaroslav@769
   588
        if (ret.equals("short")) {
jaroslav@769
   589
            return Short.class.getName();
jaroslav@769
   590
        }
jaroslav@769
   591
        if (ret.equals("char")) {
jaroslav@769
   592
            return Character.class.getName();
jaroslav@769
   593
        }
jaroslav@769
   594
        if (ret.equals("int")) {
jaroslav@769
   595
            return Integer.class.getName();
jaroslav@769
   596
        }
jaroslav@769
   597
        if (ret.equals("long")) {
jaroslav@769
   598
            return Long.class.getName();
jaroslav@769
   599
        }
jaroslav@769
   600
        if (ret.equals("float")) {
jaroslav@769
   601
            return Float.class.getName();
jaroslav@769
   602
        }
jaroslav@769
   603
        if (ret.equals("double")) {
jaroslav@769
   604
            return Double.class.getName();
jaroslav@769
   605
        }
jaroslav@769
   606
        return null;
jaroslav@769
   607
    }
jaroslav@765
   608
jaroslav@765
   609
    private boolean verifyPropName(Element e, String propName, Property[] existingProps) {
jaroslav@765
   610
        StringBuilder sb = new StringBuilder();
jaroslav@765
   611
        String sep = "";
jaroslav@765
   612
        for (Property property : existingProps) {
jaroslav@765
   613
            if (property.name().equals(propName)) {
jaroslav@765
   614
                return true;
jaroslav@765
   615
            }
jaroslav@765
   616
            sb.append(sep);
jaroslav@765
   617
            sb.append('"');
jaroslav@765
   618
            sb.append(property.name());
jaroslav@765
   619
            sb.append('"');
jaroslav@765
   620
            sep = ", ";
jaroslav@765
   621
        }
jaroslav@765
   622
        processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR,
jaroslav@765
   623
            propName + " is not one of known properties: " + sb
jaroslav@765
   624
            , e
jaroslav@765
   625
        );
jaroslav@765
   626
        return false;
jaroslav@765
   627
    }
jaroslav@770
   628
jaroslav@770
   629
    private static String findPkgName(Element e) {
jaroslav@770
   630
        for (;;) {
jaroslav@770
   631
            if (e.getKind() == ElementKind.PACKAGE) {
jaroslav@770
   632
                return ((PackageElement)e).getQualifiedName().toString();
jaroslav@770
   633
            }
jaroslav@770
   634
            e = e.getEnclosingElement();
jaroslav@490
   635
        }
jaroslav@490
   636
    }
jaroslav@879
   637
jaroslav@879
   638
    private boolean generateFunctions(
jaroslav@879
   639
        Element clazz, StringWriter body, String className, 
jaroslav@879
   640
        List<? extends Element> enclosedElements, List<String> functions
jaroslav@879
   641
    ) {
jaroslav@879
   642
        for (Element m : enclosedElements) {
jaroslav@879
   643
            if (m.getKind() != ElementKind.METHOD) {
jaroslav@879
   644
                continue;
jaroslav@879
   645
            }
jaroslav@879
   646
            ExecutableElement e = (ExecutableElement)m;
jaroslav@879
   647
            OnFunction onF = e.getAnnotation(OnFunction.class);
jaroslav@879
   648
            if (onF == null) {
jaroslav@879
   649
                continue;
jaroslav@879
   650
            }
jaroslav@879
   651
            if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@879
   652
                processingEnv.getMessager().printMessage(
jaroslav@879
   653
                    Diagnostic.Kind.ERROR, "@OnFunction method needs to be static", e
jaroslav@879
   654
                );
jaroslav@879
   655
                return false;
jaroslav@879
   656
            }
jaroslav@879
   657
            if (e.getModifiers().contains(Modifier.PRIVATE)) {
jaroslav@879
   658
                processingEnv.getMessager().printMessage(
jaroslav@879
   659
                    Diagnostic.Kind.ERROR, "@OnFunction method cannot be private", e
jaroslav@879
   660
                );
jaroslav@879
   661
                return false;
jaroslav@879
   662
            }
jaroslav@879
   663
            if (e.getReturnType().getKind() != TypeKind.VOID) {
jaroslav@879
   664
                processingEnv.getMessager().printMessage(
jaroslav@879
   665
                    Diagnostic.Kind.ERROR, "@OnFunction method should return void", e
jaroslav@879
   666
                );
jaroslav@879
   667
                return false;
jaroslav@879
   668
            }
jaroslav@879
   669
            String n = e.getSimpleName().toString();
jaroslav@927
   670
            body.append("private void ").append(n).append("(Object data, Object ev) {\n");
jaroslav@879
   671
            body.append("  ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@879
   672
            body.append(wrapParams(e, null, className, "ev", "data"));
jaroslav@879
   673
            body.append(");\n");
jaroslav@879
   674
            body.append("}\n");
jaroslav@879
   675
            
jaroslav@909
   676
            functions.add(n);
jaroslav@909
   677
            functions.add(n + "__VLjava_lang_Object_2Ljava_lang_Object_2");
jaroslav@879
   678
        }
jaroslav@879
   679
        return true;
jaroslav@879
   680
    }
jaroslav@879
   681
jaroslav@879
   682
    private CharSequence wrapParams(
jaroslav@879
   683
        ExecutableElement ee, String id, String className, String evName, String dataName
jaroslav@879
   684
    ) {
jaroslav@879
   685
        TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jaroslav@879
   686
        StringBuilder params = new StringBuilder();
jaroslav@879
   687
        boolean first = true;
jaroslav@879
   688
        for (VariableElement ve : ee.getParameters()) {
jaroslav@879
   689
            if (!first) {
jaroslav@879
   690
                params.append(", ");
jaroslav@879
   691
            }
jaroslav@879
   692
            first = false;
jaroslav@879
   693
            String toCall = null;
jaroslav@879
   694
            if (ve.asType() == stringType) {
jaroslav@879
   695
                if (ve.getSimpleName().contentEquals("id")) {
jaroslav@879
   696
                    params.append('"').append(id).append('"');
jaroslav@879
   697
                    continue;
jaroslav@879
   698
                }
jaroslav@906
   699
                toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toString(";
jaroslav@879
   700
            }
jaroslav@879
   701
            if (ve.asType().getKind() == TypeKind.DOUBLE) {
jaroslav@906
   702
                toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toDouble(";
jaroslav@879
   703
            }
jaroslav@879
   704
            if (ve.asType().getKind() == TypeKind.INT) {
jaroslav@906
   705
                toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toInt(";
jaroslav@906
   706
            }
jaroslav@906
   707
            if (dataName != null && ve.getSimpleName().contentEquals(dataName) && isModel(ve.asType())) {
jaroslav@906
   708
                toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toModel(" + ve.asType() + ".class, ";
jaroslav@879
   709
            }
jaroslav@879
   710
jaroslav@879
   711
            if (toCall != null) {
jaroslav@906
   712
                params.append(toCall);
jaroslav@906
   713
                if (dataName != null && ve.getSimpleName().contentEquals(dataName)) {
jaroslav@879
   714
                    params.append(dataName);
jaroslav@879
   715
                    params.append(", null");
jaroslav@879
   716
                } else {
jaroslav@879
   717
                    params.append(evName);
jaroslav@879
   718
                    params.append(", \"");
jaroslav@879
   719
                    params.append(ve.getSimpleName().toString());
jaroslav@879
   720
                    params.append("\"");
jaroslav@879
   721
                }
jaroslav@879
   722
                params.append(")");
jaroslav@879
   723
                continue;
jaroslav@879
   724
            }
jaroslav@929
   725
            String rn = fqn(ve.asType(), ee);
jaroslav@879
   726
            int last = rn.lastIndexOf('.');
jaroslav@879
   727
            if (last >= 0) {
jaroslav@879
   728
                rn = rn.substring(last + 1);
jaroslav@879
   729
            }
jaroslav@879
   730
            if (rn.equals(className)) {
jaroslav@879
   731
                params.append(className).append(".this");
jaroslav@879
   732
                continue;
jaroslav@879
   733
            }
jaroslav@879
   734
            processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, 
jaroslav@879
   735
                "@On method can only accept String named 'id' or " + className + " arguments",
jaroslav@879
   736
                ee
jaroslav@879
   737
            );
jaroslav@879
   738
        }
jaroslav@879
   739
        return params;
jaroslav@879
   740
    }
jaroslav@906
   741
    
jaroslav@906
   742
    private boolean isModel(TypeMirror tm) {
jaroslav@906
   743
        final Element e = processingEnv.getTypeUtils().asElement(tm);
jaroslav@907
   744
        if (e == null) {
jaroslav@907
   745
            return false;
jaroslav@907
   746
        }
jaroslav@906
   747
        for (Element ch : e.getEnclosedElements()) {
jaroslav@906
   748
            if (ch.getKind() == ElementKind.METHOD) {
jaroslav@906
   749
                ExecutableElement ee = (ExecutableElement)ch;
jaroslav@906
   750
                if (ee.getParameters().isEmpty() && ee.getSimpleName().contentEquals("modelFor")) {
jaroslav@906
   751
                    return true;
jaroslav@906
   752
                }
jaroslav@906
   753
            }
jaroslav@906
   754
        }
jaroslav@906
   755
        return models.values().contains(e.getSimpleName().toString());
jaroslav@906
   756
    }
jaroslav@909
   757
jaroslav@909
   758
    private void writeStringArray(List<String> strings, Writer w) throws IOException {
jaroslav@909
   759
        w.write("new String[] {\n");
jaroslav@909
   760
        String sep = "";
jaroslav@909
   761
        for (String n : strings) {
jaroslav@909
   762
            w.write(sep);
jaroslav@909
   763
            if (n == null) {
jaroslav@909
   764
                w.write("    null");
jaroslav@909
   765
            } else {
jaroslav@909
   766
                w.write("    \"" + n + "\"");
jaroslav@909
   767
            }
jaroslav@909
   768
            sep = ",\n";
jaroslav@909
   769
        }
jaroslav@909
   770
        w.write("\n  }");
jaroslav@909
   771
    }
jaroslav@919
   772
    
jaroslav@919
   773
    private void writeToString(Property[] props, Writer w) throws IOException {
jaroslav@919
   774
        w.write("  public String toString() {\n");
jaroslav@919
   775
        w.write("    StringBuilder sb = new StringBuilder();\n");
jaroslav@919
   776
        w.write("    sb.append('{');\n");
jaroslav@919
   777
        String sep = "";
jaroslav@919
   778
        for (Property p : props) {
jaroslav@919
   779
            w.write(sep);
jaroslav@919
   780
            w.append("    sb.append(\"" + p.name() + ": \");\n");
jaroslav@920
   781
            w.append("    sb.append(org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toJSON(prop_");
jaroslav@920
   782
            w.append(p.name()).append("));\n");
jaroslav@919
   783
            sep =    "    sb.append(',');\n";
jaroslav@919
   784
        }
jaroslav@919
   785
        w.write("    sb.append('}');\n");
jaroslav@919
   786
        w.write("    return sb.toString();\n");
jaroslav@919
   787
        w.write("  }\n");
jaroslav@919
   788
    }
jaroslav@925
   789
jaroslav@925
   790
    private String inPckName(Element e) {
jaroslav@925
   791
        StringBuilder sb = new StringBuilder();
jaroslav@925
   792
        while (e.getKind() != ElementKind.PACKAGE) {
jaroslav@925
   793
            if (sb.length() == 0) {
jaroslav@925
   794
                sb.append(e.getSimpleName());
jaroslav@925
   795
            } else {
jaroslav@925
   796
                sb.insert(0, '.');
jaroslav@925
   797
                sb.insert(0, e.getSimpleName());
jaroslav@925
   798
            }
jaroslav@925
   799
            e = e.getEnclosingElement();
jaroslav@925
   800
        }
jaroslav@925
   801
        return sb.toString();
jaroslav@925
   802
    }
jaroslav@929
   803
jaroslav@929
   804
    private String fqn(TypeMirror pt, Element relative) {
jaroslav@929
   805
        if (pt.getKind() == TypeKind.ERROR) {
jaroslav@929
   806
            final Elements eu = processingEnv.getElementUtils();
jaroslav@929
   807
            PackageElement pckg = eu.getPackageOf(relative);
jaroslav@929
   808
            return pckg.getQualifiedName() + "." + pt.toString();
jaroslav@929
   809
        }
jaroslav@929
   810
        return pt.toString();
jaroslav@929
   811
    }
jaroslav@26
   812
}