javaquery/api/src/main/java/org/apidesign/bck2brwsr/htmlpage/PageProcessor.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Sun, 07 Apr 2013 19:29:55 +0200
branchmodel
changeset 943 28aae214c202
parent 941 d1e482f73507
child 944 1e2b0dcc8326
permissions -rw-r--r--
Can parse JSON array with objects in it
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@934
    37
import javax.annotation.processing.Messager;
jaroslav@26
    38
import javax.annotation.processing.Processor;
jaroslav@26
    39
import javax.annotation.processing.RoundEnvironment;
jaroslav@26
    40
import javax.annotation.processing.SupportedAnnotationTypes;
jaroslav@117
    41
import javax.lang.model.element.AnnotationMirror;
jaroslav@26
    42
import javax.lang.model.element.Element;
jaroslav@28
    43
import javax.lang.model.element.ElementKind;
jaroslav@28
    44
import javax.lang.model.element.ExecutableElement;
jaroslav@28
    45
import javax.lang.model.element.Modifier;
jaroslav@26
    46
import javax.lang.model.element.PackageElement;
jaroslav@26
    47
import javax.lang.model.element.TypeElement;
jaroslav@491
    48
import javax.lang.model.element.VariableElement;
jaroslav@936
    49
import javax.lang.model.type.ArrayType;
jaroslav@936
    50
import javax.lang.model.type.DeclaredType;
jaroslav@490
    51
import javax.lang.model.type.MirroredTypeException;
jaroslav@813
    52
import javax.lang.model.type.TypeKind;
jaroslav@124
    53
import javax.lang.model.type.TypeMirror;
jaroslav@929
    54
import javax.lang.model.util.Elements;
jaroslav@768
    55
import javax.lang.model.util.Types;
jaroslav@26
    56
import javax.tools.Diagnostic;
jaroslav@26
    57
import javax.tools.FileObject;
jaroslav@26
    58
import javax.tools.StandardLocation;
jaroslav@491
    59
import org.apidesign.bck2brwsr.htmlpage.api.ComputedProperty;
jaroslav@770
    60
import org.apidesign.bck2brwsr.htmlpage.api.Model;
jaroslav@435
    61
import org.apidesign.bck2brwsr.htmlpage.api.On;
jaroslav@879
    62
import org.apidesign.bck2brwsr.htmlpage.api.OnFunction;
jaroslav@934
    63
import org.apidesign.bck2brwsr.htmlpage.api.OnReceive;
jaroslav@26
    64
import org.apidesign.bck2brwsr.htmlpage.api.Page;
jaroslav@490
    65
import org.apidesign.bck2brwsr.htmlpage.api.Property;
jaroslav@26
    66
import org.openide.util.lookup.ServiceProvider;
jaroslav@26
    67
jaroslav@26
    68
/** Annotation processor to process an XHTML page and generate appropriate 
jaroslav@26
    69
 * "id" file.
jaroslav@26
    70
 *
jaroslav@26
    71
 * @author Jaroslav Tulach <jtulach@netbeans.org>
jaroslav@26
    72
 */
jaroslav@26
    73
@ServiceProvider(service=Processor.class)
jaroslav@28
    74
@SupportedAnnotationTypes({
jaroslav@770
    75
    "org.apidesign.bck2brwsr.htmlpage.api.Model",
jaroslav@28
    76
    "org.apidesign.bck2brwsr.htmlpage.api.Page",
jaroslav@879
    77
    "org.apidesign.bck2brwsr.htmlpage.api.OnFunction",
jaroslav@934
    78
    "org.apidesign.bck2brwsr.htmlpage.api.OnReceive",
jaroslav@435
    79
    "org.apidesign.bck2brwsr.htmlpage.api.On"
jaroslav@28
    80
})
jaroslav@26
    81
public final class PageProcessor extends AbstractProcessor {
jaroslav@906
    82
    private final Map<Element,String> models = new WeakHashMap<>();
jaroslav@26
    83
    @Override
jaroslav@26
    84
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
jaroslav@765
    85
        boolean ok = true;
jaroslav@770
    86
        for (Element e : roundEnv.getElementsAnnotatedWith(Model.class)) {
jaroslav@770
    87
            if (!processModel(e)) {
jaroslav@770
    88
                ok = false;
jaroslav@26
    89
            }
jaroslav@26
    90
        }
jaroslav@543
    91
        for (Element e : roundEnv.getElementsAnnotatedWith(Page.class)) {
jaroslav@770
    92
            if (!processPage(e)) {
jaroslav@765
    93
                ok = false;
jaroslav@543
    94
            }
jaroslav@543
    95
        }
jaroslav@906
    96
        if (roundEnv.processingOver()) {
jaroslav@906
    97
            models.clear();
jaroslav@906
    98
        }
jaroslav@765
    99
        return ok;
jaroslav@26
   100
    }
jaroslav@26
   101
jaroslav@26
   102
    private InputStream openStream(String pkg, String name) throws IOException {
jaroslav@26
   103
        try {
jaroslav@107
   104
            FileObject fo = processingEnv.getFiler().getResource(
jaroslav@107
   105
                StandardLocation.SOURCE_PATH, pkg, name);
jaroslav@26
   106
            return fo.openInputStream();
jaroslav@26
   107
        } catch (IOException ex) {
jaroslav@26
   108
            return processingEnv.getFiler().getResource(StandardLocation.CLASS_OUTPUT, pkg, name).openInputStream();
jaroslav@26
   109
        }
jaroslav@26
   110
    }
jaroslav@934
   111
jaroslav@934
   112
    private  Messager err() {
jaroslav@934
   113
        return processingEnv.getMessager();
jaroslav@934
   114
    }
jaroslav@770
   115
    
