hg/html~html4j: json/src/main/java/org/netbeans/html/json/impl/ModelProcessor.java@4b66c5141f84 (annotated)

jtulach@0	1	/**
jaroslav@358	2	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
jtulach@0	3	*
jaroslav@365	4	* Copyright 2013-2013 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)) {
jaroslav@490	126	if (e.getKind().isField()) {
jaroslav@490	127	if (e.getKind() != ElementKind.ENUM_CONSTANT) {
jaroslav@490	128	error("Only enum constants fields can be annotated by @Model annotations", e);
jaroslav@490	129	ok = false;
jaroslav@490	130	continue;
jaroslav@490	131	}
jaroslav@490	132	if (e.getEnclosingElement().getAnnotation(Model.class) == null) {
jaroslav@490	133	error("Enclosing enum has to be annotated by @Model annotation", e);
jaroslav@490	134	ok = false;
jaroslav@490	135	continue;
jaroslav@490	136	}
jaroslav@490	137	} else if (!processModelOrEnum(e)) {
jtulach@0	138	ok = false;
jtulach@0	139	}
jtulach@0	140	}
jtulach@0	141	if (roundEnv.processingOver()) {
jtulach@0	142	models.clear();
jtulach@0	143	for (Map.Entry<Element, Prprt[]> entry : verify.entrySet()) {
jaroslav@490	144	if (!(entry.getKey() instanceof TypeElement)) {
jaroslav@490	145	continue;
jaroslav@490	146	}
jtulach@0	147	TypeElement te = (TypeElement)entry.getKey();
jtulach@0	148	String fqn = processingEnv.getElementUtils().getBinaryName(te).toString();
jtulach@0	149	Element finalElem = processingEnv.getElementUtils().getTypeElement(fqn);
jtulach@0	150	if (finalElem == null) {
jtulach@0	151	continue;
jtulach@0	152	}
jtulach@0	153	Prprt[] props;
jtulach@0	154	Model m = finalElem.getAnnotation(Model.class);
jtulach@3	155	if (m == null) {
jtulach@3	156	continue;
jtulach@3	157	}
jtulach@0	158	props = Prprt.wrap(processingEnv, finalElem, m.properties());
jtulach@0	159	for (Prprt p : props) {
jtulach@0	160	boolean[] isModel = { false };
jtulach@0	161	boolean[] isEnum = { false };
jtulach@0	162	boolean[] isPrimitive = { false };
jtulach@0	163	String t = checkType(p, isModel, isEnum, isPrimitive);
jtulach@0	164	if (isEnum[0]) {
jtulach@0	165	continue;
jtulach@0	166	}
jtulach@0	167	if (isPrimitive[0]) {
jtulach@0	168	continue;
jtulach@0	169	}
jtulach@0	170	if (isModel[0]) {
jtulach@0	171	continue;
jtulach@0	172	}
jtulach@0	173	if ("java.lang.String".equals(t)) {
jtulach@0	174	continue;
jtulach@0	175	}
jtulach@0	176	error("The type " + t + " should be defined by @Model annotation", entry.getKey());
jtulach@0	177	}
jtulach@0	178	}
jtulach@0	179	verify.clear();
jtulach@0	180	}
jtulach@0	181	return ok;
jtulach@0	182	}
jtulach@0	183
jtulach@0	184	private void error(String msg, Element e) {
jtulach@0	185	processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, msg, e);
jtulach@0	186	}
jtulach@0	187
jaroslav@490	188	private boolean processModelOrEnum(Element e) {
jaroslav@490	189	if (e.getKind() == ElementKind.ENUM) {
jaroslav@490	190	return processEnum(e);
jaroslav@490	191	} else {
jaroslav@490	192	return processModel(e);
jaroslav@490	193	}
jaroslav@490	194	}
jtulach@0	195	private boolean processModel(Element e) {
jtulach@0	196	boolean ok = true;
jtulach@0	197	Model m = e.getAnnotation(Model.class);
jtulach@0	198	if (m == null) {
jtulach@0	199	return true;
jtulach@0	200	}
jtulach@0	201	String pkg = findPkgName(e);
jtulach@0	202	Writer w;
jtulach@0	203	String className = m.className();
jtulach@0	204	models.put(e, className);
jtulach@0	205	try {
jtulach@0	206	StringWriter body = new StringWriter();
jaroslav@386	207	StringBuilder onReceiveType = new StringBuilder();
jaroslav@500	208	List<PropInfo> propsGetSet = new ArrayList<PropInfo>();
jaroslav@100	209	List<String> functions = new ArrayList<String>();
jaroslav@100	210	Map<String, Collection<String>> propsDeps = new HashMap<String, Collection<String>>();
jaroslav@100	211	Map<String, Collection<String>> functionDeps = new HashMap<String, Collection<String>>();
jtulach@0	212	Prprt[] props = createProps(e, m.properties());
jtulach@0	213
jtulach@0	214	if (!generateComputedProperties(body, props, e.getEnclosedElements(), propsGetSet, propsDeps)) {
jtulach@0	215	ok = false;
jtulach@0	216	}
jtulach@0	217	if (!generateOnChange(e, propsDeps, props, className, functionDeps)) {
jtulach@0	218	ok = false;
jtulach@0	219	}
jaroslav@502	220	if (!generateProperties(e, body, className, props, propsGetSet, null, propsDeps, functionDeps)) {
jtulach@0	221	ok = false;
jtulach@0	222	}
jtulach@0	223	if (!generateFunctions(e, body, className, e.getEnclosedElements(), functions)) {
jtulach@0	224	ok = false;
jtulach@0	225	}
jaroslav@386	226	if (!generateReceive(e, body, className, e.getEnclosedElements(), onReceiveType)) {
jaroslav@12	227	ok = false;
jaroslav@12	228	}
jaroslav@240	229	if (!generateOperation(e, body, className, e.getEnclosedElements())) {
jaroslav@240	230	ok = false;
jaroslav@240	231	}
jtulach@0	232	FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + className, e);
jtulach@0	233	w = new OutputStreamWriter(java.openOutputStream());
jtulach@0	234	try {
jtulach@0	235	w.append("package " + pkg + ";\n");
jtulach@2	236	w.append("import net.java.html.json.*;\n");
jaroslav@490	237	w.append("public final ");
jaroslav@500	238	generateClassBody(w, className, body, e, props,
jaroslav@502	239	functionDeps, propsDeps, propsGetSet, null,
jaroslav@500	240	functions, onReceiveType
jaroslav@500	241	);
jaroslav@490	242	} finally {
jaroslav@490	243	w.close();
jaroslav@490	244	}
jaroslav@490	245	} catch (IOException ex) {
jaroslav@490	246	error("Can't create " + className + ".java", e);
jaroslav@490	247	return false;
jaroslav@490	248	}
jaroslav@490	249	return ok;
jaroslav@490	250	}
jaroslav@490	251
jaroslav@500	252	private void generateClassBody(
jaroslav@500	253	Writer w, String className, StringWriter body,
jaroslav@500	254	Element e, Prprt[] props,
jaroslav@500	255	Map<String, Collection<String>> functionDeps,
jaroslav@500	256	Map<String, Collection<String>> propsDeps,
jaroslav@500	257	List<PropInfo> propsGetSet,
jaroslav@502	258	Collection<PropInfo> taken,
jaroslav@500	259	List<String> functions,
jaroslav@500	260	StringBuilder onReceiveType
jaroslav@500	261	) throws IOException {
jaroslav@490	262	w.append("class ").append(className).append(" implements Cloneable {\n");
jaroslav@490	263	w.append(" private static final Html4JavaType TYPE = new Html4JavaType();\n");
jaroslav@490	264	w.append(" private final org.apidesign.html.json.spi.Proto proto;\n");
jaroslav@490	265	w.append(body.toString());
jaroslav@490	266	w.append(" private static Class<" + inPckName(e) + "> modelFor() { return null; }\n");
jaroslav@490	267	w.append(" private ").append(className).append("(net.java.html.BrwsrCtx context) {\n");
jaroslav@490	268	w.append(" this.proto = TYPE.createProto(this, context);\n");
jaroslav@506	269	for (int i = 0; i < props.length; i++) {
jaroslav@506	270	Prprt p = props[i];
jaroslav@490	271	if (p.array()) {
jaroslav@490	272	final String tn = typeName(e, p);
jaroslav@490	273	String[] gs = toGetSet(p.name(), tn, p.array());
jaroslav@490	274	w.write(" this.prop_" + p.name() + " = proto.createList(\""
jaroslav@490	275	+ p.name() + "\"");
jaroslav@490	276	if (functionDeps.containsKey(p.name())) {
jaroslav@506	277	w.write(", " + i);
jaroslav@490	278	} else {
jaroslav@490	279	w.write(", -1");
jaroslav@490	280	}
jaroslav@490	281	Collection<String> dependants = propsDeps.get(p.name());
jaroslav@490	282	if (dependants != null) {
jaroslav@490	283	for (String depProp : dependants) {
jaroslav@490	284	w.write(", ");
jaroslav@490	285	w.write('\"');
jaroslav@490	286	w.write(depProp);
jaroslav@490	287	w.write('\"');
jaroslav@490	288	}
jaroslav@490	289	}
jaroslav@490	290	w.write(")");
jaroslav@490	291	w.write(";\n");
jaroslav@490	292	}
jaroslav@490	293	}
jaroslav@490	294	w.append(" };\n");
jaroslav@490	295	w.append(" public ").append(className).append("() {\n");
jaroslav@490	296	w.append(" this(net.java.html.BrwsrCtx.findDefault(").append(className).append(".class));\n");
jaroslav@490	297	for (Prprt p : props) {
jaroslav@490	298	if (!p.array()) {
jaroslav@490	299	boolean[] isModel = {false};
jaroslav@490	300	boolean[] isEnum = {false};
jaroslav@490	301	boolean isPrimitive[] = {false};
jaroslav@490	302	String tn = checkType(p, isModel, isEnum, isPrimitive);
jaroslav@494	303	if (isModel[0] && !isEnum[0]) {
jaroslav@490	304	w.write(" prop_" + p.name() + " = new " + tn + "();\n");
jaroslav@490	305	}
jaroslav@490	306	}
jaroslav@490	307	}
jaroslav@490	308	w.append(" };\n");
jaroslav@490	309	if (props.length > 0) {
jaroslav@490	310	w.append(" public ").append(className).append("(");
jaroslav@490	311	Prprt firstArray = null;
jaroslav@490	312	String sep = "";
jaroslav@490	313	for (Prprt p : props) {
jaroslav@490	314	if (p.array()) {
jaroslav@490	315	if (firstArray == null) {
jaroslav@490	316	firstArray = p;
jaroslav@490	317	}
jaroslav@490	318	continue;
jaroslav@490	319	}
jaroslav@490	320	String tn = typeName(e, p);
jaroslav@490	321	w.write(sep);
jaroslav@490	322	w.write(tn);
jaroslav@490	323	w.write(" " + p.name());
jaroslav@490	324	sep = ", ";
jaroslav@490	325	}
jaroslav@490	326	if (firstArray != null) {
jaroslav@490	327	String tn;
jaroslav@490	328	boolean[] isModel = {false};
jaroslav@490	329	boolean[] isEnum = {false};
jaroslav@490	330	boolean isPrimitive[] = {false};
jaroslav@490	331	tn = checkType(firstArray, isModel, isEnum, isPrimitive);
jaroslav@490	332	w.write(sep);
jaroslav@490	333	w.write(tn);
jaroslav@490	334	w.write("... " + firstArray.name());
jaroslav@490	335	}
jaroslav@490	336	w.append(") {\n");
jaroslav@490	337	w.append(" this(net.java.html.BrwsrCtx.findDefault(").append(className).append(".class));\n");
jaroslav@490	338	for (Prprt p : props) {
jaroslav@490	339	if (p.array()) {
jaroslav@490	340	continue;
jaroslav@490	341	}
jaroslav@490	342	w.write(" this.prop_" + p.name() + " = " + p.name() + ";\n");
jaroslav@490	343	}
jaroslav@490	344	if (firstArray != null) {
jaroslav@490	345	w.write(" proto.initTo(this.prop_" + firstArray.name() + ", " + firstArray.name() + ");\n");
jaroslav@490	346	}
jaroslav@490	347	w.append(" };\n");
jaroslav@490	348	}
jaroslav@490	349	w.append(" private static class Html4JavaType extends org.apidesign.html.json.spi.Proto.Type<").append(className).append("> {\n");
jaroslav@490	350	w.append(" private Html4JavaType() {\n super(").append(className).append(".class, ").
jaroslav@500	351	append(inPckName(e)).append(".class, " + (propsGetSet.size()) + ", "
jaroslav@490	352	+ (functions.size() / 2) + ");\n");
jaroslav@490	353	{
jaroslav@500	354	for (int i = 0; i < propsGetSet.size(); i++) {
jaroslav@503	355	w.append(" registerProperty(\"").append(propsGetSet.get(i).name).append("\", ");
jaroslav@503	356	w.append(i + ", " + (propsGetSet.get(i).set == null) + ");\n");
jaroslav@490	357	}
jaroslav@490	358	}
jaroslav@490	359	{
jaroslav@490	360	for (int i = 0; i < functions.size(); i += 2) {
jaroslav@490	361	w.append(" registerFunction(\"").append(functions.get(i)).append("\", ");
jaroslav@490	362	w.append((i / 2) + ");\n");
jaroslav@490	363	}
jaroslav@490	364	}
jaroslav@490	365	w.append(" }\n");
jaroslav@490	366	w.append(" @Override public void setValue(" + className + " data, int type, Object value) {\n");
jaroslav@490	367	w.append(" switch (type) {\n");
jaroslav@500	368	for (int i = 0; i < propsGetSet.size(); i++) {
jaroslav@503	369	final String set = propsGetSet.get(i).set;
jaroslav@503	370	String tn = propsGetSet.get(i).type;
jaroslav@490	371	String btn = findBoxedType(tn);
jaroslav@490	372	if (btn != null) {
jaroslav@490	373	tn = btn;
jaroslav@490	374	}
jaroslav@490	375	if (set != null) {
jaroslav@500	376	w.append(" case " + i + ": data." + strip(set) + "(TYPE.extractValue(" + tn + ".class, value)); return;\n");
jaroslav@490	377	}
jaroslav@490	378	}
jaroslav@490	379	w.append(" }\n");
jaroslav@490	380	w.append(" }\n");
jaroslav@490	381	w.append(" @Override public Object getValue(" + className + " data, int type) {\n");
jaroslav@490	382	w.append(" switch (type) {\n");
jaroslav@500	383	for (int i = 0; i < propsGetSet.size(); i++) {
jaroslav@503	384	final String get = propsGetSet.get(i).get;
jaroslav@490	385	if (get != null) {
jaroslav@500	386	w.append(" case " + i + ": return data." + strip(get) + "();\n");
jaroslav@490	387	}
jaroslav@490	388	}
jaroslav@490	389	w.append(" }\n");
jaroslav@490	390	w.append(" throw new UnsupportedOperationException();\n");
jaroslav@490	391	w.append(" }\n");
jaroslav@490	392	w.append(" @Override public void call(" + className + " model, int type, Object data, Object ev) {\n");
jaroslav@490	393	w.append(" switch (type) {\n");
jaroslav@490	394	for (int i = 0; i < functions.size(); i += 2) {
jaroslav@490	395	final String name = functions.get(i);
jaroslav@490	396	w.append(" case " + (i / 2) + ": model." + name + "(data, ev); return;\n");
jaroslav@490	397	}
jaroslav@490	398	w.append(" }\n");
jaroslav@490	399	w.append(" throw new UnsupportedOperationException();\n");
jaroslav@490	400	w.append(" }\n");
jaroslav@490	401	w.append(" @Override public org.apidesign.html.json.spi.Proto protoFor(Object obj) {\n");
jaroslav@490	402	w.append(" return ((" + className + ")obj).proto;");
jaroslav@490	403	w.append(" }\n");
jaroslav@490	404	w.append(" @Override public void onChange(" + className + " model, int type) {\n");
jaroslav@490	405	w.append(" switch (type) {\n");
jaroslav@490	406	{
jaroslav@506	407	for (int i = 0; i < props.length; i++) {
jaroslav@506	408	String pn = props[i].name();
jaroslav@506	409	Collection<String> onChange = functionDeps.get(pn);
jaroslav@490	410	if (onChange != null) {
jaroslav@490	411	w.append(" case " + i + ":\n");
jaroslav@490	412	for (String call : onChange) {
jaroslav@490	413	w.append(" ").append(call).append("\n");
jaroslav@490	414	}
jaroslav@490	415	w.write(" return;\n");
jaroslav@490	416	}
jaroslav@490	417	}
jaroslav@490	418	}
jaroslav@506	419	w.append(" }\n");
jaroslav@490	420	w.append(" }\n");
jaroslav@490	421	w.append(onReceiveType);
jaroslav@490	422	w.append(" @Override public " + className + " read(net.java.html.BrwsrCtx c, Object json) { return new " + className + "(c, json); }\n");
jaroslav@490	423	w.append(" @Override public " + className + " cloneTo(" + className + " o, net.java.html.BrwsrCtx c) { return o.clone(c); }\n");
jaroslav@490	424	w.append(" }\n");
jaroslav@490	425	w.append(" private ").append(className).append("(net.java.html.BrwsrCtx c, Object json) {\n");
jaroslav@490	426	w.append(" this(c);\n");
jaroslav@490	427	int values = 0;
jaroslav@500	428	for (int i = 0; i < propsGetSet.size(); i++) {
jaroslav@503	429	Prprt p = findPrprt(props, propsGetSet.get(i).name);
jaroslav@490	430	if (p == null) {
jaroslav@490	431	continue;
jaroslav@490	432	}
jaroslav@490	433	values++;
jaroslav@490	434	}
jaroslav@490	435	w.append(" Object[] ret = new Object[" + values + "];\n");
jaroslav@490	436	w.append(" proto.extract(json, new String[] {\n");
jaroslav@500	437	for (int i = 0; i < propsGetSet.size(); i++) {
jaroslav@503	438	Prprt p = findPrprt(props, propsGetSet.get(i).name);
jaroslav@490	439	if (p == null) {
jaroslav@490	440	continue;
jaroslav@490	441	}
jaroslav@503	442	w.append(" \"").append(propsGetSet.get(i).name).append("\",\n");
jaroslav@490	443	}
jaroslav@490	444	w.append(" }, ret);\n");
jaroslav@500	445	for (int i = 0, cnt = 0, prop = 0; i < propsGetSet.size(); i++) {
jaroslav@503	446	final String pn = propsGetSet.get(i).name;
jaroslav@490	447	Prprt p = findPrprt(props, pn);
jaroslav@490	448	if (p == null) {
jaroslav@490	449	continue;
jaroslav@490	450	}
jaroslav@501	451	if (prop == props.length) {
jaroslav@501	452	continue;
jaroslav@501	453	}
jaroslav@490	454	boolean[] isModel = { false };
jaroslav@490	455	boolean[] isEnum = { false };
jaroslav@490	456	boolean isPrimitive[] = { false };
jaroslav@490	457	String type = checkType(props[prop++], isModel, isEnum, isPrimitive);
jaroslav@490	458	if (p.array()) {
jaroslav@490	459	w.append(" if (ret[" + cnt + "] instanceof Object[]) {\n");
jaroslav@490	460	w.append(" for (Object e : ((Object[])ret[" + cnt + "])) {\n");
jaroslav@490	461	if (isModel[0]) {
jaroslav@490	462	w.append(" this.prop_").append(pn).append(".add(proto.read");
jaroslav@490	463	w.append("(" + type + ".class, e));\n");
jaroslav@490	464	} else if (isEnum[0]) {
jaroslav@490	465	w.append(" this.prop_").append(pn);
jaroslav@490	466	w.append(".add(e == null ? null : ");
jaroslav@490	467	w.append(type).append(".valueOf(TYPE.stringValue(e)));\n");
jaroslav@490	468	} else {
jaroslav@490	469	if (isPrimitive(type)) {
jaroslav@490	470	w.append(" this.prop_").append(pn).append(".add(TYPE.numberValue(e).");
jaroslav@490	471	w.append(type).append("Value());\n");
jaroslav@490	472	} else {
jaroslav@490	473	w.append(" this.prop_").append(pn).append(".add((");
jaroslav@490	474	w.append(type).append(")e);\n");
jaroslav@490	475	}
jaroslav@490	476	}
jaroslav@490	477	w.append(" }\n");
jaroslav@490	478	w.append(" }\n");
jaroslav@490	479	} else {
jaroslav@490	480	if (isEnum[0]) {
jaroslav@490	481	w.append(" try {\n");
jaroslav@490	482	w.append(" this.prop_").append(pn);
jaroslav@490	483	w.append(" = ret[" + cnt + "] == null ? null : ");
jaroslav@490	484	w.append(type).append(".valueOf(TYPE.stringValue(ret[" + cnt + "]));\n");
jaroslav@490	485	w.append(" } catch (IllegalArgumentException ex) {\n");
jaroslav@490	486	w.append(" ex.printStackTrace();\n");
jaroslav@490	487	w.append(" }\n");
jaroslav@490	488	} else if (isPrimitive(type)) {
jaroslav@490	489	w.append(" this.prop_").append(pn);
jaroslav@490	490	w.append(" = ret[" + cnt + "] == null ? ");
jaroslav@490	491	if ("char".equals(type)) {
jaroslav@490	492	w.append("0 : (TYPE.charValue(");
jaroslav@490	493	} else if ("boolean".equals(type)) {
jaroslav@490	494	w.append("false : (TYPE.boolValue(");
jaroslav@490	495	} else {
jaroslav@490	496	w.append("0 : (TYPE.numberValue(");
jaroslav@490	497	}
jaroslav@490	498	w.append("ret[" + cnt + "])).");
jaroslav@490	499	w.append(type).append("Value();\n");
jaroslav@490	500	} else if (isModel[0]) {
jaroslav@490	501	w.append(" this.prop_").append(pn).append(" = proto.read");
jaroslav@490	502	w.append("(" + type + ".class, ");
jaroslav@490	503	w.append("ret[" + cnt + "]);\n");
jaroslav@490	504	}else {
jaroslav@490	505	w.append(" this.prop_").append(pn);
jaroslav@490	506	w.append(" = (").append(type).append(')');
jaroslav@490	507	w.append("ret[" + cnt + "];\n");
jaroslav@490	508	}
jaroslav@490	509	}
jaroslav@490	510	cnt++;
jaroslav@490	511	}
jaroslav@490	512	w.append(" }\n");
jaroslav@490	513	writeToString(props, w);
jaroslav@490	514	writeClone(className, props, w);
jaroslav@490	515	w.write(" /** Activates this model instance in the current {@link \n"
jaroslav@490	516	+ "net.java.html.json.Models#bind(java.lang.Object, net.java.html.BrwsrCtx) browser context}. \n"
jaroslav@490	517	+ "In case of using Knockout technology, this means to \n"
jaroslav@490	518	+ "bind JSON like data in this model instance with Knockout tags in \n"
jaroslav@490	519	+ "the surrounding HTML page.\n"
jaroslav@490	520	+ "*/\n"
jaroslav@490	521	);
jaroslav@490	522	w.write(" public " + className + " applyBindings() {\n");
jaroslav@490	523	w.write(" proto.applyBindings();\n");
jaroslav@490	524	w.write(" return this;\n");
jaroslav@490	525	w.write(" }\n");
jaroslav@490	526	w.write(" public boolean equals(Object o) {\n");
jaroslav@490	527	w.write(" if (o == this) return true;\n");
jaroslav@490	528	w.write(" if (!(o instanceof " + className + ")) return false;\n");
jaroslav@490	529	w.write(" " + className + " p = (" + className + ")o;\n");
jaroslav@490	530	for (Prprt p : props) {
jaroslav@490	531	w.write(" if (!TYPE.isSame(prop_" + p.name() + ", p.prop_" + p.name() + ")) return false;\n");
jaroslav@490	532	}
jaroslav@490	533	w.write(" return true;\n");
jaroslav@490	534	w.write(" }\n");
jaroslav@490	535	w.write(" public int hashCode() {\n");
jaroslav@490	536	w.write(" int h = " + className + ".class.getName().hashCode();\n");
jaroslav@490	537	for (Prprt p : props) {
jaroslav@490	538	w.write(" h = TYPE.hashPlus(prop_" + p.name() + ", h);\n");
jaroslav@490	539	}
jaroslav@490	540	w.write(" return h;\n");
jaroslav@490	541	w.write(" }\n");
jaroslav@490	542	w.write("}\n");
jaroslav@490	543	}
jaroslav@490	544
jaroslav@490	545	private boolean processEnum(Element e) {
jaroslav@490	546	boolean ok = true;
jaroslav@490	547	Model m = e.getAnnotation(Model.class);
jaroslav@490	548	if (m == null) {
jaroslav@490	549	return true;
jaroslav@490	550	}
jaroslav@490	551	String pkg = findPkgName(e);
jaroslav@490	552	Writer w;
jaroslav@490	553	String className = m.className();
jaroslav@490	554	models.put(e, className);
jaroslav@490	555	try {
jaroslav@490	556	StringWriter body = new StringWriter();
jaroslav@490	557	StringBuilder onReceiveType = new StringBuilder();
jaroslav@500	558	List<PropInfo> propsGetSet = new ArrayList<PropInfo>();
jaroslav@490	559	List<String> functions = new ArrayList<String>();
jaroslav@490	560	Map<String, Collection<String>> propsDeps = new HashMap<String, Collection<String>>();
jaroslav@490	561	Map<String, Collection<String>> functionDeps = new HashMap<String, Collection<String>>();
jaroslav@490	562	Prprt[] props = createProps(e, m.properties());
jaroslav@490	563
jaroslav@490	564	if (!generateComputedProperties(body, props, e.getEnclosedElements(), propsGetSet, propsDeps)) {
jaroslav@490	565	ok = false;
jaroslav@490	566	}
jaroslav@502	567	if (!generateProperties(e, body, className, props, propsGetSet, null, propsDeps, functionDeps)) {
jaroslav@490	568	ok = false;
jaroslav@490	569	}
jaroslav@503	570	List<PropInfo> mergedProps = new ArrayList<PropInfo>(propsGetSet);
jaroslav@505	571	List<Prprt> allProps = new ArrayList<Prprt>(Arrays.asList(props));
jaroslav@503	572	for (Element ec : e.getEnclosedElements()) {
jaroslav@503	573	if (ec.getKind() != ElementKind.ENUM_CONSTANT) {
jaroslav@503	574	continue;
jaroslav@503	575	}
jaroslav@503	576	Model em = ec.getAnnotation(Model.class);
jaroslav@503	577	if (em == null) {
jaroslav@503	578	continue;
jaroslav@503	579	}
jaroslav@503	580	Prprt[] subProps = Prprt.wrap(processingEnv, ec, em.properties());
jaroslav@505	581	allProps.addAll(Arrays.asList(subProps));
jaroslav@503	582	for (Prprt prprt : subProps) {
jaroslav@503	583	PropInfo found = null;
jaroslav@503	584	for (PropInfo existing : mergedProps) {
jaroslav@503	585	if (existing.name.equals(prprt.name())) {
jaroslav@503	586	found = existing;
jaroslav@503	587	break;
jaroslav@503	588	}
jaroslav@503	589	}
jaroslav@503	590	if (found == null) {
jaroslav@503	591	final String tn = typeName(ec, prprt);
jaroslav@503	592	String[] gs = toGetSet(prprt.name(), tn, prprt.array());
jaroslav@503	593	String castTo;
jaroslav@503	594
jaroslav@503	595	if (prprt.array()) {
jaroslav@503	596	castTo = "java.util.List";
jaroslav@503	597	} else {
jaroslav@503	598	castTo = tn;
jaroslav@503	599	}
jaroslav@503	600	found = new PropInfo(
jaroslav@503	601	prprt.name(),
jaroslav@503	602	gs[2],
jaroslav@503	603	gs[3],
jaroslav@503	604	gs[0],
jaroslav@503	605	castTo
jaroslav@503	606	);
jaroslav@503	607	mergedProps.add(found);
jaroslav@503	608	}
jaroslav@503	609	found.addModel(em);
jaroslav@503	610	}
jaroslav@503	611	}
jaroslav@505	612	if (!generateOnChange(e, propsDeps, allProps.toArray(new Prprt[0]), className, functionDeps)) {
jaroslav@505	613	ok = false;
jaroslav@505	614	}
jaroslav@505	615	if (!generateFunctions(e, body, className, e.getEnclosedElements(), functions)) {
jaroslav@505	616	ok = false;
jaroslav@505	617	}
jaroslav@505	618	if (!generateReceive(e, body, className, e.getEnclosedElements(), onReceiveType)) {
jaroslav@505	619	ok = false;
jaroslav@505	620	}
jaroslav@505	621	if (!generateOperation(e, body, className, e.getEnclosedElements())) {
jaroslav@505	622	ok = false;
jaroslav@505	623	}
jaroslav@490	624	FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + className, e);
jaroslav@490	625	w = new OutputStreamWriter(java.openOutputStream());
jaroslav@490	626	try {
jaroslav@490	627	w.append("package " + pkg + ";\n");
jaroslav@490	628	w.append("import net.java.html.json.*;\n");
jtulach@0	629	w.append("public final class ").append(className).append(" implements Cloneable {\n");
jaroslav@373	630	w.append(" private static final Html4JavaType TYPE = new Html4JavaType();\n");
jaroslav@373	631	w.append(" private final org.apidesign.html.json.spi.Proto proto;\n");
jtulach@0	632	w.append(body.toString());
jtulach@0	633	w.append(" private static Class<" + inPckName(e) + "> modelFor() { return null; }\n");
jaroslav@491	634	w.append(" private ").append(className).append("(net.java.html.BrwsrCtx context, Object union) {\n");
jaroslav@497	635	w.append(" this(context, union, null);\n");
jaroslav@491	636	w.append(" union.getClass(); // null check\n");
jaroslav@497	637	w.append(" }\n");
jaroslav@497	638	w.append(" private ").append(className).append("(net.java.html.BrwsrCtx context, Object union, Object json) {\n");
jaroslav@410	639	w.append(" this.proto = TYPE.createProto(this, context);\n");
jaroslav@506	640	for (int i = 0; i < props.length; i++) {
jaroslav@506	641	Prprt p = props[i];
jaroslav@374	642	if (p.array()) {
jaroslav@374	643	final String tn = typeName(e, p);
jaroslav@374	644	String[] gs = toGetSet(p.name(), tn, p.array());
jaroslav@383	645	w.write(" this.prop_" + p.name() + " = proto.createList(\""
jaroslav@374	646	+ p.name() + "\"");
jaroslav@383	647	if (functionDeps.containsKey(p.name())) {
jaroslav@506	648	w.write(", " + i);
jaroslav@383	649	} else {
jaroslav@383	650	w.write(", -1");
jaroslav@383	651	}
jaroslav@374	652	Collection<String> dependants = propsDeps.get(p.name());
jaroslav@374	653	if (dependants != null) {
jaroslav@374	654	for (String depProp : dependants) {
jaroslav@374	655	w.write(", ");
jaroslav@374	656	w.write('\"');
jaroslav@374	657	w.write(depProp);
jaroslav@374	658	w.write('\"');
jaroslav@374	659	}
jaroslav@374	660	}
jaroslav@374	661	w.write(")");
jaroslav@374	662	w.write(";\n");
jaroslav@379	663	}
jaroslav@379	664	}
jaroslav@497	665	w.append(" this.union = union == null ? readUnion(json) : union;\n");
jaroslav@379	666	w.append(" };\n");
jaroslav@490	667	// enum
jaroslav@490	668	StringBuilder factoryHeader = new StringBuilder();
jaroslav@491	669	Prprt firstArray = null;
jaroslav@498	670	w.write(" private final Object union;\n");
jaroslav@498	671
jaroslav@499	672	{
jaroslav@499	673	Model defaultGenerated = null;
jaroslav@499	674	for (Element ec : e.getEnclosedElements()) {
jaroslav@499	675	if (ec.getKind() != ElementKind.ENUM_CONSTANT) {
jaroslav@499	676	continue;
jaroslav@499	677	}
jaroslav@499	678	Model em = ec.getAnnotation(Model.class);
jaroslav@499	679	if (em == null) {
jaroslav@499	680	continue;
jaroslav@499	681	}
jaroslav@499	682	defaultGenerated = em;
jaroslav@499	683	break;
jaroslav@498	684	}
jaroslav@499	685	w.append(" /** New instance with ").append(defaultGenerated.className()).append(" union */");
jaroslav@499	686	w.append(" public ").append(className).append("() {\n");
jaroslav@499	687	if (defaultGenerated != null) {
jaroslav@499	688	w.append(" this(net.java.html.BrwsrCtx.findDefault(").append(className).append(".class),");
jaroslav@499	689	w.append(" new ").append(defaultGenerated.className()).append("());\n");
jaroslav@499	690	} else {
jaroslav@499	691	w.append(" throw new IllegalStateException();\n");
jaroslav@499	692	error("There needs to be at least one enum constant!", e);
jaroslav@499	693	ok = false;
jaroslav@498	694	}
jaroslav@499	695	w.append(" }\n");
jaroslav@498	696	}
jaroslav@498	697	for (Element ec : e.getEnclosedElements()) {
jaroslav@498	698	if (ec.getKind() != ElementKind.ENUM_CONSTANT) {
jaroslav@498	699	continue;
jaroslav@498	700	}
jaroslav@498	701	Model em = ec.getAnnotation(Model.class);
jaroslav@498	702	if (em == null) {
jaroslav@498	703	continue;
jaroslav@498	704	}
jaroslav@498	705	factoryHeader.setLength(0);
jaroslav@490	706
jaroslav@498	707	w.append(" public ").append(className).append("(").append(em.className()).append(" union");
jaroslav@284	708	for (Prprt p : props) {
jaroslav@284	709	if (p.array()) {
jaroslav@284	710	if (firstArray == null) {
jaroslav@284	711	firstArray = p;
jaroslav@284	712	}
jaroslav@284	713	continue;
jaroslav@232	714	}
jaroslav@284	715	String tn = typeName(e, p);
jaroslav@491	716	w.write(", "); factoryHeader.append(", ");
jaroslav@490	717	w.write(tn); factoryHeader.append(tn);
jaroslav@490	718	w.write(" " + p.name()); factoryHeader.append(" " + p.name());
jaroslav@232	719	}
jaroslav@284	720	if (firstArray != null) {
jaroslav@312	721	String tn;
jaroslav@312	722	boolean[] isModel = {false};
jaroslav@312	723	boolean[] isEnum = {false};
jaroslav@312	724	boolean isPrimitive[] = {false};
jaroslav@312	725	tn = checkType(firstArray, isModel, isEnum, isPrimitive);
jaroslav@491	726	w.write(", "); factoryHeader.append(", ");
jaroslav@490	727	w.write(tn); factoryHeader.append(tn);
jaroslav@490	728	w.write("... " + firstArray.name()); factoryHeader.append("... " + firstArray.name());
jaroslav@284	729	}
jaroslav@490	730	// end of enum
jaroslav@284	731	w.append(") {\n");
jaroslav@491	732	w.append(" this(net.java.html.BrwsrCtx.findDefault(").append(className).append(".class), union);\n");
jaroslav@506	733	w.append(" ((" + em.className() + ")union).proto.onValueHasMutated(this.proto);\n");
jaroslav@284	734	for (Prprt p : props) {
jaroslav@284	735	if (p.array()) {
jaroslav@284	736	continue;
jaroslav@284	737	}
jaroslav@284	738	w.write(" this.prop_" + p.name() + " = " + p.name() + ";\n");
jaroslav@284	739	}
jaroslav@284	740	if (firstArray != null) {
jaroslav@383	741	w.write(" proto.initTo(this.prop_" + firstArray.name() + ", " + firstArray.name() + ");\n");
jaroslav@284	742	}
jaroslav@284	743	w.append(" };\n");
jaroslav@232	744	}
jaroslav@373	745	w.append(" private static class Html4JavaType extends org.apidesign.html.json.spi.Proto.Type<").append(className).append("> {\n");
jaroslav@373	746	w.append(" private Html4JavaType() {\n super(").append(className).append(".class, ").
jaroslav@503	747	append(inPckName(e)).append(".class, " + (mergedProps.size()) + ", "
jaroslav@373	748	+ (functions.size() / 2) + ");\n");
jaroslav@275	749	{
jaroslav@503	750	for (int i = 0; i < mergedProps.size(); i++) {
jaroslav@503	751	w.append(" registerProperty(\"").append(mergedProps.get(i).name).append("\", ");
jaroslav@503	752	w.append(i + ", " + (mergedProps.get(i).set == null) + ");\n");
jaroslav@275	753	}
jaroslav@17	754	}
jaroslav@275	755	{
jaroslav@275	756	for (int i = 0; i < functions.size(); i += 2) {
jaroslav@373	757	w.append(" registerFunction(\"").append(functions.get(i)).append("\", ");
jaroslav@373	758	w.append((i / 2) + ");\n");
jaroslav@275	759	}
jaroslav@20	760	}
jaroslav@373	761	w.append(" }\n");
jaroslav@374	762	w.append(" @Override public void setValue(" + className + " data, int type, Object value) {\n");
jaroslav@17	763	w.append(" switch (type) {\n");
jaroslav@503	764	for (int i = 0; i < mergedProps.size(); i++) {
jaroslav@503	765	final PropInfo mp = mergedProps.get(i);
jaroslav@503	766	final String set = mp.set;
jaroslav@503	767	String tn = mp.type;
jaroslav@198	768	String btn = findBoxedType(tn);
jaroslav@198	769	if (btn != null) {
jaroslav@198	770	tn = btn;
jaroslav@68	771	}
jaroslav@17	772	if (set != null) {
jaroslav@503	773	if (mp.models.isEmpty()) {
jaroslav@503	774	w.append(" case " + i + ": data." + strip(set) + "(TYPE.extractValue(" + tn + ".class, value)); return;\n");
jaroslav@503	775	} else {
jaroslav@503	776	w.append(" case " + i + ": {\n");
jaroslav@503	777	for (Model mdl : mp.models) {
jaroslav@503	778	w.append(" if (data.get" + mdl.className() +
jaroslav@503	779	"() != null) data.get" + mdl.className() + "()." +
jaroslav@503	780	strip(set) + "(TYPE.extractValue(" + tn +
jaroslav@503	781	".class, value));\n"
jaroslav@503	782	);
jaroslav@503	783	}
jaroslav@503	784	w.append(" return;\n");
jaroslav@503	785	w.append(" }\n");
jaroslav@503	786	}
jaroslav@17	787	}
jaroslav@17	788	}
jaroslav@17	789	w.append(" }\n");
jaroslav@17	790	w.append(" }\n");
jaroslav@374	791	w.append(" @Override public Object getValue(" + className + " data, int type) {\n");
jaroslav@17	792	w.append(" switch (type) {\n");
jaroslav@503	793	for (int i = 0; i < mergedProps.size(); i++) {
jaroslav@503	794	final PropInfo mp = mergedProps.get(i);
jaroslav@503	795	final String get = mp.get;
jaroslav@17	796	if (get != null) {
jaroslav@503	797	if (mp.models.isEmpty()) {
jaroslav@503	798	w.append(" case " + i + ": return data." + strip(get) + "();\n");
jaroslav@503	799	} else {
jaroslav@503	800	w.append(" case " + i + ": {\n");
jaroslav@503	801	for (Model mdl : mp.models) {
jaroslav@503	802	w.append(" if (data.get" + mdl.className() +
jaroslav@503	803	"() != null) return data.get" + mdl.className() + "()." +
jaroslav@503	804	strip(get) + "();\n"
jaroslav@503	805	);
jaroslav@503	806	}
jaroslav@503	807	w.append(" return null;\n");
jaroslav@503	808	w.append(" }\n");
jaroslav@503	809	}
jaroslav@17	810	}
jaroslav@17	811	}
jaroslav@17	812	w.append(" }\n");
jaroslav@17	813	w.append(" throw new UnsupportedOperationException();\n");
jaroslav@17	814	w.append(" }\n");
jaroslav@374	815	w.append(" @Override public void call(" + className + " model, int type, Object data, Object ev) {\n");
jaroslav@20	816	w.append(" switch (type) {\n");
jaroslav@20	817	for (int i = 0; i < functions.size(); i += 2) {
jaroslav@20	818	final String name = functions.get(i);
jaroslav@20	819	w.append(" case " + (i / 2) + ": model." + name + "(data, ev); return;\n");
jaroslav@20	820	}
jaroslav@20	821	w.append(" }\n");
jaroslav@20	822	w.append(" throw new UnsupportedOperationException();\n");
jaroslav@20	823	w.append(" }\n");
jaroslav@384	824	w.append(" @Override public org.apidesign.html.json.spi.Proto protoFor(Object obj) {\n");
jaroslav@384	825	w.append(" return ((" + className + ")obj).proto;");
jaroslav@384	826	w.append(" }\n");
jaroslav@383	827	w.append(" @Override public void onChange(" + className + " model, int type) {\n");
jaroslav@383	828	w.append(" switch (type) {\n");
jaroslav@383	829	{
jaroslav@506	830	for (int i = 0; i < allProps.size(); i++) {
jaroslav@506	831	String pn = allProps.get(i).name();
jaroslav@506	832	Collection<String> onChange = functionDeps.get(pn);
jaroslav@383	833	if (onChange != null) {
jaroslav@383	834	w.append(" case " + i + ":\n");
jaroslav@383	835	for (String call : onChange) {
jaroslav@383	836	w.append(" ").append(call).append("\n");
jaroslav@383	837	}
jaroslav@383	838	w.write(" return;\n");
jaroslav@383	839	}
jaroslav@383	840	}
jaroslav@383	841	}
jaroslav@506	842	w.append(" }\n");
jaroslav@383	843	w.append(" }\n");
jaroslav@386	844	w.append(onReceiveType);
jaroslav@497	845	w.append(" @Override public " + className + " read(net.java.html.BrwsrCtx c, Object json) {\n");
jaroslav@497	846	w.append(" return new " + className + "(c, null, json);\n");
jaroslav@497	847	w.append(" }\n");
jaroslav@412	848	w.append(" @Override public " + className + " cloneTo(" + className + " o, net.java.html.BrwsrCtx c) { return o.clone(c); }\n");
jaroslav@17	849	w.append(" }\n");
jaroslav@497	850	w.append(" private final Object readUnion(Object json) {\n");
jaroslav@497	851	int values = 1;
jaroslav@500	852	for (int i = 0; i < propsGetSet.size(); i++) {
jaroslav@503	853	Prprt p = findPrprt(props, propsGetSet.get(i).name);
jtulach@0	854	if (p == null) {
jtulach@0	855	continue;
jtulach@0	856	}
jtulach@0	857	values++;
jtulach@0	858	}
jtulach@0	859	w.append(" Object[] ret = new Object[" + values + "];\n");
jaroslav@385	860	w.append(" proto.extract(json, new String[] {\n");
jaroslav@497	861	w.append(" \"").append(e.getSimpleName()).append("\",\n");
jaroslav@500	862	for (int i = 0; i < propsGetSet.size(); i++) {
jaroslav@503	863	Prprt p = findPrprt(props, propsGetSet.get(i).name);
jtulach@0	864	if (p == null) {
jtulach@0	865	continue;
jtulach@0	866	}
jaroslav@503	867	w.append(" \"").append(propsGetSet.get(i).name).append("\",\n");
jtulach@0	868	}
jtulach@0	869	w.append(" }, ret);\n");
jaroslav@497	870	w.append(" Object union;\n");
jaroslav@497	871	w.append(" switch (" + inPckName(e) + ".valueOf(ret[0].toString())) {\n" );
jaroslav@497	872	for (Element ec : e.getEnclosedElements()) {
jaroslav@497	873	if (ec.getKind() != ElementKind.ENUM_CONSTANT) {
jaroslav@497	874	continue;
jaroslav@497	875	}
jaroslav@497	876	Model em = ec.getAnnotation(Model.class);
jaroslav@497	877	w.append(" case ").append(ec.getSimpleName()).append(": union = new ");
jaroslav@497	878	w.append(em.className()).append("(proto.getContext(), json); break;\n");
jaroslav@497	879	}
jaroslav@497	880	w.append(" default: throw new IllegalStateException(ret[0].toString());\n");
jaroslav@497	881	w.append(" }\n" );
jaroslav@500	882	for (int i = 0, cnt = 1, prop = 0; i < propsGetSet.size(); i++) {
jaroslav@503	883	final String pn = propsGetSet.get(i).name;
jtulach@0	884	Prprt p = findPrprt(props, pn);
jtulach@0	885	if (p == null) {
jtulach@0	886	continue;
jtulach@0	887	}
jtulach@0	888	boolean[] isModel = { false };
jtulach@0	889	boolean[] isEnum = { false };
jtulach@0	890	boolean isPrimitive[] = { false };
jtulach@0	891	String type = checkType(props[prop++], isModel, isEnum, isPrimitive);
jtulach@0	892	if (p.array()) {
jaroslav@66	893	w.append(" if (ret[" + cnt + "] instanceof Object[]) {\n");
jaroslav@66	894	w.append(" for (Object e : ((Object[])ret[" + cnt + "])) {\n");
jtulach@0	895	if (isModel[0]) {
jaroslav@385	896	w.append(" this.prop_").append(pn).append(".add(proto.read");
jaroslav@385	897	w.append("(" + type + ".class, e));\n");
jtulach@0	898	} else if (isEnum[0]) {
jaroslav@66	899	w.append(" this.prop_").append(pn);
jtulach@0	900	w.append(".add(e == null ? null : ");
jaroslav@380	901	w.append(type).append(".valueOf(TYPE.stringValue(e)));\n");
jtulach@0	902	} else {
jtulach@0	903	if (isPrimitive(type)) {
jaroslav@380	904	w.append(" this.prop_").append(pn).append(".add(TYPE.numberValue(e).");
jtulach@0	905	w.append(type).append("Value());\n");
jtulach@0	906	} else {
jaroslav@66	907	w.append(" this.prop_").append(pn).append(".add((");
jtulach@0	908	w.append(type).append(")e);\n");
jtulach@0	909	}
jtulach@0	910	}
jaroslav@66	911	w.append(" }\n");
jaroslav@66	912	w.append(" }\n");
jtulach@0	913	} else {
jtulach@0	914	if (isEnum[0]) {
jaroslav@340	915	w.append(" try {\n");
jtulach@0	916	w.append(" this.prop_").append(pn);
jtulach@0	917	w.append(" = ret[" + cnt + "] == null ? null : ");
jaroslav@380	918	w.append(type).append(".valueOf(TYPE.stringValue(ret[" + cnt + "]));\n");
jaroslav@340	919	w.append(" } catch (IllegalArgumentException ex) {\n");
jaroslav@340	920	w.append(" ex.printStackTrace();\n");
jaroslav@340	921	w.append(" }\n");
jtulach@0	922	} else if (isPrimitive(type)) {
jtulach@0	923	w.append(" this.prop_").append(pn);
jaroslav@100	924	w.append(" = ret[" + cnt + "] == null ? ");
jaroslav@100	925	if ("char".equals(type)) {
jaroslav@380	926	w.append("0 : (TYPE.charValue(");
jaroslav@100	927	} else if ("boolean".equals(type)) {
jaroslav@380	928	w.append("false : (TYPE.boolValue(");
jaroslav@100	929	} else {
jaroslav@380	930	w.append("0 : (TYPE.numberValue(");
jaroslav@100	931	}
jaroslav@197	932	w.append("ret[" + cnt + "])).");
jtulach@0	933	w.append(type).append("Value();\n");
jaroslav@136	934	} else if (isModel[0]) {
jaroslav@385	935	w.append(" this.prop_").append(pn).append(" = proto.read");
jaroslav@385	936	w.append("(" + type + ".class, ");
jaroslav@136	937	w.append("ret[" + cnt + "]);\n");
jaroslav@136	938	}else {
jtulach@0	939	w.append(" this.prop_").append(pn);
jtulach@0	940	w.append(" = (").append(type).append(')');
jtulach@0	941	w.append("ret[" + cnt + "];\n");
jtulach@0	942	}
jtulach@0	943	}
jtulach@0	944	cnt++;
jtulach@0	945	}
jaroslav@497	946	w.append(" return union;\n" );
jaroslav@373	947	w.append(" }\n");
jaroslav@495	948	writeEnumToString(e, props, w);
jaroslav@493	949	writeEnumClone(e, className, props, w);
jaroslav@311	950	w.write(" /** Activates this model instance in the current {@link \n"
jaroslav@311	951	+ "net.java.html.json.Models#bind(java.lang.Object, net.java.html.BrwsrCtx) browser context}. \n"
jaroslav@311	952	+ "In case of using Knockout technology, this means to \n"
jaroslav@311	953	+ "bind JSON like data in this model instance with Knockout tags in \n"
jaroslav@311	954	+ "the surrounding HTML page.\n"
jaroslav@311	955	+ "*/\n"
jaroslav@230	956	);
jaroslav@15	957	w.write(" public " + className + " applyBindings() {\n");
jaroslav@373	958	w.write(" proto.applyBindings();\n");
jaroslav@15	959	w.write(" return this;\n");
jaroslav@15	960	w.write(" }\n");
jaroslav@15	961	w.write(" public boolean equals(Object o) {\n");
jaroslav@15	962	w.write(" if (o == this) return true;\n");
jaroslav@281	963	w.write(" if (!(o instanceof " + className + ")) return false;\n");
jaroslav@281	964	w.write(" " + className + " p = (" + className + ")o;\n");
jaroslav@281	965	for (Prprt p : props) {
jaroslav@380	966	w.write(" if (!TYPE.isSame(prop_" + p.name() + ", p.prop_" + p.name() + ")) return false;\n");
jaroslav@281	967	}
jaroslav@281	968	w.write(" return true;\n");
jaroslav@281	969	w.write(" }\n");
jaroslav@281	970	w.write(" public int hashCode() {\n");
jaroslav@281	971	w.write(" int h = " + className + ".class.getName().hashCode();\n");
jaroslav@281	972	for (Prprt p : props) {
jaroslav@380	973	w.write(" h = TYPE.hashPlus(prop_" + p.name() + ", h);\n");
jaroslav@281	974	}
jaroslav@281	975	w.write(" return h;\n");
jaroslav@15	976	w.write(" }\n");
jaroslav@490	977	// enum
jaroslav@492	978	w.write(" public " + inPckName(e) + " get" + e.getSimpleName() + "() {\n");
jaroslav@492	979	for (Element ec : e.getEnclosedElements()) {
jaroslav@492	980	if (ec.getKind() != ElementKind.ENUM_CONSTANT) {
jaroslav@492	981	continue;
jaroslav@492	982	}
jaroslav@496	983	Model em = ec.getAnnotation(Model.class);
jaroslav@496	984	if (em == null) {
jaroslav@496	985	continue;
jaroslav@496	986	}
jaroslav@496	987	w.write(" if (union instanceof " + em.className()+ ") return " + inPckName(e) + "." + ec.getSimpleName() + ";\n");
jaroslav@492	988	}
jaroslav@492	989	w.write(" return null;\n");
jaroslav@492	990	w.write(" }\n");
jaroslav@502	991	ArrayList<PropInfo> sharedProps = new ArrayList<PropInfo>(propsGetSet);
jaroslav@502	992	final PropInfo switchProp = new PropInfo(e.getSimpleName().toString(), null, null, null, null);
jaroslav@502	993	if (sharedProps.contains(switchProp)) {
jaroslav@502	994	ok = false;
jaroslav@503	995	error("Duplicated property " + switchProp.name + " due to enum switch", e);
jaroslav@502	996	} else {
jaroslav@502	997	sharedProps.add(switchProp);
jaroslav@502	998	}
jaroslav@490	999	for (Element ec : e.getEnclosedElements()) {
jaroslav@490	1000	if (ec.getKind() != ElementKind.ENUM_CONSTANT) {
jaroslav@490	1001	continue;
jaroslav@490	1002	}
jaroslav@496	1003	Model em = ec.getAnnotation(Model.class);
jaroslav@496	1004	if (em == null) {
jaroslav@496	1005	continue;
jaroslav@496	1006	}
jaroslav@496	1007	w.write(" public " + em.className() + " get" + em.className()+ "() {\n");
jaroslav@496	1008	w.write(" return union instanceof " + em.className()+ " ? (" + em.className() + ")union : null;\n");
jaroslav@491	1009	w.write(" }\n");
jaroslav@498	1010	if (props.length > 0) {
jaroslav@498	1011	w.write(" public " + className + "(" + em.className() + " union) {\n");
jaroslav@498	1012	w.append(" this(net.java.html.BrwsrCtx.findDefault(").append(className).append(".class), union, null);\n");
jaroslav@506	1013	w.append(" ((" + em.className() + ")union).proto.onValueHasMutated(this.proto);\n");
jaroslav@498	1014	w.write(" }\n");
jaroslav@491	1015	}
jaroslav@502	1016	generateEnumConstantModel(w, ec, sharedProps);
jaroslav@490	1017	}
jaroslav@490	1018	// end of enum
jaroslav@14	1019	w.write("}\n");
jtulach@0	1020	} finally {
jtulach@0	1021	w.close();
jtulach@0	1022	}
jtulach@0	1023	} catch (IOException ex) {
jtulach@0	1024	error("Can't create " + className + ".java", e);
jtulach@0	1025	return false;
jtulach@0	1026	}
jtulach@0	1027	return ok;
jtulach@0	1028	}
jtulach@0	1029
jaroslav@502	1030	private boolean generateEnumConstantModel(Writer w, Element ec, Collection<PropInfo> taken) throws IOException {
jaroslav@490	1031	Model m = ec.getAnnotation(Model.class);
jaroslav@490	1032	if (m == null) {
jaroslav@490	1033	error("Each field in an enum needs to be annotated by @Model", ec);
jaroslav@490	1034	return false;
jaroslav@490	1035	}
jaroslav@490	1036	boolean ok = true;
jaroslav@490	1037
jaroslav@490	1038	String className = m.className();
jaroslav@490	1039
jaroslav@490	1040	StringWriter body = new StringWriter();
jaroslav@490	1041	StringBuilder onReceiveType = new StringBuilder();
jaroslav@500	1042	List<PropInfo> propsGetSet = new ArrayList<PropInfo>();
jaroslav@490	1043	List<String> functions = new ArrayList<String>();
jaroslav@490	1044	Map<String, Collection<String>> propsDeps = new HashMap<String, Collection<String>>();
jaroslav@490	1045	Map<String, Collection<String>> functionDeps = new HashMap<String, Collection<String>>();
jaroslav@490	1046	Prprt[] props = createProps(ec, m.properties());
jaroslav@490	1047
jaroslav@490	1048	// if (!generateComputedProperties(body, props, e.getEnclosedElements(), propsGetSet, propsDeps)) {
jaroslav@490	1049	// ok = false;
jaroslav@490	1050	// }
jaroslav@490	1051	// if (!generateOnChange(e, propsDeps, props, className, functionDeps)) {
jaroslav@490	1052	// ok = false;
jaroslav@490	1053	// }
jaroslav@502	1054	if (!generateProperties(ec, body, className, props, propsGetSet, taken, propsDeps, functionDeps)) {
jaroslav@490	1055	ok = false;
jaroslav@490	1056	}
jaroslav@490	1057	// if (!generateFunctions(e, body, className, e.getEnclosedElements(), functions)) {
jaroslav@490	1058	// ok = false;
jaroslav@490	1059	// }
jaroslav@490	1060	// if (!generateReceive(e, body, className, e.getEnclosedElements(), onReceiveType)) {
jaroslav@490	1061	// ok = false;
jaroslav@490	1062	// }
jaroslav@490	1063	onReceiveType.append(" @Override public void onMessage(").append(className).append(" model, int index, int type, Object data) {\n");
jaroslav@490	1064	onReceiveType.append(" }\n");
jaroslav@490	1065
jaroslav@490	1066	// if (!generateOperation(e, body, className, e.getEnclosedElements())) {
jaroslav@490	1067	// ok = false;
jaroslav@490	1068	// }
jaroslav@490	1069	w.write(" public static final ");
jaroslav@502	1070	generateClassBody(w, className, body, ec, props,
jaroslav@502	1071	functionDeps, propsDeps, propsGetSet, taken,
jaroslav@502	1072	functions, onReceiveType
jaroslav@502	1073	);
jaroslav@490	1074	return ok;
jaroslav@490	1075	}
jaroslav@490	1076
jtulach@0	1077	private boolean generateProperties(
jtulach@0	1078	Element where,
jaroslav@383	1079	Writer w, String className, Prprt[] properties,
jaroslav@502	1080	Collection<PropInfo> props, Collection<PropInfo> taken,
jtulach@0	1081	Map<String,Collection<String>> deps,
jtulach@0	1082	Map<String,Collection<String>> functionDeps
jtulach@0	1083	) throws IOException {
jtulach@0	1084	boolean ok = true;
jtulach@0	1085	for (Prprt p : properties) {
jtulach@0	1086	final String tn;
jtulach@0	1087	tn = typeName(where, p);
jtulach@0	1088	String[] gs = toGetSet(p.name(), tn, p.array());
jaroslav@17	1089	String castTo;
jaroslav@17	1090
jtulach@0	1091	if (p.array()) {
jaroslav@383	1092	w.write(" private final java.util.List<" + tn + "> prop_" + p.name() + ";\n");
jaroslav@17	1093
jaroslav@17	1094	castTo = "java.util.List";
jaroslav@66	1095	w.write(" public java.util.List<" + tn + "> " + gs[0] + "() {\n");
jaroslav@402	1096	w.write(" proto.verifyUnlocked();\n");
jaroslav@66	1097	w.write(" return prop_" + p.name() + ";\n");
jaroslav@66	1098	w.write(" }\n");
jtulach@0	1099	} else {
jaroslav@17	1100	castTo = tn;
jaroslav@66	1101	w.write(" private " + tn + " prop_" + p.name() + ";\n");
jaroslav@66	1102	w.write(" public " + tn + " " + gs[0] + "() {\n");
jaroslav@402	1103	w.write(" proto.verifyUnlocked();\n");
jaroslav@66	1104	w.write(" return prop_" + p.name() + ";\n");
jaroslav@66	1105	w.write(" }\n");
jaroslav@66	1106	w.write(" public void " + gs[1] + "(" + tn + " v) {\n");
jaroslav@402	1107	w.write(" proto.verifyUnlocked();\n");
jaroslav@380	1108	w.write(" if (TYPE.isSame(prop_" + p.name() + ", v)) return;\n");
jaroslav@66	1109	w.write(" prop_" + p.name() + " = v;\n");
jaroslav@373	1110	w.write(" proto.valueHasMutated(\"" + p.name() + "\");\n");
jtulach@0	1111	Collection<String> dependants = deps.get(p.name());
jtulach@0	1112	if (dependants != null) {
jtulach@0	1113	for (String depProp : dependants) {
jaroslav@373	1114	w.write(" proto.valueHasMutated(\"" + depProp + "\");\n");
jaroslav@66	1115	}
jaroslav@66	1116	}
jaroslav@66	1117	dependants = functionDeps.get(p.name());
jaroslav@66	1118	if (dependants != null) {
jaroslav@383	1119	w.append(className).append(" model = ").append(className).append(".this;\n");
jaroslav@66	1120	for (String call : dependants) {
jaroslav@66	1121	w.append(" ").append(call);
jtulach@0	1122	}
jtulach@0	1123	}
jtulach@0	1124	w.write(" }\n");
jtulach@0	1125	}
jaroslav@501	1126	final PropInfo pi = new PropInfo(
jaroslav@500	1127	p.name(),
jaroslav@500	1128	gs[2],
jaroslav@500	1129	gs[3],
jaroslav@500	1130	gs[0],
jaroslav@500	1131	castTo
jaroslav@501	1132	);
jaroslav@502	1133	if (props.contains(pi) \|\| (taken != null && taken.contains(pi))) {
jaroslav@501	1134	ok = false;
jaroslav@501	1135	error("Duplicated property " + p.name(), where);
jaroslav@501	1136	}
jaroslav@501	1137
jaroslav@501	1138	props.add(pi);
jtulach@0	1139	}
jtulach@0	1140	return ok;
jtulach@0	1141	}
jtulach@0	1142
jtulach@0	1143	private boolean generateComputedProperties(
jtulach@0	1144	Writer w, Prprt[] fixedProps,
jaroslav@500	1145	Collection<? extends Element> arr, Collection<PropInfo> props,
jtulach@0	1146	Map<String,Collection<String>> deps
jtulach@0	1147	) throws IOException {
jtulach@0	1148	boolean ok = true;
jtulach@0	1149	for (Element e : arr) {
jtulach@0	1150	if (e.getKind() != ElementKind.METHOD) {
jtulach@0	1151	continue;
jtulach@0	1152	}
jtulach@0	1153	if (e.getAnnotation(ComputedProperty.class) == null) {
jtulach@0	1154	continue;
jtulach@0	1155	}
jaroslav@227	1156	if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@227	1157	error("Method " + e.getSimpleName() + " has to be static when annotated by @ComputedProperty", e);
jaroslav@227	1158	ok = false;
jaroslav@227	1159	continue;
jaroslav@227	1160	}
jtulach@0	1161	ExecutableElement ee = (ExecutableElement)e;
jtulach@0	1162	final TypeMirror rt = ee.getReturnType();
jtulach@0	1163	final Types tu = processingEnv.getTypeUtils();
jtulach@0	1164	TypeMirror ert = tu.erasure(rt);
jtulach@0	1165	String tn = fqn(ert, ee);
jtulach@0	1166	boolean array = false;
jaroslav@235	1167	final TypeMirror toCheck;
jtulach@0	1168	if (tn.equals("java.util.List")) {
jtulach@0	1169	array = true;
jaroslav@235	1170	toCheck = ((DeclaredType)rt).getTypeArguments().get(0);
jaroslav@235	1171	} else {
jaroslav@235	1172	toCheck = rt;
jtulach@0	1173	}
jtulach@0	1174
jtulach@0	1175	final String sn = ee.getSimpleName().toString();
jaroslav@235	1176
jaroslav@235	1177	if (toCheck.getKind().isPrimitive()) {
jaroslav@235	1178	// OK
jaroslav@235	1179	} else {
jaroslav@235	1180	TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jaroslav@235	1181	TypeMirror enumType = processingEnv.getElementUtils().getTypeElement("java.lang.Enum").asType();
jaroslav@235	1182
jaroslav@235	1183	if (tu.isSubtype(toCheck, stringType)) {
jaroslav@235	1184	// OK
jaroslav@252	1185	} else if (tu.isSubtype(tu.erasure(toCheck), tu.erasure(enumType))) {
jaroslav@235	1186	// OK
jaroslav@236	1187	} else if (isModel(toCheck)) {
jaroslav@236	1188	// OK
jaroslav@235	1189	} else {
jaroslav@235	1190	ok = false;
jaroslav@235	1191	error(sn + " cannot return " + toCheck, e);
jaroslav@235	1192	}
jaroslav@235	1193	}
jaroslav@235	1194
jtulach@0	1195	String[] gs = toGetSet(sn, tn, array);
jtulach@0	1196
jaroslav@66	1197	w.write(" public " + tn + " " + gs[0] + "() {\n");
jtulach@0	1198	int arg = 0;
jtulach@0	1199	for (VariableElement pe : ee.getParameters()) {
jtulach@0	1200	final String dn = pe.getSimpleName().toString();
jtulach@0	1201
jtulach@0	1202	if (!verifyPropName(pe, dn, fixedProps)) {
jtulach@0	1203	ok = false;
jtulach@0	1204	}
jtulach@0	1205
jtulach@0	1206	final String dt = fqn(pe.asType(), ee);
jtulach@0	1207	String[] call = toGetSet(dn, dt, false);
jaroslav@66	1208	w.write(" " + dt + " arg" + (++arg) + " = ");
jtulach@0	1209	w.write(call[0] + "();\n");
jtulach@0	1210
jtulach@0	1211	Collection<String> depends = deps.get(dn);
jtulach@0	1212	if (depends == null) {
jaroslav@100	1213	depends = new LinkedHashSet<String>();
jtulach@0	1214	deps.put(dn, depends);
jtulach@0	1215	}
jtulach@0	1216	depends.add(sn);
jtulach@0	1217	}
jaroslav@66	1218	w.write(" try {\n");
jaroslav@373	1219	w.write(" proto.acquireLock();\n");
jaroslav@66	1220	w.write(" return " + fqn(ee.getEnclosingElement().asType(), ee) + '.' + e.getSimpleName() + "(");
jtulach@0	1221	String sep = "";
jtulach@0	1222	for (int i = 1; i <= arg; i++) {
jtulach@0	1223	w.write(sep);
jtulach@0	1224	w.write("arg" + i);
jtulach@0	1225	sep = ", ";
jtulach@0	1226	}
jtulach@0	1227	w.write(");\n");
jaroslav@66	1228	w.write(" } finally {\n");
jaroslav@373	1229	w.write(" proto.releaseLock();\n");
jaroslav@66	1230	w.write(" }\n");
jtulach@0	1231	w.write(" }\n");
jaroslav@501	1232	final PropInfo pi = new PropInfo(
jaroslav@500	1233	e.getSimpleName().toString(),
jaroslav@500	1234	gs[2],
jaroslav@500	1235	null,
jaroslav@500	1236	gs[0],
jaroslav@500	1237	tn
jaroslav@501	1238	);
jaroslav@501	1239	if (props.contains(pi)) {
jaroslav@501	1240	ok = false;
jaroslav@501	1241	error("Duplicated property " + e.getSimpleName(), e);
jaroslav@501	1242	}
jaroslav@501	1243
jaroslav@501	1244	props.add(pi);
jtulach@0	1245	}
jtulach@0	1246
jtulach@0	1247	return ok;
jtulach@0	1248	}
jtulach@0	1249
jtulach@0	1250	private static String[] toGetSet(String name, String type, boolean array) {
jtulach@0	1251	String n = Character.toUpperCase(name.charAt(0)) + name.substring(1);
jtulach@0	1252	String bck2brwsrType = "L" + type.replace('.', '_') + "_2";
jtulach@0	1253	if ("int".equals(type)) {
jtulach@0	1254	bck2brwsrType = "I";
jtulach@0	1255	}
jtulach@0	1256	if ("double".equals(type)) {
jtulach@0	1257	bck2brwsrType = "D";
jtulach@0	1258	}
jtulach@0	1259	String pref = "get";
jtulach@0	1260	if ("boolean".equals(type)) {
jtulach@0	1261	pref = "is";
jtulach@0	1262	bck2brwsrType = "Z";
jtulach@0	1263	}
jtulach@0	1264	final String nu = n.replace('.', '_');
jtulach@0	1265	if (array) {
jtulach@0	1266	return new String[] {
jtulach@0	1267	"get" + n,
jtulach@0	1268	null,
jtulach@0	1269	"get" + nu + "__Ljava_util_List_2",
jtulach@0	1270	null
jtulach@0	1271	};
jtulach@0	1272	}
jtulach@0	1273	return new String[]{
jtulach@0	1274	pref + n,
jtulach@0	1275	"set" + n,
jtulach@0	1276	pref + nu + "__" + bck2brwsrType,
jtulach@0	1277	"set" + nu + "__V" + bck2brwsrType
jtulach@0	1278	};
jtulach@0	1279	}
jtulach@0	1280
jtulach@0	1281	private String typeName(Element where, Prprt p) {
jtulach@0	1282	String ret;
jtulach@0	1283	boolean[] isModel = { false };
jtulach@0	1284	boolean[] isEnum = { false };
jtulach@0	1285	boolean isPrimitive[] = { false };
jtulach@0	1286	ret = checkType(p, isModel, isEnum, isPrimitive);
jtulach@0	1287	if (p.array()) {
jtulach@0	1288	String bt = findBoxedType(ret);
jtulach@0	1289	if (bt != null) {
jtulach@0	1290	return bt;
jtulach@0	1291	}
jtulach@0	1292	}
jtulach@0	1293	return ret;
jtulach@0	1294	}
jtulach@0	1295
jtulach@0	1296	private static String findBoxedType(String ret) {
jtulach@0	1297	if (ret.equals("boolean")) {
jtulach@0	1298	return Boolean.class.getName();
jtulach@0	1299	}
jtulach@0	1300	if (ret.equals("byte")) {
jtulach@0	1301	return Byte.class.getName();
jtulach@0	1302	}
jtulach@0	1303	if (ret.equals("short")) {
jtulach@0	1304	return Short.class.getName();
jtulach@0	1305	}
jtulach@0	1306	if (ret.equals("char")) {
jtulach@0	1307	return Character.class.getName();
jtulach@0	1308	}
jtulach@0	1309	if (ret.equals("int")) {
jtulach@0	1310	return Integer.class.getName();
jtulach@0	1311	}
jtulach@0	1312	if (ret.equals("long")) {
jtulach@0	1313	return Long.class.getName();
jtulach@0	1314	}
jtulach@0	1315	if (ret.equals("float")) {
jtulach@0	1316	return Float.class.getName();
jtulach@0	1317	}
jtulach@0	1318	if (ret.equals("double")) {
jtulach@0	1319	return Double.class.getName();
jtulach@0	1320	}
jtulach@0	1321	return null;
jtulach@0	1322	}
jtulach@0	1323
jtulach@0	1324	private boolean verifyPropName(Element e, String propName, Prprt[] existingProps) {
jtulach@0	1325	StringBuilder sb = new StringBuilder();
jtulach@0	1326	String sep = "";
jtulach@0	1327	for (Prprt Prprt : existingProps) {
jtulach@0	1328	if (Prprt.name().equals(propName)) {
jtulach@0	1329	return true;
jtulach@0	1330	}
jtulach@0	1331	sb.append(sep);
jtulach@0	1332	sb.append('"');
jtulach@0	1333	sb.append(Prprt.name());
jtulach@0	1334	sb.append('"');
jtulach@0	1335	sep = ", ";
jtulach@0	1336	}
jtulach@0	1337	error(
jtulach@0	1338	propName + " is not one of known properties: " + sb
jtulach@0	1339	, e
jtulach@0	1340	);
jtulach@0	1341	return false;
jtulach@0	1342	}
jtulach@0	1343
jtulach@0	1344	private static String findPkgName(Element e) {
jtulach@0	1345	for (;;) {
jtulach@0	1346	if (e.getKind() == ElementKind.PACKAGE) {
jtulach@0	1347	return ((PackageElement)e).getQualifiedName().toString();
jtulach@0	1348	}
jtulach@0	1349	e = e.getEnclosingElement();
jtulach@0	1350	}
jtulach@0	1351	}
jtulach@0	1352
jtulach@0	1353	private boolean generateFunctions(
jtulach@0	1354	Element clazz, StringWriter body, String className,
jtulach@0	1355	List<? extends Element> enclosedElements, List<String> functions
jtulach@0	1356	) {
jtulach@0	1357	for (Element m : enclosedElements) {
jtulach@0	1358	if (m.getKind() != ElementKind.METHOD) {
jtulach@0	1359	continue;
jtulach@0	1360	}
jtulach@0	1361	ExecutableElement e = (ExecutableElement)m;
jtulach@0	1362	Function onF = e.getAnnotation(Function.class);
jtulach@0	1363	if (onF == null) {
jtulach@0	1364	continue;
jtulach@0	1365	}
jtulach@0	1366	if (!e.getModifiers().contains(Modifier.STATIC)) {
jtulach@0	1367	error("@OnFunction method needs to be static", e);
jtulach@0	1368	return false;
jtulach@0	1369	}
jtulach@0	1370	if (e.getModifiers().contains(Modifier.PRIVATE)) {
jtulach@0	1371	error("@OnFunction method cannot be private", e);
jtulach@0	1372	return false;
jtulach@0	1373	}
jtulach@0	1374	if (e.getReturnType().getKind() != TypeKind.VOID) {
jtulach@0	1375	error("@OnFunction method should return void", e);
jtulach@0	1376	return false;
jtulach@0	1377	}
jtulach@0	1378	String n = e.getSimpleName().toString();
jaroslav@66	1379	body.append(" private void ").append(n).append("(Object data, Object ev) {\n");
jaroslav@199	1380	body.append(" ").append(((TypeElement)clazz).getQualifiedName()).append(".").append(n).append("(");
jtulach@0	1381	body.append(wrapParams(e, null, className, "ev", "data"));
jtulach@0	1382	body.append(");\n");
jaroslav@66	1383	body.append(" }\n");
jtulach@0	1384
jtulach@0	1385	functions.add(n);
jtulach@0	1386	functions.add(n + "__VLjava_lang_Object_2Ljava_lang_Object_2");
jtulach@0	1387	}
jtulach@0	1388	return true;
jtulach@0	1389	}
jtulach@0	1390
jtulach@0	1391	private boolean generateOnChange(Element clazz, Map<String,Collection<String>> propDeps,
jtulach@0	1392	Prprt[] properties, String className,
jtulach@0	1393	Map<String, Collection<String>> functionDeps
jtulach@0	1394	) {
jtulach@0	1395	for (Element m : clazz.getEnclosedElements()) {
jtulach@0	1396	if (m.getKind() != ElementKind.METHOD) {
jtulach@0	1397	continue;
jtulach@0	1398	}
jtulach@0	1399	ExecutableElement e = (ExecutableElement) m;
jtulach@0	1400	OnPropertyChange onPC = e.getAnnotation(OnPropertyChange.class);
jtulach@0	1401	if (onPC == null) {
jtulach@0	1402	continue;
jtulach@0	1403	}
jtulach@0	1404	for (String pn : onPC.value()) {
jtulach@0	1405	if (findPrprt(properties, pn) == null && findDerivedFrom(propDeps, pn).isEmpty()) {
jaroslav@504	1406	error("No property named '" + pn + "' in the model", clazz);
jtulach@0	1407	return false;
jtulach@0	1408	}
jtulach@0	1409	}
jtulach@0	1410	if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@504	1411	error("@OnPropertyChange method needs to be static", e);
jtulach@0	1412	return false;
jtulach@0	1413	}
jtulach@0	1414	if (e.getModifiers().contains(Modifier.PRIVATE)) {
jaroslav@504	1415	error("@OnPropertyChange method cannot be private", e);
jtulach@0	1416	return false;
jtulach@0	1417	}
jtulach@0	1418	if (e.getReturnType().getKind() != TypeKind.VOID) {
jaroslav@504	1419	error("@OnPropertyChange method should return void", e);
jtulach@0	1420	return false;
jtulach@0	1421	}
jtulach@0	1422	String n = e.getSimpleName().toString();
jtulach@0	1423
jtulach@0	1424
jtulach@0	1425	for (String pn : onPC.value()) {
jtulach@0	1426	StringBuilder call = new StringBuilder();
jaroslav@505	1427	call.append(" ").append(inPckName(clazz)).append(".").append(n).append("(");
jtulach@0	1428	call.append(wrapPropName(e, className, "name", pn));
jtulach@0	1429	call.append(");\n");
jtulach@0	1430
jtulach@0	1431	Collection<String> change = functionDeps.get(pn);
jtulach@0	1432	if (change == null) {
jaroslav@100	1433	change = new ArrayList<String>();
jtulach@0	1434	functionDeps.put(pn, change);
jtulach@0	1435	}
jtulach@0	1436	change.add(call.toString());
jtulach@0	1437	for (String dpn : findDerivedFrom(propDeps, pn)) {
jtulach@0	1438	change = functionDeps.get(dpn);
jtulach@0	1439	if (change == null) {
jaroslav@100	1440	change = new ArrayList<String>();
jtulach@0	1441	functionDeps.put(dpn, change);
jtulach@0	1442	}
jtulach@0	1443	change.add(call.toString());
jtulach@0	1444	}
jtulach@0	1445	}
jtulach@0	1446	}
jtulach@0	1447	return true;
jtulach@0	1448	}
jaroslav@240	1449
jaroslav@240	1450	private boolean generateOperation(Element clazz,
jaroslav@240	1451	StringWriter body, String className,
jaroslav@240	1452	List<? extends Element> enclosedElements
jaroslav@240	1453	) {
jaroslav@240	1454	for (Element m : enclosedElements) {
jaroslav@240	1455	if (m.getKind() != ElementKind.METHOD) {
jaroslav@240	1456	continue;
jaroslav@240	1457	}
jaroslav@240	1458	ExecutableElement e = (ExecutableElement)m;
jaroslav@240	1459	ModelOperation mO = e.getAnnotation(ModelOperation.class);
jaroslav@240	1460	if (mO == null) {
jaroslav@240	1461	continue;
jaroslav@240	1462	}
jaroslav@240	1463	if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@240	1464	error("@ModelOperation method needs to be static", e);
jaroslav@240	1465	return false;
jaroslav@240	1466	}
jaroslav@240	1467	if (e.getModifiers().contains(Modifier.PRIVATE)) {
jaroslav@240	1468	error("@ModelOperation method cannot be private", e);
jaroslav@240	1469	return false;
jaroslav@240	1470	}
jaroslav@240	1471	if (e.getReturnType().getKind() != TypeKind.VOID) {
jaroslav@240	1472	error("@ModelOperation method should return void", e);
jaroslav@240	1473	return false;
jaroslav@240	1474	}
jaroslav@240	1475	List<String> args = new ArrayList<String>();
jaroslav@240	1476	{
jaroslav@240	1477	body.append(" public void ").append(m.getSimpleName()).append("(");
jaroslav@240	1478	String sep = "";
jaroslav@240	1479	boolean checkFirst = true;
jaroslav@240	1480	for (VariableElement ve : e.getParameters()) {
jaroslav@240	1481	final TypeMirror type = ve.asType();
jaroslav@240	1482	CharSequence simpleName;
jaroslav@240	1483	if (type.getKind() == TypeKind.DECLARED) {
jaroslav@240	1484	simpleName = ((DeclaredType)type).asElement().getSimpleName();
jaroslav@240	1485	} else {
jaroslav@240	1486	simpleName = type.toString();
jaroslav@240	1487	}
jaroslav@240	1488	if (simpleName.toString().equals(className)) {
jaroslav@240	1489	checkFirst = false;
jaroslav@240	1490	} else {
jaroslav@240	1491	if (checkFirst) {
jaroslav@240	1492	error("First parameter of @ModelOperation method must be " + className, m);
jaroslav@240	1493	return false;
jaroslav@240	1494	}
jaroslav@240	1495	args.add(ve.getSimpleName().toString());
jaroslav@240	1496	body.append(sep).append("final ");
jaroslav@240	1497	body.append(ve.asType().toString()).append(" ");
jaroslav@240	1498	body.append(ve.toString());
jaroslav@240	1499	sep = ", ";
jaroslav@240	1500	}
jaroslav@240	1501	}
jaroslav@240	1502	body.append(") {\n");
jaroslav@381	1503	body.append(" proto.runInBrowser(new Runnable() { public void run() {\n");
jaroslav@240	1504	body.append(" ").append(clazz.getSimpleName()).append(".").append(m.getSimpleName()).append("(");
jaroslav@240	1505	body.append(className).append(".this");
jaroslav@240	1506	for (String s : args) {
jaroslav@240	1507	body.append(", ").append(s);
jaroslav@240	1508	}
jaroslav@240	1509	body.append(");\n");
jaroslav@240	1510	body.append(" }});\n");
jaroslav@240	1511	body.append(" }\n");
jaroslav@240	1512	}
jaroslav@240	1513
jaroslav@240	1514	}
jaroslav@240	1515	return true;
jaroslav@240	1516	}
jaroslav@240	1517
jtulach@0	1518
jtulach@0	1519	private boolean generateReceive(
jtulach@0	1520	Element clazz, StringWriter body, String className,
jaroslav@386	1521	List<? extends Element> enclosedElements, StringBuilder inType
jtulach@0	1522	) {
jaroslav@386	1523	inType.append(" @Override public void onMessage(").append(className).append(" model, int index, int type, Object data) {\n");
jaroslav@386	1524	inType.append(" switch (index) {\n");
jaroslav@386	1525	int index = 0;
jtulach@0	1526	for (Element m : enclosedElements) {
jtulach@0	1527	if (m.getKind() != ElementKind.METHOD) {
jtulach@0	1528	continue;
jtulach@0	1529	}
jtulach@0	1530	ExecutableElement e = (ExecutableElement)m;
jtulach@0	1531	OnReceive onR = e.getAnnotation(OnReceive.class);
jtulach@0	1532	if (onR == null) {
jtulach@0	1533	continue;
jtulach@0	1534	}
jtulach@0	1535	if (!e.getModifiers().contains(Modifier.STATIC)) {
jtulach@0	1536	error("@OnReceive method needs to be static", e);
jtulach@0	1537	return false;
jtulach@0	1538	}
jtulach@0	1539	if (e.getModifiers().contains(Modifier.PRIVATE)) {
jtulach@0	1540	error("@OnReceive method cannot be private", e);
jtulach@0	1541	return false;
jtulach@0	1542	}
jtulach@0	1543	if (e.getReturnType().getKind() != TypeKind.VOID) {
jtulach@0	1544	error("@OnReceive method should return void", e);
jtulach@0	1545	return false;
jtulach@0	1546	}
jaroslav@73	1547	if (!onR.jsonp().isEmpty() && !"GET".equals(onR.method())) {
jaroslav@73	1548	error("JSONP works only with GET transport method", e);
jaroslav@73	1549	}
jaroslav@73	1550	String dataMirror = findDataSpecified(e, onR);
jaroslav@72	1551	if ("PUT".equals(onR.method()) && dataMirror == null) {
jaroslav@70	1552	error("PUT method needs to specify a data() class", e);
jaroslav@70	1553	return false;
jaroslav@70	1554	}
jaroslav@72	1555	if ("POST".equals(onR.method()) && dataMirror == null) {
jaroslav@70	1556	error("POST method needs to specify a data() class", e);
jaroslav@70	1557	return false;
jaroslav@70	1558	}
jtulach@0	1559	String modelClass = null;
jtulach@0	1560	boolean expectsList = false;
jaroslav@100	1561	List<String> args = new ArrayList<String>();
jtulach@0	1562	{
jtulach@0	1563	for (VariableElement ve : e.getParameters()) {
jtulach@0	1564	TypeMirror modelType = null;
jaroslav@29	1565	final TypeMirror type = ve.asType();
jaroslav@29	1566	CharSequence simpleName;
jaroslav@29	1567	if (type.getKind() == TypeKind.DECLARED) {
jaroslav@29	1568	simpleName = ((DeclaredType)type).asElement().getSimpleName();
jaroslav@29	1569	} else {
jaroslav@29	1570	simpleName = type.toString();
jaroslav@29	1571	}
jaroslav@29	1572	if (simpleName.toString().equals(className)) {
jaroslav@386	1573	args.add("model");
jtulach@0	1574	} else if (isModel(ve.asType())) {
jtulach@0	1575	modelType = ve.asType();
jtulach@0	1576	} else if (ve.asType().getKind() == TypeKind.ARRAY) {
jtulach@0	1577	modelType = ((ArrayType)ve.asType()).getComponentType();
jtulach@0	1578	expectsList = true;
jaroslav@72	1579	} else if (ve.asType().toString().equals("java.lang.String")) {
jaroslav@72	1580	modelType = ve.asType();
jtulach@0	1581	}
jtulach@0	1582	if (modelType != null) {
jtulach@0	1583	if (modelClass != null) {
jtulach@0	1584	error("There can be only one model class among arguments", e);
jtulach@0	1585	} else {
jtulach@0	1586	modelClass = modelType.toString();
jtulach@0	1587	if (expectsList) {
jtulach@0	1588	args.add("arr");
jtulach@0	1589	} else {
jtulach@0	1590	args.add("arr[0]");
jtulach@0	1591	}
jtulach@0	1592	}
jtulach@0	1593	}
jtulach@0	1594	}
jtulach@0	1595	}
jtulach@0	1596	if (modelClass == null) {
jtulach@0	1597	error("The method needs to have one @Model class as parameter", e);
jtulach@0	1598	}
jtulach@0	1599	String n = e.getSimpleName().toString();
jaroslav@250	1600	if ("WebSocket".equals(onR.method())) {
jaroslav@250	1601	body.append(" /** Performs WebSocket communication. Call with <code>null</code> data parameter\n");
jaroslav@250	1602	body.append(" * to open the connection (even if not required). Call with non-null data to\n");
jaroslav@250	1603	body.append(" * send messages to server. Call again with <code>null</code> data to close the socket.\n");
jaroslav@250	1604	body.append(" */\n");
jaroslav@250	1605	}
jaroslav@66	1606	body.append(" public void ").append(n).append("(");
jaroslav@24	1607	StringBuilder urlBefore = new StringBuilder();
jaroslav@24	1608	StringBuilder urlAfter = new StringBuilder();
jtulach@0	1609	String jsonpVarName = null;
jtulach@0	1610	{
jtulach@0	1611	String sep = "";
jtulach@0	1612	boolean skipJSONP = onR.jsonp().isEmpty();
jaroslav@24	1613	for (String p : findParamNames(e, onR.url(), onR.jsonp(), urlBefore, urlAfter)) {
jtulach@0	1614	if (!skipJSONP && p.equals(onR.jsonp())) {
jtulach@0	1615	skipJSONP = true;
jtulach@0	1616	jsonpVarName = p;
jtulach@0	1617	continue;
jtulach@0	1618	}
jtulach@0	1619	body.append(sep);
jtulach@0	1620	body.append("String ").append(p);
jtulach@0	1621	sep = ", ";
jtulach@0	1622	}
jtulach@0	1623	if (!skipJSONP) {
jtulach@0	1624	error(
jtulach@0	1625	"Name of jsonp attribute ('" + onR.jsonp() +
jtulach@0	1626	"') is not used in url attribute '" + onR.url() + "'", e
jtulach@0	1627	);
jtulach@0	1628	}
jaroslav@73	1629	if (dataMirror != null) {
jaroslav@73	1630	body.append(sep).append(dataMirror.toString()).append(" data");
jaroslav@73	1631	}
jtulach@0	1632	}
jtulach@0	1633	body.append(") {\n");
jaroslav@258	1634	boolean webSocket = onR.method().equals("WebSocket");
jaroslav@258	1635	if (webSocket) {
jaroslav@387	1636	if (generateWSReceiveBody(index++, body, inType, onR, e, clazz, className, expectsList, modelClass, n, args, urlBefore, jsonpVarName, urlAfter, dataMirror)) {
jaroslav@387	1637	return false;
jaroslav@387	1638	}
jaroslav@258	1639	body.append(" }\n");
jaroslav@387	1640	body.append(" private Object ws_" + e.getSimpleName() + ";\n");
jaroslav@258	1641	} else {
jaroslav@386	1642	if (generateJSONReceiveBody(index++, body, inType, onR, e, clazz, className, expectsList, modelClass, n, args, urlBefore, jsonpVarName, urlAfter, dataMirror)) {
jaroslav@258	1643	return false;
jaroslav@258	1644	}
jaroslav@258	1645	body.append(" }\n");
jaroslav@258	1646	}
jaroslav@258	1647	}
jaroslav@386	1648	inType.append(" }\n");
jaroslav@386	1649	inType.append(" throw new UnsupportedOperationException(\"index: \" + index + \" type: \" + type);\n");
jaroslav@386	1650	inType.append(" }\n");
jaroslav@258	1651	return true;
jaroslav@258	1652	}
jaroslav@258	1653
jaroslav@386	1654	private boolean generateJSONReceiveBody(int index, StringWriter method, StringBuilder 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	1655	body.append(
jaroslav@386	1656	" case " + index + ": {\n" +
jaroslav@386	1657	" if (type == 2) { /* on error */\n" +
jaroslav@386	1658	" Exception ex = (Exception)data;\n"
jaroslav@258	1659	);
jaroslav@258	1660	if (onR.onError().isEmpty()) {
jtulach@0	1661	body.append(
jaroslav@386	1662	" ex.printStackTrace();\n"
jaroslav@245	1663	);
jaroslav@258	1664	} else {
jaroslav@258	1665	if (!findOnError(e, ((TypeElement)clazz), onR.onError(), className)) {
jaroslav@258	1666	return true;
jaroslav@245	1667	}
jaroslav@258	1668	body.append(" ").append(clazz.getSimpleName()).append(".").append(onR.onError()).append("(");
jaroslav@386	1669	body.append("model, ex);\n");
jaroslav@258	1670	}
jaroslav@258	1671	body.append(
jaroslav@386	1672	" return;\n" +
jaroslav@386	1673	" } else if (type == 1) {\n" +
jaroslav@386	1674	" Object[] ev = (Object[])data;\n"
jaroslav@258	1675	);
jaroslav@258	1676	if (expectsList) {
jaroslav@245	1677	body.append(
jaroslav@386	1678	" " + modelClass + "[] arr = new " + modelClass + "[ev.length];\n"
jaroslav@258	1679	);
jaroslav@258	1680	} else {
jaroslav@258	1681	body.append(
jaroslav@258	1682	" " + modelClass + "[] arr = { null };\n"
jaroslav@258	1683	);
jaroslav@258	1684	}
jaroslav@258	1685	body.append(
jaroslav@386	1686	" TYPE.copyJSON(model.proto.getContext(), ev, " + modelClass + ".class, arr);\n"
jaroslav@386	1687	);
jaroslav@258	1688	{
jaroslav@258	1689	body.append(" ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@258	1690	String sep = "";
jaroslav@258	1691	for (String arg : args) {
jaroslav@258	1692	body.append(sep);
jaroslav@258	1693	body.append(arg);
jaroslav@258	1694	sep = ", ";
jaroslav@74	1695	}
jaroslav@24	1696	body.append(");\n");
jtulach@0	1697	}
jaroslav@258	1698	body.append(
jaroslav@386	1699	" return;\n" +
jaroslav@258	1700	" }\n" +
jaroslav@258	1701	" }\n"
jaroslav@258	1702	);
jaroslav@386	1703	method.append(" proto.loadJSON(" + index + ",\n ");
jaroslav@386	1704	method.append(urlBefore).append(", ");
jaroslav@258	1705	if (jsonpVarName != null) {
jaroslav@386	1706	method.append(urlAfter);
jaroslav@258	1707	} else {
jaroslav@386	1708	method.append("null");
jaroslav@258	1709	}
jaroslav@258	1710	if (!"GET".equals(onR.method()) \|\| dataMirror != null) {
jaroslav@386	1711	method.append(", \"").append(onR.method()).append('"');
jaroslav@258	1712	if (dataMirror != null) {
jaroslav@386	1713	method.append(", data");
jaroslav@258	1714	} else {
jaroslav@386	1715	method.append(", null");
jaroslav@258	1716	}
jaroslav@258	1717	} else {
jaroslav@386	1718	method.append(", null, null");
jaroslav@258	1719	}
jaroslav@386	1720	method.append(");\n");
jaroslav@258	1721	return false;
jaroslav@258	1722	}
jaroslav@258	1723
jaroslav@387	1724	private boolean generateWSReceiveBody(int index, StringWriter method, StringBuilder 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	1725	body.append(
jaroslav@387	1726	" case " + index + ": {\n" +
jaroslav@387	1727	" if (type == 0) { /* on open */\n" +
jaroslav@387	1728	" ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@258	1729	{
jaroslav@258	1730	String sep = "";
jaroslav@258	1731	for (String arg : args) {
jaroslav@258	1732	body.append(sep);
jaroslav@258	1733	if (arg.startsWith("arr")) {
jaroslav@258	1734	body.append("null");
jaroslav@258	1735	} else {
jaroslav@258	1736	body.append(arg);
jaroslav@258	1737	}
jaroslav@258	1738	sep = ", ";
jaroslav@258	1739	}
jaroslav@258	1740	}
jaroslav@258	1741	body.append(");\n");
jaroslav@258	1742	body.append(
jaroslav@387	1743	" return;\n" +
jaroslav@387	1744	" } else if (type == 2) { /* on error */\n" +
jaroslav@387	1745	" Exception value = (Exception)data;\n"
jaroslav@258	1746	);
jaroslav@258	1747	if (onR.onError().isEmpty()) {
jaroslav@258	1748	body.append(
jaroslav@258	1749	" value.printStackTrace();\n"
jaroslav@258	1750	);
jaroslav@258	1751	} else {
jaroslav@258	1752	if (!findOnError(e, ((TypeElement)clazz), onR.onError(), className)) {
jaroslav@258	1753	return true;
jaroslav@258	1754	}
jaroslav@258	1755	body.append(" ").append(clazz.getSimpleName()).append(".").append(onR.onError()).append("(");
jaroslav@387	1756	body.append("model, value);\n");
jaroslav@258	1757	}
jaroslav@258	1758	body.append(
jaroslav@387	1759	" return;\n" +
jaroslav@387	1760	" } else if (type == 1) {\n" +
jaroslav@387	1761	" Object[] ev = (Object[])data;\n"
jaroslav@258	1762	);
jaroslav@258	1763	if (expectsList) {
jaroslav@258	1764	body.append(
jaroslav@387	1765	" " + modelClass + "[] arr = new " + modelClass + "[ev.length];\n"
jaroslav@258	1766	);
jaroslav@258	1767	} else {
jaroslav@258	1768	body.append(
jaroslav@258	1769	" " + modelClass + "[] arr = { null };\n"
jaroslav@258	1770	);
jaroslav@258	1771	}
jaroslav@258	1772	body.append(
jaroslav@387	1773	" TYPE.copyJSON(model.proto.getContext(), ev, " + modelClass + ".class, arr);\n"
jaroslav@387	1774	);
jaroslav@258	1775	{
jaroslav@258	1776	body.append(" ").append(clazz.getSimpleName()).append(".").append(n).append("(");
jaroslav@258	1777	String sep = "";
jaroslav@258	1778	for (String arg : args) {
jaroslav@258	1779	body.append(sep);
jaroslav@258	1780	body.append(arg);
jaroslav@258	1781	sep = ", ";
jaroslav@258	1782	}
jaroslav@258	1783	body.append(");\n");
jaroslav@258	1784	}
jaroslav@258	1785	body.append(
jaroslav@387	1786	" return;\n" +
jaroslav@387	1787	" }"
jaroslav@258	1788	);
jaroslav@258	1789	if (!onR.onError().isEmpty()) {
jaroslav@387	1790	body.append(" else if (type == 3) { /* on close */\n");
jaroslav@387	1791	body.append(" ").append(clazz.getSimpleName()).append(".").append(onR.onError()).append("(");
jaroslav@387	1792	body.append("model, null);\n");
jaroslav@258	1793	body.append(
jaroslav@397	1794	" return;" +
jaroslav@387	1795	" }"
jaroslav@258	1796	);
jaroslav@258	1797	}
jaroslav@387	1798	body.append("\n");
jaroslav@258	1799	body.append(" }\n");
jaroslav@387	1800	method.append(" if (this.ws_").append(e.getSimpleName()).append(" == null) {\n");
jaroslav@387	1801	method.append(" this.ws_").append(e.getSimpleName());
jaroslav@387	1802	method.append("= proto.wsOpen(" + index + ", ");
jaroslav@387	1803	method.append(urlBefore).append(", data);\n");
jaroslav@387	1804	method.append(" } else {\n");
jaroslav@387	1805	method.append(" proto.wsSend(this.ws_").append(e.getSimpleName()).append(", ").append(urlBefore).append(", data);\n");
jaroslav@387	1806	method.append(" }\n");
jaroslav@258	1807	return false;
jtulach@0	1808	}
jtulach@0	1809
jtulach@0	1810	private CharSequence wrapParams(
jtulach@0	1811	ExecutableElement ee, String id, String className, String evName, String dataName
jtulach@0	1812	) {
jtulach@0	1813	TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jtulach@0	1814	StringBuilder params = new StringBuilder();
jtulach@0	1815	boolean first = true;
jtulach@0	1816	for (VariableElement ve : ee.getParameters()) {
jtulach@0	1817	if (!first) {
jtulach@0	1818	params.append(", ");
jtulach@0	1819	}
jtulach@0	1820	first = false;
jtulach@0	1821	String toCall = null;
jaroslav@22	1822	String toFinish = null;
jaroslav@408	1823	boolean addNull = true;
jtulach@0	1824	if (ve.asType() == stringType) {
jtulach@0	1825	if (ve.getSimpleName().contentEquals("id")) {
jtulach@0	1826	params.append('"').append(id).append('"');
jtulach@0	1827	continue;
jtulach@0	1828	}
jaroslav@380	1829	toCall = "proto.toString(";
jtulach@0	1830	}
jtulach@0	1831	if (ve.asType().getKind() == TypeKind.DOUBLE) {
jaroslav@380	1832	toCall = "proto.toNumber(";
jaroslav@22	1833	toFinish = ".doubleValue()";
jtulach@0	1834	}
jtulach@0	1835	if (ve.asType().getKind() == TypeKind.INT) {
jaroslav@380	1836	toCall = "proto.toNumber(";
jaroslav@22	1837	toFinish = ".intValue()";
jtulach@0	1838	}
jtulach@0	1839	if (dataName != null && ve.getSimpleName().contentEquals(dataName) && isModel(ve.asType())) {
jaroslav@380	1840	toCall = "proto.toModel(" + ve.asType() + ".class, ";
jaroslav@408	1841	addNull = false;
jtulach@0	1842	}
jtulach@0	1843
jtulach@0	1844	if (toCall != null) {
jtulach@0	1845	params.append(toCall);
jtulach@0	1846	if (dataName != null && ve.getSimpleName().contentEquals(dataName)) {
jtulach@0	1847	params.append(dataName);
jaroslav@408	1848	if (addNull) {
jaroslav@408	1849	params.append(", null");
jaroslav@408	1850	}
jtulach@0	1851	} else {
jtulach@0	1852	if (evName == null) {
jtulach@0	1853	final StringBuilder sb = new StringBuilder();
jtulach@0	1854	sb.append("Unexpected string parameter name.");
jtulach@0	1855	if (dataName != null) {
jtulach@0	1856	sb.append(" Try \"").append(dataName).append("\"");
jtulach@0	1857	}
jtulach@0	1858	error(sb.toString(), ee);
jtulach@0	1859	}
jtulach@0	1860	params.append(evName);
jtulach@0	1861	params.append(", \"");
jtulach@0	1862	params.append(ve.getSimpleName().toString());
jtulach@0	1863	params.append("\"");
jtulach@0	1864	}
jtulach@0	1865	params.append(")");
jaroslav@22	1866	if (toFinish != null) {
jaroslav@22	1867	params.append(toFinish);
jaroslav@22	1868	}
jtulach@0	1869	continue;
jtulach@0	1870	}
jtulach@0	1871	String rn = fqn(ve.asType(), ee);
jtulach@0	1872	int last = rn.lastIndexOf('.');
jtulach@0	1873	if (last >= 0) {
jtulach@0	1874	rn = rn.substring(last + 1);
jtulach@0	1875	}
jtulach@0	1876	if (rn.equals(className)) {
jtulach@0	1877	params.append(className).append(".this");
jtulach@0	1878	continue;
jtulach@0	1879	}
jtulach@0	1880	error(
jaroslav@235	1881	"The annotated method can only accept " + className + " argument or argument named 'data'",
jtulach@0	1882	ee
jtulach@0	1883	);
jtulach@0	1884	}
jtulach@0	1885	return params;
jtulach@0	1886	}
jtulach@0	1887
jtulach@0	1888
jtulach@0	1889	private CharSequence wrapPropName(
jtulach@0	1890	ExecutableElement ee, String className, String propName, String propValue
jtulach@0	1891	) {
jtulach@0	1892	TypeMirror stringType = processingEnv.getElementUtils().getTypeElement("java.lang.String").asType();
jtulach@0	1893	StringBuilder params = new StringBuilder();
jtulach@0	1894	boolean first = true;
jtulach@0	1895	for (VariableElement ve : ee.getParameters()) {
jtulach@0	1896	if (!first) {
jtulach@0	1897	params.append(", ");
jtulach@0	1898	}
jtulach@0	1899	first = false;
jtulach@0	1900	if (ve.asType() == stringType) {
jtulach@0	1901	if (propName != null && ve.getSimpleName().contentEquals(propName)) {
jtulach@0	1902	params.append('"').append(propValue).append('"');
jtulach@0	1903	} else {
jtulach@0	1904	error("Unexpected string parameter name. Try \"" + propName + "\".", ee);
jtulach@0	1905	}
jtulach@0	1906	continue;
jtulach@0	1907	}
jtulach@0	1908	String rn = fqn(ve.asType(), ee);
jtulach@0	1909	int last = rn.lastIndexOf('.');
jtulach@0	1910	if (last >= 0) {
jtulach@0	1911	rn = rn.substring(last + 1);
jtulach@0	1912	}
jtulach@0	1913	if (rn.equals(className)) {
jaroslav@383	1914	params.append("model");
jtulach@0	1915	continue;
jtulach@0	1916	}
jtulach@0	1917	error(
jtulach@0	1918	"@OnPrprtChange method can only accept String or " + className + " arguments",
jtulach@0	1919	ee);
jtulach@0	1920	}
jtulach@0	1921	return params;
jtulach@0	1922	}
jtulach@0	1923
jtulach@0	1924	private boolean isModel(TypeMirror tm) {
jaroslav@13	1925	if (tm.getKind() == TypeKind.ERROR) {
jaroslav@13	1926	return true;
jaroslav@13	1927	}
jtulach@0	1928	final Element e = processingEnv.getTypeUtils().asElement(tm);
jtulach@0	1929	if (e == null) {
jtulach@0	1930	return false;
jtulach@0	1931	}
jtulach@0	1932	for (Element ch : e.getEnclosedElements()) {
jtulach@0	1933	if (ch.getKind() == ElementKind.METHOD) {
jtulach@0	1934	ExecutableElement ee = (ExecutableElement)ch;
jtulach@0	1935	if (ee.getParameters().isEmpty() && ee.getSimpleName().contentEquals("modelFor")) {
jtulach@0	1936	return true;
jtulach@0	1937	}
jtulach@0	1938	}
jtulach@0	1939	}
jtulach@0	1940	return models.values().contains(e.getSimpleName().toString());
jtulach@0	1941	}
jtulach@0	1942
jaroslav@495	1943	private void writeEnumToString(Element e, Prprt[] props, Writer w) throws IOException {
jaroslav@495	1944	w.write(" public String toString() {\n");
jaroslav@495	1945	w.write(" StringBuilder sb = new StringBuilder();\n");
jaroslav@495	1946	w.write(" sb.append(union);\n");
jaroslav@495	1947	w.write(" if (sb.length() > 1) {\n");
jaroslav@495	1948	w.write(" sb.setCharAt(sb.length() - 1, ',');\n");
jaroslav@495	1949	w.write(" } else {\n");
jaroslav@495	1950	w.write(" sb.setLength(1);\n");
jaroslav@495	1951	w.write(" }\n");
jaroslav@495	1952	w.append(" sb.append(\"\\\"").append(e.getSimpleName()).append("\\\":\\\"\").append(get");
jaroslav@495	1953	w.append(e.getSimpleName()).append("()).append(\"\\\"\");\n");
jaroslav@495	1954	for (Prprt p : props) {
jaroslav@495	1955	w.write(" sb.append(',');\n");
jaroslav@495	1956	w.append(" sb.append('\"').append(\"" + p.name() + "\")");
jaroslav@495	1957	w.append(".append('\"').append(\":\");\n");
jaroslav@495	1958	w.append(" sb.append(TYPE.toJSON(prop_");
jaroslav@495	1959	w.append(p.name()).append("));\n");
jaroslav@495	1960	}
jaroslav@495	1961	w.write(" sb.append('}');\n");
jaroslav@495	1962	w.write(" return sb.toString();\n");
jaroslav@495	1963	w.write(" }\n");
jaroslav@495	1964	}
jtulach@0	1965	private void writeToString(Prprt[] props, Writer w) throws IOException {
jtulach@0	1966	w.write(" public String toString() {\n");
jtulach@0	1967	w.write(" StringBuilder sb = new StringBuilder();\n");
jtulach@0	1968	w.write(" sb.append('{');\n");
jtulach@0	1969	String sep = "";
jtulach@0	1970	for (Prprt p : props) {
jtulach@0	1971	w.write(sep);
jaroslav@18	1972	w.append(" sb.append('\"').append(\"" + p.name() + "\")");
jaroslav@18	1973	w.append(".append('\"').append(\":\");\n");
jaroslav@380	1974	w.append(" sb.append(TYPE.toJSON(prop_");
jtulach@0	1975	w.append(p.name()).append("));\n");
jtulach@0	1976	sep = " sb.append(',');\n";
jtulach@0	1977	}
jtulach@0	1978	w.write(" sb.append('}');\n");
jtulach@0	1979	w.write(" return sb.toString();\n");
jtulach@0	1980	w.write(" }\n");
jtulach@0	1981	}
jtulach@0	1982	private void writeClone(String className, Prprt[] props, Writer w) throws IOException {
jtulach@0	1983	w.write(" public " + className + " clone() {\n");
jaroslav@373	1984	w.write(" return clone(proto.getContext());\n");
jaroslav@111	1985	w.write(" }\n");
jaroslav@111	1986	w.write(" private " + className + " clone(net.java.html.BrwsrCtx ctx) {\n");
jaroslav@491	1987	w.write(" " + className + " ret = new " + className + "(ctx, null);\n");
jtulach@0	1988	for (Prprt p : props) {
jtulach@0	1989	if (!p.array()) {
jtulach@0	1990	boolean isModel[] = { false };
jtulach@0	1991	boolean isEnum[] = { false };
jtulach@0	1992	boolean isPrimitive[] = { false };
jtulach@0	1993	checkType(p, isModel, isEnum, isPrimitive);
jtulach@0	1994	if (!isModel[0]) {
jtulach@0	1995	w.write(" ret.prop_" + p.name() + " = prop_" + p.name() + ";\n");
jtulach@0	1996	continue;
jtulach@0	1997	}
jaroslav@114	1998	w.write(" ret.prop_" + p.name() + " = prop_" + p.name() + " == null ? null : prop_" + p.name() + ".clone();\n");
jtulach@0	1999	} else {
jaroslav@383	2000	w.write(" proto.cloneList(ret.prop_" + p.name() + ", ctx, prop_" + p.name() + ");\n");
jtulach@0	2001	}
jtulach@0	2002	}
jtulach@0	2003
jtulach@0	2004	w.write(" return ret;\n");
jtulach@0	2005	w.write(" }\n");
jtulach@0	2006	}
jaroslav@493	2007	private void writeEnumClone(Element e, String className, Prprt[] props, Writer w) throws IOException {
jaroslav@493	2008	w.write(" public " + className + " clone() {\n");
jaroslav@493	2009	w.write(" return clone(proto.getContext());\n");
jaroslav@493	2010	w.write(" }\n");
jaroslav@493	2011	w.write(" private " + className + " clone(net.java.html.BrwsrCtx ctx) {\n");
jaroslav@493	2012	w.write(" Object newUnion = null;\n");
jaroslav@506	2013	w.write(" org.apidesign.html.json.spi.Proto newProto = null;\n");
jaroslav@493	2014	for (Element ec : e.getEnclosedElements()) {
jaroslav@493	2015	if (ec.getKind() != ElementKind.ENUM_CONSTANT) {
jaroslav@493	2016	continue;
jaroslav@493	2017	}
jaroslav@496	2018	Model em = ec.getAnnotation(Model.class);
jaroslav@496	2019	if (em == null) {
jaroslav@496	2020	continue;
jaroslav@496	2021	}
jaroslav@506	2022	w.write(" if (union instanceof " + em.className() + ") {\n");
jaroslav@506	2023	w.write(" newUnion = ((" + em.className() + ")union).clone(ctx);\n");
jaroslav@506	2024	w.write(" newProto = ((" + em.className() + ")newUnion).proto;\n");
jaroslav@506	2025	w.write(" }\n");
jaroslav@493	2026	}
jaroslav@493	2027	w.write(" " + className + " ret = new " + className + "(ctx, newUnion);\n");
jaroslav@506	2028	w.write(" newProto.onValueHasMutated(ret.proto);\n");
jaroslav@493	2029	for (Prprt p : props) {
jaroslav@493	2030	if (!p.array()) {
jaroslav@493	2031	boolean isModel[] = { false };
jaroslav@493	2032	boolean isEnum[] = { false };
jaroslav@493	2033	boolean isPrimitive[] = { false };
jaroslav@493	2034	checkType(p, isModel, isEnum, isPrimitive);
jaroslav@493	2035	if (!isModel[0]) {
jaroslav@493	2036	w.write(" ret.prop_" + p.name() + " = prop_" + p.name() + ";\n");
jaroslav@493	2037	continue;
jaroslav@493	2038	}
jaroslav@493	2039	w.write(" ret.prop_" + p.name() + " = prop_" + p.name() + " == null ? null : prop_" + p.name() + ".clone();\n");
jaroslav@493	2040	} else {
jaroslav@493	2041	w.write(" proto.cloneList(ret.prop_" + p.name() + ", ctx, prop_" + p.name() + ");\n");
jaroslav@493	2042	}
jaroslav@493	2043	}
jaroslav@493	2044
jaroslav@493	2045	w.write(" return ret;\n");
jaroslav@493	2046	w.write(" }\n");
jaroslav@493	2047	}
jtulach@0	2048
jtulach@0	2049	private String inPckName(Element e) {
jaroslav@490	2050	while (!(e instanceof TypeElement)) {
jaroslav@490	2051	e = e.getEnclosingElement();
jaroslav@490	2052	}
jaroslav@490	2053
jtulach@0	2054	StringBuilder sb = new StringBuilder();
jtulach@0	2055	while (e.getKind() != ElementKind.PACKAGE) {
jtulach@0	2056	if (sb.length() == 0) {
jtulach@0	2057	sb.append(e.getSimpleName());
jtulach@0	2058	} else {
jtulach@0	2059	sb.insert(0, '.');
jtulach@0	2060	sb.insert(0, e.getSimpleName());
jtulach@0	2061	}
jtulach@0	2062	e = e.getEnclosingElement();
jtulach@0	2063	}
jtulach@0	2064	return sb.toString();
jtulach@0	2065	}
jtulach@0	2066
jtulach@0	2067	private String fqn(TypeMirror pt, Element relative) {
jtulach@0	2068	if (pt.getKind() == TypeKind.ERROR) {
jtulach@0	2069	final Elements eu = processingEnv.getElementUtils();
jtulach@0	2070	PackageElement pckg = eu.getPackageOf(relative);
jtulach@0	2071	return pckg.getQualifiedName() + "." + pt.toString();
jtulach@0	2072	}
jtulach@0	2073	return pt.toString();
jtulach@0	2074	}
jtulach@0	2075
jtulach@0	2076	private String checkType(Prprt p, boolean[] isModel, boolean[] isEnum, boolean[] isPrimitive) {
jtulach@0	2077	TypeMirror tm;
jtulach@0	2078	try {
jtulach@0	2079	String ret = p.typeName(processingEnv);
jtulach@0	2080	TypeElement e = processingEnv.getElementUtils().getTypeElement(ret);
jtulach@0	2081	if (e == null) {
jtulach@0	2082	isModel[0] = true;
jtulach@0	2083	isEnum[0] = false;
jtulach@0	2084	isPrimitive[0] = false;
jtulach@0	2085	return ret;
jtulach@0	2086	}
jtulach@0	2087	tm = e.asType();
jtulach@0	2088	} catch (MirroredTypeException ex) {
jtulach@0	2089	tm = ex.getTypeMirror();
jtulach@0	2090	}
jtulach@0	2091	tm = processingEnv.getTypeUtils().erasure(tm);
jaroslav@241	2092	if (isPrimitive[0] = tm.getKind().isPrimitive()) {
jaroslav@241	2093	isEnum[0] = false;
jaroslav@241	2094	isModel[0] = false;
jaroslav@241	2095	return tm.toString();
jaroslav@241	2096	}
jtulach@0	2097	final Element e = processingEnv.getTypeUtils().asElement(tm);
jaroslav@494	2098	if (isEnum[0] = e.getKind() == ElementKind.ENUM) {
jaroslav@494	2099	return ((TypeElement)e).getQualifiedName().toString();
jaroslav@494	2100	}
jaroslav@241	2101	if (e.getKind() == ElementKind.CLASS && tm.getKind() == TypeKind.ERROR) {
jaroslav@241	2102	isModel[0] = true;
jaroslav@241	2103	isEnum[0] = false;
jaroslav@242	2104	return e.getSimpleName().toString();
jaroslav@241	2105	}
jaroslav@241	2106
jtulach@0	2107	final Model m = e == null ? null : e.getAnnotation(Model.class);
jtulach@0	2108	String ret;
jtulach@0	2109	if (m != null) {
jtulach@0	2110	ret = findPkgName(e) + '.' + m.className();
jtulach@0	2111	isModel[0] = true;
jtulach@0	2112	models.put(e, m.className());
jtulach@0	2113	} else if (findModelForMthd(e)) {
jtulach@0	2114	ret = ((TypeElement)e).getQualifiedName().toString();
jtulach@0	2115	isModel[0] = true;
jtulach@0	2116	} else {
jtulach@0	2117	ret = tm.toString();
jtulach@0	2118	}
jtulach@0	2119	return ret;
jtulach@0	2120	}
jtulach@0	2121
jtulach@0	2122	private static boolean findModelForMthd(Element clazz) {
jtulach@0	2123	if (clazz == null) {
jtulach@0	2124	return false;
jtulach@0	2125	}
jtulach@0	2126	for (Element e : clazz.getEnclosedElements()) {
jtulach@0	2127	if (e.getKind() == ElementKind.METHOD) {
jtulach@0	2128	ExecutableElement ee = (ExecutableElement)e;
jtulach@0	2129	if (
jtulach@0	2130	ee.getSimpleName().contentEquals("modelFor") &&
jtulach@0	2131	ee.getParameters().isEmpty()
jtulach@0	2132	) {
jtulach@0	2133	return true;
jtulach@0	2134	}
jtulach@0	2135	}
jtulach@0	2136	}
jtulach@0	2137	return false;
jtulach@0	2138	}
jtulach@0	2139
jaroslav@24	2140	private Iterable<String> findParamNames(
jaroslav@24	2141	Element e, String url, String jsonParam, StringBuilder... both
jaroslav@24	2142	) {
jaroslav@100	2143	List<String> params = new ArrayList<String>();
jaroslav@24	2144	int wasJSON = 0;
jtulach@0	2145
jtulach@0	2146	for (int pos = 0; ;) {
jtulach@0	2147	int next = url.indexOf('{', pos);
jtulach@0	2148	if (next == -1) {
jaroslav@24	2149	both[wasJSON].append('"')
jtulach@0	2150	.append(url.substring(pos))
jtulach@0	2151	.append('"');
jtulach@0	2152	return params;
jtulach@0	2153	}
jtulach@0	2154	int close = url.indexOf('}', next);
jtulach@0	2155	if (close == -1) {
jtulach@0	2156	error("Unbalanced '{' and '}' in " + url, e);
jtulach@0	2157	return params;
jtulach@0	2158	}
jtulach@0	2159	final String paramName = url.substring(next + 1, close);
jtulach@0	2160	params.add(paramName);
jaroslav@24	2161	if (paramName.equals(jsonParam) && !jsonParam.isEmpty()) {
jaroslav@24	2162	both[wasJSON].append('"')
jaroslav@24	2163	.append(url.substring(pos, next))
jaroslav@24	2164	.append('"');
jaroslav@24	2165	wasJSON = 1;
jaroslav@24	2166	} else {
jaroslav@24	2167	both[wasJSON].append('"')
jaroslav@24	2168	.append(url.substring(pos, next))
jaroslav@24	2169	.append("\" + ").append(paramName).append(" + ");
jaroslav@24	2170	}
jtulach@0	2171	pos = close + 1;
jtulach@0	2172	}
jtulach@0	2173	}
jtulach@0	2174
jtulach@0	2175	private static Prprt findPrprt(Prprt[] properties, String propName) {
jtulach@0	2176	for (Prprt p : properties) {
jtulach@0	2177	if (propName.equals(p.name())) {
jtulach@0	2178	return p;
jtulach@0	2179	}
jtulach@0	2180	}
jtulach@0	2181	return null;
jtulach@0	2182	}
jtulach@0	2183
jtulach@0	2184	private boolean isPrimitive(String type) {
jtulach@0	2185	return
jtulach@0	2186	"int".equals(type) \|\|
jtulach@0	2187	"double".equals(type) \|\|
jtulach@0	2188	"long".equals(type) \|\|
jtulach@0	2189	"short".equals(type) \|\|
jtulach@0	2190	"byte".equals(type) \|\|
jaroslav@100	2191	"char".equals(type) \|\|
jaroslav@100	2192	"boolean".equals(type) \|\|
jtulach@0	2193	"float".equals(type);
jtulach@0	2194	}
jtulach@0	2195
jtulach@0	2196	private static Collection<String> findDerivedFrom(Map<String, Collection<String>> propsDeps, String derivedProp) {
jaroslav@100	2197	Set<String> names = new HashSet<String>();
jtulach@0	2198	for (Map.Entry<String, Collection<String>> e : propsDeps.entrySet()) {
jtulach@0	2199	if (e.getValue().contains(derivedProp)) {
jtulach@0	2200	names.add(e.getKey());
jtulach@0	2201	}
jtulach@0	2202	}
jtulach@0	2203	return names;
jtulach@0	2204	}
jtulach@0	2205
jtulach@0	2206	private Prprt[] createProps(Element e, Property[] arr) {
jtulach@0	2207	Prprt[] ret = Prprt.wrap(processingEnv, e, arr);
jtulach@0	2208	Prprt[] prev = verify.put(e, ret);
jtulach@0	2209	if (prev != null) {
jtulach@0	2210	error("Two sets of properties for ", e);
jtulach@0	2211	}
jtulach@0	2212	return ret;
jtulach@0	2213	}
jtulach@0	2214
jaroslav@17	2215	private static String strip(String s) {
jaroslav@17	2216	int indx = s.indexOf("__");
jaroslav@17	2217	if (indx >= 0) {
jaroslav@17	2218	return s.substring(0, indx);
jaroslav@17	2219	} else {
jaroslav@17	2220	return s;
jaroslav@17	2221	}
jaroslav@17	2222	}
jaroslav@70	2223
jaroslav@73	2224	private String findDataSpecified(ExecutableElement e, OnReceive onR) {
jaroslav@73	2225	try {
jaroslav@73	2226	return onR.data().getName();
jaroslav@73	2227	} catch (MirroredTypeException ex) {
jaroslav@242	2228	final TypeMirror tm = ex.getTypeMirror();
jaroslav@242	2229	String name;
jaroslav@242	2230	final Element te = processingEnv.getTypeUtils().asElement(tm);
jaroslav@242	2231	if (te.getKind() == ElementKind.CLASS && tm.getKind() == TypeKind.ERROR) {
jaroslav@242	2232	name = te.getSimpleName().toString();
jaroslav@242	2233	} else {
jaroslav@242	2234	name = tm.toString();
jaroslav@242	2235	}
jaroslav@73	2236	return "java.lang.Object".equals(name) ? null : name;
jaroslav@73	2237	} catch (Exception ex) {
jaroslav@73	2238	// fallback
jaroslav@73	2239	}
jaroslav@73	2240
jaroslav@72	2241	AnnotationMirror found = null;
jaroslav@72	2242	for (AnnotationMirror am : e.getAnnotationMirrors()) {
jaroslav@72	2243	if (am.getAnnotationType().toString().equals(OnReceive.class.getName())) {
jaroslav@72	2244	found = am;
jaroslav@72	2245	}
jaroslav@72	2246	}
jaroslav@72	2247	if (found == null) {
jaroslav@72	2248	return null;
jaroslav@72	2249	}
jaroslav@72	2250
jaroslav@72	2251	for (Map.Entry<? extends ExecutableElement, ? extends AnnotationValue> entry : found.getElementValues().entrySet()) {
jaroslav@72	2252	ExecutableElement ee = entry.getKey();
jaroslav@72	2253	AnnotationValue av = entry.getValue();
jaroslav@72	2254	if (ee.getSimpleName().contentEquals("data")) {
jaroslav@73	2255	List<? extends Object> values = getAnnoValues(processingEnv, e, found);
jaroslav@73	2256	for (Object v : values) {
jaroslav@73	2257	String sv = v.toString();
jaroslav@73	2258	if (sv.startsWith("data = ") && sv.endsWith(".class")) {
jaroslav@73	2259	return sv.substring(7, sv.length() - 6);
jaroslav@73	2260	}
jaroslav@73	2261	}
jaroslav@73	2262	return "error";
jaroslav@72	2263	}
jaroslav@72	2264	}
jaroslav@72	2265	return null;
jaroslav@72	2266	}
jaroslav@72	2267
jaroslav@72	2268	static List<? extends Object> getAnnoValues(ProcessingEnvironment pe, Element e, AnnotationMirror am) {
jaroslav@70	2269	try {
jaroslav@72	2270	Class<?> trees = Class.forName("com.sun.tools.javac.api.JavacTrees");
jaroslav@72	2271	Method m = trees.getMethod("instance", ProcessingEnvironment.class);
jaroslav@72	2272	Object instance = m.invoke(null, pe);
jaroslav@72	2273	m = instance.getClass().getMethod("getPath", Element.class, AnnotationMirror.class);
jaroslav@72	2274	Object path = m.invoke(instance, e, am);
jaroslav@72	2275	m = path.getClass().getMethod("getLeaf");
jaroslav@72	2276	Object leaf = m.invoke(path);
jaroslav@72	2277	m = leaf.getClass().getMethod("getArguments");
jaroslav@72	2278	return (List) m.invoke(leaf);
jaroslav@72	2279	} catch (Exception ex) {
jaroslav@72	2280	return Collections.emptyList();
jaroslav@70	2281	}
jaroslav@70	2282	}
jaroslav@240	2283
jtulach@0	2284	private static class Prprt {
jtulach@0	2285	private final Element e;
jtulach@0	2286	private final AnnotationMirror tm;
jtulach@0	2287	private final Property p;
jtulach@0	2288
jtulach@0	2289	public Prprt(Element e, AnnotationMirror tm, Property p) {
jtulach@0	2290	this.e = e;
jtulach@0	2291	this.tm = tm;
jtulach@0	2292	this.p = p;
jtulach@0	2293	}
jtulach@0	2294
jtulach@0	2295	String name() {
jtulach@0	2296	return p.name();
jtulach@0	2297	}
jtulach@0	2298
jtulach@0	2299	boolean array() {
jtulach@0	2300	return p.array();
jtulach@0	2301	}
jtulach@0	2302
jtulach@0	2303	String typeName(ProcessingEnvironment env) {
jaroslav@100	2304	RuntimeException ex;
jtulach@0	2305	try {
jtulach@0	2306	return p.type().getName();
jaroslav@100	2307	} catch (IncompleteAnnotationException e) {
jaroslav@100	2308	ex = e;
jaroslav@100	2309	} catch (AnnotationTypeMismatchException e) {
jaroslav@100	2310	ex = e;
jaroslav@100	2311	}
jaroslav@100	2312	for (Object v : getAnnoValues(env, e, tm)) {
jaroslav@100	2313	String s = v.toString().replace(" ", "");
jaroslav@100	2314	if (s.startsWith("type=") && s.endsWith(".class")) {
jaroslav@100	2315	return s.substring(5, s.length() - 6);
jtulach@0	2316	}
jtulach@0	2317	}
jaroslav@100	2318	throw ex;
jtulach@0	2319	}
jtulach@0	2320
jtulach@0	2321
jtulach@0	2322	static Prprt[] wrap(ProcessingEnvironment pe, Element e, Property[] arr) {
jtulach@0	2323	if (arr.length == 0) {
jtulach@0	2324	return new Prprt[0];
jtulach@0	2325	}
jtulach@0	2326
jtulach@0	2327	List<? extends AnnotationValue> val = null;
jaroslav@490	2328	for (AnnotationMirror an : e.getAnnotationMirrors()) {
jtulach@0	2329	for (Map.Entry<? extends ExecutableElement, ? extends AnnotationValue> entry : an.getElementValues().entrySet()) {
jtulach@0	2330	if (entry.getKey().getSimpleName().contentEquals("properties")) {
jtulach@0	2331	val = (List)entry.getValue().getValue();
jtulach@0	2332	break;
jtulach@0	2333	}
jtulach@0	2334	}
jtulach@0	2335	}
jtulach@0	2336	if (val == null \|\| val.size() != arr.length) {
jtulach@0	2337	pe.getMessager().printMessage(Diagnostic.Kind.ERROR, "" + val, e);
jtulach@0	2338	return new Prprt[0];
jtulach@0	2339	}
jtulach@0	2340	Prprt[] ret = new Prprt[arr.length];
jtulach@0	2341	BIG: for (int i = 0; i < ret.length; i++) {
jtulach@0	2342	AnnotationMirror am = (AnnotationMirror)val.get(i).getValue();
jtulach@0	2343	ret[i] = new Prprt(e, am, arr[i]);
jtulach@0	2344
jtulach@0	2345	}
jtulach@0	2346	return ret;
jtulach@0	2347	}
jtulach@0	2348	}
jaroslav@70	2349
jaroslav@70	2350	@Override
jaroslav@70	2351	public Iterable<? extends Completion> getCompletions(Element element, AnnotationMirror annotation, ExecutableElement member, String userText) {
jaroslav@71	2352	final Level l = Level.FINE;
jaroslav@71	2353	LOG.log(l, " element: {0}", element);
jaroslav@71	2354	LOG.log(l, " annotation: {0}", annotation);
jaroslav@71	2355	LOG.log(l, " member: {0}", member);
jaroslav@71	2356	LOG.log(l, " userText: {0}", userText);
jaroslav@71	2357	LOG.log(l, "str: {0}", annotation.getAnnotationType().toString());
jaroslav@70	2358	if (annotation.getAnnotationType().toString().equals(OnReceive.class.getName())) {
jaroslav@70	2359	if (member.getSimpleName().contentEquals("method")) {
jaroslav@70	2360	return Arrays.asList(
jaroslav@70	2361	methodOf("GET"),
jaroslav@70	2362	methodOf("POST"),
jaroslav@70	2363	methodOf("PUT"),
jaroslav@70	2364	methodOf("DELETE"),
jaroslav@247	2365	methodOf("HEAD"),
jaroslav@247	2366	methodOf("WebSocket")
jaroslav@70	2367	);
jaroslav@70	2368	}
jaroslav@70	2369	}
jaroslav@70	2370
jaroslav@70	2371	return super.getCompletions(element, annotation, member, userText);
jaroslav@70	2372	}
jtulach@0	2373
jaroslav@70	2374	private static final Completion methodOf(String method) {
jaroslav@362	2375	ResourceBundle rb = ResourceBundle.getBundle("org.netbeans.html.json.impl.Bundle");
jaroslav@70	2376	return Completions.of('"' + method + '"', rb.getString("MSG_Completion_" + method));
jaroslav@70	2377	}
jaroslav@245	2378
jaroslav@245	2379	private boolean findOnError(ExecutableElement errElem, TypeElement te, String name, String className) {
jaroslav@245	2380	String err = null;
jaroslav@245	2381	METHODS:
jaroslav@245	2382	for (Element e : te.getEnclosedElements()) {
jaroslav@245	2383	if (e.getKind() != ElementKind.METHOD) {
jaroslav@245	2384	continue;
jaroslav@245	2385	}
jaroslav@245	2386	if (!e.getSimpleName().contentEquals(name)) {
jaroslav@245	2387	continue;
jaroslav@245	2388	}
jaroslav@245	2389	if (!e.getModifiers().contains(Modifier.STATIC)) {
jaroslav@245	2390	errElem = (ExecutableElement) e;
jaroslav@245	2391	err = "Would have to be static";
jaroslav@245	2392	continue;
jaroslav@245	2393	}
jaroslav@245	2394	ExecutableElement ee = (ExecutableElement) e;
jaroslav@245	2395	TypeMirror excType = processingEnv.getElementUtils().getTypeElement(Exception.class.getName()).asType();
jaroslav@245	2396	final List<? extends VariableElement> params = ee.getParameters();
jaroslav@245	2397	boolean error = false;
jaroslav@245	2398	if (params.size() != 2) {
jaroslav@245	2399	error = true;
jaroslav@245	2400	} else {
jaroslav@261	2401	String firstType = params.get(0).asType().toString();
jaroslav@261	2402	int lastDot = firstType.lastIndexOf('.');
jaroslav@261	2403	if (lastDot != -1) {
jaroslav@261	2404	firstType = firstType.substring(lastDot + 1);
jaroslav@261	2405	}
jaroslav@261	2406	if (!firstType.equals(className)) {
jaroslav@245	2407	error = true;
jaroslav@245	2408	}
jaroslav@245	2409	if (!processingEnv.getTypeUtils().isAssignable(excType, params.get(1).asType())) {
jaroslav@245	2410	error = true;
jaroslav@245	2411	}
jaroslav@245	2412	}
jaroslav@245	2413	if (error) {
jaroslav@245	2414	errElem = (ExecutableElement) e;
jaroslav@245	2415	err = "Error method first argument needs to be " + className + " and second Exception";
jaroslav@245	2416	continue;
jaroslav@245	2417	}
jaroslav@245	2418	return true;
jaroslav@245	2419	}
jaroslav@245	2420	if (err == null) {
jaroslav@245	2421	err = "Cannot find " + name + "(" + className + ", Exception) method in this class";
jaroslav@245	2422	}
jaroslav@245	2423	error(err, errElem);
jaroslav@245	2424	return false;
jaroslav@245	2425	}
jaroslav@500	2426
jaroslav@500	2427	private static final class PropInfo {
jaroslav@503	2428	final String name;
jaroslav@503	2429	final String get;
jaroslav@503	2430	final String set;
jaroslav@500	2431	final String p3;
jaroslav@503	2432	final String type;
jaroslav@503	2433	final List<Model> models;
jaroslav@500	2434
jaroslav@500	2435	PropInfo(
jaroslav@500	2436	String p1, String p2, Object p3, String p4, String p5
jaroslav@500	2437	) {
jaroslav@503	2438	this.name = p1;
jaroslav@503	2439	this.get = p2;
jaroslav@503	2440	this.set = p3 == null ? null : p3.toString();
jaroslav@500	2441	this.p3 = p4;
jaroslav@503	2442	this.type = p5;
jaroslav@503	2443	models = new ArrayList<Model>();
jaroslav@500	2444	}
jaroslav@501	2445
jaroslav@501	2446	@Override
jaroslav@501	2447	public int hashCode() {
jaroslav@501	2448	int hash = 7;
jaroslav@503	2449	hash = 19 * hash + (this.name != null ? this.name.hashCode() : 0);
jaroslav@501	2450	return hash;
jaroslav@501	2451	}
jaroslav@501	2452
jaroslav@501	2453	@Override
jaroslav@501	2454	public boolean equals(Object obj) {
jaroslav@501	2455	if (obj == null) {
jaroslav@501	2456	return false;
jaroslav@501	2457	}
jaroslav@501	2458	if (getClass() != obj.getClass()) {
jaroslav@501	2459	return false;
jaroslav@501	2460	}
jaroslav@501	2461	final PropInfo other = (PropInfo) obj;
jaroslav@503	2462	if ((this.name == null) ? (other.name != null) : !this.name.equals(other.name)) {
jaroslav@501	2463	return false;
jaroslav@501	2464	}
jaroslav@501	2465	return true;
jaroslav@501	2466	}
jaroslav@503	2467
jaroslav@503	2468	private void addModel(Model em) {
jaroslav@503	2469	models.add(em);
jaroslav@503	2470	}
jaroslav@501	2471	} // end of PropInfo
jtulach@0	2472	}

author	Jaroslav Tulach <jaroslav.tulach@netbeans.org>
	Thu, 30 Jan 2014 11:36:32 +0100
branch	union
changeset 506	4b66c5141f84
parent 505	501413c8638c
permissions	-rw-r--r--