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