jaroslav@770
   116
    private boolean processModel(Element e) {
jaroslav@770
   117
        boolean ok = true;
jaroslav@770
   118
        Model m = e.getAnnotation(Model.class);
jaroslav@770
   119
        if (m == null) {
jaroslav@770
   120
            return true;
jaroslav@770
   121
        }
jaroslav@770
   122
        String pkg = findPkgName(e);
jaroslav@770
   123
        Writer w;
jaroslav@770
   124
        String className = m.className();
jaroslav@941
   125
        models.put(e, className);
jaroslav@770
   126
        try {
jaroslav@770
   127
            StringWriter body = new StringWriter();
jaroslav@770
   128
            List<String> propsGetSet = new ArrayList<>();
jaroslav@914
   129
            List<String> functions = new ArrayList<>();
jaroslav@770
   130
            Map<String, Collection<String>> propsDeps = new HashMap<>();
jaroslav@770
   131
            if (!generateComputedProperties(body, m.properties(), e.getEnclosedElements(), propsGetSet, propsDeps)) {
jaroslav@770
   132
                ok = false;
jaroslav@770
   133
            }
jaroslav@875
   134
            if (!generateProperties(e, body, m.properties(), propsGetSet, propsDeps)) {
jaroslav@770
   135
                ok = false;
jaroslav@770
   136
            }
jaroslav@914
   137
            if (!generateFunctions(e, body, className, e.getEnclosedElements(), functions)) {
jaroslav@914
   138
                ok = false;
jaroslav@914
   139
            }
jaroslav@770
   140
            FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + className, e);
jaroslav@770
   141
            w = new OutputStreamWriter(java.openOutputStream());
jaroslav@770
   142
            try {
jaroslav@770
   143
                w.append("package " + pkg + ";\n");
jaroslav@770
   144
                w.append("import org.apidesign.bck2brwsr.htmlpage.api.*;\n");
jaroslav@770
   145
                w.append("import org.apidesign.bck2brwsr.htmlpage.KOList;\n");
jaroslav@908
   146
                w.append("import org.apidesign.bck2brwsr.core.JavaScriptOnly;\n");
jaroslav@930
   147
                w.append("final class ").append(className).append(" implements Cloneable {\n");
jaroslav@770
   148
                w.append("  private boolean locked;\n");
jaroslav@770
   149
                w.append("  private org.apidesign.bck2brwsr.htmlpage.Knockout ko;\n");
jaroslav@770
   150
                w.append(body.toString());
jaroslav@925
   151
                w.append("  private static Class<" + inPckName(e) + "> modelFor() { return null; }\n");
jaroslav@909
   152
                w.append("  public ").append(className).append("() {\n");
jaroslav@934
   153
                w.append("    intKnckt();\n");
jaroslav@934
   154
                w.append("  };\n");
jaroslav@934
   155
                w.append("  private void intKnckt() {\n");
jaroslav@909
   156
                w.append("    ko = org.apidesign.bck2brwsr.htmlpage.Knockout.applyBindings(this, ");
jaroslav@909
   157
                writeStringArray(propsGetSet, w);
jaroslav@909
   158
                w.append(", ");
jaroslav@914
   159
                writeStringArray(functions, w);
jaroslav@909
   160
                w.append("    );\n");
jaroslav@909
   161
                w.append("  };\n");
jaroslav@934
   162
                w.append("  ").append(className).append("(Object json) {\n");
jaroslav@934
   163
                int values = 0;
jaroslav@934
   164
                for (int i = 0; i < propsGetSet.size(); i += 4) {
jaroslav@943
   165
                    Property p = findProperty(m.properties(), propsGetSet.get(i));
jaroslav@943
   166
                    if (p == null) {
jaroslav@934
   167
                        continue;
jaroslav@934
   168
                    }
jaroslav@934
   169
                    values++;
jaroslav@934
   170
                }
jaroslav@934
   171
                w.append("    Object[] ret = new Object[" + values + "];\n");
jaroslav@934
   172
                w.append("    org.apidesign.bck2brwsr.htmlpage.ConvertTypes.extractJSON(json, new String[] {\n");
jaroslav@934
   173
                for (int i = 0; i < propsGetSet.size(); i += 4) {
jaroslav@943
   174
                    Property p = findProperty(m.properties(), propsGetSet.get(i));
jaroslav@943
   175
                    if (p == null) {
jaroslav@934
   176
                        continue;
jaroslav@934
   177
                    }
jaroslav@934
   178
                    w.append("      \"").append(propsGetSet.get(i)).append("\",\n");
jaroslav@934
   179
                }
jaroslav@934
   180
                w.append("    }, ret);\n");
jaroslav@934
   181
                for (int i = 0, cnt = 0, prop = 0; i < propsGetSet.size(); i += 4) {
jaroslav@943
   182
                    final String pn = propsGetSet.get(i);
jaroslav@943
   183
                    Property p = findProperty(m.properties(), pn);
jaroslav@943
   184
                    if (p == null) {
jaroslav@934
   185
                        continue;
jaroslav@934
   186
                    }
jaroslav@934
   187
                    boolean[] isModel = { false };
jaroslav@934
   188
                    boolean[] isEnum = { false };
jaroslav@934
   189
                    String type = checkType(m.properties()[prop++], isModel, isEnum);
jaroslav@934
   190
                    if (isEnum[0]) {
jaroslav@934
   191
//                        w.append(type).append(".valueOf((String)");
jaroslav@934
   192
//                        close = true;
jaroslav@943
   193
                        w.append("    this.prop_").append(pn);
jaroslav@943
   194
                        w.append(" = null;\n");
jaroslav@943
   195
                    } else if (p.array()) {
jaroslav@943
   196
                        w.append("if (ret[" + cnt + "] instanceof Object[]) {\n");
jaroslav@943
   197
                        w.append("  for (Object e : ((Object[])ret[" + cnt + "])) {\n");
jaroslav@943
   198
                        if (isModel[0]) {
jaroslav@943
   199
                            w.append("    this.prop_").append(pn).append(".add(new ");
jaroslav@943
   200
                            w.append(type).append("(e));\n");
jaroslav@943
   201
                        } else {
jaroslav@943
   202
                            w.append("    this.prop_").append(pn).append(".add((");
jaroslav@943
   203
                            w.append(type).append(")e);\n");
jaroslav@943
   204
                        }
jaroslav@943
   205
                        w.append("  }\n");
jaroslav@943
   206
                        w.append("}\n");
jaroslav@934
   207
                    } else {
jaroslav@943
   208
                        w.append("    this.prop_").append(pn);
jaroslav@943
   209
                        w.append(" = (").append(type).append(')');
jaroslav@943
   210
                        w.append("ret[" + cnt + "];\n");
jaroslav@934
   211
                    }
jaroslav@943
   212
                    cnt++;
jaroslav@934
   213
                }
jaroslav@934
   214
                w.append("    intKnckt();\n");
jaroslav@934
   215
                w.append("  };\n");
jaroslav@919
   216
                writeToString(m.properties(), w);
jaroslav@930
   217
                writeClone(className, m.properties(), w);
jaroslav@770
   218
                w.append("}\n");
jaroslav@770
   219
            } finally {
jaroslav@770
   220
                w.close();
jaroslav@770
   221
            }
jaroslav@770
   222
        } catch (IOException ex) {
jaroslav@934
   223
            err().printMessage(Diagnostic.Kind.ERROR, "Can't create " + className + ".java", e);
jaroslav@770
   224
            return false;
jaroslav@770
   225
        }
jaroslav@770
   226
        return ok;
jaroslav@770
   227
    }
jaroslav@770
   228
    
jaroslav@770
   229
    private boolean processPage(Element e) {
jaroslav@770
   230
        boolean ok = true;
jaroslav@770
   231
        Page p = e.getAnnotation(Page.class);
jaroslav@770
   232
        if (p == null) {
jaroslav@770
   233
            return true;
jaroslav@770
   234
        }
jaroslav@770
   235
        String pkg = findPkgName(e);
jaroslav@770
   236
jaroslav@770
   237
        ProcessPage pp;
jaroslav@770
   238
        try (InputStream is = openStream(pkg, p.xhtml())) {
jaroslav@770
   239
            pp = ProcessPage.readPage(is);
jaroslav@770
   240
            is.close();
jaroslav@770
   241
        } catch (IOException iOException) {
jaroslav@934
   242
            err().printMessage(Diagnostic.Kind.ERROR, "Can't read " + p.xhtml() + " as " + iOException.getMessage(), e);
jaroslav@770
   243
            ok = false;
jaroslav@770
   244
            pp = null;
jaroslav@770
   245
        }
jaroslav@770
   246
        Writer w;
jaroslav@770
   247
        String className = p.className();
jaroslav@770
   248
        if (className.isEmpty()) {
jaroslav@770
   249
            int indx = p.xhtml().indexOf('.');
jaroslav@770
   250
            className = p.xhtml().substring(0, indx);
jaroslav@770
   251
        }
jaroslav@770
   252
        try {
jaroslav@770
   253
            StringWriter body = new StringWriter();
jaroslav@770
   254
            List<String> propsGetSet = new ArrayList<>();
jaroslav@879
   255
            List<String> functions = new ArrayList<>();
jaroslav@770
   256
            Map<String, Collection<String>> propsDeps = new HashMap<>();
jaroslav@770
   257
            if (!generateComputedProperties(body, p.properties(), e.getEnclosedElements(), propsGetSet, propsDeps)) {
jaroslav@770
   258
                ok = false;
jaroslav@770
   259
            }
jaroslav@875
   260
            if (!generateProperties(e, body, p.properties(), propsGetSet, propsDeps)) {
jaroslav@770
   261
                ok = false;
jaroslav@770
   262
            }
jaroslav@879
   263
            if (!generateFunctions(e, body, className, e.getEnclosedElements(), functions)) {
jaroslav@879
   264
                ok = false;
jaroslav@879
   265
            }
jaroslav@934
   266
            if (!generateReceive(e, body, className, e.getEnclosedElements(), functions)) {
jaroslav@934
   267
                ok = false;
jaroslav@934
   268
            }
jaroslav@770
   269
            
jaroslav@770
   270
            FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + className, e);
jaroslav@770
   271
            w = new OutputStreamWriter(java.openOutputStream());
jaroslav@770
   272
            try {
jaroslav@770
   273
                w.append("package " + pkg + ";\n");
jaroslav@770
   274
                w.append("import org.apidesign.bck2brwsr.htmlpage.api.*;\n");
jaroslav@770
   275
                w.append("import org.apidesign.bck2brwsr.htmlpage.KOList;\n");
jaroslav@770
   276
                w.append("final class ").append(className).append(" {\n");
jaroslav@770
   277
                w.append("  private boolean locked;\n");
jaroslav@770
   278
                if (!initializeOnClick(className, (TypeElement) e, w, pp)) {
jaroslav@770
   279
                    ok = false;
jaroslav@770
   280
                } else {
jaroslav@923
   281
                    if (pp != null) for (String id : pp.ids()) {
jaroslav@770
   282
                        String tag = pp.tagNameForId(id);
jaroslav@770
   283
                        String type = type(tag);
jaroslav@770
   284
                        w.append("  ").append("public final ").
jaroslav@770
   285
                            append(type).append(' ').append(cnstnt(id)).append(" = new ").
jaroslav@770
   286
                            append(type).append("(\"").append(id).append("\");\n");
jaroslav@770
   287
                    }
jaroslav@770
   288
                }
jaroslav@770
   289
                w.append("  private org.apidesign.bck2brwsr.htmlpage.Knockout ko;\n");
jaroslav@770
   290
                w.append(body.toString());
jaroslav@770
   291
                if (!propsGetSet.isEmpty()) {
jaroslav@770
   292
                    w.write("public " + className + " applyBindings() {\n");
jaroslav@770
   293
                    w.write("  ko = org.apidesign.bck2brwsr.htmlpage.Knockout.applyBindings(");
jaroslav@770
   294
                    w.write(className + ".class, this, ");
jaroslav@909
   295
                    writeStringArray(propsGetSet, w);
jaroslav@909
   296
                    w.append(", ");
jaroslav@909
   297
                    writeStringArray(functions, w);
jaroslav@909
   298
                    w.write(");\n  return this;\n}\n");
jaroslav@770
   299
jaroslav@770
   300
                    w.write("public void triggerEvent(Element e, OnEvent ev) {\n");
jaroslav@770
   301
                    w.write("  org.apidesign.bck2brwsr.htmlpage.Knockout.triggerEvent(e.getId(), ev.getElementPropertyName());\n");
jaroslav@770
   302
                    w.write("}\n");
jaroslav@770
   303
                }
jaroslav@770
   304
                w.append("}\n");
jaroslav@770
   305
            } finally {
jaroslav@770
   306
                w.close();
jaroslav@770
   307
            }
jaroslav@770
   308
        } catch (IOException ex) {
jaroslav@934
   309
            err().printMessage(Diagnostic.Kind.ERROR, "Can't create " + className + ".java", e);
jaroslav@770
   310
            return false;
jaroslav@770
   311
        }
jaroslav@770
   312
        return ok;
jaroslav@770
   313
    }
jaroslav@26
   314
jaroslav@871
   315
    private static String type(String tag) {
jaroslav@871
   316
        if (tag.equals("title")) {
jaroslav@871
   317
            return "Title";
jaroslav@871
   318
        }
jaroslav@871
   319
        if (tag.equals("button")) {
jaroslav@871
   320
            return "Button";
jaroslav@871
   321
        }
jaroslav@871
   322
        if (tag.equals("input")) {
jaroslav@871
   323
            return "Input";
jaroslav@871
   324
        }
jaroslav@871
   325
        if (tag.equals("canvas")) {
jaroslav@871
   326
            return "Canvas";
jaroslav@871
   327
        }
jaroslav@871
   328
        if (tag.equals("img")) {
jaroslav@871
   329
            return "Image";
jaroslav@871
   330
        }
jaroslav@871
   331
        return "Element";
jaroslav@871
   332
    }
jaroslav@871
   333
jaroslav@26
   334
    private static String cnstnt(String id) {
jaroslav@892
   335
        return id.replace('.', '_').replace('-', '_');
jaroslav@26
   336
    }
jaroslav@28
   337
jaroslav@510
   338
    private boolean initializeOnClick(
jaroslav@510
   339
        String className, TypeElement type, Writer w, ProcessPage pp
jaroslav@510
   340
    ) throws IOException {
jaroslav@765
   341
        boolean ok = true;
jaroslav@124
   342
        TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jaroslav@489
   343
        { //for (Element clazz : pe.getEnclosedElements()) {
jaroslav@489
   344
          //  if (clazz.getKind() != ElementKind.CLASS) {
jaroslav@489
   345
            //    continue;
jaroslav@489
   346
           // }
jaroslav@512
   347
            w.append("  public ").append(className).append("() {\n");
jaroslav@512
   348
            StringBuilder dispatch = new StringBuilder();
jaroslav@512
   349
            int dispatchCnt = 0;
jaroslav@489
   350
            for (Element method : type.getEnclosedElements()) {
jaroslav@435
   351
                On oc = method.getAnnotation(On.class);
jaroslav@28
   352
                if (oc != null) {
jaroslav@124
   353
                    for (String id : oc.id()) {
jaroslav@765
   354
                        if (pp == null) {
jaroslav@934
   355
                            err().printMessage(Diagnostic.Kind.ERROR, "id = " + id + " not found in HTML page.");
jaroslav@765
   356
                            ok = false;
jaroslav@765
   357
                            continue;
jaroslav@765
   358
                        }
jaroslav@124
   359
                        if (pp.tagNameForId(id) == null) {
jaroslav@934
   360
                            err().printMessage(Diagnostic.Kind.ERROR, "id = " + id + " does not exist in the HTML page. Found only " + pp.ids(), method);
jaroslav@765
   361
                            ok = false;
jaroslav@765
   362
                            continue;
jaroslav@124
   363
                        }
jaroslav@124
   364
                        ExecutableElement ee = (ExecutableElement)method;
jaroslav@879
   365
                        CharSequence params = wrapParams(ee, id, className, "ev", null);
jaroslav@124
   366
                        if (!ee.getModifiers().contains(Modifier.STATIC)) {
jaroslav@934
   367
                            err().printMessage(Diagnostic.Kind.ERROR, "@On method has to be static", ee);
jaroslav@765
   368
                            ok = false;
jaroslav@765
   369
                            continue;
jaroslav@124
   370
                        }
jaroslav@124
   371
                        if (ee.getModifiers().contains(Modifier.PRIVATE)) {
jaroslav@934
   372
                            err().printMessage(Diagnostic.Kind.ERROR, "@On method can't be private", ee);
jaroslav@765
   373
                            ok = false;
jaroslav@765
   374
                            continue;
jaroslav@124
   375
                        }
jaroslav@435
   376
                        w.append("  OnEvent." + oc.event()).append(".of(").append(cnstnt(id)).
jaroslav@512
   377
                            append(").perform(new OnDispatch(" + dispatchCnt + "));\n");
jaroslav@512
   378
jaroslav@512
   379
                        dispatch.
jaroslav@512
   380
                            append("      case ").append(dispatchCnt).append(": ").
jaroslav@512
   381
                            append(type.getSimpleName().toString()).
jaroslav@512
   382
                            append('.').append(ee.getSimpleName()).append("(").
jaroslav@512
   383
                            append(params).
jaroslav@512
   384
                            append("); break;\n");
jaroslav@512
   385
                        
jaroslav@512
   386
                        dispatchCnt++;
jaroslav@512
   387
                    }
jaroslav@28
   388
                }
jaroslav@28
   389
            }
jaroslav@512
   390
            w.append("  }\n");
jaroslav@512
   391
            if (dispatchCnt > 0) {
jaroslav@813
   392
                w.append("class OnDispatch implements OnHandler {\n");
jaroslav@512
   393
                w.append("  private final int dispatch;\n");
jaroslav@512
   394
                w.append("  OnDispatch(int d) { dispatch = d; }\n");
jaroslav@813
   395
                w.append("  public void onEvent(Object ev) {\n");
jaroslav@512
   396
                w.append("    switch (dispatch) {\n");
jaroslav@512
   397
                w.append(dispatch);
jaroslav@512
   398
                w.append("    }\n");
jaroslav@512
   399
                w.append("  }\n");
jaroslav@512
   400
                w.append("}\n");
jaroslav@512
   401
            }
jaroslav@512
   402
            
jaroslav@512
   403
jaroslav@28
   404
        }
jaroslav@765
   405
        return ok;
jaroslav@28
   406
    }
jaroslav@117
   407
jaroslav@117
   408
    @Override
jaroslav@117
   409
    public Iterable<? extends Completion> getCompletions(
jaroslav@117
   410
        Element element, AnnotationMirror annotation, 
jaroslav@117
   411
        ExecutableElement member, String userText
jaroslav@117
   412
    ) {
jaroslav@117
   413
        if (!userText.startsWith("\"")) {
jaroslav@117
   414
            return Collections.emptyList();
jaroslav@117
   415
        }
jaroslav@117
   416
        
jaroslav@117
   417
        Element cls = findClass(element);
jaroslav@117
   418
        Page p = cls.getAnnotation(Page.class);
jaroslav@770
   419
        String pkg = findPkgName(cls);
jaroslav@117
   420
        ProcessPage pp;
jaroslav@117
   421
        try {
jaroslav@117
   422
            InputStream is = openStream(pkg, p.xhtml());
jaroslav@117
   423
            pp = ProcessPage.readPage(is);
jaroslav@117
   424
            is.close();
jaroslav@117
   425
        } catch (IOException iOException) {
jaroslav@117
   426
            return Collections.emptyList();
jaroslav@117
   427
        }
jaroslav@117
   428
        
jaroslav@770
   429
        List<Completion> cc = new ArrayList<>();
jaroslav@117
   430
        userText = userText.substring(1);
jaroslav@117
   431
        for (String id : pp.ids()) {
jaroslav@117
   432
            if (id.startsWith(userText)) {
jaroslav@117
   433
                cc.add(Completions.of("\"" + id + "\"", id));
jaroslav@117
   434
            }
jaroslav@117
   435
        }
jaroslav@117
   436
        return cc;
jaroslav@117
   437
    }
jaroslav@117
   438
    
jaroslav@117
   439
    private static Element findClass(Element e) {
jaroslav@117
   440
        if (e == null) {
jaroslav@117
   441
            return null;
jaroslav@117
   442
        }
jaroslav@117
   443
        Page p = e.getAnnotation(Page.class);
jaroslav@117
   444
        if (p != null) {
jaroslav@117
   445
            return e;
jaroslav@117
   446
        }
jaroslav@117
   447
        return e.getEnclosingElement();
jaroslav@117
   448
    }
jaroslav@490
   449
jaroslav@770
   450
    private boolean generateProperties(
jaroslav@875
   451
        Element where,
jaroslav@770
   452
        Writer w, Property[] properties,
jaroslav@770
   453
        Collection<String> props, Map<String,Collection<String>> deps
jaroslav@492
   454
    ) throws IOException {
jaroslav@770
   455
        boolean ok = true;
jaroslav@490
   456
        for (Property p : properties) {
jaroslav@770
   457
            final String tn;
jaroslav@875
   458
            tn = typeName(where, p);
jaroslav@760
   459
            String[] gs = toGetSet(p.name(), tn, p.array());
jaroslav@490
   460
jaroslav@760
   461
            if (p.array()) {
jaroslav@761
   462
                w.write("private KOList<" + tn + "> prop_" + p.name() + " = new KOList<" + tn + ">(\""
jaroslav@761
   463
                    + p.name() + "\"");
jaroslav@761
   464
                final Collection<String> dependants = deps.get(p.name());
jaroslav@761
   465
                if (dependants != null) {
jaroslav@761
   466
                    for (String depProp : dependants) {
jaroslav@761
   467
                        w.write(", ");
jaroslav@761
   468
                        w.write('\"');
jaroslav@761
   469
                        w.write(depProp);
jaroslav@761
   470
                        w.write('\"');
jaroslav@761
   471
                    }
jaroslav@498
   472
                }
jaroslav@761
   473
                w.write(");\n");
jaroslav@760
   474
                w.write("public java.util.List<" + tn + "> " + gs[0] + "() {\n");
jaroslav@760
   475
                w.write("  if (locked) throw new IllegalStateException();\n");
jaroslav@761
   476
                w.write("  prop_" + p.name() + ".assign(ko);\n");
jaroslav@760
   477
                w.write("  return prop_" + p.name() + ";\n");
jaroslav@760
   478
                w.write("}\n");
jaroslav@760
   479
            } else {
jaroslav@760
   480
                w.write("private " + tn + " prop_" + p.name() + ";\n");
jaroslav@760
   481
                w.write("public " + tn + " " + gs[0] + "() {\n");
jaroslav@760
   482
                w.write("  if (locked) throw new IllegalStateException();\n");
jaroslav@760
   483
                w.write("  return prop_" + p.name() + ";\n");
jaroslav@760
   484
                w.write("}\n");
jaroslav@760
   485
                w.write("public void " + gs[1] + "(" + tn + " v) {\n");
jaroslav@760
   486
                w.write("  if (locked) throw new IllegalStateException();\n");
jaroslav@760
   487
                w.write("  prop_" + p.name() + " = v;\n");
jaroslav@760
   488
                w.write("  if (ko != null) {\n");
jaroslav@760
   489
                w.write("    ko.valueHasMutated(\"" + p.name() + "\");\n");
jaroslav@760
   490
                final Collection<String> dependants = deps.get(p.name());
jaroslav@760
   491
                if (dependants != null) {
jaroslav@760
   492
                    for (String depProp : dependants) {
jaroslav@760
   493
                        w.write("    ko.valueHasMutated(\"" + depProp + "\");\n");
jaroslav@760
   494
                    }
jaroslav@543
   495
                }
jaroslav@760
   496
                w.write("  }\n");
jaroslav@760
   497
                w.write("}\n");
jaroslav@498
   498
            }
jaroslav@492
   499
            
jaroslav@492
   500
            props.add(p.name());
jaroslav@492
   501
            props.add(gs[2]);
jaroslav@492
   502
            props.add(gs[3]);
jaroslav@530
   503
            props.add(gs[0]);
jaroslav@490
   504
        }
jaroslav@770
   505
        return ok;
jaroslav@490
   506
    }
jaroslav@490
   507
