org.vesalainen.code.Processor Maven / Gradle / Ivy
/*
* Copyright (C) 2014 Timo Vesalainen
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see classnames = new HashSet<>();
private static final AtomicInteger sequence = new AtomicInteger();
private Elements elements;
private Types types;
@Override
public synchronized void init(ProcessingEnvironment pe)
{
super.init(pe);
elements = pe.getElementUtils();
types = pe.getTypeUtils();
}
@Override
public boolean process(Set annotations, RoundEnvironment roundEnv)
{
Messager msg = processingEnv.getMessager();
for (TypeElement te : annotations)
{
for (Element e : roundEnv.getElementsAnnotatedWith(te))
{
TypeElement type = (TypeElement) e;
try
{
msg.printMessage(Diagnostic.Kind.NOTE, "processing", type);
System.err.println("processing "+type);
switch (te.getQualifiedName().toString())
{
case "org.vesalainen.code.BeanProxyClass":
generateBeanProxy(type, processingEnv);
break;
case "org.vesalainen.code.TransactionalSetterClass":
generateTransactionalSetter(type, processingEnv);
break;
case "org.vesalainen.code.PropertyDispatcherClass":
generatePropertyDispatcher(type, processingEnv);
break;
default:
throw new UnsupportedOperationException(te.getQualifiedName().toString()+" not supported");
}
}
catch (Exception ex)
{
ex.printStackTrace();
String m = ex.getMessage();
m = m != null ? m : ex.toString();
msg.printMessage(Diagnostic.Kind.ERROR, m, e);
}
}
}
return true;
}
private void generatePropertyDispatcher(TypeElement cls, ProcessingEnvironment processingEnv) throws IOException
{
PropertyDispatcherClass annotation = cls.getAnnotation(PropertyDispatcherClass.class);
if (annotation == null)
{
throw new IllegalArgumentException("@"+PropertyDispatcherClass.class.getSimpleName()+" missing in cls");
}
TypeElement transactional = elements.getTypeElement(Transactional.class.getCanonicalName());
Filer filer = processingEnv.getFiler();
String value = annotation.value();
JavaFileObject sourceFile = filer.createSourceFile(value);
try (Writer writer = sourceFile.openWriter())
{
CodePrinter mp = new CodePrinter(writer);
int idx = value.lastIndexOf('.');
String classname = value.substring(idx+1);
String pgk = value.substring(0, idx);
List methods = getMethods(cls);
List interfaces = cls.getInterfaces();
mp.println("package "+pgk+";");
mp.println("import org.vesalainen.util.EnumMapList;");
mp.println("import org.vesalainen.util.Transactional;");
mp.println("import org.vesalainen.code.PropertySetter;");
mp.println("import java.util.Arrays;");
mp.println("import javax.annotation.Generated;");
mp.println("@Generated(");
mp.println("\tvalue=\""+Processor.class.getCanonicalName()+"\"");
mp.println("\t, comments=\"Generated for "+cls+"\"");
Date date = new Date();
mp.println("\t, date=\""+date+"\"");
mp.println(")");
CodePrinter cp = mp.createClass(EnumSet.of(PUBLIC), classname, cls);
Set en = new TreeSet<>();
Integer[] sizes = new Integer[]{0, 0, 0, 0, 0, 0, 0, 0, 0};
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
String name = m.getSimpleName().toString();
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
en.add(getEnum(name));
try
{
VariableElement ve = parameters.get(0);
TypeMirror tm = ve.asType();
TypeKind tk = tm.getKind();
JavaType jt = JavaType.valueOf(tk.name());
sizes[jt.ordinal()]++;
}
catch (IllegalArgumentException ex)
{
}
}
}
cp.print("private enum Prop {");
cp.print(", ", en);
cp.println("};");
cp.println("private final EnumMapList observers = new EnumMapList<>(Prop.class);");
cp.println("public "+classname+"()");
cp.println("{");
CodePrinter cons = cp.createSub("}");
cons.print("super(new int[] {");
cons.print(", ", sizes);
cons.println("});");
cons.flush();
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
cp.println("@Override");
CodePrinter cm = cp.createMethod(EnumSet.of(PUBLIC), m);
String name = m.getSimpleName().toString();
String enumname = getEnum(name);
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
VariableElement ve = parameters.get(0);
cm.println("if (observers.containsKey(Prop."+enumname+"))");
cm.println("{");
CodePrinter ifobs = cm.createSub("}");
ifobs.println("set(Prop."+enumname+".ordinal(), "+ve.getSimpleName()+");");
ifobs.flush();
}
else
{
if (
name.equals("commit") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
cm.println("Arrays.fill(ind, 0);");
cm.println("Prop[] values = Prop.values();");
cm.println("for (int ii=0;ii params = ee.getParameters();
TypeMirror rt = ee.getReturnType();
String sn = ee.getSimpleName().toString();
if (
sn.startsWith("set") &&
params.size() == 1 &&
rt.getKind() == VOID
)
{
JavaType jt = JavaType.valueOf(params.get(0).asType().getKind().name());
String aEnum = getEnum(sn);
String prop = getProperty(sn);
cs.println("case "+aEnum+":");
cs.println("{");
CodePrinter csw = cs.createSub("}");
String code = jt.getCode();
int ordinal = jt.ordinal();
csw.println(code+"[] a = ("+code+"[])arr["+ordinal+"];");
csw.println("for (PropertySetter ps : observers.get(Prop."+aEnum+"))");
csw.println("{");
CodePrinter csf = csw.createSub("}");
csf.println("ps.set(\""+prop+"\", a[ind["+ordinal+"]++]);");
csf.flush();
csw.println("break;");
csw.flush();
}
}
cs.println("default:");
CodePrinter csw = cs.createSub("");
csw.println("throw new UnsupportedOperationException(\"should not happen\");");
csw.flush();
cs.flush();
cc.flush();
cm.println("for (Transactional tr : transactionalObservers)");
cm.println("{");
CodePrinter ctr = cm.createSub("}");
ctr.println("tr.commit("+parameters.get(0).getSimpleName()+");");
ctr.flush();
cm.println("clear();");
}
else
{
if (
name.equals("rollback") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
cm.println("clear();");
cm.println("for (Transactional tr : transactionalObservers)");
cm.println("{");
CodePrinter ctr = cm.createSub("}");
ctr.println("tr.rollback("+parameters.get(0).getSimpleName()+");");
ctr.flush();
}
else
{
String sa = String.class.getCanonicalName()+"[]";
if (
name.equals("addObserver") &&
parameters.size() == 2 &&
PropertySetter.class.getCanonicalName().equals(parameters.get(0).asType().toString()) &&
sa.equals(parameters.get(1).asType().toString()) &&
returnType.getKind() == VOID
)
{
Name arg0 = parameters.get(0).getSimpleName();
Name arg1 = parameters.get(1).getSimpleName();
cm.println("super.addObserver("+arg0+");");
cm.println("for (String p : "+arg1+")");
CodePrinter cao = cm.createSub("}");
cao.println("{");
cao.println("Prop pr = Prop.valueOf(Character.toUpperCase(p.charAt(0))+p.substring(1));");
cao.println("observers.add(pr, "+arg0+");");
cao.flush();
}
else
{
cm.println("throw new UnsupportedOperationException(\"not supported.\");");
}
}
}
}
cm.flush();
}
cp.flush();
}
}
private void generateTransactionalSetter(TypeElement cls, ProcessingEnvironment processingEnv) throws IOException
{
TransactionalSetterClass annotation = cls.getAnnotation(TransactionalSetterClass.class);
if (annotation == null)
{
throw new IllegalArgumentException("@"+TransactionalSetterClass.class.getSimpleName()+" missing in cls");
}
TypeElement transactional = elements.getTypeElement(Transactional.class.getCanonicalName());
Filer filer = processingEnv.getFiler();
String value = annotation.value();
JavaFileObject sourceFile = filer.createSourceFile(value);
try (Writer writer = sourceFile.openWriter())
{
CodePrinter mp = new CodePrinter(writer);
int idx = value.lastIndexOf('.');
String classname = value.substring(idx+1);
String pgk = value.substring(0, idx);
List methods = getMethods(cls);
List interfaces = cls.getInterfaces();
if (interfaces.size() != 1)
{
throw new IllegalArgumentException("interface count != 1 (="+interfaces.size()+")");
}
TypeMirror theInterface = interfaces.get(0);
mp.println("package "+pgk+";");
mp.println("import java.util.Arrays;");
mp.println("import javax.annotation.Generated;");
mp.println("@Generated(");
mp.println("\tvalue=\""+Processor.class.getCanonicalName()+"\"");
mp.println("\t, comments=\"Generated for "+cls+"\"");
Date date = new Date();
mp.println("\t, date=\""+date+"\"");
mp.println(")");
CodePrinter cp = mp.createClass(EnumSet.of(PUBLIC), classname, cls);
Set en = new TreeSet<>();
Integer[] sizes = new Integer[]{0, 0, 0, 0, 0, 0, 0, 0, 0};
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
String name = m.getSimpleName().toString();
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
en.add(getEnum(name));
try
{
VariableElement ve = parameters.get(0);
TypeMirror tm = ve.asType();
TypeKind tk = tm.getKind();
JavaType jt = JavaType.valueOf(tk.name());
sizes[jt.ordinal()]++;
}
catch (IllegalArgumentException ex)
{
}
}
}
cp.print("private enum Prop {");
cp.print(", ", en);
cp.println("};");
cp.println("public "+classname+"()");
cp.println("{");
CodePrinter cons = cp.createSub("}");
cons.print("super(new int[] {");
cons.print(", ", sizes);
cons.println("});");
cons.flush();
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
cp.println("@Override");
CodePrinter cm = cp.createMethod(EnumSet.of(PUBLIC), m);
String name = m.getSimpleName().toString();
String enumname = getEnum(name);
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
VariableElement ve = parameters.get(0);
cm.println("set(Prop."+enumname+".ordinal(), "+ve.getSimpleName()+");");
}
else
{
if (
name.equals("commit") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
cm.println("Arrays.fill(ind, 0);");
cm.println("Prop[] values = Prop.values();");
String iname = theInterface.toString();
cm.println(iname+" interf = ("+iname+")intf;");
cm.println("for (int ii=0;ii params = ee.getParameters();
TypeMirror rt = ee.getReturnType();
String sn = ee.getSimpleName().toString();
if (
sn.startsWith("set") &&
params.size() == 1 &&
rt.getKind() == VOID
)
{
JavaType jt = JavaType.valueOf(params.get(0).asType().getKind().name());
String aEnum = getEnum(sn);
cs.println("case "+aEnum+":");
cs.println("{");
CodePrinter csw = cs.createSub("}");
String code = jt.getCode();
int ordinal = jt.ordinal();
csw.println(code+"[] a = ("+code+"[])arr["+ordinal+"];");
csw.println("interf."+sn+"("+cast(params.get(0).asType())+"a[ind["+ordinal+"]++]);");
csw.println("break;");
csw.flush();
}
}
cs.println("default:");
CodePrinter csw = cs.createSub("");
csw.println("throw new UnsupportedOperationException(\"should not happen\");");
csw.flush();
cs.flush();
cc.flush();
cm.println("clear();");
if (types.isAssignable(theInterface, transactional.asType()))
{
cm.println("interf.commit("+parameters.get(0).getSimpleName()+");");
}
}
else
{
if (
name.equals("rollback") &&
parameters.size() == 1 &&
"java.lang.String".equals(parameters.get(0).asType().toString()) &&
returnType.getKind() == VOID
)
{
cm.println("clear();");
if (types.isAssignable(theInterface, transactional.asType()))
{
String iname = theInterface.toString();
cm.println(iname+" interf = ("+iname+")intf;");
cm.println("interf.rollback("+parameters.get(0).getSimpleName()+");");
}
}
else
{
cm.println("throw new UnsupportedOperationException(\"not supported.\");");
}
}
}
cm.flush();
}
cp.flush();
}
}
private void generateBeanProxy(TypeElement cls, ProcessingEnvironment processingEnv) throws IOException
{
BeanProxyClass annotation = cls.getAnnotation(BeanProxyClass.class);
if (annotation == null)
{
throw new IllegalArgumentException("@"+BeanProxyClass.class.getSimpleName()+" missing in cls");
}
Filer filer = processingEnv.getFiler();
String value = annotation.value();
JavaFileObject sourceFile = filer.createSourceFile(value);
try (Writer writer = sourceFile.openWriter())
{
CodePrinter mp = new CodePrinter(writer);
int idx = value.lastIndexOf('.');
String classname = value.substring(idx+1);
String pgk = value.substring(0, idx);
List methods = getMethods(cls);
mp.println("package "+pgk+";");
mp.println("import javax.annotation.Generated;");
mp.println("@Generated(");
mp.println("\tvalue=\""+Processor.class.getCanonicalName()+"\"");
mp.println("\t, comments=\"Generated for "+cls+"\"");
Date date = new Date();
mp.println("\t, date=\""+date+"\"");
mp.println(")");
CodePrinter cp = mp.createClass(EnumSet.of(PUBLIC), classname, cls);
for (ExecutableElement m : methods)
{
List parameters = m.getParameters();
TypeMirror returnType = m.getReturnType();
cp.println("@Override");
CodePrinter cm = cp.createMethod(EnumSet.of(PUBLIC), m);
String name = m.getSimpleName().toString();
String property = getProperty(name);
if (
name.startsWith("set") &&
parameters.size() == 1 &&
returnType.getKind() == VOID
)
{
VariableElement ve = parameters.get(0);
cm.println("set(\""+property+"\", "+ve.getSimpleName()+");");
}
else
{
if (
name.startsWith("get") &&
parameters.isEmpty() &&
returnType.getKind() != VOID
)
{
cm.println("return "+cast(returnType)+getter(returnType.getKind())+"(\""+property+"\");");
}
else
{
cm.println("throw new UnsupportedOperationException(\"not supported.\");");
}
}
cm.flush();
}
cp.flush();
}
}
private List getMethods(TypeElement cls)
{
List list = new ArrayList<>();
for (ExecutableElement m : ElementFilter.methodsIn(elements.getAllMembers(cls)))
{
if (m.getModifiers().contains(ABSTRACT))
{
list.add(m);
}
}
return list;
}
private synchronized String getUniqueClassname(String name)
{
String res = name;
while (classnames.contains(res))
{
res = name+sequence.incrementAndGet();
}
classnames.add(res);
return res;
}
private Set getImports(List methods)
{
Set set = new TreeSet<>();
for (ExecutableElement ee : methods)
{
add(set, ee.getReturnType());
for (TypeMirror tm : ee.getThrownTypes())
{
add(set, tm);
}
for (VariableElement ve : ee.getParameters())
{
add(set, ve.asType());
}
}
return set;
}
private static void add(Set set, TypeMirror tm)
{
if (tm.getKind() == DECLARED)
{
String cl = tm.toString();
int idx = cl.lastIndexOf('.');
if (idx != -1)
{
if ("java.lang".equals(cl.substring(0, idx)))
{
return;
}
}
set.add(cl);
}
}
private String getProperty(String name)
{
return Character.toLowerCase(name.charAt(3))+name.substring(4);
}
private String getEnum(String name)
{
return name.substring(3);
}
private String getter(TypeKind kind)
{
return "get"+getTypename(kind);
}
private String getTypename(TypeKind kind)
{
switch (kind)
{
case ARRAY:
case DECLARED:
return "Object";
case BOOLEAN:
return "Boolean";
case BYTE:
return "Byte";
case CHAR:
return "Char";
case DOUBLE:
return "Double";
case FLOAT:
return "Float";
case INT:
return "Int";
case LONG:
return "Long";
case SHORT:
return "Short";
default:
throw new IllegalArgumentException("not type "+kind);
}
}
private String cast(TypeMirror tm)
{
switch (tm.getKind())
{
case ARRAY:
case DECLARED:
return "("+tm.toString()+")";
default:
return "";
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy