json/src/main/java/org/netbeans/html/json/impl/ModelProcessor.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Mon, 16 Dec 2013 16:59:43 +0100
branchnetbeans
changeset 362 92fb71afdc0e
parent 358 json/src/main/java/org/apidesign/html/json/impl/ModelProcessor.java@80702021b851
child 365 5c93ad8c7a15
permissions -rw-r--r--
Moving implementation classes into org.netbeans.html namespace
jtulach@0
     1
/**
jaroslav@358
     2
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
jtulach@0
     3
 *
jaroslav@358
     4
 * Copyright 1997-2010 Oracle and/or its affiliates. All rights reserved.
jtulach@0
     5
 *
jaroslav@358
     6
 * Oracle and Java are registered trademarks of Oracle and/or its affiliates.
jaroslav@358
     7
 * Other names may be trademarks of their respective owners.
jtulach@0
     8
 *
jaroslav@358
     9
 * The contents of this file are subject to the terms of either the GNU
jaroslav@358
    10
 * General Public License Version 2 only ("GPL") or the Common
jaroslav@358
    11
 * Development and Distribution License("CDDL") (collectively, the
jaroslav@358
    12
 * "License"). You may not use this file except in compliance with the
jaroslav@358
    13
 * License. You can obtain a copy of the License at
jaroslav@358
    14
 * http://www.netbeans.org/cddl-gplv2.html
jaroslav@358
    15
 * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
jaroslav@358
    16
 * specific language governing permissions and limitations under the
jaroslav@358
    17
 * License.  When distributing the software, include this License Header
jaroslav@358
    18
 * Notice in each file and include the License file at
jaroslav@358
    19
 * nbbuild/licenses/CDDL-GPL-2-CP.  Oracle designates this
jaroslav@358
    20
 * particular file as subject to the "Classpath" exception as provided
jaroslav@358
    21
 * by Oracle in the GPL Version 2 section of the License file that
jaroslav@358
    22
 * accompanied this code. If applicable, add the following below the
jaroslav@358
    23
 * License Header, with the fields enclosed by brackets [] replaced by
jaroslav@358
    24
 * your own identifying information:
jaroslav@358
    25
 * "Portions Copyrighted [year] [name of copyright owner]"
jaroslav@358
    26
 *
jaroslav@358
    27
 * Contributor(s):
jaroslav@358
    28
 *
jaroslav@358
    29
 * The Original Software is NetBeans. The Initial Developer of the Original
jaroslav@358
    30
 * Software is Oracle. Portions Copyright 2013-2013 Oracle. All Rights Reserved.
jaroslav@358
    31
 *
jaroslav@358
    32
 * If you wish your version of this file to be governed by only the CDDL
jaroslav@358
    33
 * or only the GPL Version 2, indicate your decision by adding
jaroslav@358
    34
 * "[Contributor] elects to include this software in this distribution
jaroslav@358
    35
 * under the [CDDL or GPL Version 2] license." If you do not indicate a
jaroslav@358
    36
 * single choice of license, a recipient has the option to distribute
jaroslav@358
    37
 * your version of this file under either the CDDL, the GPL Version 2 or
jaroslav@358
    38
 * to extend the choice of license to its licensees as provided above.
jaroslav@358
    39
 * However, if you add GPL Version 2 code and therefore, elected the GPL
jaroslav@358
    40
 * Version 2 license, then the option applies only if the new code is
jaroslav@358
    41
 * made subject to such option by the copyright holder.
jtulach@0
    42
 */
jaroslav@362
    43
package org.netbeans.html.json.impl;
jtulach@0
    44
jtulach@0
    45
import java.io.IOException;
jtulach@0
    46
import java.io.OutputStreamWriter;
jtulach@0
    47
import java.io.StringWriter;
jtulach@0
    48
import java.io.Writer;
jtulach@0
    49
import java.lang.annotation.AnnotationTypeMismatchException;
jtulach@0
    50
import java.lang.annotation.IncompleteAnnotationException;
jtulach@0
    51
import java.lang.reflect.Method;
jtulach@0
    52
import java.util.ArrayList;
jaroslav@70
    53
import java.util.Arrays;
jtulach@0
    54
import java.util.Collection;
jtulach@0
    55
import java.util.Collections;
jtulach@0
    56
import java.util.HashMap;
jtulach@0
    57
import java.util.HashSet;
jtulach@0
    58
import java.util.LinkedHashSet;
jtulach@0
    59
import java.util.List;
jtulach@0
    60
import java.util.Map;
jaroslav@70
    61
import java.util.ResourceBundle;
jtulach@0
    62
import java.util.Set;
jtulach@0
    63
import java.util.WeakHashMap;
jaroslav@71
    64
import java.util.logging.Level;
jaroslav@70
    65
import java.util.logging.Logger;
jtulach@0
    66
import javax.annotation.processing.AbstractProcessor;
jaroslav@70
    67
import javax.annotation.processing.Completion;
jaroslav@70
    68
import javax.annotation.processing.Completions;
jtulach@0
    69
import javax.annotation.processing.ProcessingEnvironment;
jtulach@0
    70
import javax.annotation.processing.Processor;
jtulach@0
    71
import javax.annotation.processing.RoundEnvironment;
jtulach@0
    72
import javax.annotation.processing.SupportedAnnotationTypes;
jaroslav@58
    73
import javax.annotation.processing.SupportedSourceVersion;
jaroslav@58
    74
import javax.lang.model.SourceVersion;
jtulach@0
    75
import javax.lang.model.element.AnnotationMirror;
jtulach@0
    76
import javax.lang.model.element.AnnotationValue;
jtulach@0
    77
import javax.lang.model.element.Element;
jtulach@0
    78
import javax.lang.model.element.ElementKind;
jtulach@0
    79
import javax.lang.model.element.ExecutableElement;
jtulach@0
    80
import javax.lang.model.element.Modifier;
jtulach@0
    81
import javax.lang.model.element.PackageElement;
jtulach@0
    82
import javax.lang.model.element.TypeElement;
jtulach@0
    83
import javax.lang.model.element.VariableElement;
jtulach@0
    84
import javax.lang.model.type.ArrayType;
jaroslav@29
    85
import javax.lang.model.type.DeclaredType;
jtulach@0
    86
import javax.lang.model.type.MirroredTypeException;
jtulach@0
    87
import javax.lang.model.type.TypeKind;
jtulach@0
    88
import javax.lang.model.type.TypeMirror;
jtulach@0
    89
import javax.lang.model.util.Elements;
jtulach@0
    90
import javax.lang.model.util.Types;
jtulach@0
    91
import javax.tools.Diagnostic;
jtulach@0
    92
import javax.tools.FileObject;
jtulach@0
    93
import net.java.html.json.ComputedProperty;
jtulach@0
    94
import net.java.html.json.Model;
jtulach@0
    95
import net.java.html.json.Function;
jaroslav@240
    96
import net.java.html.json.ModelOperation;
jtulach@0
    97
import net.java.html.json.OnPropertyChange;
jtulach@0
    98
import net.java.html.json.OnReceive;
jtulach@0
    99
import net.java.html.json.Property;
jtulach@0
   100
import org.openide.util.lookup.ServiceProvider;
jtulach@0
   101
jaroslav@70
   102
/** Annotation processor to process {@link Model @Model} annotations and
jaroslav@70
   103
 * generate appropriate model classes.
jtulach@0
   104
 *
jtulach@0
   105
 * @author Jaroslav Tulach <jtulach@netbeans.org>
jtulach@0
   106
 */
jtulach@0
   107
@ServiceProvider(service=Processor.class)
jaroslav@100
   108
@SupportedSourceVersion(SourceVersion.RELEASE_6)
jtulach@0
   109
@SupportedAnnotationTypes({
jtulach@1
   110
    "net.java.html.json.Model",
jaroslav@240
   111
    "net.java.html.json.ModelOperation",
jtulach@1
   112
    "net.java.html.json.Function",
jtulach@1
   113
    "net.java.html.json.OnReceive",
jtulach@1
   114
    "net.java.html.json.OnPropertyChange",
jtulach@1
   115
    "net.java.html.json.ComputedProperty",
jtulach@1
   116
    "net.java.html.json.Property"
jtulach@0
   117
})
jtulach@3
   118
public final class ModelProcessor extends AbstractProcessor {
jaroslav@70
   119
    private static final Logger LOG = Logger.getLogger(ModelProcessor.class.getName());
jaroslav@100
   120
    private final Map<Element,String> models = new WeakHashMap<Element,String>();
jaroslav@100
   121
    private final Map<Element,Prprt[]> verify = new WeakHashMap<Element,Prprt[]>();
jtulach@0
   122
    @Override
jtulach@0
   123
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
jtulach@0
   124
        boolean ok = true;
jtulach@0
   125
        for (Element e : roundEnv.getElementsAnnotatedWith(Model.class)) {
jtulach@0
   126
            if (!processModel(e)) {
jtulach@0
   127
                ok = false;
jtulach@0
   128
            }
jtulach@0
   129
        }
jtulach@0
   130
        if (roundEnv.processingOver()) {
jtulach@0
   131
            models.clear();
jtulach@0
   132
            for (Map.Entry<Element, Prprt[]> entry : verify.entrySet()) {
jtulach@0
   133
                TypeElement te = (TypeElement)entry.getKey();
jtulach@0
   134
                String fqn = processingEnv.getElementUtils().getBinaryName(te).toString();
jtulach@0
   135
                Element finalElem = processingEnv.getElementUtils().getTypeElement(fqn);
jtulach@0
   136
                if (finalElem == null) {
jtulach@0
   137
                    continue;
jtulach@0
   138
                }
jtulach@0
   139
                Prprt[] props;
jtulach@0
   140
                Model m = finalElem.getAnnotation(Model.class);
jtulach@3
   141
                if (m == null) {
jtulach@3
   142
                    continue;
jtulach@3
   143
                }
jtulach@0
   144
                props = Prprt.wrap(processingEnv, finalElem, m.properties());
jtulach@0
   145
                for (Prprt p : props) {
jtulach@0
   146
                    boolean[] isModel = { false };
jtulach@0
   147
                    boolean[] isEnum = { false };
jtulach@0
   148
                    boolean[] isPrimitive = { false };
jtulach@0
   149
                    String t = checkType(p, isModel, isEnum, isPrimitive);
jtulach@0
   150
                    if (isEnum[0]) {
jtulach@0
   151
                        continue;
jtulach@0
   152
                    }
jtulach@0
   153
                    if (isPrimitive[0]) {
jtulach@0
   154
                        continue;
jtulach@0
   155
                    }
jtulach@0
   156
                    if (isModel[0]) {
jtulach@0
   157
                        continue;
jtulach@0
   158
                    }
jtulach@0
   159
                    if ("java.lang.String".equals(t)) {
jtulach@0
   160
                        continue;
jtulach@0
   161
                    }
jtulach@0
   162
                    error("The type " + t + " should be defined by @Model annotation", entry.getKey());
jtulach@0
   163
                }
jtulach@0
   164
            }
jtulach@0
   165
            verify.clear();
jtulach@0
   166
        }
jtulach@0
   167
        return ok;
jtulach@0
   168
    }
jtulach@0
   169
jtulach@0
   170
    private void error(String msg, Element e) {
jtulach@0
   171
        processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, msg, e);
jtulach@0
   172
    }
jtulach@0
   173
    
jtulach@0
   174
    private boolean processModel(Element e) {
jtulach@0
   175
        boolean ok = true;
jtulach@0
   176
        Model m = e.getAnnotation(Model.class);
jtulach@0
   177
        if (m == null) {
jtulach@0
   178
            return true;
jtulach@0
   179
        }
jtulach@0
   180
        String pkg = findPkgName(e);
jtulach@0
   181
        Writer w;
jtulach@0
   182
        String className = m.className();
jtulach@0
   183
        models.put(e, className);
jtulach@0
   184
        try {
jtulach@0
   185
            StringWriter body = new StringWriter();
jaroslav@100
   186
            List<String> propsGetSet = new ArrayList<String>();
jaroslav@100
   187
            List<String> functions = new ArrayList<String>();
jaroslav@100
   188
            Map<String, Collection<String>> propsDeps = new HashMap<String, Collection<String>>();
jaroslav@100
   189
            Map<String, Collection<String>> functionDeps = new HashMap<String, Collection<String>>();
jtulach@0
   190
            Prprt[] props = createProps(e, m.properties());
jtulach@0
   191
            
jtulach@0
   192
            if (!generateComputedProperties(body, props, e.getEnclosedElements(), propsGetSet, propsDeps)) {
jtulach@0
   193
                ok = false;
jtulach@0
   194
            }
jtulach@0
   195
            if (!generateOnChange(e, propsDeps, props, className, functionDeps)) {
jtulach@0
   196
                ok = false;
jtulach@0
   197
            }
jtulach@0
   198
            if (!generateProperties(e, body, props, propsGetSet, propsDeps, functionDeps)) {
jtulach@0
   199
                ok = false;
jtulach@0
   200
            }
jtulach@0
   201
            if (!generateFunctions(e, body, className, e.getEnclosedElements(), functions)) {
jtulach@0
   202
                ok = false;
jtulach@0
   203
            }
jaroslav@12
   204
            if (!generateReceive(e, body, className, e.getEnclosedElements(), functions)) {
jaroslav@12
   205
                ok = false;
jaroslav@12
   206
            }
jaroslav@240
   207
            if (!generateOperation(e, body, className, e.getEnclosedElements())) {
jaroslav@240
   208
                ok = false;
jaroslav@240
   209
            }
jtulach@0
   210
            FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + className, e);
jtulach@0
   211
            w = new OutputStreamWriter(java.openOutputStream());
jtulach@0
   212
            try {
jtulach@0
   213
                w.append("package " + pkg + ";\n");
jtulach@2
   214
                w.append("import net.java.html.json.*;\n");
jtulach@0
   215
                w.append("public final class ").append(className).append(" implements Cloneable {\n");
jtulach@0
   216
                w.append("  private boolean locked;\n");
jaroslav@110
   217
                w.append("  private net.java.html.BrwsrCtx context;\n");
jaroslav@362
   218
                w.append("  private org.netbeans.html.json.impl.Bindings[] ko = { null };\n");
jtulach@0
   219
                w.append(body.toString());
jtulach@0
   220
                w.append("  private static Class<" + inPckName(e) + "> modelFor() { return null; }\n");
jaroslav@111
   221
                w.append("  private ").append(className).append("(net.java.html.BrwsrCtx context) {\n");
jaroslav@5
   222
                w.append("    this.context = context;\n");
jtulach@0
   223
                w.append("  };\n");
jaroslav@111
   224
                w.append("  public ").append(className).append("() {\n");
jaroslav@111
   225
                w.append("    this(net.java.html.BrwsrCtx.findDefault(").append(className).append(".class));\n");
jaroslav@231
   226
                for (Prprt p : props) {
jaroslav@231
   227
                    if (p.array()) {
jaroslav@231
   228
                        continue;
jaroslav@231
   229
                    }
jaroslav@231
   230
                    boolean[] isModel = {false};
jaroslav@231
   231
                    boolean[] isEnum = {false};
jaroslav@231
   232
                    boolean isPrimitive[] = {false};
jaroslav@231
   233
                    String tn = checkType(p, isModel, isEnum, isPrimitive);
jaroslav@231
   234
                    if (isModel[0]) {
jaroslav@231
   235
                        w.write("    prop_" + p.name() + " = new " + tn + "();\n");
jaroslav@231
   236
                    }
jaroslav@231
   237
                }
jaroslav@111
   238
                w.append("  };\n");
jaroslav@284
   239
                if (props.length > 0) {
jaroslav@284
   240
                    w.append("  public ").append(className).append("(");
jaroslav@284
   241
                    Prprt firstArray = null;
jaroslav@284
   242
                    String sep = "";
jaroslav@284
   243
                    for (Prprt p : props) {
jaroslav@284
   244
                        if (p.array()) {
jaroslav@284
   245
                            if (firstArray == null) {
jaroslav@284
   246
                                firstArray = p;
jaroslav@284
   247
                            }
jaroslav@284
   248
                            continue;
jaroslav@232
   249
                        }
jaroslav@284
   250
                        String tn = typeName(e, p);
jaroslav@284
   251
                        w.write(sep);
jaroslav@284
   252
                        w.write(tn);
jaroslav@284
   253
                        w.write(" " + p.name());
jaroslav@284
   254
                        sep = ", ";
jaroslav@232
   255
                    }
jaroslav@284
   256
                    if (firstArray != null) {
jaroslav@312
   257
                        String tn;
jaroslav@312
   258
                        boolean[] isModel = {false};
jaroslav@312
   259
                        boolean[] isEnum = {false};
jaroslav@312
   260
                        boolean isPrimitive[] = {false};
jaroslav@312
   261
                        tn = checkType(firstArray, isModel, isEnum, isPrimitive);
jaroslav@284
   262
                        w.write(sep);
jaroslav@284
   263
                        w.write(tn);
jaroslav@284
   264
                        w.write("... " + firstArray.name());
jaroslav@284
   265
                    }
jaroslav@284
   266
                    w.append(") {\n");
jaroslav@284
   267
                    w.append("    this(net.java.html.BrwsrCtx.findDefault(").append(className).append(".class));\n");
jaroslav@284
   268
                    for (Prprt p : props) {
jaroslav@284
   269
                        if (p.array()) {
jaroslav@284
   270
                            continue;
jaroslav@284
   271
                        }
jaroslav@284
   272
                        w.write("    this.prop_" + p.name() + " = " + p.name() + ";\n");
jaroslav@284
   273
                    }
jaroslav@284
   274
                    if (firstArray != null) {
jaroslav@284
   275
                        w.write("    this.prop_" + firstArray.name() + ".init(" + firstArray.name() + ");\n");
jaroslav@284
   276
                    }
jaroslav@284
   277
                    w.append("  };\n");
jaroslav@232
   278
                }
jaroslav@362
   279
                w.append("  private org.netbeans.html.json.impl.Bindings intKnckt() {\n");
jaroslav@278
   280
                w.append("    if (ko[0] != null) return ko[0];\n");
jaroslav@362
   281
                w.append("    ko[0] = org.netbeans.html.json.impl.Bindings.apply(context, this);\n");
jaroslav@275
   282
                {
jaroslav@275
   283
                    w.append("    org.apidesign.html.json.spi.PropertyBinding[] propArr = {\n");
jaroslav@275
   284
                    for (int i = 0; i < propsGetSet.size(); i += 5) {
jaroslav@278
   285
                        w.append("      ko[0].registerProperty(\"").append(propsGetSet.get(i)).append("\", this, new P(");
jaroslav@275
   286
                        w.append((i / 5) + "), " + (propsGetSet.get(i + 2) == null) + "),\n");
jaroslav@275
   287
                    }
jaroslav@275
   288
                    w.append("    };\n");
jaroslav@17
   289
                }
jaroslav@275
   290
                {
jaroslav@275
   291
                    w.append("    org.apidesign.html.json.spi.FunctionBinding[] funcArr = {\n");
jaroslav@275
   292
                    for (int i = 0; i < functions.size(); i += 2) {
jaroslav@278
   293
                        w.append("      ko[0].registerFunction(\"").append(functions.get(i)).append("\", this, new P(");
jaroslav@275
   294
                        w.append((i / 2) + ")),\n");
jaroslav@275
   295
                    }
jaroslav@275
   296
                    w.append("    };\n");
jaroslav@20
   297
                }
jaroslav@278
   298
                w.append("    ko[0].finish(this, propArr, funcArr);\n");
jaroslav@278
   299
                w.append("    return ko[0];\n");
jtulach@0
   300
                w.append("  };\n");
jaroslav@362
   301
                w.append("  private static final class P implements org.netbeans.html.json.impl.SetAndGet<" + className + ">,\n");
jaroslav@362
   302
                w.append("  org.netbeans.html.json.impl.Callback<" + className + ">,\n");
jaroslav@362
   303
                w.append("  org.netbeans.html.json.impl.FromJSON<" + className + "> {\n");
jaroslav@17
   304
                w.append("    private final int type;\n");
jaroslav@17
   305
                w.append("    P(int t) { type = t; };\n");
jaroslav@17
   306
                w.append("    public void setValue(" + className + " data, Object value) {\n");
jaroslav@17
   307
                w.append("      switch (type) {\n");
jaroslav@17
   308
                for (int i = 0; i < propsGetSet.size(); i += 5) {
jaroslav@17
   309
                    final String set = propsGetSet.get(i + 2);
jaroslav@68
   310
                    String tn = propsGetSet.get(i + 4);
jaroslav@198
   311
                    String btn = findBoxedType(tn);
jaroslav@198
   312
                    if (btn != null) {
jaroslav@198
   313
                        tn = btn;
jaroslav@68
   314
                    }
jaroslav@17
   315
                    if (set != null) {
jaroslav@362
   316
                        w.append("        case " + (i / 5) + ": data." + strip(set) + "(org.netbeans.html.json.impl.JSON.extractValue(" + tn + ".class, value)); return;\n");
jaroslav@17
   317
                    }
jaroslav@17
   318
                }
jaroslav@17
   319
                w.append("      }\n");
jaroslav@17
   320
                w.append("    }\n");
jaroslav@17
   321
                w.append("    public Object getValue(" + className + " data) {\n");
jaroslav@17
   322
                w.append("      switch (type) {\n");
jaroslav@17
   323
                for (int i = 0; i < propsGetSet.size(); i += 5) {
jaroslav@17
   324
                    final String get = propsGetSet.get(i + 1);
jaroslav@17
   325
                    if (get != null) {
jaroslav@17
   326
                        w.append("        case " + (i / 5) + ": return data." + strip(get) + "();\n");
jaroslav@17
   327
                    }
jaroslav@17
   328
                }
jaroslav@17
   329
                w.append("      }\n");
jaroslav@17
   330
                w.append("      throw new UnsupportedOperationException();\n");
jaroslav@17
   331
                w.append("    }\n");
jaroslav@20
   332
                w.append("    public void call(" + className + " model, Object data, Object ev) {\n");
jaroslav@20
   333
                w.append("      switch (type) {\n");
jaroslav@20
   334
                for (int i = 0; i < functions.size(); i += 2) {
jaroslav@20
   335
                    final String name = functions.get(i);
jaroslav@20
   336
                    w.append("        case " + (i / 2) + ": model." + name + "(data, ev); return;\n");
jaroslav@20
   337
                }
jaroslav@20
   338
                w.append("      }\n");
jaroslav@20
   339
                w.append("      throw new UnsupportedOperationException();\n");
jaroslav@20
   340
                w.append("    }\n");
jaroslav@27
   341
                w.append("    public Class<" + className + "> factoryFor() { return " + className + ".class; }\n");
jaroslav@110
   342
                w.append("    public " + className + " read(net.java.html.BrwsrCtx c, Object json) { return new " + className + "(c, json); }\n");
jaroslav@111
   343
                w.append("    public " + className + " cloneTo(Object o, net.java.html.BrwsrCtx c) { return ((" + className + ")o).clone(c); }\n");
jaroslav@17
   344
                w.append("  }\n");
jaroslav@362
   345
                w.append("  static { org.netbeans.html.json.impl.JSON.register(new P(0)); }\n");
jaroslav@110
   346
                w.append("  private ").append(className).append("(net.java.html.BrwsrCtx c, Object json) {\n");
jaroslav@25
   347
                w.append("    this.context = c;\n");
jtulach@0
   348
                int values = 0;
jaroslav@17
   349
                for (int i = 0; i < propsGetSet.size(); i += 5) {
jtulach@0
   350
                    Prprt p = findPrprt(props, propsGetSet.get(i));
jtulach@0
   351
                    if (p == null) {
jtulach@0
   352
                        continue;
jtulach@0
   353
                    }
jtulach@0
   354
                    values++;
jtulach@0
   355
                }
jtulach@0
   356
                w.append("    Object[] ret = new Object[" + values + "];\n");
jaroslav@362
   357
                w.append("    org.netbeans.html.json.impl.JSON.extract(context, json, new String[] {\n");
jaroslav@17
   358
                for (int i = 0; i < propsGetSet.size(); i += 5) {
jtulach@0
   359
                    Prprt p = findPrprt(props, propsGetSet.get(i));
jtulach@0
   360
                    if (p == null) {
jtulach@0
   361
                        continue;
jtulach@0
   362
                    }
jtulach@0
   363
                    w.append("      \"").append(propsGetSet.get(i)).append("\",\n");
jtulach@0
   364
                }
jtulach@0
   365
                w.append("    }, ret);\n");
jaroslav@17
   366
                for (int i = 0, cnt = 0, prop = 0; i < propsGetSet.size(); i += 5) {
jtulach@0
   367
                    final String pn = propsGetSet.get(i);
jtulach@0
   368
                    Prprt p = findPrprt(props, pn);
jtulach@0
   369
                    if (p == null) {
jtulach@0
   370
                        continue;
jtulach@0
   371
                    }
jtulach@0
   372
                    boolean[] isModel = { false };
jtulach@0
   373
                    boolean[] isEnum = { false };
jtulach@0
   374
                    boolean isPrimitive[] = { false };
jtulach@0
   375
                    String type = checkType(props[prop++], isModel, isEnum, isPrimitive);
jtulach@0
   376
                    if (p.array()) {
jaroslav@66
   377
                        w.append("    if (ret[" + cnt + "] instanceof Object[]) {\n");
jaroslav@66
   378
                        w.append("      for (Object e : ((Object[])ret[" + cnt + "])) {\n");
jtulach@0
   379
                        if (isModel[0]) {
jaroslav@362
   380
                            w.append("        this.prop_").append(pn).append(".add(org.netbeans.html.json.impl.JSON.read");
jaroslav@27
   381
                            w.append("(c, " + type + ".class, e));\n");
jtulach@0
   382
                        } else if (isEnum[0]) {
jaroslav@66
   383
                            w.append("        this.prop_").append(pn);
jtulach@0
   384
                            w.append(".add(e == null ? null : ");
jaroslav@362
   385
                            w.append(type).append(".valueOf(org.netbeans.html.json.impl.JSON.stringValue(e)));\n");
jtulach@0
   386
                        } else {
jtulach@0
   387
                            if (isPrimitive(type)) {
jaroslav@362
   388
                                w.append("        this.prop_").append(pn).append(".add(org.netbeans.html.json.impl.JSON.numberValue(e).");
jtulach@0
   389
                                w.append(type).append("Value());\n");
jtulach@0
   390
                            } else {
jaroslav@66
   391
                                w.append("        this.prop_").append(pn).append(".add((");
jtulach@0
   392
                                w.append(type).append(")e);\n");
jtulach@0
   393
                            }
jtulach@0
   394
                        }
jaroslav@66
   395
                        w.append("      }\n");
jaroslav@66
   396
                        w.append("    }\n");
jtulach@0
   397
                    } else {
jtulach@0
   398
                        if (isEnum[0]) {
jaroslav@340
   399
                            w.append("    try {\n");
jtulach@0
   400
                            w.append("    this.prop_").append(pn);
jtulach@0
   401
                            w.append(" = ret[" + cnt + "] == null ? null : ");
jaroslav@362
   402
                            w.append(type).append(".valueOf(org.netbeans.html.json.impl.JSON.stringValue(ret[" + cnt + "]));\n");
jaroslav@340
   403
                            w.append("    } catch (IllegalArgumentException ex) {\n");
jaroslav@340
   404
                            w.append("      ex.printStackTrace();\n");
jaroslav@340
   405
                            w.append("    }\n");
jtulach@0
   406
                        } else if (isPrimitive(type)) {
jtulach@0
   407
                            w.append("    this.prop_").append(pn);
jaroslav@100
   408
                            w.append(" = ret[" + cnt + "] == null ? ");
jaroslav@100
   409
                            if ("char".equals(type)) {
jaroslav@362
   410
                                w.append("0 : (org.netbeans.html.json.impl.JSON.charValue(");
jaroslav@100
   411
                            } else if ("boolean".equals(type)) {
jaroslav@362
   412
                                w.append("false : (org.netbeans.html.json.impl.JSON.boolValue(");
jaroslav@100
   413
                            } else {
jaroslav@362
   414
                                w.append("0 : (org.netbeans.html.json.impl.JSON.numberValue(");
jaroslav@100
   415
                            }
jaroslav@197
   416
                            w.append("ret[" + cnt + "])).");
jtulach@0
   417
                            w.append(type).append("Value();\n");
jaroslav@136
   418
                        } else if (isModel[0]) {
jaroslav@362
   419
                            w.append("    this.prop_").append(pn).append(" = org.netbeans.html.json.impl.JSON.read");
jaroslav@136
   420
                            w.append("(c, " + type + ".class, ");
jaroslav@136
   421
                            w.append("ret[" + cnt + "]);\n");
jaroslav@136
   422
                        }else {
jtulach@0
   423
                            w.append("    this.prop_").append(pn);
jtulach@0
   424
                            w.append(" = (").append(type).append(')');
jtulach@0
   425
                            w.append("ret[" + cnt + "];\n");
jtulach@0
   426
                        }
jtulach@0
   427
                    }
jtulach@0
   428
                    cnt++;
jtulach@0
   429
                }
jtulach@0
   430
                w.append("  };\n");
jtulach@0
   431
                writeToString(props, w);
jtulach@0
   432
                writeClone(className, props, w);
jaroslav@311
   433
                w.write("  /** Activates this model instance in the current {@link \n"
jaroslav@311
   434
                    + "net.java.html.json.Models#bind(java.lang.Object, net.java.html.BrwsrCtx) browser context}. \n"
jaroslav@311
   435
                    + "In case of using Knockout technology, this means to \n"
jaroslav@311
   436
                    + "bind JSON like data in this model instance with Knockout tags in \n"
jaroslav@311
   437
                    + "the surrounding HTML page.\n"
jaroslav@311
   438
                    + "*/\n"
jaroslav@230
   439
                );
jaroslav@15
   440
                w.write("  public " + className + " applyBindings() {\n");
jaroslav@31
   441
                w.write("    intKnckt().applyBindings();\n");
jaroslav@15
   442
                w.write("    return this;\n");
jaroslav@15
   443
                w.write("  }\n");
jaroslav@15
   444
                w.write("  public boolean equals(Object o) {\n");
jaroslav@15
   445
                w.write("    if (o == this) return true;\n");
jaroslav@362
   446
                w.write("    if (o instanceof org.netbeans.html.json.impl.WrapperObject) {\n");
jaroslav@362
   447
                w.write("      ((org.netbeans.html.json.impl.WrapperObject)o).setRealObject(intKnckt().koData());\n");
jaroslav@281
   448
                w.write("      return false;\n");
jaroslav@15
   449
                w.write("    }\n");
jaroslav@281
   450
                w.write("    if (!(o instanceof " + className + ")) return false;\n");
jaroslav@281
   451
                w.write("    " + className + " p = (" + className + ")o;\n");
jaroslav@281
   452
                for (Prprt p : props) {
jaroslav@362
   453
                    w.write("    if (!org.netbeans.html.json.impl.JSON.isSame(prop_" + p.name() + ", p.prop_" + p.name() + ")) return false;\n");
jaroslav@281
   454
                }
jaroslav@281
   455
                w.write("    return true;\n");
jaroslav@281
   456
                w.write("  }\n");
jaroslav@281
   457
                w.write("  public int hashCode() {\n");
jaroslav@281
   458
                w.write("    int h = " + className + ".class.getName().hashCode();\n");
jaroslav@281
   459
                for (Prprt p : props) {
jaroslav@362
   460
                    w.write("    h = org.netbeans.html.json.impl.JSON.hashPlus(prop_" + p.name() + ", h);\n");
jaroslav@281
   461
                }
jaroslav@281
   462
                w.write("    return h;\n");
jaroslav@15
   463
                w.write("  }\n");
jaroslav@14
   464
                w.write("}\n");
jtulach@0
   465
            } finally {
jtulach@0
   466
                w.close();
jtulach@0
   467
            }
jtulach@0
   468
        } catch (IOException ex) {
jtulach@0
   469
            error("Can't create " + className + ".java", e);
jtulach@0
   470
            return false;
jtulach@0
   471
        }
jtulach@0
   472
        return ok;
jtulach@0
   473
    }
jtulach@0
   474
    
jtulach@0
   475
    private boolean generateProperties(
jtulach@0
   476
        Element where,
jtulach@0
   477
        Writer w, Prprt[] properties,
jtulach@0
   478
        Collection<String> props, 
jtulach@0
   479
        Map<String,Collection<String>> deps,
jtulach@0
   480
        Map<String,Collection<String>> functionDeps
jtulach@0
   481
    ) throws IOException {
jtulach@0
   482
        boolean ok = true;
jtulach@0
   483
        for (Prprt p : properties) {
jtulach@0
   484
            final String tn;
jtulach@0
   485
            tn = typeName(where, p);
jtulach@0
   486
            String[] gs = toGetSet(p.name(), tn, p.array());
jaroslav@17
   487
            String castTo;
jaroslav@17
   488
            
jtulach@0
   489
            if (p.array()) {
jaroslav@362
   490
                w.write("  private org.netbeans.html.json.impl.JSONList<" + tn + "> prop_" + p.name() + " = new org.netbeans.html.json.impl.JSONList<" + tn + ">(ko, \""
jtulach@0
   491
                    + p.name() + "\"");
jtulach@0
   492
                Collection<String> dependants = deps.get(p.name());
jtulach@0
   493
                if (dependants != null) {
jtulach@0
   494
                    for (String depProp : dependants) {
jtulach@0
   495
                        w.write(", ");
jtulach@0
   496
                        w.write('\"');
jtulach@0
   497
                        w.write(depProp);
jtulach@0
   498
                        w.write('\"');
jtulach@0
   499
                    }
jtulach@0
   500
                }
jtulach@0
   501
                w.write(")");
jtulach@0
   502
                
jtulach@0
   503
                dependants = functionDeps.get(p.name());
jtulach@0
   504
                if (dependants != null) {
jtulach@0
   505
                    w.write(".onChange(new Runnable() { public void run() {\n");
jtulach@0
   506
                    for (String call : dependants) {
jaroslav@66
   507
                        w.append("  ").append(call);
jtulach@0
   508
                    }
jaroslav@66
   509
                    w.write("  }})");
jtulach@0
   510
                }
jtulach@0
   511
                w.write(";\n");
jaroslav@17
   512
            
jaroslav@17
   513
                castTo = "java.util.List";
jaroslav@66
   514
                w.write("  public java.util.List<" + tn + "> " + gs[0] + "() {\n");
jaroslav@66
   515
                w.write("    if (locked) throw new IllegalStateException();\n");
jaroslav@66
   516
                w.write("    return prop_" + p.name() + ";\n");
jaroslav@66
   517
                w.write("  }\n");
jtulach@0
   518
            } else {
jaroslav@17
   519
                castTo = tn;
jaroslav@66
   520
                w.write("  private " + tn + " prop_" + p.name() + ";\n");
jaroslav@66
   521
                w.write("  public " + tn + " " + gs[0] + "() {\n");
jaroslav@66
   522
                w.write("    if (locked) throw new IllegalStateException();\n");
jaroslav@66
   523
                w.write("    return prop_" + p.name() + ";\n");
jaroslav@66
   524
                w.write("  }\n");
jaroslav@66
   525
                w.write("  public void " + gs[1] + "(" + tn + " v) {\n");
jaroslav@66
   526
                w.write("    if (locked) throw new IllegalStateException();\n");
jaroslav@362
   527
                w.write("    if (org.netbeans.html.json.impl.JSON.isSame(prop_" + p.name() + ", v)) return;\n");
jaroslav@66
   528
                w.write("    prop_" + p.name() + " = v;\n");
jaroslav@362
   529
                w.write("    org.netbeans.html.json.impl.Bindings b = ko[0];\n");
jaroslav@66
   530
                w.write("    if (b != null) {\n");
jaroslav@66
   531
                w.write("      b.valueHasMutated(\"" + p.name() + "\");\n");
jtulach@0
   532
                Collection<String> dependants = deps.get(p.name());
jtulach@0
   533
                if (dependants != null) {
jtulach@0
   534
                    for (String depProp : dependants) {
jaroslav@66
   535
                        w.write("      b.valueHasMutated(\"" + depProp + "\");\n");
jaroslav@66
   536
                    }
jaroslav@66
   537
                }
jaroslav@66
   538
                w.write("    }\n");
jaroslav@66
   539
                dependants = functionDeps.get(p.name());
jaroslav@66
   540
                if (dependants != null) {
jaroslav@66
   541
                    for (String call : dependants) {
jaroslav@66
   542
                        w.append("  ").append(call);
jtulach@0
   543
                    }
jtulach@0
   544
                }
jtulach@0
   545
                w.write("  }\n");
jtulach@0
   546
            }
jtulach@0
   547
            
jtulach@0
   548
            props.add(p.name());
jtulach@0
   549
            props.add(gs[2]);
jtulach@0
   550
            props.add(gs[3]);
jtulach@0
   551
            props.add(gs[0]);
jaroslav@17
   552
            props.add(castTo);
jtulach@0
   553
        }
jtulach@0
   554
        return ok;
jtulach@0
   555
    }
jtulach@0
   556
jtulach@0
   557
    private boolean generateComputedProperties(
jtulach@0
   558
        Writer w, Prprt[] fixedProps,
jtulach@0
   559
        Collection<? extends Element> arr, Collection<String> props,
jtulach@0
   560
        Map<String,Collection<String>> deps
jtulach@0
   561
    ) throws IOException {
jtulach@0
   562
        boolean ok = true;
jtulach@0
   563
        for (Element e : arr) {
jtulach@0
   564
            if (e.getKind() != ElementKind.METHOD) {
jtulach@0
   565
                continue;
jtulach@0
   566
            }
jtulach@0
   567
            if (e.getAnnotation(ComputedProperty.class) == null) {
jtulach@0
   568
                continue;
jtulach@0
   569
            }
jaroslav@227
   570
            if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@227
   571
                error("Method " + e.getSimpleName() + " has to be static when annotated by @ComputedProperty", e);
jaroslav@227
   572
                ok = false;
jaroslav@227
   573
                continue;
jaroslav@227
   574
            }
jtulach@0
   575
            ExecutableElement ee = (ExecutableElement)e;
jtulach@0
   576
            final TypeMirror rt = ee.getReturnType();
jtulach@0
   577
            final Types tu = processingEnv.getTypeUtils();
jtulach@0
   578
            TypeMirror ert = tu.erasure(rt);
jtulach@0
   579
            String tn = fqn(ert, ee);
jtulach@0
   580
            boolean array = false;
jaroslav@235
   581
            final TypeMirror toCheck;
jtulach@0
   582
            if (tn.equals("java.util.List")) {
jtulach@0
   583
                array = true;
jaroslav@235
   584
                toCheck = ((DeclaredType)rt).getTypeArguments().get(0);
jaroslav@235
   585
            } else {
jaroslav@235
   586
                toCheck = rt;
jtulach@0
   587
            }
jtulach@0
   588
            
jtulach@0
   589
            final String sn = ee.getSimpleName().toString();
jaroslav@235
   590
            
jaroslav@235
   591
            if (toCheck.getKind().isPrimitive()) {
jaroslav@235
   592
                // OK
jaroslav@235
   593
            } else {
jaroslav@235
   594
                TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jaroslav@235
   595
                TypeMirror enumType = processingEnv.getElementUtils().getTypeElement("java.lang.Enum").asType();
jaroslav@235
   596
jaroslav@235
   597
                if (tu.isSubtype(toCheck, stringType)) {
jaroslav@235
   598
                    // OK
jaroslav@252
   599
                } else if (tu.isSubtype(tu.erasure(toCheck), tu.erasure(enumType))) {
jaroslav@235
   600
                    // OK
jaroslav@236
   601
                } else if (isModel(toCheck)) {
jaroslav@236
   602
                    // OK
jaroslav@235
   603
                } else {
jaroslav@235
   604
                    ok = false;
jaroslav@235
   605
                    error(sn + " cannot return " + toCheck, e);
jaroslav@235
   606
                }
jaroslav@235
   607
            }
jaroslav@235
   608
            
jtulach@0
   609
            String[] gs = toGetSet(sn, tn, array);
jtulach@0
   610
            
jaroslav@66
   611
            w.write("  public " + tn + " " + gs[0] + "() {\n");
jaroslav@66
   612
            w.write("    if (locked) throw new IllegalStateException();\n");
jtulach@0
   613
            int arg = 0;
jtulach@0
   614
            for (VariableElement pe : ee.getParameters()) {
jtulach@0
   615
                final String dn = pe.getSimpleName().toString();
jtulach@0
   616
                
jtulach@0
   617
                if (!verifyPropName(pe, dn, fixedProps)) {
jtulach@0
   618
                    ok = false;
jtulach@0
   619
                }
jtulach@0
   620
                
jtulach@0
   621
                final String dt = fqn(pe.asType(), ee);
jtulach@0
   622
                String[] call = toGetSet(dn, dt, false);
jaroslav@66
   623
                w.write("    " + dt + " arg" + (++arg) + " = ");
jtulach@0
   624
                w.write(call[0] + "();\n");
jtulach@0
   625
                
jtulach@0
   626
                Collection<String> depends = deps.get(dn);
jtulach@0
   627
                if (depends == null) {
jaroslav@100
   628
                    depends = new LinkedHashSet<String>();
jtulach@0
   629
                    deps.put(dn, depends);
jtulach@0
   630
                }
jtulach@0
   631
                depends.add(sn);
jtulach@0
   632
            }
jaroslav@66
   633
            w.write("    try {\n");
jaroslav@66
   634
            w.write("      locked = true;\n");
jaroslav@66
   635
            w.write("      return " + fqn(ee.getEnclosingElement().asType(), ee) + '.' + e.getSimpleName() + "(");
jtulach@0
   636
            String sep = "";
jtulach@0
   637
            for (int i = 1; i <= arg; i++) {
jtulach@0
   638
                w.write(sep);
jtulach@0
   639
                w.write("arg" + i);
jtulach@0
   640
                sep = ", ";
jtulach@0
   641
            }
jtulach@0
   642
            w.write(");\n");
jaroslav@66
   643
            w.write("    } finally {\n");
jaroslav@66
   644
            w.write("      locked = false;\n");
jaroslav@66
   645
            w.write("    }\n");
jtulach@0
   646
            w.write("  }\n");
jtulach@0
   647
jtulach@0
   648
            props.add(e.getSimpleName().toString());
jtulach@0
   649
            props.add(gs[2]);
jtulach@0
   650
            props.add(null);
jtulach@0
   651
            props.add(gs[0]);
jaroslav@17
   652
            props.add(tn);
jtulach@0
   653
        }
jtulach@0
   654
        
jtulach@0
   655
        return ok;
jtulach@0
   656
    }
jtulach@0
   657
jtulach@0
   658
    private static String[] toGetSet(String name, String type, boolean array) {
jtulach@0
   659
        String n = Character.toUpperCase(name.charAt(0)) + name.substring(1);
jtulach@0
   660
        String bck2brwsrType = "L" + type.replace('.', '_') + "_2";
jtulach@0
   661
        if ("int".equals(type)) {
jtulach@0
   662
            bck2brwsrType = "I";
jtulach@0
   663
        }
jtulach@0
   664
        if ("double".equals(type)) {
jtulach@0
   665
            bck2brwsrType = "D";
jtulach@0
   666
        }
jtulach@0
   667
        String pref = "get";
jtulach@0
   668
        if ("boolean".equals(type)) {
jtulach@0
   669
            pref = "is";
jtulach@0
   670
            bck2brwsrType = "Z";
jtulach@0
   671
        }
jtulach@0
   672
        final String nu = n.replace('.', '_');
jtulach@0
   673
        if (array) {
jtulach@0
   674
            return new String[] { 
jtulach@0
   675
                "get" + n,
jtulach@0
   676
                null,
jtulach@0
   677
                "get" + nu + "__Ljava_util_List_2",
jtulach@0
   678
                null
jtulach@0
   679
            };
jtulach@0
   680
        }
jtulach@0
   681
        return new String[]{
jtulach@0
   682
            pref + n, 
jtulach@0
   683
            "set" + n, 
jtulach@0
   684
            pref + nu + "__" + bck2brwsrType,
jtulach@0
   685
            "set" + nu + "__V" + bck2brwsrType
jtulach@0
   686
        };
jtulach@0
   687
    }
jtulach@0
   688
jtulach@0
   689
    private String typeName(Element where, Prprt p) {
jtulach@0
   690
        String ret;
jtulach@0
   691
        boolean[] isModel = { false };
jtulach@0
   692
        boolean[] isEnum = { false };
jtulach@0
   693
        boolean isPrimitive[] = { false };
jtulach@0
   694
        ret = checkType(p, isModel, isEnum, isPrimitive);
jtulach@0
   695
        if (p.array()) {
jtulach@0
   696
            String bt = findBoxedType(ret);
jtulach@0
   697
            if (bt != null) {
jtulach@0
   698
                return bt;
jtulach@0
   699
            }
jtulach@0
   700
        }
jtulach@0
   701
        return ret;
jtulach@0
   702
    }
jtulach@0
   703
    
jtulach@0
   704
    private static String findBoxedType(String ret) {
jtulach@0
   705
        if (ret.equals("boolean")) {
jtulach@0
   706
            return Boolean.class.getName();
jtulach@0
   707
        }
jtulach@0
   708
        if (ret.equals("byte")) {
jtulach@0
   709
            return Byte.class.getName();
jtulach@0
   710
        }
jtulach@0
   711
        if (ret.equals("short")) {
jtulach@0
   712
            return Short.class.getName();
jtulach@0
   713
        }
jtulach@0
   714
        if (ret.equals("char")) {
jtulach@0
   715
            return Character.class.getName();
jtulach@0
   716
        }
jtulach@0
   717
        if (ret.equals("int")) {
jtulach@0
   718
            return Integer.class.getName();
jtulach@0
   719
        }
jtulach@0
   720
        if (ret.equals("long")) {
jtulach@0
   721
            return Long.class.getName();
jtulach@0
   722
        }
jtulach@0
   723
        if (ret.equals("float")) {
jtulach@0
   724
            return Float.class.getName();
jtulach@0
   725
        }
jtulach@0
   726
        if (ret.equals("double")) {
jtulach@0
   727
            return Double.class.getName();
jtulach@0
   728
        }
jtulach@0
   729
        return null;
jtulach@0
   730
    }
jtulach@0
   731
jtulach@0
   732
    private boolean verifyPropName(Element e, String propName, Prprt[] existingProps) {
jtulach@0
   733
        StringBuilder sb = new StringBuilder();
jtulach@0
   734
        String sep = "";
jtulach@0
   735
        for (Prprt Prprt : existingProps) {
jtulach@0
   736
            if (Prprt.name().equals(propName)) {
jtulach@0
   737
                return true;
jtulach@0
   738
            }
jtulach@0
   739
            sb.append(sep);
jtulach@0
   740
            sb.append('"');
jtulach@0
   741
            sb.append(Prprt.name());
jtulach@0
   742
            sb.append('"');
jtulach@0
   743
            sep = ", ";
jtulach@0
   744
        }
jtulach@0
   745
        error(
jtulach@0
   746
            propName + " is not one of known properties: " + sb
jtulach@0
   747
            , e
jtulach@0
   748
        );
jtulach@0
   749
        return false;
jtulach@0
   750
    }
jtulach@0
   751
jtulach@0
   752
    private static String findPkgName(Element e) {
jtulach@0
   753
        for (;;) {
jtulach@0
   754
            if (e.getKind() == ElementKind.PACKAGE) {
jtulach@0
   755
                return ((PackageElement)e).getQualifiedName().toString();
jtulach@0
   756
            }
jtulach@0
   757
            e = e.getEnclosingElement();
jtulach@0
   758
        }
jtulach@0
   759
    }
jtulach@0
   760
jtulach@0
   761
    private boolean generateFunctions(
jtulach@0
   762
        Element clazz, StringWriter body, String className, 
jtulach@0
   763
        List<? extends Element> enclosedElements, List<String> functions
jtulach@0
   764
    ) {
jtulach@0
   765
        for (Element m : enclosedElements) {
jtulach@0
   766
            if (m.getKind() != ElementKind.METHOD) {
jtulach@0
   767
                continue;
jtulach@0
   768
            }
jtulach@0
   769
            ExecutableElement e = (ExecutableElement)m;
jtulach@0
   770
            Function onF = e.getAnnotation(Function.class);
jtulach@0
   771
            if (onF == null) {
jtulach@0
   772
                continue;
jtulach@0
   773
            }
jtulach@0
   774
            if (!e.getModifiers().contains(Modifier.STATIC)) {
jtulach@0
   775
                error("@OnFunction method needs to be static", e);
jtulach@0
   776
                return false;
jtulach@0
   777
            }
jtulach@0
   778
            if (e.getModifiers().contains(Modifier.PRIVATE)) {
jtulach@0
   779
                error("@OnFunction method cannot be private", e);
jtulach@0
   780
                return false;
jtulach@0
   781
            }
jtulach@0
   782
            if (e.getReturnType().getKind() != TypeKind.VOID) {
jtulach@0
   783
                error("@OnFunction method should return void", e);
jtulach@0
   784
                return false;
jtulach@0
   785
            }
jtulach@0
   786
            String n = e.getSimpleName().toString();
jaroslav@66
   787
            body.append("  private void ").append(n).append("(Object data, Object ev) {\n");
jaroslav@199
   788
            body.append("    ").append(((TypeElement)clazz).getQualifiedName()).append(".").append(n).append("(");
jtulach@0
   789
            body.append(wrapParams(e, null, className, "ev", "data"));
jtulach@0
   790
            body.append(");\n");
jaroslav@66
   791
            body.append("  }\n");
jtulach@0
   792
            
jtulach@0
   793
            functions.add(n);
jtulach@0
   794
            functions.add(n + "__VLjava_lang_Object_2Ljava_lang_Object_2");
jtulach@0
   795
        }
jtulach@0
   796
        return true;
jtulach@0
   797
    }
jtulach@0
   798
jtulach@0
   799
    private boolean generateOnChange(Element clazz, Map<String,Collection<String>> propDeps,
jtulach@0
   800
        Prprt[] properties, String className, 
jtulach@0
   801
        Map<String, Collection<String>> functionDeps
jtulach@0
   802
    ) {
jtulach@0
   803
        for (Element m : clazz.getEnclosedElements()) {
jtulach@0
   804
            if (m.getKind() != ElementKind.METHOD) {
jtulach@0
   805
                continue;
jtulach@0
   806
            }
jtulach@0
   807
            ExecutableElement e = (ExecutableElement) m;
jtulach@0
   808
            OnPropertyChange onPC = e.getAnnotation(OnPropertyChange.class);
jtulach@0
   809
            if (onPC == null) {
jtulach@0
   810
                continue;
jtulach@0
   811
            }
jtulach@0
   812
            for (String pn : onPC.value()) {
jtulach@0
   813
                if (findPrprt(properties, pn) == null && findDerivedFrom(propDeps, pn).isEmpty()) {
jtulach@0
   814
                    error("No Prprt named '" + pn + "' in the model", clazz);
jtulach@0
   815
                    return false;
jtulach@0
   816
                }
jtulach@0
   817
            }
jtulach@0
   818
            if (!e.getModifiers().contains(Modifier.STATIC)) {
jtulach@0
   819
                error("@OnPrprtChange method needs to be static", e);
jtulach@0
   820
                return false;
jtulach@0
   821
            }
jtulach@0
   822
            if (e.getModifiers().contains(Modifier.PRIVATE)) {
jtulach@0
   823
                error("@OnPrprtChange method cannot be private", e);
jtulach@0
   824
                return false;
jtulach@0
   825
            }
jtulach@0
   826
            if (e.getReturnType().getKind() != TypeKind.VOID) {
jtulach@0
   827
                error("@OnPrprtChange method should return void", e);
jtulach@0
   828
                return false;
jtulach@0
   829
            }
jtulach@0
   830
            String n = e.getSimpleName().toString();
jtulach@0
   831
            
jtulach@0
   832
            
jtulach@0
   833
            for (String pn : onPC.value()) {
jtulach@0
   834
                StringBuilder call = new StringBuilder();
jtulach@0
   835
                call.append("  ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jtulach@0
   836
                call.append(wrapPropName(e, className, "name", pn));
jtulach@0
   837
                call.append(");\n");
jtulach@0
   838
                
jtulach@0
   839
                Collection<String> change = functionDeps.get(pn);
jtulach@0
   840
                if (change == null) {
jaroslav@100
   841
                    change = new ArrayList<String>();
jtulach@0
   842
                    functionDeps.put(pn, change);
jtulach@0
   843
                }
jtulach@0
   844
                change.add(call.toString());
jtulach@0
   845
                for (String dpn : findDerivedFrom(propDeps, pn)) {
jtulach@0
   846
                    change = functionDeps.get(dpn);
jtulach@0
   847
                    if (change == null) {
jaroslav@100
   848
                        change = new ArrayList<String>();
jtulach@0
   849
                        functionDeps.put(dpn, change);
jtulach@0
   850
                    }
jtulach@0
   851
                    change.add(call.toString());
jtulach@0
   852
                }
jtulach@0
   853
            }
jtulach@0
   854
        }
jtulach@0
   855
        return true;
jtulach@0
   856
    }
jaroslav@240
   857
jaroslav@240
   858
    private boolean generateOperation(Element clazz, 
jaroslav@240
   859
        StringWriter body, String className, 
jaroslav@240
   860
        List<? extends Element> enclosedElements
jaroslav@240
   861
    ) {
jaroslav@240
   862
        for (Element m : enclosedElements) {
jaroslav@240
   863
            if (m.getKind() != ElementKind.METHOD) {
jaroslav@240
   864
                continue;
jaroslav@240
   865
            }
jaroslav@240
   866
            ExecutableElement e = (ExecutableElement)m;
jaroslav@240
   867
            ModelOperation mO = e.getAnnotation(ModelOperation.class);
jaroslav@240
   868
            if (mO == null) {
jaroslav@240
   869
                continue;
jaroslav@240
   870
            }
jaroslav@240
   871
            if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@240
   872
                error("@ModelOperation method needs to be static", e);
jaroslav@240
   873
                return false;
jaroslav@240
   874
            }
jaroslav@240
   875
            if (e.getModifiers().contains(Modifier.PRIVATE)) {
jaroslav@240
   876
                error("@ModelOperation method cannot be private", e);
jaroslav@240
   877
                return false;
jaroslav@240
   878
            }
jaroslav@240
   879
            if (e.getReturnType().getKind() != TypeKind.VOID) {
jaroslav@240
   880
                error("@ModelOperation method should return void", e);
jaroslav@240
   881
                return false;
jaroslav@240
   882
            }
jaroslav@240
   883
            List<String> args = new ArrayList<String>();
jaroslav@240
   884
            {
jaroslav@240
   885
                body.append("  public void ").append(m.getSimpleName()).append("(");
jaroslav@240
   886
                String sep = "";
jaroslav@240
   887
                boolean checkFirst = true;
jaroslav@240
   888
                for (VariableElement ve : e.getParameters()) {
jaroslav@240
   889
                    final TypeMirror type = ve.asType();
jaroslav@240
   890
                    CharSequence simpleName;
jaroslav@240
   891
                    if (type.getKind() == TypeKind.DECLARED) {
jaroslav@240
   892
                        simpleName = ((DeclaredType)type).asElement().getSimpleName();
jaroslav@240
   893
                    } else {
jaroslav@240
   894
                        simpleName = type.toString();
jaroslav@240
   895
                    }
jaroslav@240
   896
                    if (simpleName.toString().equals(className)) {
jaroslav@240
   897
                        checkFirst = false;
jaroslav@240
   898
                    } else {
jaroslav@240
   899
                        if (checkFirst) {
jaroslav@240
   900
                            error("First parameter of @ModelOperation method must be " + className, m);
jaroslav@240
   901
                            return false;
jaroslav@240
   902
                        }
jaroslav@240
   903
                        args.add(ve.getSimpleName().toString());
jaroslav@240
   904
                        body.append(sep).append("final ");
jaroslav@240
   905
                        body.append(ve.asType().toString()).append(" ");
jaroslav@240
   906
                        body.append(ve.toString());
jaroslav@240
   907
                        sep = ", ";
jaroslav@240
   908
                    }
jaroslav@240
   909
                }
jaroslav@240
   910
                body.append(") {\n");
jaroslav@362
   911
                body.append("    org.netbeans.html.json.impl.JSON.runInBrowser(this.context, new Runnable() { public void run() {\n");
jaroslav@240
   912
                body.append("      ").append(clazz.getSimpleName()).append(".").append(m.getSimpleName()).append("(");
jaroslav@240
   913
                body.append(className).append(".this");
jaroslav@240
   914
                for (String s : args) {
jaroslav@240
   915
                    body.append(", ").append(s);
jaroslav@240
   916
                }
jaroslav@240
   917
                body.append(");\n");
jaroslav@240
   918
                body.append("    }});\n");
jaroslav@240
   919
                body.append("  }\n");
jaroslav@240
   920
            }
jaroslav@240
   921
            
jaroslav@240
   922
        }
jaroslav@240
   923
        return true;
jaroslav@240
   924
    }
jaroslav@240
   925
    
jtulach@0
   926
    
jtulach@0
   927
    private boolean generateReceive(
jtulach@0
   928
        Element clazz, StringWriter body, String className, 
jtulach@0
   929
        List<? extends Element> enclosedElements, List<String> functions
jtulach@0
   930
    ) {
jtulach@0
   931
        for (Element m : enclosedElements) {
jtulach@0
   932
            if (m.getKind() != ElementKind.METHOD) {
jtulach@0
   933
                continue;
jtulach@0
   934
            }
jtulach@0
   935
            ExecutableElement e = (ExecutableElement)m;
jtulach@0
   936
            OnReceive onR = e.getAnnotation(OnReceive.class);
jtulach@0
   937
            if (onR == null) {
jtulach@0
   938
                continue;
jtulach@0
   939
            }
jtulach@0
   940
            if (!e.getModifiers().contains(Modifier.STATIC)) {
jtulach@0
   941
                error("@OnReceive method needs to be static", e);
jtulach@0
   942
                return false;
jtulach@0
   943
            }
jtulach@0
   944
            if (e.getModifiers().contains(Modifier.PRIVATE)) {
jtulach@0
   945
                error("@OnReceive method cannot be private", e);
jtulach@0
   946
                return false;
jtulach@0
   947
            }
jtulach@0
   948
            if (e.getReturnType().getKind() != TypeKind.VOID) {
jtulach@0
   949
                error("@OnReceive method should return void", e);
jtulach@0
   950
                return false;
jtulach@0
   951
            }
jaroslav@73
   952
            if (!onR.jsonp().isEmpty() && !"GET".equals(onR.method())) {
jaroslav@73
   953
                error("JSONP works only with GET transport method", e);
jaroslav@73
   954
            }
jaroslav@73
   955
            String dataMirror = findDataSpecified(e, onR);
jaroslav@72
   956
            if ("PUT".equals(onR.method()) && dataMirror == null) {
jaroslav@70
   957
                error("PUT method needs to specify a data() class", e);
jaroslav@70
   958
                return false;
jaroslav@70
   959
            }
jaroslav@72
   960
            if ("POST".equals(onR.method()) && dataMirror == null) {
jaroslav@70
   961
                error("POST method needs to specify a data() class", e);
jaroslav@70
   962
                return false;
jaroslav@70
   963
            }
jtulach@0
   964
            String modelClass = null;
jtulach@0
   965
            boolean expectsList = false;
jaroslav@100
   966
            List<String> args = new ArrayList<String>();
jtulach@0
   967
            {
jtulach@0
   968
                for (VariableElement ve : e.getParameters()) {
jtulach@0
   969
                    TypeMirror modelType = null;
jaroslav@29
   970
                    final TypeMirror type = ve.asType();
jaroslav@29
   971
                    CharSequence simpleName;
jaroslav@29
   972
                    if (type.getKind() == TypeKind.DECLARED) {
jaroslav@29
   973
                        simpleName = ((DeclaredType)type).asElement().getSimpleName();
jaroslav@29
   974
                    } else {
jaroslav@29
   975
                        simpleName = type.toString();
jaroslav@29
   976
                    }
jaroslav@29
   977
                    if (simpleName.toString().equals(className)) {
jtulach@0
   978
                        args.add(className + ".this");
jtulach@0
   979
                    } else if (isModel(ve.asType())) {
jtulach@0
   980
                        modelType = ve.asType();
jtulach@0
   981
                    } else if (ve.asType().getKind() == TypeKind.ARRAY) {
jtulach@0
   982
                        modelType = ((ArrayType)ve.asType()).getComponentType();
jtulach@0
   983
                        expectsList = true;
jaroslav@72
   984
                    } else if (ve.asType().toString().equals("java.lang.String")) {
jaroslav@72
   985
                        modelType = ve.asType();
jtulach@0
   986
                    }
jtulach@0
   987
                    if (modelType != null) {
jtulach@0
   988
                        if (modelClass != null) {
jtulach@0
   989
                            error("There can be only one model class among arguments", e);
jtulach@0
   990
                        } else {
jtulach@0
   991
                            modelClass = modelType.toString();
jtulach@0
   992
                            if (expectsList) {
jtulach@0
   993
                                args.add("arr");
jtulach@0
   994
                            } else {
jtulach@0
   995
                                args.add("arr[0]");
jtulach@0
   996
                            }
jtulach@0
   997
                        }
jtulach@0
   998
                    }
jtulach@0
   999
                }
jtulach@0
  1000
            }
jtulach@0
  1001
            if (modelClass == null) {
jtulach@0
  1002
                error("The method needs to have one @Model class as parameter", e);
jtulach@0
  1003
            }
jtulach@0
  1004
            String n = e.getSimpleName().toString();
jaroslav@250
  1005
            if ("WebSocket".equals(onR.method())) {
jaroslav@250
  1006
                body.append("  /** Performs WebSocket communication. Call with <code>null</code> data parameter\n");
jaroslav@250
  1007
                body.append("  * to open the connection (even if not required). Call with non-null data to\n");
jaroslav@250
  1008
                body.append("  * send messages to server. Call again with <code>null</code> data to close the socket.\n");
jaroslav@250
  1009
                body.append("  */\n");
jaroslav@250
  1010
            }
jaroslav@66
  1011
            body.append("  public void ").append(n).append("(");
jaroslav@24
  1012
            StringBuilder urlBefore = new StringBuilder();
jaroslav@24
  1013
            StringBuilder urlAfter = new StringBuilder();
jtulach@0
  1014
            String jsonpVarName = null;
jtulach@0
  1015
            {
jtulach@0
  1016
                String sep = "";
jtulach@0
  1017
                boolean skipJSONP = onR.jsonp().isEmpty();
jaroslav@24
  1018
                for (String p : findParamNames(e, onR.url(), onR.jsonp(), urlBefore, urlAfter)) {
jtulach@0
  1019
                    if (!skipJSONP && p.equals(onR.jsonp())) {
jtulach@0
  1020
                        skipJSONP = true;
jtulach@0
  1021
                        jsonpVarName = p;
jtulach@0
  1022
                        continue;
jtulach@0
  1023
                    }
jtulach@0
  1024
                    body.append(sep);
jtulach@0
  1025
                    body.append("String ").append(p);
jtulach@0
  1026
                    sep = ", ";
jtulach@0
  1027
                }
jtulach@0
  1028
                if (!skipJSONP) {
jtulach@0
  1029
                    error(
jtulach@0
  1030
                        "Name of jsonp attribute ('" + onR.jsonp() + 
jtulach@0
  1031
                        "') is not used in url attribute '" + onR.url() + "'", e
jtulach@0
  1032
                    );
jtulach@0
  1033
                }
jaroslav@73
  1034
                if (dataMirror != null) {
jaroslav@73
  1035
                    body.append(sep).append(dataMirror.toString()).append(" data");
jaroslav@73
  1036
                }
jtulach@0
  1037
            }
jtulach@0
  1038
            body.append(") {\n");
jaroslav@258
  1039
            boolean webSocket = onR.method().equals("WebSocket");
jaroslav@258
  1040
            if (webSocket) {
jaroslav@258
  1041
                if (generateWSReceiveBody(body, onR, e, clazz, className, expectsList, modelClass, n, args, urlBefore, jsonpVarName, urlAfter, dataMirror)) {
jaroslav@258
  1042
                    return false;
jaroslav@258
  1043
                }
jaroslav@258
  1044
                body.append("  }\n");
jaroslav@362
  1045
                body.append("  private org.netbeans.html.json.impl.JSON.WS ws_" + e.getSimpleName() + ";\n");
jaroslav@258
  1046
            } else {
jaroslav@258
  1047
                if (generateJSONReceiveBody(body, onR, e, clazz, className, expectsList, modelClass, n, args, urlBefore, jsonpVarName, urlAfter, dataMirror)) {
jaroslav@258
  1048
                    return false;
jaroslav@258
  1049
                }
jaroslav@258
  1050
                body.append("  }\n");
jaroslav@258
  1051
            }
jaroslav@258
  1052
        }
jaroslav@258
  1053
        return true;
jaroslav@258
  1054
    }
jaroslav@258
  1055
jaroslav@258
  1056
    private boolean generateJSONReceiveBody(StringWriter body, OnReceive onR, ExecutableElement e, Element clazz, String className, boolean expectsList, String modelClass, String n, List<String> args, StringBuilder urlBefore, String jsonpVarName, StringBuilder urlAfter, String dataMirror) {
jaroslav@258
  1057
        body.append(
jaroslav@362
  1058
            "    class ProcessResult extends org.netbeans.html.json.impl.RcvrJSON {\n" +
jaroslav@258
  1059
            "      @Override\n" +
jaroslav@362
  1060
            "      public void onError(org.netbeans.html.json.impl.RcvrJSON.MsgEvnt ev) {\n" +
jaroslav@258
  1061
            "        Exception value = ev.getException();\n"
jaroslav@258
  1062
            );
jaroslav@258
  1063
        if (onR.onError().isEmpty()) {
jtulach@0
  1064
            body.append(
jaroslav@255
  1065
                "        value.printStackTrace();\n"
jaroslav@245
  1066
                );
jaroslav@258
  1067
        } else {
jaroslav@258
  1068
            if (!findOnError(e, ((TypeElement)clazz), onR.onError(), className)) {
jaroslav@258
  1069
                return true;
jaroslav@245
  1070
            }
jaroslav@258
  1071
            body.append("        ").append(clazz.getSimpleName()).append(".").append(onR.onError()).append("(");
jaroslav@258
  1072
            body.append(className).append(".this, value);\n");
jaroslav@258
  1073
        }
jaroslav@258
  1074
        body.append(
jaroslav@258
  1075
            "      }\n" +
jaroslav@258
  1076
            "      @Override\n" +
jaroslav@362
  1077
            "      public void onMessage(org.netbeans.html.json.impl.RcvrJSON.MsgEvnt ev) {\n"
jaroslav@258
  1078
            );
jaroslav@258
  1079
        if (expectsList) {
jaroslav@245
  1080
            body.append(
jaroslav@258
  1081
                "        " + modelClass + "[] arr = new " + modelClass + "[ev.dataSize()];\n"
jaroslav@258
  1082
                );
jaroslav@258
  1083
        } else {
jaroslav@258
  1084
            body.append(
jaroslav@258
  1085
                "        " + modelClass + "[] arr = { null };\n"
jaroslav@258
  1086
                );
jaroslav@258
  1087
        }
jaroslav@258
  1088
        body.append(
jaroslav@258
  1089
            "        ev.dataRead(context, " + modelClass + ".class, arr);\n"
jaroslav@245
  1090
            );
jaroslav@258
  1091
        {
jaroslav@258
  1092
            body.append("        ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@258
  1093
            String sep = "";
jaroslav@258
  1094
            for (String arg : args) {
jaroslav@258
  1095
                body.append(sep);
jaroslav@258
  1096
                body.append(arg);
jaroslav@258
  1097
                sep = ", ";
jaroslav@74
  1098
            }
jaroslav@24
  1099
            body.append(");\n");
jtulach@0
  1100
        }
jaroslav@258
  1101
        body.append(
jaroslav@258
  1102
            "      }\n" +
jaroslav@258
  1103
            "    }\n"
jaroslav@258
  1104
            );
jaroslav@258
  1105
        body.append("    ProcessResult pr = new ProcessResult();\n");
jaroslav@362
  1106
        body.append("    org.netbeans.html.json.impl.JSON.loadJSON(context, pr,\n        ");
jaroslav@258
  1107
        body.append(urlBefore).append(", ");
jaroslav@258
  1108
        if (jsonpVarName != null) {
jaroslav@258
  1109
            body.append(urlAfter);
jaroslav@258
  1110
        } else {
jaroslav@258
  1111
            body.append("null");
jaroslav@258
  1112
        }
jaroslav@258
  1113
        if (!"GET".equals(onR.method()) || dataMirror != null) {
jaroslav@258
  1114
            body.append(", \"").append(onR.method()).append('"');
jaroslav@258
  1115
            if (dataMirror != null) {
jaroslav@258
  1116
                body.append(", data");
jaroslav@258
  1117
            } else {
jaroslav@258
  1118
                body.append(", null");
jaroslav@258
  1119
            }
jaroslav@258
  1120
        } else {
jaroslav@258
  1121
            body.append(", null, null");
jaroslav@258
  1122
        }
jaroslav@258
  1123
        body.append(");\n");
jaroslav@258
  1124
        return false;
jaroslav@258
  1125
    }
jaroslav@258
  1126
    
jaroslav@258
  1127
    private boolean generateWSReceiveBody(StringWriter body, OnReceive onR, ExecutableElement e, Element clazz, String className, boolean expectsList, String modelClass, String n, List<String> args, StringBuilder urlBefore, String jsonpVarName, StringBuilder urlAfter, String dataMirror) {
jaroslav@258
  1128
        body.append(
jaroslav@362
  1129
            "    class ProcessResult extends org.netbeans.html.json.impl.RcvrJSON {\n" +
jaroslav@258
  1130
            "      @Override\n" +
jaroslav@362
  1131
            "      public void onOpen(org.netbeans.html.json.impl.RcvrJSON.MsgEvnt ev) {\n"
jaroslav@258
  1132
        );
jaroslav@258
  1133
        body.append("        ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@258
  1134
        {
jaroslav@258
  1135
            String sep = "";
jaroslav@258
  1136
            for (String arg : args) {
jaroslav@258
  1137
                body.append(sep);
jaroslav@258
  1138
                if (arg.startsWith("arr")) {
jaroslav@258
  1139
                    body.append("null");
jaroslav@258
  1140
                } else {
jaroslav@258
  1141
                    body.append(arg);
jaroslav@258
  1142
                }
jaroslav@258
  1143
                sep = ", ";
jaroslav@258
  1144
            }
jaroslav@258
  1145
        }
jaroslav@258
  1146
        body.append(");\n");
jaroslav@258
  1147
        body.append(
jaroslav@258
  1148
            "      }\n" +
jaroslav@258
  1149
            "      @Override\n" +
jaroslav@362
  1150
            "      public void onError(org.netbeans.html.json.impl.RcvrJSON.MsgEvnt ev) {\n" +
jaroslav@258
  1151
            "        Exception value = ev.getException();\n"
jaroslav@258
  1152
            );
jaroslav@258
  1153
        if (onR.onError().isEmpty()) {
jaroslav@258
  1154
            body.append(
jaroslav@258
  1155
                "        value.printStackTrace();\n"
jaroslav@258
  1156
                );
jaroslav@258
  1157
        } else {
jaroslav@258
  1158
            if (!findOnError(e, ((TypeElement)clazz), onR.onError(), className)) {
jaroslav@258
  1159
                return true;
jaroslav@258
  1160
            }
jaroslav@258
  1161
            body.append("        ").append(clazz.getSimpleName()).append(".").append(onR.onError()).append("(");
jaroslav@258
  1162
            body.append(className).append(".this, value);\n");
jaroslav@258
  1163
        }
jaroslav@258
  1164
        body.append(
jaroslav@258
  1165
            "      }\n" +
jaroslav@258
  1166
            "      @Override\n" +
jaroslav@362
  1167
            "      public void onMessage(org.netbeans.html.json.impl.RcvrJSON.MsgEvnt ev) {\n"
jaroslav@258
  1168
        );
jaroslav@258
  1169
        if (expectsList) {
jaroslav@258
  1170
            body.append(
jaroslav@258
  1171
                "        " + modelClass + "[] arr = new " + modelClass + "[ev.dataSize()];\n"
jaroslav@258
  1172
                );
jaroslav@258
  1173
        } else {
jaroslav@258
  1174
            body.append(
jaroslav@258
  1175
                "        " + modelClass + "[] arr = { null };\n"
jaroslav@258
  1176
                );
jaroslav@258
  1177
        }
jaroslav@258
  1178
        body.append(
jaroslav@258
  1179
            "        ev.dataRead(context, " + modelClass + ".class, arr);\n"
jaroslav@258
  1180
            );
jaroslav@258
  1181
        {
jaroslav@258
  1182
            body.append("        ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@258
  1183
            String sep = "";
jaroslav@258
  1184
            for (String arg : args) {
jaroslav@258
  1185
                body.append(sep);
jaroslav@258
  1186
                body.append(arg);
jaroslav@258
  1187
                sep = ", ";
jaroslav@258
  1188
            }
jaroslav@258
  1189
            body.append(");\n");
jaroslav@258
  1190
        }
jaroslav@258
  1191
        body.append(
jaroslav@258
  1192
            "      }\n"
jaroslav@258
  1193
        );
jaroslav@258
  1194
        if (!onR.onError().isEmpty()) {
jaroslav@258
  1195
            body.append(
jaroslav@258
  1196
                "      @Override\n"
jaroslav@362
  1197
              + "      public void onClose(org.netbeans.html.json.impl.RcvrJSON.MsgEvnt ev) {\n"
jaroslav@258
  1198
            );
jaroslav@258
  1199
            body.append("        ").append(clazz.getSimpleName()).append(".").append(onR.onError()).append("(");
jaroslav@258
  1200
            body.append(className).append(".this, null);\n");
jaroslav@258
  1201
            body.append(
jaroslav@258
  1202
                "      }\n"
jaroslav@258
  1203
            );
jaroslav@258
  1204
        }
jaroslav@258
  1205
        body.append("    }\n");
jaroslav@258
  1206
        body.append("    if (this.ws_").append(e.getSimpleName()).append(" == null) {\n");
jaroslav@258
  1207
        body.append("      ProcessResult pr = new ProcessResult();\n");
jaroslav@258
  1208
        body.append("      this.ws_").append(e.getSimpleName());
jaroslav@362
  1209
        body.append("= org.netbeans.html.json.impl.JSON.openWS(context, pr,\n        ");
jaroslav@258
  1210
        body.append(urlBefore).append(", data);\n");
jaroslav@258
  1211
        body.append("    } else {\n");
jaroslav@262
  1212
        body.append("      this.ws_").append(e.getSimpleName()).append(".send(context, ").append(urlBefore).append(", data);\n");
jaroslav@258
  1213
        body.append("    }\n");
jaroslav@258
  1214
        return false;
jtulach@0
  1215
    }
jtulach@0
  1216
jtulach@0
  1217
    private CharSequence wrapParams(
jtulach@0
  1218
        ExecutableElement ee, String id, String className, String evName, String dataName
jtulach@0
  1219
    ) {
jtulach@0
  1220
        TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jtulach@0
  1221
        StringBuilder params = new StringBuilder();
jtulach@0
  1222
        boolean first = true;
jtulach@0
  1223
        for (VariableElement ve : ee.getParameters()) {
jtulach@0
  1224
            if (!first) {
jtulach@0
  1225
                params.append(", ");
jtulach@0
  1226
            }
jtulach@0
  1227
            first = false;
jtulach@0
  1228
            String toCall = null;
jaroslav@22
  1229
            String toFinish = null;
jtulach@0
  1230
            if (ve.asType() == stringType) {
jtulach@0
  1231
                if (ve.getSimpleName().contentEquals("id")) {
jtulach@0
  1232
                    params.append('"').append(id).append('"');
jtulach@0
  1233
                    continue;
jtulach@0
  1234
                }
jaroslav@362
  1235
                toCall = "org.netbeans.html.json.impl.JSON.toString(context, ";
jtulach@0
  1236
            }
jtulach@0
  1237
            if (ve.asType().getKind() == TypeKind.DOUBLE) {
jaroslav@362
  1238
                toCall = "org.netbeans.html.json.impl.JSON.toNumber(context, ";
jaroslav@22
  1239
                toFinish = ".doubleValue()";
jtulach@0
  1240
            }
jtulach@0
  1241
            if (ve.asType().getKind() == TypeKind.INT) {
jaroslav@362
  1242
                toCall = "org.netbeans.html.json.impl.JSON.toNumber(context, ";
jaroslav@22
  1243
                toFinish = ".intValue()";
jtulach@0
  1244
            }
jtulach@0
  1245
            if (dataName != null && ve.getSimpleName().contentEquals(dataName) && isModel(ve.asType())) {
jaroslav@362
  1246
                toCall = "org.netbeans.html.json.impl.JSON.toModel(context, " + ve.asType() + ".class, ";
jtulach@0
  1247
            }
jtulach@0
  1248
jtulach@0
  1249
            if (toCall != null) {
jtulach@0
  1250
                params.append(toCall);
jtulach@0
  1251
                if (dataName != null && ve.getSimpleName().contentEquals(dataName)) {
jtulach@0
  1252
                    params.append(dataName);
jtulach@0
  1253
                    params.append(", null");
jtulach@0
  1254
                } else {
jtulach@0
  1255
                    if (evName == null) {
jtulach@0
  1256
                        final StringBuilder sb = new StringBuilder();
jtulach@0
  1257
                        sb.append("Unexpected string parameter name.");
jtulach@0
  1258
                        if (dataName != null) {
jtulach@0
  1259
                            sb.append(" Try \"").append(dataName).append("\"");
jtulach@0
  1260
                        }
jtulach@0
  1261
                        error(sb.toString(), ee);
jtulach@0
  1262
                    }
jtulach@0
  1263
                    params.append(evName);
jtulach@0
  1264
                    params.append(", \"");
jtulach@0
  1265
                    params.append(ve.getSimpleName().toString());
jtulach@0
  1266
                    params.append("\"");
jtulach@0
  1267
                }
jtulach@0
  1268
                params.append(")");
jaroslav@22
  1269
                if (toFinish != null) {
jaroslav@22
  1270
                    params.append(toFinish);
jaroslav@22
  1271
                }
jtulach@0
  1272
                continue;
jtulach@0
  1273
            }
jtulach@0
  1274
            String rn = fqn(ve.asType(), ee);
jtulach@0
  1275
            int last = rn.lastIndexOf('.');
jtulach@0
  1276
            if (last >= 0) {
jtulach@0
  1277
                rn = rn.substring(last + 1);
jtulach@0
  1278
            }
jtulach@0
  1279
            if (rn.equals(className)) {
jtulach@0
  1280
                params.append(className).append(".this");
jtulach@0
  1281
                continue;
jtulach@0
  1282
            }
jtulach@0
  1283
            error(
jaroslav@235
  1284
                "The annotated method can only accept " + className + " argument or argument named 'data'",
jtulach@0
  1285
                ee
jtulach@0
  1286
            );
jtulach@0
  1287
        }
jtulach@0
  1288
        return params;
jtulach@0
  1289
    }
jtulach@0
  1290
    
jtulach@0
  1291
    
jtulach@0
  1292
    private CharSequence wrapPropName(
jtulach@0
  1293
        ExecutableElement ee, String className, String propName, String propValue
jtulach@0
  1294
    ) {
jtulach@0
  1295
        TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jtulach@0
  1296
        StringBuilder params = new StringBuilder();
jtulach@0
  1297
        boolean first = true;
jtulach@0
  1298
        for (VariableElement ve : ee.getParameters()) {
jtulach@0
  1299
            if (!first) {
jtulach@0
  1300
                params.append(", ");
jtulach@0
  1301
            }
jtulach@0
  1302
            first = false;
jtulach@0
  1303
            if (ve.asType() == stringType) {
jtulach@0
  1304
                if (propName != null && ve.getSimpleName().contentEquals(propName)) {
jtulach@0
  1305
                    params.append('"').append(propValue).append('"');
jtulach@0
  1306
                } else {
jtulach@0
  1307
                    error("Unexpected string parameter name. Try \"" + propName + "\".", ee);
jtulach@0
  1308
                }
jtulach@0
  1309
                continue;
jtulach@0
  1310
            }
jtulach@0
  1311
            String rn = fqn(ve.asType(), ee);
jtulach@0
  1312
            int last = rn.lastIndexOf('.');
jtulach@0
  1313
            if (last >= 0) {
jtulach@0
  1314
                rn = rn.substring(last + 1);
jtulach@0
  1315
            }
jtulach@0
  1316
            if (rn.equals(className)) {
jtulach@0
  1317
                params.append(className).append(".this");
jtulach@0
  1318
                continue;
jtulach@0
  1319
            }
jtulach@0
  1320
            error(
jtulach@0
  1321
                "@OnPrprtChange method can only accept String or " + className + " arguments",
jtulach@0
  1322
                ee);
jtulach@0
  1323
        }
jtulach@0
  1324
        return params;
jtulach@0
  1325
    }
jtulach@0
  1326
    
jtulach@0
  1327
    private boolean isModel(TypeMirror tm) {
jaroslav@13
  1328
        if (tm.getKind() == TypeKind.ERROR) {
jaroslav@13
  1329
            return true;
jaroslav@13
  1330
        }
jtulach@0
  1331
        final Element e = processingEnv.getTypeUtils().asElement(tm);
jtulach@0
  1332
        if (e == null) {
jtulach@0
  1333
            return false;
jtulach@0
  1334
        }
jtulach@0
  1335
        for (Element ch : e.getEnclosedElements()) {
jtulach@0
  1336
            if (ch.getKind() == ElementKind.METHOD) {
jtulach@0
  1337
                ExecutableElement ee = (ExecutableElement)ch;
jtulach@0
  1338
                if (ee.getParameters().isEmpty() && ee.getSimpleName().contentEquals("modelFor")) {
jtulach@0
  1339
                    return true;
jtulach@0
  1340
                }
jtulach@0
  1341
            }
jtulach@0
  1342
        }
jtulach@0
  1343
        return models.values().contains(e.getSimpleName().toString());
jtulach@0
  1344
    }
jtulach@0
  1345
    
jtulach@0
  1346
    private void writeToString(Prprt[] props, Writer w) throws IOException {
jtulach@0
  1347
        w.write("  public String toString() {\n");
jtulach@0
  1348
        w.write("    StringBuilder sb = new StringBuilder();\n");
jtulach@0
  1349
        w.write("    sb.append('{');\n");
jtulach@0
  1350
        String sep = "";
jtulach@0
  1351
        for (Prprt p : props) {
jtulach@0
  1352
            w.write(sep);
jaroslav@18
  1353
            w.append("    sb.append('\"').append(\"" + p.name() + "\")");
jaroslav@18
  1354
                w.append(".append('\"').append(\":\");\n");
jaroslav@362
  1355
            w.append("    sb.append(org.netbeans.html.json.impl.JSON.toJSON(prop_");
jtulach@0
  1356
            w.append(p.name()).append("));\n");
jtulach@0
  1357
            sep =    "    sb.append(',');\n";
jtulach@0
  1358
        }
jtulach@0
  1359
        w.write("    sb.append('}');\n");
jtulach@0
  1360
        w.write("    return sb.toString();\n");
jtulach@0
  1361
        w.write("  }\n");
jtulach@0
  1362
    }
jtulach@0
  1363
    private void writeClone(String className, Prprt[] props, Writer w) throws IOException {
jtulach@0
  1364
        w.write("  public " + className + " clone() {\n");
jaroslav@111
  1365
        w.write("    return clone(context);\n");
jaroslav@111
  1366
        w.write("  }\n");
jaroslav@111
  1367
        w.write("  private " + className + " clone(net.java.html.BrwsrCtx ctx) {\n");
jaroslav@111
  1368
        w.write("    " + className + " ret = new " + className + "(ctx);\n");
jtulach@0
  1369
        for (Prprt p : props) {
jtulach@0
  1370
            if (!p.array()) {
jtulach@0
  1371
                boolean isModel[] = { false };
jtulach@0
  1372
                boolean isEnum[] = { false };
jtulach@0
  1373
                boolean isPrimitive[] = { false };
jtulach@0
  1374
                checkType(p, isModel, isEnum, isPrimitive);
jtulach@0
  1375
                if (!isModel[0]) {
jtulach@0
  1376
                    w.write("    ret.prop_" + p.name() + " = prop_" + p.name() + ";\n");
jtulach@0
  1377
                    continue;
jtulach@0
  1378
                }
jaroslav@114
  1379
                w.write("    ret.prop_" + p.name() + " =  prop_" + p.name() + "  == null ? null : prop_" + p.name() + ".clone();\n");
jtulach@0
  1380
            } else {
jaroslav@282
  1381
                w.write("    ret.prop_" + p.name() + ".cloneAll(ctx, prop_" + p.name() + ");\n");
jtulach@0
  1382
            }
jtulach@0
  1383
        }
jtulach@0
  1384
        
jtulach@0
  1385
        w.write("    return ret;\n");
jtulach@0
  1386
        w.write("  }\n");
jtulach@0
  1387
    }
jtulach@0
  1388
jtulach@0
  1389
    private String inPckName(Element e) {
jtulach@0
  1390
        StringBuilder sb = new StringBuilder();
jtulach@0
  1391
        while (e.getKind() != ElementKind.PACKAGE) {
jtulach@0
  1392
            if (sb.length() == 0) {
jtulach@0
  1393
                sb.append(e.getSimpleName());
jtulach@0
  1394
            } else {
jtulach@0
  1395
                sb.insert(0, '.');
jtulach@0
  1396
                sb.insert(0, e.getSimpleName());
jtulach@0
  1397
            }
jtulach@0
  1398
            e = e.getEnclosingElement();
jtulach@0
  1399
        }
jtulach@0
  1400
        return sb.toString();
jtulach@0
  1401
    }
jtulach@0
  1402
jtulach@0
  1403
    private String fqn(TypeMirror pt, Element relative) {
jtulach@0
  1404
        if (pt.getKind() == TypeKind.ERROR) {
jtulach@0
  1405
            final Elements eu = processingEnv.getElementUtils();
jtulach@0
  1406
            PackageElement pckg = eu.getPackageOf(relative);
jtulach@0
  1407
            return pckg.getQualifiedName() + "." + pt.toString();
jtulach@0
  1408
        }
jtulach@0
  1409
        return pt.toString();
jtulach@0
  1410
    }
jtulach@0
  1411
jtulach@0
  1412
    private String checkType(Prprt p, boolean[] isModel, boolean[] isEnum, boolean[] isPrimitive) {
jtulach@0
  1413
        TypeMirror tm;
jtulach@0
  1414
        try {
jtulach@0
  1415
            String ret = p.typeName(processingEnv);
jtulach@0
  1416
            TypeElement e = processingEnv.getElementUtils().getTypeElement(ret);
jtulach@0
  1417
            if (e == null) {
jtulach@0
  1418
                isModel[0] = true;
jtulach@0
  1419
                isEnum[0] = false;
jtulach@0
  1420
                isPrimitive[0] = false;
jtulach@0
  1421
                return ret;
jtulach@0
  1422
            }
jtulach@0
  1423
            tm = e.asType();
jtulach@0
  1424
        } catch (MirroredTypeException ex) {
jtulach@0
  1425
            tm = ex.getTypeMirror();
jtulach@0
  1426
        }
jtulach@0
  1427
        tm = processingEnv.getTypeUtils().erasure(tm);
jaroslav@241
  1428
        if (isPrimitive[0] = tm.getKind().isPrimitive()) {
jaroslav@241
  1429
            isEnum[0] = false;
jaroslav@241
  1430
            isModel[0] = false;
jaroslav@241
  1431
            return tm.toString();
jaroslav@241
  1432
        }
jtulach@0
  1433
        final Element e = processingEnv.getTypeUtils().asElement(tm);
jaroslav@241
  1434
        if (e.getKind() == ElementKind.CLASS && tm.getKind() == TypeKind.ERROR) {
jaroslav@241
  1435
            isModel[0] = true;
jaroslav@241
  1436
            isEnum[0] = false;
jaroslav@242
  1437
            return e.getSimpleName().toString();
jaroslav@241
  1438
        }
jaroslav@241
  1439
        
jtulach@0
  1440
        final Model m = e == null ? null : e.getAnnotation(Model.class);
jtulach@0
  1441
        String ret;
jtulach@0
  1442
        if (m != null) {
jtulach@0
  1443
            ret = findPkgName(e) + '.' + m.className();
jtulach@0
  1444
            isModel[0] = true;
jtulach@0
  1445
            models.put(e, m.className());
jtulach@0
  1446
        } else if (findModelForMthd(e)) {
jtulach@0
  1447
            ret = ((TypeElement)e).getQualifiedName().toString();
jtulach@0
  1448
            isModel[0] = true;
jtulach@0
  1449
        } else {
jtulach@0
  1450
            ret = tm.toString();
jtulach@0
  1451
        }
jtulach@0
  1452
        TypeMirror enm = processingEnv.getElementUtils().getTypeElement("java.lang.Enum").asType();
jtulach@0
  1453
        enm = processingEnv.getTypeUtils().erasure(enm);
jtulach@0
  1454
        isEnum[0] = processingEnv.getTypeUtils().isSubtype(tm, enm);
jtulach@0
  1455
        return ret;
jtulach@0
  1456
    }
jtulach@0
  1457
    
jtulach@0
  1458
    private static boolean findModelForMthd(Element clazz) {
jtulach@0
  1459
        if (clazz == null) {
jtulach@0
  1460
            return false;
jtulach@0
  1461
        }
jtulach@0
  1462
        for (Element e : clazz.getEnclosedElements()) {
jtulach@0
  1463
            if (e.getKind() == ElementKind.METHOD) {
jtulach@0
  1464
                ExecutableElement ee = (ExecutableElement)e;
jtulach@0
  1465
                if (
jtulach@0
  1466
                    ee.getSimpleName().contentEquals("modelFor") &&
jtulach@0
  1467
                    ee.getParameters().isEmpty()
jtulach@0
  1468
                ) {
jtulach@0
  1469
                    return true;
jtulach@0
  1470
                }
jtulach@0
  1471
            }
jtulach@0
  1472
        }
jtulach@0
  1473
        return false;
jtulach@0
  1474
    }
jtulach@0
  1475
jaroslav@24
  1476
    private Iterable<String> findParamNames(
jaroslav@24
  1477
        Element e, String url, String jsonParam, StringBuilder... both
jaroslav@24
  1478
    ) {
jaroslav@100
  1479
        List<String> params = new ArrayList<String>();
jaroslav@24
  1480
        int wasJSON = 0;
jtulach@0
  1481
jtulach@0
  1482
        for (int pos = 0; ;) {
jtulach@0
  1483
            int next = url.indexOf('{', pos);
jtulach@0
  1484
            if (next == -1) {
jaroslav@24
  1485
                both[wasJSON].append('"')
jtulach@0
  1486
                    .append(url.substring(pos))
jtulach@0
  1487
                    .append('"');
jtulach@0
  1488
                return params;
jtulach@0
  1489
            }
jtulach@0
  1490
            int close = url.indexOf('}', next);
jtulach@0
  1491
            if (close == -1) {
jtulach@0
  1492
                error("Unbalanced '{' and '}' in " + url, e);
jtulach@0
  1493
                return params;
jtulach@0
  1494
            }
jtulach@0
  1495
            final String paramName = url.substring(next + 1, close);
jtulach@0
  1496
            params.add(paramName);
jaroslav@24
  1497
            if (paramName.equals(jsonParam) && !jsonParam.isEmpty()) {
jaroslav@24
  1498
                both[wasJSON].append('"')
jaroslav@24
  1499
                    .append(url.substring(pos, next))
jaroslav@24
  1500
                    .append('"');
jaroslav@24
  1501
                wasJSON = 1;
jaroslav@24
  1502
            } else {
jaroslav@24
  1503
                both[wasJSON].append('"')
jaroslav@24
  1504
                    .append(url.substring(pos, next))
jaroslav@24
  1505
                    .append("\" + ").append(paramName).append(" + ");
jaroslav@24
  1506
            }
jtulach@0
  1507
            pos = close + 1;
jtulach@0
  1508
        }
jtulach@0
  1509
    }
jtulach@0
  1510
jtulach@0
  1511
    private static Prprt findPrprt(Prprt[] properties, String propName) {
jtulach@0
  1512
        for (Prprt p : properties) {
jtulach@0
  1513
            if (propName.equals(p.name())) {
jtulach@0
  1514
                return p;
jtulach@0
  1515
            }
jtulach@0
  1516
        }
jtulach@0
  1517
        return null;
jtulach@0
  1518
    }
jtulach@0
  1519
jtulach@0
  1520
    private boolean isPrimitive(String type) {
jtulach@0
  1521
        return 
jtulach@0
  1522
            "int".equals(type) ||
jtulach@0
  1523
            "double".equals(type) ||
jtulach@0
  1524
            "long".equals(type) ||
jtulach@0
  1525
            "short".equals(type) ||
jtulach@0
  1526
            "byte".equals(type) ||
jaroslav@100
  1527
            "char".equals(type) ||
jaroslav@100
  1528
            "boolean".equals(type) ||
jtulach@0
  1529
            "float".equals(type);
jtulach@0
  1530
    }
jtulach@0
  1531
jtulach@0
  1532
    private static Collection<String> findDerivedFrom(Map<String, Collection<String>> propsDeps, String derivedProp) {
jaroslav@100
  1533
        Set<String> names = new HashSet<String>();
jtulach@0
  1534
        for (Map.Entry<String, Collection<String>> e : propsDeps.entrySet()) {
jtulach@0
  1535
            if (e.getValue().contains(derivedProp)) {
jtulach@0
  1536
                names.add(e.getKey());
jtulach@0
  1537
            }
jtulach@0
  1538
        }
jtulach@0
  1539
        return names;
jtulach@0
  1540
    }
jtulach@0
  1541
    
jtulach@0
  1542
    private Prprt[] createProps(Element e, Property[] arr) {
jtulach@0
  1543
        Prprt[] ret = Prprt.wrap(processingEnv, e, arr);
jtulach@0
  1544
        Prprt[] prev = verify.put(e, ret);
jtulach@0
  1545
        if (prev != null) {
jtulach@0
  1546
            error("Two sets of properties for ", e);
jtulach@0
  1547
        }
jtulach@0
  1548
        return ret;
jtulach@0
  1549
    }
jtulach@0
  1550
    
jaroslav@17
  1551
    private static String strip(String s) {
jaroslav@17
  1552
        int indx = s.indexOf("__");
jaroslav@17
  1553
        if (indx >= 0) {
jaroslav@17
  1554
            return s.substring(0, indx);
jaroslav@17
  1555
        } else {
jaroslav@17
  1556
            return s;
jaroslav@17
  1557
        }
jaroslav@17
  1558
    }
jaroslav@70
  1559
jaroslav@73
  1560
    private String findDataSpecified(ExecutableElement e, OnReceive onR) {
jaroslav@73
  1561
        try {
jaroslav@73
  1562
            return onR.data().getName();
jaroslav@73
  1563
        } catch (MirroredTypeException ex) {
jaroslav@242
  1564
            final TypeMirror tm = ex.getTypeMirror();
jaroslav@242
  1565
            String name;
jaroslav@242
  1566
            final Element te = processingEnv.getTypeUtils().asElement(tm);
jaroslav@242
  1567
            if (te.getKind() == ElementKind.CLASS && tm.getKind() == TypeKind.ERROR) {
jaroslav@242
  1568
                name = te.getSimpleName().toString();
jaroslav@242
  1569
            } else {
jaroslav@242
  1570
                name = tm.toString();
jaroslav@242
  1571
            }
jaroslav@73
  1572
            return "java.lang.Object".equals(name) ? null : name;
jaroslav@73
  1573
        } catch (Exception ex) {
jaroslav@73
  1574
            // fallback
jaroslav@73
  1575
        }
jaroslav@73
  1576
        
jaroslav@72
  1577
        AnnotationMirror found = null;
jaroslav@72
  1578
        for (AnnotationMirror am : e.getAnnotationMirrors()) {
jaroslav@72
  1579
            if (am.getAnnotationType().toString().equals(OnReceive.class.getName())) {
jaroslav@72
  1580
                found = am;
jaroslav@72
  1581
            }
jaroslav@72
  1582
        }
jaroslav@72
  1583
        if (found == null) {
jaroslav@72
  1584
            return null;
jaroslav@72
  1585
        }
jaroslav@72
  1586
        
jaroslav@72
  1587
        for (Map.Entry<? extends ExecutableElement, ? extends AnnotationValue> entry : found.getElementValues().entrySet()) {
jaroslav@72
  1588
            ExecutableElement ee = entry.getKey();
jaroslav@72
  1589
            AnnotationValue av = entry.getValue();
jaroslav@72
  1590
            if (ee.getSimpleName().contentEquals("data")) {
jaroslav@73
  1591
                List<? extends Object> values = getAnnoValues(processingEnv, e, found);
jaroslav@73
  1592
                for (Object v : values) {
jaroslav@73
  1593
                    String sv = v.toString();
jaroslav@73
  1594
                    if (sv.startsWith("data = ") && sv.endsWith(".class")) {
jaroslav@73
  1595
                        return sv.substring(7, sv.length() - 6);
jaroslav@73
  1596
                    }
jaroslav@73
  1597
                }
jaroslav@73
  1598
                return "error";
jaroslav@72
  1599
            }
jaroslav@72
  1600
        }
jaroslav@72
  1601
        return null;
jaroslav@72
  1602
    }
jaroslav@72
  1603
jaroslav@72
  1604
    static List<? extends Object> getAnnoValues(ProcessingEnvironment pe, Element e, AnnotationMirror am) {
jaroslav@70
  1605
        try {
jaroslav@72
  1606
            Class<?> trees = Class.forName("com.sun.tools.javac.api.JavacTrees");
jaroslav@72
  1607
            Method m = trees.getMethod("instance", ProcessingEnvironment.class);
jaroslav@72
  1608
            Object instance = m.invoke(null, pe);
jaroslav@72
  1609
            m = instance.getClass().getMethod("getPath", Element.class, AnnotationMirror.class);
jaroslav@72
  1610
            Object path = m.invoke(instance, e, am);
jaroslav@72
  1611
            m = path.getClass().getMethod("getLeaf");
jaroslav@72
  1612
            Object leaf = m.invoke(path);
jaroslav@72
  1613
            m = leaf.getClass().getMethod("getArguments");
jaroslav@72
  1614
            return (List) m.invoke(leaf);
jaroslav@72
  1615
        } catch (Exception ex) {
jaroslav@72
  1616
            return Collections.emptyList();
jaroslav@70
  1617
        }
jaroslav@70
  1618
    }
jaroslav@240
  1619
jtulach@0
  1620
    private static class Prprt {
jtulach@0
  1621
        private final Element e;
jtulach@0
  1622
        private final AnnotationMirror tm;
jtulach@0
  1623
        private final Property p;
jtulach@0
  1624
jtulach@0
  1625
        public Prprt(Element e, AnnotationMirror tm, Property p) {
jtulach@0
  1626
            this.e = e;
jtulach@0
  1627
            this.tm = tm;
jtulach@0
  1628
            this.p = p;
jtulach@0
  1629
        }
jtulach@0
  1630
        
jtulach@0
  1631
        String name() {
jtulach@0
  1632
            return p.name();
jtulach@0
  1633
        }
jtulach@0
  1634
        
jtulach@0
  1635
        boolean array() {
jtulach@0
  1636
            return p.array();
jtulach@0
  1637
        }
jtulach@0
  1638
jtulach@0
  1639
        String typeName(ProcessingEnvironment env) {
jaroslav@100
  1640
            RuntimeException ex;
jtulach@0
  1641
            try {
jtulach@0
  1642
                return p.type().getName();
jaroslav@100
  1643
            } catch (IncompleteAnnotationException e) {
jaroslav@100
  1644
                ex = e;
jaroslav@100
  1645
            } catch (AnnotationTypeMismatchException e) {
jaroslav@100
  1646
                ex = e;
jaroslav@100
  1647
            }
jaroslav@100
  1648
            for (Object v : getAnnoValues(env, e, tm)) {
jaroslav@100
  1649
                String s = v.toString().replace(" ", "");
jaroslav@100
  1650
                if (s.startsWith("type=") && s.endsWith(".class")) {
jaroslav@100
  1651
                    return s.substring(5, s.length() - 6);
jtulach@0
  1652
                }
jtulach@0
  1653
            }
jaroslav@100
  1654
            throw ex;
jtulach@0
  1655
        }
jtulach@0
  1656
        
jtulach@0
  1657
        
jtulach@0
  1658
        static Prprt[] wrap(ProcessingEnvironment pe, Element e, Property[] arr) {
jtulach@0
  1659
            if (arr.length == 0) {
jtulach@0
  1660
                return new Prprt[0];
jtulach@0
  1661
            }
jtulach@0
  1662
            
jtulach@0
  1663
            if (e.getKind() != ElementKind.CLASS) {
jtulach@0
  1664
                throw new IllegalStateException("" + e.getKind());
jtulach@0
  1665
            }
jtulach@0
  1666
            TypeElement te = (TypeElement)e;
jtulach@0
  1667
            List<? extends AnnotationValue> val = null;
jtulach@0
  1668
            for (AnnotationMirror an : te.getAnnotationMirrors()) {
jtulach@0
  1669
                for (Map.Entry<? extends ExecutableElement, ? extends AnnotationValue> entry : an.getElementValues().entrySet()) {
jtulach@0
  1670
                    if (entry.getKey().getSimpleName().contentEquals("properties")) {
jtulach@0
  1671
                        val = (List)entry.getValue().getValue();
jtulach@0
  1672
                        break;
jtulach@0
  1673
                    }
jtulach@0
  1674
                }
jtulach@0
  1675
            }
jtulach@0
  1676
            if (val == null || val.size() != arr.length) {
jtulach@0
  1677
                pe.getMessager().printMessage(Diagnostic.Kind.ERROR, "" + val, e);
jtulach@0
  1678
                return new Prprt[0];
jtulach@0
  1679
            }
jtulach@0
  1680
            Prprt[] ret = new Prprt[arr.length];
jtulach@0
  1681
            BIG: for (int i = 0; i < ret.length; i++) {
jtulach@0
  1682
                AnnotationMirror am = (AnnotationMirror)val.get(i).getValue();
jtulach@0
  1683
                ret[i] = new Prprt(e, am, arr[i]);
jtulach@0
  1684
                
jtulach@0
  1685
            }
jtulach@0
  1686
            return ret;
jtulach@0
  1687
        }
jtulach@0
  1688
    }
jaroslav@70
  1689
jaroslav@70
  1690
    @Override
jaroslav@70
  1691
    public Iterable<? extends Completion> getCompletions(Element element, AnnotationMirror annotation, ExecutableElement member, String userText) {
jaroslav@71
  1692
        final Level l = Level.FINE;
jaroslav@71
  1693
        LOG.log(l, " element: {0}", element);
jaroslav@71
  1694
        LOG.log(l, " annotation: {0}", annotation);
jaroslav@71
  1695
        LOG.log(l, " member: {0}", member);
jaroslav@71
  1696
        LOG.log(l, " userText: {0}", userText);
jaroslav@71
  1697
        LOG.log(l, "str: {0}", annotation.getAnnotationType().toString());
jaroslav@70
  1698
        if (annotation.getAnnotationType().toString().equals(OnReceive.class.getName())) {
jaroslav@70
  1699
            if (member.getSimpleName().contentEquals("method")) {
jaroslav@70
  1700
                return Arrays.asList(
jaroslav@70
  1701
                    methodOf("GET"),
jaroslav@70
  1702
                    methodOf("POST"),
jaroslav@70
  1703
                    methodOf("PUT"),
jaroslav@70
  1704
                    methodOf("DELETE"),
jaroslav@247
  1705
                    methodOf("HEAD"),
jaroslav@247
  1706
                    methodOf("WebSocket")
jaroslav@70
  1707
                );
jaroslav@70
  1708
            }
jaroslav@70
  1709
        }
jaroslav@70
  1710
        
jaroslav@70
  1711
        return super.getCompletions(element, annotation, member, userText);
jaroslav@70
  1712
    }
jtulach@0
  1713
    
jaroslav@70
  1714
    private static final Completion methodOf(String method) {
jaroslav@362
  1715
        ResourceBundle rb = ResourceBundle.getBundle("org.netbeans.html.json.impl.Bundle");
jaroslav@70
  1716
        return Completions.of('"' + method + '"', rb.getString("MSG_Completion_" + method));
jaroslav@70
  1717
    }
jaroslav@245
  1718
    
jaroslav@245
  1719
    private boolean findOnError(ExecutableElement errElem, TypeElement te, String name, String className) {
jaroslav@245
  1720
        String err = null;
jaroslav@245
  1721
        METHODS:
jaroslav@245
  1722
        for (Element e : te.getEnclosedElements()) {
jaroslav@245
  1723
            if (e.getKind() != ElementKind.METHOD) {
jaroslav@245
  1724
                continue;
jaroslav@245
  1725
            }
jaroslav@245
  1726
            if (!e.getSimpleName().contentEquals(name)) {
jaroslav@245
  1727
                continue;
jaroslav@245
  1728
            }
jaroslav@245
  1729
            if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@245
  1730
                errElem = (ExecutableElement) e;
jaroslav@245
  1731
                err = "Would have to be static";
jaroslav@245
  1732
                continue;
jaroslav@245
  1733
            }
jaroslav@245
  1734
            ExecutableElement ee = (ExecutableElement) e;
jaroslav@245
  1735
            TypeMirror excType = processingEnv.getElementUtils().getTypeElement(Exception.class.getName()).asType();
jaroslav@245
  1736
            final List<? extends VariableElement> params = ee.getParameters();
jaroslav@245
  1737
            boolean error = false;
jaroslav@245
  1738
            if (params.size() != 2) {
jaroslav@245
  1739
                error = true;
jaroslav@245
  1740
            } else {
jaroslav@261
  1741
                String firstType = params.get(0).asType().toString();
jaroslav@261
  1742
                int lastDot = firstType.lastIndexOf('.');
jaroslav@261
  1743
                if (lastDot != -1) {
jaroslav@261
  1744
                    firstType = firstType.substring(lastDot + 1);
jaroslav@261
  1745
                }
jaroslav@261
  1746
                if (!firstType.equals(className)) {
jaroslav@245
  1747
                    error = true;
jaroslav@245
  1748
                }
jaroslav@245
  1749
                if (!processingEnv.getTypeUtils().isAssignable(excType, params.get(1).asType())) {
jaroslav@245
  1750
                    error = true;
jaroslav@245
  1751
                }
jaroslav@245
  1752
            }
jaroslav@245
  1753
            if (error) {
jaroslav@245
  1754
                errElem = (ExecutableElement) e;
jaroslav@245
  1755
                err = "Error method first argument needs to be " + className + " and second Exception";
jaroslav@245
  1756
                continue;
jaroslav@245
  1757
            }
jaroslav@245
  1758
            return true;
jaroslav@245
  1759
        }
jaroslav@245
  1760
        if (err == null) {
jaroslav@245
  1761
            err = "Cannot find " + name + "(" + className + ", Exception) method in this class";
jaroslav@245
  1762
        }
jaroslav@245
  1763
        error(err, errElem);
jaroslav@245
  1764
        return false;
jaroslav@245
  1765
    }
jaroslav@245
  1766
    
jtulach@0
  1767
}