jaroslav@498
   508
    private boolean generateComputedProperties(
jaroslav@765
   509
        Writer w, Property[] fixedProps,
jaroslav@765
   510
        Collection<? extends Element> arr, Collection<String> props,
jaroslav@498
   511
        Map<String,Collection<String>> deps
jaroslav@492
   512
    ) throws IOException {
jaroslav@765
   513
        boolean ok = true;
jaroslav@491
   514
        for (Element e : arr) {
jaroslav@491
   515
            if (e.getKind() != ElementKind.METHOD) {
jaroslav@491
   516
                continue;
jaroslav@491
   517
            }
jaroslav@491
   518
            if (e.getAnnotation(ComputedProperty.class) == null) {
jaroslav@491
   519
                continue;
jaroslav@491
   520
            }
jaroslav@491
   521
            ExecutableElement ee = (ExecutableElement)e;
jaroslav@767
   522
            final TypeMirror rt = ee.getReturnType();
jaroslav@768
   523
            final Types tu = processingEnv.getTypeUtils();
jaroslav@768
   524
            TypeMirror ert = tu.erasure(rt);
jaroslav@929
   525
            String tn = fqn(ert, ee);
jaroslav@768
   526
            boolean array = false;
jaroslav@768
   527
            if (tn.equals("java.util.List")) {
jaroslav@768
   528
                array = true;
jaroslav@768
   529
            }
jaroslav@768
   530
            
jaroslav@498
   531
            final String sn = ee.getSimpleName().toString();
jaroslav@768
   532
            String[] gs = toGetSet(sn, tn, array);
jaroslav@498
   533
            
jaroslav@505
   534
            w.write("public " + tn + " " + gs[0] + "() {\n");
jaroslav@500
   535
            w.write("  if (locked) throw new IllegalStateException();\n");
jaroslav@500
   536
            int arg = 0;
jaroslav@491
   537
            for (VariableElement pe : ee.getParameters()) {
jaroslav@498
   538
                final String dn = pe.getSimpleName().toString();
jaroslav@765
   539
                
jaroslav@765
   540
                if (!verifyPropName(pe, dn, fixedProps)) {
jaroslav@765
   541
                    ok = false;
jaroslav@765
   542
                }
jaroslav@765
   543
                
jaroslav@929
   544
                final String dt = fqn(pe.asType(), ee);
jaroslav@760
   545
                String[] call = toGetSet(dn, dt, false);
jaroslav@500
   546
                w.write("  " + dt + " arg" + (++arg) + " = ");
jaroslav@500
   547
                w.write(call[0] + "();\n");
jaroslav@498
   548
                
jaroslav@498
   549
                Collection<String> depends = deps.get(dn);
jaroslav@498
   550
                if (depends == null) {
jaroslav@770
   551
                    depends = new LinkedHashSet<>();
jaroslav@498
   552
                    deps.put(dn, depends);
jaroslav@498
   553
                }
jaroslav@498
   554
                depends.add(sn);
jaroslav@491
   555
            }
jaroslav@500
   556
            w.write("  try {\n");
jaroslav@500
   557
            w.write("    locked = true;\n");
jaroslav@500
   558
            w.write("    return " + e.getEnclosingElement().getSimpleName() + '.' + e.getSimpleName() + "(");
jaroslav@500
   559
            String sep = "";
jaroslav@500
   560
            for (int i = 1; i <= arg; i++) {
jaroslav@500
   561
                w.write(sep);
jaroslav@500
   562
                w.write("arg" + i);
jaroslav@500
   563
                sep = ", ";
jaroslav@500
   564
            }
jaroslav@491
   565
            w.write(");\n");
jaroslav@500
   566
            w.write("  } finally {\n");
jaroslav@500
   567
            w.write("    locked = false;\n");
jaroslav@500
   568
            w.write("  }\n");
jaroslav@491
   569
            w.write("}\n");
jaroslav@768
   570
jaroslav@492
   571
            props.add(e.getSimpleName().toString());
jaroslav@492
   572
            props.add(gs[2]);
jaroslav@492
   573
            props.add(null);
jaroslav@530
   574
            props.add(gs[0]);
jaroslav@491
   575
        }
jaroslav@498
   576
        
jaroslav@765
   577
        return ok;
jaroslav@491
   578
    }
jaroslav@491
   579
jaroslav@760
   580
    private static String[] toGetSet(String name, String type, boolean array) {
jaroslav@491
   581
        String n = Character.toUpperCase(name.charAt(0)) + name.substring(1);
jaroslav@492
   582
        String bck2brwsrType = "L" + type.replace('.', '_') + "_2";
jaroslav@492
   583
        if ("int".equals(type)) {
jaroslav@492
   584
            bck2brwsrType = "I";
jaroslav@492
   585
        }
jaroslav@492
   586
        if ("double".equals(type)) {
jaroslav@492
   587
            bck2brwsrType = "D";
jaroslav@492
   588
        }
jaroslav@492
   589
        String pref = "get";
jaroslav@492
   590
        if ("boolean".equals(type)) {
jaroslav@492
   591
            pref = "is";
jaroslav@492
   592
            bck2brwsrType = "Z";
jaroslav@492
   593
        }
jaroslav@498
   594
        final String nu = n.replace('.', '_');
jaroslav@760
   595
        if (array) {
jaroslav@760
   596
            return new String[] { 
jaroslav@760
   597
                "get" + n,
jaroslav@760
   598
                null,
jaroslav@887
   599
                "get" + nu + "__Ljava_util_List_2",
jaroslav@760
   600
                null
jaroslav@760
   601
            };
jaroslav@760
   602
        }
jaroslav@492
   603
        return new String[]{
jaroslav@492
   604
            pref + n, 
jaroslav@492
   605
            "set" + n, 
jaroslav@498
   606
            pref + nu + "__" + bck2brwsrType,
jaroslav@498
   607
            "set" + nu + "__V" + bck2brwsrType
jaroslav@492
   608
        };
jaroslav@490
   609
    }
jaroslav@490
   610
jaroslav@875
   611
    private String typeName(Element where, Property p) {
jaroslav@764
   612
        String ret;
jaroslav@930
   613
        boolean[] isModel = { false };
jaroslav@930
   614
        boolean[] isEnum = { false };
jaroslav@930
   615
        ret = checkType(p, isModel, isEnum);
jaroslav@764
   616
        if (p.array()) {
jaroslav@769
   617
            String bt = findBoxedType(ret);
jaroslav@769
   618
            if (bt != null) {
jaroslav@769
   619
                return bt;
jaroslav@764
   620
            }
jaroslav@764
   621
        }
jaroslav@930
   622
        if (!isModel[0] && !"java.lang.String".equals(ret) && !isEnum[0]) {
jaroslav@770
   623
            String bt = findBoxedType(ret);
jaroslav@770
   624
            if (bt == null) {
jaroslav@934
   625
                err().printMessage(
jaroslav@875
   626
                    Diagnostic.Kind.ERROR, 
jaroslav@875
   627
                    "Only primitive types supported in the mapping. Not " + ret,
jaroslav@875
   628
                    where
jaroslav@875
   629
                );
jaroslav@770
   630
            }
jaroslav@769
   631
        }
jaroslav@764
   632
        return ret;
jaroslav@543
   633
    }
jaroslav@769
   634
    
jaroslav@769
   635
    private static String findBoxedType(String ret) {
jaroslav@876
   636
        if (ret.equals("boolean")) {
jaroslav@876
   637
            return Boolean.class.getName();
jaroslav@876
   638
        }
jaroslav@769
   639
        if (ret.equals("byte")) {
jaroslav@769
   640
            return Byte.class.getName();
jaroslav@769
   641
        }
jaroslav@769
   642
        if (ret.equals("short")) {
jaroslav@769
   643
            return Short.class.getName();
jaroslav@769
   644
        }
jaroslav@769
   645
        if (ret.equals("char")) {
jaroslav@769
   646
            return Character.class.getName();
jaroslav@769
   647
        }
jaroslav@769
   648
        if (ret.equals("int")) {
jaroslav@769
   649
            return Integer.class.getName();
jaroslav@769
   650
        }
jaroslav@769
   651
        if (ret.equals("long")) {
jaroslav@769
   652
            return Long.class.getName();
jaroslav@769
   653
        }
jaroslav@769
   654
        if (ret.equals("float")) {
jaroslav@769
   655
            return Float.class.getName();
jaroslav@769
   656
        }
jaroslav@769
   657
        if (ret.equals("double")) {
jaroslav@769
   658
            return Double.class.getName();
jaroslav@769
   659
        }
jaroslav@769
   660
        return null;
jaroslav@769
   661
    }
jaroslav@765
   662
jaroslav@765
   663
    private boolean verifyPropName(Element e, String propName, Property[] existingProps) {
jaroslav@765
   664
        StringBuilder sb = new StringBuilder();
jaroslav@765
   665
        String sep = "";
jaroslav@765
   666
        for (Property property : existingProps) {
jaroslav@765
   667
            if (property.name().equals(propName)) {
jaroslav@765
   668
                return true;
jaroslav@765
   669
            }
jaroslav@765
   670
            sb.append(sep);
jaroslav@765
   671
            sb.append('"');
jaroslav@765
   672
            sb.append(property.name());
jaroslav@765
   673
            sb.append('"');
jaroslav@765
   674
            sep = ", ";
jaroslav@765
   675
        }
jaroslav@934
   676
        err().printMessage(Diagnostic.Kind.ERROR,
jaroslav@765
   677
            propName + " is not one of known properties: " + sb
jaroslav@765
   678
            , e
jaroslav@765
   679
        );
jaroslav@765
   680
        return false;
jaroslav@765
   681
    }
jaroslav@770
   682
jaroslav@770
   683
    private static String findPkgName(Element e) {
jaroslav@770
   684
        for (;;) {
jaroslav@770
   685
            if (e.getKind() == ElementKind.PACKAGE) {
jaroslav@770
   686
                return ((PackageElement)e).getQualifiedName().toString();
jaroslav@770
   687
            }
jaroslav@770
   688
            e = e.getEnclosingElement();
jaroslav@490
   689
        }
jaroslav@490
   690
    }
jaroslav@879
   691
jaroslav@879
   692
    private boolean generateFunctions(
jaroslav@879
   693
        Element clazz, StringWriter body, String className, 
jaroslav@879
   694
        List<? extends Element> enclosedElements, List<String> functions
jaroslav@879
   695
    ) {
jaroslav@879
   696
        for (Element m : enclosedElements) {
jaroslav@879
   697
            if (m.getKind() != ElementKind.METHOD) {
jaroslav@879
   698
                continue;
jaroslav@879
   699
            }
jaroslav@879
   700
            ExecutableElement e = (ExecutableElement)m;
jaroslav@879
   701
            OnFunction onF = e.getAnnotation(OnFunction.class);
jaroslav@879
   702
            if (onF == null) {
jaroslav@879
   703
                continue;
jaroslav@879
   704
            }
jaroslav@879
   705
            if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@934
   706
                err().printMessage(
jaroslav@879
   707
                    Diagnostic.Kind.ERROR, "@OnFunction method needs to be static", e
jaroslav@879
   708
                );
jaroslav@879
   709
                return false;
jaroslav@879
   710
            }
jaroslav@879
   711
            if (e.getModifiers().contains(Modifier.PRIVATE)) {
jaroslav@934
   712
                err().printMessage(
jaroslav@879
   713
                    Diagnostic.Kind.ERROR, "@OnFunction method cannot be private", e
jaroslav@879
   714
                );
jaroslav@879
   715
                return false;
jaroslav@879
   716
            }
jaroslav@879
   717
            if (e.getReturnType().getKind() != TypeKind.VOID) {
jaroslav@934
   718
                err().printMessage(
jaroslav@879
   719
                    Diagnostic.Kind.ERROR, "@OnFunction method should return void", e
jaroslav@879
   720
                );
jaroslav@879
   721
                return false;
jaroslav@879
   722
            }
jaroslav@879
   723
            String n = e.getSimpleName().toString();
jaroslav@927
   724
            body.append("private void ").append(n).append("(Object data, Object ev) {\n");
jaroslav@879
   725
            body.append("  ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@879
   726
            body.append(wrapParams(e, null, className, "ev", "data"));
jaroslav@879
   727
            body.append(");\n");
jaroslav@879
   728
            body.append("}\n");
jaroslav@879
   729
            
jaroslav@909
   730
            functions.add(n);
jaroslav@909
   731
            functions.add(n + "__VLjava_lang_Object_2Ljava_lang_Object_2");
jaroslav@879
   732
        }
jaroslav@879
   733
        return true;
jaroslav@879
   734
    }
jaroslav@879
   735
jaroslav@934
   736
    private boolean generateReceive(
jaroslav@934
   737
        Element clazz, StringWriter body, String className, 
jaroslav@934
   738
        List<? extends Element> enclosedElements, List<String> functions
jaroslav@934
   739
    ) {
jaroslav@934
   740
        for (Element m : enclosedElements) {
jaroslav@934
   741
            if (m.getKind() != ElementKind.METHOD) {
jaroslav@934
   742
                continue;
jaroslav@934
   743
            }
jaroslav@934
   744
            ExecutableElement e = (ExecutableElement)m;
jaroslav@934
   745
            OnReceive onR = e.getAnnotation(OnReceive.class);
jaroslav@934
   746
            if (onR == null) {
jaroslav@934
   747
                continue;
jaroslav@934
   748
            }
jaroslav@934
   749
            if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@934
   750
                err().printMessage(
jaroslav@934
   751
                    Diagnostic.Kind.ERROR, "@OnReceive method needs to be static", e
jaroslav@934
   752
                );
jaroslav@934
   753
                return false;
jaroslav@934
   754
            }
jaroslav@934
   755
            if (e.getModifiers().contains(Modifier.PRIVATE)) {
jaroslav@934
   756
                err().printMessage(
jaroslav@934
   757
                    Diagnostic.Kind.ERROR, "@OnReceive method cannot be private", e
jaroslav@934
   758
                );
jaroslav@934
   759
                return false;
jaroslav@934
   760
            }
jaroslav@934
   761
            if (e.getReturnType().getKind() != TypeKind.VOID) {
jaroslav@934
   762
                err().printMessage(
jaroslav@934
   763
                    Diagnostic.Kind.ERROR, "@OnReceive method should return void", e
jaroslav@934
   764
                );
jaroslav@934
   765
                return false;
jaroslav@934
   766
            }
jaroslav@934
   767
            String modelClass = null;
jaroslav@936
   768
            boolean expectsList = false;
jaroslav@934
   769
            List<String> args = new ArrayList<>();
jaroslav@934
   770
            {
jaroslav@934
   771
                for (VariableElement ve : e.getParameters()) {
jaroslav@936
   772
                    TypeMirror modelType = null;
jaroslav@934
   773
                    if (ve.asType().toString().equals(className)) {
jaroslav@934
   774
                        args.add(className + ".this");
jaroslav@934
   775
                    } else if (isModel(ve.asType())) {
jaroslav@936
   776
                        modelType = ve.asType();
jaroslav@936
   777
                    } else if (ve.asType().getKind() == TypeKind.ARRAY) {
jaroslav@936
   778
                        modelType = ((ArrayType)ve.asType()).getComponentType();
jaroslav@936
   779
                        expectsList = true;
jaroslav@936
   780
                    }
jaroslav@936
   781
                    if (modelType != null) {
jaroslav@934
   782
                        if (modelClass != null) {
jaroslav@934
   783
                            err().printMessage(Diagnostic.Kind.ERROR, "There can be only one model class among arguments", e);
jaroslav@934
   784
                        } else {
jaroslav@936
   785
                            modelClass = modelType.toString();
jaroslav@936
   786
                            if (expectsList) {
jaroslav@936
   787
                                args.add("arr");
jaroslav@936
   788
                            } else {
jaroslav@936
   789
                                args.add("arr[0]");
jaroslav@936
   790
                            }
jaroslav@934
   791
                        }
jaroslav@934
   792
                    }
jaroslav@934
   793
                }
jaroslav@934
   794
            }
jaroslav@936
   795
            if (modelClass == null) {
jaroslav@936
   796
                err().printMessage(Diagnostic.Kind.ERROR, "The method needs to have one @Model class as parameter", e);
jaroslav@936
   797
            }
jaroslav@934
   798
            String n = e.getSimpleName().toString();
jaroslav@934
   799
            body.append("public void ").append(n).append("(");
jaroslav@934
   800
            StringBuilder assembleURL = new StringBuilder();
jaroslav@934
   801
            {
jaroslav@934
   802
                String sep = "";
jaroslav@934
   803
                for (String p : findParamNames(e, onR.url(), assembleURL)) {
jaroslav@934
   804
                    body.append(sep);
jaroslav@934
   805
                    body.append("String ").append(p);
jaroslav@934
   806
                    sep = ", ";
jaroslav@934
   807
                }
jaroslav@934
   808
            }
jaroslav@934
   809
            body.append(") {\n");
jaroslav@934
   810
            body.append("  final Object[] result = { null };\n");
jaroslav@934
   811
            body.append(
jaroslav@934
   812
                "  class ProcessResult implements Runnable {\n" +
jaroslav@934
   813
                "    @Override\n" +
jaroslav@934
   814
                "    public void run() {\n" +
jaroslav@936
   815
                "      Object value = result[0];\n");
jaroslav@936
   816
            body.append(
jaroslav@936
   817
                "      " + modelClass + "[] arr;\n");
jaroslav@936
   818
            body.append(
jaroslav@934
   819
                "      if (value instanceof Object[]) {\n" +
jaroslav@936
   820
                "        Object[] data = ((Object[])value);\n" +
jaroslav@936
   821
                "        arr = new " + modelClass + "[data.length];\n" +
jaroslav@936
   822
                "        for (int i = 0; i < data.length; i++) {\n" +
jaroslav@941
   823
                "          arr[i] = new " + modelClass + "(data[i]);\n" +
jaroslav@936
   824
                "        }\n" +
jaroslav@936
   825
                "      } else {\n" +
jaroslav@936
   826
                "        arr = new " + modelClass + "[1];\n" +
jaroslav@936
   827
                "        arr[0] = new " + modelClass + "(value);\n" +
jaroslav@936
   828
                "      }\n"
jaroslav@936
   829
            );
jaroslav@934
   830
            {
jaroslav@934
   831
                body.append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@934
   832
                String sep = "";
jaroslav@934
   833
                for (String arg : args) {
jaroslav@934
   834
                    body.append(sep);
jaroslav@934
   835
                    body.append(arg);
jaroslav@934
   836
                    sep = ", ";
jaroslav@934
   837
                }
jaroslav@934
   838
                body.append(");\n");
jaroslav@934
   839
            }
jaroslav@934
   840
            body.append(
jaroslav@934
   841
                "    }\n" +
jaroslav@934
   842
                "  }\n"
jaroslav@934
   843
            );
jaroslav@934
   844
            body.append("  org.apidesign.bck2brwsr.htmlpage.ConvertTypes.loadJSON(\n      ");
jaroslav@934
   845
            body.append(assembleURL);
jaroslav@934
   846
            body.append(", result, new ProcessResult()\n  );\n");
jaroslav@934
   847
//            body.append("  ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@934
   848
//            body.append(wrapParams(e, null, className, "ev", "data"));
jaroslav@934
   849
//            body.append(");\n");
jaroslav@934
   850
            body.append("}\n");
jaroslav@934
   851
        }
jaroslav@934
   852
        return true;
jaroslav@934
   853
    }
jaroslav@934
   854
jaroslav@879
   855
    private CharSequence wrapParams(
jaroslav@879
   856
        ExecutableElement ee, String id, String className, String evName, String dataName
jaroslav@879
   857
    ) {
jaroslav@879
   858
        TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jaroslav@879
   859
        StringBuilder params = new StringBuilder();
jaroslav@879
   860
        boolean first = true;
jaroslav@879
   861
        for (VariableElement ve : ee.getParameters()) {
jaroslav@879
   862
            if (!first) {
jaroslav@879
   863
                params.append(", ");
jaroslav@879
   864
            }
jaroslav@879
   865
            first = false;
jaroslav@879
   866
            String toCall = null;
jaroslav@879
   867
            if (ve.asType() == stringType) {
jaroslav@879
   868
                if (ve.getSimpleName().contentEquals("id")) {
jaroslav@879
   869
                    params.append('"').append(id).append('"');
jaroslav@879
   870
                    continue;
jaroslav@879
   871
                }
jaroslav@906
   872
                toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toString(";
jaroslav@879
   873
            }
jaroslav@879
   874
            if (ve.asType().getKind() == TypeKind.DOUBLE) {
jaroslav@906
   875
                toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toDouble(";
jaroslav@879
   876
            }
jaroslav@879
   877
            if (ve.asType().getKind() == TypeKind.INT) {
jaroslav@906
   878
                toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toInt(";
jaroslav@906
   879
            }
jaroslav@906
   880
            if (dataName != null && ve.getSimpleName().contentEquals(dataName) && isModel(ve.asType())) {
jaroslav@906
   881
                toCall = "org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toModel(" + ve.asType() + ".class, ";
jaroslav@879
   882
            }
jaroslav@879
   883
jaroslav@879
   884
            if (toCall != null) {
jaroslav@906
   885
                params.append(toCall);
jaroslav@906
   886
                if (dataName != null && ve.getSimpleName().contentEquals(dataName)) {
jaroslav@879
   887
                    params.append(dataName);
jaroslav@879
   888
                    params.append(", null");
jaroslav@879
   889
                } else {
jaroslav@879
   890
                    params.append(evName);
jaroslav@879
   891
                    params.append(", \"");
jaroslav@879
   892
                    params.append(ve.getSimpleName().toString());
jaroslav@879
   893
                    params.append("\"");
jaroslav@879
   894
                }
jaroslav@879
   895
                params.append(")");
jaroslav@879
   896
                continue;
jaroslav@879
   897
            }
jaroslav@929
   898
            String rn = fqn(ve.asType(), ee);
jaroslav@879
   899
            int last = rn.lastIndexOf('.');
jaroslav@879
   900
            if (last >= 0) {
jaroslav@879
   901
                rn = rn.substring(last + 1);
jaroslav@879
   902
            }
jaroslav@879
   903
            if (rn.equals(className)) {
jaroslav@879
   904
                params.append(className).append(".this");
jaroslav@879
   905
                continue;
jaroslav@879
   906
            }
jaroslav@934
   907
            err().printMessage(Diagnostic.Kind.ERROR, 
jaroslav@879
   908
                "@On method can only accept String named 'id' or " + className + " arguments",
jaroslav@879
   909
                ee
jaroslav@879
   910
            );
jaroslav@879
   911
        }
jaroslav@879
   912
        return params;
jaroslav@879
   913
    }
jaroslav@906
   914
    
jaroslav@906
   915
    private boolean isModel(TypeMirror tm) {
jaroslav@906
   916
        final Element e = processingEnv.getTypeUtils().asElement(tm);
jaroslav@907
   917
        if (e == null) {
jaroslav@907
   918
            return false;
jaroslav@907
   919
        }
jaroslav@906
   920
        for (Element ch : e.getEnclosedElements()) {
jaroslav@906
   921
            if (ch.getKind() == ElementKind.METHOD) {
jaroslav@906
   922
                ExecutableElement ee = (ExecutableElement)ch;
jaroslav@906
   923
                if (ee.getParameters().isEmpty() && ee.getSimpleName().contentEquals("modelFor")) {
jaroslav@906
   924
                    return true;
jaroslav@906
   925
                }
jaroslav@906
   926
            }
jaroslav@906
   927
        }
jaroslav@906
   928
        return models.values().contains(e.getSimpleName().toString());
jaroslav@906
   929
    }
jaroslav@909
   930
jaroslav@909
   931
    private void writeStringArray(List<String> strings, Writer w) throws IOException {
jaroslav@909
   932
        w.write("new String[] {\n");
jaroslav@909
   933
        String sep = "";
jaroslav@909
   934
        for (String n : strings) {
jaroslav@909
   935
            w.write(sep);
jaroslav@909
   936
            if (n == null) {
jaroslav@909
   937
                w.write("    null");
jaroslav@909
   938
            } else {
jaroslav@909
   939
                w.write("    \"" + n + "\"");
jaroslav@909
   940
            }
jaroslav@909
   941
            sep = ",\n";
jaroslav@909
   942
        }
jaroslav@909
   943
        w.write("\n  }");
jaroslav@909
   944
    }
jaroslav@919
   945
    
jaroslav@919
   946
    private void writeToString(Property[] props, Writer w) throws IOException {
jaroslav@919
   947
        w.write("  public String toString() {\n");
jaroslav@919
   948
        w.write("    StringBuilder sb = new StringBuilder();\n");
jaroslav@919
   949
        w.write("    sb.append('{');\n");
jaroslav@919
   950
        String sep = "";
jaroslav@919
   951
        for (Property p : props) {
jaroslav@919
   952
            w.write(sep);
jaroslav@919
   953
            w.append("    sb.append(\"" + p.name() + ": \");\n");
jaroslav@920
   954
            w.append("    sb.append(org.apidesign.bck2brwsr.htmlpage.ConvertTypes.toJSON(prop_");
jaroslav@920
   955
            w.append(p.name()).append("));\n");
jaroslav@919
   956
            sep =    "    sb.append(',');\n";
jaroslav@919
   957
        }
jaroslav@919
   958
        w.write("    sb.append('}');\n");
jaroslav@919
   959
        w.write("    return sb.toString();\n");
jaroslav@919
   960
        w.write("  }\n");
jaroslav@919
   961
    }
jaroslav@930
   962
    private void writeClone(String className, Property[] props, Writer w) throws IOException {
jaroslav@930
   963
        w.write("  public " + className + " clone() {\n");
jaroslav@930
   964
        w.write("    " + className + " ret = new " + className + "();\n");
jaroslav@930
   965
        for (Property p : props) {
jaroslav@930
   966
            if (!p.array()) {
jaroslav@930
   967
                boolean isModel[] = { false };
jaroslav@930
   968
                boolean isEnum[] = { false };
jaroslav@930
   969
                checkType(p, isModel, isEnum);
jaroslav@930
   970
                if (!isModel[0]) {
jaroslav@930
   971
                    w.write("    ret.prop_" + p.name() + " = prop_" + p.name() + ";\n");
jaroslav@930
   972
                    continue;
jaroslav@930
   973
                }
jaroslav@930
   974
            }
jaroslav@930
   975
            w.write("    ret.prop_" + p.name() + " = prop_" + p.name() + ".clone();\n");
jaroslav@930
   976
        }
jaroslav@930
   977
        
jaroslav@930
   978
        w.write("    return ret;\n");
jaroslav@930
   979
        w.write("  }\n");
jaroslav@930
   980
    }
jaroslav@925
   981
jaroslav@925
   982
    private String inPckName(Element e) {
jaroslav@925
   983
        StringBuilder sb = new StringBuilder();
jaroslav@925
   984
        while (e.getKind() != ElementKind.PACKAGE) {
jaroslav@925
   985
            if (sb.length() == 0) {
jaroslav@925
   986
                sb.append(e.getSimpleName());
jaroslav@925
   987
            } else {
jaroslav@925
   988
                sb.insert(0, '.');
jaroslav@925
   989
                sb.insert(0, e.getSimpleName());
jaroslav@925
   990
            }
jaroslav@925
   991
            e = e.getEnclosingElement();
jaroslav@925
   992
        }
jaroslav@925
   993
        return sb.toString();
jaroslav@925
   994
    }
jaroslav@929
   995
jaroslav@929
   996
    private String fqn(TypeMirror pt, Element relative) {
jaroslav@929
   997
        if (pt.getKind() == TypeKind.ERROR) {
jaroslav@929
   998
            final Elements eu = processingEnv.getElementUtils();
jaroslav@929
   999
            PackageElement pckg = eu.getPackageOf(relative);
jaroslav@929
  1000
            return pckg.getQualifiedName() + "." + pt.toString();
jaroslav@929
  1001
        }
jaroslav@929
  1002
        return pt.toString();
jaroslav@929
  1003
    }
jaroslav@930
  1004
jaroslav@930
  1005
    private String checkType(Property p, boolean[] isModel, boolean[] isEnum) {
jaroslav@930
  1006
        String ret;
jaroslav@930
  1007
        try {
jaroslav@930
  1008
            ret = p.type().getName();
jaroslav@930
  1009
        } catch (MirroredTypeException ex) {
jaroslav@930
  1010
            TypeMirror tm = processingEnv.getTypeUtils().erasure(ex.getTypeMirror());
jaroslav@930
  1011
            final Element e = processingEnv.getTypeUtils().asElement(tm);
jaroslav@930
  1012
            final Model m = e == null ? null : e.getAnnotation(Model.class);
jaroslav@930
  1013
            if (m != null) {
jaroslav@930
  1014
                ret = findPkgName(e) + '.' + m.className();
jaroslav@930
  1015
                isModel[0] = true;
jaroslav@930
  1016
                models.put(e, m.className());
jaroslav@930
  1017
            } else {
jaroslav@930
  1018
                ret = tm.toString();
jaroslav@930
  1019
            }
jaroslav@930
  1020
            TypeMirror enm = processingEnv.getElementUtils().getTypeElement("java.lang.Enum").asType();
jaroslav@930
  1021
            enm = processingEnv.getTypeUtils().erasure(enm);
jaroslav@930
  1022
            isEnum[0] = processingEnv.getTypeUtils().isSubtype(tm, enm);
jaroslav@930
  1023
        }
jaroslav@930
  1024
        return ret;
jaroslav@930
  1025
    }
jaroslav@934
  1026
jaroslav@934
  1027
    private Iterable<String> findParamNames(Element e, String url, StringBuilder assembleURL) {
jaroslav@934
  1028
        List<String> params = new ArrayList<>();
jaroslav@934
  1029
jaroslav@934
  1030
        for (int pos = 0; ;) {
jaroslav@934
  1031
            int next = url.indexOf('{', pos);
jaroslav@934
  1032
            if (next == -1) {
jaroslav@934
  1033
                assembleURL.append('"')
jaroslav@934
  1034
                    .append(url.substring(pos))
jaroslav@934
  1035
                    .append('"');
jaroslav@934
  1036
                return params;
jaroslav@934
  1037
            }
jaroslav@934
  1038
            int close = url.indexOf('}', next);
jaroslav@934
  1039
            if (close == -1) {
jaroslav@934
  1040
                err().printMessage(Diagnostic.Kind.ERROR, "Unbalanced '{' and '}' in " + url, e);
jaroslav@934
  1041
                return params;
jaroslav@934
  1042
            }
jaroslav@934
  1043
            final String paramName = url.substring(next + 1, close);
jaroslav@934
  1044
            params.add(paramName);
jaroslav@934
  1045
            assembleURL.append('"')
jaroslav@934
  1046
                .append(url.substring(pos, next))
jaroslav@934
  1047
                .append("\" + ").append(paramName).append(" + ");
jaroslav@934
  1048
            pos = close + 1;
jaroslav@934
  1049
        }
jaroslav@934
  1050
    }
jaroslav@943
  1051
jaroslav@943
  1052
    private static Property findProperty(Property[] properties, String propName) {
jaroslav@943
  1053
        for (Property p : properties) {
jaroslav@943
  1054
            if (propName.equals(p.name())) {
jaroslav@943
  1055
                return p;
jaroslav@943
  1056
            }
jaroslav@943
  1057
        }
jaroslav@943
  1058
        return null;
jaroslav@943
  1059
    }
jaroslav@26
  1060
}